mirror of
https://github.com/zvx-echo6/central.git
synced 2026-05-21 18:14:44 +02:00
feat(gui): implement first-run setup wizard (1b-8)
Add a 5-step setup wizard that replaces the single-step /setup: 1. Create Operator - create initial operator account 2. System Settings - configure map tile URL and attribution 3. API Keys - optionally add API keys for adapters 4. Configure Adapters - enable/disable adapters with region picker 5. Finish Setup - review and complete setup Key changes: - Update middleware to handle wizard URL structure and step routing - Add wizard routes for each step with proper auth checks - Create new templates using base_wizard.html for consistent styling - Add audit events for system.update and setup.complete - Update tests for new middleware behavior Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
parent
96ec88883c
commit
62116ca6a4
12 changed files with 1840 additions and 27 deletions
|
|
@ -14,6 +14,8 @@ STREAM_UPDATE = "stream.update"
|
||||||
API_KEY_CREATE = "api_key.create"
|
API_KEY_CREATE = "api_key.create"
|
||||||
API_KEY_ROTATE = "api_key.rotate"
|
API_KEY_ROTATE = "api_key.rotate"
|
||||||
API_KEY_DELETE = "api_key.delete"
|
API_KEY_DELETE = "api_key.delete"
|
||||||
|
SYSTEM_UPDATE = "system.update"
|
||||||
|
SETUP_COMPLETE = "setup.complete"
|
||||||
|
|
||||||
|
|
||||||
async def write_audit(
|
async def write_audit(
|
||||||
|
|
|
||||||
|
|
@ -12,11 +12,10 @@ from central.gui.db import get_pool
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
# Paths that don't require setup to be complete
|
# Paths that don't require setup to be complete
|
||||||
SETUP_EXEMPT_PATHS = {"/setup", "/health"}
|
SETUP_EXEMPT_PREFIXES = ("/static/", "/setup")
|
||||||
SETUP_EXEMPT_PREFIXES = ("/static/",)
|
|
||||||
|
|
||||||
# Paths that don't require authentication
|
# Paths that don't require authentication
|
||||||
AUTH_EXEMPT_PATHS = {"/setup", "/login", "/health"}
|
AUTH_EXEMPT_PATHS = {"/setup/operator", "/login", "/health"}
|
||||||
AUTH_EXEMPT_PREFIXES = ("/static/",)
|
AUTH_EXEMPT_PREFIXES = ("/static/",)
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -30,6 +29,35 @@ def _is_exempt(path: str, exempt_paths: set, exempt_prefixes: tuple) -> bool:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def _get_wizard_redirect_step(conn) -> str:
|
||||||
|
"""Determine which wizard step to redirect to based on DB state."""
|
||||||
|
# Check if any operators exist
|
||||||
|
op_count = await conn.fetchval("SELECT COUNT(*) FROM config.operators")
|
||||||
|
if op_count == 0:
|
||||||
|
return "/setup/operator"
|
||||||
|
|
||||||
|
# Check if system settings have been configured (map_tile_url not default)
|
||||||
|
sys_row = await conn.fetchrow(
|
||||||
|
"SELECT map_tile_url FROM config.system WHERE id = true"
|
||||||
|
)
|
||||||
|
default_tile = "https://tile.openstreetmap.org/{z}/{x}/{y}.png"
|
||||||
|
if sys_row is None or sys_row["map_tile_url"] == default_tile:
|
||||||
|
return "/setup/system"
|
||||||
|
|
||||||
|
# Keys step is optional, so check adapters have been reviewed
|
||||||
|
# We consider adapters reviewed if any adapter has a non-null updated_at
|
||||||
|
# (meaning it was explicitly saved during setup)
|
||||||
|
adapters_touched = await conn.fetchval(
|
||||||
|
"SELECT COUNT(*) FROM config.adapters WHERE updated_at IS NOT NULL"
|
||||||
|
)
|
||||||
|
if adapters_touched == 0:
|
||||||
|
# Go to keys first, then adapters
|
||||||
|
return "/setup/keys"
|
||||||
|
|
||||||
|
# All steps done, go to finish
|
||||||
|
return "/setup/finish"
|
||||||
|
|
||||||
|
|
||||||
class SetupGateMiddleware(BaseHTTPMiddleware):
|
class SetupGateMiddleware(BaseHTTPMiddleware):
|
||||||
"""Redirect to /setup if setup is not complete."""
|
"""Redirect to /setup if setup is not complete."""
|
||||||
|
|
||||||
|
|
@ -55,12 +83,30 @@ class SetupGateMiddleware(BaseHTTPMiddleware):
|
||||||
return await call_next(request)
|
return await call_next(request)
|
||||||
|
|
||||||
if not setup_complete:
|
if not setup_complete:
|
||||||
# Setup not complete - only allow exempt paths
|
# Setup not complete - only allow setup paths and static/health
|
||||||
if not _is_exempt(path, SETUP_EXEMPT_PATHS, SETUP_EXEMPT_PREFIXES):
|
if path.startswith("/setup"):
|
||||||
|
# Allow all /setup/* paths (handler will enforce auth)
|
||||||
|
# But /setup with no subpath should redirect to appropriate step
|
||||||
|
if path == "/setup" or path == "/setup/":
|
||||||
|
try:
|
||||||
|
async with pool.acquire() as conn:
|
||||||
|
redirect_step = await _get_wizard_redirect_step(conn)
|
||||||
|
return RedirectResponse(url=redirect_step, status_code=302)
|
||||||
|
except Exception:
|
||||||
|
logger.warning("Failed to determine wizard step", exc_info=True)
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
return await call_next(request)
|
||||||
|
elif path == "/health" or path.startswith("/static/"):
|
||||||
|
return await call_next(request)
|
||||||
|
elif path == "/login":
|
||||||
|
# During setup, login redirects to /setup
|
||||||
|
return RedirectResponse(url="/setup", status_code=302)
|
||||||
|
else:
|
||||||
|
# All other paths redirect to /setup
|
||||||
return RedirectResponse(url="/setup", status_code=302)
|
return RedirectResponse(url="/setup", status_code=302)
|
||||||
else:
|
else:
|
||||||
# Setup complete - redirect /setup to /
|
# Setup complete - redirect /setup* to /
|
||||||
if path == "/setup":
|
if path.startswith("/setup"):
|
||||||
return RedirectResponse(url="/", status_code=302)
|
return RedirectResponse(url="/", status_code=302)
|
||||||
|
|
||||||
return await call_next(request)
|
return await call_next(request)
|
||||||
|
|
|
||||||
|
|
@ -28,7 +28,9 @@ from central.gui.audit import (
|
||||||
AUTH_LOGOUT,
|
AUTH_LOGOUT,
|
||||||
AUTH_PASSWORD_CHANGE,
|
AUTH_PASSWORD_CHANGE,
|
||||||
OPERATOR_CREATE,
|
OPERATOR_CREATE,
|
||||||
|
SETUP_COMPLETE,
|
||||||
STREAM_UPDATE,
|
STREAM_UPDATE,
|
||||||
|
SYSTEM_UPDATE,
|
||||||
write_audit,
|
write_audit,
|
||||||
)
|
)
|
||||||
from central.gui.db import get_pool
|
from central.gui.db import get_pool
|
||||||
|
|
@ -252,32 +254,37 @@ async def dashboard_polls(request: Request) -> HTMLResponse:
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@router.get("/setup", response_class=HTMLResponse)
|
# =============================================================================
|
||||||
async def setup_form(
|
# Setup Wizard routes
|
||||||
|
# =============================================================================
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/setup/operator", response_class=HTMLResponse)
|
||||||
|
async def setup_operator_form(
|
||||||
request: Request,
|
request: Request,
|
||||||
csrf_protect: CsrfProtect = Depends(),
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
) -> HTMLResponse:
|
) -> HTMLResponse:
|
||||||
"""Render the setup form."""
|
"""Render the setup operator form (step 1)."""
|
||||||
templates = _get_templates()
|
templates = _get_templates()
|
||||||
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
response = templates.TemplateResponse(
|
response = templates.TemplateResponse(
|
||||||
request=request,
|
request=request,
|
||||||
name="setup.html",
|
name="setup_operator.html",
|
||||||
context={"csrf_token": csrf_token, "error": None},
|
context={"csrf_token": csrf_token, "error": None, "form_data": None},
|
||||||
)
|
)
|
||||||
csrf_protect.set_csrf_cookie(signed_token, response)
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
return response
|
return response
|
||||||
|
|
||||||
|
|
||||||
@router.post("/setup")
|
@router.post("/setup/operator")
|
||||||
async def setup_submit(
|
async def setup_operator_submit(
|
||||||
request: Request,
|
request: Request,
|
||||||
username: str = Form(...),
|
username: str = Form(...),
|
||||||
password: str = Form(...),
|
password: str = Form(...),
|
||||||
confirm_password: str = Form(...),
|
confirm_password: str = Form(...),
|
||||||
csrf_protect: CsrfProtect = Depends(),
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
) -> Response:
|
) -> Response:
|
||||||
"""Process the setup form."""
|
"""Process the setup operator form (step 1)."""
|
||||||
templates = _get_templates()
|
templates = _get_templates()
|
||||||
pool = get_pool()
|
pool = get_pool()
|
||||||
|
|
||||||
|
|
@ -298,8 +305,12 @@ async def setup_submit(
|
||||||
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
response = templates.TemplateResponse(
|
response = templates.TemplateResponse(
|
||||||
request=request,
|
request=request,
|
||||||
name="setup.html",
|
name="setup_operator.html",
|
||||||
context={"csrf_token": csrf_token, "error": error},
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"error": error,
|
||||||
|
"form_data": {"username": username},
|
||||||
|
},
|
||||||
status_code=200,
|
status_code=200,
|
||||||
)
|
)
|
||||||
csrf_protect.set_csrf_cookie(signed_token, response)
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
|
@ -336,15 +347,651 @@ async def setup_submit(
|
||||||
# Create session
|
# Create session
|
||||||
token, expires_at = await create_session(conn, operator_id, lifetime_days)
|
token, expires_at = await create_session(conn, operator_id, lifetime_days)
|
||||||
|
|
||||||
|
# Redirect to next step with session cookie
|
||||||
|
response = RedirectResponse(url="/setup/system", status_code=302)
|
||||||
|
_set_session_cookie(response, token, lifetime_days * 86400)
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/setup/system", response_class=HTMLResponse)
|
||||||
|
async def setup_system_form(
|
||||||
|
request: Request,
|
||||||
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
|
) -> HTMLResponse:
|
||||||
|
"""Render the system settings form (step 2)."""
|
||||||
|
# Require authentication for this step
|
||||||
|
operator = getattr(request.state, "operator", None)
|
||||||
|
if operator is None:
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
|
||||||
|
templates = _get_templates()
|
||||||
|
pool = get_pool()
|
||||||
|
|
||||||
|
async with pool.acquire() as conn:
|
||||||
|
row = await conn.fetchrow(
|
||||||
|
"SELECT map_tile_url, map_attribution FROM config.system WHERE id = true"
|
||||||
|
)
|
||||||
|
system = {
|
||||||
|
"map_tile_url": row["map_tile_url"] if row else "https://tile.openstreetmap.org/{z}/{x}/{y}.png",
|
||||||
|
"map_attribution": row["map_attribution"] if row else "© OpenStreetMap contributors",
|
||||||
|
}
|
||||||
|
|
||||||
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
|
response = templates.TemplateResponse(
|
||||||
|
request=request,
|
||||||
|
name="setup_system.html",
|
||||||
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"error": None,
|
||||||
|
"errors": None,
|
||||||
|
"form_data": None,
|
||||||
|
"system": system,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/setup/system")
|
||||||
|
async def setup_system_submit(
|
||||||
|
request: Request,
|
||||||
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
|
) -> Response:
|
||||||
|
"""Process the system settings form (step 2)."""
|
||||||
|
# Require authentication for this step
|
||||||
|
operator = getattr(request.state, "operator", None)
|
||||||
|
if operator is None:
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
|
||||||
|
templates = _get_templates()
|
||||||
|
pool = get_pool()
|
||||||
|
|
||||||
|
await csrf_protect.validate_csrf(request)
|
||||||
|
|
||||||
|
form = await request.form()
|
||||||
|
map_tile_url = form.get("map_tile_url", "").strip()
|
||||||
|
map_attribution = form.get("map_attribution", "").strip()
|
||||||
|
|
||||||
|
form_data = {
|
||||||
|
"map_tile_url": map_tile_url,
|
||||||
|
"map_attribution": map_attribution,
|
||||||
|
}
|
||||||
|
|
||||||
|
errors: dict[str, str] = {}
|
||||||
|
|
||||||
|
# Validate map_tile_url
|
||||||
|
if not map_tile_url:
|
||||||
|
errors["map_tile_url"] = "Map tile URL is required"
|
||||||
|
elif "{z}" not in map_tile_url or "{x}" not in map_tile_url or "{y}" not in map_tile_url:
|
||||||
|
errors["map_tile_url"] = "URL must contain {z}, {x}, and {y} placeholders"
|
||||||
|
|
||||||
|
# Validate map_attribution
|
||||||
|
if not map_attribution:
|
||||||
|
errors["map_attribution"] = "Map attribution is required"
|
||||||
|
|
||||||
|
async with pool.acquire() as conn:
|
||||||
|
if errors:
|
||||||
|
row = await conn.fetchrow(
|
||||||
|
"SELECT map_tile_url, map_attribution FROM config.system WHERE id = true"
|
||||||
|
)
|
||||||
|
system = {
|
||||||
|
"map_tile_url": row["map_tile_url"] if row else "",
|
||||||
|
"map_attribution": row["map_attribution"] if row else "",
|
||||||
|
}
|
||||||
|
|
||||||
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
|
response = templates.TemplateResponse(
|
||||||
|
request=request,
|
||||||
|
name="setup_system.html",
|
||||||
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"error": None,
|
||||||
|
"errors": errors,
|
||||||
|
"form_data": form_data,
|
||||||
|
"system": system,
|
||||||
|
},
|
||||||
|
status_code=200,
|
||||||
|
)
|
||||||
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
return response
|
||||||
|
|
||||||
|
# Get current values for audit
|
||||||
|
old_row = await conn.fetchrow(
|
||||||
|
"SELECT map_tile_url, map_attribution FROM config.system WHERE id = true"
|
||||||
|
)
|
||||||
|
before = {
|
||||||
|
"map_tile_url": old_row["map_tile_url"] if old_row else None,
|
||||||
|
"map_attribution": old_row["map_attribution"] if old_row else None,
|
||||||
|
}
|
||||||
|
|
||||||
|
# Update system settings
|
||||||
|
await conn.execute(
|
||||||
|
"""
|
||||||
|
UPDATE config.system
|
||||||
|
SET map_tile_url = $1, map_attribution = $2
|
||||||
|
WHERE id = true
|
||||||
|
""",
|
||||||
|
map_tile_url,
|
||||||
|
map_attribution,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Write audit log
|
||||||
|
await write_audit(
|
||||||
|
conn,
|
||||||
|
SYSTEM_UPDATE,
|
||||||
|
operator_id=operator.id,
|
||||||
|
target="system",
|
||||||
|
before=before,
|
||||||
|
after={"map_tile_url": map_tile_url, "map_attribution": map_attribution},
|
||||||
|
)
|
||||||
|
|
||||||
|
return RedirectResponse(url="/setup/keys", status_code=302)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/setup/keys", response_class=HTMLResponse)
|
||||||
|
async def setup_keys_form(
|
||||||
|
request: Request,
|
||||||
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
|
) -> HTMLResponse:
|
||||||
|
"""Render the API keys form (step 3)."""
|
||||||
|
# Require authentication for this step
|
||||||
|
operator = getattr(request.state, "operator", None)
|
||||||
|
if operator is None:
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
|
||||||
|
from central.crypto import encrypt
|
||||||
|
|
||||||
|
templates = _get_templates()
|
||||||
|
pool = get_pool()
|
||||||
|
|
||||||
|
async with pool.acquire() as conn:
|
||||||
|
rows = await conn.fetch(
|
||||||
|
"SELECT alias, created_at FROM config.api_keys ORDER BY alias"
|
||||||
|
)
|
||||||
|
keys = [{"alias": row["alias"], "created_at": row["created_at"]} for row in rows]
|
||||||
|
|
||||||
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
|
response = templates.TemplateResponse(
|
||||||
|
request=request,
|
||||||
|
name="setup_keys.html",
|
||||||
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"keys": keys,
|
||||||
|
"errors": None,
|
||||||
|
"form_data": None,
|
||||||
|
"success": None,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/setup/keys")
|
||||||
|
async def setup_keys_submit(
|
||||||
|
request: Request,
|
||||||
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
|
) -> Response:
|
||||||
|
"""Process the API keys form (step 3)."""
|
||||||
|
# Require authentication for this step
|
||||||
|
operator = getattr(request.state, "operator", None)
|
||||||
|
if operator is None:
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
|
||||||
|
await csrf_protect.validate_csrf(request)
|
||||||
|
|
||||||
|
form = await request.form()
|
||||||
|
action = form.get("action", "add")
|
||||||
|
|
||||||
|
# If action is "next", redirect to adapters step
|
||||||
|
if action == "next":
|
||||||
|
return RedirectResponse(url="/setup/adapters", status_code=302)
|
||||||
|
|
||||||
|
from central.crypto import encrypt
|
||||||
|
|
||||||
|
templates = _get_templates()
|
||||||
|
pool = get_pool()
|
||||||
|
|
||||||
|
# Otherwise, add a new key
|
||||||
|
alias = form.get("alias", "").strip()
|
||||||
|
plaintext_key = form.get("plaintext_key", "")
|
||||||
|
|
||||||
|
form_data = {"alias": alias}
|
||||||
|
errors: dict[str, str] = {}
|
||||||
|
|
||||||
|
# Validate alias
|
||||||
|
if not alias:
|
||||||
|
errors["alias"] = "Alias is required"
|
||||||
|
elif len(alias) > 64:
|
||||||
|
errors["alias"] = "Alias must be at most 64 characters"
|
||||||
|
elif not ALIAS_REGEX.match(alias):
|
||||||
|
errors["alias"] = "Alias must contain only letters, numbers, and underscores"
|
||||||
|
|
||||||
|
# Validate plaintext_key
|
||||||
|
if not plaintext_key:
|
||||||
|
errors["plaintext_key"] = "API key is required"
|
||||||
|
elif len(plaintext_key) > 4096:
|
||||||
|
errors["plaintext_key"] = "API key must be at most 4096 characters"
|
||||||
|
|
||||||
|
async with pool.acquire() as conn:
|
||||||
|
if not errors:
|
||||||
|
# Check if alias already exists
|
||||||
|
existing = await conn.fetchrow(
|
||||||
|
"SELECT alias FROM config.api_keys WHERE alias = $1",
|
||||||
|
alias,
|
||||||
|
)
|
||||||
|
if existing:
|
||||||
|
errors["alias"] = "An API key with this alias already exists"
|
||||||
|
|
||||||
|
keys = await conn.fetch(
|
||||||
|
"SELECT alias, created_at FROM config.api_keys ORDER BY alias"
|
||||||
|
)
|
||||||
|
keys = [{"alias": row["alias"], "created_at": row["created_at"]} for row in keys]
|
||||||
|
|
||||||
|
if errors:
|
||||||
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
|
response = templates.TemplateResponse(
|
||||||
|
request=request,
|
||||||
|
name="setup_keys.html",
|
||||||
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"keys": keys,
|
||||||
|
"errors": errors,
|
||||||
|
"form_data": form_data,
|
||||||
|
"success": None,
|
||||||
|
},
|
||||||
|
status_code=200,
|
||||||
|
)
|
||||||
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
return response
|
||||||
|
|
||||||
|
# Encrypt the key
|
||||||
|
encrypted_value = encrypt(plaintext_key.encode())
|
||||||
|
|
||||||
|
# Insert the new key
|
||||||
|
row = await conn.fetchrow(
|
||||||
|
"""
|
||||||
|
INSERT INTO config.api_keys (alias, encrypted_value)
|
||||||
|
VALUES ($1, $2)
|
||||||
|
RETURNING created_at
|
||||||
|
""",
|
||||||
|
alias,
|
||||||
|
encrypted_value,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Write audit log (no plaintext!)
|
||||||
|
await write_audit(
|
||||||
|
conn,
|
||||||
|
API_KEY_CREATE,
|
||||||
|
operator_id=operator.id,
|
||||||
|
target=alias,
|
||||||
|
before=None,
|
||||||
|
after={"alias": alias, "created_at": row["created_at"].isoformat()},
|
||||||
|
)
|
||||||
|
|
||||||
|
# Refresh keys list
|
||||||
|
keys = await conn.fetch(
|
||||||
|
"SELECT alias, created_at FROM config.api_keys ORDER BY alias"
|
||||||
|
)
|
||||||
|
keys = [{"alias": row["alias"], "created_at": row["created_at"]} for row in keys]
|
||||||
|
|
||||||
|
# Re-render with success message
|
||||||
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
|
response = templates.TemplateResponse(
|
||||||
|
request=request,
|
||||||
|
name="setup_keys.html",
|
||||||
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"keys": keys,
|
||||||
|
"errors": None,
|
||||||
|
"form_data": None,
|
||||||
|
"success": f"API key '{alias}' added successfully.",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/setup/adapters", response_class=HTMLResponse)
|
||||||
|
async def setup_adapters_form(
|
||||||
|
request: Request,
|
||||||
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
|
) -> HTMLResponse:
|
||||||
|
"""Render the adapters configuration form (step 4)."""
|
||||||
|
# Require authentication for this step
|
||||||
|
operator = getattr(request.state, "operator", None)
|
||||||
|
if operator is None:
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
|
||||||
|
templates = _get_templates()
|
||||||
|
pool = get_pool()
|
||||||
|
|
||||||
|
async with pool.acquire() as conn:
|
||||||
|
rows = await conn.fetch(
|
||||||
|
"""
|
||||||
|
SELECT name, enabled, cadence_s, settings
|
||||||
|
FROM config.adapters
|
||||||
|
ORDER BY name
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
adapters = []
|
||||||
|
for row in rows:
|
||||||
|
settings = row["settings"] or {}
|
||||||
|
adapters.append({
|
||||||
|
"name": row["name"],
|
||||||
|
"enabled": row["enabled"],
|
||||||
|
"cadence_s": row["cadence_s"],
|
||||||
|
"settings": settings,
|
||||||
|
})
|
||||||
|
|
||||||
|
# Get API keys for dropdown
|
||||||
|
api_keys = await conn.fetch(
|
||||||
|
"SELECT alias FROM config.api_keys ORDER BY alias"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Get map tile settings
|
||||||
|
sys_row = await conn.fetchrow(
|
||||||
|
"SELECT map_tile_url, map_attribution FROM config.system WHERE id = true"
|
||||||
|
)
|
||||||
|
tile_url = sys_row["map_tile_url"] if sys_row else "https://tile.openstreetmap.org/{z}/{x}/{y}.png"
|
||||||
|
tile_attribution = sys_row["map_attribution"] if sys_row else "© OpenStreetMap contributors"
|
||||||
|
|
||||||
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
|
response = templates.TemplateResponse(
|
||||||
|
request=request,
|
||||||
|
name="setup_adapters.html",
|
||||||
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"adapters": adapters,
|
||||||
|
"api_keys": [{"alias": k["alias"]} for k in api_keys],
|
||||||
|
"valid_satellites": _get_valid_satellites(),
|
||||||
|
"valid_feeds": sorted(_get_valid_feeds()),
|
||||||
|
"tile_url": tile_url,
|
||||||
|
"tile_attribution": tile_attribution,
|
||||||
|
"error": None,
|
||||||
|
"errors": None,
|
||||||
|
"form_data": None,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/setup/adapters")
|
||||||
|
async def setup_adapters_submit(
|
||||||
|
request: Request,
|
||||||
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
|
) -> Response:
|
||||||
|
"""Process the adapters configuration form (step 4)."""
|
||||||
|
# Require authentication for this step
|
||||||
|
operator = getattr(request.state, "operator", None)
|
||||||
|
if operator is None:
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
|
||||||
|
templates = _get_templates()
|
||||||
|
pool = get_pool()
|
||||||
|
|
||||||
|
await csrf_protect.validate_csrf(request)
|
||||||
|
|
||||||
|
form = await request.form()
|
||||||
|
errors: dict[str, str] = {}
|
||||||
|
|
||||||
|
async with pool.acquire() as conn:
|
||||||
|
# Get current adapters
|
||||||
|
rows = await conn.fetch(
|
||||||
|
"""
|
||||||
|
SELECT name, enabled, cadence_s, settings
|
||||||
|
FROM config.adapters
|
||||||
|
ORDER BY name
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
for row in rows:
|
||||||
|
adapter_name = row["name"]
|
||||||
|
current_settings = row["settings"] or {}
|
||||||
|
new_settings = dict(current_settings)
|
||||||
|
|
||||||
|
# Parse enabled
|
||||||
|
enabled = f"{adapter_name}_enabled" in form
|
||||||
|
|
||||||
|
# Parse cadence
|
||||||
|
cadence_str = form.get(f"{adapter_name}_cadence_s", "")
|
||||||
|
try:
|
||||||
|
cadence_s = int(cadence_str)
|
||||||
|
if cadence_s < 60 or cadence_s > 3600:
|
||||||
|
errors[f"{adapter_name}_cadence_s"] = "Cadence must be between 60 and 3600 seconds"
|
||||||
|
except ValueError:
|
||||||
|
errors[f"{adapter_name}_cadence_s"] = "Cadence must be a valid integer"
|
||||||
|
cadence_s = row["cadence_s"]
|
||||||
|
|
||||||
|
# Adapter-specific validation
|
||||||
|
if adapter_name == "nws":
|
||||||
|
contact_email = form.get(f"{adapter_name}_contact_email", "").strip()
|
||||||
|
if enabled:
|
||||||
|
if not contact_email:
|
||||||
|
errors[f"{adapter_name}_contact_email"] = "Contact email is required when enabled"
|
||||||
|
elif not EMAIL_REGEX.match(contact_email):
|
||||||
|
errors[f"{adapter_name}_contact_email"] = "Invalid email format"
|
||||||
|
else:
|
||||||
|
new_settings["contact_email"] = contact_email
|
||||||
|
else:
|
||||||
|
new_settings["contact_email"] = contact_email if contact_email else current_settings.get("contact_email")
|
||||||
|
|
||||||
|
elif adapter_name == "firms":
|
||||||
|
api_key_alias = form.get(f"{adapter_name}_api_key_alias", "").strip()
|
||||||
|
satellites = form.getlist(f"{adapter_name}_satellites")
|
||||||
|
|
||||||
|
if api_key_alias:
|
||||||
|
key_exists = await conn.fetchrow(
|
||||||
|
"SELECT 1 FROM config.api_keys WHERE alias = $1",
|
||||||
|
api_key_alias,
|
||||||
|
)
|
||||||
|
if not key_exists:
|
||||||
|
errors[f"{adapter_name}_api_key_alias"] = f"API key alias '{api_key_alias}' does not exist"
|
||||||
|
else:
|
||||||
|
new_settings["api_key_alias"] = api_key_alias
|
||||||
|
else:
|
||||||
|
new_settings["api_key_alias"] = None
|
||||||
|
|
||||||
|
# Validate satellites
|
||||||
|
valid_sats = set(_get_valid_satellites())
|
||||||
|
invalid_sats = [s for s in satellites if s not in valid_sats]
|
||||||
|
if invalid_sats:
|
||||||
|
errors[f"{adapter_name}_satellites"] = f"Invalid satellites: {', '.join(invalid_sats)}"
|
||||||
|
else:
|
||||||
|
new_settings["satellites"] = satellites
|
||||||
|
|
||||||
|
elif adapter_name == "usgs_quake":
|
||||||
|
feed = form.get(f"{adapter_name}_feed", "").strip()
|
||||||
|
valid_feeds = _get_valid_feeds()
|
||||||
|
if feed not in valid_feeds:
|
||||||
|
errors[f"{adapter_name}_feed"] = f"Invalid feed"
|
||||||
|
else:
|
||||||
|
new_settings["feed"] = feed
|
||||||
|
|
||||||
|
# Region validation
|
||||||
|
region_north_str = form.get(f"{adapter_name}_region_north", "").strip()
|
||||||
|
region_south_str = form.get(f"{adapter_name}_region_south", "").strip()
|
||||||
|
region_east_str = form.get(f"{adapter_name}_region_east", "").strip()
|
||||||
|
region_west_str = form.get(f"{adapter_name}_region_west", "").strip()
|
||||||
|
|
||||||
|
try:
|
||||||
|
region_north = float(region_north_str)
|
||||||
|
region_south = float(region_south_str)
|
||||||
|
region_east = float(region_east_str)
|
||||||
|
region_west = float(region_west_str)
|
||||||
|
|
||||||
|
if not (-90 <= region_south < region_north <= 90):
|
||||||
|
errors[f"{adapter_name}_region"] = "Invalid latitude: south must be less than north, both between -90 and 90"
|
||||||
|
elif not (-180 <= region_west < region_east <= 180):
|
||||||
|
errors[f"{adapter_name}_region"] = "Invalid longitude: west must be less than east, both between -180 and 180"
|
||||||
|
else:
|
||||||
|
new_settings["region"] = {
|
||||||
|
"north": region_north,
|
||||||
|
"south": region_south,
|
||||||
|
"east": region_east,
|
||||||
|
"west": region_west,
|
||||||
|
}
|
||||||
|
except ValueError:
|
||||||
|
errors[f"{adapter_name}_region"] = "Region coordinates must be valid numbers"
|
||||||
|
|
||||||
|
# Store parsed data for re-render on error or update
|
||||||
|
if not errors.get(f"{adapter_name}_cadence_s"):
|
||||||
|
# Update adapter
|
||||||
|
await conn.execute(
|
||||||
|
"""
|
||||||
|
UPDATE config.adapters
|
||||||
|
SET enabled = $1, cadence_s = $2, settings = $3, updated_at = now()
|
||||||
|
WHERE name = $4
|
||||||
|
""",
|
||||||
|
enabled,
|
||||||
|
cadence_s,
|
||||||
|
new_settings,
|
||||||
|
adapter_name,
|
||||||
|
)
|
||||||
|
|
||||||
|
# If any errors, re-render
|
||||||
|
if errors:
|
||||||
|
adapters = []
|
||||||
|
rows = await conn.fetch(
|
||||||
|
"""
|
||||||
|
SELECT name, enabled, cadence_s, settings
|
||||||
|
FROM config.adapters
|
||||||
|
ORDER BY name
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
for row in rows:
|
||||||
|
settings = row["settings"] or {}
|
||||||
|
adapters.append({
|
||||||
|
"name": row["name"],
|
||||||
|
"enabled": row["enabled"],
|
||||||
|
"cadence_s": row["cadence_s"],
|
||||||
|
"settings": settings,
|
||||||
|
})
|
||||||
|
|
||||||
|
api_keys = await conn.fetch(
|
||||||
|
"SELECT alias FROM config.api_keys ORDER BY alias"
|
||||||
|
)
|
||||||
|
|
||||||
|
sys_row = await conn.fetchrow(
|
||||||
|
"SELECT map_tile_url, map_attribution FROM config.system WHERE id = true"
|
||||||
|
)
|
||||||
|
tile_url = sys_row["map_tile_url"] if sys_row else "https://tile.openstreetmap.org/{z}/{x}/{y}.png"
|
||||||
|
tile_attribution = sys_row["map_attribution"] if sys_row else "© OpenStreetMap contributors"
|
||||||
|
|
||||||
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
|
response = templates.TemplateResponse(
|
||||||
|
request=request,
|
||||||
|
name="setup_adapters.html",
|
||||||
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"adapters": adapters,
|
||||||
|
"api_keys": [{"alias": k["alias"]} for k in api_keys],
|
||||||
|
"valid_satellites": _get_valid_satellites(),
|
||||||
|
"valid_feeds": sorted(_get_valid_feeds()),
|
||||||
|
"tile_url": tile_url,
|
||||||
|
"tile_attribution": tile_attribution,
|
||||||
|
"error": "Please fix the errors below.",
|
||||||
|
"errors": errors,
|
||||||
|
"form_data": form,
|
||||||
|
},
|
||||||
|
status_code=200,
|
||||||
|
)
|
||||||
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
return response
|
||||||
|
|
||||||
|
return RedirectResponse(url="/setup/finish", status_code=302)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/setup/finish", response_class=HTMLResponse)
|
||||||
|
async def setup_finish_form(
|
||||||
|
request: Request,
|
||||||
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
|
) -> HTMLResponse:
|
||||||
|
"""Render the finish setup page (step 5)."""
|
||||||
|
# Require authentication for this step
|
||||||
|
operator = getattr(request.state, "operator", None)
|
||||||
|
if operator is None:
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
|
||||||
|
templates = _get_templates()
|
||||||
|
pool = get_pool()
|
||||||
|
|
||||||
|
async with pool.acquire() as conn:
|
||||||
|
# Get counts
|
||||||
|
operator_count = await conn.fetchval("SELECT COUNT(*) FROM config.operators")
|
||||||
|
key_count = await conn.fetchval("SELECT COUNT(*) FROM config.api_keys")
|
||||||
|
|
||||||
|
# Get system settings
|
||||||
|
sys_row = await conn.fetchrow(
|
||||||
|
"SELECT map_tile_url FROM config.system WHERE id = true"
|
||||||
|
)
|
||||||
|
system = {
|
||||||
|
"map_tile_url": sys_row["map_tile_url"] if sys_row else "",
|
||||||
|
}
|
||||||
|
|
||||||
|
# Get adapters
|
||||||
|
rows = await conn.fetch(
|
||||||
|
"""
|
||||||
|
SELECT name, enabled, cadence_s
|
||||||
|
FROM config.adapters
|
||||||
|
ORDER BY name
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
adapters = [
|
||||||
|
{
|
||||||
|
"name": row["name"],
|
||||||
|
"enabled": row["enabled"],
|
||||||
|
"cadence_s": row["cadence_s"],
|
||||||
|
}
|
||||||
|
for row in rows
|
||||||
|
]
|
||||||
|
|
||||||
|
csrf_token, signed_token = csrf_protect.generate_csrf_tokens()
|
||||||
|
response = templates.TemplateResponse(
|
||||||
|
request=request,
|
||||||
|
name="setup_finish.html",
|
||||||
|
context={
|
||||||
|
"csrf_token": csrf_token,
|
||||||
|
"operator_count": operator_count,
|
||||||
|
"key_count": key_count,
|
||||||
|
"system": system,
|
||||||
|
"adapters": adapters,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
csrf_protect.set_csrf_cookie(signed_token, response)
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/setup/finish")
|
||||||
|
async def setup_finish_submit(
|
||||||
|
request: Request,
|
||||||
|
csrf_protect: CsrfProtect = Depends(),
|
||||||
|
) -> Response:
|
||||||
|
"""Complete the setup wizard."""
|
||||||
|
# Require authentication for this step
|
||||||
|
operator = getattr(request.state, "operator", None)
|
||||||
|
if operator is None:
|
||||||
|
return RedirectResponse(url="/setup/operator", status_code=302)
|
||||||
|
|
||||||
|
pool = get_pool()
|
||||||
|
|
||||||
|
await csrf_protect.validate_csrf(request)
|
||||||
|
|
||||||
|
async with pool.acquire() as conn:
|
||||||
# Mark setup complete
|
# Mark setup complete
|
||||||
await conn.execute(
|
await conn.execute(
|
||||||
"UPDATE config.system SET setup_complete = true WHERE id = true"
|
"UPDATE config.system SET setup_complete = true WHERE id = true"
|
||||||
)
|
)
|
||||||
|
|
||||||
# Redirect with session cookie
|
# Write audit log
|
||||||
response = RedirectResponse(url="/", status_code=302)
|
await write_audit(
|
||||||
_set_session_cookie(response, token, lifetime_days * 86400)
|
conn,
|
||||||
return response
|
SETUP_COMPLETE,
|
||||||
|
operator_id=operator.id,
|
||||||
|
target="system",
|
||||||
|
)
|
||||||
|
|
||||||
|
return RedirectResponse(url="/", status_code=302)
|
||||||
|
|
||||||
|
|
||||||
@router.get("/login", response_class=HTMLResponse)
|
@router.get("/login", response_class=HTMLResponse)
|
||||||
|
|
|
||||||
6
src/central/gui/templates/_wizard_header.html
Normal file
6
src/central/gui/templates/_wizard_header.html
Normal file
|
|
@ -0,0 +1,6 @@
|
||||||
|
<article style="margin-bottom: 2rem;">
|
||||||
|
<header>
|
||||||
|
<strong>Step {{ step }} of 5</strong> — {{ step_name }}
|
||||||
|
</header>
|
||||||
|
<progress value="{{ step }}" max="5" style="margin-bottom: 0;"></progress>
|
||||||
|
</article>
|
||||||
24
src/central/gui/templates/base_wizard.html
Normal file
24
src/central/gui/templates/base_wizard.html
Normal file
|
|
@ -0,0 +1,24 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en" data-theme="light">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>{% block title %}Central - Setup{% endblock %}</title>
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@picocss/pico@2/css/pico.min.css">
|
||||||
|
<script src="https://unpkg.com/htmx.org@2.0.4"></script>
|
||||||
|
{% block head %}{% endblock %}
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<nav class="container">
|
||||||
|
<ul>
|
||||||
|
<li><strong>Central</strong></li>
|
||||||
|
</ul>
|
||||||
|
<ul>
|
||||||
|
<li>Setup Wizard</li>
|
||||||
|
</ul>
|
||||||
|
</nav>
|
||||||
|
<main class="container">
|
||||||
|
{% block content %}{% endblock %}
|
||||||
|
</main>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
217
src/central/gui/templates/setup_adapters.html
Normal file
217
src/central/gui/templates/setup_adapters.html
Normal file
|
|
@ -0,0 +1,217 @@
|
||||||
|
{% extends "base_wizard.html" %}
|
||||||
|
|
||||||
|
{% block title %}Central - Configure Adapters{% endblock %}
|
||||||
|
|
||||||
|
{% block head %}
|
||||||
|
<link rel="stylesheet" href="https://unpkg.com/leaflet@1.9.4/dist/leaflet.css" integrity="sha256-p4NxAoJBhIIN+hmNHrzRCf9tD/miZyoHS5obTRR9BMY=" crossorigin="">
|
||||||
|
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/leaflet.draw/1.0.4/leaflet.draw.css" integrity="sha512-gc3xjCmIy673V6MyOAZhIW93xhM9ei1I+gLbmFjUHIjocENRsLX/QUE1htk5q1XV2D/iie/VQ8DXI6Uj8GB1Og==" crossorigin="anonymous">
|
||||||
|
<script src="https://unpkg.com/leaflet@1.9.4/dist/leaflet.js" integrity="sha256-20nQCchB9co0qIjJZRGuk2/Z9VM+kNiyxNV1lvTlZBo=" crossorigin=""></script>
|
||||||
|
<script src="https://cdnjs.cloudflare.com/ajax/libs/leaflet.draw/1.0.4/leaflet.draw.js" integrity="sha512-ozq8xQKq6urvuU6jNgkfqAmT7jKN2XumbrX1JiB3TnF7tI48DPI4Ber9dLJ0ikXiRg9G9Vl2jXwqjZ5LDGQ3g==" crossorigin="anonymous"></script>
|
||||||
|
{% endblock %}
|
||||||
|
|
||||||
|
{% block content %}
|
||||||
|
{% with step=4, step_name="Configure Adapters" %}
|
||||||
|
{% include "_wizard_header.html" %}
|
||||||
|
{% endwith %}
|
||||||
|
|
||||||
|
<article>
|
||||||
|
<header>
|
||||||
|
<h1>Configure Adapters</h1>
|
||||||
|
<p>Enable and configure data source adapters. Each adapter polls an external API and normalizes events.</p>
|
||||||
|
</header>
|
||||||
|
|
||||||
|
{% if error %}
|
||||||
|
<p style="color: var(--pico-color-red-500);">{{ error }}</p>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<form action="/setup/adapters" method="post">
|
||||||
|
<input type="hidden" name="csrf_token" value="{{ csrf_token }}">
|
||||||
|
|
||||||
|
{% for adapter in adapters %}
|
||||||
|
<details open style="margin-bottom: 2rem;">
|
||||||
|
<summary><strong>{{ adapter.name }}</strong></summary>
|
||||||
|
|
||||||
|
<div style="padding: 1rem; border-left: 3px solid var(--pico-primary);">
|
||||||
|
<label>
|
||||||
|
<input type="checkbox" name="{{ adapter.name }}_enabled"
|
||||||
|
{% if form_data and form_data.get(adapter.name + '_enabled') %}checked
|
||||||
|
{% elif not form_data and adapter.enabled %}checked{% endif %}>
|
||||||
|
Enabled
|
||||||
|
</label>
|
||||||
|
{% if errors and errors.get(adapter.name + '_enabled') %}
|
||||||
|
<small style="color: var(--pico-color-red-500); display: block;">{{ errors[adapter.name + '_enabled'] }}</small>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<label for="{{ adapter.name }}_cadence_s">Cadence (seconds)</label>
|
||||||
|
<input type="number" id="{{ adapter.name }}_cadence_s" name="{{ adapter.name }}_cadence_s"
|
||||||
|
value="{{ form_data.get(adapter.name + '_cadence_s') if form_data else adapter.cadence_s }}"
|
||||||
|
min="60" max="3600">
|
||||||
|
{% if errors and errors.get(adapter.name + '_cadence_s') %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors[adapter.name + '_cadence_s'] }}</small>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
{% if adapter.name == 'nws' %}
|
||||||
|
<label for="{{ adapter.name }}_contact_email">Contact Email</label>
|
||||||
|
<input type="email" id="{{ adapter.name }}_contact_email" name="{{ adapter.name }}_contact_email"
|
||||||
|
value="{{ form_data.get(adapter.name + '_contact_email') if form_data else adapter.settings.contact_email }}">
|
||||||
|
{% if errors and errors.get(adapter.name + '_contact_email') %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors[adapter.name + '_contact_email'] }}</small>
|
||||||
|
{% endif %}
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
{% if adapter.name == 'firms' %}
|
||||||
|
<label for="{{ adapter.name }}_api_key_alias">API Key Alias</label>
|
||||||
|
<select id="{{ adapter.name }}_api_key_alias" name="{{ adapter.name }}_api_key_alias">
|
||||||
|
<option value="">(none)</option>
|
||||||
|
{% for key in api_keys %}
|
||||||
|
<option value="{{ key.alias }}"
|
||||||
|
{% if (form_data.get(adapter.name + '_api_key_alias') if form_data else adapter.settings.api_key_alias) == key.alias %}selected{% endif %}>
|
||||||
|
{{ key.alias }}
|
||||||
|
</option>
|
||||||
|
{% endfor %}
|
||||||
|
</select>
|
||||||
|
{% if errors and errors.get(adapter.name + '_api_key_alias') %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors[adapter.name + '_api_key_alias'] }}</small>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<label>Satellites</label>
|
||||||
|
{% for sat in valid_satellites %}
|
||||||
|
<label style="display: inline-block; margin-right: 1rem;">
|
||||||
|
<input type="checkbox" name="{{ adapter.name }}_satellites" value="{{ sat }}"
|
||||||
|
{% if sat in (form_data.getlist(adapter.name + '_satellites') if form_data else adapter.settings.satellites or []) %}checked{% endif %}>
|
||||||
|
{{ sat }}
|
||||||
|
</label>
|
||||||
|
{% endfor %}
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
{% if adapter.name == 'usgs_quake' %}
|
||||||
|
<label for="{{ adapter.name }}_feed">Feed</label>
|
||||||
|
<select id="{{ adapter.name }}_feed" name="{{ adapter.name }}_feed">
|
||||||
|
{% for f in valid_feeds %}
|
||||||
|
<option value="{{ f }}"
|
||||||
|
{% if (form_data.get(adapter.name + '_feed') if form_data else adapter.settings.feed) == f %}selected{% endif %}>
|
||||||
|
{{ f }}
|
||||||
|
</option>
|
||||||
|
{% endfor %}
|
||||||
|
</select>
|
||||||
|
{% if errors and errors.get(adapter.name + '_feed') %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors[adapter.name + '_feed'] }}</small>
|
||||||
|
{% endif %}
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<h4>Region</h4>
|
||||||
|
{% set region = form_data if form_data else adapter.settings.region %}
|
||||||
|
<div id="region-picker-{{ adapter.name }}"
|
||||||
|
data-adapter="{{ adapter.name }}"
|
||||||
|
data-north="{{ form_data.get(adapter.name + '_region_north') if form_data else (adapter.settings.region.north if adapter.settings.region else 49.5) }}"
|
||||||
|
data-south="{{ form_data.get(adapter.name + '_region_south') if form_data else (adapter.settings.region.south if adapter.settings.region else 31.0) }}"
|
||||||
|
data-east="{{ form_data.get(adapter.name + '_region_east') if form_data else (adapter.settings.region.east if adapter.settings.region else -102.0) }}"
|
||||||
|
data-west="{{ form_data.get(adapter.name + '_region_west') if form_data else (adapter.settings.region.west if adapter.settings.region else -124.5) }}"
|
||||||
|
data-tile-url="{{ tile_url }}"
|
||||||
|
data-tile-attr="{{ tile_attribution }}">
|
||||||
|
|
||||||
|
<div id="region-map-{{ adapter.name }}" style="height: 300px; margin-bottom: 1rem;"></div>
|
||||||
|
|
||||||
|
<div class="grid">
|
||||||
|
<div>
|
||||||
|
<label>North</label>
|
||||||
|
<input type="number" name="{{ adapter.name }}_region_north" step="0.0001" min="-90" max="90" readonly
|
||||||
|
value="{{ form_data.get(adapter.name + '_region_north') if form_data else (adapter.settings.region.north if adapter.settings.region else 49.5) }}">
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<label>South</label>
|
||||||
|
<input type="number" name="{{ adapter.name }}_region_south" step="0.0001" min="-90" max="90" readonly
|
||||||
|
value="{{ form_data.get(adapter.name + '_region_south') if form_data else (adapter.settings.region.south if adapter.settings.region else 31.0) }}">
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<label>East</label>
|
||||||
|
<input type="number" name="{{ adapter.name }}_region_east" step="0.0001" min="-180" max="180" readonly
|
||||||
|
value="{{ form_data.get(adapter.name + '_region_east') if form_data else (adapter.settings.region.east if adapter.settings.region else -102.0) }}">
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<label>West</label>
|
||||||
|
<input type="number" name="{{ adapter.name }}_region_west" step="0.0001" min="-180" max="180" readonly
|
||||||
|
value="{{ form_data.get(adapter.name + '_region_west') if form_data else (adapter.settings.region.west if adapter.settings.region else -124.5) }}">
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
{% if errors and errors.get(adapter.name + '_region') %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors[adapter.name + '_region'] }}</small>
|
||||||
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</details>
|
||||||
|
{% endfor %}
|
||||||
|
|
||||||
|
<div style="display: flex; gap: 1rem; margin-top: 1rem;">
|
||||||
|
<a href="/setup/keys" role="button" class="outline">← Back</a>
|
||||||
|
<button type="submit">Next →</button>
|
||||||
|
</div>
|
||||||
|
</form>
|
||||||
|
</article>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
document.addEventListener('DOMContentLoaded', function() {
|
||||||
|
const adapters = ['nws', 'firms', 'usgs_quake'];
|
||||||
|
|
||||||
|
adapters.forEach(function(adapterName) {
|
||||||
|
const container = document.getElementById('region-picker-' + adapterName);
|
||||||
|
if (!container) return;
|
||||||
|
|
||||||
|
const savedNorth = parseFloat(container.dataset.north);
|
||||||
|
const savedSouth = parseFloat(container.dataset.south);
|
||||||
|
const savedEast = parseFloat(container.dataset.east);
|
||||||
|
const savedWest = parseFloat(container.dataset.west);
|
||||||
|
const tileUrl = container.dataset.tileUrl || 'https://tile.openstreetmap.org/{z}/{x}/{y}.png';
|
||||||
|
const tileAttr = container.dataset.tileAttr || '© OpenStreetMap contributors';
|
||||||
|
|
||||||
|
const centerLat = (savedNorth + savedSouth) / 2;
|
||||||
|
const centerLng = (savedEast + savedWest) / 2;
|
||||||
|
const mapEl = document.getElementById('region-map-' + adapterName);
|
||||||
|
const map = L.map(mapEl).setView([centerLat, centerLng], 4);
|
||||||
|
|
||||||
|
L.tileLayer(tileUrl, {
|
||||||
|
attribution: tileAttr,
|
||||||
|
maxZoom: 18
|
||||||
|
}).addTo(map);
|
||||||
|
|
||||||
|
const bounds = L.latLngBounds(
|
||||||
|
L.latLng(savedSouth, savedWest),
|
||||||
|
L.latLng(savedNorth, savedEast)
|
||||||
|
);
|
||||||
|
map.fitBounds(bounds.pad(0.1));
|
||||||
|
|
||||||
|
const rectangle = L.rectangle(bounds, {
|
||||||
|
color: '#3388ff',
|
||||||
|
weight: 2,
|
||||||
|
fillOpacity: 0.2
|
||||||
|
}).addTo(map);
|
||||||
|
|
||||||
|
rectangle.editing.enable();
|
||||||
|
|
||||||
|
const northInput = container.querySelector('input[name="' + adapterName + '_region_north"]');
|
||||||
|
const southInput = container.querySelector('input[name="' + adapterName + '_region_south"]');
|
||||||
|
const eastInput = container.querySelector('input[name="' + adapterName + '_region_east"]');
|
||||||
|
const westInput = container.querySelector('input[name="' + adapterName + '_region_west"]');
|
||||||
|
|
||||||
|
function updateInputs() {
|
||||||
|
const b = rectangle.getBounds();
|
||||||
|
northInput.value = b.getNorth().toFixed(4);
|
||||||
|
southInput.value = b.getSouth().toFixed(4);
|
||||||
|
eastInput.value = b.getEast().toFixed(4);
|
||||||
|
westInput.value = b.getWest().toFixed(4);
|
||||||
|
}
|
||||||
|
|
||||||
|
rectangle.on('edit', updateInputs);
|
||||||
|
updateInputs();
|
||||||
|
|
||||||
|
// Fix map size when details is opened
|
||||||
|
const details = container.closest('details');
|
||||||
|
if (details) {
|
||||||
|
details.addEventListener('toggle', function() {
|
||||||
|
setTimeout(function() { map.invalidateSize(); }, 100);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
{% endblock %}
|
||||||
69
src/central/gui/templates/setup_finish.html
Normal file
69
src/central/gui/templates/setup_finish.html
Normal file
|
|
@ -0,0 +1,69 @@
|
||||||
|
{% extends "base_wizard.html" %}
|
||||||
|
|
||||||
|
{% block title %}Central - Finish Setup{% endblock %}
|
||||||
|
|
||||||
|
{% block content %}
|
||||||
|
{% with step=5, step_name="Finish Setup" %}
|
||||||
|
{% include "_wizard_header.html" %}
|
||||||
|
{% endwith %}
|
||||||
|
|
||||||
|
<article>
|
||||||
|
<header>
|
||||||
|
<h1>Setup Complete</h1>
|
||||||
|
<p>Review your configuration and finish the setup wizard.</p>
|
||||||
|
</header>
|
||||||
|
|
||||||
|
<h2>Summary</h2>
|
||||||
|
|
||||||
|
<table>
|
||||||
|
<tbody>
|
||||||
|
<tr>
|
||||||
|
<th>Operators</th>
|
||||||
|
<td>{{ operator_count }} configured</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<th>API Keys</th>
|
||||||
|
<td>{{ key_count }} configured</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<th>Map Tile URL</th>
|
||||||
|
<td style="word-break: break-all;">{{ system.map_tile_url }}</td>
|
||||||
|
</tr>
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
|
||||||
|
<h3>Adapters</h3>
|
||||||
|
<table>
|
||||||
|
<thead>
|
||||||
|
<tr>
|
||||||
|
<th>Adapter</th>
|
||||||
|
<th>Status</th>
|
||||||
|
<th>Cadence</th>
|
||||||
|
</tr>
|
||||||
|
</thead>
|
||||||
|
<tbody>
|
||||||
|
{% for adapter in adapters %}
|
||||||
|
<tr>
|
||||||
|
<td><strong>{{ adapter.name }}</strong></td>
|
||||||
|
<td>
|
||||||
|
{% if adapter.enabled %}
|
||||||
|
<span style="color: var(--pico-color-green-500);">Enabled</span>
|
||||||
|
{% else %}
|
||||||
|
<span style="color: var(--pico-color-grey-500);">Disabled</span>
|
||||||
|
{% endif %}
|
||||||
|
</td>
|
||||||
|
<td>{{ adapter.cadence_s }}s</td>
|
||||||
|
</tr>
|
||||||
|
{% endfor %}
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
|
||||||
|
<form action="/setup/finish" method="post">
|
||||||
|
<input type="hidden" name="csrf_token" value="{{ csrf_token }}">
|
||||||
|
<div style="display: flex; gap: 1rem; margin-top: 2rem;">
|
||||||
|
<a href="/setup/adapters" role="button" class="outline">← Back</a>
|
||||||
|
<button type="submit">Finish Setup</button>
|
||||||
|
</div>
|
||||||
|
</form>
|
||||||
|
</article>
|
||||||
|
{% endblock %}
|
||||||
84
src/central/gui/templates/setup_keys.html
Normal file
84
src/central/gui/templates/setup_keys.html
Normal file
|
|
@ -0,0 +1,84 @@
|
||||||
|
{% extends "base_wizard.html" %}
|
||||||
|
|
||||||
|
{% block title %}Central - API Keys{% endblock %}
|
||||||
|
|
||||||
|
{% block content %}
|
||||||
|
{% with step=3, step_name="API Keys" %}
|
||||||
|
{% include "_wizard_header.html" %}
|
||||||
|
{% endwith %}
|
||||||
|
|
||||||
|
<article>
|
||||||
|
<header>
|
||||||
|
<h1>API Keys</h1>
|
||||||
|
<p>Add API keys for adapters that require external service credentials (e.g., FIRMS).</p>
|
||||||
|
</header>
|
||||||
|
|
||||||
|
{% if success %}
|
||||||
|
<p style="color: var(--pico-color-green-500);">{{ success }}</p>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
{% if keys %}
|
||||||
|
<h2>Existing Keys</h2>
|
||||||
|
<table>
|
||||||
|
<thead>
|
||||||
|
<tr>
|
||||||
|
<th>Alias</th>
|
||||||
|
<th>Created</th>
|
||||||
|
</tr>
|
||||||
|
</thead>
|
||||||
|
<tbody>
|
||||||
|
{% for key in keys %}
|
||||||
|
<tr>
|
||||||
|
<td><strong>{{ key.alias }}</strong></td>
|
||||||
|
<td>{{ key.created_at.strftime('%Y-%m-%d %H:%M') if key.created_at else '(never)' }}</td>
|
||||||
|
</tr>
|
||||||
|
{% endfor %}
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
{% else %}
|
||||||
|
<p><em>No API keys configured yet.</em></p>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<h2>Add New Key</h2>
|
||||||
|
<form action="/setup/keys" method="post">
|
||||||
|
<input type="hidden" name="csrf_token" value="{{ csrf_token }}">
|
||||||
|
<input type="hidden" name="action" value="add">
|
||||||
|
|
||||||
|
<div class="grid">
|
||||||
|
<div>
|
||||||
|
<label for="alias">Alias</label>
|
||||||
|
<input type="text" id="alias" name="alias" placeholder="e.g., firms"
|
||||||
|
value="{{ form_data.alias if form_data else '' }}" maxlength="64">
|
||||||
|
{% if errors and errors.alias %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors.alias }}</small>
|
||||||
|
{% else %}
|
||||||
|
<small>Letters, numbers, and underscores only.</small>
|
||||||
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<label for="plaintext_key">API Key</label>
|
||||||
|
<input type="password" id="plaintext_key" name="plaintext_key"
|
||||||
|
placeholder="Paste your API key">
|
||||||
|
{% if errors and errors.plaintext_key %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors.plaintext_key }}</small>
|
||||||
|
{% else %}
|
||||||
|
<small>Will be encrypted before storage.</small>
|
||||||
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button type="submit" class="outline">Add Key</button>
|
||||||
|
</form>
|
||||||
|
|
||||||
|
<hr>
|
||||||
|
|
||||||
|
<form action="/setup/keys" method="post">
|
||||||
|
<input type="hidden" name="csrf_token" value="{{ csrf_token }}">
|
||||||
|
<input type="hidden" name="action" value="next">
|
||||||
|
<div style="display: flex; gap: 1rem;">
|
||||||
|
<a href="/setup/system" role="button" class="outline">← Back</a>
|
||||||
|
<button type="submit">Next →</button>
|
||||||
|
</div>
|
||||||
|
</form>
|
||||||
|
</article>
|
||||||
|
{% endblock %}
|
||||||
45
src/central/gui/templates/setup_operator.html
Normal file
45
src/central/gui/templates/setup_operator.html
Normal file
|
|
@ -0,0 +1,45 @@
|
||||||
|
{% extends "base_wizard.html" %}
|
||||||
|
|
||||||
|
{% block title %}Central - Create Operator{% endblock %}
|
||||||
|
|
||||||
|
{% block content %}
|
||||||
|
{% with step=1, step_name="Create Operator" %}
|
||||||
|
{% include "_wizard_header.html" %}
|
||||||
|
{% endwith %}
|
||||||
|
|
||||||
|
<article>
|
||||||
|
<header>
|
||||||
|
<h1>Create Operator Account</h1>
|
||||||
|
<p>Create the initial operator account to manage Central.</p>
|
||||||
|
</header>
|
||||||
|
|
||||||
|
{% if error %}
|
||||||
|
<p style="color: var(--pico-color-red-500);">{{ error }}</p>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<form action="/setup/operator" method="post">
|
||||||
|
<input type="hidden" name="csrf_token" value="{{ csrf_token }}">
|
||||||
|
|
||||||
|
<label for="username">
|
||||||
|
Username
|
||||||
|
<input type="text" id="username" name="username" required
|
||||||
|
autocomplete="username" autofocus value="{{ form_data.username if form_data else '' }}">
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<label for="password">
|
||||||
|
Password
|
||||||
|
<input type="password" id="password" name="password" required
|
||||||
|
autocomplete="new-password" minlength="8">
|
||||||
|
<small>Minimum 8 characters</small>
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<label for="confirm_password">
|
||||||
|
Confirm Password
|
||||||
|
<input type="password" id="confirm_password" name="confirm_password" required
|
||||||
|
autocomplete="new-password">
|
||||||
|
</label>
|
||||||
|
|
||||||
|
<button type="submit">Create Operator →</button>
|
||||||
|
</form>
|
||||||
|
</article>
|
||||||
|
{% endblock %}
|
||||||
49
src/central/gui/templates/setup_system.html
Normal file
49
src/central/gui/templates/setup_system.html
Normal file
|
|
@ -0,0 +1,49 @@
|
||||||
|
{% extends "base_wizard.html" %}
|
||||||
|
|
||||||
|
{% block title %}Central - System Settings{% endblock %}
|
||||||
|
|
||||||
|
{% block content %}
|
||||||
|
{% with step=2, step_name="System Settings" %}
|
||||||
|
{% include "_wizard_header.html" %}
|
||||||
|
{% endwith %}
|
||||||
|
|
||||||
|
<article>
|
||||||
|
<header>
|
||||||
|
<h1>System Settings</h1>
|
||||||
|
<p>Configure map tile provider for the region picker.</p>
|
||||||
|
</header>
|
||||||
|
|
||||||
|
{% if error %}
|
||||||
|
<p style="color: var(--pico-color-red-500);">{{ error }}</p>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<form action="/setup/system" method="post">
|
||||||
|
<input type="hidden" name="csrf_token" value="{{ csrf_token }}">
|
||||||
|
|
||||||
|
<label for="map_tile_url">
|
||||||
|
Map Tile URL
|
||||||
|
<input type="text" id="map_tile_url" name="map_tile_url"
|
||||||
|
value="{{ form_data.map_tile_url if form_data else system.map_tile_url }}" required>
|
||||||
|
<small>Use {z}, {x}, {y} placeholders. Example: https://tile.openstreetmap.org/{z}/{x}/{y}.png</small>
|
||||||
|
</label>
|
||||||
|
{% if errors and errors.map_tile_url %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors.map_tile_url }}</small>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<label for="map_attribution">
|
||||||
|
Map Attribution
|
||||||
|
<input type="text" id="map_attribution" name="map_attribution"
|
||||||
|
value="{{ form_data.map_attribution if form_data else system.map_attribution }}" required>
|
||||||
|
<small>Credit the map provider (required by most tile services).</small>
|
||||||
|
</label>
|
||||||
|
{% if errors and errors.map_attribution %}
|
||||||
|
<small style="color: var(--pico-color-red-500);">{{ errors.map_attribution }}</small>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<div style="display: flex; gap: 1rem; margin-top: 1rem;">
|
||||||
|
<a href="/setup/operator" role="button" class="outline">← Back</a>
|
||||||
|
<button type="submit">Next →</button>
|
||||||
|
</div>
|
||||||
|
</form>
|
||||||
|
</article>
|
||||||
|
{% endblock %}
|
||||||
|
|
@ -12,8 +12,8 @@ class TestSetupGateMiddleware:
|
||||||
"""Tests for SetupGateMiddleware."""
|
"""Tests for SetupGateMiddleware."""
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_allows_setup_route_when_incomplete(self):
|
async def test_allows_setup_subpath_when_incomplete(self):
|
||||||
"""SetupGateMiddleware allows /setup when setup_complete=False."""
|
"""SetupGateMiddleware allows /setup/operator when setup_complete=False."""
|
||||||
mock_pool = MagicMock()
|
mock_pool = MagicMock()
|
||||||
mock_conn = MagicMock()
|
mock_conn = MagicMock()
|
||||||
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": False})
|
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": False})
|
||||||
|
|
@ -21,6 +21,31 @@ class TestSetupGateMiddleware:
|
||||||
mock_conn.__aexit__ = AsyncMock()
|
mock_conn.__aexit__ = AsyncMock()
|
||||||
mock_pool.acquire = MagicMock(return_value=mock_conn)
|
mock_pool.acquire = MagicMock(return_value=mock_conn)
|
||||||
|
|
||||||
|
with patch("central.gui.middleware.get_pool", return_value=mock_pool):
|
||||||
|
app = FastAPI()
|
||||||
|
|
||||||
|
@app.get("/setup/operator")
|
||||||
|
async def setup_operator():
|
||||||
|
return {"message": "operator"}
|
||||||
|
|
||||||
|
app.add_middleware(SetupGateMiddleware)
|
||||||
|
client = TestClient(app)
|
||||||
|
|
||||||
|
response = client.get("/setup/operator")
|
||||||
|
assert response.status_code == 200
|
||||||
|
assert response.json() == {"message": "operator"}
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_redirects_setup_base_to_wizard_step(self):
|
||||||
|
"""SetupGateMiddleware redirects /setup to wizard step when incomplete."""
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_conn = MagicMock()
|
||||||
|
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": False})
|
||||||
|
mock_conn.fetchval = AsyncMock(return_value=0) # No operators
|
||||||
|
mock_conn.__aenter__ = AsyncMock(return_value=mock_conn)
|
||||||
|
mock_conn.__aexit__ = AsyncMock()
|
||||||
|
mock_pool.acquire = MagicMock(return_value=mock_conn)
|
||||||
|
|
||||||
with patch("central.gui.middleware.get_pool", return_value=mock_pool):
|
with patch("central.gui.middleware.get_pool", return_value=mock_pool):
|
||||||
app = FastAPI()
|
app = FastAPI()
|
||||||
|
|
||||||
|
|
@ -28,12 +53,16 @@ class TestSetupGateMiddleware:
|
||||||
async def setup():
|
async def setup():
|
||||||
return {"message": "setup"}
|
return {"message": "setup"}
|
||||||
|
|
||||||
|
@app.get("/setup/operator")
|
||||||
|
async def setup_operator():
|
||||||
|
return {"message": "operator"}
|
||||||
|
|
||||||
app.add_middleware(SetupGateMiddleware)
|
app.add_middleware(SetupGateMiddleware)
|
||||||
client = TestClient(app)
|
client = TestClient(app, follow_redirects=False)
|
||||||
|
|
||||||
response = client.get("/setup")
|
response = client.get("/setup")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 302
|
||||||
assert response.json() == {"message": "setup"}
|
assert response.headers["location"] == "/setup/operator"
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_allows_health_when_incomplete(self):
|
async def test_allows_health_when_incomplete(self):
|
||||||
|
|
@ -135,7 +164,7 @@ class TestSetupGateMiddleware:
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_redirects_setup_when_complete(self):
|
async def test_redirects_setup_when_complete(self):
|
||||||
"""SetupGateMiddleware redirects /setup to / when setup_complete=True."""
|
"""SetupGateMiddleware redirects /setup/* to / when setup_complete=True."""
|
||||||
mock_pool = MagicMock()
|
mock_pool = MagicMock()
|
||||||
mock_conn = MagicMock()
|
mock_conn = MagicMock()
|
||||||
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": True})
|
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": True})
|
||||||
|
|
@ -154,9 +183,18 @@ class TestSetupGateMiddleware:
|
||||||
async def setup():
|
async def setup():
|
||||||
return {"message": "setup"}
|
return {"message": "setup"}
|
||||||
|
|
||||||
|
@app.get("/setup/operator")
|
||||||
|
async def setup_operator():
|
||||||
|
return {"message": "operator"}
|
||||||
|
|
||||||
app.add_middleware(SetupGateMiddleware)
|
app.add_middleware(SetupGateMiddleware)
|
||||||
client = TestClient(app, follow_redirects=False)
|
client = TestClient(app, follow_redirects=False)
|
||||||
|
|
||||||
|
# Both /setup and /setup/operator should redirect to /
|
||||||
response = client.get("/setup")
|
response = client.get("/setup")
|
||||||
assert response.status_code == 302
|
assert response.status_code == 302
|
||||||
assert response.headers["location"] == "/"
|
assert response.headers["location"] == "/"
|
||||||
|
|
||||||
|
response = client.get("/setup/operator")
|
||||||
|
assert response.status_code == 302
|
||||||
|
assert response.headers["location"] == "/"
|
||||||
|
|
|
||||||
586
tests/test_wizard.py
Normal file
586
tests/test_wizard.py
Normal file
|
|
@ -0,0 +1,586 @@
|
||||||
|
"""Tests for the first-run setup wizard."""
|
||||||
|
|
||||||
|
from datetime import datetime, timezone
|
||||||
|
from unittest.mock import AsyncMock, MagicMock, patch
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from central.gui.routes import (
|
||||||
|
setup_operator_form,
|
||||||
|
setup_operator_submit,
|
||||||
|
setup_system_form,
|
||||||
|
setup_system_submit,
|
||||||
|
setup_keys_form,
|
||||||
|
setup_keys_submit,
|
||||||
|
setup_adapters_form,
|
||||||
|
setup_adapters_submit,
|
||||||
|
setup_finish_form,
|
||||||
|
setup_finish_submit,
|
||||||
|
)
|
||||||
|
from central.gui.middleware import SetupGateMiddleware, _get_wizard_redirect_step
|
||||||
|
|
||||||
|
|
||||||
|
class TestWizardStepRedirect:
|
||||||
|
"""Test wizard step redirect logic."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_no_operators_redirects_to_operator(self):
|
||||||
|
"""When no operators exist, redirect to /setup/operator."""
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchval.side_effect = [0] # No operators
|
||||||
|
|
||||||
|
result = await _get_wizard_redirect_step(mock_conn)
|
||||||
|
assert result == "/setup/operator"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_default_tile_url_redirects_to_system(self):
|
||||||
|
"""When map_tile_url is default, redirect to /setup/system."""
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchval.side_effect = [1] # Has operator
|
||||||
|
mock_conn.fetchrow.return_value = {
|
||||||
|
"map_tile_url": "https://tile.openstreetmap.org/{z}/{x}/{y}.png"
|
||||||
|
}
|
||||||
|
|
||||||
|
result = await _get_wizard_redirect_step(mock_conn)
|
||||||
|
assert result == "/setup/system"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_no_adapters_touched_redirects_to_keys(self):
|
||||||
|
"""When no adapters have been updated, redirect to /setup/keys."""
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchval.side_effect = [1, 0] # Has operator, no adapters touched
|
||||||
|
mock_conn.fetchrow.return_value = {
|
||||||
|
"map_tile_url": "https://custom.example.com/{z}/{x}/{y}.png"
|
||||||
|
}
|
||||||
|
|
||||||
|
result = await _get_wizard_redirect_step(mock_conn)
|
||||||
|
assert result == "/setup/keys"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_all_steps_complete_redirects_to_finish(self):
|
||||||
|
"""When all steps done, redirect to /setup/finish."""
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchval.side_effect = [1, 1] # Has operator, adapters touched
|
||||||
|
mock_conn.fetchrow.return_value = {
|
||||||
|
"map_tile_url": "https://custom.example.com/{z}/{x}/{y}.png"
|
||||||
|
}
|
||||||
|
|
||||||
|
result = await _get_wizard_redirect_step(mock_conn)
|
||||||
|
assert result == "/setup/finish"
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupOperatorForm:
|
||||||
|
"""Test operator creation form (step 1)."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_get_returns_form(self):
|
||||||
|
"""GET /setup/operator returns the form."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
|
||||||
|
mock_templates = MagicMock()
|
||||||
|
mock_templates.TemplateResponse.return_value = MagicMock()
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.generate_csrf_tokens.return_value = ("token", "signed")
|
||||||
|
mock_csrf.set_csrf_cookie = MagicMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes._get_templates", return_value=mock_templates):
|
||||||
|
result = await setup_operator_form(mock_request, mock_csrf)
|
||||||
|
|
||||||
|
mock_templates.TemplateResponse.assert_called_once()
|
||||||
|
call_args = mock_templates.TemplateResponse.call_args
|
||||||
|
context = call_args.kwargs.get("context", call_args[1].get("context"))
|
||||||
|
assert context["csrf_token"] == "token"
|
||||||
|
assert context["error"] is None
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupOperatorSubmit:
|
||||||
|
"""Test operator creation submission."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_password_mismatch_shows_error(self):
|
||||||
|
"""POST with password mismatch re-renders with error."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_templates = MagicMock()
|
||||||
|
mock_templates.TemplateResponse.return_value = MagicMock()
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.validate_csrf = AsyncMock()
|
||||||
|
mock_csrf.generate_csrf_tokens.return_value = ("token", "signed")
|
||||||
|
mock_csrf.set_csrf_cookie = MagicMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes._get_templates", return_value=mock_templates):
|
||||||
|
with patch("central.gui.routes.get_pool", return_value=mock_pool):
|
||||||
|
result = await setup_operator_submit(
|
||||||
|
mock_request,
|
||||||
|
username="admin",
|
||||||
|
password="password123",
|
||||||
|
confirm_password="different",
|
||||||
|
csrf_protect=mock_csrf,
|
||||||
|
)
|
||||||
|
|
||||||
|
call_args = mock_templates.TemplateResponse.call_args
|
||||||
|
context = call_args.kwargs.get("context", call_args[1].get("context"))
|
||||||
|
assert context["error"] == "Passwords do not match"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_valid_creates_operator_and_redirects(self):
|
||||||
|
"""POST with valid data creates operator and redirects to /setup/system."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchrow.side_effect = [
|
||||||
|
{"id": 1}, # INSERT RETURNING id
|
||||||
|
{"session_lifetime_days": 90}, # system settings
|
||||||
|
]
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_pool.acquire.return_value.__aenter__.return_value = mock_conn
|
||||||
|
mock_pool.acquire.return_value.__aexit__.return_value = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.validate_csrf = AsyncMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes.get_pool", return_value=mock_pool):
|
||||||
|
with patch("central.gui.routes.hash_password", return_value="hashed"):
|
||||||
|
with patch("central.gui.routes.create_session", new_callable=AsyncMock) as mock_session:
|
||||||
|
mock_session.return_value = ("session_token", datetime.now())
|
||||||
|
with patch("central.gui.routes.write_audit", new_callable=AsyncMock):
|
||||||
|
result = await setup_operator_submit(
|
||||||
|
mock_request,
|
||||||
|
username="admin",
|
||||||
|
password="password123",
|
||||||
|
confirm_password="password123",
|
||||||
|
csrf_protect=mock_csrf,
|
||||||
|
)
|
||||||
|
|
||||||
|
assert result.status_code == 302
|
||||||
|
assert result.headers["location"] == "/setup/system"
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupSystemForm:
|
||||||
|
"""Test system settings form (step 2)."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_unauthenticated_redirects_to_operator(self):
|
||||||
|
"""GET /setup/system without auth redirects to /setup/operator."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
|
||||||
|
result = await setup_system_form(mock_request, mock_csrf)
|
||||||
|
assert result.status_code == 302
|
||||||
|
assert result.headers["location"] == "/setup/operator"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_authenticated_returns_form(self):
|
||||||
|
"""GET /setup/system with auth returns the form."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = MagicMock(id=1, username="admin")
|
||||||
|
|
||||||
|
mock_templates = MagicMock()
|
||||||
|
mock_templates.TemplateResponse.return_value = MagicMock()
|
||||||
|
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchrow.return_value = {
|
||||||
|
"map_tile_url": "https://tile.openstreetmap.org/{z}/{x}/{y}.png",
|
||||||
|
"map_attribution": "© OpenStreetMap contributors",
|
||||||
|
}
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_pool.acquire.return_value.__aenter__.return_value = mock_conn
|
||||||
|
mock_pool.acquire.return_value.__aexit__.return_value = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.generate_csrf_tokens.return_value = ("token", "signed")
|
||||||
|
mock_csrf.set_csrf_cookie = MagicMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes._get_templates", return_value=mock_templates):
|
||||||
|
with patch("central.gui.routes.get_pool", return_value=mock_pool):
|
||||||
|
result = await setup_system_form(mock_request, mock_csrf)
|
||||||
|
|
||||||
|
mock_templates.TemplateResponse.assert_called_once()
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupSystemSubmit:
|
||||||
|
"""Test system settings submission."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_missing_placeholders_shows_error(self):
|
||||||
|
"""POST without {z},{x},{y} placeholders shows error."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = MagicMock(id=1, username="admin")
|
||||||
|
|
||||||
|
form_data = MagicMock()
|
||||||
|
form_data.get = lambda k, default="": {
|
||||||
|
"map_tile_url": "https://example.com/tiles",
|
||||||
|
"map_attribution": "Test",
|
||||||
|
}.get(k, default)
|
||||||
|
mock_request.form = AsyncMock(return_value=form_data)
|
||||||
|
|
||||||
|
mock_templates = MagicMock()
|
||||||
|
mock_templates.TemplateResponse.return_value = MagicMock()
|
||||||
|
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchrow.return_value = {
|
||||||
|
"map_tile_url": "",
|
||||||
|
"map_attribution": "",
|
||||||
|
}
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_pool.acquire.return_value.__aenter__.return_value = mock_conn
|
||||||
|
mock_pool.acquire.return_value.__aexit__.return_value = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.validate_csrf = AsyncMock()
|
||||||
|
mock_csrf.generate_csrf_tokens.return_value = ("token", "signed")
|
||||||
|
mock_csrf.set_csrf_cookie = MagicMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes._get_templates", return_value=mock_templates):
|
||||||
|
with patch("central.gui.routes.get_pool", return_value=mock_pool):
|
||||||
|
result = await setup_system_submit(mock_request, mock_csrf)
|
||||||
|
|
||||||
|
call_args = mock_templates.TemplateResponse.call_args
|
||||||
|
context = call_args.kwargs.get("context", call_args[1].get("context"))
|
||||||
|
assert "map_tile_url" in context["errors"]
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_valid_updates_and_redirects(self):
|
||||||
|
"""POST with valid data updates system and redirects to /setup/keys."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = MagicMock(id=1, username="admin")
|
||||||
|
|
||||||
|
form_data = MagicMock()
|
||||||
|
form_data.get = lambda k, default="": {
|
||||||
|
"map_tile_url": "https://example.com/{z}/{x}/{y}.png",
|
||||||
|
"map_attribution": "Test Attribution",
|
||||||
|
}.get(k, default)
|
||||||
|
mock_request.form = AsyncMock(return_value=form_data)
|
||||||
|
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchrow.return_value = {
|
||||||
|
"map_tile_url": "old_url",
|
||||||
|
"map_attribution": "old_attr",
|
||||||
|
}
|
||||||
|
mock_conn.execute = AsyncMock()
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_pool.acquire.return_value.__aenter__.return_value = mock_conn
|
||||||
|
mock_pool.acquire.return_value.__aexit__.return_value = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.validate_csrf = AsyncMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes.get_pool", return_value=mock_pool):
|
||||||
|
with patch("central.gui.routes.write_audit", new_callable=AsyncMock):
|
||||||
|
result = await setup_system_submit(mock_request, mock_csrf)
|
||||||
|
|
||||||
|
assert result.status_code == 302
|
||||||
|
assert result.headers["location"] == "/setup/keys"
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupKeysForm:
|
||||||
|
"""Test API keys form (step 3)."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_unauthenticated_redirects_to_operator(self):
|
||||||
|
"""GET /setup/keys without auth redirects to /setup/operator."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
|
||||||
|
result = await setup_keys_form(mock_request, mock_csrf)
|
||||||
|
assert result.status_code == 302
|
||||||
|
assert result.headers["location"] == "/setup/operator"
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupKeysSubmit:
|
||||||
|
"""Test API keys submission."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_next_action_redirects_to_adapters(self):
|
||||||
|
"""POST with action=next redirects to /setup/adapters."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = MagicMock(id=1, username="admin")
|
||||||
|
|
||||||
|
form_data = MagicMock()
|
||||||
|
form_data.get = lambda k, default="": {"action": "next"}.get(k, default)
|
||||||
|
mock_request.form = AsyncMock(return_value=form_data)
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.validate_csrf = AsyncMock()
|
||||||
|
|
||||||
|
# No need to mock get_pool since action="next" returns before it's called
|
||||||
|
result = await setup_keys_submit(mock_request, mock_csrf)
|
||||||
|
assert result.status_code == 302
|
||||||
|
assert result.headers["location"] == "/setup/adapters"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_add_key_creates_and_rerenders(self):
|
||||||
|
"""POST with action=add creates key and re-renders with success."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = MagicMock(id=1, username="admin")
|
||||||
|
|
||||||
|
form_data = MagicMock()
|
||||||
|
form_data.get = lambda k, default="": {
|
||||||
|
"action": "add",
|
||||||
|
"alias": "testkey",
|
||||||
|
"plaintext_key": "secret123",
|
||||||
|
}.get(k, default)
|
||||||
|
mock_request.form = AsyncMock(return_value=form_data)
|
||||||
|
|
||||||
|
mock_templates = MagicMock()
|
||||||
|
mock_templates.TemplateResponse.return_value = MagicMock()
|
||||||
|
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchrow.side_effect = [
|
||||||
|
None, # No existing key
|
||||||
|
{"created_at": datetime(2026, 5, 18, 12, 0, tzinfo=timezone.utc)},
|
||||||
|
]
|
||||||
|
mock_conn.fetch.side_effect = [
|
||||||
|
[], # First list
|
||||||
|
[{"alias": "testkey", "created_at": datetime(2026, 5, 18, 12, 0, tzinfo=timezone.utc)}], # After insert
|
||||||
|
]
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_pool.acquire.return_value.__aenter__.return_value = mock_conn
|
||||||
|
mock_pool.acquire.return_value.__aexit__.return_value = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.validate_csrf = AsyncMock()
|
||||||
|
mock_csrf.generate_csrf_tokens.return_value = ("token", "signed")
|
||||||
|
mock_csrf.set_csrf_cookie = MagicMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes._get_templates", return_value=mock_templates):
|
||||||
|
with patch("central.gui.routes.get_pool", return_value=mock_pool):
|
||||||
|
with patch("central.crypto.encrypt", return_value=b"encrypted"):
|
||||||
|
with patch("central.gui.routes.write_audit", new_callable=AsyncMock):
|
||||||
|
result = await setup_keys_submit(mock_request, mock_csrf)
|
||||||
|
|
||||||
|
call_args = mock_templates.TemplateResponse.call_args
|
||||||
|
context = call_args.kwargs.get("context", call_args[1].get("context"))
|
||||||
|
assert context["success"] == "API key 'testkey' added successfully."
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupAdaptersForm:
|
||||||
|
"""Test adapters configuration form (step 4)."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_unauthenticated_redirects_to_operator(self):
|
||||||
|
"""GET /setup/adapters without auth redirects to /setup/operator."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
|
||||||
|
result = await setup_adapters_form(mock_request, mock_csrf)
|
||||||
|
assert result.status_code == 302
|
||||||
|
assert result.headers["location"] == "/setup/operator"
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupFinishForm:
|
||||||
|
"""Test finish page (step 5)."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_unauthenticated_redirects_to_operator(self):
|
||||||
|
"""GET /setup/finish without auth redirects to /setup/operator."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
|
||||||
|
result = await setup_finish_form(mock_request, mock_csrf)
|
||||||
|
assert result.status_code == 302
|
||||||
|
assert result.headers["location"] == "/setup/operator"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_authenticated_shows_summary(self):
|
||||||
|
"""GET /setup/finish with auth shows summary."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = MagicMock(id=1, username="admin")
|
||||||
|
|
||||||
|
mock_templates = MagicMock()
|
||||||
|
mock_templates.TemplateResponse.return_value = MagicMock()
|
||||||
|
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.fetchval.side_effect = [1, 2] # 1 operator, 2 keys
|
||||||
|
mock_conn.fetchrow.return_value = {"map_tile_url": "https://example.com/{z}/{x}/{y}.png"}
|
||||||
|
mock_conn.fetch.return_value = [
|
||||||
|
{"name": "nws", "enabled": True, "cadence_s": 300},
|
||||||
|
{"name": "firms", "enabled": False, "cadence_s": 600},
|
||||||
|
]
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_pool.acquire.return_value.__aenter__.return_value = mock_conn
|
||||||
|
mock_pool.acquire.return_value.__aexit__.return_value = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.generate_csrf_tokens.return_value = ("token", "signed")
|
||||||
|
mock_csrf.set_csrf_cookie = MagicMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes._get_templates", return_value=mock_templates):
|
||||||
|
with patch("central.gui.routes.get_pool", return_value=mock_pool):
|
||||||
|
result = await setup_finish_form(mock_request, mock_csrf)
|
||||||
|
|
||||||
|
call_args = mock_templates.TemplateResponse.call_args
|
||||||
|
context = call_args.kwargs.get("context", call_args[1].get("context"))
|
||||||
|
assert context["operator_count"] == 1
|
||||||
|
assert context["key_count"] == 2
|
||||||
|
assert len(context["adapters"]) == 2
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupFinishSubmit:
|
||||||
|
"""Test setup completion."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_marks_setup_complete_and_redirects(self):
|
||||||
|
"""POST /setup/finish marks setup_complete=true and redirects to /."""
|
||||||
|
mock_request = MagicMock()
|
||||||
|
mock_request.state.operator = MagicMock(id=1, username="admin")
|
||||||
|
|
||||||
|
mock_conn = AsyncMock()
|
||||||
|
mock_conn.execute = AsyncMock()
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_pool.acquire.return_value.__aenter__.return_value = mock_conn
|
||||||
|
mock_pool.acquire.return_value.__aexit__.return_value = None
|
||||||
|
|
||||||
|
mock_csrf = MagicMock()
|
||||||
|
mock_csrf.validate_csrf = AsyncMock()
|
||||||
|
|
||||||
|
with patch("central.gui.routes.get_pool", return_value=mock_pool):
|
||||||
|
with patch("central.gui.routes.write_audit", new_callable=AsyncMock) as mock_audit:
|
||||||
|
result = await setup_finish_submit(mock_request, mock_csrf)
|
||||||
|
|
||||||
|
assert result.status_code == 302
|
||||||
|
assert result.headers["location"] == "/"
|
||||||
|
mock_conn.execute.assert_called_once()
|
||||||
|
mock_audit.assert_called_once()
|
||||||
|
|
||||||
|
|
||||||
|
class TestSetupGateMiddlewareWizard:
|
||||||
|
"""Test SetupGateMiddleware with wizard paths."""
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_allows_setup_operator_when_incomplete(self):
|
||||||
|
"""SetupGateMiddleware allows /setup/operator when setup_complete=False."""
|
||||||
|
from starlette.testclient import TestClient
|
||||||
|
from fastapi import FastAPI
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_conn = MagicMock()
|
||||||
|
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": False})
|
||||||
|
mock_conn.__aenter__ = AsyncMock(return_value=mock_conn)
|
||||||
|
mock_conn.__aexit__ = AsyncMock()
|
||||||
|
mock_pool.acquire = MagicMock(return_value=mock_conn)
|
||||||
|
|
||||||
|
with patch("central.gui.middleware.get_pool", return_value=mock_pool):
|
||||||
|
app = FastAPI()
|
||||||
|
|
||||||
|
@app.get("/setup/operator")
|
||||||
|
async def setup_operator():
|
||||||
|
return {"message": "operator form"}
|
||||||
|
|
||||||
|
app.add_middleware(SetupGateMiddleware)
|
||||||
|
client = TestClient(app)
|
||||||
|
|
||||||
|
response = client.get("/setup/operator")
|
||||||
|
assert response.status_code == 200
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_redirects_base_setup_to_wizard_step(self):
|
||||||
|
"""SetupGateMiddleware redirects /setup to appropriate wizard step."""
|
||||||
|
from starlette.testclient import TestClient
|
||||||
|
from fastapi import FastAPI
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_conn = MagicMock()
|
||||||
|
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": False})
|
||||||
|
mock_conn.fetchval = AsyncMock(return_value=0) # No operators
|
||||||
|
mock_conn.__aenter__ = AsyncMock(return_value=mock_conn)
|
||||||
|
mock_conn.__aexit__ = AsyncMock()
|
||||||
|
mock_pool.acquire = MagicMock(return_value=mock_conn)
|
||||||
|
|
||||||
|
with patch("central.gui.middleware.get_pool", return_value=mock_pool):
|
||||||
|
app = FastAPI()
|
||||||
|
|
||||||
|
@app.get("/setup")
|
||||||
|
async def setup():
|
||||||
|
return {"message": "base setup"}
|
||||||
|
|
||||||
|
@app.get("/setup/operator")
|
||||||
|
async def setup_operator():
|
||||||
|
return {"message": "operator"}
|
||||||
|
|
||||||
|
app.add_middleware(SetupGateMiddleware)
|
||||||
|
client = TestClient(app, follow_redirects=False)
|
||||||
|
|
||||||
|
response = client.get("/setup")
|
||||||
|
assert response.status_code == 302
|
||||||
|
assert response.headers["location"] == "/setup/operator"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_redirects_login_to_setup_when_incomplete(self):
|
||||||
|
"""SetupGateMiddleware redirects /login to /setup when setup_complete=False."""
|
||||||
|
from starlette.testclient import TestClient
|
||||||
|
from fastapi import FastAPI
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_conn = MagicMock()
|
||||||
|
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": False})
|
||||||
|
mock_conn.__aenter__ = AsyncMock(return_value=mock_conn)
|
||||||
|
mock_conn.__aexit__ = AsyncMock()
|
||||||
|
mock_pool.acquire = MagicMock(return_value=mock_conn)
|
||||||
|
|
||||||
|
with patch("central.gui.middleware.get_pool", return_value=mock_pool):
|
||||||
|
app = FastAPI()
|
||||||
|
|
||||||
|
@app.get("/login")
|
||||||
|
async def login():
|
||||||
|
return {"message": "login"}
|
||||||
|
|
||||||
|
@app.get("/setup")
|
||||||
|
async def setup():
|
||||||
|
return {"message": "setup"}
|
||||||
|
|
||||||
|
app.add_middleware(SetupGateMiddleware)
|
||||||
|
client = TestClient(app, follow_redirects=False)
|
||||||
|
|
||||||
|
response = client.get("/login")
|
||||||
|
assert response.status_code == 302
|
||||||
|
assert response.headers["location"] == "/setup"
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_redirects_all_setup_paths_when_complete(self):
|
||||||
|
"""SetupGateMiddleware redirects /setup/* to / when setup_complete=True."""
|
||||||
|
from starlette.testclient import TestClient
|
||||||
|
from fastapi import FastAPI
|
||||||
|
|
||||||
|
mock_pool = MagicMock()
|
||||||
|
mock_conn = MagicMock()
|
||||||
|
mock_conn.fetchrow = AsyncMock(return_value={"setup_complete": True})
|
||||||
|
mock_conn.__aenter__ = AsyncMock(return_value=mock_conn)
|
||||||
|
mock_conn.__aexit__ = AsyncMock()
|
||||||
|
mock_pool.acquire = MagicMock(return_value=mock_conn)
|
||||||
|
|
||||||
|
with patch("central.gui.middleware.get_pool", return_value=mock_pool):
|
||||||
|
app = FastAPI()
|
||||||
|
|
||||||
|
@app.get("/")
|
||||||
|
async def index():
|
||||||
|
return {"message": "home"}
|
||||||
|
|
||||||
|
@app.get("/setup/operator")
|
||||||
|
async def setup_operator():
|
||||||
|
return {"message": "operator"}
|
||||||
|
|
||||||
|
app.add_middleware(SetupGateMiddleware)
|
||||||
|
client = TestClient(app, follow_redirects=False)
|
||||||
|
|
||||||
|
response = client.get("/setup/operator")
|
||||||
|
assert response.status_code == 302
|
||||||
|
assert response.headers["location"] == "/"
|
||||||
Loading…
Add table
Add a link
Reference in a new issue