agy-cli-usage 0.1.0 → 0.2.0

package/README.md

@@ -39,20 +39,23 @@ agy-usage --no-cache               # 5분 캐시 무시하고 강제 조회
 
 `agy`는 OAuth 토큰을 OS 키링(zalando/go-keyring 규약, service=`gemini`/account=`antigravity`)에 저장한다.
 
-1. **직접 API (기본)** — 키링에서 토큰을 읽고(만료 시 자동 refresh), `agy`와 동일하게:
+1. **직접 API (기본)** — 토큰을 읽고(만료 시 자동 refresh), `agy`와 동일하게:
    - `POST https://<host>/v1internal:loadCodeAssist` → `cloudaicompanionProject` 획득
    - `POST https://<host>/v1internal:retrieveUserQuotaSummary {project}` → 쿼타
    - 호스트: `daily-cloudcode-pa.googleapis.com`(현재 CLI) 또는 `cloudcode-pa.googleapis.com`
-2. **PTY 폴백** — 키링을 읽을 수 없거나(헤드리스 등) 내부 API가 바뀌면, `agy`를 가상 터미널로 띄워 `/usage`를 보내고, `@xterm/headless`로 alt-screen을 재구성해 패널을 파싱한다.
+   - 토큰 소스: **키링 → 토큰 파일 → (실패 시) PTY** 순으로 자동 시도
+2. **PTY 폴백** — 위 토큰 소스를 모두 읽을 수 없거나 내부 API가 바뀌면, `agy`를 가상 터미널로 띄워 `/usage`를 보내고, `@xterm/headless`로 alt-screen을 재구성해 패널을 파싱한다.
 
 ### 크로스플랫폼 자격증명
 
-| OS | 키링 백엔드 | 비고 |
+| OS | 토큰 소스 | 비고 |
 |----|-----------|------|
 | macOS | Keychain | `@napi-rs/keyring`, CLI 폴백 `security` |
 | Windows | Credential Manager | `@napi-rs/keyring` |
 | Linux (데스크톱) | Secret Service | `@napi-rs/keyring`, CLI 폴백 `secret-tool` |
-| Linux (헤드리스) | — | Secret Service 부재 시 `--source pty`로 우회 (python3 필요) |
+| Linux (헤드리스) | **토큰 파일** | 키링 부재 시 `~/.gemini/antigravity-cli/antigravity-oauth-token`(순수 JSON) 읽음 → **API 경로 동작**. `AGY_OAUTH_TOKEN_FILE`로 경로 override |
+
+> 헤드리스 서버에서도 토큰 파일을 읽어 빠른 API 경로를 쓴다. 파일조차 없으면 `--source pty`로 우회(python3 필요).
 
 ## HTTP 엔드포인트 (선택)
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agy-cli-usage",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Headless usage/quota monitor for the Antigravity CLI (agy) — reads Cloud Code quota directly, with a PTY fallback. No IDE required.",
   "type": "module",
   "bin": {

package/src/credentials.js

@@ -10,10 +10,15 @@
 // Read backends, tried in order:
 //   1. @napi-rs/keyring native module  (macOS / Windows / Linux Secret Service)
 //   2. OS CLI fallback                  (`security` on macOS, `secret-tool` on Linux)
-// If every backend fails (e.g. a headless Linux box with no Secret Service),
-// the caller falls back to the PTY path which drives `agy` itself.
+//   3. File fallback                    (headless Linux: agy can't reach a keyring
+//                                        and writes the token to a plain-JSON file)
+// If every backend fails, the caller falls back to the PTY path which drives
+// `agy` itself.
 
 import { execFileSync } from 'node:child_process';
+import { readFileSync, existsSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
 
 // OAuth client for the Antigravity CLI. This is an installed/desktop ("public")
 // OAuth client: per Google's own docs the client secret of an installed app is
@@ -68,17 +73,39 @@ function readViaCli() {
   return null;
 }
 
+// On headless Linux (no Secret Service) agy persists the token to a plain-JSON
+// file instead of the keyring. Same payload shape, no `go-keyring-base64:` prefix.
+function readViaFile() {
+  const candidates = [
+    process.env.AGY_OAUTH_TOKEN_FILE,
+    join(homedir(), '.gemini', 'antigravity-cli', 'antigravity-oauth-token'),
+  ].filter(Boolean);
+  for (const path of candidates) {
+    try {
+      if (existsSync(path)) {
+        const content = readFileSync(path, 'utf8').trim();
+        if (content) return content;
+      }
+    } catch {
+      // unreadable (perms) — try next candidate
+    }
+  }
+  return null;
+}
+
 async function readRawSecret() {
   const fromNapi = await readViaNapiEsm();
   if (fromNapi) return fromNapi;
   const fromCli = readViaCli();
   if (fromCli) return fromCli;
+  const fromFile = readViaFile();
+  if (fromFile) return fromFile;
   return null;
 }
 
 // --- decode ------------------------------------------------------------------
 
-function decodeSecret(raw) {
+export function decodeSecret(raw) {
   const payload = raw.startsWith(B64_PREFIX)
     ? Buffer.from(raw.slice(B64_PREFIX.length), 'base64').toString('utf8')
     : raw;
@@ -138,8 +165,9 @@ export async function getAccessToken() {
   const raw = await readRawSecret();
   if (!raw) {
     throw new CredentialError(
-      'Could not read agy credential from the OS keyring. ' +
-        'Is agy logged in on this machine? (headless servers may lack a keyring — use --source pty)',
+      'Could not read agy credential from the OS keyring or token file. ' +
+        'Is agy logged in on this machine? (set AGY_OAUTH_TOKEN_FILE to override the path, ' +
+        'or use --source pty)',
     );
   }
   const cred = decodeSecret(raw);