admin / Syanpse-Vanguard

"""Shared startup/shutdown for every service role. Builds the storage engine and
broker, initializes schema, seeds the bootstrap admin, and ensures the consumer
group exists."""
from __future__ import annotations

import logging

from sv.auth.users import seed_bootstrap_admin
from sv.broker import Broker, build_broker
from sv.config import settings
from sv.storage.base import StorageEngine, build_engine

log = logging.getLogger("sv.bootstrap")


async def startup(with_broker: bool = True) -> tuple[StorageEngine, Broker | None]:
    logging.basicConfig(level=settings.log_level)
    engine = build_engine()
    await engine.init_schema()
    await seed_bootstrap_admin(engine)
    broker: Broker | None = None
    if with_broker:
        broker = build_broker()
        await broker.ensure_group()
    log.info("startup complete (storage=%s broker=%s)",
             settings.storage_backend, settings.broker if with_broker else "none")
    return engine, broker


async def shutdown(engine: StorageEngine, broker: Broker | None) -> None:
    if broker is not None:
        await broker.close()
    await engine.close()