const { PrismaClient } = require('@prisma/client')
const fs = require('fs')
const path = require('path')

const prisma = new PrismaClient()

async function updateFileSizes() {
  console.log('开始扫描文件并更新数据库...\n')
  
  try {
    // 获取所有素材记录
    const materials = await prisma.material.findMany({
      select: {
        id: true,
        zipFilePath: true,
        zipFileName: true,
        zipFileSize: true,
      }
    })

    console.log(`找到 ${materials.length} 条素材记录\n`)

    let updatedCount = 0
    let notFoundCount = 0
    let unchangedCount = 0

    for (const material of materials) {
      try {
        // 构建文件完整路径
        const filePath = path.join(process.cwd(), material.zipFilePath)
        
        // 检查文件是否存在
        if (!fs.existsSync(filePath)) {
          console.log(`❌ 文件不存在: ${material.zipFileName}`)
          console.log(`   路径: ${filePath}\n`)
          notFoundCount++
          continue
        }

        // 获取文件大小
        const stats = fs.statSync(filePath)
        const actualSize = BigInt(stats.size)
        const currentSize = BigInt(material.zipFileSize)

        // 如果大小不同，更新数据库
        if (actualSize !== currentSize) {
          await prisma.material.update({
            where: { id: material.id },
            data: { zipFileSize: actualSize }
          })

          console.log(`✅ 更新: ${material.zipFileName}`)
          console.log(`   原大小: ${formatSize(currentSize)}`)
          console.log(`   新大小: ${formatSize(actualSize)}\n`)
          updatedCount++
        } else {
          unchangedCount++
        }
      } catch (error) {
        console.error(`错误处理文件 ${material.zipFileName}:`, error.message)
      }
    }

    console.log('\n========================================')
    console.log('扫描完成！')
    console.log(`总计: ${materials.length} 条记录`)
    console.log(`✅ 已更新: ${updatedCount} 条`)
    console.log(`➡️  未变化: ${unchangedCount} 条`)
    console.log(`❌ 文件不存在: ${notFoundCount} 条`)
    console.log('========================================\n')

  } catch (error) {
    console.error('执行失败:', error)
  } finally {
    await prisma.$disconnect()
  }
}

function formatSize(bytes) {
  const size = Number(bytes)
  if (size === 0) return '0 B'
  const units = ['B', 'KB', 'MB', 'GB', 'TB']
  const i = Math.floor(Math.log(size) / Math.log(1024))
  return `${(size / Math.pow(1024, i)).toFixed(2)} ${units[i]}`
}

// 执行脚本
updateFileSizes()