triflux 10.3.4 → 10.7.0

package/hud/constants.mjs

@@ -6,24 +6,77 @@ import { join } from "node:path";
 
 export const VERSION = "2.0";
 
-export const QOS_PATH = join(homedir(), ".omc", "state", "cli_qos_profile.json");
-export const ACCOUNTS_CONFIG_PATH = join(homedir(), ".omc", "router", "accounts.json");
-export const ACCOUNTS_STATE_PATH = join(homedir(), ".omc", "state", "cli_accounts_state.json");
+export const QOS_PATH = join(
+  homedir(),
+  ".omc",
+  "state",
+  "cli_qos_profile.json",
+);
+export const ACCOUNTS_CONFIG_PATH = join(
+  homedir(),
+  ".omc",
+  "router",
+  "accounts.json",
+);
+export const ACCOUNTS_STATE_PATH = join(
+  homedir(),
+  ".omc",
+  "state",
+  "cli_accounts_state.json",
+);
 
 // tfx-multi 상태 (v2.2 HUD 통합)
-export const TEAM_STATE_PATH = join(homedir(), ".claude", "cache", "tfx-hub", "team-state.json");
-export const CONTEXT_MONITOR_CACHE_PATH = join(homedir(), ".claude", "cache", "tfx-hub", "context-monitor.json");
-export const CONTEXT_MONITOR_LEGACY_PATH = join(homedir(), ".omc", "state", "context-monitor.json");
+export const TEAM_STATE_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "tfx-hub",
+  "team-state.json",
+);
+export const CONTEXT_MONITOR_CACHE_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "tfx-hub",
+  "context-monitor.json",
+);
+export const CONTEXT_MONITOR_LEGACY_PATH = join(
+  homedir(),
+  ".omc",
+  "state",
+  "context-monitor.json",
+);
 export const CONTEXT_MONITOR_LOG_DIR = join(homedir(), ".omc", "logs");
 
 // 원격 프로브 캐시 (tfx-remote-spawn)
-export const REMOTE_ENV_CACHE_DIR = join(homedir(), ".claude", "cache", "tfx-hub", "remote-env");
+export const REMOTE_ENV_CACHE_DIR = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "tfx-hub",
+  "remote-env",
+);
 export const REMOTE_ENV_CACHE_TTL_MS = 24 * 60 * 60 * 1000; // 24시간
 
 // Claude OAuth Usage API (api.anthropic.com/api/oauth/usage)
-export const CLAUDE_CREDENTIALS_PATH = join(homedir(), ".claude", ".credentials.json");
-export const CLAUDE_USAGE_CACHE_PATH = join(homedir(), ".claude", "cache", "claude-usage-cache.json");
-export const OMC_PLUGIN_USAGE_CACHE_PATH = join(homedir(), ".claude", "plugins", "oh-my-claudecode", ".usage-cache.json");
+export const CLAUDE_CREDENTIALS_PATH = join(
+  homedir(),
+  ".claude",
+  ".credentials.json",
+);
+export const CLAUDE_USAGE_CACHE_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "claude-usage-cache.json",
+);
+export const OMC_PLUGIN_USAGE_CACHE_PATH = join(
+  homedir(),
+  ".claude",
+  "plugins",
+  "oh-my-claudecode",
+  ".usage-cache.json",
+);
 export const CLAUDE_USAGE_STALE_MS_SOLO = 5 * 60 * 1000; // OMC 없을 때: 5분 캐시
 export const CLAUDE_USAGE_STALE_MS_WITH_OMC = 15 * 60 * 1000; // OMC 있을 때: 15분 (OMC가 30초마다 갱신)
 export const CLAUDE_USAGE_429_BACKOFF_MS = 10 * 60 * 1000; // 429 에러 시 10분 backoff
@@ -38,29 +91,105 @@ export const ONE_DAY_MS = 24 * 60 * 60 * 1000;
 export const DEFAULT_OAUTH_CLIENT_ID = "9d1c250a-e61b-44d9-88ed-5944d1962f5e";
 
 export const CODEX_AUTH_PATH = join(homedir(), ".codex", "auth.json");
-export const CODEX_QUOTA_CACHE_PATH = join(homedir(), ".claude", "cache", "codex-rate-limits-cache.json");
+export const CODEX_QUOTA_CACHE_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "codex-rate-limits-cache.json",
+);
 export const CODEX_QUOTA_STALE_MS = 30 * 1000; // 30초
 export const CODEX_MIN_BUCKETS = 2;
 
 // Spawn lock (중복 refresh 방지)
-export const CODEX_REFRESH_LOCK_PATH = join(homedir(), ".claude", "cache", ".codex-refresh-lock");
-export const GEMINI_QUOTA_REFRESH_LOCK_PATH = join(homedir(), ".claude", "cache", ".gemini-quota-refresh-lock");
-export const GEMINI_SESSION_REFRESH_LOCK_PATH = join(homedir(), ".claude", "cache", ".gemini-session-refresh-lock");
+export const CLAUDE_REFRESH_LOCK_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  ".claude-refresh-lock",
+);
+export const CODEX_REFRESH_LOCK_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  ".codex-refresh-lock",
+);
+export const GEMINI_QUOTA_REFRESH_LOCK_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  ".gemini-quota-refresh-lock",
+);
+export const GEMINI_SESSION_REFRESH_LOCK_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  ".gemini-session-refresh-lock",
+);
 export const SPAWN_LOCK_TTL_MS = 30 * 1000; // 30초 spawn dedup
 
 // Gemini 쿼터 API 관련
 export const GEMINI_OAUTH_PATH = join(homedir(), ".gemini", "oauth_creds.json");
-export const GEMINI_QUOTA_CACHE_PATH = join(homedir(), ".claude", "cache", "gemini-quota-cache.json");
-export const GEMINI_PROJECT_CACHE_PATH = join(homedir(), ".claude", "cache", "gemini-project-id.json");
-export const GEMINI_SESSION_CACHE_PATH = join(homedir(), ".claude", "cache", "gemini-session-cache.json");
-export const GEMINI_RPM_TRACKER_PATH = join(homedir(), ".claude", "cache", "gemini-rpm-tracker.json");
-export const SV_ACCUMULATOR_PATH = join(homedir(), ".claude", "cache", "sv-accumulator.json");
+export const GEMINI_QUOTA_CACHE_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "gemini-quota-cache.json",
+);
+export const GEMINI_PROJECT_CACHE_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "gemini-project-id.json",
+);
+export const GEMINI_SESSION_CACHE_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "gemini-session-cache.json",
+);
+export const GEMINI_RPM_TRACKER_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "gemini-rpm-tracker.json",
+);
+export const SV_ACCUMULATOR_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  "sv-accumulator.json",
+);
 // 이전 .omc/ 경로 fallback (기존 환경 호환)
-export const LEGACY_GEMINI_QUOTA_CACHE = join(homedir(), ".omc", "state", "gemini_quota_cache.json");
-export const LEGACY_GEMINI_PROJECT_CACHE = join(homedir(), ".omc", "state", "gemini_project_id.json");
-export const LEGACY_GEMINI_SESSION_CACHE = join(homedir(), ".omc", "state", "gemini_session_tokens_cache.json");
-export const LEGACY_GEMINI_RPM_TRACKER = join(homedir(), ".omc", "state", "gemini_rpm_tracker.json");
-export const LEGACY_SV_ACCUMULATOR = join(homedir(), ".omc", "state", "sv-accumulator.json");
+export const LEGACY_GEMINI_QUOTA_CACHE = join(
+  homedir(),
+  ".omc",
+  "state",
+  "gemini_quota_cache.json",
+);
+export const LEGACY_GEMINI_PROJECT_CACHE = join(
+  homedir(),
+  ".omc",
+  "state",
+  "gemini_project_id.json",
+);
+export const LEGACY_GEMINI_SESSION_CACHE = join(
+  homedir(),
+  ".omc",
+  "state",
+  "gemini_session_tokens_cache.json",
+);
+export const LEGACY_GEMINI_RPM_TRACKER = join(
+  homedir(),
+  ".omc",
+  "state",
+  "gemini_rpm_tracker.json",
+);
+export const LEGACY_SV_ACCUMULATOR = join(
+  homedir(),
+  ".omc",
+  "state",
+  "sv-accumulator.json",
+);
 
 export const GEMINI_RPM_WINDOW_MS = 60 * 1000; // 60초 슬라이딩 윈도우
 export const GEMINI_QUOTA_STALE_MS = 5 * 60 * 1000; // 5분
@@ -90,5 +219,12 @@ export const ROWS_BUDGET_MEDIUM = 28;
 export const ROWS_BUDGET_SMALL = 22;
 
 // Gemini Pro 풀 공유 그룹: 같은 remainingFraction을 공유하는 모델 ID들
-export const GEMINI_PRO_POOL = new Set(["gemini-2.5-pro", "gemini-3-pro-preview", "gemini-3.1-pro-preview"]);
-export const GEMINI_FLASH_POOL = new Set(["gemini-2.5-flash", "gemini-3-flash-preview"]);
+export const GEMINI_PRO_POOL = new Set([
+  "gemini-2.5-pro",
+  "gemini-3-pro-preview",
+  "gemini-3.1-pro-preview",
+]);
+export const GEMINI_FLASH_POOL = new Set([
+  "gemini-2.5-flash",
+  "gemini-3-flash-preview",
+]);

package/hud/context-monitor.mjs

@@ -1,7 +1,6 @@
+import { randomUUID } from "node:crypto";
 import { mkdirSync, writeFileSync } from "node:fs";
 import { dirname, join } from "node:path";
-import { randomUUID } from "node:crypto";
-import { homedir } from "node:os";
 
 import {
   CONTEXT_MONITOR_CACHE_PATH,
@@ -72,21 +71,24 @@ function normalizeUsage(usage) {
   const input = Number(usage.input_tokens ?? usage.inputTokens ?? 0);
   const output = Number(usage.output_tokens ?? usage.outputTokens ?? 0);
   const cacheCreation = Number(
-    usage.cache_creation_input_tokens
-      ?? usage.cacheCreationInputTokens
-      ?? usage.cache_creation_tokens
-      ?? 0,
+    usage.cache_creation_input_tokens ??
+      usage.cacheCreationInputTokens ??
+      usage.cache_creation_tokens ??
+      0,
   );
   const cacheRead = Number(
-    usage.cache_read_input_tokens
-      ?? usage.cacheReadInputTokens
-      ?? usage.cache_read_tokens
-      ?? 0,
+    usage.cache_read_input_tokens ??
+      usage.cacheReadInputTokens ??
+      usage.cache_read_tokens ??
+      0,
+  );
+  const totalCandidate = Number(
+    usage.total_tokens ?? usage.totalTokens ?? Number.NaN,
   );
-  const totalCandidate = Number(usage.total_tokens ?? usage.totalTokens ?? Number.NaN);
-  const total = Number.isFinite(totalCandidate) && totalCandidate > 0
-    ? totalCandidate
-    : input + output + cacheCreation + cacheRead;
+  const total =
+    Number.isFinite(totalCandidate) && totalCandidate > 0
+      ? totalCandidate
+      : input + output + cacheCreation + cacheRead;
   if (!Number.isFinite(total) || total <= 0) return null;
   return {
     input: Math.max(0, Math.round(input)),
@@ -159,7 +161,10 @@ function extractFileKeys(args) {
 function detectSkillHints(payloadText) {
   if (!payloadText) return [];
   const matches = payloadText.match(/\$[a-z0-9_-]+/gi) || [];
-  return Array.from(new Set(matches.map((m) => m.replace(/^\$/, "")))).slice(0, 5);
+  return Array.from(new Set(matches.map((m) => m.replace(/^\$/, "")))).slice(
+    0,
+    5,
+  );
 }
 
 export function estimateTokens(input) {
@@ -172,21 +177,34 @@ export function parseUsageFromPayload(payload) {
 
 export function classifyContextThreshold(percent) {
   const p = clampThresholdPercent(percent);
-  if (p >= WARNING_LEVELS.critical.min) return { level: "critical", message: WARNING_LEVELS.critical.message };
-  if (p >= WARNING_LEVELS.warn.min) return { level: "warn", message: WARNING_LEVELS.warn.message };
-  if (p >= WARNING_LEVELS.info.min) return { level: "info", message: WARNING_LEVELS.info.message };
+  if (p >= WARNING_LEVELS.critical.min)
+    return { level: "critical", message: WARNING_LEVELS.critical.message };
+  if (p >= WARNING_LEVELS.warn.min)
+    return { level: "warn", message: WARNING_LEVELS.warn.message };
+  if (p >= WARNING_LEVELS.info.min)
+    return { level: "info", message: WARNING_LEVELS.info.message };
   return { level: "ok", message: "" };
 }
 
 export function formatContextUsage(usedTokens, limitTokens, percent = null) {
   const used = Math.max(0, Math.round(Number(usedTokens) || 0));
-  const limit = Math.max(1, Math.round(Number(limitTokens) || DEFAULT_CONTEXT_LIMIT));
-  const pct = percent == null ? clampPercent((used / limit) * 100) : clampPercent(percent);
+  const limit = Math.max(
+    1,
+    Math.round(Number(limitTokens) || DEFAULT_CONTEXT_LIMIT),
+  );
+  const pct =
+    percent == null
+      ? clampPercent((used / limit) * 100)
+      : clampPercent(percent);
   return `${formatTokenCount(used)}/${formatTokenCount(limit)} (${pct}%)`;
 }
 
 export function readContextMonitorSnapshot() {
-  return readJsonMigrate(CONTEXT_MONITOR_CACHE_PATH, CONTEXT_MONITOR_LEGACY_PATH, null);
+  return readJsonMigrate(
+    CONTEXT_MONITOR_CACHE_PATH,
+    CONTEXT_MONITOR_LEGACY_PATH,
+    null,
+  );
 }
 
 function getStdinContextUsage(stdin) {
@@ -194,9 +212,10 @@ function getStdinContextUsage(stdin) {
   const nativePercent = Number(stdin?.context_window?.used_percentage);
   const usage = stdin?.context_window?.current_usage || {};
   const explicitUsed = Number(usage.total_tokens || 0);
-  const calculatedUsed = Number(usage.input_tokens || 0)
-    + Number(usage.cache_creation_input_tokens || 0)
-    + Number(usage.cache_read_input_tokens || 0);
+  const calculatedUsed =
+    Number(usage.input_tokens || 0) +
+    Number(usage.cache_creation_input_tokens || 0) +
+    Number(usage.cache_read_input_tokens || 0);
   const usedTokens = explicitUsed > 0 ? explicitUsed : calculatedUsed;
 
   if (limitTokens > 0 && usedTokens > 0) {
@@ -225,12 +244,11 @@ export function buildContextUsageView(stdin, snapshot = null) {
   const monitor = snapshot || readContextMonitorSnapshot();
   const fallbackLimit = Number(monitor?.limitTokens || DEFAULT_CONTEXT_LIMIT);
 
-  const usedTokens = stdinUsage?.usedTokens
-    ?? Number(monitor?.usedTokens || 0);
-  const limitTokens = stdinUsage?.limitTokens
-    ?? Math.max(1, fallbackLimit);
-  const percent = stdinUsage?.percent
-    ?? (limitTokens > 0 ? clampPercent((usedTokens / limitTokens) * 100) : 0);
+  const usedTokens = stdinUsage?.usedTokens ?? Number(monitor?.usedTokens || 0);
+  const limitTokens = stdinUsage?.limitTokens ?? Math.max(1, fallbackLimit);
+  const percent =
+    stdinUsage?.percent ??
+    (limitTokens > 0 ? clampPercent((usedTokens / limitTokens) * 100) : 0);
 
   const warning = classifyContextThreshold(percent);
   return {
@@ -240,10 +258,14 @@ export function buildContextUsageView(stdin, snapshot = null) {
     display: formatContextUsage(usedTokens, limitTokens, percent),
     warningLevel: warning.level,
     warningMessage: warning.message,
-    warningTag: warning.level === "warn" ? "⚠ 압축 권장"
-      : warning.level === "critical" ? "‼ 분할 권장"
-      : warning.level === "info" ? "ℹ 절반 이상 사용"
-      : "",
+    warningTag:
+      warning.level === "warn"
+        ? "⚠ 압축 권장"
+        : warning.level === "critical"
+          ? "‼ 분할 권장"
+          : warning.level === "info"
+            ? "ℹ 절반 이상 사용"
+            : "",
     source: stdinUsage?.source || (monitor ? "monitor" : "none"),
   };
 }
@@ -295,7 +317,10 @@ export function createContextMonitor(options = {}) {
     const percent = clampPercent((state.usedTokens / state.limitTokens) * 100);
     const warning = classifyContextThreshold(percent);
     const ts = new Date().toISOString().replace(/[:.]/g, "-");
-    const reportPath = join(logsDir, `context-usage-${state.sessionId}-${ts}.json`);
+    const reportPath = join(
+      logsDir,
+      `context-usage-${state.sessionId}-${ts}.json`,
+    );
     safeWriteJson(reportPath, {
       sessionId: state.sessionId,
       reason,
@@ -329,17 +354,31 @@ export function createContextMonitor(options = {}) {
     toolName = "",
   } = {}) => {
     const started = process.hrtime.bigint();
-    const reqObj = typeof requestBody === "object" ? requestBody : safeJsonParse(String(requestBody || ""));
-    const resObj = typeof responseBody === "object" ? responseBody : safeJsonParse(String(responseBody || ""));
+    const reqObj =
+      typeof requestBody === "object"
+        ? requestBody
+        : safeJsonParse(String(requestBody || ""));
+    const resObj =
+      typeof responseBody === "object"
+        ? responseBody
+        : safeJsonParse(String(responseBody || ""));
 
     const usage = parseUsageFromPayload(resObj);
-    const requestTokens = requestBytes > 0 ? toTokenEstimate(requestBytes) : toTokenEstimate(requestBody);
-    const responseTokens = usage?.total ?? (responseBytes > 0 ? toTokenEstimate(responseBytes) : toTokenEstimate(responseBody));
+    const requestTokens =
+      requestBytes > 0
+        ? toTokenEstimate(requestBytes)
+        : toTokenEstimate(requestBody);
+    const responseTokens =
+      usage?.total ??
+      (responseBytes > 0
+        ? toTokenEstimate(responseBytes)
+        : toTokenEstimate(responseBody));
     const totalTokens = Math.max(0, requestTokens + responseTokens);
 
     const method = reqObj?.method || reqObj?.params?.name || "";
     const name = toolName || reqObj?.params?.name || reqObj?.tool || "";
-    const args = reqObj?.params?.arguments || reqObj?.arguments || reqObj?.params || {};
+    const args =
+      reqObj?.params?.arguments || reqObj?.arguments || reqObj?.params || {};
     const payloadText = normalizeText(requestBody).slice(0, MAX_CAPTURE_BYTES);
     const skills = detectSkillHints(payloadText);
     const files = extractFileKeys(args);
@@ -383,7 +422,9 @@ export function createContextMonitor(options = {}) {
 
   if (registerExitHooks) {
     const flushOnExit = () => {
-      try { flush("process.exit"); } catch {}
+      try {
+        flush("process.exit");
+      } catch {}
     };
     process.once("exit", flushOnExit);
     process.once("SIGINT", flushOnExit);