vmware-api-simulator/app/web/routes.py

"""Browser console for exercising the simulator API."""

from __future__ import annotations

from pathlib import Path
from typing import Annotated

from asyncpg import Pool  # type: ignore[import-untyped]
from fastapi import APIRouter, HTTPException, Query, Request
from fastapi.responses import HTMLResponse, JSONResponse

from app.api.registry import HandlerRegistry
from app.config import Settings
from app.contracts.runtime import apply_runtime_contract_locked, contract_store_root
from app.contracts.source import SourceError
from app.db.pool import AsyncpgDatabase
from app.dependencies import get_database
from app.simulation.seed import apply_seed, build_profile, simulation_state_summary
from app.web.assets import compact_console_html, console_html
from app.web.compatibility_catalog import compatibility_payload
from app.version import get_app_version
from app.web.contract_catalog import catalog_payload, list_majors, load_snapshot, method_payload

router = APIRouter(tags=["Simulator"])


@router.get("/", response_class=HTMLResponse, include_in_schema=True)
async def console() -> HTMLResponse:
    """Interactive API console and cluster overview."""

    return HTMLResponse(
        console_html(),
        headers={"Cache-Control": "no-store"},
    )


@router.get("/console.html", response_class=HTMLResponse, include_in_schema=False)
async def compact_console() -> HTMLResponse:
    """Compact session console for quick REST smoke."""

    return HTMLResponse(
        compact_console_html(),
        headers={"Cache-Control": "no-store"},
    )


def _use_vsphere_plane(request: Request) -> bool:
    settings = _settings(request)
    return settings is None or not bool(getattr(settings, "enable_pve_stub", False))


@router.get("/ui/api/versions", include_in_schema=False)
async def ui_versions(request: Request) -> JSONResponse:
    settings = _settings(request)
    runtime_version = _runtime_version(request)
    if _use_vsphere_plane(request):
        from app.vsphere.contracts.catalog import list_vsphere_majors

        payload = list_vsphere_majors(runtime_version=runtime_version)
    else:
        payload = list_majors(runtime_version=runtime_version, settings=settings)
    if isinstance(payload, dict):
        payload = {**payload, "app_version": get_app_version()}
    return JSONResponse(payload)


@router.get("/ui/api/catalog", include_in_schema=False)
async def ui_catalog(
    request: Request,
    major: Annotated[int, Query(ge=6, le=9)],
) -> JSONResponse:
    if _use_vsphere_plane(request):
        from app.vsphere.contracts.catalog import vsphere_catalog_payload

        return JSONResponse(vsphere_catalog_payload(major))
    settings = _settings(request)
    try:
        snapshot = await load_snapshot(major, _store_root(request), settings=settings)
    except SourceError as error:
        raise HTTPException(status_code=503, detail=str(error)) from error
    implemented = getattr(request.app.state, "implemented_methods", None)
    return JSONResponse(
        catalog_payload(snapshot, major, implemented_methods=implemented, settings=settings)
    )


@router.get("/ui/api/method", include_in_schema=False)
async def ui_method(
    request: Request,
    major: Annotated[int, Query(ge=6, le=9)],
    path: Annotated[str, Query(min_length=1)],
    verb: Annotated[str, Query(min_length=1)],
) -> JSONResponse:
    if _use_vsphere_plane(request):
        from app.vsphere.contracts.catalog import vsphere_method_payload

        return JSONResponse(
            vsphere_method_payload(
                major=major,
                path=path,
                verb=verb,
                runtime_version=_runtime_version(request),
            )
        )
    settings = _settings(request)
    try:
        snapshot = await load_snapshot(major, _store_root(request), settings=settings)
    except SourceError as error:
        raise HTTPException(status_code=503, detail=str(error)) from error
    runtime_version = _runtime_version(request)
    implemented = getattr(request.app.state, "implemented_methods", None)
    try:
        payload = method_payload(
            snapshot,
            major=major,
            path=path,
            verb=verb,
            runtime_version=runtime_version,
            implemented_methods=implemented,
        )
    except KeyError as error:
        raise HTTPException(status_code=404, detail=f"unknown contract method: {error}") from error
    return JSONResponse(payload)


@router.get("/ui/api/compatibility", include_in_schema=False)
async def ui_compatibility(
    request: Request,
    major: Annotated[int, Query(ge=6, le=9)],
) -> JSONResponse:
    if _use_vsphere_plane(request):
        from app.vsphere.contracts.compatibility import vsphere_compatibility_payload

        return JSONResponse(
            vsphere_compatibility_payload(
                major,
                runtime_version=_runtime_version(request),
            )
        )
    settings = _settings(request)
    try:
        snapshot = await load_snapshot(major, _store_root(request), settings=settings)
    except SourceError as error:
        raise HTTPException(status_code=503, detail=str(error)) from error
    implemented = getattr(request.app.state, "implemented_methods", None)
    runtime_report = getattr(request.app.state, "compatibility_report", None)
    runtime_version = _runtime_version(request)
    return JSONResponse(
        compatibility_payload(
            snapshot,
            major,
            implemented_methods=implemented,
            runtime_report=runtime_report,
            runtime_version=runtime_version,
            settings=settings,
        )
    )


@router.post("/ui/api/contract/apply", include_in_schema=False)
async def ui_contract_apply(
    request: Request,
    major: Annotated[int, Query(ge=6, le=9)],
) -> JSONResponse:
    """Hot-swap the in-memory runtime contract to a catalog major (memory-only)."""

    if _use_vsphere_plane(request):
        from app.vsphere.contracts.matrix import VERSIONS, catalog_entries_for_major

        meta = VERSIONS.get(major) or VERSIONS[9]
        entries = catalog_entries_for_major(major)
        request.app.state.vsphere_contract_major = major
        request.app.state.runtime_source_version = meta["version"]
        request.app.state.vsphere_implemented_methods = {(e["verb"], e["path"]) for e in entries}
        return JSONResponse(
            {
                "ok": True,
                "major": major,
                "runtime_version": meta["version"],
                "plane": "vsphere-rest",
                "path_count": len({e["path"] for e in entries}),
                "method_count": len(entries),
            }
        )

    settings = _settings(request)
    handlers = getattr(request.app.state, "handlers", None)
    if (
        settings is None
        or settings.contract_snapshot is None
        or not isinstance(handlers, HandlerRegistry)
    ):
        raise HTTPException(status_code=503, detail="runtime contract is not available")
    store_root = _store_root(request)
    try:
        snapshot = await load_snapshot(major, store_root, settings=settings)
    except SourceError as error:
        raise HTTPException(status_code=503, detail=str(error)) from error
    await apply_runtime_contract_locked(
        request.app,
        snapshot,
        handlers=handlers,
        store_root=store_root,
        fallback=settings.contract_fallback,
        settings=settings,
        require_evidence_match=False,
        register_admin=True,
    )
    method_count = sum(len(path.methods) for path in snapshot.paths)
    return JSONResponse(
        {
            "ok": True,
            "major": major,
            "runtime_version": snapshot.source_version,
            "path_count": len(snapshot.paths),
            "method_count": method_count,
        }
    )


@router.get("/ui/api/demo/state", include_in_schema=False)
async def ui_demo_state(request: Request) -> JSONResponse:
    from app.vsphere.seed import vsphere_state_summary

    settings = _settings(request)
    try:
        vsphere = await vsphere_state_summary(get_database(request))
    except Exception as error:
        raise HTTPException(status_code=503, detail=f"database is not ready: {error}") from error
    if settings is not None and getattr(settings, "enable_pve_stub", False):
        try:
            pool = _database_pool(request)
        except Exception as error:
            raise HTTPException(status_code=503, detail=str(error)) from error
        async with pool.acquire() as connection:
            pve = await simulation_state_summary(connection)
        profile = vsphere.get("profile_hint") or "minimal"
        loaded = profile in {"small", "large", "big", "demo-cluster", "demo", "enterprise"}
        return JSONResponse(
            {
                "vsphere": vsphere,
                "proxmox_stub": pve,
                "loaded": loaded,
                "label": f"{vsphere.get('hosts', 0)} hosts · {vsphere.get('vms', 0)} VMs",
                "profile": profile,
            }
        )
    profile = vsphere.get("profile_hint") or "minimal"
    loaded = profile in {"small", "large", "big", "demo-cluster", "demo", "enterprise"}
    return JSONResponse(
        {
            "vsphere": vsphere,
            "loaded": loaded,
            "label": f"{vsphere.get('hosts', 0)} hosts · {vsphere.get('vms', 0)} VMs",
            "profile": profile,
        }
    )


@router.post("/ui/api/demo/load", include_in_schema=False)
async def ui_demo_load(request: Request) -> JSONResponse:
    from app.vsphere.seed import seed_vsphere_inventory

    settings = _settings(request)
    summary: dict = {}
    size_raw = request.query_params.get("size") or request.query_params.get("profile")
    if not size_raw:
        try:
            body = await request.json()
        except Exception:
            body = {}
        if isinstance(body, dict):
            size_raw = body.get("size") or body.get("profile")
    profile_name = str(size_raw or "large").strip().lower() or "large"
    if profile_name in {"demo-cluster", "demo", "enterprise"}:
        profile_name = "big"
    if profile_name not in {"small", "large", "big"}:
        raise HTTPException(
            status_code=400,
            detail=f"unknown size {profile_name!r}; sizes: small, large, big",
        )
    if settings is not None and getattr(settings, "enable_pve_stub", False):
        pool = _database_pool(request)
        pve_name = "demo-cluster" if profile_name == "big" else profile_name
        try:
            profile = build_profile(pve_name)
        except Exception:
            profile = build_profile("demo-cluster")
        async with pool.acquire() as connection:
            await apply_seed(connection, profile)
            summary = await simulation_state_summary(connection)
    try:
        vsphere = await seed_vsphere_inventory(
            get_database(request), force=True, profile=profile_name
        )
    except Exception as error:
        raise HTTPException(status_code=503, detail=f"database is not ready: {error}") from error
    return JSONResponse(
        {"ok": True, "profile": profile_name, "summary": summary, "vsphere": vsphere}
    )


@router.post("/ui/api/demo/unload", include_in_schema=False)
async def ui_demo_unload(request: Request) -> JSONResponse:
    """Reset seed, wiping API-created state first."""
    from app.vsphere.seed import seed_vsphere_inventory

    settings = _settings(request)
    summary: dict = {}
    profile_name = "minimal"
    try:
        if settings is not None and getattr(settings, "enable_pve_stub", False):
            pool = _database_pool(request)
            profile = build_profile("minimal")
            async with pool.acquire() as connection:
                await apply_seed(connection, profile)
                summary = await simulation_state_summary(connection)
            profile_name = profile.name
        vsphere = await seed_vsphere_inventory(get_database(request), force=True, profile="minimal")
    except HTTPException:
        raise
    except Exception as error:
        raise HTTPException(
            status_code=503, detail=f"failed to remove demo data: {error}"
        ) from error
    return JSONResponse(
        {"ok": True, "profile": profile_name, "summary": summary, "vsphere": vsphere}
    )


@router.post("/ui/api/vsphere/seed", include_in_schema=False)
async def ui_vsphere_seed(request: Request) -> JSONResponse:
    """(Re)seed native vSphere inventory used by /api and /sdk."""

    from app.vsphere.seed import seed_vsphere_inventory

    profile = request.query_params.get("profile") or "large"
    try:
        result = await seed_vsphere_inventory(get_database(request), force=True, profile=profile)
    except Exception as error:
        raise HTTPException(status_code=500, detail=str(error)) from error
    return JSONResponse({"ok": True, **result})


def _database_pool(request: Request) -> Pool:
    database = get_database(request)
    if not isinstance(database, AsyncpgDatabase):
        raise HTTPException(status_code=503, detail="database is not available")
    return database.pool


def _settings(request: Request) -> Settings | None:
    return getattr(request.app.state, "settings", None)


def _runtime_version(request: Request) -> str | None:
    active = getattr(request.app.state, "runtime_source_version", None)
    if isinstance(active, str) and active:
        return active
    settings = _settings(request)
    if settings is None or settings.contract_snapshot is None:
        return None
    from app.contracts.model import Snapshot

    snapshot = Snapshot.model_validate_json(settings.contract_snapshot.read_bytes())
    return snapshot.source_version


def _store_root(request: Request) -> Path:
    stored = getattr(request.app.state, "contract_store_root", None)
    if isinstance(stored, Path):
        return stored
    settings = _settings(request)
    if settings is not None:
        return contract_store_root(settings)
    return Path("contracts")