triflux 3.3.0-dev.8 → 4.0.0

package/hub/tools.mjs

@@ -177,9 +177,9 @@ export function createTools(store, router, hitl, pipe = null) {
     },
 
     // ── 6. handoff ──
-    {
-      name: 'handoff',
-      description: '다른 에이전트에게 작업을 인계합니다. acceptance_criteria로 완료 기준 지정 가능',
+    {
+      name: 'handoff',
+      description: '다른 에이전트에게 작업을 인계합니다. acceptance_criteria로 완료 기준 지정 가능',
       inputSchema: {
         type: 'object',
         required: ['from', 'to', 'topic', 'task'],
@@ -196,90 +196,90 @@ export function createTools(store, router, hitl, pipe = null) {
           correlation_id:      { type: 'string' },
         },
       },
-      handler: wrap('HANDOFF_FAILED', (args) => {
-        return router.handleHandoff(args);
-      }),
-    },
-
-    // ── 7. assign_async ──
-    {
-      name: 'assign_async',
-      description: 'AWS CAO 스타일 비차단 assign job을 생성하고 워커에게 실시간 전달합니다',
-      inputSchema: {
-        type: 'object',
-        required: ['supervisor_agent', 'worker_agent', 'task'],
-        properties: {
-          supervisor_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
-          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
-          topic: { type: 'string', pattern: '^[a-zA-Z0-9._:-]+$', default: 'assign.job' },
-          task: { type: 'string', minLength: 1, maxLength: 20000 },
-          payload: { type: 'object' },
-          priority: { type: 'integer', minimum: 1, maximum: 9, default: 5 },
-          ttl_ms: { type: 'integer', minimum: 1000, maximum: 86400000, default: 600000 },
-          timeout_ms: { type: 'integer', minimum: 1000, maximum: 86400000, default: 600000 },
-          max_retries: { type: 'integer', minimum: 0, maximum: 20, default: 0 },
-          trace_id: { type: 'string' },
-          correlation_id: { type: 'string' },
-        },
-      },
-      handler: wrap('ASSIGN_ASYNC_FAILED', (args) => {
-        return router.assignAsync(args);
-      }),
-    },
-
-    // ── 8. assign_result ──
-    {
-      name: 'assign_result',
-      description: 'assign job의 진행/완료 결과를 보고합니다. completed + metadata.result 관례를 지원합니다',
-      inputSchema: {
-        type: 'object',
-        required: ['job_id', 'status'],
-        properties: {
-          job_id: { type: 'string', minLength: 1, maxLength: 128 },
-          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
-          status: { type: 'string', enum: ['queued', 'running', 'in_progress', 'completed', 'succeeded', 'success', 'failed', 'error', 'timed_out', 'timeout'] },
-          attempt: { type: 'integer', minimum: 1 },
-          result: {},
-          error: {},
-          metadata: { type: 'object' },
-          payload: { type: 'object' },
-        },
-      },
-      handler: wrap('ASSIGN_RESULT_FAILED', (args) => {
-        return router.reportAssignResult(args);
-      }),
-    },
-
-    // ── 9. assign_status ──
-    {
-      name: 'assign_status',
-      description: 'assign job 단건 상태 또는 supervisor/worker/status 기준 목록을 조회합니다',
-      inputSchema: {
-        type: 'object',
-        properties: {
-          job_id: { type: 'string', minLength: 1, maxLength: 128 },
-          supervisor_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
-          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
-          status: { type: 'string', enum: ['queued', 'running', 'succeeded', 'failed', 'timed_out'] },
-          statuses: {
-            type: 'array',
-            items: { type: 'string', enum: ['queued', 'running', 'succeeded', 'failed', 'timed_out'] },
-            maxItems: 8,
-          },
-          trace_id: { type: 'string' },
-          correlation_id: { type: 'string' },
-          limit: { type: 'integer', minimum: 1, maximum: 100, default: 50 },
-        },
-      },
-      handler: wrap('ASSIGN_STATUS_FAILED', (args) => {
-        return router.getAssignStatus(args);
-      }),
-    },
-
-    // ── 10. request_human_input ──
-    {
-      name: 'request_human_input',
-      description: '사용자에게 입력을 요청합니다 (CAPTCHA, 승인, 자격증명, 선택, 텍스트)',
+      handler: wrap('HANDOFF_FAILED', (args) => {
+        return router.handleHandoff(args);
+      }),
+    },
+
+    // ── 7. assign_async ──
+    {
+      name: 'assign_async',
+      description: 'AWS CAO 스타일 비차단 assign job을 생성하고 워커에게 실시간 전달합니다',
+      inputSchema: {
+        type: 'object',
+        required: ['supervisor_agent', 'worker_agent', 'task'],
+        properties: {
+          supervisor_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          topic: { type: 'string', pattern: '^[a-zA-Z0-9._:-]+$', default: 'assign.job' },
+          task: { type: 'string', minLength: 1, maxLength: 20000 },
+          payload: { type: 'object' },
+          priority: { type: 'integer', minimum: 1, maximum: 9, default: 5 },
+          ttl_ms: { type: 'integer', minimum: 1000, maximum: 86400000, default: 600000 },
+          timeout_ms: { type: 'integer', minimum: 1000, maximum: 86400000, default: 600000 },
+          max_retries: { type: 'integer', minimum: 0, maximum: 20, default: 0 },
+          trace_id: { type: 'string' },
+          correlation_id: { type: 'string' },
+        },
+      },
+      handler: wrap('ASSIGN_ASYNC_FAILED', (args) => {
+        return router.assignAsync(args);
+      }),
+    },
+
+    // ── 8. assign_result ──
+    {
+      name: 'assign_result',
+      description: 'assign job의 진행/완료 결과를 보고합니다. completed + metadata.result 관례를 지원합니다',
+      inputSchema: {
+        type: 'object',
+        required: ['job_id', 'status'],
+        properties: {
+          job_id: { type: 'string', minLength: 1, maxLength: 128 },
+          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          status: { type: 'string', enum: ['queued', 'running', 'in_progress', 'completed', 'succeeded', 'success', 'failed', 'error', 'timed_out', 'timeout'] },
+          attempt: { type: 'integer', minimum: 1 },
+          result: {},
+          error: {},
+          metadata: { type: 'object' },
+          payload: { type: 'object' },
+        },
+      },
+      handler: wrap('ASSIGN_RESULT_FAILED', (args) => {
+        return router.reportAssignResult(args);
+      }),
+    },
+
+    // ── 9. assign_status ──
+    {
+      name: 'assign_status',
+      description: 'assign job 단건 상태 또는 supervisor/worker/status 기준 목록을 조회합니다',
+      inputSchema: {
+        type: 'object',
+        properties: {
+          job_id: { type: 'string', minLength: 1, maxLength: 128 },
+          supervisor_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          worker_agent: { type: 'string', pattern: '^[a-zA-Z0-9._:-]{3,64}$' },
+          status: { type: 'string', enum: ['queued', 'running', 'succeeded', 'failed', 'timed_out'] },
+          statuses: {
+            type: 'array',
+            items: { type: 'string', enum: ['queued', 'running', 'succeeded', 'failed', 'timed_out'] },
+            maxItems: 8,
+          },
+          trace_id: { type: 'string' },
+          correlation_id: { type: 'string' },
+          limit: { type: 'integer', minimum: 1, maximum: 100, default: 50 },
+        },
+      },
+      handler: wrap('ASSIGN_STATUS_FAILED', (args) => {
+        return router.getAssignStatus(args);
+      }),
+    },
+
+    // ── 10. request_human_input ──
+    {
+      name: 'request_human_input',
+      description: '사용자에게 입력을 요청합니다 (CAPTCHA, 승인, 자격증명, 선택, 텍스트)',
       inputSchema: {
         type: 'object',
         required: ['requester_agent', 'kind', 'prompt', 'requested_schema', 'deadline_ms', 'default_action'],
@@ -295,12 +295,12 @@ export function createTools(store, router, hitl, pipe = null) {
           correlation_id:     { type: 'string' },
         },
       },
-      handler: wrap('HITL_REQUEST_FAILED', (args) => {
-        return hitl.requestHumanInput(args);
-      }),
-    },
-
-    // ── 11. submit_human_input ──
+      handler: wrap('HITL_REQUEST_FAILED', (args) => {
+        return hitl.requestHumanInput(args);
+      }),
+    },
+
+    // ── 11. submit_human_input ──
     {
       name: 'submit_human_input',
       description: '사용자 입력 요청에 응답합니다 (accept, decline, cancel)',
@@ -319,7 +319,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 12. team_info ──
+    // ── 12. team_info ──
     {
       name: 'team_info',
       description: 'Claude Native Teams 메타/멤버/경로 정보를 조회합니다',
@@ -337,10 +337,10 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 13. team_task_list ──
-    {
-      name: 'team_task_list',
-      description: 'Claude Native Teams task 목록을 owner/status 조건으로 조회합니다. 실패 판정은 completed + metadata.result도 함께 확인해야 합니다',
+    // ── 13. team_task_list ──
+    {
+      name: 'team_task_list',
+      description: 'Claude Native Teams task 목록을 owner/status 조건으로 조회합니다. 실패 판정은 completed + metadata.result도 함께 확인해야 합니다',
       inputSchema: {
         type: 'object',
         required: ['team_name'],
@@ -361,10 +361,10 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 14. team_task_update ──
-    {
-      name: 'team_task_update',
-      description: 'Claude Native Teams task를 claim/update 합니다. status: "failed" 입력은 completed + metadata.result="failed"로 정규화됩니다',
+    // ── 14. team_task_update ──
+    {
+      name: 'team_task_update',
+      description: 'Claude Native Teams task를 claim/update 합니다. status: "failed" 입력은 completed + metadata.result="failed"로 정규화됩니다',
       inputSchema: {
         type: 'object',
         required: ['team_name', 'task_id'],
@@ -389,7 +389,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 15. team_send_message ──
+    // ── 15. team_send_message ──
     {
       name: 'team_send_message',
       description: 'Claude Native Teams inbox에 메시지를 append 합니다',
@@ -410,7 +410,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 16. pipeline_state ──
+    // ── 16. pipeline_state ──
     {
       name: 'pipeline_state',
       description: '파이프라인 상태를 조회합니다 (--thorough 모드)',
@@ -430,7 +430,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 17. pipeline_advance ──
+    // ── 17. pipeline_advance ──
     {
       name: 'pipeline_advance',
       description: '파이프라인을 다음 단계로 전이합니다 (전이 규칙 + fix loop 바운딩 적용)',
@@ -449,7 +449,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 18. pipeline_init ──
+    // ── 18. pipeline_init ──
     {
       name: 'pipeline_init',
       description: '새 파이프라인을 초기화합니다 (기존 상태 덮어쓰기)',
@@ -472,7 +472,7 @@ export function createTools(store, router, hitl, pipe = null) {
       }),
     },
 
-    // ── 19. pipeline_list ──
+    // ── 19. pipeline_list ──
     {
       name: 'pipeline_list',
       description: '활성 파이프라인 목록을 조회합니다',

package/hub/tray.mjs

@@ -0,0 +1,332 @@
+#!/usr/bin/env node
+
+import _SysTrayModule from "systray2";
+const SysTray = _SysTrayModule.default || _SysTrayModule;
+import { exec } from "node:child_process";
+import { readFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { join, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+
+const DASHBOARD_URL = "http://127.0.0.1:27888/dashboard";
+const HUB_STATUS_URL = "http://127.0.0.1:27888/status";
+const POLL_INTERVAL_MS = 10_000;
+const HUB_TIMEOUT_MS = 3_000;
+const AIMD_INITIAL = 3;
+const AIMD_MIN = 1;
+const AIMD_MAX = 10;
+const AIMD_WINDOW_MS = 30 * 60 * 1000;
+
+const CACHE_DIR = join(homedir(), ".claude", "cache");
+const BATCH_EVENTS_FILE = join(CACHE_DIR, "batch-events.jsonl");
+const CODEX_RATE_LIMITS_FILE = join(CACHE_DIR, "codex-rate-limits-cache.json");
+const GEMINI_QUOTA_FILE = join(CACHE_DIR, "gemini-quota-cache.json");
+const CLAUDE_USAGE_FILE = join(CACHE_DIR, "claude-usage-cache.json");
+
+const TRAY_ICON_BASE64 = "AAABAAEAICAAAAEAIAADAQAAFgAAAIlQTkcNChoKAAAADUlIRFIAAAAgAAAAIAgGAAAAc3p69AAAAMpJREFUeJzV1UEKgzAQheEcwnXP4a17gl6n6yy7U1IIqDSTeW/m0TYwK8X/E6OW8m9rWW5Pa74SlWLYeBgRDcOQ7V42a+SIGSALkwKIQKDnroJQmy4bAgO8EDnAA4EBkV3do7VWeAOfAO0Cx/EC+vnMG2QCLND12Pp4vUcK+DQ9fBxqI47uDI23oV7F2fNVxF2A64zCTBwCWGE2Pv0WzKKp8ba8wWg4DIiGKUBWdBjP+i+E4z8BUCJccRUCimdC6HAGIiWOYiRR5doBauXshzcEs0UAAAAASUVORK5CYII=";
+
+function clampPercent(value) {
+  if (!Number.isFinite(Number(value))) return null;
+  return Math.max(0, Math.min(100, Math.round(Number(value))));
+}
+
+function readJson(filePath) {
+  try {
+    return JSON.parse(readFileSync(filePath, "utf8"));
+  } catch {
+    return null;
+  }
+}
+
+function readLines(filePath) {
+  try {
+    return readFileSync(filePath, "utf8").split(/\r?\n/).filter(Boolean);
+  } catch {
+    return [];
+  }
+}
+
+function isSuccessResult(result) {
+  return result === "success" || result === "success_with_warnings";
+}
+
+function getAimdBatchSize(now = Date.now()) {
+  const sinceMs = now - AIMD_WINDOW_MS;
+  const events = readLines(BATCH_EVENTS_FILE)
+    .map((line) => {
+      try {
+        return JSON.parse(line);
+      } catch {
+        return null;
+      }
+    })
+    .filter((event) => event && Number(event.ts) >= sinceMs);
+
+  if (events.length === 0) return AIMD_INITIAL;
+
+  let batchSize = AIMD_INITIAL;
+  for (const event of events) {
+    if (isSuccessResult(event.result)) {
+      batchSize = Math.min(AIMD_MAX, batchSize + 1);
+    } else {
+      batchSize = Math.max(AIMD_MIN, Math.floor(batchSize * 0.5));
+    }
+  }
+  return batchSize;
+}
+
+function getCodexPercent() {
+  const data = readJson(CODEX_RATE_LIMITS_FILE);
+  const buckets = data?.buckets && typeof data.buckets === "object" ? data.buckets : null;
+  const primaryBucket = buckets?.codex ?? Object.values(buckets ?? {})[0] ?? null;
+  return clampPercent(primaryBucket?.primary?.used_percent);
+}
+
+function pickGeminiBucket(data) {
+  const buckets = Array.isArray(data?.buckets) ? data.buckets : [];
+  if (buckets.length === 0) return null;
+
+  const preferredModels = [
+    "gemini-3-flash-preview",
+    "gemini-2.5-flash",
+    "gemini-3-flash",
+    "gemini-2.5-flash-lite",
+  ];
+
+  for (const modelId of preferredModels) {
+    const match = buckets.find((bucket) => bucket?.modelId === modelId);
+    if (match) return match;
+  }
+
+  return buckets.find((bucket) => String(bucket?.modelId ?? "").includes("flash")) ?? buckets[0];
+}
+
+function getGeminiPercent() {
+  const data = readJson(GEMINI_QUOTA_FILE);
+  const bucket = pickGeminiBucket(data);
+  if (!bucket) return null;
+  return clampPercent((1 - Number(bucket.remainingFraction ?? 1)) * 100);
+}
+
+function getClaudePercent() {
+  const data = readJson(CLAUDE_USAGE_FILE);
+  return clampPercent(data?.data?.fiveHourPercent);
+}
+
+async function getHubStatusLabel() {
+  try {
+    const response = await fetch(HUB_STATUS_URL, {
+      signal: AbortSignal.timeout(HUB_TIMEOUT_MS),
+    });
+    if (!response.ok) throw new Error(`HTTP ${response.status}`);
+
+    const data = await response.json();
+    const state = typeof data?.hub?.state === "string" ? data.hub.state : "connected";
+    const sessions = Number.isFinite(Number(data?.sessions)) ? Number(data.sessions) : null;
+    return sessions == null ? `Hub: ${state}` : `Hub: ${state} | S:${sessions}`;
+  } catch {
+    return "Hub 미연결";
+  }
+}
+
+function formatTooltipPercent(value) {
+  return value == null ? "--%" : `${value}%`;
+}
+
+function formatMenuPercent(value) {
+  return value == null ? "--%" : `${value}%`;
+}
+
+function buildTooltip(snapshot) {
+  return `tfx AIMD:${snapshot.aimd}/10 | C:${formatTooltipPercent(snapshot.claude)} X:${formatTooltipPercent(snapshot.codex)} G:${formatTooltipPercent(snapshot.gemini)}`;
+}
+
+function buildUsageTitle(snapshot) {
+  return `C: ${formatMenuPercent(snapshot.claude)} | X: ${formatMenuPercent(snapshot.codex)} | G: ${formatMenuPercent(snapshot.gemini)}`;
+}
+
+function openDashboard() {
+  exec(`start "" "${DASHBOARD_URL}"`, {
+    shell: process.env.ComSpec || "cmd.exe",
+    windowsHide: true,
+  }, () => {});
+}
+
+const openDashboardItem = {
+  title: "대시보드 열기",
+  tooltip: "브라우저에서 대시보드 열기",
+  enabled: true,
+  click: openDashboard,
+};
+
+const aimdItem = {
+  title: "AIMD: 3/10",
+  tooltip: "최근 30분 AIMD 동시 워커",
+  enabled: false,
+};
+
+const quotaItem = {
+  title: "C: --% | X: --% | G: --%",
+  tooltip: "Claude | Codex | Gemini 사용률",
+  enabled: false,
+};
+
+const hubItem = {
+  title: "Hub 미연결",
+  tooltip: "Hub 연결 상태",
+  enabled: false,
+};
+
+const refreshItem = {
+  title: "새로고침",
+  tooltip: "캐시 재읽기",
+  enabled: true,
+  click: () => {
+    void scheduleRefresh();
+  },
+};
+
+const exitItem = {
+  title: "종료",
+  tooltip: "트레이 종료",
+  enabled: true,
+  click: () => {
+    void shutdown("menu-exit");
+  },
+};
+
+const menu = {
+  icon: TRAY_ICON_BASE64,
+  title: "tfx",
+  tooltip: "tfx AIMD:3/10 | C:--% X:--% G:--%",
+  items: [
+    openDashboardItem,
+    SysTray.separator,
+    aimdItem,
+    quotaItem,
+    hubItem,
+    SysTray.separator,
+    refreshItem,
+    exitItem,
+  ],
+};
+
+let systray = null;
+let pollTimer = null;
+let refreshPromise = null;
+let shuttingDown = false;
+
+async function refreshMenu() {
+  const snapshot = {
+    aimd: getAimdBatchSize(),
+    codex: getCodexPercent(),
+    gemini: getGeminiPercent(),
+    claude: getClaudePercent(),
+    hubLabel: await getHubStatusLabel(),
+  };
+
+  aimdItem.title = `AIMD: ${snapshot.aimd}/10`;
+  quotaItem.title = buildUsageTitle(snapshot);
+  hubItem.title = snapshot.hubLabel;
+
+  if (systray) {
+    await systray.sendAction({ type: "update-item", item: aimdItem });
+    await systray.sendAction({ type: "update-item", item: quotaItem });
+    await systray.sendAction({ type: "update-item", item: hubItem });
+    await systray.sendAction({ type: "update-item-and-title", item: { title: buildTooltip(snapshot) } });
+  }
+}
+
+function scheduleRefresh() {
+  if (refreshPromise) return refreshPromise;
+  refreshPromise = refreshMenu().catch((error) => {
+    console.error(`[tfx-tray] refresh failed: ${error.message}`);
+  }).finally(() => {
+    refreshPromise = null;
+  });
+  return refreshPromise;
+}
+
+async function shutdown(reason = "shutdown") {
+  if (shuttingDown) return;
+  shuttingDown = true;
+
+  if (pollTimer) {
+    clearInterval(pollTimer);
+    pollTimer = null;
+  }
+
+  try {
+    if (systray && !systray.killed) {
+      await systray.kill(false);
+    }
+  } catch (error) {
+    console.error(`[tfx-tray] ${reason} cleanup failed: ${error.message}`);
+  } finally {
+    systray = null;
+    process.exit(0);
+  }
+}
+
+export async function startTray() {
+  if (process.platform !== "win32") {
+    throw new Error("tray command is only supported on Windows.");
+  }
+
+  systray = new SysTray({
+    menu,
+    debug: false,
+    copyDir: false,
+  });
+
+  await systray.ready();
+
+  systray.onError((error) => {
+    console.error(`[tfx-tray] ${error.message}`);
+  });
+
+  systray.onExit((code, signal) => {
+    if (shuttingDown) return;
+    const detail = signal ? `signal ${signal}` : `code ${code ?? 0}`;
+    console.error(`[tfx-tray] tray exited unexpectedly (${detail})`);
+    process.exit(typeof code === "number" ? code : 1);
+  });
+  await systray.onClick((action) => {
+    if (action.item?.click) {
+      action.item.click();
+      return;
+    }
+
+    if (action.item?.__id === openDashboardItem.__id) {
+      openDashboard();
+    }
+  });
+
+  await scheduleRefresh();
+
+  pollTimer = setInterval(() => {
+    void scheduleRefresh();
+  }, POLL_INTERVAL_MS);
+  pollTimer.unref();
+
+  process.on("SIGINT", () => {
+    void shutdown("SIGINT");
+  });
+  process.on("SIGTERM", () => {
+    void shutdown("SIGTERM");
+  });
+
+  return {
+    systray,
+    stop: shutdown,
+  };
+}
+
+const selfRun = process.argv[1] && resolve(process.argv[1]) === resolve(fileURLToPath(import.meta.url));
+
+if (selfRun) {
+  startTray().catch((error) => {
+    console.error(`[tfx-tray] start failed: ${error.message}`);
+    process.exit(1);
+  });
+}