specflow-dev-service 0.0.0-beta.10

package/dist/orchestrator.js

@@ -0,0 +1,679 @@
+'use strict';
+
+var node_child_process = require('node:child_process');
+var node_fs = require('node:fs');
+var node_path = require('node:path');
+var defaults = require('./config/defaults.js');
+var loader = require('./config/loader.js');
+var logger = require('./logger.js');
+require('cross-spawn');
+var client = require('./runner/client.js');
+
+/**
+ * ForgeService — 核心协调器
+ *
+ * 职责：
+ *   1. 解析 specflow.config.ts 配置文件
+ *   2. 前端启动逻辑（端口管理 + Vite 进程管理）
+ *   3. 后端热更新逻辑（air）
+ *
+ * 日志原则：只输出关键信息，静默处理正常流程
+ */
+
+
+// ============================================================
+// 类型定义
+// ============================================================
+
+// ============================================================
+// Go 进程管理器（基于 air，极致精简）
+// ============================================================
+
+class GoProcessManager {
+  process = null;
+  state = 'idle';
+  restartCount = 0;
+  /** 端口真正就绪的 Promise（running 信号后开始等待） */
+  portReadyResolver = null;
+  portReadyPromise = null;
+  constructor(serviceDir, port, env, options) {
+    this.serviceDir = serviceDir;
+    this.port = port;
+    this.env = env;
+    this.options = options;
+  }
+  get isRunning() {
+    return this.state === 'running' && this.process !== null && !this.process.killed;
+  }
+  get pid() {
+    return this.process?.pid ?? undefined;
+  }
+  async start() {
+    if (this.state === 'starting' || this.state === 'running') return;
+    this.state = 'starting';
+
+    // 用户自定义命令优先
+    if (this.options?.command) {
+      await this.startCustom();
+      return;
+    }
+
+    // 默认使用 air
+    await this.startWithAir();
+  }
+  async stop() {
+    if (this.state === 'stopped' || this.state === 'idle') return;
+    this.state = 'stopped';
+    if (this.process) {
+      this.killProcess();
+      this.process = null;
+    }
+    this.state = 'idle';
+  }
+
+  /** 使用 air 启动 */
+  async startWithAir() {
+    // 检查并自动安装 air
+    if (!(await this.ensureAirInstalled())) return;
+
+    // 自动生成 .air.toml（不存在时）
+    this.ensureAirToml();
+    const args = this.buildAirArgs();
+    this.process = node_child_process.spawn('air', args, {
+      cwd: this.serviceDir,
+      stdio: ['ignore', 'pipe', 'pipe'],
+      env: this.env,
+      // 不额外注入 PORT，由 Go 程序自身配置决定端口
+      windowsHide: true,
+      shell: process.platform === 'win32'
+    });
+    this.setupAirOutputHandler();
+
+    // 创建端口就绪 Promise（output handler 的 running... 信号会触发 resolve）
+    this.portReadyPromise = new Promise(resolve => {
+      this.portReadyResolver = resolve;
+    });
+
+    // 等待 air 启动完成
+    await new Promise(res => setTimeout(res, 1200));
+    if (this.process && !this.process.killed) {
+      this.state = 'running';
+      this.restartCount++;
+      // 等待端口真正就绪（覆盖 MySQL/Redis/tRPC 初始化窗口）
+      await this.portReadyPromise;
+    }
+  }
+
+  /**
+   * 自动生成 .air.toml 配置文件（仅当不存在时）
+   *
+   * 设计原则：
+   *   - 生成的配置与用户手动编写的完全等价，无任何降级
+   *   - 平台自适应：Windows 生成 .exe，Linux/Mac 无后缀
+   *   - 输出同时写 stdout+stderr（forge-service 通过 pipe 捕获）
+   */
+  ensureAirToml() {
+    const tomlPath = node_path.resolve(this.serviceDir, '.air.toml');
+    const binExt = process.platform === 'win32' ? '.exe' : '';
+    const tmpDir = 'tmp';
+    const binName = `./tmp/main${binExt}`;
+
+    // 检测损坏的旧版本：如果文件存在但包含未替换的模板变量，则重新生成
+    if (node_fs.existsSync(tomlPath)) {
+      try {
+        const existing = require('node:fs').readFileSync(tomlPath, 'utf-8');
+        if (existing.includes('${binExt}') || existing.includes('${tmpDir}')) {
+          // 损坏的旧版 → 删除并重新生成
+          require('node:fs').unlinkSync(tomlPath);
+        } else {
+          return; // 已有有效配置，不覆盖
+        }
+      } catch {
+        /* 读取失败，重新生成 */
+      }
+    }
+
+    // 确保 tmp 目录存在
+    const tmpFullPath = node_path.resolve(this.serviceDir, tmpDir);
+    if (!node_fs.existsSync(tmpFullPath)) {
+      node_fs.mkdirSync(tmpFullPath, {
+        recursive: true
+      });
+    }
+    const tomlContent = ['# Air 配置文件 - Go 热更新（自动生成，可手动修改）', '', 'root = "."', `tmp_dir = "${tmpDir}"`, 'working_dir = "."', '', '[build]', `  entrypoint = "${binName}"`, `  cmd = "go build -o ./tmp/main${binExt} ."`, '  delay = 1000', `  exclude_dir = ["${tmpDir}", "vendor", "client", "bin", "log", "node_modules", ".git"]`, '  include_ext = ["go", "yaml", "yml", "json", "toml", "mod", "sum"]', '  kill_delay = "3s"', '  build_delay = "6s"', '  log = ""', '  send_interrupt = true', '  stop_on_error = false', '', '[log]', '  time = false', '  outputs = ["stdout", "stderr"]', '', '[color]', '  main = "magenta"', '  watcher = "cyan"', '  build = "yellow"', '  runner = "green"', '', '[misc]', '  clean_on_exit = true'].join('\n');
+    node_fs.writeFileSync(tomlPath, tomlContent, 'utf-8');
+    logger.logProgress('go', 'config', `.air.toml 已生成 (${binName})`);
+  }
+
+  /** 构建 air 参数（始终使用 .air.toml） */
+  buildAirArgs() {
+    // 始终使用项目中的 .air.toml（ensureAirToml 保证其存在）
+    const args = ['-c', '.air.toml'];
+    if (this.options?.args) args.push(...this.options.args);
+    return args;
+  }
+
+  /** 检查并自动安装 air */
+  async ensureAirInstalled() {
+    // 兼容旧版 air（v1.65.x 用 -v，新版用 --version）
+    const versionFlags = ['-v', '--version'];
+    for (const flag of versionFlags) {
+      try {
+        node_child_process.execSync(`air ${flag}`, {
+          stdio: 'pipe',
+          timeout: 2000
+        });
+        return true; // 已安装
+      } catch {
+        // 这个 flag 不支持，试下一个
+      }
+    }
+
+    // 都没成功 → 尝试自动安装
+    logger.logProgress('go', 'installing air', 'go install github.com/air-verse/air@latest');
+    try {
+      await new Promise((resolvePromise, reject) => {
+        const installProc = node_child_process.spawn('go', ['install', 'github.com/air-verse/air@latest'], {
+          stdio: 'pipe',
+          env: this.env,
+          windowsHide: true
+        });
+        installProc.on('close', code => {
+          if (code === 0) resolvePromise();else reject(new Error(`air 安装失败 (exit ${code})`));
+        });
+        installProc.on('error', reject);
+      });
+
+      // 验证安装
+      for (const flag of versionFlags) {
+        try {
+          node_child_process.execSync(`air ${flag}`, {
+            stdio: 'pipe',
+            timeout: 2000
+          });
+          return true;
+        } catch {
+          /* continue */
+        }
+      }
+      throw new Error('安装后验证失败');
+    } catch (err) {
+      logger.logError(`[go] air 安装失败: ${err instanceof Error ? err.message : String(err)}`);
+      logger.logError('[go] 请手动执行: go install github.com/air-verse/air@latest');
+      this.state = 'idle';
+      return false;
+    }
+  }
+
+  /**
+   * 处理 air 输出 — 用户友好模式
+   *
+   * 正常情况只显示：
+   *   [go] compiling...        （编译进度）
+   *   [go] ✓ ready  http://localhost:11080  （启动成功+端口）
+   *   [go] ✓ reloaded         （热更新成功）
+   *
+   * 出错才显示：
+   *   [go] ✗ compile error:   （编译失败，显示具体错误）
+   *   [go] ✗ crashed          （运行时崩溃，显示panic）
+   */
+  setupAirOutputHandler() {
+    if (!this.process) return;
+    const prefix = '\x1b[36m[go]\x1b[0m';
+    const isTTY = process.stdout.isTTY === true;
+    let phase = 'idle';
+    let errorLines = []; // 编译错误（只在 building 收集）
+    let buildDots = 0;
+    let buildTimer = null;
+    const clearLine = () => {
+      if (isTTY) process.stdout.write('\r\x1b[K');
+    };
+    const startBuildProgress = () => {
+      if (buildTimer) return;
+      buildDots = 0;
+      if (isTTY) {
+        buildTimer = setInterval(() => {
+          buildDots = (buildDots + 1) % 4;
+          process.stdout.write(`\r\x1b[K${prefix} compiling${'.'.repeat(buildDots)}${' '.repeat(3 - buildDots)}`);
+        }, 300);
+      } else {
+        process.stdout.write(`${prefix} compiling...\n`);
+      }
+    };
+    const stopBuildProgress = () => {
+      if (buildTimer) {
+        clearInterval(buildTimer);
+        buildTimer = null;
+      }
+      clearLine();
+    };
+
+    /** 等待端口真正就绪后输出 ready 并 resolve portReadyPromise */
+    const waitForPortReady = () => {
+      const {
+        port
+      } = this;
+      const maxWait = 15000; // 最多等 15s（覆盖慢速 DB/Redis 初始化）
+      const start = Date.now();
+      const check = () => {
+        // 先检查进程是否还活着（避免死循环）
+        if (!this.process || this.process.killed) return;
+        try {
+          const socket = require('net').createConnection(port, '127.0.0.1');
+          socket.on('connect', () => {
+            socket.destroy();
+            // 端口就绪 → 输出 + 通知 startWithAir 继续
+            if (this.restartCount > 1) {
+              console.log(`${prefix} \x1b[32m✓ reloaded\x1b[0m`);
+            } else {
+              console.log(`${prefix} \x1b[32m✓ ready\x1b[0m  \x1b[4mhttp://localhost:${port}\x1b[0m`);
+            }
+            this.portReadyResolver?.();
+          });
+          socket.on('error', () => {
+            socket.destroy();
+            if (Date.now() - start < maxWait) setTimeout(check, 300);else {
+              // 超时：仍然放行（可能是服务没监听此端口），打印警告
+              console.log(`${prefix} \x1b[33m⚠ port :${port} not responding in ${maxWait / 1000}s, proceeding anyway\x1b[0m`);
+              this.portReadyResolver?.();
+            }
+          });
+          setTimeout(() => {
+            try {
+              socket.destroy();
+            } catch {
+              /* noop */
+            }
+          }, 2000);
+        } catch {
+          // net 模块异常，直接放行
+          this.portReadyResolver?.();
+        }
+      };
+      check();
+    };
+
+    // ---- 核心行处理 ----
+    const stripAnsi = s => s.replace(/\x1b\[[0-9;]*[a-zA-Z]/g, '').replace(/\r/g, '');
+    const stripAirPrefix = s => s.replace(/^[\s]*[✗✓]?\s*\[air\]\s*/i, '');
+    const isAirNoise = t => /^[_/\\ |]+$/.test(t) || /^(watching|!exclude|cleaning|see you)/i.test(t) || t.startsWith('air') || /^v\d+\.\d+/.test(t) || /^[|/_\\-]+/.test(t) && t.length < 40 || /built with Go/i.test(t);
+    const handlePhaseSignal = t => {
+      if (/^building\.\.\./.test(t)) {
+        phase = 'building';
+        errorLines = [];
+        startBuildProgress();
+        return true;
+      }
+      if (/^running\.\.\./.test(t)) {
+        stopBuildProgress();
+        phase = 'running';
+        waitForPortReady();
+        return true;
+      }
+      const exitMatch = t.match(/Process Exit with Code:\s*(\d+)/);
+      if (exitMatch) {
+        stopBuildProgress();
+        const code = exitMatch[1];
+        phase = code === '0' ? 'idle' : 'failed';
+        if (code !== '0' && errorLines.length > 0) {
+          console.log(`\n${prefix} \x1b[31m✗ compile error:\x1b[0m`);
+          for (const line of errorLines.slice(0, 10)) {
+            console.log(`  \x1b[31m${line}\x1b[0m`);
+          }
+          errorLines = [];
+        }
+        return true;
+      }
+      return false;
+    };
+    const handleBuildingPhase = t => {
+      if (/^\d{4}-\d{2}-\d{2}/.test(t)) return;
+      if (/\.go:\d+:\d+:/.test(t) || /^#\s/.test(t)) {
+        errorLines.push(t);
+        return;
+      }
+      if (/cannot\s+(find|import)\s+package|:\s*undefined:|declared\s+and\s+not\s+used|imported\s+and\s+not\s+used/.test(t)) {
+        errorLines.push(t);
+      }
+    };
+    const handleLine = raw => {
+      const t = stripAirPrefix(stripAnsi(raw)).trim();
+      if (!t || isAirNoise(t)) return;
+      if (handlePhaseSignal(t)) return;
+      if (phase === 'building') handleBuildingPhase(t);
+    };
+
+    // ---- 绑定流 ----
+    let buf = '';
+    const drain = chunk => {
+      buf += chunk.toString();
+      const lines = buf.split('\n');
+      buf = lines.pop() || '';
+      for (const line of lines) handleLine(line);
+    };
+    this.process.stdout?.on('data', drain);
+    this.process.stderr?.on('data', drain);
+    this.process.on('error', () => {
+      stopBuildProgress();
+      this.state = 'idle';
+    });
+    this.process.on('exit', (_code, signal) => {
+      stopBuildProgress();
+      if (signal !== 'SIGTERM' && signal !== 'SIGINT') {
+        logger.logError(`[go] air stopped`);
+      }
+      this.process = null;
+      this.state = 'idle';
+    });
+  }
+
+  /** 自定义命令启动 */
+  async startCustom() {
+    const cmd = this.options.command;
+    const args = this.options.args || [];
+    this.process = node_child_process.spawn(cmd, args, {
+      cwd: this.serviceDir,
+      stdio: ['ignore', 'pipe', 'pipe'],
+      env: this.env,
+      windowsHide: true,
+      shell: process.platform === 'win32'
+    });
+    this.setupCustomOutputHandler(cmd);
+    await new Promise(res => setTimeout(res, 1500));
+    if (this.process && !this.process.killed) {
+      this.state = 'running';
+      this.restartCount++;
+      logger.logProgress('go', 'ready', `http://localhost:${this.port} (${cmd})`);
+    }
+  }
+  setupCustomOutputHandler(cmd) {
+    if (!this.process) return;
+    const prefix = `\x1b[36m[go]\x1b[0m`;
+
+    // ---- 绑定 stdout + stderr，复用 handleLine 过滤 ----
+    let buf = '';
+    const flush = () => {
+      if (!buf.trim()) {
+        buf = '';
+        return;
+      }
+      for (const line of buf.split('\n')) {
+        const t = line.trim();
+        if (!t) continue;
+        // 自定义命令：透传关键信息，过滤噪音
+        if (/launch\s+success|listening|serving|started|ready|error|panic|fatal/i.test(t)) {
+          console.log(`${prefix} ${t}`);
+        }
+      }
+      buf = '';
+    };
+    this.process.stdout?.on('data', d => {
+      buf += d.toString();
+      const nl = buf.lastIndexOf('\n');
+      if (nl !== -1) {
+        const chunk = buf.slice(0, nl);
+        buf = buf.slice(nl + 1);
+        for (const line of chunk.split('\n')) {
+          const t = line.trim();
+          if (!t) continue;
+          if (/launch\s+success|listening|serving|started|ready|error|panic|fatal/i.test(t)) {
+            console.log(`${prefix} ${t}`);
+          }
+        }
+      }
+    });
+    this.process.stderr?.on('data', d => {
+      const text = d.toString().trim();
+      if (text) logger.logError(`[${cmd}] ${text}`);
+    });
+    this.process.on('error', err => {
+      logger.logError(`[go] ${err.message}`);
+      this.state = 'idle';
+    });
+    this.process.on('exit', () => {
+      flush();
+      this.process = null;
+      this.state = 'idle';
+    });
+  }
+  killProcess() {
+    if (!this.process || this.process.killed) return;
+    try {
+      if (process.platform === 'win32') {
+        node_child_process.execSync(`taskkill /PID ${this.process.pid} /T /F`, {
+          stdio: 'pipe',
+          timeout: 5000,
+          windowsHide: true
+        });
+      } else {
+        this.process.kill('SIGTERM');
+        // 给进程 1 秒优雅退出
+        setTimeout(() => {
+          if (this.process && !this.process.killed) {
+            this.process.kill('SIGKILL');
+          }
+        }, 1000);
+      }
+    } catch {
+      // 进程可能已退出，忽略错误
+    }
+  }
+}
+
+// ============================================================
+// ForgeService 主类
+// ============================================================
+
+class ForgeService {
+  running = false;
+  clientHandle = null;
+  goManager = null;
+  constructor(config, cwd, mode) {
+    this.config = config;
+    this.cwd = cwd;
+    this.mode = mode;
+  }
+
+  /** 停止所有服务 */
+  async stop() {
+    if (this.goManager) {
+      await this.goManager.stop();
+      this.goManager = null;
+    }
+    if (this.clientHandle) {
+      client.stopClient(this.clientHandle);
+      this.clientHandle = null;
+    }
+    this.running = false;
+  }
+
+  /** 获取状态 */
+  getStatus() {
+    return {
+      running: this.running,
+      config: {
+        clientPort: this.config.client.port,
+        servicePort: this.config.service.port
+      },
+      client: this.getClientStatus(),
+      service: this.getServiceStatus()
+    };
+  }
+
+  /** 启动所有服务（Go 后端先启动，前端后启动） */
+  async start(options) {
+    const opt = {
+      client: true,
+      service: true,
+      ...options
+    };
+
+    // === 第零步：按 preset 注入自定义环境变量（前端/Go 进程均可读取） ===
+    this.injectEnvPreset();
+
+    // === 第一步：启动 Go 后端 ===
+    if (opt.service) {
+      logger.logProgress('go', 'starting', `http://localhost:${this.config.service.port}`);
+      await this.startGoService();
+      // GoProcessManager.start() 内部已等待端口就绪（最多 15s）
+      if (this.goManager?.isRunning) {
+        logger.logProgress('go', 'ready', `http://localhost:${this.config.service.port}`);
+      } else {
+        logger.logWarn(`[go] failed — http://localhost:${this.config.service.port} 未就绪`);
+      }
+    }
+
+    // === 第二步：启动前端 ===
+    if (opt.client) {
+      logger.logProgress('client', 'starting', `http://localhost:${this.config.client.port}`);
+      const clientConfig = {
+        port: this.config.client.port,
+        dir: node_path.resolve(this.cwd, this.config.client.dir)
+      };
+      this.clientHandle = client.startClient(clientConfig);
+    }
+    this.running = true;
+  }
+
+  /** 按 preset 名称注入环境变量到 process.env */
+  injectEnvPreset() {
+    const presets = this.config.env || {};
+    const presetKeys = Object.keys(presets);
+    if (presetKeys.length === 0) return;
+
+    // 确定使用哪个 preset
+    const name = this.mode || presetKeys[0];
+    const vars = presets[name];
+    if (!vars) {
+      logger.logError(`[specflow] 环境变量分组 "${name}" 不存在，可选: ${presetKeys.join(', ')}`);
+      return;
+    }
+    for (const [key, value] of Object.entries(vars)) {
+      process.env[key] = value;
+    }
+    logger.logProgress('specflow', 'env', `"${name}" (${Object.keys(vars).length} 个变量)`);
+  }
+  getClientStatus() {
+    const {
+      port
+    } = this.config.client;
+    const pid = this.clientHandle?.child?.pid;
+    return {
+      port,
+      running: pid !== null && pid !== undefined,
+      pid
+    };
+  }
+  getServiceStatus() {
+    const {
+      port
+    } = this.config.service;
+    const pid = this.goManager?.pid;
+    return {
+      port,
+      running: this.goManager?.isRunning ?? false,
+      pid
+    };
+  }
+  async waitForPort(port, timeoutMs) {
+    const start = Date.now();
+    while (Date.now() - start < timeoutMs) {
+      try {
+        const socket = require('net').createConnection(port, '127.0.0.1');
+        await new Promise((resolve, reject) => {
+          socket.on('connect', () => {
+            socket.destroy();
+            resolve();
+          });
+          socket.on('error', reject);
+          setTimeout(() => {
+            socket.destroy();
+            reject(new Error('timeout'));
+          }, 2000);
+        });
+        return; // 端口可达
+      } catch {
+        // 端口未就绪，等 500ms 再试
+        await new Promise(r => setTimeout(r, 500));
+      }
+    }
+  }
+
+  // ---- Go 服务启动 ----
+
+  async startGoService() {
+    const serviceDir = node_path.resolve(this.cwd, this.config.service.dir);
+    if (!node_fs.existsSync(serviceDir)) return;
+
+    // 检查 main.go 或 go.mod 是否存在
+    const hasMainGo = node_fs.existsSync(node_path.resolve(serviceDir, 'main.go'));
+    const hasGoMod = node_fs.existsSync(node_path.resolve(serviceDir, 'go.mod'));
+    if (!hasMainGo && !hasGoMod) return;
+    const goEnv = {
+      ...process.env
+      // 不注入 PORT 环境变量 — 让 Go 程序使用自身配置文件（yaml/toml）中的端口设置
+      // 强制注入 PORT 会导致 tRPC 等框架的行为与直接运行 air 时不一致
+    };
+
+    // 解析 dev 配置
+    const {
+      dev
+    } = this.config.service;
+    let devOpts;
+    if (typeof dev === 'string' && dev !== 'air') {
+      // "make dev" 或 "./scripts/dev.sh"（显式自定义命令）
+      // 注意：'air' 是保留字，表示使用内置 air 管理（走 startWithAir 路径）
+      devOpts = {
+        command: dev
+      };
+    } else if (dev && typeof dev === 'object') {
+      // { command: 'make', args: ['dev'] }
+      devOpts = dev;
+    }
+    // dev === 'air' 或 dev 未配置 → 默认使用内置 startWithAir()
+
+    this.goManager = new GoProcessManager(serviceDir, this.config.service.port, goEnv, devOpts);
+    try {
+      await this.goManager.start();
+    } catch (err) {
+      logger.logError(`[go] ${err instanceof Error ? err.message : String(err)}`);
+    }
+  }
+}
+
+// ============================================================
+// 工厂函数
+// ============================================================
+
+async function createForgeService(cwd, configPath, mode) {
+  const root = cwd || process.cwd();
+  const config = configPath ? await loadConfigFromPath(configPath, root) : await loader.loadConfig(root);
+  return new ForgeService(config, root, mode);
+}
+async function loadConfigFromPath(configPath, cwd) {
+  const {
+    parseConfig
+  } = await import('confmix');
+  const {
+    config: userConfig
+  } = await parseConfig(configPath);
+  return {
+    client: {
+      ...defaults.defaultConfig.client,
+      ...(userConfig.client || {})
+    },
+    service: {
+      ...defaults.defaultConfig.service,
+      ...(userConfig.service || {})
+    },
+    env: {
+      ...defaults.defaultConfig.env,
+      ...(userConfig.env || {})
+    },
+    configDir: cwd
+  };
+}
+
+exports.ForgeService = ForgeService;
+exports.createForgeService = createForgeService;
+//# sourceMappingURL=orchestrator.js.map