"""会话及对话轮次，用于多轮上下文和后续持久化"""

from datetime import datetime
from typing import Annotated, Literal

from pydantic import BaseModel, Field, StringConstraints

from .base import JsonObject, NonBlankString

Title = Annotated[
    str, StringConstraints(strip_whitespace=True, min_length=1, max_length=100)
]


class TurnChange(BaseModel):
    """本轮交付的一处修改，同一组件的多个动作合并成一条；记的是交付了，不是前端已执行"""

    node_id: str  # 画布写 "canvas"
    type: str  # 组件类型，画布写 "canvas"
    title: str | None = None  # 交付时画面上的名字或内容，删除的组件取原始大屏上的
    change: Literal["added", "modified", "deleted"]


class TurnQuery(BaseModel):
    """问答里成功执行的一次查询：只存 SQL 和结果的形状，不存结果"""

    sql: str
    columns: list[str]
    rows: int  # 返回的行数
    truncated: bool  # 结果超过返回上限，只返回了一部分


class Turn(BaseModel):
    """一次任务对应一轮对话，同一个 thread_id 多次写入时覆盖同一轮"""

    conversation_id: str
    thread_id: str
    message_id: str
    prompt: str  # 用户原始输入
    outcome: Literal["completed", "waiting_for_input", "failed", "cancelled"]
    summary: str  # 助手一侧的完整回复（回答、追问或失败说明），不截断；不代表前端已执行
    created_at: datetime
    # 轮次记录：由代码从运行结果推出，不经过模型；只有完成的轮次才有 changes、queries
    changes: list[TurnChange] = Field(default_factory=list)  # 交付的修改
    queries: list[TurnQuery] = Field(default_factory=list)  # 问答查询，按执行顺序
    # 本任务里追问过的问题和用户的回答（含选中的选项）；恢复后整轮被覆盖，问答记在这里
    clarifications: list[JsonObject] = Field(default_factory=list)


class ConversationCreate(BaseModel):
    # 会话归属的大屏；不传则不归属任何大屏（兼容旧调用）
    screen_id: NonBlankString | None = None


class ConversationRename(BaseModel):
    title: Title


class ConversationCreated(BaseModel):
    conversation_id: str


class ConversationDetail(BaseModel):
    conversation_id: str
    turns: list[Turn]


class ConversationSummary(BaseModel):
    """历史会话列表项，只包含已产生对话的会话"""

    conversation_id: str
    title: str  # 用户改过名用改的，否则取自首轮用户输入，过长时截断
    screen_id: str | None = None
    updated_at: datetime  # 最近一轮对话的创建时间


class ConversationList(BaseModel):
    conversations: list[ConversationSummary]
    total: int  # 符合条件的会话总数（不受 limit / offset 影响），管理页翻页用