from datetime import datetime
from typing import Any

from sqlalchemy import JSON, BigInteger, DateTime, Integer, String
from sqlalchemy.orm import Mapped, mapped_column

from app.model.base import Base


class ConfigRelease(Base):
    __tablename__ = "config_release"
    id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
    release_no: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
    title: Mapped[str] = mapped_column(String(128), nullable=False)
    change_summary: Mapped[str] = mapped_column(String(1000), nullable=False)
    status: Mapped[str] = mapped_column(String(20), nullable=False, default="draft")
    created_by: Mapped[int] = mapped_column(BigInteger, nullable=False)
    reviewer_id: Mapped[int | None] = mapped_column(BigInteger)
    reviewed_at: Mapped[datetime | None] = mapped_column(DateTime)
    activated_at: Mapped[datetime | None] = mapped_column(DateTime)
    supersedes_release_id: Mapped[int | None] = mapped_column(BigInteger)
    rollback_of_release_id: Mapped[int | None] = mapped_column(BigInteger)
    created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
    updated_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)


class ModelEndpointConfig(Base):
    __tablename__ = "model_endpoint_config"
    id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
    endpoint_code: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
    provider: Mapped[str] = mapped_column(String(32), nullable=False)
    model_name: Mapped[str] = mapped_column(String(128), nullable=False)
    base_url: Mapped[str] = mapped_column(String(500), nullable=False)
    secret_ref: Mapped[str] = mapped_column(String(256), nullable=False)
    capabilities: Mapped[list[Any]] = mapped_column(JSON, nullable=False)
    status: Mapped[str] = mapped_column(String(16), nullable=False, default="disabled")
    timeout_ms: Mapped[int] = mapped_column(Integer, nullable=False, default=15000)


class PlatformConfigItem(Base):
    __tablename__ = "platform_config_item"
    id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
    release_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
    namespace: Mapped[str] = mapped_column(String(64), nullable=False)
    config_key: Mapped[str] = mapped_column(String(128), nullable=False)
    value_json: Mapped[dict[str, Any]] = mapped_column(JSON, nullable=False)
    schema_version: Mapped[str] = mapped_column(String(32), nullable=False)
    checksum: Mapped[str] = mapped_column(String(64), nullable=False)
    created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)


class PromptTemplateVersion(Base):
    __tablename__ = "prompt_template_version"
    id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
    release_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
    prompt_code: Mapped[str] = mapped_column(String(64), nullable=False)
    task_type: Mapped[str] = mapped_column(String(48), nullable=False)
    agent_type: Mapped[str | None] = mapped_column(String(32))
    version: Mapped[int] = mapped_column(Integer, nullable=False)
    system_prompt: Mapped[str] = mapped_column(String(65535), nullable=False)
    user_prompt_template: Mapped[str] = mapped_column(String(65535), nullable=False)
    input_schema: Mapped[dict[str, Any] | None] = mapped_column(JSON)
    output_schema: Mapped[dict[str, Any] | None] = mapped_column(JSON)
    checksum: Mapped[str] = mapped_column(String(64), nullable=False)
    created_by: Mapped[int] = mapped_column(BigInteger, nullable=False)
    created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)


class AgentIntentConfig(Base):
    __tablename__ = "agent_intent_config"
    id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
    agent_type: Mapped[str] = mapped_column(String(32), nullable=False)
    intent_code: Mapped[str] = mapped_column(String(64), nullable=False)
    allowed_tools: Mapped[list[Any] | None] = mapped_column(JSON)
    confidence_threshold: Mapped[float] = mapped_column(nullable=False)
    status: Mapped[str] = mapped_column(String(16), nullable=False)
    version: Mapped[int] = mapped_column(Integer, nullable=False)