xlxumu/backend/database/database-manager.js

const mysql = require('mysql2/promise');
const fs = require('fs');
const path = require('path');
require('dotenv').config();

// 数据库连接配置
const dbConfig = {
  host: process.env.DB_HOST,
  port: process.env.DB_PORT,
  user: process.env.DB_USER,
  password: process.env.DB_PASSWORD,
  database: process.env.DB_NAME,
  charset: 'utf8mb4',
  multipleStatements: true
};

async function initializeDatabase() {
  let connection = null;
  
  try {
    console.log('🔗 正在连接数据库...');
    console.log(`📍 连接地址: ${dbConfig.host}:${dbConfig.port}`);
    console.log(`📊 数据库名: ${dbConfig.database}`);
    
    // 创建数据库连接
    connection = await mysql.createConnection(dbConfig);
    console.log('✅ 数据库连接成功！');
    
    // 读取初始化脚本
    const sqlFilePath = path.join(__dirname, 'init_tables.sql');
    console.log(`📄 读取SQL脚本: ${sqlFilePath}`);
    
    if (!fs.existsSync(sqlFilePath)) {
      throw new Error(`SQL脚本文件不存在: ${sqlFilePath}`);
    }
    
    const sqlScript = fs.readFileSync(sqlFilePath, 'utf8');
    console.log(`📋 SQL脚本大小: ${(sqlScript.length / 1024).toFixed(2)} KB`);
    
    // 执行初始化脚本
    console.log('🚀 开始执行数据库初始化...');
    const [results] = await connection.execute(sqlScript);
    
    console.log('✅ 数据库初始化完成！');
    
    // 验证表创建情况
    console.log('🔍 验证表创建情况...');
    const [tables] = await connection.execute(
      'SELECT TABLE_NAME, TABLE_COMMENT FROM information_schema.TABLES WHERE TABLE_SCHEMA = ? ORDER BY TABLE_NAME',
      [dbConfig.database]
    );
    
    console.log(`📊 成功创建 ${tables.length} 张表:`);
    tables.forEach((table, index) => {
      console.log(`  ${index + 1}. ${table.TABLE_NAME} - ${table.TABLE_COMMENT || '无注释'}`);
    });
    
    // 检查初始数据
    console.log('\n🔍 检查初始数据...');
    
    // 检查角色表
    const [roles] = await connection.execute('SELECT COUNT(*) as count FROM roles');
    console.log(`👥 角色数量: ${roles[0].count}`);
    
    // 检查权限表
    const [permissions] = await connection.execute('SELECT COUNT(*) as count FROM permissions');
    console.log(`🔐 权限数量: ${permissions[0].count}`);
    
    // 检查用户表
    const [users] = await connection.execute('SELECT COUNT(*) as count FROM users');
    console.log(`👤 用户数量: ${users[0].count}`);
    
    if (users[0].count > 0) {
      const [adminUser] = await connection.execute(
        'SELECT username, real_name, user_type FROM users WHERE user_type = "admin" LIMIT 1'
      );
      if (adminUser.length > 0) {
        console.log(`🔧 管理员用户: ${adminUser[0].username} (${adminUser[0].real_name})`);
      }
    }
    
    console.log('\n🎉 数据库初始化成功完成！');
    console.log('📝 下一步可以启动API服务器进行测试');
    
  } catch (error) {
    console.error('❌ 数据库初始化失败:', error.message);
    
    if (error.code === 'ENOTFOUND') {
      console.error('🌐 网络连接问题：无法解析数据库主机名');
    } else if (error.code === 'ER_ACCESS_DENIED_ERROR') {
      console.error('🔐 认证失败：用户名或密码错误');
    } else if (error.code === 'ECONNREFUSED') {
      console.error('🚫 连接被拒绝：数据库服务器可能未运行或端口被封锁');
    } else if (error.code === 'ER_BAD_DB_ERROR') {
      console.error('🗃️ 数据库不存在：请先创建目标数据库');
    } else if (error.message.includes('Access denied')) {
      console.error('🛡️ IP访问限制：请检查数据库白名单设置');
      console.error('💡 解决方案：在腾讯云控制台添加当前IP到数据库白名单');
    }
    
    process.exit(1);
  } finally {
    if (connection) {
      await connection.end();
      console.log('🔚 数据库连接已关闭');
    }
  }
}

// 测试数据库连接
async function testConnection() {
  let connection = null;
  
  try {
    console.log('🧪 测试数据库连接...');
    connection = await mysql.createConnection(dbConfig);
    
    const [result] = await connection.execute('SELECT 1 as test, NOW() as current_time');
    console.log('✅ 连接测试成功！');
    console.log(`⏰ 数据库时间: ${result[0].current_time}`);
    
    const [versionResult] = await connection.execute('SELECT VERSION() as version');
    console.log(`🗄️ MySQL版本: ${versionResult[0].version}`);
    
    return true;
  } catch (error) {
    console.error('❌ 连接测试失败:', error.message);
    return false;
  } finally {
    if (connection) {
      await connection.end();
    }
  }
}

// 删除所有表（危险操作，仅用于重置）
async function dropAllTables() {
  let connection = null;
  
  try {
    console.log('⚠️ 警告：即将删除所有表！');
    console.log('3秒后开始执行...');
    await new Promise(resolve => setTimeout(resolve, 3000));
    
    connection = await mysql.createConnection(dbConfig);
    
    // 获取所有表
    const [tables] = await connection.execute(
      'SELECT TABLE_NAME FROM information_schema.TABLES WHERE TABLE_SCHEMA = ?',
      [dbConfig.database]
    );
    
    if (tables.length === 0) {
      console.log('📭 数据库中没有表需要删除');
      return;
    }
    
    // 禁用外键检查
    await connection.execute('SET FOREIGN_KEY_CHECKS = 0');
    
    // 删除所有表
    for (const table of tables) {
      console.log(`🗑️ 删除表: ${table.TABLE_NAME}`);
      await connection.execute(`DROP TABLE IF EXISTS \`${table.TABLE_NAME}\``);
    }
    
    // 启用外键检查
    await connection.execute('SET FOREIGN_KEY_CHECKS = 1');
    
    console.log('✅ 所有表已删除');
    
  } catch (error) {
    console.error('❌ 删除表失败:', error.message);
  } finally {
    if (connection) {
      await connection.end();
    }
  }
}

// 主函数
async function main() {
  const command = process.argv[2];
  
  switch (command) {
    case 'test':
      await testConnection();
      break;
    case 'init':
      await initializeDatabase();
      break;
    case 'reset':
      console.log('⚠️ 确认要重置数据库吗？这将删除所有数据！');
      console.log('如果确认，请在5秒内按Ctrl+C取消，否则将继续执行...');
      await new Promise(resolve => setTimeout(resolve, 5000));
      await dropAllTables();
      await initializeDatabase();
      break;
    default:
      console.log('🔧 锡林郭勒盟智慧养殖平台 - 数据库管理工具');
      console.log('');
      console.log('使用方法:');
      console.log('  node database-manager.js test   - 测试数据库连接');
      console.log('  node database-manager.js init   - 初始化数据库表');
      console.log('  node database-manager.js reset  - 重置数据库（删除所有表后重新创建）');
      console.log('');
      console.log('环境变量配置:');
      console.log(`  DB_HOST: ${process.env.DB_HOST || '未设置'}`);
      console.log(`  DB_PORT: ${process.env.DB_PORT || '未设置'}`);
      console.log(`  DB_USER: ${process.env.DB_USER || '未设置'}`);
      console.log(`  DB_NAME: ${process.env.DB_NAME || '未设置'}`);
      break;
  }
}

// 优雅处理进程退出
process.on('SIGINT', () => {
  console.log('\n👋 程序被用户中断');
  process.exit(0);
});

process.on('unhandledRejection', (reason, promise) => {
  console.error('❌ 未处理的Promise拒绝:', reason);
  process.exit(1);
});

// 运行主函数
main().catch(error => {
  console.error('❌ 程序执行失败:', error.message);
  process.exit(1);
});