"""
SQLAlchemy ORM 数据模型
对应数据库表结构 (基于 DATABASE_DESIGN.md)
"""

import uuid
from datetime import datetime, timezone
from enum import Enum as PyEnum
from sqlalchemy import Column, String, Text, DateTime, Integer, Enum, Index, ForeignKey, Boolean
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import relationship

from .database import Base


class TaskStatus(str, PyEnum):
    """任务状态枚举"""
    GENERATING = "generating" # 生成中 - TaskProcess 生成阶段
    EXECUTING = "executing"  # 执行中 - 任务执行阶段
    STOPPED = "stopped"      # 已停止 - 用户手动停止执行
    COMPLETED = "completed"   # 已完成 - 任务正常完成


def utc_now():
    """获取当前 UTC 时间"""
    return datetime.now(timezone.utc)


class MultiAgentTask(Base):
    """多智能体任务记录模型"""
    __tablename__ = "multi_agent_tasks"

    task_id = Column(String(64), primary_key=True)
    user_id = Column(String(64), nullable=False, index=True)
    query = Column(Text, nullable=False)
    agents_info = Column(JSONB, nullable=False)
    task_outline = Column(JSONB)
    assigned_agents = Column(JSONB)
    agent_scores = Column(JSONB)
    result = Column(JSONB)
    status = Column(
        Enum(TaskStatus, name="task_status_enum", create_type=False),
        default=TaskStatus.GENERATING,
        nullable=False
    )
    execution_count = Column(Integer, default=0, nullable=False)
    generation_id = Column(String(64))
    execution_id = Column(String(64))
    rehearsal_log = Column(JSONB)
    branches = Column(JSONB)  # 任务大纲探索分支数据
    is_pinned = Column(Boolean, default=False, nullable=False)  # 置顶标志
    created_at = Column(DateTime(timezone=True), default=utc_now)
    updated_at = Column(DateTime(timezone=True), default=utc_now, onupdate=utc_now)

    __table_args__ = (
        Index("idx_multi_agent_tasks_status", "status"),
        Index("idx_multi_agent_tasks_generation_id", "generation_id"),
        Index("idx_multi_agent_tasks_execution_id", "execution_id"),
    )

    def to_dict(self) -> dict:
        """转换为字典"""
        return {
            "task_id": self.task_id,
            "user_id": self.user_id,
            "query": self.query,
            "agents_info": self.agents_info,
            "task_outline": self.task_outline,
            "assigned_agents": self.assigned_agents,
            "agent_scores": self.agent_scores,
            "result": self.result,
            "status": self.status.value if self.status else None,
            "execution_count": self.execution_count,
            "generation_id": self.generation_id,
            "execution_id": self.execution_id,
            "rehearsal_log": self.rehearsal_log,
            "branches": self.branches,
            "is_pinned": self.is_pinned,
            "created_at": self.created_at.isoformat() if self.created_at else None,
            "updated_at": self.updated_at.isoformat() if self.updated_at else None,
        }


class ExportRecord(Base):
    """导出记录模型"""
    __tablename__ = "export_records"

    id = Column(Integer, primary_key=True, autoincrement=True)
    task_id = Column(String(64), nullable=False, index=True)          # 关联任务ID
    user_id = Column(String(64), nullable=False, index=True)          # 用户ID
    export_type = Column(String(32), nullable=False)                  # 导出类型: doc/markdown/mindmap/infographic/excel/ppt
    file_name = Column(String(256), nullable=False)                   # 文件名
    file_path = Column(String(512), nullable=False)                    # 服务器存储路径
    file_url = Column(String(512))                                     # 访问URL
    file_size = Column(Integer, default=0)                            # 文件大小(字节)
    created_at = Column(DateTime(timezone=True), default=utc_now)

    __table_args__ = (
        Index("idx_export_records_task_user", "task_id", "user_id"),
    )

    def to_dict(self) -> dict:
        """转换为字典"""
        return {
            "id": self.id,
            "task_id": self.task_id,
            "user_id": self.user_id,
            "export_type": self.export_type,
            "file_name": self.file_name,
            "file_path": self.file_path,
            "file_url": self.file_url,
            "file_size": self.file_size,
            "created_at": self.created_at.isoformat() if self.created_at else None,
        }


class UserAgent(Base):
    """用户保存的智能体配置模型 (可选表)"""
    __tablename__ = "user_agents"

    id = Column(String(64), primary_key=True)
    user_id = Column(String(64), nullable=False, index=True)
    agent_name = Column(String(100), nullable=False)
    agent_config = Column(JSONB, nullable=False)
    created_at = Column(DateTime(timezone=True), default=utc_now)

    __table_args__ = (
        Index("idx_user_agents_user_created", "user_id", "created_at"),
    )

    def to_dict(self) -> dict:
        """转换为字典"""
        return {
            "id": self.id,
            "user_id": self.user_id,
            "agent_name": self.agent_name,
            "agent_config": self.agent_config,
            "created_at": self.created_at.isoformat() if self.created_at else None,
        }