@infly/vue2-vite 1.0.0

package/src/runner/workspace.cjs

@@ -0,0 +1,346 @@
+const fs = require('node:fs');
+const path = require('node:path');
+const { spawn } = require('node:child_process');
+
+function resolvePackageViteBin() {
+  try {
+    const packageJsonPath = require.resolve('vite/package.json', {
+      paths: [__dirname],
+    });
+    const viteBin = path.join(path.dirname(packageJsonPath), 'bin', 'vite.js');
+    if (!fs.existsSync(viteBin)) {
+      throw new Error(`Vite CLI 不存在: ${viteBin}`);
+    }
+    return viteBin;
+  } catch (error) {
+    throw new Error(
+      `无法从 @infly/vue2-vite 的直接依赖解析 Vite CLI；请重新运行 pnpm install。`
+      + ` ${error.message}`,
+      { cause: error },
+    );
+  }
+}
+
+function createWorkspaceRunner({
+  registry,
+  getTargetConfig,
+  createConfigPath,
+  defaultRootDir = process.cwd(),
+  resolveViteBin = resolvePackageViteBin,
+  spawnProcess = spawn,
+}) {
+  if (!Array.isArray(registry)) throw new TypeError('registry 必须是数组');
+  if (typeof getTargetConfig !== 'function') {
+    throw new TypeError('getTargetConfig 必须是函数');
+  }
+  if (typeof createConfigPath !== 'function') {
+    throw new TypeError('createConfigPath 必须是函数');
+  }
+
+  function validateOutputDir(app, outputDir, rootDir = defaultRootDir) {
+    if (!app.outputRoot) throw new Error(`应用 ${app.id} 未登记 outputRoot`);
+    const resolvedRoot = path.resolve(rootDir, app.outputRoot);
+    const resolvedOutput = path.resolve(rootDir, outputDir);
+    const relative = path.relative(resolvedRoot, resolvedOutput);
+    if (!relative || relative.startsWith('..') || path.isAbsolute(relative)) {
+      throw new Error(`拒绝未登记或不安全的输出目录: ${resolvedOutput}`);
+    }
+    const registered = Object.values(app.targets).some(
+      (targetConfig) => path.resolve(rootDir, targetConfig.outputDir) === resolvedOutput,
+    );
+    if (!registered) {
+      throw new Error(`输出目录不属于 ${app.id} 的注册 target: ${resolvedOutput}`);
+    }
+    return resolvedOutput;
+  }
+
+  function matchAppPackage(pkg) {
+    const packageDir = (pkg.dir || '').replace(/\\/g, '/');
+    return registry.find((app) => {
+      const appDir = app.subProjectPath.replace(/\\/g, '/');
+      return packageDir === appDir || pkg.name === app.id || packageDir.endsWith(appDir);
+    });
+  }
+
+  function resolveConfigPath(app, rootDir) {
+    const configPath = path.resolve(createConfigPath(app, rootDir));
+    if (!fs.existsSync(configPath)) {
+      throw new Error(`Vite 配置文件不存在: ${configPath}`);
+    }
+    return configPath;
+  }
+
+  function resolveApplicationVersion(app, rootDir) {
+    const packageJsonPath = path.resolve(rootDir, app.subProjectPath, 'package.json');
+    let packageJson;
+    try {
+      packageJson = JSON.parse(fs.readFileSync(packageJsonPath, 'utf8'));
+    } catch (error) {
+      throw new Error(`无法读取应用 package.json: ${packageJsonPath}`, { cause: error });
+    }
+    if (typeof packageJson.version !== 'string' || packageJson.version.length === 0) {
+      throw new Error(`应用 package.json 缺少有效 version: ${packageJsonPath}`);
+    }
+    return packageJson.version;
+  }
+
+  function spawnViteDev(app, options = {}) {
+    const rootDir = options.rootDir || defaultRootDir;
+    const target = options.target || app.defaultTarget;
+    const targetConfig = getTargetConfig(app, target);
+    const mode = options.mode || 'development';
+    const configPath = resolveConfigPath(app, rootDir);
+    const appVersion = resolveApplicationVersion(app, rootDir);
+    const env = {
+      ...process.env,
+      INFLY_VITE_APP_ID: app.id,
+      INFLY_VITE_APP_VERSION: appVersion,
+      VUE_APP_PLATFORM: target,
+      NODE_ENV: mode,
+      ENV: mode,
+    };
+    const args = [
+      resolveViteBin(),
+      '--config',
+      configPath,
+      '--mode',
+      mode,
+      '--port',
+      String(options.port || targetConfig.port),
+    ];
+    if (options.host) {
+      args.push('--host', String(options.host));
+    }
+    return spawnProcess(process.execPath, args, {
+      cwd: rootDir,
+      stdio: 'inherit',
+      env,
+    });
+  }
+
+  function spawnViteBuild(app, options = {}) {
+    return new Promise((resolve, reject) => {
+      const rootDir = options.rootDir || defaultRootDir;
+      const target = options.target || app.defaultTarget;
+      const mode = options.mode || 'production';
+      const viteMode = mode === 'prod'
+        ? 'production'
+        : mode === 'stage'
+          ? 'staging'
+          : mode;
+      let configPath;
+      try {
+        configPath = resolveConfigPath(app, rootDir);
+      } catch (error) {
+        reject(error);
+        return;
+      }
+
+      const targetConfig = getTargetConfig(app, target);
+      let appVersion;
+      try {
+        appVersion = resolveApplicationVersion(app, rootDir);
+      } catch (error) {
+        reject(error);
+        return;
+      }
+      const outputDir = options.outputDir || targetConfig.outputDir;
+      let safeOutputDir;
+      try {
+        safeOutputDir = validateOutputDir(app, outputDir, rootDir);
+      } catch (error) {
+        reject(error);
+        return;
+      }
+
+      const env = {
+        ...process.env,
+        INFLY_VITE_APP_ID: app.id,
+        INFLY_VITE_APP_VERSION: appVersion,
+        VUE_APP_PLATFORM: target,
+        NODE_ENV: viteMode === 'development' ? 'development' : 'production',
+        ENV: mode,
+      };
+      const args = [
+        resolveViteBin(),
+        'build',
+        '--config',
+        configPath,
+        '--mode',
+        viteMode,
+        '--outDir',
+        safeOutputDir,
+      ];
+
+      if (options.dryRun) {
+        console.log(`[dry-run] @infly/vue2-vite build: ${args.join(' ')}`);
+        console.log(
+          `[dry-run] env: INFLY_VITE_APP_ID=${app.id}`
+          + ` INFLY_VITE_APP_VERSION=${appVersion}`
+          + ` VUE_APP_PLATFORM=${target} NODE_ENV=${env.NODE_ENV}`,
+        );
+        resolve({ code: 0, signal: null });
+        return;
+      }
+
+      const child = spawnProcess(process.execPath, args, {
+        cwd: rootDir,
+        stdio: 'inherit',
+        env,
+      });
+      buildProcesses.add(child);
+      child.on('exit', (code, signal) => {
+        buildProcesses.delete(child);
+        if (code === 0) {
+          resolve({ code, signal });
+          return;
+        }
+        reject(new Error(
+          `Vite build 失败 (${app.id}, target=${target}): exit code ${code}`
+          + `${signal ? `, signal ${signal}` : ''}`,
+        ));
+      });
+      child.on('error', (error) => {
+        buildProcesses.delete(child);
+        reject(new Error(`Vite build 进程错误 (${app.id}): ${error.message}`));
+      });
+    });
+  }
+
+  // 构建期活跃子进程（信号处理用）
+  const buildProcesses = new Set();
+
+  function killAll(processes) {
+    for (const child of processes) {
+      if (!child.killed && child.exitCode === null) {
+        child._killed = true;
+        child.kill('SIGTERM');
+      }
+    }
+  }
+
+  function viteDev(packages, options = {}) {
+    const processes = [];
+    for (const pkg of packages) {
+      const app = matchAppPackage(pkg);
+      if (!app) {
+        console.warn(`[vite-dev] 跳过未注册的应用: ${pkg.name}`);
+        continue;
+      }
+      const targetLabel = pkg.target || options.target;
+      console.log(`[vite-dev] 启动 ${app.id}${targetLabel ? ` (target=${targetLabel})` : ''} (@infly/vue2-vite)`);
+      const child = spawnViteDev(app, { ...options, target: targetLabel });
+      processes.push(child);
+      child.on('exit', (code, signal) => {
+        if (code !== 0 && !child._killed) {
+          console.error(
+            `[vite-dev] ${app.id} 异常退出 (code=${code}, signal=${signal})，终止所有进程`,
+          );
+          killAll(processes);
+          process.exit(code || 1);
+        }
+      });
+    }
+
+    for (const signal of ['SIGINT', 'SIGTERM', 'SIGHUP']) {
+      process.on(signal, () => {
+        killAll(processes);
+        process.exit(0);
+      });
+    }
+    return processes;
+  }
+
+  async function viteBuild(packages, options = {}) {
+    const mode = options.mode || 'stage';
+    const concurrency = options.preset === 'all' ? 1 : 2;
+
+    // 信号处理：Ctrl+C 时终止构建子进程并立即退出（与 dev 链一致）
+    const onBuildSignal = (signal) => {
+      killAll([...buildProcesses]);
+      process.exit(signal === 'SIGINT' ? 130 : 143);
+    };
+    for (const signal of ['SIGINT', 'SIGTERM']) {
+      if (!process.listeners(signal).includes(onBuildSignal)) {
+        process.on(signal, onBuildSignal);
+      }
+    }
+    const tasks = [];
+    for (const pkg of packages) {
+      const app = matchAppPackage(pkg);
+      if (!app) {
+        console.warn(`[vite-build] 跳过未注册的应用: ${pkg.name}`);
+        continue;
+      }
+      const pkgTarget = pkg.target || options.target;
+      if (pkgTarget) {
+        getTargetConfig(app, pkgTarget);
+        tasks.push({ app, target: pkgTarget });
+      } else if (options.preset) {
+        tasks.push(...Object.keys(app.targets).map((target) => ({ app, target })));
+      } else {
+        tasks.push({ app, target: app.defaultTarget });
+      }
+    }
+
+    if (tasks.length === 0) {
+      console.log('[vite-build] 没有需要 Vite 构建的任务');
+      return;
+    }
+
+    const groups = new Map();
+    for (const task of tasks) {
+      if (!groups.has(task.app.id)) groups.set(task.app.id, []);
+      groups.get(task.app.id).push(task);
+    }
+    const appList = [...groups.values()];
+    const errors = [];
+    let index = 0;
+
+    async function runNext() {
+      const appTasks = appList[index++];
+      if (!appTasks) return;
+      try {
+        for (const task of appTasks) {
+          const targetConfig = getTargetConfig(task.app, task.target);
+          console.log(`[vite-build] 构建 ${task.app.id} (target=${task.target})`);
+          await spawnViteBuild(task.app, {
+            ...options,
+            target: task.target,
+            mode,
+            outputDir: targetConfig.outputDir,
+          });
+        }
+      } catch (error) {
+        errors.push(error.message);
+      }
+      await runNext();
+    }
+
+    await Promise.all(
+      Array.from(
+        { length: Math.min(concurrency, appList.length) },
+        () => runNext(),
+      ),
+    );
+    if (errors.length > 0) {
+      throw new Error(`Vite build 失败:\n${errors.join('\n')}`);
+    }
+  }
+
+  return {
+    matchAppPackage,
+    spawnViteDev,
+    spawnViteBuild,
+    viteDev,
+    viteBuild,
+    killAll,
+    resolveViteBin,
+    validateOutputDir,
+  };
+}
+
+module.exports = {
+  createWorkspaceRunner,
+  resolvePackageViteBin,
+};