triflux 10.9.7 → 10.9.8

package/hub/account-broker.mjs

@@ -4,7 +4,7 @@
 // Singleton export. All state changes create new objects (immutable pattern).
 
 import { EventEmitter } from "node:events";
-import { existsSync, readFileSync } from "node:fs";
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { homedir } from "node:os";
 import { join, sep } from "node:path";
 import * as z from "zod";
@@ -55,6 +55,39 @@ const LEASE_TTL_MS = 30 * 60 * 1000; // 30 minutes
 const CIRCUIT_WINDOW_MS = 10 * 60_000; // 10 minutes
 const CIRCUIT_MAX_FAILURES = 3;
 const AUTH_BASE_PATH = join(homedir(), ".claude", "cache", "tfx-hub");
+const STATE_PERSIST_PATH = join(AUTH_BASE_PATH, "broker-state.json");
+
+// ── State persistence ────────────────────────────────────────────
+
+function persistState(stateMap) {
+  try {
+    const now = Date.now();
+    const entries = {};
+    for (const [id, acct] of stateMap) {
+      // 활성 쿨다운 또는 circuit open만 저장 (불필요한 데이터 제거)
+      if (acct.cooldownUntil > now || acct.circuitOpenedAt > 0 || acct.totalSessions > 0) {
+        entries[id] = {
+          cooldownUntil: acct.cooldownUntil,
+          circuitOpenedAt: acct.circuitOpenedAt,
+          failureTimestamps: acct.failureTimestamps,
+          totalSessions: acct.totalSessions,
+          lastUsedAt: acct.lastUsedAt,
+        };
+      }
+    }
+    mkdirSync(AUTH_BASE_PATH, { recursive: true });
+    writeFileSync(STATE_PERSIST_PATH, JSON.stringify({ ts: now, entries }));
+  } catch { /* best-effort */ }
+}
+
+function loadPersistedState() {
+  try {
+    if (!existsSync(STATE_PERSIST_PATH)) return null;
+    return JSON.parse(readFileSync(STATE_PERSIST_PATH, "utf8"));
+  } catch {
+    return null;
+  }
+}
 
 // ── env var resolution ───────────────────────────────────────────
 
@@ -101,7 +134,11 @@ class AccountBroker extends EventEmitter {
       ...(parsed.gemini || []).map((a) => ({ ...a, provider: "gemini" })),
     ];
 
+    const persisted = loadPersistedState();
+    const pEntries = persisted?.entries || {};
+
     for (const account of allAccounts) {
+      const saved = pEntries[account.id];
       this.#state.set(account.id, {
         id: account.id,
         provider: account.provider,
@@ -113,13 +150,12 @@ class AccountBroker extends EventEmitter {
         tier: account.tier ?? "unknown",
         busy: false,
         leasedAt: null,
-        cooldownUntil: 0,
-        // per-account circuit breaker state
-        failureTimestamps: [], // timestamp array for window-based decay
-        circuitOpenedAt: 0,
+        cooldownUntil: saved?.cooldownUntil ?? 0,
+        failureTimestamps: saved?.failureTimestamps ?? [],
+        circuitOpenedAt: saved?.circuitOpenedAt ?? 0,
         circuitTrialInFlight: false,
-        lastUsedAt: 0,
-        totalSessions: 0,
+        lastUsedAt: saved?.lastUsedAt ?? 0,
+        totalSessions: saved?.totalSessions ?? 0,
       });
     }
   }
@@ -343,6 +379,7 @@ class AccountBroker extends EventEmitter {
     };
 
     this.#state.set(accountId, updated);
+    persistState(this.#state);
     this.emit("release", { id: accountId, ok });
   }
 
@@ -357,6 +394,7 @@ class AccountBroker extends EventEmitter {
       leasedAt: null,
       cooldownUntil: Date.now() + coolMs,
     });
+    persistState(this.#state);
   }
 
   // ── snapshot ──────────────────────────────────────────────────

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "triflux",
-  "version": "10.9.7",
+  "version": "10.9.8",
   "description": "CLI-first multi-model orchestrator for Claude Code — route tasks to Codex, Gemini, and Claude",
   "type": "module",
   "bin": {

package/scripts/setup.mjs

@@ -64,6 +64,43 @@ const REQUIRED_CODEX_PROFILES = [
 
 const HUD_SYNC_EXCLUDES = new Set(["omc-hud.mjs", "omc-hud.mjs.bak"]);
 
+/**
+ * hub/workers/*.mjs + hub/ 루트의 worker 의존성 파일을 자동 스캔.
+ * 수동 리스트 대신 glob으로 탐색하여 파일 추가 시 sync 누락 방지.
+ */
+function scanHubWorkerFiles(pluginRoot, claudeDir) {
+  const results = [];
+  const hubRoot = join(pluginRoot, "hub");
+  if (!existsSync(hubRoot)) return results;
+
+  // hub/workers/*.mjs 전체
+  const workersDir = join(hubRoot, "workers");
+  if (existsSync(workersDir)) {
+    for (const f of readdirSync(workersDir).sort()) {
+      if (!f.endsWith(".mjs")) continue;
+      results.push({
+        src: join(workersDir, f),
+        dst: join(claudeDir, "scripts", "hub", "workers", f),
+        label: `hub/workers/${f}`,
+      });
+    }
+  }
+
+  // hub/ 루트: worker가 import하는 의존성 (cli-adapter-base, platform 등)
+  const hubRootDeps = ["cli-adapter-base.mjs", "platform.mjs", "account-broker.mjs"];
+  for (const f of hubRootDeps) {
+    if (existsSync(join(hubRoot, f))) {
+      results.push({
+        src: join(hubRoot, f),
+        dst: join(claudeDir, "scripts", "hub", f),
+        label: `hub/${f}`,
+      });
+    }
+  }
+
+  return results;
+}
+
 function scanHudFiles(pluginRoot, claudeDir) {
   const hudRoot = join(pluginRoot, "hud");
   if (!existsSync(hudRoot)) return [];
@@ -124,51 +161,7 @@ const SYNC_MAP = [
     dst: join(CLAUDE_DIR, "scripts", "tfx-route-worker.mjs"),
     label: "tfx-route-worker.mjs",
   },
-  {
-    src: join(PLUGIN_ROOT, "hub", "workers", "codex-mcp.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "workers", "codex-mcp.mjs"),
-    label: "hub/workers/codex-mcp.mjs",
-  },
-  {
-    src: join(PLUGIN_ROOT, "hub", "workers", "delegator-mcp.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "workers", "delegator-mcp.mjs"),
-    label: "hub/workers/delegator-mcp.mjs",
-  },
-  {
-    src: join(PLUGIN_ROOT, "hub", "workers", "interface.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "workers", "interface.mjs"),
-    label: "hub/workers/interface.mjs",
-  },
-  {
-    src: join(PLUGIN_ROOT, "hub", "workers", "gemini-worker.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "workers", "gemini-worker.mjs"),
-    label: "hub/workers/gemini-worker.mjs",
-  },
-  {
-    src: join(PLUGIN_ROOT, "hub", "workers", "claude-worker.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "workers", "claude-worker.mjs"),
-    label: "hub/workers/claude-worker.mjs",
-  },
-  {
-    src: join(PLUGIN_ROOT, "hub", "workers", "worker-utils.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "workers", "worker-utils.mjs"),
-    label: "hub/workers/worker-utils.mjs",
-  },
-  {
-    src: join(PLUGIN_ROOT, "hub", "workers", "factory.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "workers", "factory.mjs"),
-    label: "hub/workers/factory.mjs",
-  },
-  {
-    src: join(PLUGIN_ROOT, "hub", "cli-adapter-base.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "cli-adapter-base.mjs"),
-    label: "hub/cli-adapter-base.mjs",
-  },
-  {
-    src: join(PLUGIN_ROOT, "hub", "platform.mjs"),
-    dst: join(CLAUDE_DIR, "scripts", "hub", "platform.mjs"),
-    label: "hub/platform.mjs",
-  },
+  ...scanHubWorkerFiles(PLUGIN_ROOT, CLAUDE_DIR),
   ...scanHudFiles(PLUGIN_ROOT, CLAUDE_DIR),
   {
     src: join(PLUGIN_ROOT, "scripts", "notion-read.mjs"),