triflux 10.3.4 → 10.7.0

package/hud/providers/gemini.mjs

@@ -1,31 +1,47 @@
 // ============================================================================
 // Gemini 쿼터 API / 세션 토큰 / RPM 트래커
 // ============================================================================
-import { existsSync, readFileSync, readdirSync } from "node:fs";
+
+import { spawn } from "node:child_process";
+import { existsSync, readdirSync, readFileSync } from "node:fs";
+import https from "node:https";
 import { homedir } from "node:os";
 import { join } from "node:path";
-import https from "node:https";
-import { spawn } from "node:child_process";
 import {
-  GEMINI_OAUTH_PATH, GEMINI_QUOTA_CACHE_PATH, GEMINI_PROJECT_CACHE_PATH,
-  GEMINI_SESSION_CACHE_PATH, GEMINI_RPM_TRACKER_PATH,
-  LEGACY_GEMINI_QUOTA_CACHE, LEGACY_GEMINI_PROJECT_CACHE,
-  LEGACY_GEMINI_SESSION_CACHE, LEGACY_GEMINI_RPM_TRACKER,
-  GEMINI_RPM_WINDOW_MS, GEMINI_QUOTA_STALE_MS, GEMINI_SESSION_STALE_MS,
   GEMINI_API_TIMEOUT_MS,
-  GEMINI_REFRESH_FLAG, GEMINI_SESSION_REFRESH_FLAG,
-  GEMINI_QUOTA_REFRESH_LOCK_PATH, GEMINI_SESSION_REFRESH_LOCK_PATH, SPAWN_LOCK_TTL_MS,
+  GEMINI_OAUTH_PATH,
+  GEMINI_PROJECT_CACHE_PATH,
+  GEMINI_QUOTA_CACHE_PATH,
+  GEMINI_QUOTA_REFRESH_LOCK_PATH,
+  GEMINI_QUOTA_STALE_MS,
+  GEMINI_REFRESH_FLAG,
+  GEMINI_RPM_TRACKER_PATH,
+  GEMINI_RPM_WINDOW_MS,
+  GEMINI_SESSION_CACHE_PATH,
+  GEMINI_SESSION_REFRESH_FLAG,
+  GEMINI_SESSION_REFRESH_LOCK_PATH,
+  GEMINI_SESSION_STALE_MS,
+  LEGACY_GEMINI_PROJECT_CACHE,
+  LEGACY_GEMINI_QUOTA_CACHE,
+  LEGACY_GEMINI_RPM_TRACKER,
+  LEGACY_GEMINI_SESSION_CACHE,
+  SPAWN_LOCK_TTL_MS,
 } from "../constants.mjs";
 import {
-  readJson, writeJsonSafe, readJsonMigrate, makeHash, clampPercent,
-  decodeJwtEmail, createHttpsPost,
+  clampPercent,
+  createHttpsPost,
+  decodeJwtEmail,
+  makeHash,
+  readJson,
+  readJsonMigrate,
+  writeJsonSafe,
 } from "../utils.mjs";
 
 const httpsPost = createHttpsPost(https, GEMINI_API_TIMEOUT_MS);
 
 // Gemini 모델별 RPM 한도 (실측 기반: Pro 25, Flash 300)
 export function getGeminiRpmLimit(model) {
-  if (model && model.includes("pro")) return 25;
+  if (model?.includes("pro")) return 25;
   return 300; // Flash 기본
 }
 
@@ -49,21 +65,36 @@ export function deriveGeminiLimits(bucket) {
   if (bucket.remainingAmount != null) {
     const remaining = parseInt(bucket.remainingAmount, 10);
     const limit = fraction > 0 ? Math.round(remaining / fraction) : 0;
-    return { usedPct, remaining, limit, resetTime: bucket.resetTime, modelId: bucket.modelId };
+    return {
+      usedPct,
+      remaining,
+      limit,
+      resetTime: bucket.resetTime,
+      modelId: bucket.modelId,
+    };
   }
-  return { usedPct, remaining: null, limit: null, resetTime: bucket.resetTime, modelId: bucket.modelId };
+  return {
+    usedPct,
+    remaining: null,
+    limit: null,
+    resetTime: bucket.resetTime,
+    modelId: bucket.modelId,
+  };
 }
 
 export function getGeminiEmail() {
   try {
     const oauth = readJson(GEMINI_OAUTH_PATH, null);
     return decodeJwtEmail(oauth?.id_token);
-  } catch { return null; }
+  } catch {
+    return null;
+  }
 }
 
 export function buildGeminiAuthContext(accountId) {
   const oauth = readJson(GEMINI_OAUTH_PATH, null);
-  const tokenSource = oauth?.refresh_token || oauth?.id_token || oauth?.access_token || "";
+  const tokenSource =
+    oauth?.refresh_token || oauth?.id_token || oauth?.access_token || "";
   const tokenFingerprint = tokenSource ? makeHash(tokenSource) : "none";
   const cacheKey = `${accountId || "gemini-main"}::${tokenFingerprint}`;
   return { oauth, tokenFingerprint, cacheKey };
@@ -78,11 +109,17 @@ export async function fetchGeminiQuota(accountId, options = {}) {
   const forceRefresh = options.forceRefresh === true;
 
   // 1. 캐시 확인 (계정/토큰별)
-  const cache = readJsonMigrate(GEMINI_QUOTA_CACHE_PATH, LEGACY_GEMINI_QUOTA_CACHE, null);
-  if (!forceRefresh
-    && cache?.cacheKey === cacheKey
-    && cache?.timestamp
-    && (Date.now() - cache.timestamp < GEMINI_QUOTA_STALE_MS)) {
+  const cache = readJsonMigrate(
+    GEMINI_QUOTA_CACHE_PATH,
+    LEGACY_GEMINI_QUOTA_CACHE,
+    null,
+  );
+  if (
+    !forceRefresh &&
+    cache?.cacheKey === cacheKey &&
+    cache?.timestamp &&
+    Date.now() - cache.timestamp < GEMINI_QUOTA_STALE_MS
+  ) {
     return cache;
   }
 
@@ -117,12 +154,22 @@ export async function fetchGeminiQuota(accountId, options = {}) {
       oauth.access_token,
     );
     const id = loadRes?.cloudaicompanionProject;
-    if (id) writeJsonSafe(GEMINI_PROJECT_CACHE_PATH, { cacheKey, projectId: id, timestamp: Date.now() });
+    if (id)
+      writeJsonSafe(GEMINI_PROJECT_CACHE_PATH, {
+        cacheKey,
+        projectId: id,
+        timestamp: Date.now(),
+      });
     return id || null;
   };
 
-  const projCache = readJsonMigrate(GEMINI_PROJECT_CACHE_PATH, LEGACY_GEMINI_PROJECT_CACHE, null);
-  let projectId = projCache?.cacheKey === cacheKey ? projCache?.projectId : null;
+  const projCache = readJsonMigrate(
+    GEMINI_PROJECT_CACHE_PATH,
+    LEGACY_GEMINI_PROJECT_CACHE,
+    null,
+  );
+  let projectId =
+    projCache?.cacheKey === cacheKey ? projCache?.projectId : null;
   if (!projectId) projectId = await fetchProjectId();
   if (!projectId) return cache;
 
@@ -146,7 +193,11 @@ export async function fetchGeminiQuota(accountId, options = {}) {
 
   if (!quotaRes?.buckets) {
     // API 응답에 buckets 없음: 에러 코드 또는 응답 내용을 캐시에 기록
-    const apiError = quotaRes?.error?.message || quotaRes?.error?.code || quotaRes?.error || "no buckets in response";
+    const apiError =
+      quotaRes?.error?.message ||
+      quotaRes?.error?.code ||
+      quotaRes?.error ||
+      "no buckets in response";
     writeJsonSafe(GEMINI_QUOTA_CACHE_PATH, {
       ...(cache || {}),
       timestamp: cache?.timestamp || Date.now(),
@@ -176,12 +227,17 @@ export async function fetchGeminiQuota(accountId, options = {}) {
 export function readGeminiRpm(model) {
   try {
     // 새 경로 → 레거시 경로 fallback
-    const rpmPath = existsSync(GEMINI_RPM_TRACKER_PATH) ? GEMINI_RPM_TRACKER_PATH
-      : existsSync(LEGACY_GEMINI_RPM_TRACKER) ? LEGACY_GEMINI_RPM_TRACKER : null;
+    const rpmPath = existsSync(GEMINI_RPM_TRACKER_PATH)
+      ? GEMINI_RPM_TRACKER_PATH
+      : existsSync(LEGACY_GEMINI_RPM_TRACKER)
+        ? LEGACY_GEMINI_RPM_TRACKER
+        : null;
     if (!rpmPath) return { count: 0, percent: 0, remainingSec: 60 };
     const raw = readFileSync(rpmPath, "utf-8");
     const parsed = JSON.parse(raw);
-    const timestamps = Array.isArray(parsed.timestamps) ? parsed.timestamps : [];
+    const timestamps = Array.isArray(parsed.timestamps)
+      ? parsed.timestamps
+      : [];
     const now = Date.now();
     const recent = timestamps.filter((t) => now - t < GEMINI_RPM_WINDOW_MS);
     const count = recent.length;
@@ -189,9 +245,15 @@ export function readGeminiRpm(model) {
     const percent = clampPercent(Math.round((count / rpmLimit) * 100));
     // 가장 오래된 엔트리가 윈도우에서 빠지기까지 남은 초 (0건이면 0s)
     // 5초 단위 반올림으로 HUD 깜빡임 감소
-    const rawRemainingSec = recent.length > 0
-      ? Math.max(0, Math.ceil((GEMINI_RPM_WINDOW_MS - (now - Math.min(...recent))) / 1000))
-      : 0;
+    const rawRemainingSec =
+      recent.length > 0
+        ? Math.max(
+            0,
+            Math.ceil(
+              (GEMINI_RPM_WINDOW_MS - (now - Math.min(...recent))) / 1000,
+            ),
+          )
+        : 0;
     const remainingSec = Math.ceil(rawRemainingSec / 5) * 5;
     return { count, percent, remainingSec };
   } catch {
@@ -200,7 +262,11 @@ export function readGeminiRpm(model) {
 }
 
 export function readGeminiQuotaSnapshot(accountId, authContext) {
-  const cache = readJsonMigrate(GEMINI_QUOTA_CACHE_PATH, LEGACY_GEMINI_QUOTA_CACHE, null);
+  const cache = readJsonMigrate(
+    GEMINI_QUOTA_CACHE_PATH,
+    LEGACY_GEMINI_QUOTA_CACHE,
+    null,
+  );
   if (!cache?.buckets) {
     return { quota: null, shouldRefresh: true };
   }
@@ -209,19 +275,24 @@ export function readGeminiQuotaSnapshot(accountId, authContext) {
   const isLegacyCache = !cache.cacheKey;
   const keyMatched = cache.cacheKey === cacheKey;
   const cacheTs = Number(cache.timestamp);
-  const ageMs = Number.isFinite(cacheTs) ? Date.now() - cacheTs : Number.MAX_SAFE_INTEGER;
+  const ageMs = Number.isFinite(cacheTs)
+    ? Date.now() - cacheTs
+    : Number.MAX_SAFE_INTEGER;
   const isFresh = ageMs < GEMINI_QUOTA_STALE_MS;
 
   if (keyMatched) {
     // resetTime이 지난 버킷의 remainingFraction을 1로 보정 (stale 캐시 방지)
     if (Array.isArray(cache.buckets)) {
       const now = Date.now();
-      const patchedBuckets = cache.buckets.map(b =>
+      const patchedBuckets = cache.buckets.map((b) =>
         b?.resetTime && new Date(b.resetTime).getTime() <= now
           ? { ...b, remainingFraction: 1 }
-          : b
+          : b,
       );
-      return { quota: { ...cache, buckets: patchedBuckets }, shouldRefresh: !isFresh };
+      return {
+        quota: { ...cache, buckets: patchedBuckets },
+        shouldRefresh: !isFresh,
+      };
     }
     return { quota: cache, shouldRefresh: !isFresh };
   }
@@ -238,16 +309,24 @@ export function scheduleGeminiQuotaRefresh(accountId) {
   // 스폰 락: 30초 내 이미 스폰했으면 중복 방지
   try {
     if (existsSync(GEMINI_QUOTA_REFRESH_LOCK_PATH)) {
-      const lockAge = Date.now() - readJson(GEMINI_QUOTA_REFRESH_LOCK_PATH, {}).t;
+      const lockAge =
+        Date.now() - readJson(GEMINI_QUOTA_REFRESH_LOCK_PATH, {}).t;
       if (lockAge < SPAWN_LOCK_TTL_MS) return;
     }
     writeJsonSafe(GEMINI_QUOTA_REFRESH_LOCK_PATH, { t: Date.now() });
-  } catch { /* 락 실패 무시 — 스폰 진행 */ }
+  } catch {
+    /* 락 실패 무시 — 스폰 진행 */
+  }
 
   try {
     const child = spawn(
       process.execPath,
-      [scriptPath, GEMINI_REFRESH_FLAG, "--account", accountId || "gemini-main"],
+      [
+        scriptPath,
+        GEMINI_REFRESH_FLAG,
+        "--account",
+        accountId || "gemini-main",
+      ],
       {
         detached: true,
         stdio: "ignore",
@@ -266,7 +345,11 @@ export function scheduleGeminiQuotaRefresh(accountId) {
 }
 
 export function readGeminiSessionSnapshot() {
-  const cache = readJsonMigrate(GEMINI_SESSION_CACHE_PATH, LEGACY_GEMINI_SESSION_CACHE, null);
+  const cache = readJsonMigrate(
+    GEMINI_SESSION_CACHE_PATH,
+    LEGACY_GEMINI_SESSION_CACHE,
+    null,
+  );
   if (!cache?.session) {
     return { session: null, shouldRefresh: true };
   }
@@ -290,20 +373,29 @@ export function scheduleGeminiSessionRefresh() {
   // 스폰 락: 30초 내 이미 스폰했으면 중복 방지
   try {
     if (existsSync(GEMINI_SESSION_REFRESH_LOCK_PATH)) {
-      const lockAge = Date.now() - readJson(GEMINI_SESSION_REFRESH_LOCK_PATH, {}).t;
+      const lockAge =
+        Date.now() - readJson(GEMINI_SESSION_REFRESH_LOCK_PATH, {}).t;
       if (lockAge < SPAWN_LOCK_TTL_MS) return;
     }
     writeJsonSafe(GEMINI_SESSION_REFRESH_LOCK_PATH, { t: Date.now() });
-  } catch { /* 락 실패 무시 — 스폰 진행 */ }
+  } catch {
+    /* 락 실패 무시 — 스폰 진행 */
+  }
 
   try {
-    const child = spawn(process.execPath, [scriptPath, GEMINI_SESSION_REFRESH_FLAG], {
-      detached: true,
-      stdio: "ignore",
-      windowsHide: true,
-    });
+    const child = spawn(
+      process.execPath,
+      [scriptPath, GEMINI_SESSION_REFRESH_FLAG],
+      {
+        detached: true,
+        stdio: "ignore",
+        windowsHide: true,
+      },
+    );
     child.unref();
-  } catch { /* 백그라운드 실행 실패 무시 */ }
+  } catch {
+    /* 백그라운드 실행 실패 무시 */
+  }
 }
 
 // ============================================================================
@@ -315,27 +407,47 @@ export function scanGeminiSessionTokens() {
   let best = null;
   let bestTime = 0;
   try {
-    const dirs = readdirSync(tmpDir).filter((d) => existsSync(join(tmpDir, d, "chats")));
+    const dirs = readdirSync(tmpDir).filter((d) =>
+      existsSync(join(tmpDir, d, "chats")),
+    );
     for (const dir of dirs) {
       const chatsDir = join(tmpDir, dir, "chats");
       let files;
-      try { files = readdirSync(chatsDir).filter((f) => f.endsWith(".json")); } catch { continue; }
+      try {
+        files = readdirSync(chatsDir).filter((f) => f.endsWith(".json"));
+      } catch {
+        continue;
+      }
       for (const file of files) {
         try {
           const data = JSON.parse(readFileSync(join(chatsDir, file), "utf-8"));
           const updatedAt = new Date(data.lastUpdated || 0).getTime();
           if (updatedAt <= bestTime) continue;
-          let input = 0, output = 0;
+          let input = 0,
+            output = 0;
           let model = "unknown";
           for (const msg of data.messages || []) {
-            if (msg.tokens) { input += msg.tokens.input || 0; output += msg.tokens.output || 0; }
+            if (msg.tokens) {
+              input += msg.tokens.input || 0;
+              output += msg.tokens.output || 0;
+            }
             if (msg.model) model = msg.model;
           }
           bestTime = updatedAt;
-          best = { input, output, total: input + output, model, lastUpdated: data.lastUpdated };
-        } catch { /* 무시 */ }
+          best = {
+            input,
+            output,
+            total: input + output,
+            model,
+            lastUpdated: data.lastUpdated,
+          };
+        } catch {
+          /* 무시 */
+        }
       }
     }
-  } catch { /* 무시 */ }
+  } catch {
+    /* 무시 */
+  }
   return best;
 }