"""
对话相关 API 路由
"""
from datetime import datetime, timezone
from typing import List, Optional
from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select
import uuid

from database import get_async_db, Conversation, Segment, User
from database.models import Conversation as ConversationModel, Segment as SegmentModel

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


@router.post("")
async def create_conversation(
    user_id: str,
    db: AsyncSession = Depends(get_async_db)
):
    """创建新对话"""
    conversation = ConversationModel(
        id=str(uuid.uuid4()),
        user_id=user_id,
        started_at=datetime.now(timezone.utc),
        status="active"
    )
    db.add(conversation)
    await db.commit()
    await db.refresh(conversation)
    
    return {
        "id": conversation.id,
        "user_id": conversation.user_id,
        "started_at": conversation.started_at.isoformat(),
        "status": conversation.status
    }


@router.get("/{conversation_id}")
async def get_conversation(
    conversation_id: str,
    db: AsyncSession = Depends(get_async_db)
):
    """获取对话详情"""
    conversation = await db.get(ConversationModel, conversation_id)
    if not conversation:
        raise HTTPException(status_code=404, detail="Conversation not found")
    
    return {
        "id": conversation.id,
        "user_id": conversation.user_id,
        "started_at": conversation.started_at.isoformat(),
        "ended_at": conversation.ended_at.isoformat() if conversation.ended_at else None,
        "duration_seconds": conversation.duration_seconds,
        "summary": conversation.summary,
        "status": conversation.status,
        "current_topic": conversation.current_topic,
        "conversation_stage": conversation.conversation_stage
    }


@router.post("/{conversation_id}/end")
async def end_conversation(
    conversation_id: str,
    db: AsyncSession = Depends(get_async_db)
):
    """结束对话"""
    conversation = await db.get(ConversationModel, conversation_id)
    if not conversation:
        raise HTTPException(status_code=404, detail="Conversation not found")
    
    conversation.status = "ended"
    conversation.ended_at = datetime.now(timezone.utc)
    
    if conversation.started_at:
        duration = (conversation.ended_at - conversation.started_at).total_seconds()
        conversation.duration_seconds = int(duration)
    
    await db.commit()
    
    return {
        "id": conversation.id,
        "status": conversation.status,
        "ended_at": conversation.ended_at.isoformat(),
        "duration_seconds": conversation.duration_seconds
    }