"""工具管理相关数据模型"""
import uuid
from datetime import datetime
from enum import StrEnum

from sqlalchemy import Column, String, Text, DateTime, JSON, ForeignKey, Integer, Float, Boolean
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import relationship

from app.db import Base


class ToolType(StrEnum):
    """工具类型枚举"""
    BUILTIN = "builtin"
    CUSTOM = "custom"
    MCP = "mcp"


class ToolStatus(StrEnum):
    """工具状态枚举"""
    AVAILABLE = "available"  # 可用（已配置且已启用）
    UNCONFIGURED = "unconfigured"  # 未配置
    CONFIGURED_DISABLED = "configured_disabled"  # 已配置未启用
    ERROR = "error"  # 错误状态

    @classmethod
    def get_all_statuses(cls):
        """获取所有工具状态"""
        return [status.value for status in cls]

    @classmethod
    def get_all_statuses_with_labels(cls):
        """获取所有工具状态及其文本描述"""
        return [
            {"value": cls.AVAILABLE.value, "label": "可用"},
            {"value": cls.UNCONFIGURED.value, "label": "未配置"},
            {"value": cls.CONFIGURED_DISABLED.value, "label": "已配置未启用"},
            {"value": cls.ERROR.value, "label": "错误状态"}
        ]

    @classmethod
    def is_valid_status(cls, status):
        """检查状态是否有效"""
        return status in cls._value2member_map_

    @classmethod
    def get_active_statuses(cls):
        """获取所有活跃状态"""
        return [cls.AVAILABLE.value]

    @classmethod
    def get_inactive_statuses(cls):
        """获取所有非活跃状态"""
        return [cls.UNCONFIGURED.value, cls.CONFIGURED_DISABLED.value, cls.ERROR.value]


class AuthType(StrEnum):
    """认证类型枚举"""
    NONE = "none"
    API_KEY = "api_key"
    BEARER_TOKEN = "bearer_token"
    BASIC_AUTH = "basic_auth"

    @classmethod
    def get_all_types(cls):
        """获取所有认证类型"""
        return [auth_type.value for auth_type in cls]

    @classmethod
    def get_all_types_with_labels(cls):
        """获取所有认证类型及其文本描述"""
        return [
            {"value": cls.NONE.value, "label": "无需认证"},
            {"value": cls.API_KEY.value, "label": "API Key"},
            {"value": cls.BEARER_TOKEN.value, "label": "Bearer Token"},
            {"value": cls.BASIC_AUTH.value, "label": "Basic Auth"}
        ]

    @classmethod
    def is_valid_types(cls, auth_type):
        """检查认证类型是否有效"""
        return auth_type in cls._value2member_map_


class ExecutionStatus(StrEnum):
    """执行状态枚举"""
    PENDING = "pending"
    RUNNING = "running"
    COMPLETED = "completed"
    FAILED = "failed"
    TIMEOUT = "timeout"


class ToolConfig(Base):
    """工具配置基础模型"""
    __tablename__ = "tool_configs"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    name = Column(String(255), nullable=False, index=True)
    description = Column(Text)
    icon = Column(String(255))  # 工具图标
    tool_type = Column(String(50), nullable=False, index=True)
    tenant_id = Column(UUID(as_uuid=True), ForeignKey("tenants.id"), nullable=False, index=True)  # 必须属于租户
    status = Column(String(50), default=ToolStatus.UNCONFIGURED.value, nullable=False, index=True)  # 工具状态
    
    # 工具特定配置（JSON格式存储）
    config_data = Column(JSON, default=dict)

    # 元数据
    version = Column(String(50), default="1.0.0")
    tags = Column(JSON, default=list)  # 标签列表
    
    # 时间戳
    created_at = Column(DateTime, default=datetime.now, nullable=False)
    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now, nullable=False)
    
    # 关联关系
    tenant = relationship("Tenants", back_populates="tool_configs")
    executions = relationship("ToolExecution", back_populates="tool_config", cascade="all, delete-orphan")
    
    def __repr__(self):
        return f"<ToolConfig(id={self.id}, name={self.name}, type={self.tool_type}, status={self.status})>"


class BuiltinToolConfig(Base):
    """内置工具配置模型"""
    __tablename__ = "builtin_tool_configs"

    id = Column(UUID(as_uuid=True), ForeignKey("tool_configs.id"), primary_key=True)
    tool_class = Column(String(255), nullable=False)  # 工具类名
    parameters = Column(JSON, default=dict)  # 工具参数配置
    is_enabled = Column(Boolean, default=False, server_default='false', nullable=False)  # 启用开关
    requires_config = Column(Boolean, default=False, server_default='false', nullable=False)  # 是否需要配置

    # 关联关系
    base_config = relationship("ToolConfig", foreign_keys=[id])
    
    def __repr__(self):
        return f"<BuiltinToolConfig(id={self.id}, tool_class={self.tool_class}, enabled={self.is_enabled})>"


class CustomToolConfig(Base):
    """自定义工具配置模型"""
    __tablename__ = "custom_tool_configs"

    id = Column(UUID(as_uuid=True), ForeignKey("tool_configs.id"), primary_key=True)
    schema_url = Column(String(1000))  # OpenAPI schema URL
    schema_content = Column(JSON)  # OpenAPI schema 内容
    
    # 认证配置
    auth_type = Column(String(50), default=AuthType.NONE.value, nullable=False)
    auth_config = Column(JSON, default=dict)  # 认证配置（加密存储）
    
    # API配置
    base_url = Column(String(1000))  # API基础URL
    timeout = Column(Integer, default=30)  # 超时时间（秒）
    
    # 关联关系
    base_config = relationship("ToolConfig", foreign_keys=[id])
    
    def __repr__(self):
        return f"<CustomToolConfig(id={self.id}, auth_type={self.auth_type})>"


class MCPToolConfig(Base):
    """MCP工具配置模型"""
    __tablename__ = "mcp_tool_configs"

    id = Column(UUID(as_uuid=True), ForeignKey("tool_configs.id"), primary_key=True)
    server_url = Column(String(1000), nullable=False)  # MCP服务器URL
    connection_config = Column(JSON, default=dict)  # 连接配置（包含认证信息）
    
    # 服务状态
    last_health_check = Column(DateTime)
    health_status = Column(String(50), default="unknown")
    error_message = Column(Text)
    
    # 可用工具列表
    available_tools = Column(JSON, default=list)
    
    # 关联关系
    base_config = relationship("ToolConfig", foreign_keys=[id])
    
    def __repr__(self):
        return f"<MCPToolConfig(id={self.id}, server_url={self.server_url})>"


class ToolExecution(Base):
    """工具执行记录模型"""
    __tablename__ = "tool_executions"

    id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    tool_config_id = Column(UUID(as_uuid=True), ForeignKey("tool_configs.id"), nullable=False, index=True)
    
    # 执行信息
    execution_id = Column(String(255), nullable=False, index=True)  # 执行ID（可用于关联工作流等）
    status = Column(String(50), default=ExecutionStatus.PENDING.value, nullable=False, index=True)
    
    # 输入输出
    input_data = Column(JSON)  # 输入参数
    output_data = Column(JSON)  # 输出结果
    error_message = Column(Text)  # 错误信息
    
    # 性能指标
    started_at = Column(DateTime, nullable=False, index=True)
    completed_at = Column(DateTime)
    execution_time = Column(Float)  # 执行时间（秒）
    
    # Token使用情况（如果适用）
    token_usage = Column(JSON)
    
    # 用户信息
    user_id = Column(UUID(as_uuid=True), index=True, nullable=True)
    workspace_id = Column(UUID(as_uuid=True), ForeignKey("workspaces.id"), nullable=False, index=True)
    
    # 关联关系
    tool_config = relationship("ToolConfig", back_populates="executions")
    workspace = relationship("Workspace")
    
    def __repr__(self):
        return f"<ToolExecution(id={self.id}, status={self.status}, tool={self.tool_config_id})>"


# class ToolDependency(Base):
#     """工具依赖关系模型"""
#     __tablename__ = "tool_dependencies"
#
#     id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
#     tool_id = Column(UUID(as_uuid=True), ForeignKey("tool_configs.id"), nullable=False)
#     depends_on_tool_id = Column(UUID(as_uuid=True), ForeignKey("tool_configs.id"), nullable=False)
#
#     # 依赖类型和版本要求
#     dependency_type = Column(String(50), default="required")  # required, optional
#     version_constraint = Column(String(100))  # 版本约束，如 ">=1.0.0"
#
#     # 时间戳
#     created_at = Column(DateTime, default=datetime.now, nullable=False)
#
#     # 关联关系
#     tool = relationship("ToolConfig", foreign_keys=[tool_id])
#     depends_on_tool = relationship("ToolConfig", foreign_keys=[depends_on_tool_id])
#
#     def __repr__(self):
#         return f"<ToolDependency(tool={self.tool_id}, depends_on={self.depends_on_tool_id})>"


# class PluginConfig(Base):
#     """插件配置模型"""
#     __tablename__ = "plugin_configs"
#
#     id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
#     name = Column(String(255), nullable=False, unique=True, index=True)
#     description = Column(Text)
#
#     # 插件信息
#     plugin_path = Column(String(1000), nullable=False)  # 插件文件路径
#     entry_point = Column(String(255), nullable=False)  # 入口点
#     version = Column(String(50), default="1.0.0")
#
#     # 状态
#     is_enabled = Column(Boolean, default=True, nullable=False)
#     is_loaded = Column(Boolean, default=False, nullable=False)
#     load_error = Column(Text)  # 加载错误信息
#
#     # 配置
#     config_schema = Column(JSON)  # 配置schema
#     config_data = Column(JSON, default=dict)  # 配置数据
#
#     # 依赖
#     dependencies = Column(JSON, default=list)  # 依赖的其他插件
#
#     # 时间戳
#     created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
#     updated_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)
#     last_loaded_at = Column(DateTime)
#
#     def __repr__(self):
#         return f"<PluginConfig(id={self.id}, name={self.name}, version={self.version})>"