codee-agent 0.1.0__py3-none-any.whl

codee/admin_api.py

@@ -0,0 +1,61 @@
+"""Plain HTTP endpoints served alongside the Reflex pages.
+
+OAuth callbacks can't be Reflex pages: the authorization code would have to
+travel through the browser and into a page's ``on_load`` before anything could
+be done with it. Handling them as backend routes keeps the code and the client
+secret server-side, and lets the browser be bounced straight back to /settings.
+
+Reflex mounts its own ASGI app underneath this Starlette app (see
+``api_transformer`` in :mod:`codee.admin`), so these routes are matched first
+and everything else falls through to the UI.
+"""
+from urllib.parse import urlencode
+
+from starlette.applications import Starlette
+from starlette.requests import Request
+from starlette.responses import RedirectResponse
+from starlette.routing import Route
+
+from codee.admin_service import AdminService
+from codee_tasks_azure_devops.oauth import CALLBACK_PATH
+
+SETTINGS_PATH = "/settings"
+
+# Its own service instance: this route runs outside any Reflex session, and
+# every handler re-reads settings from disk anyway.
+_service = AdminService()
+
+
+def _back_to_settings(connected: bool, message: str) -> RedirectResponse:
+    """Return to the settings page carrying the outcome for the UI to toast."""
+    query = urlencode(
+        {"azure": "connected" if connected else "error", "message": message})
+    # 303: the callback is a GET, and the browser should land on /settings as a
+    # fresh GET rather than replaying anything.
+    return RedirectResponse(f"{SETTINGS_PATH}?{query}", status_code=303)
+
+
+def azure_devops_callback(request: Request) -> RedirectResponse:
+    """Entra ID redirects here with ?code&state, or with ?error on refusal.
+
+    Defined as a sync handler on purpose: the token exchange is a blocking HTTP
+    call, so Starlette runs it in a worker thread instead of stalling the event
+    loop that serves the rest of the admin UI.
+    """
+    params = request.query_params
+    if params.get("error"):
+        description = params.get("error_description") or params["error"]
+        return _back_to_settings(False, description.splitlines()[0])
+
+    code, state = params.get("code"), params.get("state")
+    if not code or not state:
+        return _back_to_settings(
+            False, "Azure DevOps did not return an authorization code.")
+
+    connected, message = _service.complete_azure_authorization(code, state)
+    return _back_to_settings(connected, message)
+
+
+api_app = Starlette(routes=[
+    Route(CALLBACK_PATH, azure_devops_callback, methods=["GET"]),
+])

codee/admin_cli.py

@@ -0,0 +1,86 @@
+"""Console-script launcher for the Reflex admin UI."""
+import argparse
+import os
+import sys
+from pathlib import Path
+
+from codee.admin_service import DEFAULT_ADMIN_PORT
+from codee_main_context.context import data_dir, project_root
+from codee_main_context.logging import (
+    configure_logging, enable_debug, get_logger, verbose_debug_enabled)
+
+
+log = get_logger(__name__)
+
+
+RXCONFIG = """import reflex as rx
+
+config = rx.Config(
+    app_name="codee_admin",
+    telemetry_enabled=False,
+    show_built_with_reflex=False,
+    plugins=[
+        rx.plugins.RadixThemesPlugin(
+            theme=rx.theme(
+                accent_color="green",
+                gray_color="sage",
+                radius="small",
+                appearance="inherit",
+            )
+        ),
+    ],
+    disable_plugins=[rx.plugins.SitemapPlugin],
+)
+"""
+
+
+def _runtime_directory(root: Path) -> Path:
+    runtime = data_dir(root) / "reflex"
+    runtime.mkdir(parents=True, exist_ok=True)
+    config_path = runtime / "rxconfig.py"
+    if not config_path.exists() or config_path.read_text() != RXCONFIG:
+        config_path.write_text(RXCONFIG)
+    return runtime
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(prog="codee-admin")
+    parser.add_argument("--port", "--server.port",
+                        dest="port", type=int, default=DEFAULT_ADMIN_PORT)
+    parser.add_argument("-d", "--debug", action="store_true",
+                        help="log debug messages")
+    arguments, reflex_arguments = parser.parse_known_args()
+
+    if arguments.debug:
+        enable_debug()
+    configure_logging()
+
+    root = project_root().resolve()
+    codee_data = data_dir(root).resolve()
+    os.environ["CODEE_PROJECT_ROOT"] = str(root)
+    os.environ["CODEE_DATA_DIR"] = str(codee_data)
+    os.environ["REFLEX_API_URL"] = f"http://127.0.0.1:{arguments.port}"
+    os.chdir(_runtime_directory(root))
+
+    from reflex.reflex import cli
+
+    sys.argv = [
+        "reflex",
+        "run",
+        "--env",
+        "prod",
+        "--single-port",
+        "--frontend-port",
+        str(arguments.port),
+        # Reflex keeps its own logger; only CODEE_DEBUG=all turns it up, since
+        # its debug output is far noisier than ours.
+        *(["--loglevel", "debug"] if verbose_debug_enabled() else []),
+        *reflex_arguments,
+    ]
+    log.debug("root=%s data=%s reflex argv=%s", root, codee_data, sys.argv)
+    cli()
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())