triflux 3.2.0-dev.1 → 3.2.0-dev.3

package/hub/team/native-supervisor.mjs

@@ -0,0 +1,300 @@
+// hub/team/native-supervisor.mjs — tmux 없이 멀티 CLI를 직접 띄우는 네이티브 팀 런타임
+import { createServer } from "node:http";
+import { spawn } from "node:child_process";
+import { mkdirSync, readFileSync, writeFileSync, createWriteStream } from "node:fs";
+import { dirname, join } from "node:path";
+
+function parseArgs(argv) {
+  const out = {};
+  for (let i = 0; i < argv.length; i++) {
+    const cur = argv[i];
+    if (cur === "--config" && argv[i + 1]) {
+      out.config = argv[++i];
+    }
+  }
+  return out;
+}
+
+async function readJson(path) {
+  return JSON.parse(readFileSync(path, "utf8"));
+}
+
+function safeText(v, fallback = "") {
+  if (v == null) return fallback;
+  return String(v);
+}
+
+function nowMs() {
+  return Date.now();
+}
+
+const args = parseArgs(process.argv.slice(2));
+if (!args.config) {
+  console.error("사용법: node native-supervisor.mjs --config <path>");
+  process.exit(1);
+}
+
+const config = await readJson(args.config);
+const {
+  sessionName,
+  runtimeFile,
+  logsDir,
+  startupDelayMs = 3000,
+  members = [],
+} = config;
+
+mkdirSync(logsDir, { recursive: true });
+mkdirSync(dirname(runtimeFile), { recursive: true });
+
+const startedAt = nowMs();
+const processMap = new Map();
+
+function memberStateSnapshot() {
+  const states = [];
+  for (const m of members) {
+    const state = processMap.get(m.name);
+    states.push({
+      name: m.name,
+      role: m.role,
+      cli: m.cli,
+      agentId: m.agentId,
+      command: m.command,
+      pid: state?.child?.pid || null,
+      status: state?.status || "unknown",
+      exitCode: state?.exitCode ?? null,
+      lastPreview: state?.lastPreview || "",
+      logFile: state?.logFile || null,
+      errFile: state?.errFile || null,
+    });
+  }
+  return states;
+}
+
+function writeRuntime(controlPort) {
+  const runtime = {
+    sessionName,
+    supervisorPid: process.pid,
+    controlUrl: `http://127.0.0.1:${controlPort}`,
+    startedAt,
+    members: memberStateSnapshot(),
+  };
+  writeFileSync(runtimeFile, JSON.stringify(runtime, null, 2) + "\n");
+}
+
+function spawnMember(member) {
+  const outPath = join(logsDir, `${member.name}.out.log`);
+  const errPath = join(logsDir, `${member.name}.err.log`);
+
+  const outWs = createWriteStream(outPath, { flags: "a" });
+  const errWs = createWriteStream(errPath, { flags: "a" });
+
+  const child = spawn(member.command, {
+    shell: true,
+    env: {
+      ...process.env,
+      TERM: process.env.TERM && process.env.TERM !== "dumb" ? process.env.TERM : "xterm-256color",
+    },
+    stdio: ["pipe", "pipe", "pipe"],
+    windowsHide: true,
+  });
+
+  const state = {
+    member,
+    child,
+    outWs,
+    errWs,
+    logFile: outPath,
+    errFile: errPath,
+    status: "running",
+    exitCode: null,
+    lastPreview: "",
+  };
+
+  child.stdout.on("data", (buf) => {
+    outWs.write(buf);
+    const txt = safeText(buf).trim();
+    if (txt) {
+      const lines = txt.split(/\r?\n/).filter(Boolean);
+      if (lines.length) state.lastPreview = lines[lines.length - 1].slice(0, 280);
+    }
+  });
+
+  child.stderr.on("data", (buf) => {
+    errWs.write(buf);
+    const txt = safeText(buf).trim();
+    if (txt) {
+      const lines = txt.split(/\r?\n/).filter(Boolean);
+      if (lines.length) state.lastPreview = `[err] ${lines[lines.length - 1].slice(0, 260)}`;
+    }
+  });
+
+  child.on("exit", (code) => {
+    state.status = "exited";
+    state.exitCode = code;
+    try { outWs.end(); } catch {}
+    try { errWs.end(); } catch {}
+    maybeAutoShutdown();
+  });
+
+  processMap.set(member.name, state);
+}
+
+function sendInput(memberName, text) {
+  const state = processMap.get(memberName);
+  if (!state) return { ok: false, error: "member_not_found" };
+  if (state.status !== "running") return { ok: false, error: "member_not_running" };
+  try {
+    state.child.stdin.write(`${safeText(text)}\n`);
+    return { ok: true };
+  } catch (e) {
+    return { ok: false, error: e.message };
+  }
+}
+
+function interruptMember(memberName) {
+  const state = processMap.get(memberName);
+  if (!state) return { ok: false, error: "member_not_found" };
+  if (state.status !== "running") return { ok: false, error: "member_not_running" };
+
+  let signaled = false;
+  try {
+    signaled = state.child.kill("SIGINT");
+  } catch {
+    signaled = false;
+  }
+
+  if (!signaled) {
+    try {
+      state.child.stdin.write("\u0003");
+      signaled = true;
+    } catch {
+      signaled = false;
+    }
+  }
+
+  return signaled ? { ok: true } : { ok: false, error: "interrupt_failed" };
+}
+
+let isShuttingDown = false;
+
+function maybeAutoShutdown() {
+  if (isShuttingDown) return;
+  const allExited = [...processMap.values()].every((s) => s.status === "exited");
+  if (!allExited) return;
+  shutdown();
+}
+
+function shutdown() {
+  if (isShuttingDown) return;
+  isShuttingDown = true;
+
+  for (const state of processMap.values()) {
+    if (state.status === "running") {
+      try { state.child.stdin.write("exit\n"); } catch {}
+      try { state.child.kill("SIGTERM"); } catch {}
+    }
+    try { state.outWs.end(); } catch {}
+    try { state.errWs.end(); } catch {}
+  }
+
+  setTimeout(() => {
+    for (const state of processMap.values()) {
+      if (state.status === "running") {
+        try { state.child.kill("SIGKILL"); } catch {}
+      }
+    }
+    process.exit(0);
+  }, 1200).unref();
+}
+
+for (const member of members) {
+  spawnMember(member);
+}
+
+const server = createServer(async (req, res) => {
+  const send = (code, obj) => {
+    res.writeHead(code, { "Content-Type": "application/json" });
+    res.end(JSON.stringify(obj));
+  };
+
+  if (req.method === "GET" && (req.url === "/" || req.url === "/status")) {
+    return send(200, {
+      ok: true,
+      data: {
+        sessionName,
+        supervisorPid: process.pid,
+        uptimeMs: nowMs() - startedAt,
+        members: memberStateSnapshot(),
+      },
+    });
+  }
+
+  if (req.method !== "POST") {
+    return send(405, { ok: false, error: "method_not_allowed" });
+  }
+
+  let body = {};
+  try {
+    const chunks = [];
+    for await (const c of req) chunks.push(c);
+    const raw = Buffer.concat(chunks).toString("utf8") || "{}";
+    body = JSON.parse(raw);
+  } catch {
+    return send(400, { ok: false, error: "invalid_json" });
+  }
+
+  if (req.url === "/send") {
+    const { member, text } = body;
+    const r = sendInput(member, text);
+    return send(r.ok ? 200 : 400, r);
+  }
+
+  if (req.url === "/interrupt") {
+    const { member } = body;
+    const r = interruptMember(member);
+    return send(r.ok ? 200 : 400, r);
+  }
+
+  if (req.url === "/control") {
+    const { member, command = "", reason = "" } = body;
+    const controlMsg = `[LEAD CONTROL] command=${command}${reason ? ` reason=${reason}` : ""}`;
+    const a = sendInput(member, controlMsg);
+    if (!a.ok) return send(400, a);
+    if (String(command).toLowerCase() === "interrupt") {
+      const b = interruptMember(member);
+      if (!b.ok) return send(400, b);
+    }
+    return send(200, { ok: true });
+  }
+
+  if (req.url === "/stop") {
+    send(200, { ok: true });
+    shutdown();
+    return;
+  }
+
+  return send(404, { ok: false, error: "not_found" });
+});
+
+server.listen(0, "127.0.0.1", () => {
+  const address = server.address();
+  const port = typeof address === "object" && address ? address.port : null;
+  if (!port) {
+    console.error("native supervisor 포트 바인딩 실패");
+    process.exit(1);
+  }
+
+  writeRuntime(port);
+
+  // CLI 초기화 후 프롬프트 주입
+  setTimeout(() => {
+    for (const m of members) {
+      if (m.prompt) {
+        sendInput(m.name, m.prompt);
+      }
+    }
+  }, startupDelayMs).unref();
+});
+
+process.on("SIGINT", shutdown);
+process.on("SIGTERM", shutdown);

package/hub/team/native.mjs

@@ -0,0 +1,92 @@
+// hub/team/native.mjs — Claude Native Teams 래퍼
+// teammate 프롬프트 템플릿 + 팀 설정 빌더
+//
+// Claude Code 네이티브 Agent Teams (CLAUDE_CODE_EXPERIMENTAL_AGENT_TEAMS=1)
+// 환경에서 teammate를 Codex/Gemini CLI 래퍼로 구성하는 유틸리티.
+// SKILL.md가 인라인 프롬프트를 사용하므로, 이 모듈은 CLI(tfx team --native)에서
+// 팀 설정을 프로그래밍적으로 생성할 때 사용한다.
+
+const ROUTE_SCRIPT = "~/.claude/scripts/tfx-route.sh";
+
+/**
+ * CLI 타입별 teammate 프롬프트 생성
+ * @param {'codex'|'gemini'|'claude'} cli — CLI 타입
+ * @param {object} opts
+ * @param {string} opts.subtask — 서브태스크 설명
+ * @param {string} [opts.role] — 역할 (executor, designer, reviewer 등)
+ * @param {string} [opts.teamName] — 팀 이름
+ * @returns {string} teammate 프롬프트
+ */
+export function buildTeammatePrompt(cli, opts = {}) {
+  const { subtask, role = "executor", teamName = "tfx-team" } = opts;
+
+  if (cli === "claude") {
+    return `너는 ${teamName}의 Claude 워커이다.
+
+[작업] ${subtask}
+
+[실행]
+1. TaskList에서 pending 작업을 확인하고 claim (TaskUpdate: in_progress)
+2. Glob, Grep, Read, Bash 등 도구로 직접 수행
+3. 완료 시 TaskUpdate(status: completed) + SendMessage로 리드에게 보고
+4. 추가 작업이 있으면 반복
+
+에러 시 TaskUpdate(status: failed) + SendMessage로 보고.`;
+  }
+
+  const label = cli === "codex" ? "Codex" : "Gemini";
+  const escaped = subtask.replace(/'/g, "'\\''");
+
+  return `너는 ${teamName}의 ${label} 워커이다.
+
+[작업] ${subtask}
+
+[실행]
+1. TaskList에서 pending 작업을 확인하고 claim (TaskUpdate: in_progress)
+2. Bash("bash ${ROUTE_SCRIPT} ${role} '${escaped}' auto")로 실행
+3. 결과 확인 후 TaskUpdate(status: completed) + SendMessage로 리드에게 보고
+4. 추가 pending 작업이 있으면 반복
+
+[규칙]
+- 실제 구현은 ${label} CLI가 수행 — 너는 실행+보고 역할
+- 에러 시 TaskUpdate(status: failed) + SendMessage로 보고`;
+}
+
+/**
+ * teammate 이름 생성
+ * @param {'codex'|'gemini'|'claude'} cli
+ * @param {number} index — 0-based
+ * @returns {string}
+ */
+export function buildTeammateName(cli, index) {
+  return `${cli}-worker-${index + 1}`;
+}
+
+/**
+ * 트리아지 결과에서 팀 멤버 설정 생성
+ * @param {string} teamName — 팀 이름
+ * @param {Array<{cli: string, subtask: string, role?: string}>} assignments
+ * @returns {{ name: string, members: Array<{name: string, cli: string, prompt: string}> }}
+ */
+export function buildTeamConfig(teamName, assignments) {
+  return {
+    name: teamName,
+    members: assignments.map((a, i) => ({
+      name: buildTeammateName(a.cli, i),
+      cli: a.cli,
+      prompt: buildTeammatePrompt(a.cli, {
+        subtask: a.subtask,
+        role: a.role || "executor",
+        teamName,
+      }),
+    })),
+  };
+}
+
+/**
+ * 팀 이름 생성 (타임스탬프 기반)
+ * @returns {string}
+ */
+export function generateTeamName() {
+  return `tfx-${Date.now().toString(36).slice(-6)}`;
+}