@triflux/core 10.50.2 → 10.51.0

package/hub/lib/process-utils.mjs

@@ -488,68 +488,6 @@ function hasExactGbrainServe(commandLine) {
   );
 }
 
-function hasGbrainServeCommand(commandLine) {
-  return /(^|\s)gbrain\s+serve(\s|$)/i.test(commandLine.trim());
-}
-
-function hasFsmonitorDaemonCommand(commandLine) {
-  return FSMONITOR_DAEMON_PATTERN.test(commandLine);
-}
-
-function categoryForShardProcess(proc, context) {
-  const name = normalizeName(proc.name);
-  const commandLine = normalizeCommandLine(proc.commandLine);
-  const worktreePath = normalizeCommandLine(context.worktreePath || "");
-  const shardMarker = context.shardName
-    ? `.codex-swarm\\wt-${context.shardName}`
-    : "";
-  const hasWorktree = worktreePath && commandLine.includes(worktreePath);
-  const hasShardMarker = shardMarker && commandLine.includes(shardMarker);
-
-  if (name === "node.exe" || name === "node") {
-    if (hasWorktree || hasShardMarker) return "node";
-  }
-
-  if (name === "bash.exe" || name === "bash" || name === "sh") {
-    if (hasWorktree) return "bash";
-  }
-
-  if (name === "conhost.exe" || name === "conhost") {
-    if (
-      context.sessionIds?.length > 0 &&
-      context.sessionIds.some((id) => id && commandLine.includes(id))
-    ) {
-      return "conhost";
-    }
-  }
-
-  if (name === "bun.exe" || name === "bun") {
-    if ((hasWorktree || hasShardMarker) && hasGbrainServeCommand(commandLine)) {
-      return "bun";
-    }
-  }
-
-  if (name === "git.exe" || name === "git") {
-    if (
-      (hasWorktree || hasShardMarker) &&
-      hasFsmonitorDaemonCommand(commandLine)
-    ) {
-      return "git";
-    }
-  }
-
-  return null;
-}
-
-function killPid(
-  pid,
-  { killFn = process.kill, protectedPids = new Set() } = {},
-) {
-  if (protectedPids.has(pid)) return false;
-  killFn(pid, "SIGKILL");
-  return true;
-}
-
 /**
  * Return a process tree rooted at `rootPid`.
  *
@@ -730,67 +668,6 @@ export function findProcessesByCommandLine(
     });
 }
 
-/**
- * Cleanup processes associated with a swarm shard.
- *
- * Sequence: kill known top PID snapshots, scan scoped command lines, skip
- * protected PIDs, and kill only narrowly gated runtime categories.
- *
- * @param {{worktreePath?: string, sessionIds?: string[], topPids?: Array<number | {pid: number}>, runId?: string, shardName?: string, dryRun?: boolean, isWindows?: boolean, spawnSyncFn?: typeof spawnSync, killFn?: typeof process.kill, protectedPids?: Set<number>}} opts
- * @returns {{killed: number, scanned: number, skipped: number, byCategory: {node: number, bash: number, conhost: number, bun: number, git: number}}}
- */
-export function cleanupShardProcesses({
-  worktreePath,
-  sessionIds = [],
-  topPids = [],
-  runId,
-  shardName,
-  dryRun = false,
-  isWindows = IS_WINDOWS,
-  spawnSyncFn = spawnSync,
-  killFn = process.kill,
-  protectedPids,
-} = {}) {
-  const protectedSet = getProtectedPids({
-    protectedPids,
-    isWindows,
-    spawnSyncFn,
-    includeAncestorScan: true,
-  });
-  const byCategory = { node: 0, bash: 0, conhost: 0, bun: 0, git: 0 };
-  let killed = 0;
-  let skipped = 0;
-
-  if (!dryRun && topPids.length > 0) {
-    killed += killProcessTreeSnapshot(topPids, {
-      killFn,
-      protectedPids: protectedSet,
-    }).killed;
-  }
-
-  const scanned = getProcessSnapshot({ isWindows, spawnSyncFn });
-
-  for (const proc of scanned) {
-    const category = categoryForShardProcess(proc, {
-      worktreePath,
-      sessionIds,
-      shardName,
-    });
-    if (!category) continue;
-    byCategory[category]++;
-    if (protectedSet.has(proc.pid)) {
-      skipped++;
-      continue;
-    }
-    if (dryRun) continue;
-    try {
-      if (killPid(proc.pid, { killFn, protectedPids: protectedSet })) killed++;
-    } catch {}
-  }
-
-  return { killed, scanned: scanned.length, skipped, byCategory };
-}
-
 /**
  * Cleanup narrowly gated orphan runtime processes.
  *

package/hub/lib/spawn-trace.mjs

@@ -7,13 +7,14 @@ import { join } from "node:path";
 const LOG_DIR = join(homedir(), ".triflux", "logs");
 const DEDUPE_WINDOW_MS = 5_000;
 const RATE_WINDOW_MS = 1_000;
+const DEFAULT_MAX_SPAWN_PER_SEC = 100;
 // multi-worker headless dispatch 가 1 초 안에 ~25-30+ psmux/tmux 명령을 호출한다.
 // 2-worker dispatchBatch 가 default 30 limit 을 초과해 `rate_limit` throw 로 mac
 // 호환성 회귀 (smoke test 발견, 2026-05-15). 100 으로 상향 — 폭주 안전망은
 // 여전히 (default 100/sec) 유지, env override 로 보수 설정 가능.
 export let MAX_SPAWN_PER_SEC = resolvePositiveInteger(
   process.env.TRIFLUX_MAX_SPAWN_RATE,
-  100,
+  DEFAULT_MAX_SPAWN_PER_SEC,
 );
 export const MAX_TOTAL_DESCENDANTS = resolvePositiveInteger(
   process.env.TRIFLUX_MAX_DESCENDANTS,
@@ -160,7 +161,7 @@ export function getMaxSpawnPerSec() {
 export function reload() {
   MAX_SPAWN_PER_SEC = resolvePositiveInteger(
     process.env.TRIFLUX_MAX_SPAWN_RATE,
-    30,
+    DEFAULT_MAX_SPAWN_PER_SEC,
   );
   return getMaxSpawnPerSec();
 }

package/hub/lib/timeout-defaults.mjs

@@ -5,7 +5,6 @@ export const MAX_DURATION_MS = 86_400_000;
 export const DEFAULT_ASSIGN_TIMEOUT_MS = 600_000;
 export const DEFAULT_ASSIGN_TTL_MS = 600_000;
 export const DEFAULT_HANDOFF_TTL_MS = 600_000;
-export const DEFAULT_SWARM_LOCK_TTL_MS = 600_000;
 export const DEFAULT_REGISTER_TIMEOUT_SEC = 600;
 export const REGISTER_HEARTBEAT_GRACE_MS = 120_000;
 export const DEFAULT_STALL_INTERVENTION_SEC = 1_200;

package/hub/role-contract.mjs

@@ -216,8 +216,6 @@ export function reduceRoleReachabilityTransition(state, trigger) {
     : { state: transition.to, action: transition.action };
 }
 
-export const reduceRoleReachability = reduceRoleReachabilityTransition;
-
 function snapshotLegacyDetails(snapshot) {
   const legacy = isRecord(snapshot.legacy) ? snapshot.legacy : {};
   return {

package/hud/cli-policy.mjs

@@ -41,25 +41,3 @@ export function resolveHudCliVisibility(env = process.env, opts = {}) {
     return { showCodex: true, antigravityAllowed: true, policy: null };
   }
 }
-
-/**
- * antigravity / gemini 슬롯 행을 그릴지 판정한다.
- *
- * antigravity 가 허용이면 기존 동작 그대로 항상 그린다. 정책으로 꺼졌을 때만
- * gemini 폴백 정보가 하나라도 있는지 보고, 전부 비어 있으면 행을 생략한다.
- *
- * @param {{
- *   antigravityAllowed?: boolean,
- *   antigravityReady?: boolean,
- *   geminiEmail?: unknown,
- * }} [args]
- * @returns {boolean}
- */
-export function shouldRenderGeminiFallbackRow({
-  antigravityAllowed = true,
-  antigravityReady = false,
-  geminiEmail = null,
-} = {}) {
-  if (antigravityAllowed || antigravityReady) return true;
-  return Boolean(geminiEmail);
-}

package/hud/constants.mjs

@@ -31,9 +31,6 @@ export const CLAUDE_BAND_MARKER_DIR = join(
 );
 export const CLAUDE_BAND_MARKER_TTL_MS = 12 * 60 * 60 * 1000;
 
-// tfx-multi 세션 상태 디렉터리
-export const TEAM_STATE_DIR =
-  process.env.TFX_HUB_PID_DIR || join(homedir(), ".claude", "cache", "tfx-hub");
 export const CONTEXT_MONITOR_CACHE_PATH = join(
   homedir(),
   ".claude",
@@ -149,11 +146,9 @@ export const CODEX_REFRESH_LOCK_PATH = join(
 );
 export const SPAWN_LOCK_TTL_MS = 30 * 1000; // 30초 spawn dedup
 
-export const GEMINI_OAUTH_PATH = join(homedir(), ".gemini", "oauth_creds.json");
 export const ANTIGRAVITY_OAUTH_PATHS = [
   join(homedir(), ".gemini", "antigravity-cli", "oauth_creds.json"),
   join(homedir(), ".gemini", "antigravity-cli", "credentials.json"),
-  GEMINI_OAUTH_PATH,
 ];
 export const ANTIGRAVITY_KEYCHAIN_SERVICE = "gemini";
 export const ANTIGRAVITY_KEYCHAIN_ACCOUNT = "antigravity";
@@ -163,33 +158,23 @@ export const ANTIGRAVITY_SETTINGS_PATH = join(
   "antigravity-cli",
   "settings.json",
 );
-// Antigravity CLI 현재 장착 모델 라벨 → HUD 2자 약어 매핑
-export const ANTIGRAVITY_MODEL_ABBREV = {
-  // Flash 세대와 등급을 구분한다.
-  "Gemini 3.8 Flash (High)": "8h",
-  "Gemini 3.8 Flash (Medium)": "8m",
-  "Gemini 3.8 Flash (Low)": "8l",
-  "Gemini 3.7 Flash (High)": "7h",
-  "Gemini 3.7 Flash (Medium)": "7m",
-  "Gemini 3.7 Flash (Low)": "7l",
-  "Gemini 3.6 Flash (High)": "6h",
-  "Gemini 3.6 Flash (Medium)": "6m",
-  "Gemini 3.6 Flash (Low)": "6l",
-  "Gemini 3.5 Flash (High)": "Fh",
-  "Gemini 3.5 Flash (Medium)": "Fm",
-  "Gemini 3.5 Flash (Low)": "Fl",
-};
-export const TFX_PREFLIGHT_CACHE_PATH = join(
+export const ANTIGRAVITY_QUOTA_CACHE_PATH = join(
   homedir(),
   ".claude",
   "cache",
-  "tfx-preflight.json",
+  "antigravity-quota-cache.json",
 );
-export const TFX_PREFLIGHT_CACHE_STALE_MS = 60 * 60 * 1000;
-
+export const ANTIGRAVITY_REFRESH_LOCK_PATH = join(
+  homedir(),
+  ".claude",
+  "cache",
+  ".antigravity-refresh-lock",
+);
+export const ANTIGRAVITY_QUOTA_STALE_MS = 5 * 60 * 1000;
+export const ANTIGRAVITY_REFRESH_FLAG = "--refresh-antigravity-quota";
 export const ACCOUNT_LABEL_WIDTH = 10;
 export const PROVIDER_PREFIX_WIDTH = 2;
-export const PERCENT_CELL_WIDTH = 3;
+export const PERCENT_CELL_WIDTH = 4;
 export const TIME_CELL_INNER_WIDTH = 6;
 
 export const CLAUDE_REFRESH_FLAG = "--refresh-claude-usage";

package/hud/hud-qos-status.mjs

@@ -3,10 +3,7 @@
 import { existsSync, statSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
-import {
-  resolveHudCliVisibility,
-  shouldRenderGeminiFallbackRow,
-} from "./cli-policy.mjs";
+import { resolveHudCliVisibility } from "./cli-policy.mjs";
 import {
   bold,
   claudeOrange,
@@ -19,12 +16,11 @@ import {
 import {
   ACCOUNTS_CONFIG_PATH,
   ACCOUNTS_STATE_PATH,
+  ANTIGRAVITY_REFRESH_FLAG,
   CLAUDE_BAND_MARKER_DIR,
   CLAUDE_BAND_MARKER_TTL_MS,
   CLAUDE_REFRESH_FLAG,
   CODEX_REFRESH_FLAG,
-  TFX_PREFLIGHT_CACHE_PATH,
-  TFX_PREFLIGHT_CACHE_STALE_MS,
 } from "./constants.mjs";
 import { buildContextUsageView } from "./context-monitor.mjs";
 import {
@@ -41,15 +37,14 @@ import {
 } from "./providers/codex.mjs";
 import {
   getAntigravityAccountLabel,
-  getAntigravityCurrentModel,
-  getAntigravityModelAbbrev,
-  getGeminiEmail,
+  readAntigravityQuotaSnapshot,
+  refreshAntigravityQuotaCache,
+  scheduleAntigravityQuotaRefresh,
 } from "./providers/gemini.mjs";
 import {
   getClaudeRows,
   getMicroLine,
   getProviderRow,
-  getTeamRow,
   renderAlignedRows,
 } from "./renderers.mjs";
 import { selectTier } from "./terminal.mjs";
@@ -64,22 +59,21 @@ async function main() {
     await refreshCodexRateLimitsCache();
     return;
   }
+  if (process.argv.includes(ANTIGRAVITY_REFRESH_FLAG)) {
+    refreshAntigravityQuotaCache();
+    return;
+  }
 
   const stdinPromise = readStdinJson();
-  const preflightCache = readJson(TFX_PREFLIGHT_CACHE_PATH, null);
-  const preflightTimestamp = Number(preflightCache?.timestamp);
-  const preflightFresh =
-    Number.isFinite(preflightTimestamp) &&
-    Date.now() - preflightTimestamp <= TFX_PREFLIGHT_CACHE_STALE_MS;
   const { showCodex, antigravityAllowed } = resolveHudCliVisibility();
-  const antigravityReady =
-    antigravityAllowed &&
-    preflightFresh &&
-    preflightCache?.antigravity?.ok === true;
   const accountsConfig = readJson(ACCOUNTS_CONFIG_PATH, { providers: {} });
   const accountsState = readJson(ACCOUNTS_STATE_PATH, { providers: {} });
   const claudeUsageSnapshot = readClaudeUsageSnapshot();
   const codexSnapshot = readCodexRateLimitSnapshot();
+  const antigravitySnapshot = antigravityAllowed
+    ? readAntigravityQuotaSnapshot()
+    : null;
+  if (antigravitySnapshot?.shouldRefresh) scheduleAntigravityQuotaRefresh();
   // 설정이 없는 홈에서는 갱신 프로세스를 시작하지 않는다.
   if (
     claudeUsageSnapshot.shouldRefresh &&
@@ -102,21 +96,12 @@ async function main() {
     : null;
   const codexBuckets = codexSnapshot.buckets;
   const currentTier = selectTier();
-  const geminiEmail = getGeminiEmail();
-  const showGeminiRow = shouldRenderGeminiFallbackRow({
-    antigravityAllowed,
-    antigravityReady,
-    geminiEmail,
-  });
-
   if (currentTier === "nano") {
-    const microLine = getMicroLine(
-      contextView,
-      claudeUsage,
-      codexBuckets,
-      antigravityReady,
-      { showCodex, showGemini: showGeminiRow },
-    );
+    const microLine = getMicroLine(contextView, claudeUsage, codexBuckets, {
+      showCodex,
+      showAntigravity: antigravityAllowed,
+      antigravityQuota: antigravitySnapshot?.data,
+    });
     process.stdout.write(`\x1b[0m${microLine}\n`);
     return;
   }
@@ -140,38 +125,25 @@ async function main() {
       ),
     );
   }
-  let geminiRowIndex = -1;
-  if (showGeminiRow) {
-    geminiRowIndex = rows.length;
+  if (antigravityAllowed) {
     rows.push(
       getProviderRow(
         currentTier,
-        antigravityReady ? "antigravity" : "gemini",
-        antigravityReady ? "a" : "g",
+        "antigravity",
+        "a",
         geminiBlue,
         accountsConfig,
         accountsState,
-        antigravityReady
-          ? {
-              currentAbbrev: getAntigravityModelAbbrev(
-                getAntigravityCurrentModel(),
-              ),
-            }
-          : null,
-        antigravityReady ? getAntigravityAccountLabel() : geminiEmail,
+        antigravitySnapshot?.data,
+        getAntigravityAccountLabel(),
       ),
     );
   }
-  const teamRow = getTeamRow(currentTier);
-  if (teamRow) rows.push(teamRow);
 
   const outputLines = renderAlignedRows(rows);
   if (!codexBuckets && outputLines[codexRowIndex] != null) {
     outputLines[codexRowIndex] = `${DIM}${outputLines[codexRowIndex]}${RESET}`;
   }
-  if (!antigravityReady && outputLines[geminiRowIndex] != null) {
-    outputLines[geminiRowIndex] = dim(outputLines[geminiRowIndex]);
-  }
   // 알림 배너와 TUI 스타일이 HUD 내용에 겹치지 않도록 한다.
   const leadingBreaks = contextView.percent >= 85 ? "\n\n" : "\n";
   const resetLines = outputLines.map((line) => `\x1b[0m${line}`);

package/hud/providers/gemini.mjs

@@ -1,10 +1,15 @@
+import { execFileSync, spawn } from "node:child_process";
+import { existsSync } from "node:fs";
 import {
-  ANTIGRAVITY_MODEL_ABBREV,
   ANTIGRAVITY_OAUTH_PATHS,
+  ANTIGRAVITY_QUOTA_CACHE_PATH,
+  ANTIGRAVITY_QUOTA_STALE_MS,
+  ANTIGRAVITY_REFRESH_FLAG,
+  ANTIGRAVITY_REFRESH_LOCK_PATH,
   ANTIGRAVITY_SETTINGS_PATH,
-  GEMINI_OAUTH_PATH,
+  SPAWN_LOCK_TTL_MS,
 } from "../constants.mjs";
-import { decodeJwtEmail, readJson } from "../utils.mjs";
+import { decodeJwtEmail, readJson, writeJsonSafe } from "../utils.mjs";
 
 function getCredentialEmail(credential) {
   return (
@@ -17,10 +22,6 @@ function getCredentialEmail(credential) {
   );
 }
 
-export function getGeminiEmail() {
-  return getCredentialEmail(readJson(GEMINI_OAUTH_PATH, null));
-}
-
 // GCP 프로젝트 인증이면 계정 이메일 대신 프로젝트 ID를 보여 준다.
 export function getAntigravityAccountLabel() {
   const project = readJson(ANTIGRAVITY_SETTINGS_PATH, null)?.gcp?.project;
@@ -36,20 +37,104 @@ export function getAntigravityAccountLabel() {
   return null;
 }
 
-export function getAntigravityCurrentModel() {
-  const settings = readJson(ANTIGRAVITY_SETTINGS_PATH, null);
-  return settings?.model || null;
+function isGcpProjectAuth() {
+  return Boolean(readJson(ANTIGRAVITY_SETTINGS_PATH, null)?.gcp?.project);
+}
+
+export function readAntigravityQuotaSnapshot() {
+  // 프로젝트 과금의 기본 응답을 개인 플랜의 0% 사용률로 표시하지 않는다.
+  if (isGcpProjectAuth()) return { data: null, shouldRefresh: false };
+  const cache = readJson(ANTIGRAVITY_QUOTA_CACHE_PATH, null);
+  const matched = cache?.accountLabel === getAntigravityAccountLabel();
+  const stale =
+    !cache?.timestamp ||
+    Date.now() - cache.timestamp >= ANTIGRAVITY_QUOTA_STALE_MS;
+  const model = readJson(ANTIGRAVITY_SETTINGS_PATH, null)?.model;
+  const buckets = matched && Array.isArray(cache?.buckets) ? cache.buckets : [];
+  const bucket = buckets
+    .filter(
+      (item) =>
+        model &&
+        item &&
+        (item?.name === model || item?.id === model) &&
+        Number.isFinite(item.remaining_fraction) &&
+        item.remaining_fraction >= 0 &&
+        item.remaining_fraction <= 1 &&
+        Date.parse(item.reset_time) > Date.now(),
+    )
+    .sort((a, b) => a.remaining_fraction - b.remaining_fraction)[0];
+  return {
+    data: bucket
+      ? {
+          usedPercent: Math.round((1 - bucket.remaining_fraction) * 100),
+          resetTime: bucket.reset_time,
+          stale,
+        }
+      : null,
+    shouldRefresh: !matched || stale,
+  };
+}
+
+// /usage는 모델 턴을 시작하지 않는 공식 print-mode 조회다.
+export function refreshAntigravityQuotaCache(execFileSyncFn = execFileSync) {
+  if (isGcpProjectAuth()) return null;
+  const accountLabel = getAntigravityAccountLabel();
+  let buckets = null;
+  try {
+    const result = JSON.parse(
+      execFileSyncFn("agy", ["-p", "/usage", "--output-format", "json"], {
+        encoding: "utf8",
+        timeout: 15_000,
+        maxBuffer: 1024 * 1024,
+        stdio: ["ignore", "pipe", "ignore"],
+        windowsHide: true,
+      }),
+    );
+    if (
+      result.status === "SUCCESS" &&
+      result.command?.name === "usage" &&
+      Array.isArray(result.command.data?.groups)
+    ) {
+      buckets = result.command.data.groups.flatMap((group) =>
+        Array.isArray(group?.buckets) ? group.buckets : [],
+      );
+    }
+  } catch {
+    // 실패도 캐시해 인증 오류나 CLI 부재 시 매 렌더마다 실행하지 않는다.
+  }
+  if (accountLabel === getAntigravityAccountLabel()) {
+    writeJsonSafe(ANTIGRAVITY_QUOTA_CACHE_PATH, {
+      timestamp: Date.now(),
+      accountLabel,
+      buckets,
+    });
+  }
+  return buckets;
 }
 
-export function getAntigravityModelAbbrev(label) {
-  if (!label) return "??";
-  const known = ANTIGRAVITY_MODEL_ABBREV[label];
-  if (known) return known;
-  const parens = label.match(/\(([^)]+)\)/);
-  const head = label.match(/^(\w+)/);
-  if (head && parens) {
-    return `${head[1][0].toUpperCase()}${parens[1][0].toLowerCase()}`;
+export function scheduleAntigravityQuotaRefresh() {
+  const scriptPath = process.argv[1];
+  if (
+    !scriptPath ||
+    !existsSync(ANTIGRAVITY_SETTINGS_PATH) ||
+    isGcpProjectAuth()
+  )
+    return;
+  const lock = readJson(ANTIGRAVITY_REFRESH_LOCK_PATH, null);
+  if (Date.now() - Number(lock?.t || 0) < SPAWN_LOCK_TTL_MS) return;
+  writeJsonSafe(ANTIGRAVITY_REFRESH_LOCK_PATH, { t: Date.now() });
+  try {
+    const child = spawn(
+      process.execPath,
+      [scriptPath, ANTIGRAVITY_REFRESH_FLAG],
+      {
+        detached: true,
+        stdio: "ignore",
+        windowsHide: true,
+      },
+    );
+    child.unref();
+  } catch {
+    // 다음 렌더에서 다시 시도한다.
   }
-  if (head) return head[1].slice(0, 2);
-  return "??";
 }