src/api/v1/room.py

"""Room management API endpoints."""

from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select
from sqlalchemy.orm import selectinload
from src.db.postgres.connection import get_db
from src.db.postgres.models import Room, ChatMessage, MessageSource
from src.middlewares.logging import get_logger, log_execution
from pydantic import BaseModel
from typing import List, Optional
from datetime import datetime
import uuid

logger = get_logger("room_api")

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


class MessageSourceResponse(BaseModel):
    document_id: Optional[str]
    filename: Optional[str]
    page_label: Optional[str]


class ChatMessageResponse(BaseModel):
    id: str
    role: str
    content: str
    created_at: str
    sources: List[MessageSourceResponse] = []


class RoomResponse(BaseModel):
    id: str
    title: str
    created_at: str
    updated_at: str | None
    messages: List[ChatMessageResponse] = []


class CreateRoomRequest(BaseModel):
    user_id: str
    title: str = "New Chat"


@router.get("/rooms/{user_id}", response_model=List[RoomResponse])
@log_execution(logger)
async def list_rooms(
    user_id: str,
    db: AsyncSession = Depends(get_db)
):
    """List all rooms for a user."""
    result = await db.execute(
        select(Room)
        .where(Room.user_id == user_id, Room.status == "active")
        .order_by(Room.updated_at.desc())
    )
    rooms = result.scalars().all()

    return [
        RoomResponse(
            id=room.id,
            title=room.title,
            created_at=room.created_at.isoformat(),
            updated_at=room.updated_at.isoformat() if room.updated_at else None
        )
        for room in rooms
    ]


@router.get("/room/{room_id}", response_model=RoomResponse)
@log_execution(logger)
async def get_room(
    room_id: str,
    db: AsyncSession = Depends(get_db)
):
    """Get a specific room with its chat history."""
    result = await db.execute(
        select(Room)
        .where(Room.id == room_id)
        .options(selectinload(Room.messages).selectinload(ChatMessage.sources))
    )
    room = result.scalars().first()

    if not room:
        raise HTTPException(
            status_code=404,
            detail="Room not found"
        )

    messages = sorted(room.messages, key=lambda m: m.created_at)

    return RoomResponse(
        id=room.id,
        title=room.title,
        created_at=room.created_at.isoformat(),
        updated_at=room.updated_at.isoformat() if room.updated_at else None,
        messages=[
            ChatMessageResponse(
                id=msg.id,
                role=msg.role,
                content=msg.content,
                created_at=msg.created_at.isoformat(),
                sources=[
                    MessageSourceResponse(
                        document_id=src.document_id,
                        filename=src.filename,
                        page_label=src.page_label,
                    )
                    for src in msg.sources
                ],
            )
            for msg in messages
        ]
    )


@router.delete("/room/{room_id}")
@log_execution(logger)
async def delete_room(
    room_id: str,
    user_id: str,
    db: AsyncSession = Depends(get_db)
):
    """Soft-delete a room by setting its status to inactive."""
    result = await db.execute(
        select(Room).where(Room.id == room_id)
    )
    room = result.scalars().first()

    if not room:
        raise HTTPException(status_code=404, detail="Room not found")

    if room.user_id != user_id:
        raise HTTPException(status_code=403, detail="Access denied")

    room.status = "inactive"
    await db.commit()

    return {"status": "success", "message": "Room deleted successfully"}


@router.post("/room/create")
@log_execution(logger)
async def create_room(
    request: CreateRoomRequest,
    db: AsyncSession = Depends(get_db)
):
    """Create a new room."""
    room = Room(
        id=str(uuid.uuid4()),
        user_id=request.user_id,
        title=request.title
    )
    db.add(room)
    await db.commit()
    await db.refresh(room)

    return {
        "status": "success",
        "message": "Room created successfully",
        "data": RoomResponse(
            id=room.id,
            title=room.title,
            created_at=room.created_at.isoformat(),
            updated_at=None
        )
    }