from __future__ import annotations

from datetime import datetime
from decimal import Decimal

from sqlalchemy import DateTime, ForeignKey, Index, Integer, Numeric, String, text
from sqlalchemy.orm import Mapped, mapped_column

from app.models.base import Base, TimestampMixin


class CreditRecord(Base, TimestampMixin):
    __tablename__ = "credit_records"
    __table_args__ = (
        Index("uq_credit_records_user_biz_key", "user_id", "biz_key", unique=True),
        Index("ix_credit_records_user_refund_for_biz_key", "user_id", "refund_for_biz_key"),
        Index(
            "uq_credit_records_user_refund_target_kind",
            "user_id", "refund_for_biz_key", "refund_kind",
            unique=True,
            postgresql_where=text("type = 'refund' AND refund_for_biz_key IS NOT NULL"),
        ),
        Index("ix_credit_records_related_type", "related_id", "type"),
        Index("ix_credit_records_owner", "owner_type", "owner_id"),
        Index("ix_credit_records_subject_media", "credit_subject", "media_type"),
        Index("ix_credit_records_source_module_scene", "source_module", "billing_scene"),
        Index("ix_credit_records_user_kind_time", "user_type_snapshot", "frontend_user_kind_snapshot", "created_at"),
        Index("ix_credit_records_team_time", "team_id_snapshot", "created_at"),
        Index("ix_credit_records_user_kind_team_time", "user_type_snapshot", "frontend_user_kind_snapshot", "team_id_snapshot", "created_at"),
    )

    id: Mapped[str] = mapped_column(String(32), primary_key=True)
    user_id: Mapped[str] = mapped_column(
        String(32), ForeignKey("users.id", ondelete="CASCADE"), index=True
    )
    type: Mapped[str] = mapped_column(String(24), index=True)
    amount: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False)
    balance_delta: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False, default=Decimal("0.00"), server_default="0")
    expired_amount: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False, default=Decimal("0.00"), server_default="0")
    balance_after: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False)
    description: Mapped[str] = mapped_column(String(512))
    related_id: Mapped[str | None] = mapped_column(String(64), nullable=True)
    request_time: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True, index=True)

    biz_key: Mapped[str | None] = mapped_column(String(180), nullable=True, index=True)
    refund_for_biz_key: Mapped[str | None] = mapped_column(String(180), nullable=True, index=True)
    refund_kind: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)

    owner_type: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    owner_id: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    attempt_no: Mapped[int | None] = mapped_column(Integer, nullable=True)

    charge_kind: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
    charge_action: Mapped[str | None] = mapped_column(String(24), nullable=True)
    credit_subject: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
    media_type: Mapped[str | None] = mapped_column(String(16), nullable=True, index=True)
    billing_scene: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    scene_name_snapshot: Mapped[str | None] = mapped_column(String(128), nullable=True)
    credit_level_snapshot: Mapped[str | None] = mapped_column(String(32), nullable=True)

    source_module: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    source_project_id: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    source_step_id: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
    source_step_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)

    token_usage_id: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
    input_tokens: Mapped[int | None] = mapped_column(Integer, nullable=True)
    output_tokens: Mapped[int | None] = mapped_column(Integer, nullable=True)
    total_tokens: Mapped[int | None] = mapped_column(Integer, nullable=True)
    llm_call_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
    llm_success_call_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
    llm_failed_call_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
    llm_billing_execution_id: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)

    engine_type: Mapped[str | None] = mapped_column(String(16), nullable=True)
    engine_id: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
    engine_name: Mapped[str | None] = mapped_column(String(128), nullable=True)
    engine_provider: Mapped[str | None] = mapped_column(String(64), nullable=True)
    engine_model_name: Mapped[str | None] = mapped_column(String(128), nullable=True)

    user_type_snapshot: Mapped[str | None] = mapped_column(String(16), nullable=True, index=True)
    frontend_user_kind_snapshot: Mapped[str | None] = mapped_column(String(16), nullable=True, index=True)
    team_id_snapshot: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
    team_name_snapshot: Mapped[str | None] = mapped_column(String(128), nullable=True)