from __future__ import annotations

from datetime import datetime
from decimal import Decimal

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

from app.enums.credit_subscription import CreditSubscriptionStatus
from app.models.base import Base, TimestampMixin


class UserCreditSubscription(Base, TimestampMixin):
    __tablename__ = "user_credit_subscriptions"
    __table_args__ = (
        Index("ix_user_credit_subscriptions_user_active", "user_id", "status", "expires_at"),
        Index("ix_user_credit_subscriptions_team_active", "team_id", "status", "expires_at"),
        Index("ix_user_credit_subscriptions_expire_due", "status", "expires_at", "id"),
        Index("ix_user_credit_subscriptions_grant_due", "status", "next_grant_at", "id"),
        Index("uq_user_credit_subscriptions_payment", "payment_order_id", unique=True),
        Index("uq_user_credit_subscriptions_no", "subscription_no", unique=True),
    )

    id: Mapped[str] = mapped_column(String(32), primary_key=True)
    subscription_no: Mapped[str] = mapped_column(
        String(32), nullable=False, comment="订阅业务实例编号，供用户/客服/开发定位"
    )
    user_id: Mapped[str] = mapped_column(
        String(32), ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True,
        comment="成交时的个人用户；团队订阅为成交时队长",
    )
    team_id: Mapped[str | None] = mapped_column(
        String(32), ForeignKey("teams.id", ondelete="RESTRICT"), nullable=True
    )
    team_manager_id_snapshot: 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
    )
    payment_order_id: Mapped[str] = mapped_column(
        String(32), ForeignKey("payment_orders.id", ondelete="RESTRICT"), nullable=False
    )
    status: Mapped[str] = mapped_column(
        String(32), nullable=False, default=CreditSubscriptionStatus.PENDING.value,
        server_default=CreditSubscriptionStatus.PENDING.value,
    )
    purchase_scene: Mapped[str] = mapped_column(String(24), nullable=False)

    product_type_snapshot: Mapped[str] = mapped_column(String(24), nullable=False)
    product_name_snapshot: Mapped[str] = mapped_column(String(96), nullable=False)
    tier_code: Mapped[str] = mapped_column(String(32), nullable=False)
    tier_rank: Mapped[int] = mapped_column(nullable=False)
    billing_cycle: Mapped[str] = mapped_column(String(24), nullable=False)

    anchor_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    start_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    expires_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    next_grant_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)

    monthly_grant_credits_snapshot: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False)
    monthly_total_credits_snapshot: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False)
    quantity_snapshot: Mapped[int] = mapped_column(nullable=False, default=1, server_default="1")
    grant_count: Mapped[int] = mapped_column(nullable=False)
    granted_count: Mapped[int] = mapped_column(nullable=False, default=0, server_default="0")

    first_purchase_price_snapshot: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False)
    regular_price_snapshot: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False)
    activity_price_snapshot: Mapped[Decimal | None] = mapped_column(Numeric(20, 2), nullable=True)
    actual_unit_price_snapshot: Mapped[Decimal] = mapped_column(Numeric(20, 6), nullable=False)
    paid_amount_snapshot: Mapped[Decimal] = mapped_column(Numeric(20, 2), nullable=False)
    product_snapshot_json: Mapped[dict] = mapped_column(JSON, nullable=False)