web/src/components/composites/purchase/shared/sku.ts

import {ProductItem} from '@/actions/product'
import {PurchaseKind, PurchaseMode} from './resource'

export type PurchaseSkuItem = {
  code: string
  mode: PurchaseMode
  live: string
  expire: string
  price: string
  count_min: number
  discount: number
}

export type PurchaseSkuData = {
  items: PurchaseSkuItem[]
  priceMap: Map<string, string>
  countMinMap: Map<string, number>
  discountMap: Map<string, number>
  modeList: PurchaseMode[]
  liveList: string[]
  expireList: string[]
}

export function parsePurchaseSkuList(kind: PurchaseKind, skuList: ProductItem['skus']): PurchaseSkuData {
  if (!skuList?.length) {
    throw new Error('没有套餐数据')
  }

  const items: PurchaseSkuItem[] = []
  const priceMap = new Map<string, string>()
  const countMinMap = new Map<string, number>()
  const discountMap = new Map<string, number>()
  const modeSet = new Set<PurchaseMode>()
  const liveSet = new Set<number>()
  const expireSet = new Set<number>()

  for (const sku of skuList) {
    const params = new URLSearchParams(sku.code)
    const mode = parsePurchaseSkuMode(params.get('mode'))
    const live = Number(params.get('live') || '0')
    const expire = Number(params.get('expire') || '0')

    if (!mode || live <= 0) {
      continue
    }

    const liveValue = String(live)
    const expireValue = mode === '1' ? String(expire || '0') : '0'
    const code = getPurchaseSkuKey({
      mode,
      live: liveValue,
      expire: expireValue,
    })
    const countMin = typeof sku.count_min === 'number' ? sku.count_min : Number(sku.count_min) || 0
    countMinMap.set(code, countMin)

    const skuDiscount = sku.discount ?? 100
    items.push({
      code,
      mode,
      live: liveValue,
      expire: expireValue,
      price: sku.price,
      count_min: countMin,
      discount: skuDiscount,
    })
    priceMap.set(code, sku.price)
    discountMap.set(code, skuDiscount)
    modeSet.add(mode)

    liveSet.add(live)

    if (kind === 'short') {
      if (mode === '1' && expire > 0) {
        expireSet.add(expire)
      }
    }
    else if (expire > 0) {
      expireSet.add(expire)
    }
  }

  if (items.length === 0) {
    throw new Error('没有可用的套餐数据')
  }

  return {
    items,
    priceMap,
    countMinMap,
    discountMap,
    modeList: (['2', '1'] as const).filter(mode => modeSet.has(mode)),
    liveList: sortNumericValues(liveSet),
    expireList: sortNumericValues(expireSet),
  }
}

function parsePurchaseSkuMode(mode: string | null): PurchaseMode | null {
  if (mode === 'time') {
    return '1'
  }

  if (mode === 'quota') {
    return '2'
  }

  return null
}

function sortNumericValues(values: Iterable<number>) {
  return Array.from(values).sort((a, b) => a - b).map(String)
}

export function getPurchaseSkuKey(props: {
  mode: PurchaseMode
  live: string
  expire: string
}) {
  const params = new URLSearchParams()
  params.set('mode', props.mode === '1' ? 'time' : 'quota')
  params.set('live', props.live || '0')
  params.set('expire', props.mode === '1' ? props.expire || '0' : '0')
  return params.toString()
}

export function getAvailablePurchaseLives(skuData: PurchaseSkuData, props: {
  mode: PurchaseMode
  expire?: string
}) {
  return sortNumericValues(new Set(
    skuData.items
      .filter(item => item.mode === props.mode)
      .filter(item => !props.expire || item.expire === props.expire)
      .map(item => Number(item.live)),
  ))
}

export function getAvailablePurchaseExpires(skuData: PurchaseSkuData, props: {
  mode: PurchaseMode
  live?: string
}) {
  return sortNumericValues(new Set(
    skuData.items
      .filter(item => item.mode === props.mode)
      .filter(item => !props.live || item.live === props.live)
      .filter(item => item.expire !== '0')
      .map(item => Number(item.expire)),
  ))
}

export function hasPurchaseSku(skuData: PurchaseSkuData, props: {
  mode: PurchaseMode
  live: string
  expire: string
}) {
  return skuData.priceMap.has(getPurchaseSkuKey(props))
}

export function getPurchaseSkuPrice(priceMap: Map<string, string>, props: {
  mode: PurchaseMode
  live: string
  expire: string
}) {
  return priceMap.get(getPurchaseSkuKey(props))
}

export function getPurchaseSkuDiscount(discountMap: Map<string, number>, props: {
  mode: PurchaseMode
  live: string
  expire: string
}) {
  return discountMap.get(getPurchaseSkuKey(props))
}

export function formatPurchaseLiveLabel(live: string, kind: PurchaseKind) {
  const minutes = Number(live)

  if (kind === 'long') {
    return `${minutes / 60} 小时`
  }

  if (minutes % 60 === 0) {
    return `${minutes / 60} 小时`
  }

  return `${minutes} 分钟`
}

export function getPurchaseSkuCountMin(
  skuData: PurchaseSkuData,
  props: {mode: PurchaseMode, live: string, expire: string},
): number {
  const key = getPurchaseSkuKey(props)
  return skuData.countMinMap.get(key) ?? 0
}