from __future__ import annotations

from datetime import datetime, timezone

from sqlalchemy import Boolean, DateTime, Float, ForeignKey, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column

from app.core.db.common import CommonBase


class AccountingAISemanticDecision(CommonBase):
    __tablename__ = "accounting_ai_semantic_decisions"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False, index=True)
    client_id: Mapped[int] = mapped_column(ForeignKey("clients.id", ondelete="CASCADE"), nullable=False, index=True)

    source_type: Mapped[str] = mapped_column(String(30), nullable=False, index=True)
    source_record_id: Mapped[int] = mapped_column(Integer, nullable=False, index=True)
    source_fingerprint: Mapped[str] = mapped_column(String(80), nullable=False, default="", index=True)

    provider_name: Mapped[str] = mapped_column(String(60), nullable=False, default="")
    model_name: Mapped[str] = mapped_column(String(120), nullable=False, default="")
    prompt_version: Mapped[str] = mapped_column(String(40), nullable=False, default="phase12-v1")

    deterministic_nature_id: Mapped[int | None] = mapped_column(
        ForeignKey("accounting_natures.id", ondelete="SET NULL"), nullable=True, index=True
    )
    deterministic_confidence: Mapped[int] = mapped_column(Integer, nullable=False, default=0)

    ai_nature_id: Mapped[int | None] = mapped_column(
        ForeignKey("accounting_natures.id", ondelete="SET NULL"), nullable=True, index=True
    )
    ai_nature_code: Mapped[str] = mapped_column(String(80), nullable=False, default="")
    ai_confidence: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    combined_confidence: Mapped[int] = mapped_column(Integer, nullable=False, default=0)

    capital_revenue: Mapped[str] = mapped_column(String(20), nullable=False, default="")
    business_personal: Mapped[str] = mapped_column(String(20), nullable=False, default="business")
    concise_reason: Mapped[str] = mapped_column(Text, nullable=False, default="")
    alternatives_json: Mapped[str | None] = mapped_column(Text, nullable=True)
    review_flags_json: Mapped[str | None] = mapped_column(Text, nullable=True)
    context_json: Mapped[str | None] = mapped_column(Text, nullable=True)
    raw_response_json: Mapped[str | None] = mapped_column(Text, nullable=True)

    applied_to_source: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, index=True)
    skipped_reason: Mapped[str | None] = mapped_column(Text, nullable=True)
    status: Mapped[str] = mapped_column(String(30), nullable=False, default="completed", index=True)
    error_message: Mapped[str | None] = mapped_column(Text, nullable=True)

    final_nature_id: Mapped[int | None] = mapped_column(
        ForeignKey("accounting_natures.id", ondelete="SET NULL"), nullable=True, index=True
    )
    final_ledger_name: Mapped[str] = mapped_column(String(255), nullable=False, default="")
    suggestion_accepted: Mapped[bool | None] = mapped_column(Boolean, nullable=True, index=True)
    reviewed_by_user_id: Mapped[int | None] = mapped_column(
        ForeignKey("users.id", ondelete="SET NULL"), nullable=True
    )
    reviewed_at_utc: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)

    input_tokens: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    output_tokens: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
    latency_ms: Mapped[int] = mapped_column(Integer, nullable=False, default=0)

    created_by_user_id: Mapped[int | None] = mapped_column(
        ForeignKey("users.id", ondelete="SET NULL"), nullable=True
    )
    created_at_utc: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, default=lambda: datetime.now(timezone.utc), index=True
    )