ai-rpg/backend/app/api/worlds.py

"""Worlds routes: CRUD + world builder flow."""
from __future__ import annotations

from typing import List
from uuid import UUID

from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from fastapi import APIRouter, Depends, HTTPException

from app.db import get_db_dep
from app.deps import get_current_user
from app.engine.world_builder import commit_world_builder, continue_world_builder, start_world_builder
from app.models import User, World
from app.schemas import (
    WorldBuilderCommit,
    WorldBuilderMessage,
    WorldBuilderReply,
    WorldBuilderStart,
    WorldCreate,
    WorldOut,
    WorldUpdate,
)

router = APIRouter(prefix="/api/worlds", tags=["worlds"])


@router.get("", response_model=List[WorldOut])
async def list_worlds(
    db: AsyncSession = Depends(get_db_dep),
    user: User = Depends(get_current_user),
):
    result = await db.execute(
        select(World).where(World.owner_id == user.id).order_by(World.updated_at.desc())
    )
    return result.scalars().all()


@router.get("/{world_id}", response_model=WorldOut)
async def get_world(
    world_id: UUID,
    db: AsyncSession = Depends(get_db_dep),
    user: User = Depends(get_current_user),
):
    result = await db.execute(select(World).where(World.id == world_id))
    world = result.scalars().first()
    if not world:
        raise HTTPException(status_code=404, detail="world_not_found")
    if world.owner_id != user.id and not user.is_admin:
        raise HTTPException(status_code=403, detail="forbidden")
    return world


@router.post("", response_model=WorldOut, status_code=201)
async def create_world(
    payload: WorldCreate,
    db: AsyncSession = Depends(get_db_dep),
    user: User = Depends(get_current_user),
):
    world = World(
        owner_id=user.id,
        name=payload.name,
        language=payload.language,
        definition={},
        state={},
        status="draft",
        preset_id=payload.preset_id,
    )
    db.add(world)
    await db.commit()
    await db.refresh(world)
    return world


@router.patch("/{world_id}", response_model=WorldOut)
async def update_world(
    world_id: UUID,
    payload: WorldUpdate,
    db: AsyncSession = Depends(get_db_dep),
    user: User = Depends(get_current_user),
):
    result = await db.execute(select(World).where(World.id == world_id))
    world = result.scalars().first()
    if not world:
        raise HTTPException(status_code=404, detail="world_not_found")
    if world.owner_id != user.id and not user.is_admin:
        raise HTTPException(status_code=403, detail="forbidden")
    for field, value in payload.model_dump(exclude_unset=True).items():
        setattr(world, field, value)
    await db.commit()
    await db.refresh(world)
    return world


@router.delete("/{world_id}", status_code=204)
async def delete_world(
    world_id: UUID,
    db: AsyncSession = Depends(get_db_dep),
    user: User = Depends(get_current_user),
):
    result = await db.execute(select(World).where(World.id == world_id))
    world = result.scalars().first()
    if not world:
        raise HTTPException(status_code=404, detail="world_not_found")
    if world.owner_id != user.id and not user.is_admin:
        raise HTTPException(status_code=403, detail="forbidden")
    await db.delete(world)
    await db.commit()


# === World Builder flow ===

@router.post("/builder/start", response_model=WorldBuilderReply)
async def builder_start(
    payload: WorldBuilderStart,
    db: AsyncSession = Depends(get_db_dep),
    user: User = Depends(get_current_user),
):
    try:
        return await start_world_builder(
            db=db,
            user=user,
            world_name=payload.world_name,
            language=payload.language,
            preset_id=payload.preset_id,
            setting_brief=payload.setting_brief,
            character_brief=payload.character_brief,
            rules_brief=payload.rules_brief,
            notes=payload.notes,
        )
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"builder_start_failed: {e}")


@router.post("/builder/continue", response_model=WorldBuilderReply)
async def builder_continue(
    payload: WorldBuilderMessage,
    db: AsyncSession = Depends(get_db_dep),
    user: User = Depends(get_current_user),
):
    try:
        return await continue_world_builder(db=db, user=user, session_id=payload.session_id, user_message=payload.message)
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"builder_continue_failed: {e}")


@router.post("/builder/commit", response_model=WorldOut)
async def builder_commit(
    payload: WorldBuilderCommit,
    db: AsyncSession = Depends(get_db_dep),
    user: User = Depends(get_current_user),
):
    try:
        return await commit_world_builder(db=db, user=user, session_id=payload.session_id, name=payload.name)
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"builder_commit_failed: {e}")