init

custom_components/openid/http_helper.py

@@ -0,0 +1,84 @@
+"""Patch the built-in /auth/authorize and /auth/login_flow pages to load our JS helper."""
+
+from http import HTTPStatus
+import json
+import logging
+
+from aiohttp.web import Request, Response
+
+from homeassistant.core import HomeAssistant
+
+from .const import CONF_BLOCK_LOGIN, CONF_OPENID_TEXT, DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+
+def override_authorize_login_flow(hass: HomeAssistant) -> None:
+    """Patch the build-in /auth/login_flow page to not return any actual login data."""
+
+    _original_post_function = None
+
+    async def post(request: Request) -> Response:
+        if not hass.data[DOMAIN].get(CONF_BLOCK_LOGIN, False):
+            content = json.loads((await _original_post_function(request)).text)
+        else:
+            content = {
+                "type": "form",
+                "flow_id": None,
+                "handler": [None],
+                "data_schema": [],
+                "errors": {},
+                "description_placeholders": None,
+                "last_step": None,
+                "preview": None,
+                "step_id": "init",
+            }
+
+        content[CONF_BLOCK_LOGIN] = hass.data[DOMAIN].get(CONF_BLOCK_LOGIN, False)
+        content[CONF_OPENID_TEXT] = hass.data[DOMAIN].get(
+            CONF_OPENID_TEXT, "OpenID / OAuth2 Authentication"
+        )
+
+        return Response(
+            status=HTTPStatus.OK,
+            body=json.dumps(content),
+            content_type="application/json",
+        )
+
+    # Swap out the existing GET handler on /auth/authorize
+    for resource in hass.http.app.router._resources:  # noqa: SLF001
+        if getattr(resource, "canonical", None) == "/auth/login_flow":
+            post_handler = resource._routes.get("POST")  # noqa: SLF001
+            # Replace the underlying coroutine fn.
+            _original_post_function = post_handler._handler  # noqa: SLF001
+            post_handler._handler = post  # noqa: SLF001
+            # Reset the routes map to ensure only our GET exists.
+            resource._routes = {"POST": post_handler}  # noqa: SLF001
+            _LOGGER.debug("Overrode /auth/login_flow route")
+            break
+
+
+def override_authorize_route(hass: HomeAssistant) -> None:
+    """Patch the built-in /auth/authorize page to load our JS helper."""
+
+    async def get(request: Request) -> Response:
+        content = hass.data[DOMAIN]["authorize_template"]
+
+        # Inject script before </head>
+        content = content.replace(
+            "</head>",
+            '<script src="/openid/authorize.js"></script></head>',
+        )
+
+        return Response(status=HTTPStatus.OK, body=content, content_type="text/html")
+
+    # Swap out the existing GET handler on /auth/authorize
+    for resource in hass.http.app.router._resources:  # noqa: SLF001
+        if getattr(resource, "canonical", None) == "/auth/authorize":
+            get_handler = resource._routes.get("GET")  # noqa: SLF001
+            # Replace the underlying coroutine fn.
+            get_handler._handler = get  # noqa: SLF001
+            # Reset the routes map to ensure only our GET exists.
+            resource._routes = {"GET": get_handler}  # noqa: SLF001
+            _LOGGER.debug("Overrode /auth/authorize route – custom JS injected")
+            break