// 数据库清理脚本
const { PrismaClient } = require('@prisma/client')

const prisma = new PrismaClient()

async function cleanup() {
  try {
    console.log('开始数据库清理...\n')

    // 1. 清理过期的邀请码（已使用且过期超过30天）
    console.log('1. 清理过期的邀请码...')
    const expiredInvitations = await prisma.$executeRawUnsafe(`
      DELETE FROM invitations 
      WHERE "usedAt" IS NOT NULL 
        AND "expiresAt" IS NOT NULL 
        AND "expiresAt" < NOW() - INTERVAL '30 days'
    `)
    console.log(`   删除了 ${expiredInvitations} 条过期邀请码`)

    // 2. 清理旧的弹幕数据（超过90天）
    console.log('2. 清理旧的弹幕数据...')
    const oldDanmakus = await prisma.$executeRawUnsafe(`
      DELETE FROM danmakus 
      WHERE "createdAt" < NOW() - INTERVAL '90 days'
    `)
    console.log(`   删除了 ${oldDanmakus} 条旧弹幕`)

    // 3. 获取清理后的统计信息
    console.log('\n3. 获取清理后的统计信息...')
    const stats = await prisma.$queryRawUnsafe(`
      SELECT 
        'invitations' as table_name,
        COUNT(*)::bigint as count
      FROM invitations
      UNION ALL
      SELECT 
        'danmakus' as table_name,
        COUNT(*)::bigint as count
      FROM danmakus
      UNION ALL
      SELECT 
        'materials' as table_name,
        COUNT(*)::bigint as count
      FROM materials
      UNION ALL
      SELECT 
        'users' as table_name,
        COUNT(*)::bigint as count
      FROM users
    `)

    console.log('\n清理后的数据统计:')
    stats.forEach((stat) => {
      console.log(`   ${stat.table_name}: ${stat.count}`)
    })

    console.log('\n数据库清理完成！')
  } catch (error) {
    console.error('清理失败:', error)
    process.exit(1)
  } finally {
    await prisma.$disconnect()
  }
}

cleanup()