require('dotenv').config();
const pool = require('../src/config/database');

async function cleanDatabase() {
  const connection = await pool.getConnection();
  
  try {
    console.log('⚠️  警告：此操作将清空所有数据（保留管理员账号）\n');
    
    // 清空表（保留结构）
    await connection.query('SET FOREIGN_KEY_CHECKS = 0');
    
    await connection.query('DELETE FROM operation_logs');
    console.log('✅ 清空操作日志表');
    
    await connection.query('DELETE FROM attachments');
    console.log('✅ 清空附件表');
    
    await connection.query('DELETE FROM approvals');
    console.log('✅ 清空审批表');
    
    await connection.query('DELETE FROM pool_claim_records');
    console.log('✅ 清空公海池领取记录表');
    
    await connection.query('DELETE FROM followup_records');
    console.log('✅ 清空跟进记录表');
    
    await connection.query('DELETE FROM customers');
    console.log('✅ 清空客户表');
    
    await connection.query("DELETE FROM users WHERE username != 'admin'");
    console.log('✅ 清空用户表（保留管理员）');
    
    await connection.query('SET FOREIGN_KEY_CHECKS = 1');
    
    console.log('\n🎉 数据库清理完成！\n');
    console.log('管理员账号已保留: admin / admin123\n');
    
  } catch (error) {
    console.error('清理数据库失败:', error);
    process.exit(1);
  } finally {
    connection.release();
    await pool.end();
  }
}

cleanDatabase();