from __future__ import annotations

from datetime import datetime
from decimal import Decimal

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

from app.enums.common import PaymentOrderSourceEnum
from app.models.base import Base, TimestampMixin


class PaymentOrder(Base, TimestampMixin):
    __tablename__ = "payment_orders"
    __table_args__ = (
        Index("idx_payorder_user_status_created", "user_id", "status", "created_at"),
        Index("idx_payorder_status_created", "status", "created_at"),
        Index("ix_payorder_source_status_created", "order_source", "status", "created_at"),
        Index("ix_payorder_team_status_created", "team_id_snapshot", "status", "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
    )
    order_no: Mapped[str] = mapped_column(String(64), unique=True)
    amount: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False, comment="实际整单实收金额")
    credits: Mapped[Decimal] = mapped_column(
        Numeric(20, 2), nullable=False, default=Decimal("0.00"), server_default="0"
    )
    payment_method: Mapped[str] = mapped_column(String(16))
    order_source: Mapped[str] = mapped_column(
        String(24), nullable=False, default=PaymentOrderSourceEnum.ONLINE_PAYMENT.value,
        server_default=PaymentOrderSourceEnum.ONLINE_PAYMENT.value,
    )
    status: Mapped[str] = mapped_column(String(32), default="pending")
    paid_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
    trade_no: Mapped[str | None] = mapped_column(String(128), nullable=True)
    refund_trade_no: Mapped[str | None] = mapped_column(String(128), nullable=True)
    refunded_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
    refund_amount: Mapped[Decimal | None] = mapped_column(Numeric(20, 2), nullable=True)
    refund_entitlement_status: Mapped[str | None] = mapped_column(String(32), nullable=True)

    product_id: Mapped[str | None] = mapped_column(
        String(32), ForeignKey("credit_products.id", ondelete="SET NULL"), nullable=True, index=True
    )
    product_type: Mapped[str | None] = mapped_column(String(24), nullable=True, index=True)
    purchase_scene: Mapped[str | None] = mapped_column(String(24), nullable=True)
    price_type: Mapped[str | None] = mapped_column(String(24), nullable=True)
    product_code_snapshot: Mapped[str | None] = mapped_column(String(64), nullable=True)
    product_name_snapshot: Mapped[str | None] = mapped_column(String(96), nullable=True)
    product_snapshot_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)

    quantity: Mapped[int] = mapped_column(Integer, nullable=False, default=1, server_default="1")
    quoted_unit_price_snapshot: Mapped[Decimal | None] = mapped_column(Numeric(20, 2), nullable=True)
    quoted_amount_snapshot: Mapped[Decimal | None] = mapped_column(Numeric(20, 2), nullable=True)
    actual_unit_price_snapshot: Mapped[Decimal | None] = mapped_column(Numeric(20, 6), nullable=True)
    team_id_snapshot: Mapped[str | None] = mapped_column(
        String(32), ForeignKey("teams.id", ondelete="RESTRICT"), nullable=True
    )

    operator_admin_id: Mapped[str | None] = mapped_column(
        String(32), ForeignKey("users.id", ondelete="SET NULL"), nullable=True
    )
    offline_trade_no: Mapped[str | None] = mapped_column(String(128), nullable=True)
    offline_payment_detail: Mapped[str | None] = mapped_column(String(128), nullable=True)
    remark: Mapped[str | None] = mapped_column(String(512), nullable=True)

    subscription_id: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
    fulfillment_status: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
    fulfilled_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)