import path from 'path'
import { promises as fs } from 'fs'
import { scanHistoryMaterials } from '@/lib/history-material-import'
import { prisma } from '@/lib/prisma'

function parseArgs() {
  const args = process.argv.slice(2)
  const getArg = (name: string) => {
    const match = args.find((arg) => arg.startsWith(`--${name}=`))
    return match ? match.slice(name.length + 3) : ''
  }

  const materialsRoot = getArg('materials-root') || 'S:\\大货营销策划图片'
  const cpRoot = getArg('cp-root') || '\\\\192.168.58.116\\cp-gb'
  const yyaRoot = getArg('yya-root') || '\\\\192.168.58.116\\yya'
  const output =
    getArg('output') || path.join(process.cwd(), 'output', `history-material-scan-${Date.now()}.json`)
  const limitRaw = getArg('limit')
  const spusRaw = getArg('spus')

  return {
    materialsRoot,
    cpRoot,
    yyaRoot,
    output,
    limit: limitRaw ? Number(limitRaw) : undefined,
    spuFilter: spusRaw ? spusRaw.split(',').map((item) => item.trim()).filter(Boolean) : undefined,
  }
}

async function main() {
  const options = parseArgs()

  console.log('历史素材扫描开始')
  console.log(`- 素材目录: ${options.materialsRoot}`)
  console.log(`- 店铺参考目录: ${options.cpRoot}`)
  console.log(`- 辅助参考目录: ${options.yyaRoot}`)

  const records = await scanHistoryMaterials(options)
  await fs.mkdir(path.dirname(options.output), { recursive: true })
  await fs.writeFile(options.output, JSON.stringify(records, null, 2), 'utf8')

  const summary = records.reduce(
    (acc, record) => {
      acc.total += 1
      acc[record.status] += 1
      return acc
    },
    { total: 0, ready: 0, needs_review: 0, skipped: 0 } as Record<string, number>,
  )

  console.log('\n扫描完成')
  console.log(`- 总记录数: ${summary.total}`)
  console.log(`- 可直接导入: ${summary.ready}`)
  console.log(`- 需要人工确认: ${summary.needs_review}`)
  console.log(`- 已跳过: ${summary.skipped}`)
  console.log(`- 输出文件: ${options.output}`)

  const preview = records.slice(0, 12).map((record) => ({
    status: record.status,
    spu: record.spu,
    store: record.guessedStore,
    main: record.mainCount,
    sku: record.skuCount,
    detail: record.detailCount,
    path: record.relativeSourcePath,
  }))
  console.table(preview)
}

main()
  .catch((error) => {
    console.error('历史素材扫描失败:', error)
    process.exitCode = 1
  })
  .finally(async () => {
    await prisma.$disconnect()
    process.exit(process.exitCode ?? 0)
  })