397 lines
14 KiB
Python
397 lines
14 KiB
Python
"""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.openstack.demo_cloud import openstack_demo_summary, seed_openstack_demo
|
|
from app.openstack.seed import seed_openstack
|
|
from app.web.assets import console_html
|
|
from app.web.compatibility_catalog import compatibility_payload
|
|
from app.web.contract_catalog import catalog_payload, list_majors, load_snapshot, method_payload
|
|
|
|
router = APIRouter(tags=["Simulator"])
|
|
|
|
|
|
@router.get("/console", 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("/ui/api/versions", include_in_schema=False)
|
|
async def ui_versions(request: Request) -> JSONResponse:
|
|
from app.web.openstack_catalog import openstack_series_majors
|
|
|
|
runtime_version = _runtime_version(request)
|
|
# Prefer OpenStack contract packs when present.
|
|
try:
|
|
return JSONResponse(openstack_series_majors(runtime_version))
|
|
except Exception:
|
|
settings = _settings(request)
|
|
return JSONResponse(list_majors(runtime_version=runtime_version, settings=settings))
|
|
|
|
|
|
@router.get("/ui/api/catalog", include_in_schema=False)
|
|
async def ui_catalog(
|
|
request: Request,
|
|
major: Annotated[int, Query(ge=6, le=9)],
|
|
) -> JSONResponse:
|
|
from app.web.openstack_catalog import openstack_catalog_payload
|
|
|
|
try:
|
|
return JSONResponse(openstack_catalog_payload(major))
|
|
except FileNotFoundError:
|
|
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:
|
|
from app.web.openstack_catalog import openstack_method_payload
|
|
|
|
runtime_version = _runtime_version(request)
|
|
try:
|
|
return JSONResponse(
|
|
openstack_method_payload(
|
|
major=major,
|
|
path=path,
|
|
verb=verb,
|
|
runtime_version=runtime_version,
|
|
)
|
|
)
|
|
except (FileNotFoundError, KeyError):
|
|
pass
|
|
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)
|
|
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:
|
|
from app.web.openstack_catalog import openstack_compatibility_payload
|
|
|
|
runtime_version = _runtime_version(request)
|
|
# Prefer OpenStack pack coverage (Yoga→Dalmatian).
|
|
try:
|
|
return JSONResponse(
|
|
openstack_compatibility_payload(
|
|
major,
|
|
runtime_version=runtime_version,
|
|
schema_ops_mounted=getattr(request.app.state, "openstack_schema_ops", None),
|
|
)
|
|
)
|
|
except (FileNotFoundError, KeyError, ValueError):
|
|
pass
|
|
|
|
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)
|
|
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).
|
|
|
|
Prefer OpenStack series packs when present; fall back to legacy Proxmox
|
|
snapshot swap when ``CONTRACT_SNAPSHOT`` / handler registry are configured.
|
|
"""
|
|
|
|
from app.openstack.contract_loader import series_for_major
|
|
from app.openstack.schema_engine import remount_schema_services
|
|
|
|
# OpenStack pack path (Yoga=6 … Dalmatian=9).
|
|
try:
|
|
series = series_for_major(major)
|
|
except Exception:
|
|
series = None
|
|
if series:
|
|
async with request.app.state.contract_swap_lock:
|
|
try:
|
|
summary = remount_schema_services(request.app, series)
|
|
except FileNotFoundError as exc:
|
|
raise HTTPException(status_code=404, detail=str(exc)) from exc
|
|
request.app.state.openstack_schema_ops = summary.get(
|
|
"routes_mounted", summary.get("operation_count", 0)
|
|
)
|
|
request.app.state.runtime_version = f"openstack-{series}"
|
|
return JSONResponse(
|
|
{
|
|
"ok": True,
|
|
"major": major,
|
|
"series": series,
|
|
"runtime_version": f"openstack-{series}",
|
|
"path_count": summary.get("service_count"),
|
|
"method_count": summary.get("routes_mounted", summary.get("operation_count")),
|
|
**{k: v for k, v in summary.items() if k not in {"ok"}},
|
|
}
|
|
)
|
|
|
|
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:
|
|
pool = _database_pool(request)
|
|
async with pool.acquire() as connection:
|
|
return JSONResponse(await openstack_demo_summary(connection))
|
|
|
|
|
|
@router.get("/ui/api/openstack/contracts", include_in_schema=False)
|
|
async def ui_openstack_contracts(request: Request) -> JSONResponse:
|
|
"""Active OpenStack API contract pack + available series."""
|
|
|
|
from app.openstack.contract_loader import ensure_loaded, get_runtime, list_series
|
|
|
|
ensure_loaded("dalmatian")
|
|
runtime = get_runtime()
|
|
return JSONResponse(
|
|
{
|
|
"active": runtime.summary(),
|
|
"available": list_series(),
|
|
"schema_ops_mounted": getattr(request.app.state, "openstack_schema_ops", 0),
|
|
}
|
|
)
|
|
|
|
|
|
@router.post("/ui/api/openstack/contracts/activate", include_in_schema=False)
|
|
async def ui_openstack_contracts_activate(request: Request) -> JSONResponse:
|
|
"""Hot-swap the active OpenStack series contract pack."""
|
|
|
|
from app.openstack.schema_engine import remount_schema_services
|
|
|
|
payload = await request.json()
|
|
series = str(payload.get("series") or "").lower().strip()
|
|
if not series:
|
|
raise HTTPException(status_code=400, detail="series is required")
|
|
async with request.app.state.contract_swap_lock:
|
|
try:
|
|
summary = remount_schema_services(request.app, series)
|
|
except FileNotFoundError as exc:
|
|
raise HTTPException(status_code=404, detail=str(exc)) from exc
|
|
request.app.state.openstack_schema_ops = summary.get(
|
|
"routes_mounted", summary.get("operation_count", 0)
|
|
)
|
|
request.app.state.runtime_version = f"openstack-{series}"
|
|
return JSONResponse({"ok": True, "runtime_version": f"openstack-{series}", **summary})
|
|
|
|
|
|
@router.post("/ui/api/openstack/microversions", include_in_schema=False)
|
|
async def ui_openstack_microversions(request: Request) -> JSONResponse:
|
|
"""Set or clear a per-service microversion override for the lab."""
|
|
|
|
from app.openstack.contract_loader import ensure_loaded, get_runtime
|
|
|
|
ensure_loaded("dalmatian")
|
|
payload = await request.json()
|
|
service = str(payload.get("service") or "").lower().strip()
|
|
version = payload.get("version")
|
|
if not service:
|
|
raise HTTPException(status_code=400, detail="service is required")
|
|
runtime = get_runtime()
|
|
if service not in runtime.packs:
|
|
raise HTTPException(status_code=404, detail=f"unknown service {service}")
|
|
runtime.set_microversion(service, None if version in (None, "", "default") else str(version))
|
|
return JSONResponse({"ok": True, "active": runtime.summary()})
|
|
|
|
|
|
@router.post("/ui/api/demo/load", include_in_schema=False)
|
|
async def ui_demo_load(request: Request) -> JSONResponse:
|
|
"""Load synthetic OpenStack cloud (size: small | large | big)."""
|
|
|
|
from app.openstack.demo_cloud import DEMO_SIZE_DEFAULT, resolve_demo_size
|
|
|
|
size = DEMO_SIZE_DEFAULT
|
|
try:
|
|
payload = await request.json()
|
|
except Exception:
|
|
payload = {}
|
|
if isinstance(payload, dict) and payload.get("size"):
|
|
size = str(payload["size"])
|
|
elif request.query_params.get("size"):
|
|
size = str(request.query_params.get("size"))
|
|
try:
|
|
resolve_demo_size(size)
|
|
except ValueError as error:
|
|
raise HTTPException(status_code=400, detail=str(error)) from error
|
|
|
|
pool = _database_pool(request)
|
|
try:
|
|
async with pool.acquire() as connection:
|
|
async with connection.transaction():
|
|
result = await seed_openstack_demo(connection, size=size)
|
|
summary = await openstack_demo_summary(connection)
|
|
except Exception as error:
|
|
raise HTTPException(
|
|
status_code=500, detail=f"failed to load OpenStack demo cloud: {error}"
|
|
) from error
|
|
return JSONResponse(
|
|
{
|
|
"ok": True,
|
|
"profile": result["profile"],
|
|
"size": result.get("size"),
|
|
"summary": summary,
|
|
"seed": result,
|
|
}
|
|
)
|
|
|
|
|
|
@router.post("/ui/api/demo/unload", include_in_schema=False)
|
|
async def ui_demo_unload(request: Request) -> JSONResponse:
|
|
"""Reset OpenStack state to the minimal lab seed."""
|
|
|
|
pool = _database_pool(request)
|
|
try:
|
|
async with pool.acquire() as connection:
|
|
async with connection.transaction():
|
|
from app.openstack.demo_cloud import clear_openstack_state
|
|
|
|
await clear_openstack_state(connection)
|
|
result = await seed_openstack(connection)
|
|
summary = await openstack_demo_summary(connection)
|
|
except Exception as error:
|
|
raise HTTPException(
|
|
status_code=500, detail=f"failed to reset to minimal cluster: {error}"
|
|
) from error
|
|
return JSONResponse(
|
|
{"ok": True, "profile": result.get("profile", "minimal"), "summary": summary}
|
|
)
|
|
|
|
|
|
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:
|
|
for attr in ("runtime_source_version", "runtime_version"):
|
|
active = getattr(request.app.state, attr, None)
|
|
if isinstance(active, str) and active:
|
|
return active
|
|
# Prefer active OpenStack pack series when Proxmox snapshot is absent.
|
|
try:
|
|
from app.openstack.contract_loader import get_runtime
|
|
|
|
runtime = get_runtime()
|
|
if runtime.series:
|
|
return f"openstack-{runtime.series}"
|
|
except Exception:
|
|
pass
|
|
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")
|