"""AI PMO 工作项模型。"""

from __future__ import annotations

from typing import Any, Optional

from sqlalchemy import BigInteger, ForeignKey, Integer, UniqueConstraint, VARCHAR
from sqlalchemy.orm import Mapped, mapped_column

from .base import Base
from .types import JSONB


class WorkItem(Base):
    __tablename__ = "work_items"
    __table_args__ = (
        UniqueConstraint("tenant_id", "source_system", "source_message_id", name="uq_work_items_source_message"),
        UniqueConstraint("tenant_id", "ingest_event_id", name="uq_work_items_ingest_event"),
    )

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    ingest_event_id: Mapped[int] = mapped_column(
        BigInteger,
        ForeignKey("ingest_events.id", ondelete="CASCADE"),
        nullable=False,
        index=True,
    )
    source_system: Mapped[str] = mapped_column(VARCHAR(32), nullable=False, index=True)
    source_message_id: Mapped[str] = mapped_column(VARCHAR(128), nullable=False, index=True)
    state: Mapped[str] = mapped_column(VARCHAR(32), nullable=False, default="received", index=True)
    intent: Mapped[Optional[str]] = mapped_column(VARCHAR(64), nullable=True, index=True)
    decision_level: Mapped[Optional[str]] = mapped_column(VARCHAR(20), nullable=True, index=True)
    project_id: Mapped[Optional[int]] = mapped_column(
        BigInteger,
        ForeignKey("projects.id"),
        nullable=True,
        index=True,
    )
    signal_id: Mapped[Optional[int]] = mapped_column(
        BigInteger,
        ForeignKey("signals.id"),
        nullable=True,
        index=True,
    )
    owner_user_id: Mapped[Optional[int]] = mapped_column(
        BigInteger,
        ForeignKey("users.id"),
        nullable=True,
        index=True,
    )
    dedupe_key: Mapped[Optional[str]] = mapped_column(VARCHAR(255), nullable=True, index=True)
    context_snapshot: Mapped[Optional[Any]] = mapped_column(JSONB, nullable=True)
    action_bundle: Mapped[Optional[Any]] = mapped_column(JSONB, nullable=True)
    execution_result: Mapped[Optional[Any]] = mapped_column(JSONB, nullable=True)
    review_summary: Mapped[Optional[Any]] = mapped_column(JSONB, nullable=True)
    reply_template_key: Mapped[Optional[str]] = mapped_column(VARCHAR(64), nullable=True)