claude-relay-service/scripts/migrate-apikey-expiry.js

#!/usr/bin/env node

/**
 * 数据迁移脚本：为现有 API Key 设置默认有效期
 *
 * 使用方法：
 * node scripts/migrate-apikey-expiry.js [--days=30] [--dry-run]
 *
 * 参数：
 * --days: 设置默认有效期天数（默认30天）
 * --dry-run: 仅模拟运行，不实际修改数据
 */

const redis = require('../src/models/redis')
const logger = require('../src/utils/logger')
const readline = require('readline')

// 解析命令行参数
const args = process.argv.slice(2)
const params = {}
args.forEach((arg) => {
  const [key, value] = arg.split('=')
  params[key.replace('--', '')] = value || true
})

const DEFAULT_DAYS = params.days ? parseInt(params.days) : 30
const DRY_RUN = params['dry-run'] === true

// 创建 readline 接口用于用户确认
const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout
})

async function askConfirmation(question) {
  return new Promise((resolve) => {
    rl.question(`${question} (yes/no): `, (answer) => {
      resolve(answer.toLowerCase() === 'yes' || answer.toLowerCase() === 'y')
    })
  })
}

async function migrateApiKeys() {
  try {
    logger.info('🔄 Starting API Key expiry migration...')
    logger.info(`📅 Default expiry period: ${DEFAULT_DAYS} days`)
    logger.info(`🔍 Mode: ${DRY_RUN ? 'DRY RUN (no changes will be made)' : 'LIVE RUN'}`)

    // 连接 Redis
    await redis.connect()
    logger.success('✅ Connected to Redis')

    // 获取所有 API Keys
    const apiKeys = await redis.getAllApiKeys()
    logger.info(`📊 Found ${apiKeys.length} API Keys in total`)

    // 统计信息
    const stats = {
      total: apiKeys.length,
      needsMigration: 0,
      alreadyHasExpiry: 0,
      migrated: 0,
      errors: 0
    }

    // 需要迁移的 Keys
    const keysToMigrate = []

    // 分析每个 API Key
    for (const key of apiKeys) {
      if (!key.expiresAt || key.expiresAt === 'null' || key.expiresAt === '') {
        keysToMigrate.push(key)
        stats.needsMigration++
        logger.info(`📌 API Key "${key.name}" (${key.id}) needs migration`)
      } else {
        stats.alreadyHasExpiry++
        const expiryDate = new Date(key.expiresAt)
        logger.info(
          `✓ API Key "${key.name}" (${key.id}) already has expiry: ${expiryDate.toLocaleString()}`
        )
      }
    }

    if (keysToMigrate.length === 0) {
      logger.success('✨ No API Keys need migration!')
      return
    }

    // 显示迁移摘要
    console.log(`\n${'='.repeat(60)}`)
    console.log('📋 Migration Summary:')
    console.log('='.repeat(60))
    console.log(`Total API Keys: ${stats.total}`)
    console.log(`Already have expiry: ${stats.alreadyHasExpiry}`)
    console.log(`Need migration: ${stats.needsMigration}`)
    console.log(`Default expiry: ${DEFAULT_DAYS} days from now`)
    console.log(`${'='.repeat(60)}\n`)

    // 如果不是 dry run，请求确认
    if (!DRY_RUN) {
      const confirmed = await askConfirmation(
        `⚠️  This will set expiry dates for ${keysToMigrate.length} API Keys. Continue?`
      )

      if (!confirmed) {
        logger.warn('❌ Migration cancelled by user')
        return
      }
    }

    // 计算新的过期时间
    const newExpiryDate = new Date()
    newExpiryDate.setDate(newExpiryDate.getDate() + DEFAULT_DAYS)
    const newExpiryISO = newExpiryDate.toISOString()

    logger.info(`\n🚀 Starting migration... New expiry date: ${newExpiryDate.toLocaleString()}`)

    // 执行迁移
    for (const key of keysToMigrate) {
      try {
        if (!DRY_RUN) {
          // 直接更新 Redis 中的数据
          // 使用 hset 更新单个字段
          await redis.client.hset(`apikey:${key.id}`, 'expiresAt', newExpiryISO)
          logger.success(`✅ Migrated: "${key.name}" (${key.id})`)
        } else {
          logger.info(`[DRY RUN] Would migrate: "${key.name}" (${key.id})`)
        }
        stats.migrated++
      } catch (error) {
        logger.error(`❌ Error migrating "${key.name}" (${key.id}):`, error.message)
        stats.errors++
      }
    }

    // 显示最终结果
    console.log(`\n${'='.repeat(60)}`)
    console.log('✅ Migration Complete!')
    console.log('='.repeat(60))
    console.log(`Successfully migrated: ${stats.migrated}`)
    console.log(`Errors: ${stats.errors}`)
    console.log(`New expiry date: ${newExpiryDate.toLocaleString()}`)
    console.log(`${'='.repeat(60)}\n`)

    if (DRY_RUN) {
      logger.warn('⚠️  This was a DRY RUN. No actual changes were made.')
      logger.info('💡 Run without --dry-run flag to apply changes.')
    }
  } catch (error) {
    logger.error('💥 Migration failed:', error)
    process.exit(1)
  } finally {
    // 清理
    rl.close()
    await redis.disconnect()
    logger.info('👋 Disconnected from Redis')
  }
}

// 显示帮助信息
if (params.help) {
  console.log(`
API Key Expiry Migration Script

This script adds expiry dates to existing API Keys that don't have one.

Usage:
  node scripts/migrate-apikey-expiry.js [options]

Options:
  --days=NUMBER    Set default expiry days (default: 30)
  --dry-run        Simulate the migration without making changes
  --help           Show this help message

Examples:
  # Set 30-day expiry for all API Keys without expiry
  node scripts/migrate-apikey-expiry.js

  # Set 90-day expiry
  node scripts/migrate-apikey-expiry.js --days=90

  # Test run without making changes
  node scripts/migrate-apikey-expiry.js --dry-run
`)
  process.exit(0)
}

// 运行迁移
migrateApiKeys().catch((error) => {
  logger.error('💥 Unexpected error:', error)
  process.exit(1)
})