import { logger } from 'node-karin'

import { Networks } from '@/module'
import type { KuaishouDataTypes } from '@/types'

export interface ExtendedKuaishouOptionsType {
  type: KuaishouDataTypes[keyof KuaishouDataTypes]
  [x: string]: any
}

/**
 *
 * @param url 分享链接
 * @param log 输出日志，默认true
 * @returns
 */
export const getKuaishouID = async (url: string, log = true) => {
  const longLink = await new Networks({ url }).getLongLink()
  let result = {} as ExtendedKuaishouOptionsType
  switch (true) {
    case /photoId=(.*)/.test(longLink): {
      const workid = /photoId=([^&]+)/.exec(longLink)
      result = {
        type: 'one_work',
        photoId: workid ? workid[1] : undefined
      }
      break
    }
    // 匹配 kuaishou.com/short-video/{id} 格式的链接
    case /kuaishou\.com\/short-video\/([^?]+)/.test(longLink): {
      const workid = /short-video\/([^?]+)/.exec(longLink)
      result = {
        type: 'one_work',
        photoId: workid ? workid[1] : undefined
      }
      break
    }
    default: {
      logger.warn('无法获取作品ID')
      break
    }
  }

  if (log) {
    console.log(result)
  }
  return result
}