triflux 3.2.0-dev.9 → 3.3.0-dev.1

package/hub/team/pane.mjs

@@ -1,5 +1,5 @@
-// hub/team/pane.mjs — pane별 CLI 실행 + stdin 주입
-// 의존성: child_process, fs, os, path (Node.js 내장)만 사용
+// hub/team/pane.mjs — pane별 CLI 실행 + stdin 주입
+// 의존성: child_process, fs, os, path (Node.js 내장)만 사용
 import { writeFileSync, unlinkSync, mkdirSync } from "node:fs";
 import { join } from "node:path";
 import { tmpdir } from "node:os";
@@ -16,29 +16,29 @@ function getPsmuxSessionName(target) {
 
 /** Windows 경로를 멀티플렉서용 경로로 변환 */
 function toMuxPath(p) {
-  if (process.platform !== "win32") return p;
-
-  const mux = detectMultiplexer();
-
-  // psmux는 Windows 네이티브 경로 그대로 사용
-  if (mux === "psmux") return p;
-
-  const normalized = p.replace(/\\/g, "/");
-  const m = normalized.match(/^([A-Za-z]):\/(.*)$/);
-  if (!m) return normalized;
-
-  const drive = m[1].toLowerCase();
-  const rest = m[2];
-
-  // wsl tmux는 /mnt/c/... 경로를 사용
-  if (mux === "wsl-tmux") {
-    return `/mnt/${drive}/${rest}`;
-  }
-
-  // Git Bash/MSYS tmux는 /c/... 경로를 사용
-  return `/${drive}/${rest}`;
-}
-
+  if (process.platform !== "win32") return p;
+
+  const mux = detectMultiplexer();
+
+  // psmux는 Windows 네이티브 경로 그대로 사용
+  if (mux === "psmux") return p;
+
+  const normalized = p.replace(/\\/g, "/");
+  const m = normalized.match(/^([A-Za-z]):\/(.*)$/);
+  if (!m) return normalized;
+
+  const drive = m[1].toLowerCase();
+  const rest = m[2];
+
+  // wsl tmux는 /mnt/c/... 경로를 사용
+  if (mux === "wsl-tmux") {
+    return `/mnt/${drive}/${rest}`;
+  }
+
+  // Git Bash/MSYS tmux는 /c/... 경로를 사용
+  return `/${drive}/${rest}`;
+}
+
 /** 멀티플렉서 커맨드 실행 (session.mjs와 동일 패턴) */
 function muxExec(args, opts = {}) {
   const exec = detectMultiplexer() === "psmux" ? psmuxExec : tmuxExec;
@@ -46,65 +46,81 @@ function muxExec(args, opts = {}) {
     encoding: "utf8",
     timeout: 10000,
     stdio: ["pipe", "pipe", "pipe"],
-    ...opts,
-  });
-}
-
-/**
- * CLI 에이전트 시작 커맨드 생성
- * @param {'codex'|'gemini'|'claude'} cli
- * @param {{ trustMode?: boolean }} [options]
- * @returns {string} 실행할 셸 커맨드
- */
-export function buildCliCommand(cli, options = {}) {
-  const { trustMode = false } = options;
-
-  switch (cli) {
-    case "codex":
-      // trust 모드에서는 승인/샌드박스 우회 + alt-screen 비활성화
-      return trustMode
-        ? "codex --dangerously-bypass-approvals-and-sandbox --no-alt-screen"
-        : "codex";
-    case "gemini":
-      // interactive 모드 — MCP는 ~/.gemini/settings.json에 사전 등록
-      return "gemini";
-    case "claude":
-      // interactive 모드
-      return "claude";
-    default:
-      return cli; // 커스텀 CLI 허용
-  }
-}
-
-/**
+    ...opts,
+  });
+}
+
+/**
+ * CLI 에이전트 시작 커맨드 생성
+ * @param {'codex'|'gemini'|'claude'} cli
+ * @param {{ trustMode?: boolean }} [options]
+ * @returns {string} 실행할 셸 커맨드
+ */
+export function buildCliCommand(cli, options = {}) {
+  const { trustMode = false } = options;
+
+  switch (cli) {
+    case "codex":
+      // trust 모드에서는 승인/샌드박스 우회 + alt-screen 비활성화
+      return trustMode
+        ? "codex --dangerously-bypass-approvals-and-sandbox --no-alt-screen"
+        : "codex";
+    case "gemini":
+      // interactive 모드 — MCP는 ~/.gemini/settings.json에 사전 등록
+      return "gemini";
+    case "claude":
+      // interactive 모드
+      return "claude";
+    default:
+      return cli; // 커스텀 CLI 허용
+  }
+}
+
+/**
  * pane에 CLI 시작
- * @param {string} target — 예: tfx-team-abc:0.1
- * @param {string} command — 실행할 커맨드
- */
+ * @param {string} target — 예: tfx-multi-abc:0.1
+ * @param {string} command — 실행할 커맨드
+ */
 export function startCliInPane(target, command) {
   // CLI 시작도 buffer paste를 재사용해 셸/플랫폼별 quoting 차이를 제거한다.
   injectPrompt(target, command);
 }
-
-/**
- * pane에 프롬프트 주입 (load-buffer + paste-buffer 방식)
- * 멀티라인 + 특수문자 안전, 크기 제한 없음
- * @param {string} target — 예: tfx-team-abc:0.1
- * @param {string} prompt — 주입할 텍스트
- */
-export function injectPrompt(target, prompt) {
-  // 임시 파일에 프롬프트 저장
-  const tmpDir = join(tmpdir(), "tfx-team");
-  mkdirSync(tmpDir, { recursive: true });
-
-  // pane ID를 파일명에 포함 (충돌 방지)
-  const safeTarget = target.replace(/[:.]/g, "-");
-  const tmpFile = join(tmpDir, `prompt-${safeTarget}-${Date.now()}.txt`);
-
+
+/**
+ * pane에 프롬프트 주입 (load-buffer + paste-buffer 방식)
+ * 멀티라인 + 특수문자 안전, 크기 제한 없음
+ * @param {string} target — 예: tfx-multi-abc:0.1
+ * @param {string} prompt — 주입할 텍스트
+ */
+/**
+ * pane에 프롬프트 주입
+ * @param {string} target — 예: tfx-multi-abc:0.1
+ * @param {string} prompt — 주입할 텍스트
+ * @param {object} [opts]
+ * @param {boolean} [opts.useFileRef] — true면 TUI용 @file 참조 방식 (psmux 전용)
+ */
+export function injectPrompt(target, prompt, { useFileRef = false } = {}) {
+  const tmpDir = join(tmpdir(), "tfx-multi");
+  mkdirSync(tmpDir, { recursive: true });
+
+  const safeTarget = target.replace(/[:.]/g, "-");
+  const tmpFile = join(tmpDir, `prompt-${safeTarget}-${Date.now()}.txt`);
+
+  // psmux + TUI 앱: @file 참조로 주입 (paste-buffer는 TUI와 호환 안 됨)
+  if (detectMultiplexer() === "psmux" && useFileRef) {
+    writeFileSync(tmpFile, prompt, "utf8");
+    const filePath = tmpFile.replace(/\\/g, "/");
+    psmuxExec(["select-pane", "-t", target]);
+    psmuxExec(["send-keys", "-t", target, "-l", `@${filePath}`]);
+    psmuxExec(["send-keys", "-t", target, "Enter"]);
+    // TUI가 파일을 읽을 시간을 주고 정리
+    setTimeout(() => { try { unlinkSync(tmpFile); } catch {} }, 10000);
+    return;
+  }
+
   try {
     writeFileSync(tmpFile, prompt, "utf8");
 
-    // psmux는 buffer 명령에 세션 컨텍스트가 필요하다.
     if (detectMultiplexer() === "psmux") {
       const sessionName = getPsmuxSessionName(target);
       psmuxExec(["load-buffer", "-t", sessionName, toMuxPath(tmpFile)]);
@@ -119,20 +135,15 @@ export function injectPrompt(target, prompt) {
     muxExec(`paste-buffer -t ${target}`);
     muxExec(`send-keys -t ${target} Enter`);
   } finally {
-    // 임시 파일 정리
-    try {
-      unlinkSync(tmpFile);
-    } catch {
-      // 정리 실패 무시
-    }
-  }
-}
-
-/**
- * pane에 키 입력 전송
- * @param {string} target — 예: tfx-team-abc:0.1
- * @param {string} keys — tmux 키 표현 (예: 'C-c', 'Enter')
- */
+    try { unlinkSync(tmpFile); } catch {}
+  }
+}
+
+/**
+ * pane에 키 입력 전송
+ * @param {string} target — 예: tfx-multi-abc:0.1
+ * @param {string} keys — tmux 키 표현 (예: 'C-c', 'Enter')
+ */
 export function sendKeys(target, keys) {
   muxExec(`send-keys -t ${target} ${keys}`);
 }

package/hub/team/psmux.mjs

@@ -3,6 +3,8 @@
 import { execSync, spawnSync } from "node:child_process";
 
 const PSMUX_BIN = process.env.PSMUX_BIN || "psmux";
+const GIT_BASH = process.env.GIT_BASH_PATH || "C:\\Program Files\\Git\\bin\\bash.exe";
+const IS_WINDOWS = process.platform === "win32";
 
 function quoteArg(value) {
   const str = String(value);
@@ -199,14 +201,14 @@ export function psmuxSessionExists(sessionName) {
 }
 
 /**
- * tfx-team- 접두사 psmux 세션 목록
+ * tfx-multi- 접두사 psmux 세션 목록
  * @returns {string[]}
  */
 export function listPsmuxSessions() {
   try {
     return parseSessionSummaries(psmuxExec("list-sessions"))
       .map((session) => session.sessionName)
-      .filter((sessionName) => sessionName.startsWith("tfx-team-"));
+      .filter((sessionName) => sessionName.startsWith("tfx-multi-"));
   } catch {
     return [];
   }
@@ -268,6 +270,7 @@ export function getPsmuxSessionAttachedCount(sessionName) {
 export function configurePsmuxKeybindings(sessionName, opts = {}) {
   const { inProcess = false, taskListCommand = "" } = opts;
   const cond = `#{==:#{session_name},${sessionName}}`;
+  const target = `${sessionName}:0`;
   const bindNext = inProcess
     ? `'select-pane -t :.+ \\; resize-pane -Z'`
     : `'select-pane -t :.+'`;
@@ -275,23 +278,229 @@ export function configurePsmuxKeybindings(sessionName, opts = {}) {
     ? `'select-pane -t :.- \\; resize-pane -Z'`
     : `'select-pane -t :.-'`;
 
-  psmuxExec(`bind-key -T root -n S-Down if-shell -F '${cond}' ${bindNext} 'send-keys S-Down'`);
-  psmuxExec(`bind-key -T root -n S-Up if-shell -F '${cond}' ${bindPrev} 'send-keys S-Up'`);
-  psmuxExec(`bind-key -T root -n S-Right if-shell -F '${cond}' ${bindNext} 'send-keys S-Right'`);
-  psmuxExec(`bind-key -T root -n S-Left if-shell -F '${cond}' ${bindPrev} 'send-keys S-Left'`);
-  psmuxExec(`bind-key -T root -n BTab if-shell -F '${cond}' ${bindPrev} 'send-keys BTab'`);
-  psmuxExec(`bind-key -T root -n Escape if-shell -F '${cond}' 'send-keys C-c' 'send-keys Escape'`);
+  // psmux는 세션별 서버이므로 -t target으로 세션 컨텍스트를 전달해야 한다
+  const bindSafe = (cmd) => {
+    try { psmuxExec(`-t ${quoteArg(target)} ${cmd}`); } catch { /* 미지원 시 무시 */ }
+  };
+
+  bindSafe(`bind-key -T root -n S-Down if-shell -F '${cond}' ${bindNext} 'send-keys S-Down'`);
+  bindSafe(`bind-key -T root -n S-Up if-shell -F '${cond}' ${bindPrev} 'send-keys S-Up'`);
+  bindSafe(`bind-key -T root -n S-Right if-shell -F '${cond}' ${bindNext} 'send-keys S-Right'`);
+  bindSafe(`bind-key -T root -n S-Left if-shell -F '${cond}' ${bindPrev} 'send-keys S-Left'`);
+  bindSafe(`bind-key -T root -n BTab if-shell -F '${cond}' ${bindPrev} 'send-keys BTab'`);
+  bindSafe(`bind-key -T root -n Escape if-shell -F '${cond}' 'send-keys C-c' 'send-keys Escape'`);
 
   if (taskListCommand) {
     const escaped = taskListCommand.replace(/'/g, "'\\''");
+    bindSafe(
+      `bind-key -T root -n C-t if-shell -F '${cond}' "display-popup -E '${escaped}'" "send-keys C-t"`
+    );
+  }
+}
+
+// ─── 하이브리드 모드 워커 관리 함수 ───
+
+/**
+ * psmux 세션의 새 pane에서 워커 실행
+ * @param {string} sessionName - 대상 psmux 세션 이름
+ * @param {string} workerName - 워커 식별용 pane 타이틀
+ * @param {string} cmd - 실행할 커맨드
+ * @returns {{ paneId: string, workerName: string }}
+ */
+export function spawnWorker(sessionName, workerName, cmd) {
+  if (!hasPsmux()) {
+    throw new Error("psmux가 설치되어 있지 않습니다. psmux를 먼저 설치하세요.");
+  }
+
+  // remain-on-exit: 종료된 pane이 즉시 사라지는 것 방지
+  try {
+    psmuxExec(`set-option -t ${quoteArg(sessionName)} remain-on-exit on`);
+  } catch { /* 미지원 시 무시 */ }
+
+  // Windows: pane 기본셸이 PowerShell → Git Bash로 래핑
+  // psmux가 이스케이프 시퀀스를 처리하므로 포워드 슬래시로 변환
+  const shellCmd = IS_WINDOWS
+    ? `& '${GIT_BASH.replace(/\\/g, '/')}' -l -c '${cmd.replace(/'/g, "'\\''")}'`
+    : cmd;
+
+  try {
+    // 배열 형태 spawnSync → 쉘 해석 우회 (백슬래시 경로 보존)
+    const paneTarget = psmux([
+      "split-window", "-t", sessionName,
+      "-P", "-F", "#{session_name}:#{window_index}.#{pane_index}",
+      shellCmd,
+    ]);
+    psmux(["select-pane", "-t", paneTarget, "-T", workerName]);
+    return { paneId: paneTarget, workerName };
+  } catch (err) {
+    throw new Error(`워커 생성 실패 (session=${sessionName}, worker=${workerName}): ${err.message}`);
+  }
+}
+
+/**
+ * 워커 pane 실행 상태 확인
+ * @param {string} sessionName - 대상 psmux 세션 이름
+ * @param {string} workerName - 워커 pane 타이틀
+ * @returns {{ status: "running"|"exited", exitCode: number|null, paneId: string }}
+ */
+export function getWorkerStatus(sessionName, workerName) {
+  if (!hasPsmux()) {
+    throw new Error("psmux가 설치되어 있지 않습니다.");
+  }
+  try {
+    const output = psmuxExec(
+      `list-panes -t ${quoteArg(sessionName)} -F "#{pane_title}\t#{session_name}:#{window_index}.#{pane_index}\t#{pane_dead}\t#{pane_dead_status}"`
+    );
+    const lines = output.split("\n").filter(Boolean);
+    for (const line of lines) {
+      const [title, paneId, dead, deadStatus] = line.split("\t");
+      if (title === workerName) {
+        const isDead = dead === "1";
+        return {
+          status: isDead ? "exited" : "running",
+          exitCode: isDead ? parseInt(deadStatus, 10) || 0 : null,
+          paneId,
+        };
+      }
+    }
+    throw new Error(`워커를 찾을 수 없습니다: ${workerName}`);
+  } catch (err) {
+    if (err.message.includes("워커를 찾을 수 없습니다")) throw err;
+    throw new Error(`워커 상태 조회 실패 (session=${sessionName}, worker=${workerName}): ${err.message}`);
+  }
+}
+
+/**
+ * 워커 pane 프로세스 강제 종료
+ * @param {string} sessionName - 대상 psmux 세션 이름
+ * @param {string} workerName - 워커 pane 타이틀
+ * @returns {{ killed: boolean }}
+ */
+export function killWorker(sessionName, workerName) {
+  if (!hasPsmux()) {
+    throw new Error("psmux가 설치되어 있지 않습니다.");
+  }
+  try {
+    const { paneId, status } = getWorkerStatus(sessionName, workerName);
+
+    // 이미 종료된 워커 → pane 정리만 수행
+    if (status === "exited") {
+      try { psmuxExec(`kill-pane -t ${quoteArg(paneId)}`); } catch { /* 무시 */ }
+      return { killed: true };
+    }
+
+    // running → C-c 우아한 종료 시도
     try {
-      psmuxExec(
-        `bind-key -T root -n C-t if-shell -F '${cond}' "display-popup -E '${escaped}'" "send-keys C-t"`
-      );
+      psmuxExec(`send-keys -t ${quoteArg(paneId)} C-c`);
     } catch {
-      psmuxExec(
-        `bind-key -T root -n C-t if-shell -F '${cond}' 'display-message "tfx team tasks 명령으로 태스크 확인"' 'send-keys C-t'`
-      );
+      // send-keys 실패 무시
+    }
+    // 1초 대기 후 pane 강제 종료
+    spawnSync("sleep", ["1"], { stdio: "ignore", windowsHide: true });
+    try {
+      psmuxExec(`kill-pane -t ${quoteArg(paneId)}`);
+    } catch {
+      // 이미 종료된 pane — 무시
+    }
+    return { killed: true };
+  } catch (err) {
+    // 워커를 찾을 수 없음 → 이미 종료된 것으로 간주
+    if (err.message.includes("워커를 찾을 수 없습니다")) {
+      return { killed: true };
+    }
+    throw new Error(`워커 종료 실패 (session=${sessionName}, worker=${workerName}): ${err.message}`);
+  }
+}
+
+/**
+ * 워커 pane 출력 마지막 N줄 캡처
+ * @param {string} sessionName - 대상 psmux 세션 이름
+ * @param {string} workerName - 워커 pane 타이틀
+ * @param {number} lines - 캡처할 줄 수 (기본 50)
+ * @returns {string} 캡처된 출력
+ */
+export function captureWorkerOutput(sessionName, workerName, lines = 50) {
+  if (!hasPsmux()) {
+    throw new Error("psmux가 설치되어 있지 않습니다.");
+  }
+  try {
+    const { paneId } = getWorkerStatus(sessionName, workerName);
+    return psmuxExec(`capture-pane -t ${quoteArg(paneId)} -p -S -${lines}`);
+  } catch (err) {
+    if (err.message.includes("워커를 찾을 수 없습니다")) throw err;
+    throw new Error(`출력 캡처 실패 (session=${sessionName}, worker=${workerName}): ${err.message}`);
+  }
+}
+
+// ─── CLI 진입점 ───
+
+if (process.argv[1] && process.argv[1].endsWith("psmux.mjs")) {
+  const [,, cmd, ...args] = process.argv;
+
+  // CLI 인자 파싱 헬퍼
+  function getArg(name) {
+    const idx = args.indexOf(`--${name}`);
+    return idx !== -1 && idx + 1 < args.length ? args[idx + 1] : null;
+  }
+
+  try {
+    switch (cmd) {
+      case "spawn": {
+        const session = getArg("session");
+        const name = getArg("name");
+        const workerCmd = getArg("cmd");
+        if (!session || !name || !workerCmd) {
+          console.error("사용법: node psmux.mjs spawn --session <세션> --name <워커명> --cmd <커맨드>");
+          process.exit(1);
+        }
+        const result = spawnWorker(session, name, workerCmd);
+        console.log(JSON.stringify(result, null, 2));
+        break;
+      }
+      case "status": {
+        const session = getArg("session");
+        const name = getArg("name");
+        if (!session || !name) {
+          console.error("사용법: node psmux.mjs status --session <세션> --name <워커명>");
+          process.exit(1);
+        }
+        const result = getWorkerStatus(session, name);
+        console.log(JSON.stringify(result, null, 2));
+        break;
+      }
+      case "kill": {
+        const session = getArg("session");
+        const name = getArg("name");
+        if (!session || !name) {
+          console.error("사용법: node psmux.mjs kill --session <세션> --name <워커명>");
+          process.exit(1);
+        }
+        const result = killWorker(session, name);
+        console.log(JSON.stringify(result, null, 2));
+        break;
+      }
+      case "output": {
+        const session = getArg("session");
+        const name = getArg("name");
+        const lines = parseInt(getArg("lines") || "50", 10);
+        if (!session || !name) {
+          console.error("사용법: node psmux.mjs output --session <세션> --name <워커명> [--lines <줄수>]");
+          process.exit(1);
+        }
+        console.log(captureWorkerOutput(session, name, lines));
+        break;
+      }
+      default:
+        console.error("사용법: node psmux.mjs spawn|status|kill|output [args]");
+        console.error("");
+        console.error("  spawn   --session <세션> --name <워커명> --cmd <커맨드>");
+        console.error("  status  --session <세션> --name <워커명>");
+        console.error("  kill    --session <세션> --name <워커명>");
+        console.error("  output  --session <세션> --name <워커명> [--lines <줄수>]");
+        process.exit(1);
     }
+  } catch (err) {
+    console.error(`오류: ${err.message}`);
+    process.exit(1);
   }
 }