triflux 3.2.0-dev.8 → 3.2.0-dev.9

package/scripts/tfx-route-worker.mjs

@@ -0,0 +1,161 @@
+#!/usr/bin/env node
+// tfx-route-worker.mjs — tfx-route.sh용 subprocess worker 러너
+
+import { readFileSync, existsSync } from 'node:fs';
+import { dirname, resolve } from 'node:path';
+import { fileURLToPath, pathToFileURL } from 'node:url';
+
+const SCRIPT_DIR = dirname(fileURLToPath(import.meta.url));
+const FACTORY_CANDIDATES = [
+  resolve(SCRIPT_DIR, '../hub/workers/factory.mjs'),
+  resolve(SCRIPT_DIR, './hub/workers/factory.mjs'),
+];
+
+// MCP transport 실패 시 tfx-route.sh가 exec fallback을 수행할 수 있도록
+// CODEX_MCP_TRANSPORT_EXIT_CODE(70)으로 종료한다.
+const MCP_TRANSPORT_EXIT_CODE = 70;
+
+let createWorker = null;
+
+for (const candidate of FACTORY_CANDIDATES) {
+  if (!existsSync(candidate)) continue;
+  try {
+    ({ createWorker } = await import(pathToFileURL(candidate).href));
+  } catch (err) {
+    // 의존성 누락 (예: @modelcontextprotocol/sdk) → fallback 가능하도록 exit 70
+    if (err.code === 'ERR_MODULE_NOT_FOUND') {
+      process.stderr.write(`[tfx-route-worker] 모듈 로드 실패: ${err.message}\n`);
+      process.exit(MCP_TRANSPORT_EXIT_CODE);
+    }
+    throw err;
+  }
+  break;
+}
+
+if (!createWorker) {
+  process.stderr.write('[tfx-route-worker] worker factory를 찾지 못했습니다.\n');
+  process.exit(MCP_TRANSPORT_EXIT_CODE);
+}
+
+function parseArgs(argv) {
+  const args = {
+    allowedMcpServerNames: [],
+    mcpConfig: [],
+  };
+
+  for (let index = 0; index < argv.length; index += 1) {
+    const token = argv[index];
+    const next = argv[index + 1];
+
+    switch (token) {
+      case '--type':
+        args.type = next;
+        index += 1;
+        break;
+      case '--command':
+        args.command = next;
+        index += 1;
+        break;
+      case '--command-args-json':
+        args.commandArgsJson = next;
+        index += 1;
+        break;
+      case '--model':
+        args.model = next;
+        index += 1;
+        break;
+      case '--timeout-ms':
+        args.timeoutMs = Number(next);
+        index += 1;
+        break;
+      case '--approval-mode':
+        args.approvalMode = next;
+        index += 1;
+        break;
+      case '--permission-mode':
+        args.permissionMode = next;
+        index += 1;
+        break;
+      case '--allow-dangerously-skip-permissions':
+        args.allowDangerouslySkipPermissions = true;
+        break;
+      case '--allowed-mcp-server-name':
+        args.allowedMcpServerNames.push(next);
+        index += 1;
+        break;
+      case '--mcp-config':
+        args.mcpConfig.push(next);
+        index += 1;
+        break;
+      case '--cwd':
+        args.cwd = next;
+        index += 1;
+        break;
+      default:
+        throw new Error(`Unknown argument: ${token}`);
+    }
+  }
+
+  if (!args.type) {
+    throw new Error('--type is required');
+  }
+
+  return args;
+}
+
+function parseJsonArray(raw, label) {
+  if (!raw) return [];
+  try {
+    const parsed = JSON.parse(raw);
+    if (!Array.isArray(parsed)) {
+      throw new Error(`${label} must be a JSON array`);
+    }
+    return parsed.map((item) => String(item));
+  } catch (error) {
+    throw new Error(`${label} parse failed: ${error.message}`);
+  }
+}
+
+function readPromptFromStdin() {
+  return readFileSync(0, 'utf8');
+}
+
+function resolveDefaultMcpConfig(cwd) {
+  const candidate = resolve(cwd, '.mcp.json');
+  return existsSync(candidate) ? [candidate] : [];
+}
+
+const args = parseArgs(process.argv.slice(2));
+const prompt = readPromptFromStdin();
+
+const worker = createWorker(args.type, {
+  command: args.command,
+  commandArgs: parseJsonArray(args.commandArgsJson, '--command-args-json'),
+  model: args.model,
+  timeoutMs: args.timeoutMs,
+  approvalMode: args.approvalMode,
+  permissionMode: args.permissionMode,
+  allowDangerouslySkipPermissions: args.allowDangerouslySkipPermissions,
+  allowedMcpServerNames: args.allowedMcpServerNames,
+  mcpConfig: args.type === 'claude' && args.mcpConfig.length === 0
+    ? resolveDefaultMcpConfig(args.cwd || process.cwd())
+    : args.mcpConfig,
+  cwd: args.cwd || process.cwd(),
+});
+
+try {
+  const result = await worker.run(prompt);
+  if (result.response) {
+    process.stdout.write(result.response);
+    if (!result.response.endsWith('\n')) process.stdout.write('\n');
+  }
+} catch (error) {
+  if (error.stderr) {
+    process.stderr.write(String(error.stderr));
+    if (!String(error.stderr).endsWith('\n')) process.stderr.write('\n');
+  }
+  process.stderr.write(`${error.message}\n`);
+  process.exitCode = error.code === 'ETIMEDOUT' ? 124 : 1;
+} finally {
+  try { await worker.stop(); } catch {}
+}