xiaoxia-saas/apps/web/src/pages/subscription/hooks/useSubscription.ts

import { useState, useEffect, useCallback } from "react"
import { message } from "antd"
import {
  getCurrentSubscription,
  changePlan,
  toggleAutoRenew,
  cancelSubscription,
  type SubscriptionInfo,
  type PlanType,
  type BillingCycle,
} from "@/api/subscription"

/**
 * 订阅管理 Hook
 * 封装订阅信息查询、套餐变更、自动续费切换、取消订阅等逻辑
 */
export function useSubscription() {
  const [subscription, setSubscription] = useState<SubscriptionInfo | null>(null)
  const [loading, setLoading] = useState(true)
  const [submitting, setSubmitting] = useState(false)
  const [selectedPlan, setSelectedPlan] = useState<PlanType>("standard")
  const [billingCycle, setBillingCycle] = useState<BillingCycle>("monthly")

  const loadSubscription = useCallback(async () => {
    try {
      const data = await getCurrentSubscription()
      setSubscription(data)
      setSelectedPlan(data.plan_id)
    } catch (err: unknown) {
      if (!(err as { __msgShown?: boolean })?.__msgShown) message.error("获取订阅信息失败")
    } finally {
      setLoading(false)
    }
  }, [])

  useEffect(() => {
    loadSubscription()
  }, [loadSubscription])

  const handleUpgrade = useCallback(async () => {
    if (!subscription) return
    if (selectedPlan === subscription.plan_id && billingCycle === subscription.billing_cycle) {
      message.info("当前已是该套餐")
      return
    }
    // 由调用方决定是否弹确认框
  }, [subscription, selectedPlan, billingCycle])

  const executeChangePlan = useCallback(async () => {
    try {
      setSubmitting(true)
      const res = await changePlan({
        target_plan_id: selectedPlan,
        billing_cycle: billingCycle,
      })
      if (res.success) {
        message.success(res.message)
        setSubscription(res.new_subscription ?? null)
      } else {
        message.error(res.message)
      }
    } catch (err: unknown) {
      if (!(err as { __msgShown?: boolean })?.__msgShown) message.error("套餐变更失败，请重试")
    } finally {
      setSubmitting(false)
    }
  }, [selectedPlan, billingCycle])

  const handleToggleAutoRenew = useCallback(
    async (enabled: boolean) => {
      try {
        const res = await toggleAutoRenew(enabled)
        message.success(res.message)
        if (subscription) {
          setSubscription({ ...subscription, auto_renew: enabled })
        }
      } catch (err: unknown) {
        if (!(err as { __msgShown?: boolean })?.__msgShown) message.error("操作失败")
      }
    },
    [subscription],
  )

  const handleCancel = useCallback(async () => {
    try {
      const res = await cancelSubscription()
      message.success(res.message)
      return true
    } catch (err: unknown) {
      if (!(err as { __msgShown?: boolean })?.__msgShown) message.error("取消失败")
      return false
    }
  }, [])

  return {
    // 状态
    subscription,
    loading,
    submitting,
    selectedPlan,
    billingCycle,
    setSelectedPlan,
    setBillingCycle,
    // 操作
    loadSubscription,
    handleUpgrade,
    executeChangePlan,
    handleToggleAutoRenew,
    handleCancel,
  }
}