initial commit

custom_components/hacs/websocket/critical.py

@@ -0,0 +1,56 @@
+"""Register info websocket commands."""
+from __future__ import annotations
+
+from typing import Any
+
+from homeassistant.components import websocket_api
+from homeassistant.core import HomeAssistant
+import homeassistant.helpers.config_validation as cv
+import voluptuous as vol
+
+from ..utils.store import async_load_from_store, async_save_to_store
+
+
+@websocket_api.websocket_command(
+    {
+        vol.Required("type"): "hacs/critical/list",
+    }
+)
+@websocket_api.require_admin
+@websocket_api.async_response
+async def hacs_critical_list(
+    hass: HomeAssistant,
+    connection: websocket_api.ActiveConnection,
+    msg: dict[str, Any],
+):
+    """List critical repositories."""
+    connection.send_message(
+        websocket_api.result_message(
+            msg["id"],
+            (await async_load_from_store(hass, "critical") or []),
+        )
+    )
+
+
+@websocket_api.websocket_command(
+    {
+        vol.Required("type"): "hacs/critical/acknowledge",
+        vol.Optional("repository"): cv.string,
+    }
+)
+@websocket_api.require_admin
+@websocket_api.async_response
+async def hacs_critical_acknowledge(
+    hass: HomeAssistant,
+    connection: websocket_api.ActiveConnection,
+    msg: dict[str, Any],
+):
+    """Acknowledge critical repository."""
+    repository = msg["repository"]
+
+    critical = await async_load_from_store(hass, "critical")
+    for repo in critical:
+        if repository == repo["repository"]:
+            repo["acknowledged"] = True
+    await async_save_to_store(hass, "critical", critical)
+    connection.send_message(websocket_api.result_message(msg["id"], critical))