claude-relay-service/scripts/setup.js

const fs = require('fs')
const path = require('path')
const crypto = require('crypto')
const chalk = require('chalk')
const ora = require('ora')

const config = require('../config/config')

async function setup() {
  console.log(chalk.blue.bold('\n🚀 Claude Relay Service 初始化设置\n'))

  const spinner = ora('正在进行初始化设置...').start()

  try {
    // 1. 创建必要目录
    const directories = ['logs', 'data', 'temp']

    directories.forEach((dir) => {
      const dirPath = path.join(__dirname, '..', dir)
      if (!fs.existsSync(dirPath)) {
        fs.mkdirSync(dirPath, { recursive: true })
      }
    })

    // 2. 生成环境配置文件
    if (!fs.existsSync(path.join(__dirname, '..', '.env'))) {
      const envTemplate = fs.readFileSync(path.join(__dirname, '..', '.env.example'), 'utf8')

      // 生成随机密钥
      const jwtSecret = crypto.randomBytes(64).toString('hex')
      const encryptionKey = crypto.randomBytes(32).toString('hex')

      const envContent = envTemplate
        .replace('your-jwt-secret-here', jwtSecret)
        .replace('your-encryption-key-here', encryptionKey)

      fs.writeFileSync(path.join(__dirname, '..', '.env'), envContent)
    }

    // 3. 生成或使用环境变量中的管理员凭据
    const adminUsername =
      process.env.ADMIN_USERNAME || `cr_admin_${crypto.randomBytes(4).toString('hex')}`
    const adminPassword =
      process.env.ADMIN_PASSWORD ||
      crypto
        .randomBytes(16)
        .toString('base64')
        .replace(/[^a-zA-Z0-9]/g, '')
        .substring(0, 16)

    // 如果使用了环境变量，显示提示
    if (process.env.ADMIN_USERNAME || process.env.ADMIN_PASSWORD) {
      console.log(chalk.yellow('\n📌 使用环境变量中的管理员凭据'))
    }

    // 4. 创建初始化完成标记文件
    const initData = {
      initializedAt: new Date().toISOString(),
      adminUsername,
      adminPassword,
      version: '1.0.0'
    }

    fs.writeFileSync(
      path.join(__dirname, '..', 'data', 'init.json'),
      JSON.stringify(initData, null, 2)
    )

    spinner.succeed('初始化设置完成')

    console.log(chalk.green('\n✅ 设置完成！\n'))
    console.log(chalk.yellow('📋 重要信息：\n'))
    console.log(`   管理员用户名: ${chalk.cyan(adminUsername)}`)
    console.log(`   管理员密码:   ${chalk.cyan(adminPassword)}`)

    // 如果是自动生成的凭据，强调需要保存
    if (!process.env.ADMIN_USERNAME && !process.env.ADMIN_PASSWORD) {
      console.log(chalk.red('\n⚠️  请立即保存这些凭据！首次登录后建议修改密码。'))
      console.log(
        chalk.yellow(
          '\n💡 提示: 也可以通过环境变量 ADMIN_USERNAME 和 ADMIN_PASSWORD 预设管理员凭据。\n'
        )
      )
    } else {
      console.log(chalk.green('\n✅ 已使用预设的管理员凭据。\n'))
    }

    console.log(chalk.blue('🚀 启动服务：\n'))
    console.log('   npm start              - 启动生产服务')
    console.log('   npm run dev            - 启动开发服务')
    console.log('   npm run cli admin      - 管理员CLI工具\n')

    console.log(chalk.blue('🌐 访问地址：\n'))
    console.log(`   Web管理界面: http://localhost:${config.server.port}/web`)
    console.log(`   API端点:     http://localhost:${config.server.port}/api/v1/messages`)
    console.log(`   健康检查:    http://localhost:${config.server.port}/health\n`)
  } catch (error) {
    spinner.fail('初始化设置失败')
    console.error(chalk.red('❌ 错误:'), error.message)
    process.exit(1)
  }
}

// 检查是否已初始化
function checkInitialized() {
  const initFile = path.join(__dirname, '..', 'data', 'init.json')
  if (fs.existsSync(initFile)) {
    const initData = JSON.parse(fs.readFileSync(initFile, 'utf8'))
    console.log(chalk.yellow('⚠️  服务已经初始化过了！'))
    console.log(`   初始化时间: ${new Date(initData.initializedAt).toLocaleString()}`)
    console.log(`   管理员用户名: ${initData.adminUsername}`)
    console.log('\n如需重新初始化，请删除 data/init.json 文件后再运行此命令。')
    console.log(chalk.red('\n⚠️  重要提示：'))
    console.log('   1. 删除 init.json 文件后运行 npm run setup')
    console.log('   2. 生成新的账号密码后，需要重启服务才能生效')
    console.log('   3. 使用 npm run service:restart 重启服务\n')
    return true
  }
  return false
}

if (require.main === module) {
  if (!checkInitialized()) {
    setup()
  }
}

module.exports = { setup, checkInitialized }