pqs-9100_client/scripts/mysql-process-manager.js

const { spawn, exec } = require('child_process');
const path = require('path');
const fs = require('fs');
const { resolveRuntimeStrategy } = require('./path-utils');

/**
 * MySQL 进程管理器
 * 使用进程模式启动 MySQL，无需管理员权限
 */
class MySQLProcessManager {
  constructor(logWindowManager = null) {
    const isDev = !process.resourcesPath;
    const baseDir = isDev
      ? path.join(__dirname, '..')
      : path.dirname(process.resourcesPath);
    const pathStrategy = resolveRuntimeStrategy(baseDir);

    this.baseDir = baseDir;
    this.pathStrategy = pathStrategy;
    this.sourceMysqlPath = path.join(baseDir, 'mysql');
    this.mysqlPath = pathStrategy.usesSafePaths
      ? path.join(pathStrategy.safeRuntimeRoot, 'mysql')
      : this.sourceMysqlPath;
    this.binPath = path.join(this.mysqlPath, 'bin');
    this.dataPath = path.join(this.mysqlPath, 'data');
    this.configFile = path.join(this.mysqlPath, 'my.ini');
    this.logWindowManager = logWindowManager;

    // 进程模式：保存 MySQL 进程引用
    this.mysqlProcess = null;
    this.currentPort = null;
  }

  // 睡眠函数
  sleep(ms) {
    return new Promise(resolve => setTimeout(resolve, ms));
  }

  // 日志输出辅助方法
  log(type, message) {
    console.log(`[MySQL] ${message}`);
    if (this.logWindowManager) {
      this.logWindowManager.addLog(type, `[MySQL] ${message}`);
    }
  }

  /**
   * 递归复制目录。
   * 仅当文件不存在、大小变化或源文件时间更新时才覆盖，避免每次启动全量重写。
   */
  copyDirectorySync(sourceDir, targetDir, options = {}) {
    const {
      excludeTopLevelDirs = new Set(),
      excludeFileNames = new Set(),
      excludeFileExtensions = new Set()
    } = options;

    if (!fs.existsSync(sourceDir)) {
      return;
    }

    if (!fs.existsSync(targetDir)) {
      fs.mkdirSync(targetDir, { recursive: true });
    }

    const entries = fs.readdirSync(sourceDir, { withFileTypes: true });

    entries.forEach((entry) => {
      const sourcePath = path.join(sourceDir, entry.name);
      const targetPath = path.join(targetDir, entry.name);
      const isTopLevelEntry = sourceDir === this.sourceMysqlPath;

      if (entry.isDirectory()) {
        if (isTopLevelEntry && excludeTopLevelDirs.has(entry.name)) {
          return;
        }

        this.copyDirectorySync(sourcePath, targetPath, options);
        return;
      }

      const extension = path.extname(entry.name).toLowerCase();
      if (excludeFileNames.has(entry.name) || excludeFileExtensions.has(extension)) {
        return;
      }

      let shouldCopy = !fs.existsSync(targetPath);
      if (!shouldCopy) {
        const sourceStat = fs.statSync(sourcePath);
        const targetStat = fs.statSync(targetPath);
        shouldCopy = sourceStat.size !== targetStat.size || sourceStat.mtimeMs > targetStat.mtimeMs;
      }

      if (shouldCopy) {
        fs.copyFileSync(sourcePath, targetPath);
      }
    });
  }

  /**
   * 非 ASCII 路径下，将 MySQL 运行环境同步到英文安全目录。
   */
  ensureRuntimeEnvironment() {
    if (!this.pathStrategy.usesSafePaths) {
      return;
    }

    this.log('system', '检测到应用路径包含非 ASCII 字符，启用 MySQL 英文安全运行目录');
    this.log('system', `MySQL 源目录: ${this.sourceMysqlPath}`);
    this.log('system', `MySQL 运行目录: ${this.mysqlPath}`);
    this.log('system', `MySQL 数据目录: ${this.dataPath}`);

    fs.mkdirSync(this.mysqlPath, { recursive: true });
    fs.mkdirSync(path.dirname(this.dataPath), { recursive: true });

    this.copyDirectorySync(this.sourceMysqlPath, this.mysqlPath, {
      excludeTopLevelDirs: new Set(['data', 'data_backup']),
      excludeFileNames: new Set(['my.ini', 'mysql_error.log', 'mysql_slow.log']),
      excludeFileExtensions: new Set(['.pid', '.err'])
    });

    this.ensureSeedData();
  }

  /**
   * 首次进入英文安全路径模式时，将打包内预置数据库复制到安全数据目录。
   */
  ensureSeedData() {
    const hasTargetData = fs.existsSync(this.dataPath) && fs.readdirSync(this.dataPath).length > 0;
    if (hasTargetData) {
      this.log('system', '检测到已有 MySQL 安全数据目录，直接复用');
      return;
    }

    const sourceDataPath = path.join(this.sourceMysqlPath, 'data');
    if (fs.existsSync(sourceDataPath) && fs.readdirSync(sourceDataPath).length > 0) {
      this.log('system', '首次启动，正在复制预置 MySQL 数据到安全数据目录...');
      this.copyDirectorySync(sourceDataPath, this.dataPath);
      this.log('success', '预置 MySQL 数据复制完成');
      return;
    }

    fs.mkdirSync(this.dataPath, { recursive: true });
    this.log('warn', '未找到预置 MySQL 数据目录，后续将按空库初始化');
  }

  /**
   * 执行命令并返回Promise
   */
  execCommand(command) {
    return new Promise((resolve, reject) => {
      exec(command, { encoding: 'utf8' }, (error, stdout, stderr) => {
        if (error) {
          const err = new Error(error.message || stderr || stdout || 'Command execution failed');
          err.code = error.code;
          err.stdout = stdout;
          err.stderr = stderr;
          reject(err);
        } else {
          resolve({ stdout, stderr });
        }
      });
    });
  }

  /**
   * 读取配置文件中的端口
   */
  readPortFromConfig() {
    try {
      if (fs.existsSync(this.configFile)) {
        const content = fs.readFileSync(this.configFile, 'utf-8');
        const match = content.match(/port\s*=\s*(\d+)/);
        return match ? parseInt(match[1]) : 3306;
      }
    } catch (error) {
      console.error('[MySQL] Failed to read port from config:', error);
    }
    return 3306;
  }

  /**
   * 生成 my.ini 配置文件
   */
  generateMyIni(port) {
    const basedir = this.mysqlPath.replace(/\\/g, '/');
    const datadir = this.dataPath.replace(/\\/g, '/');

    const config = `[mysqld]
# 基础路径配置
basedir=${basedir}
datadir=${datadir}

# 网络配置
port=${port}
bind-address=0.0.0.0

# 字符集配置
character-set-server=utf8mb4
collation-server=utf8mb4_unicode_ci

# 性能优化
max_connections=200
innodb_buffer_pool_size=128M
innodb_log_file_size=64M

# 日志配置
log_error=${basedir}/mysql_error.log
slow_query_log=1
slow_query_log_file=${basedir}/mysql_slow.log
long_query_time=2

# 安全配置
sql_mode=STRICT_TRANS_TABLES,NO_ZERO_IN_DATE,NO_ZERO_DATE,ERROR_FOR_DIVISION_BY_ZERO,NO_ENGINE_SUBSTITUTION

[client]
default-character-set=utf8mb4
`;

    fs.writeFileSync(this.configFile, config, 'utf-8');
    this.log('system', `生成配置文件，端口: ${port}`);
  }

  /**
   * 初始化数据库
   */
  async initializeDatabase() {
    return new Promise((resolve, reject) => {
      this.log('system', '正在初始化 MySQL 数据库...');

      const mysqld = path.join(this.binPath, 'mysqld.exe');
      const initProcess = spawn(mysqld, [
        '--defaults-file=' + this.configFile,
        '--initialize-insecure'
      ], {
        cwd: this.mysqlPath,
        stdio: 'pipe'
      });

      initProcess.stdout.on('data', (data) => {
        this.log('system', data.toString().trim());
      });

      initProcess.stderr.on('data', (data) => {
        this.log('system', data.toString().trim());
      });

      initProcess.on('close', (code) => {
        if (code === 0) {
          this.log('success', '数据库初始化成功');
          resolve();
        } else {
          reject(new Error(`数据库初始化失败，退出码: ${code}`));
        }
      });

      initProcess.on('error', (err) => {
        reject(err);
      });
    });
  }

  /**
   * 检查数据库是否已初始化
   */
  isDatabaseInitialized() {
    return fs.existsSync(this.dataPath) && fs.readdirSync(this.dataPath).length > 0;
  }

  /**
   * 检查是否有残留的 mysqld 进程
   */
  async checkAndKillOrphanProcess() {
    try {
      // 检查是否有 mysqld.exe 进程在运行
      const { stdout } = await this.execCommand('tasklist /FI "IMAGENAME eq mysqld.exe" /FO CSV /NH');

      if (stdout.includes('mysqld.exe')) {
        this.log('warn', '检测到残留的 MySQL 进程，正在清理...');

        // 尝试优雅关闭
        try {
          const mysqladmin = path.join(this.binPath, 'mysqladmin.exe');
          await this.execCommand(`"${mysqladmin}" -u root shutdown`);
          await this.sleep(2000);
          this.log('success', '残留进程已优雅关闭');
        } catch (e) {
          // 优雅关闭失败，强制杀死
          this.log('warn', '优雅关闭失败，强制终止...');
          await this.execCommand('taskkill /F /IM mysqld.exe');
          await this.sleep(1000);
          this.log('success', '残留进程已强制终止');
        }
      }
    } catch (error) {
      // tasklist 命令失败通常意味着没有找到进程，这是正常的
    }
  }

  /**
   * 进程模式：启动 MySQL 进程
   */
  async startMySQLProcess() {
    return new Promise((resolve, reject) => {
      const mysqld = path.join(this.binPath, 'mysqld.exe');

      this.log('system', '正在启动 MySQL 进程...');
      this.log('system', `可执行文件: ${mysqld}`);
      this.log('system', `配置文件: ${this.configFile}`);

      // 检查文件是否存在
      if (!fs.existsSync(mysqld)) {
        return reject(new Error(`mysqld.exe 不存在: ${mysqld}`));
      }
      if (!fs.existsSync(this.configFile)) {
        return reject(new Error(`配置文件不存在: ${this.configFile}`));
      }

      this.mysqlProcess = spawn(mysqld, [
        `--defaults-file=${this.configFile}`,
        '--console'
      ], {
        cwd: this.mysqlPath,
        stdio: ['ignore', 'pipe', 'pipe'],
        windowsHide: true
      });

      let startupComplete = false;
      let startupTimeout = null;

      // 监听 stderr（MySQL 主要输出在 stderr）
      this.mysqlProcess.stderr.on('data', (data) => {
        const msg = data.toString().trim();
        if (msg) {
          // 检查启动成功标志
          if (msg.includes('ready for connections') || msg.includes('port:')) {
            if (!startupComplete) {
              startupComplete = true;
              if (startupTimeout) clearTimeout(startupTimeout);
              this.log('success', 'MySQL 进程已就绪');
              resolve(true);
            }
          }

          // 输出日志（过滤一些不重要的信息）
          if (!msg.includes('[Note]') || msg.includes('error') || msg.includes('ready')) {
            this.log('system', `[mysqld] ${msg}`);
          }
        }
      });

      this.mysqlProcess.stdout.on('data', (data) => {
        const msg = data.toString().trim();
        if (msg) {
          this.log('system', `[mysqld] ${msg}`);
        }
      });

      this.mysqlProcess.on('error', (err) => {
        this.log('error', `MySQL 进程启动失败: ${err.message}`);
        this.mysqlProcess = null;
        reject(err);
      });

      this.mysqlProcess.on('exit', (code, signal) => {
        this.log('system', `MySQL 进程退出，代码: ${code}, 信号: ${signal}`);
        this.mysqlProcess = null;

        // 如果还没完成启动就退出了，说明启动失败
        if (!startupComplete) {
          reject(new Error(`MySQL 进程异常退出，代码: ${code}`));
        }
      });

      // 超时处理（30秒）
      startupTimeout = setTimeout(() => {
        if (!startupComplete && this.mysqlProcess) {
          // 进程还在运行，认为启动成功
          startupComplete = true;
          this.log('success', 'MySQL 进程启动完成（超时检测）');
          resolve(true);
        }
      }, 30000);
    });
  }

  /**
   * 进程模式：停止 MySQL 进程
   */
  async stopMySQLProcess() {
    if (!this.mysqlProcess) {
      this.log('system', 'MySQL 进程未运行');
      return;
    }

    this.log('system', '正在停止 MySQL 进程...');

    try {
      // 方式1：通过 mysqladmin 优雅关闭
      const mysqladmin = path.join(this.binPath, 'mysqladmin.exe');
      await this.execCommand(`"${mysqladmin}" -u root shutdown`);

      // 等待进程退出
      await this.sleep(3000);

      if (this.mysqlProcess && !this.mysqlProcess.killed) {
        // 如果还没退出，强制杀死
        this.log('warn', '优雅关闭超时，强制终止...');
        this.mysqlProcess.kill('SIGTERM');
        await this.sleep(1000);
      }

      this.log('success', 'MySQL 进程已关闭');
    } catch (error) {
      // mysqladmin 关闭失败，强制杀死进程
      this.log('warn', `优雅关闭失败: ${error.message}，强制终止...`);

      if (this.mysqlProcess) {
        this.mysqlProcess.kill('SIGTERM');
        await this.sleep(1000);
      }

      // 确保进程被杀死
      try {
        await this.execCommand('taskkill /F /IM mysqld.exe');
        await this.sleep(2000); // 等待文件句柄释放
      } catch (e) {
        // 忽略
      }
    }

    // 最终检查：确保所有 mysqld.exe 进程都被终止
    try {
      const { stdout } = await this.execCommand('tasklist /FI "IMAGENAME eq mysqld.exe" /FO CSV /NH');
      if (stdout.includes('mysqld.exe')) {
        this.log('warn', '检测到残留进程，强制清理...');
        await this.execCommand('taskkill /F /IM mysqld.exe');
        await this.sleep(2000); // 额外等待以确保文件句柄释放
      }
    } catch (e) {
      // 忽略
    }

    this.mysqlProcess = null;
    this.currentPort = null;
  }

  /**
   * 检查 MySQL 进程是否在运行
   */
  isMySQLRunning() {
    return this.mysqlProcess !== null && !this.mysqlProcess.killed;
  }

  /**
   * 主流程：确保 MySQL 运行（进程模式）
   * @param {function} findAvailablePort - 查找可用端口的函数
   * @param {function} waitForPort - 等待端口就绪的函数
   * @returns {Promise<number>} 返回 MySQL 运行的端口
   */
  async ensureServiceRunning(findAvailablePort, waitForPort) {
    this.ensureRuntimeEnvironment();

    this.log('system', '开始 MySQL 进程检查流程（进程模式）');
    this.log('system', `路径策略: ${this.pathStrategy.usesSafePaths ? '英文安全路径模式' : '应用目录直启模式'}`);
    this.log('system', `MySQL 路径: ${this.mysqlPath}`);
    this.log('system', `配置文件: ${this.configFile}`);

    // 1. 检查是否已有 MySQL 进程在运行
    if (this.isMySQLRunning()) {
      const port = this.currentPort || this.readPortFromConfig();
      this.log('success', `MySQL 进程已在运行，端口: ${port}`);
      return port;
    }

    // 2. 检查并清理残留进程
    await this.checkAndKillOrphanProcess();

    // 3. 查找可用端口
    const port = await findAvailablePort(3306, 100);
    this.log('system', `找到可用端口: ${port}`);

    // 4. 检查数据库是否已初始化
    if (!this.isDatabaseInitialized()) {
      this.log('system', '数据库未初始化，开始初始化...');
      this.generateMyIni(port);
      await this.initializeDatabase();
    } else {
      this.log('system', '检测到已有数据库目录');
      // 更新配置文件（可能端口变了）
      this.generateMyIni(port);
    }

    // 5. 启动 MySQL 进程
    this.log('system', '启动 MySQL 进程...');
    await this.startMySQLProcess();

    // 6. 等待端口就绪
    const finalPort = this.readPortFromConfig();
    this.currentPort = finalPort;

    this.log('system', `等待端口 ${finalPort} 就绪...`);
    const portReady = await waitForPort(finalPort, 30000);

    if (!portReady) {
      throw new Error(`MySQL 端口 ${finalPort} 未能在 30 秒内就绪`);
    }

    this.log('success', `MySQL 进程启动成功，端口: ${finalPort}`);
    return finalPort;
  }
}

module.exports = MySQLProcessManager;