ai-rpg/app/main.py

"""FastAPI app factory and lifespan (DB + Qdrant init, settings seed)."""

from __future__ import annotations

import os
from contextlib import asynccontextmanager
from collections.abc import AsyncIterator
from pathlib import Path

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from sqlalchemy import select, text

from app import __version__
from app.api import admin, auth, misc, presets, sessions, worlds
from app.config import get_settings
from app.core.logging import configure_logging, get_logger
from app.core.qdrant_client import dispose_qdrant_client, init_qdrant_collections
from app.core.settings_service import (
    get_admin_setup_token,
    seed_default_settings,
)
from app.db import dispose_engine, get_sessionmaker
from app.models import Setting, User

_logger = get_logger(__name__)


async def _apply_schema_fixups(engine) -> None:
    """Apply idempotent ALTER statements for schema fixes that create_all
    cannot handle (e.g. changing NOT NULL → NULL on existing tables).

    These run on every startup. Each statement is wrapped in try/except so
    it silently succeeds if the fix was already applied.
    """
    fixups = [
        # world_presets.owner_id: was NOT NULL, now nullable (for system presets)
        "ALTER TABLE world_presets ALTER COLUMN owner_id DROP NOT NULL",
    ]
    async with engine.begin() as conn:
        for sql in fixups:
            try:
                await conn.execute(text(sql))
            except Exception:  # noqa: BLE001
                pass  # already applied, or table doesn't exist yet


@asynccontextmanager
async def lifespan(app: FastAPI) -> AsyncIterator[None]:
    """Application startup / shutdown lifecycle."""
    cfg = get_settings()
    configure_logging()
    _logger.info("app_starting", version=__version__, debug=cfg.debug)

    # Startup -------------------------------------------------------------
    # Step 0: Create all DB tables (idempotent — equivalent to alembic upgrade head).
    # This must run BEFORE any settings query, otherwise seed_default_settings
    # crashes with "relation 'settings' does not exist".
    from app.db import get_engine
    from app.migrations.versions._001_initial_schema import create_all_tables

    try:
        engine = get_engine()
        await create_all_tables(engine)
        await _apply_schema_fixups(engine)
        _logger.info("db_tables_ready")
    except Exception as e:  # noqa: BLE001
        _logger.error("db_tables_create_failed", error=str(e))

    sm = get_sessionmaker()
    try:
        async with sm() as session:
            # 1) seed settings (idempotent)
            await seed_default_settings(session)
            # 2) ensure admin setup token
            token = await get_admin_setup_token(session)
            # 3) check if any admin exists — ALWAYS print the setup URL
            #    (user requested: even if admin exists, show the token for
            #    reference / debugging)
            from sqlalchemy import func

            admins_count = (
                await session.execute(
                    select(func.count(User.id)).where(User.is_admin.is_(True))
                )
            ).scalar_one()
            if admins_count == 0:
                _logger.warning("no_admin_yet", setup_url=f"/register/admin?token={token}")
                print(f"\n{'=' * 60}")
                print(f"  AI-RPG — No admin user yet.")
                print(f"  Open this URL to create the first admin:")
                print(f"    http://localhost:8080/register/admin?token={token}")
                print(f"{'=' * 60}\n")
            else:
                _logger.info("admins_present", count=admins_count)
                # Still print the token URL for reference
                print(f"\n  Admin already exists. Admin register URL (for reference):")
                print(f"    /register/admin?token={token}")
                print(f"  (This URL is blocked since an admin already exists.)\n")
            # 4) seed builtin presets (idempotent)
            from app.migrations.seed import seed_builtin_presets

            try:
                await seed_builtin_presets(session)
                _logger.info("presets_seeded")
            except Exception as e:  # noqa: BLE001
                _logger.warning("presets_seed_failed", error=str(e))
            # 5) Init Qdrant collections
            from app.core.settings_service import get_setting

            dim = int(await get_setting(session, "embeddings.dimension") or 256)
            try:
                await init_qdrant_collections(dim)
                _logger.info("qdrant_collections_ready", dimension=dim)
            except Exception as e:  # noqa: BLE001
                _logger.warning("qdrant_init_failed", error=str(e))
    except Exception as e:  # noqa: BLE001
        _logger.error("startup_failed", error=str(e))
        pass

    yield

    # Shutdown ------------------------------------------------------------
    _logger.info("app_stopping")
    await dispose_qdrant_client()
    await dispose_engine()


def create_app() -> FastAPI:
    """Build and return the FastAPI application."""
    cfg = get_settings()
    app = FastAPI(
        title=cfg.app_name,
        version=__version__,
        description="AI-RPG — text RPG with an LLM Game Master.",
        lifespan=lifespan,
        docs_url="/api/docs",
        redoc_url=None,
        openapi_url="/api/openapi.json",
    )

    # CORS
    app.add_middleware(
        CORSMiddleware,
        allow_origins=cfg.cors_origins_list,
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )

    # Static asset serving (icons / uploads)
    assets_dir = Path(cfg.assets_dir)
    assets_dir.mkdir(parents=True, exist_ok=True)
    app.mount("/static/assets", StaticFiles(directory=str(assets_dir)), name="assets")

    # Routers
    app.include_router(misc.router)
    app.include_router(auth.router)
    app.include_router(worlds.router)
    app.include_router(sessions.router)
    app.include_router(presets.router)
    app.include_router(admin.router)

    return app


app = create_app()