from datetime import date, datetime
from decimal import Decimal

from sqlalchemy import Date, DateTime, Integer, Numeric, String, Text
from sqlalchemy.orm import Mapped, mapped_column

from app.core.database import Base


class TimestampMixin:
    created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
    updated_at: Mapped[datetime] = mapped_column(
        DateTime, default=datetime.utcnow, onupdate=datetime.utcnow
    )


class Project(Base, TimestampMixin):
    __tablename__ = "projects"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    name: Mapped[str] = mapped_column(String(255), index=True)
    owner: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
    status: Mapped[str] = mapped_column(String(64), default="立项", index=True)
    priority: Mapped[str] = mapped_column(String(32), default="P2")
    progress_percent: Mapped[int] = mapped_column(Integer, default=0)
    risk_level: Mapped[str] = mapped_column(String(32), default="low", index=True)
    budget_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
    actual_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
    start_date: Mapped[date | None] = mapped_column(Date, nullable=True)
    due_date: Mapped[date | None] = mapped_column(Date, nullable=True, index=True)
    description: Mapped[str | None] = mapped_column(Text, nullable=True)
    source_system: Mapped[str] = mapped_column(String(64), default="internal")
    external_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)


class WorkTask(Base, TimestampMixin):
    __tablename__ = "work_tasks"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    title: Mapped[str] = mapped_column(String(255), index=True)
    project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    owner: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
    status: Mapped[str] = mapped_column(String(64), default="待办", index=True)
    priority: Mapped[str] = mapped_column(String(32), default="P2")
    due_date: Mapped[date | None] = mapped_column(Date, nullable=True, index=True)
    completed_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
    blocker: Mapped[str | None] = mapped_column(Text, nullable=True)
    description: Mapped[str | None] = mapped_column(Text, nullable=True)


class Procurement(Base, TimestampMixin):
    __tablename__ = "procurements"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    name: Mapped[str] = mapped_column(String(255), index=True)
    applicant: Mapped[str | None] = mapped_column(String(128), nullable=True)
    project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    supplier_name: Mapped[str | None] = mapped_column(String(255), nullable=True)
    budget_subject: Mapped[str | None] = mapped_column(String(128), nullable=True)
    expected_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
    actual_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
    approval_status: Mapped[str] = mapped_column(String(64), default="草稿", index=True)
    delivery_status: Mapped[str] = mapped_column(String(64), default="未到货", index=True)
    payment_status: Mapped[str] = mapped_column(String(64), default="未付款", index=True)
    comparison_summary: Mapped[str | None] = mapped_column(Text, nullable=True)


class Expense(Base, TimestampMixin):
    __tablename__ = "expenses"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    expense_type: Mapped[str] = mapped_column(String(128), index=True)
    amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
    applicant: Mapped[str | None] = mapped_column(String(128), nullable=True)
    department: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
    project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    budget_subject: Mapped[str | None] = mapped_column(String(128), nullable=True)
    payment_account: Mapped[str | None] = mapped_column(String(128), nullable=True)
    invoice_status: Mapped[str] = mapped_column(String(64), default="未收票")
    approval_status: Mapped[str] = mapped_column(String(64), default="草稿", index=True)
    payment_status: Mapped[str] = mapped_column(String(64), default="未付款", index=True)


class FundAccount(Base, TimestampMixin):
    __tablename__ = "fund_accounts"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    name: Mapped[str] = mapped_column(String(255), index=True)
    account_type: Mapped[str] = mapped_column(String(64), default="bank")
    current_balance: Mapped[Decimal] = mapped_column(Numeric(16, 2), default=0)
    expected_receivable: Mapped[Decimal] = mapped_column(Numeric(16, 2), default=0)
    expected_payable: Mapped[Decimal] = mapped_column(Numeric(16, 2), default=0)
    safety_line: Mapped[Decimal] = mapped_column(Numeric(16, 2), default=0)
    risk_level: Mapped[str] = mapped_column(String(32), default="low", index=True)
    note: Mapped[str | None] = mapped_column(Text, nullable=True)


class Policy(Base, TimestampMixin):
    __tablename__ = "policies"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    title: Mapped[str] = mapped_column(String(255), index=True)
    policy_type: Mapped[str | None] = mapped_column(String(128), nullable=True)
    owner_department: Mapped[str | None] = mapped_column(String(128), nullable=True)
    version: Mapped[str] = mapped_column(String(32), default="v1.0")
    status: Mapped[str] = mapped_column(String(64), default="草案", index=True)
    effective_date: Mapped[date | None] = mapped_column(Date, nullable=True)
    feishu_doc_url: Mapped[str | None] = mapped_column(String(1024), nullable=True)
    summary: Mapped[str | None] = mapped_column(Text, nullable=True)


class Standard(Base, TimestampMixin):
    __tablename__ = "standards"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    title: Mapped[str] = mapped_column(String(255), index=True)
    standard_type: Mapped[str | None] = mapped_column(String(128), nullable=True)
    applies_to: Mapped[str | None] = mapped_column(String(255), nullable=True)
    status: Mapped[str] = mapped_column(String(64), default="有效", index=True)
    check_items: Mapped[str | None] = mapped_column(Text, nullable=True)
    remediation: Mapped[str | None] = mapped_column(Text, nullable=True)
    policy_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)


class PerformanceMetric(Base, TimestampMixin):
    __tablename__ = "performance_metrics"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    name: Mapped[str] = mapped_column(String(255), index=True)
    applies_to_role: Mapped[str | None] = mapped_column(String(128), nullable=True)
    formula: Mapped[str | None] = mapped_column(Text, nullable=True)
    weight: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
    data_source: Mapped[str | None] = mapped_column(String(255), nullable=True)
    auto_score: Mapped[Decimal] = mapped_column(Numeric(8, 2), default=0)
    confirmed_score: Mapped[Decimal | None] = mapped_column(Numeric(8, 2), nullable=True)
    status: Mapped[str] = mapped_column(String(64), default="草稿", index=True)


class Supplier(Base, TimestampMixin):
    __tablename__ = "suppliers"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    name: Mapped[str] = mapped_column(String(255), index=True)
    category: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
    contact: Mapped[str | None] = mapped_column(String(128), nullable=True)
    quality_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
    delivery_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
    price_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
    risk_level: Mapped[str] = mapped_column(String(32), default="low", index=True)
    blacklist_status: Mapped[str] = mapped_column(String(32), default="normal", index=True)