triflux 3.1.0-dev.5 → 3.2.0-dev.2

package/hub/team/nativeProxy.mjs

@@ -0,0 +1,455 @@
+// hub/team/nativeProxy.mjs
+// Claude Native Teams 파일을 Hub tool/REST에서 안전하게 읽고 쓰기 위한 유틸.
+
+import {
+  existsSync,
+  mkdirSync,
+  readdirSync,
+  readFileSync,
+  renameSync,
+  statSync,
+  unlinkSync,
+  writeFileSync,
+  openSync,
+  closeSync,
+} from 'node:fs';
+import { basename, dirname, join } from 'node:path';
+import { homedir } from 'node:os';
+import { randomUUID } from 'node:crypto';
+
+const TEAM_NAME_RE = /^[a-z0-9][a-z0-9-]*$/;
+const CLAUDE_HOME = join(homedir(), '.claude');
+const TEAMS_ROOT = join(CLAUDE_HOME, 'teams');
+const TASKS_ROOT = join(CLAUDE_HOME, 'tasks');
+
+function err(code, message, extra = {}) {
+  return { ok: false, error: { code, message, ...extra } };
+}
+
+function validateTeamName(teamName) {
+  if (!TEAM_NAME_RE.test(String(teamName || ''))) {
+    throw new Error('INVALID_TEAM_NAME');
+  }
+}
+
+function readJsonSafe(path) {
+  try {
+    return JSON.parse(readFileSync(path, 'utf8'));
+  } catch {
+    return null;
+  }
+}
+
+function atomicWriteJson(path, value) {
+  mkdirSync(dirname(path), { recursive: true });
+  const tmp = `${path}.tmp-${process.pid}-${Date.now()}`;
+  writeFileSync(tmp, `${JSON.stringify(value, null, 2)}\n`, 'utf8');
+  renameSync(tmp, path);
+}
+
+function sleepMs(ms) {
+  const end = Date.now() + ms;
+  while (Date.now() < end) {}
+}
+
+function withFileLock(lockPath, fn, retries = 20, delayMs = 25) {
+  let fd = null;
+  for (let i = 0; i < retries; i += 1) {
+    try {
+      fd = openSync(lockPath, 'wx');
+      break;
+    } catch (e) {
+      if (e?.code !== 'EEXIST' || i === retries - 1) throw e;
+      sleepMs(delayMs);
+    }
+  }
+
+  try {
+    return fn();
+  } finally {
+    try { if (fd != null) closeSync(fd); } catch {}
+    try { unlinkSync(lockPath); } catch {}
+  }
+}
+
+function getLeadSessionId(config) {
+  return config?.leadSessionId
+    || config?.lead_session_id
+    || config?.lead?.lead_session_id
+    || config?.lead?.sessionId
+    || null;
+}
+
+export function resolveTeamPaths(teamName) {
+  validateTeamName(teamName);
+
+  const teamDir = join(TEAMS_ROOT, teamName);
+  const configPath = join(teamDir, 'config.json');
+  const inboxesDir = join(teamDir, 'inboxes');
+  const config = readJsonSafe(configPath);
+  const leadSessionId = getLeadSessionId(config);
+
+  const byTeam = join(TASKS_ROOT, teamName);
+  const byLeadSession = leadSessionId ? join(TASKS_ROOT, leadSessionId) : null;
+
+  let tasksDir = byTeam;
+  let tasksDirResolution = 'not_found';
+  if (existsSync(byTeam)) {
+    tasksDirResolution = 'team_name';
+  } else if (byLeadSession && existsSync(byLeadSession)) {
+    tasksDir = byLeadSession;
+    tasksDirResolution = 'lead_session_id';
+  }
+
+  return {
+    team_dir: teamDir,
+    config_path: configPath,
+    inboxes_dir: inboxesDir,
+    tasks_dir: tasksDir,
+    tasks_dir_resolution: tasksDirResolution,
+    lead_session_id: leadSessionId,
+    config,
+  };
+}
+
+function collectTaskFiles(tasksDir) {
+  if (!existsSync(tasksDir)) return [];
+  return readdirSync(tasksDir)
+    .filter((name) => name.endsWith('.json'))
+    .filter((name) => !name.endsWith('.lock'))
+    .filter((name) => name !== '.highwatermark')
+    .map((name) => join(tasksDir, name));
+}
+
+function locateTaskFile(tasksDir, taskId) {
+  const direct = join(tasksDir, `${taskId}.json`);
+  if (existsSync(direct)) return direct;
+
+  const files = collectTaskFiles(tasksDir);
+  for (const file of files) {
+    if (basename(file, '.json') === taskId) return file;
+    const json = readJsonSafe(file);
+    if (json && String(json.id || '') === taskId) return file;
+  }
+  return null;
+}
+
+function isObject(v) {
+  return !!v && typeof v === 'object' && !Array.isArray(v);
+}
+
+export function teamInfo(args = {}) {
+  const { team_name, include_members = true, include_paths = true } = args;
+  try {
+    validateTeamName(team_name);
+  } catch {
+    return err('INVALID_TEAM_NAME', 'team_name 형식이 올바르지 않습니다');
+  }
+
+  const paths = resolveTeamPaths(team_name);
+  if (!existsSync(paths.team_dir)) {
+    return err('TEAM_NOT_FOUND', `팀 디렉토리가 없습니다: ${paths.team_dir}`);
+  }
+
+  const members = Array.isArray(paths.config?.members) ? paths.config.members : [];
+  const leadAgentId = paths.config?.leadAgentId
+    || paths.config?.lead_agent_id
+    || members[0]?.agentId
+    || null;
+
+  return {
+    ok: true,
+    data: {
+      team: {
+        team_name,
+        description: paths.config?.description || null,
+      },
+      lead: {
+        lead_agent_id: leadAgentId,
+        lead_session_id: paths.lead_session_id,
+      },
+      ...(include_members ? { members } : {}),
+      ...(include_paths ? {
+        paths: {
+          config_path: paths.config_path,
+          tasks_dir: paths.tasks_dir,
+          inboxes_dir: paths.inboxes_dir,
+          tasks_dir_resolution: paths.tasks_dir_resolution,
+        },
+      } : {}),
+    },
+  };
+}
+
+export function teamTaskList(args = {}) {
+  const {
+    team_name,
+    owner,
+    statuses = [],
+    include_internal = false,
+    limit = 200,
+  } = args;
+
+  try {
+    validateTeamName(team_name);
+  } catch {
+    return err('INVALID_TEAM_NAME', 'team_name 형식이 올바르지 않습니다');
+  }
+
+  const paths = resolveTeamPaths(team_name);
+  if (paths.tasks_dir_resolution === 'not_found') {
+    return err('TASKS_DIR_NOT_FOUND', `task 디렉토리를 찾지 못했습니다: ${team_name}`);
+  }
+
+  const statusSet = new Set((statuses || []).map((s) => String(s)));
+  const maxCount = Math.max(1, Math.min(Number(limit) || 200, 1000));
+  let parseWarnings = 0;
+
+  const tasks = [];
+  for (const file of collectTaskFiles(paths.tasks_dir)) {
+    const json = readJsonSafe(file);
+    if (!json || !isObject(json)) {
+      parseWarnings += 1;
+      continue;
+    }
+
+    if (!include_internal && json?.metadata?._internal === true) continue;
+    if (owner && String(json.owner || '') !== String(owner)) continue;
+    if (statusSet.size > 0 && !statusSet.has(String(json.status || ''))) continue;
+
+    let mtime = Date.now();
+    try { mtime = statSync(file).mtimeMs; } catch {}
+
+    tasks.push({
+      ...json,
+      task_file: file,
+      mtime_ms: mtime,
+    });
+  }
+
+  tasks.sort((a, b) => Number(b.mtime_ms || 0) - Number(a.mtime_ms || 0));
+  const sliced = tasks.slice(0, maxCount);
+
+  return {
+    ok: true,
+    data: {
+      tasks: sliced,
+      count: sliced.length,
+      parse_warnings: parseWarnings,
+      tasks_dir: paths.tasks_dir,
+      tasks_dir_resolution: paths.tasks_dir_resolution,
+    },
+  };
+}
+
+export function teamTaskUpdate(args = {}) {
+  const {
+    team_name,
+    task_id,
+    claim = false,
+    owner,
+    status,
+    subject,
+    description,
+    activeForm,
+    add_blocks = [],
+    add_blocked_by = [],
+    metadata_patch,
+    if_match_mtime_ms,
+    actor,
+  } = args;
+
+  try {
+    validateTeamName(team_name);
+  } catch {
+    return err('INVALID_TEAM_NAME', 'team_name 형식이 올바르지 않습니다');
+  }
+
+  if (!String(task_id || '').trim()) {
+    return err('INVALID_TASK_ID', 'task_id가 필요합니다');
+  }
+
+  const paths = resolveTeamPaths(team_name);
+  if (paths.tasks_dir_resolution === 'not_found') {
+    return err('TASKS_DIR_NOT_FOUND', `task 디렉토리를 찾지 못했습니다: ${team_name}`);
+  }
+
+  const taskFile = locateTaskFile(paths.tasks_dir, String(task_id));
+  if (!taskFile) {
+    return err('TASK_NOT_FOUND', `task를 찾지 못했습니다: ${task_id}`);
+  }
+
+  const lockFile = `${taskFile}.lock`;
+
+  try {
+    return withFileLock(lockFile, () => {
+      const before = readJsonSafe(taskFile);
+      if (!before || !isObject(before)) {
+        return err('INVALID_TASK_FILE', `task 파일 파싱 실패: ${taskFile}`);
+      }
+
+      let beforeMtime = Date.now();
+      try { beforeMtime = statSync(taskFile).mtimeMs; } catch {}
+
+      if (if_match_mtime_ms != null && Number(if_match_mtime_ms) !== Number(beforeMtime)) {
+        return err('MTIME_CONFLICT', 'if_match_mtime_ms가 일치하지 않습니다', {
+          task_file: taskFile,
+          mtime_ms: beforeMtime,
+        });
+      }
+
+      const after = JSON.parse(JSON.stringify(before));
+      let claimed = false;
+      let updated = false;
+
+      if (claim) {
+        const requestedOwner = String(owner || actor || '');
+        const ownerNow = String(before.owner || '');
+        const ownerCompatible = ownerNow === '' || requestedOwner === '' || ownerNow === requestedOwner;
+        const statusPending = String(before.status || '') === 'pending';
+
+        if (!statusPending || !ownerCompatible) {
+          return err('CLAIM_CONFLICT', 'task claim 충돌', {
+            task_before: before,
+            task_file: taskFile,
+            mtime_ms: beforeMtime,
+          });
+        }
+
+        if (requestedOwner) after.owner = requestedOwner;
+        after.status = status || 'in_progress';
+        claimed = true;
+        updated = true;
+      }
+
+      if (owner != null && String(after.owner || '') !== String(owner)) {
+        after.owner = owner;
+        updated = true;
+      }
+      if (status != null && String(after.status || '') !== String(status)) {
+        after.status = status;
+        updated = true;
+      }
+      if (subject != null && String(after.subject || '') !== String(subject)) {
+        after.subject = subject;
+        updated = true;
+      }
+      if (description != null && String(after.description || '') !== String(description)) {
+        after.description = description;
+        updated = true;
+      }
+      if (activeForm != null && String(after.activeForm || '') !== String(activeForm)) {
+        after.activeForm = activeForm;
+        updated = true;
+      }
+
+      if (Array.isArray(add_blocks) && add_blocks.length > 0) {
+        const current = Array.isArray(after.blocks) ? [...after.blocks] : [];
+        for (const item of add_blocks) {
+          if (!current.includes(item)) current.push(item);
+        }
+        after.blocks = current;
+        updated = true;
+      }
+
+      if (Array.isArray(add_blocked_by) && add_blocked_by.length > 0) {
+        const current = Array.isArray(after.blockedBy) ? [...after.blockedBy] : [];
+        for (const item of add_blocked_by) {
+          if (!current.includes(item)) current.push(item);
+        }
+        after.blockedBy = current;
+        updated = true;
+      }
+
+      if (isObject(metadata_patch)) {
+        const base = isObject(after.metadata) ? after.metadata : {};
+        after.metadata = { ...base, ...metadata_patch };
+        updated = true;
+      }
+
+      if (updated) {
+        atomicWriteJson(taskFile, after);
+      }
+
+      let afterMtime = beforeMtime;
+      try { afterMtime = statSync(taskFile).mtimeMs; } catch {}
+
+      return {
+        ok: true,
+        data: {
+          claimed,
+          updated,
+          task_before: before,
+          task_after: updated ? after : before,
+          task_file: taskFile,
+          mtime_ms: afterMtime,
+        },
+      };
+    });
+  } catch (e) {
+    return err('TASK_UPDATE_FAILED', e.message);
+  }
+}
+
+function sanitizeRecipientName(v) {
+  return String(v || 'team-lead').replace(/[\\/:*?"<>|]/g, '_');
+}
+
+export function teamSendMessage(args = {}) {
+  const {
+    team_name,
+    from,
+    to = 'team-lead',
+    text,
+    summary,
+    color = 'blue',
+  } = args;
+
+  try {
+    validateTeamName(team_name);
+  } catch {
+    return err('INVALID_TEAM_NAME', 'team_name 형식이 올바르지 않습니다');
+  }
+
+  if (!String(from || '').trim()) return err('INVALID_FROM', 'from이 필요합니다');
+  if (!String(text || '').trim()) return err('INVALID_TEXT', 'text가 필요합니다');
+
+  const paths = resolveTeamPaths(team_name);
+  if (!existsSync(paths.team_dir)) {
+    return err('TEAM_NOT_FOUND', `팀 디렉토리가 없습니다: ${paths.team_dir}`);
+  }
+
+  const recipient = sanitizeRecipientName(to);
+  const inboxFile = join(paths.inboxes_dir, `${recipient}.json`);
+  const queue = readJsonSafe(inboxFile);
+  const list = Array.isArray(queue) ? queue : [];
+
+  const message = {
+    id: randomUUID(),
+    from: String(from),
+    text: String(text),
+    ...(summary ? { summary: String(summary) } : {}),
+    timestamp: new Date().toISOString(),
+    color: String(color || 'blue'),
+    read: false,
+  };
+  list.push(message);
+
+  try {
+    atomicWriteJson(inboxFile, list);
+  } catch (e) {
+    return err('SEND_MESSAGE_FAILED', e.message);
+  }
+
+  const unreadCount = list.filter((m) => m?.read !== true).length;
+  return {
+    ok: true,
+    data: {
+      message_id: message.id,
+      recipient,
+      inbox_file: inboxFile,
+      queued_at: message.timestamp,
+      unread_count: unreadCount,
+    },
+  };
+}

package/hub/team/orchestrator.mjs

@@ -0,0 +1,166 @@
+// hub/team/orchestrator.mjs — 작업 분배 + 프롬프트 구성
+// 의존성: pane.mjs만 사용
+import { injectPrompt } from "./pane.mjs";
+
+/**
+ * 작업 분해 (LLM 없이 구분자 기반)
+ * @param {string} taskDescription — 전체 작업 설명
+ * @param {number} agentCount — 에이전트 수
+ * @returns {string[]} 각 에이전트의 서브태스크
+ */
+export function decomposeTask(taskDescription, agentCount) {
+  if (agentCount <= 0) return [];
+  if (agentCount === 1) return [taskDescription];
+
+  // '+', ',', '\n' 기준으로 분리
+  const parts = taskDescription
+    .split(/[+,\n]+/)
+    .map((s) => s.trim())
+    .filter(Boolean);
+
+  if (parts.length === 0) return [taskDescription];
+
+  // 에이전트보다 서브태스크가 적으면 마지막 에이전트에 전체 태스크 부여
+  if (parts.length < agentCount) {
+    const result = [...parts];
+    while (result.length < agentCount) {
+      result.push(taskDescription);
+    }
+    return result;
+  }
+
+  // 에이전트보다 서브태스크가 많으면 앞에서부터 N개, 나머지는 마지막에 합침
+  if (parts.length > agentCount) {
+    const result = parts.slice(0, agentCount - 1);
+    result.push(parts.slice(agentCount - 1).join(" + "));
+    return result;
+  }
+
+  return parts;
+}
+
+/**
+ * 리드(보통 claude) 초기 프롬프트 생성
+ * @param {string} taskDescription
+ * @param {object} config
+ * @param {string} config.agentId
+ * @param {string} config.hubUrl
+ * @param {string} config.teammateMode
+ * @param {Array<{agentId:string, cli:string, subtask:string}>} config.workers
+ * @returns {string}
+ */
+export function buildLeadPrompt(taskDescription, config) {
+  const { agentId, hubUrl, teammateMode = "tmux", workers = [] } = config;
+
+  const roster = workers
+    .map((w, i) => `${i + 1}. ${w.agentId} (${w.cli}) — ${w.subtask}`)
+    .join("\n") || "- (워커 없음)";
+
+  const workerIds = workers.map((w) => w.agentId).join(", ");
+
+  return `리드 에이전트: ${agentId}
+
+목표: ${taskDescription}
+모드: ${teammateMode}
+
+워커:
+${roster}
+
+규칙:
+- 가능한 짧고 핵심만 지시/요약(토큰 절약)
+- 워커 제어 메시지 표준:
+  publish(from="${agentId}", to="<worker-agent-id>", topic="lead.control", payload={command:"interrupt|stop|pause|resume", reason:"..."})
+- 워커 결과 수집:
+  poll_messages(agent_id="${agentId}", wait_ms=1000, max_messages=20)
+- 최종 결과는 topic="task.result"를 모아 통합
+
+권장 워커 ID: ${workerIds || "(없음)"}
+Hub: ${hubUrl}
+지금 즉시 워커를 배정하고 병렬 진행을 관리하라.`;
+}
+
+/**
+ * 워커 초기 프롬프트 생성
+ * @param {string} subtask — 이 에이전트의 서브태스크
+ * @param {object} config
+ * @param {string} config.cli — codex/gemini/claude
+ * @param {string} config.agentId — 에이전트 식별자
+ * @param {string} config.hubUrl — Hub URL
+ * @returns {string}
+ */
+export function buildPrompt(subtask, config) {
+  const { cli, agentId, hubUrl } = config;
+
+  const hubBase = hubUrl.replace("/mcp", "");
+
+  return `워커: ${agentId} (${cli})
+작업: ${subtask}
+
+필수 규칙:
+1) 간결하게 작업(불필요한 장문 설명 금지)
+2) 시작 즉시 register 실행:
+   register(agent_id="${agentId}", cli="${cli}", capabilities=["code"], topics=["lead.control","task.result"], heartbeat_ttl_ms=600000)
+3) 주기적으로 수신함 확인:
+   poll_messages(agent_id="${agentId}", wait_ms=1000, max_messages=10)
+4) lead.control 수신 시 즉시 반응
+   - interrupt: 즉시 중단, 진행상태 요약 publish
+   - stop: 작업 종료, 최종 요약 publish 후 대기
+   - pause: 작업 일시정지
+   - resume: 작업 재개
+5) 완료 시 결과 발행:
+   publish(from="${agentId}", to="topic:task.result", topic="task.result", payload={summary:"..."})
+
+MCP가 없으면 REST 폴백:
+- POST ${hubBase}/bridge/register
+- POST ${hubBase}/bridge/result
+
+지금 작업을 시작하라.`;
+}
+
+/**
+ * 팀 오케스트레이션 실행 — 각 pane에 프롬프트 주입
+ * @param {string} sessionName — tmux 세션 이름
+ * @param {Array<{target: string, cli: string, subtask: string}>} assignments
+ * @param {object} opts
+ * @param {string} opts.hubUrl — Hub URL
+ * @param {{target:string, cli:string, task:string}|null} opts.lead
+ * @param {string} opts.teammateMode
+ * @returns {Promise<void>}
+ */
+export async function orchestrate(sessionName, assignments, opts = {}) {
+  const {
+    hubUrl = "http://127.0.0.1:27888/mcp",
+    lead = null,
+    teammateMode = "tmux",
+  } = opts;
+
+  const workers = assignments.map(({ target, cli, subtask }) => ({
+    target,
+    cli,
+    subtask,
+    agentId: `${cli}-${target.split(".").pop()}`,
+  }));
+
+  if (lead?.target) {
+    const leadAgentId = `${lead.cli || "claude"}-${lead.target.split(".").pop()}`;
+    const leadPrompt = buildLeadPrompt(lead.task || "팀 작업 조율", {
+      agentId: leadAgentId,
+      hubUrl,
+      teammateMode,
+      workers: workers.map((w) => ({ agentId: w.agentId, cli: w.cli, subtask: w.subtask })),
+    });
+    injectPrompt(lead.target, leadPrompt);
+    await new Promise((r) => setTimeout(r, 100));
+  }
+
+  for (const worker of workers) {
+    const prompt = buildPrompt(worker.subtask, {
+      cli: worker.cli,
+      agentId: worker.agentId,
+      hubUrl,
+      sessionName,
+    });
+    injectPrompt(worker.target, prompt);
+    await new Promise((r) => setTimeout(r, 100));
+  }
+}