"""PMFlow OS change review models."""

from __future__ import annotations

from datetime import datetime
from typing import Any, Optional

from sqlalchemy import BigInteger, Float, ForeignKey, Integer, Text, TIMESTAMP, VARCHAR
from sqlalchemy.orm import Mapped, mapped_column

from .base import Base
from .types import JSONB


class PMFlowChangeSet(Base):
    __tablename__ = "pmflow_change_sets"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
    thread_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_threads.id"), nullable=False, index=True)
    source_message_id: Mapped[int] = mapped_column(
        BigInteger, ForeignKey("pmflow_thread_messages.id"), nullable=False, index=True
    )
    status: Mapped[str] = mapped_column(VARCHAR(32), nullable=False, default="open", index=True)
    summary: Mapped[str] = mapped_column(Text, nullable=False)
    headline: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
    extraction_scope: Mapped[str] = mapped_column(VARCHAR(64), nullable=False, default="latest_message")
    routing_meta: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
    llm_call_id: Mapped[Optional[str]] = mapped_column(VARCHAR(128), nullable=True)
    created_by: Mapped[int] = mapped_column(BigInteger, ForeignKey("users.id"), nullable=False, index=True)
    reviewed_at: Mapped[Optional[datetime]] = mapped_column(TIMESTAMP, nullable=True)
    committed_at: Mapped[Optional[datetime]] = mapped_column(TIMESTAMP, nullable=True)
    committed_by: Mapped[Optional[int]] = mapped_column(BigInteger, ForeignKey("users.id"), nullable=True, index=True)


class PMFlowChangeItem(Base):
    __tablename__ = "pmflow_change_items"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    change_set_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_change_sets.id"), nullable=False, index=True)
    project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
    group: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
    object_type: Mapped[str] = mapped_column(VARCHAR(64), nullable=False)
    object_id: Mapped[Optional[str]] = mapped_column(VARCHAR(128), nullable=True)
    title: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
    summary: Mapped[str] = mapped_column(Text, nullable=False)
    diff: Mapped[str] = mapped_column(Text, nullable=False)
    payload: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
    edited_payload: Mapped[Optional[Any]] = mapped_column(JSONB, nullable=True)
    confidence: Mapped[Optional[float]] = mapped_column(Float, nullable=True)
    risk_level: Mapped[str] = mapped_column(VARCHAR(16), nullable=False, default="medium")
    status: Mapped[str] = mapped_column(VARCHAR(32), nullable=False, default="pending_review", index=True)
    reviewed_at: Mapped[Optional[datetime]] = mapped_column(TIMESTAMP, nullable=True)


class PMFlowApproval(Base):
    __tablename__ = "pmflow_approvals"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
    change_set_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_change_sets.id"), nullable=False, index=True)
    change_item_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_change_items.id"), nullable=False, index=True)
    actor_user_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("users.id"), nullable=False, index=True)
    action: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
    decision: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
    before_status: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
    after_status: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
    before_payload: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
    after_payload: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
    reason: Mapped[Optional[str]] = mapped_column(Text, nullable=True)


class PMFlowEvidenceLink(Base):
    __tablename__ = "pmflow_evidence_links"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
    change_set_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_change_sets.id"), nullable=False, index=True)
    change_item_id: Mapped[Optional[int]] = mapped_column(
        BigInteger, ForeignKey("pmflow_change_items.id"), nullable=True, index=True
    )
    source_type: Mapped[str] = mapped_column(VARCHAR(64), nullable=False, index=True)
    source_id: Mapped[str] = mapped_column(VARCHAR(128), nullable=False, index=True)
    title: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
    excerpt: Mapped[str] = mapped_column(Text, nullable=False)
    display_label: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
    evidence_metadata: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)


__all__ = ["PMFlowChangeSet", "PMFlowChangeItem", "PMFlowApproval", "PMFlowEvidenceLink"]