from __future__ import annotations

from enum import StrEnum


class CreditProductType(StrEnum):
    SUBSCRIPTION = "subscription"
    TEAM_SUBSCRIPTION = "team_subscription"
    CREDIT_ADDON = "credit_addon"


CREDIT_PRODUCT_TYPE_LABELS = {
    CreditProductType.SUBSCRIPTION.value: "个人订阅套餐",
    CreditProductType.TEAM_SUBSCRIPTION.value: "团队订阅套餐",
    CreditProductType.CREDIT_ADDON.value: "积分增值包",
}


class SubscriptionBillingCycle(StrEnum):
    MONTHLY = "monthly"
    QUARTERLY = "quarterly"
    YEARLY = "yearly"


SUBSCRIPTION_BILLING_CYCLE_LABELS = {
    SubscriptionBillingCycle.MONTHLY.value: "月卡",
    SubscriptionBillingCycle.QUARTERLY.value: "季卡",
    SubscriptionBillingCycle.YEARLY.value: "年卡",
}


SUBSCRIPTION_GRANT_COUNT = {
    SubscriptionBillingCycle.MONTHLY.value: 1,
    SubscriptionBillingCycle.QUARTERLY.value: 3,
    SubscriptionBillingCycle.YEARLY.value: 12,
}


class SubscriptionTierCode(StrEnum):
    STARTER = "starter"
    STANDARD = "standard"
    ADVANCED = "advanced"
    SUPER = "super"


SUBSCRIPTION_TIER_LABELS = {
    SubscriptionTierCode.STARTER.value: "入门",
    SubscriptionTierCode.STANDARD.value: "标准",
    SubscriptionTierCode.ADVANCED.value: "高级",
    SubscriptionTierCode.SUPER.value: "超级",
}


class ProductPriceType(StrEnum):
    FIRST_PURCHASE = "first_purchase"
    REGULAR = "regular"
    ACTIVITY = "activity"


PRODUCT_PRICE_TYPE_LABELS = {
    ProductPriceType.FIRST_PURCHASE.value: "首购价",
    ProductPriceType.REGULAR.value: "常规价",
    ProductPriceType.ACTIVITY.value: "活动价",
}