@slkiser/opencode-quota 1.0.0

package/dist/lib/google-token-cache.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-token-cache.d.ts","sourceRoot":"","sources":["../../src/lib/google-token-cache.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAOH,MAAM,WAAW,2BAA2B;IAC1C,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAqBD,wBAAgB,uBAAuB,IAAI,MAAM,CAEhD;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE;IAC1C,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,GAAG,MAAM,CAUT;AA4CD,wBAAsB,oBAAoB,CAAC,MAAM,EAAE;IACjD,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,2BAA2B,GAAG,IAAI,CAAC,CAO9C;AAED,wBAAsB,oBAAoB,CAAC,MAAM,EAAE;IACjD,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,2BAA2B,CAAC;CACpC,GAAG,OAAO,CAAC,IAAI,CAAC,CAKhB;AAED,wBAAsB,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC,CAG3D"}

package/dist/lib/google-token-cache.js

@@ -0,0 +1,100 @@
+/**
+ * Persistent access-token cache for Google Antigravity accounts.
+ *
+ * Why:
+ * - Antigravity quota is multi-account; each account needs its own access token.
+ * - Refreshing on every toast is noisy and increases timeout risk.
+ * - We persist access tokens so restarts don't force immediate refresh.
+ */
+import { readFile, writeFile, mkdir } from "fs/promises";
+import { homedir } from "os";
+import { dirname, join } from "path";
+import { createHash } from "crypto";
+const CACHE_VERSION = 1;
+let memCache = null;
+let loadPromise = null;
+function getCacheBaseDir() {
+    const home = homedir();
+    if (process.platform === "win32") {
+        return process.env.LOCALAPPDATA || join(home, "AppData", "Local");
+    }
+    return process.env.XDG_CACHE_HOME || join(home, ".cache");
+}
+export function getGoogleTokenCachePath() {
+    return join(getCacheBaseDir(), "opencode-quota", "google-access-tokens.json");
+}
+export function makeAccountCacheKey(params) {
+    const emailPart = (params.email ?? "").trim().toLowerCase();
+    const hash = createHash("sha256")
+        .update(params.refreshToken)
+        .update("\n")
+        .update(params.projectId)
+        .digest("hex")
+        .slice(0, 16);
+    // Keep a human hint without making it sensitive.
+    return `${emailPart}::${params.projectId}::${hash}`;
+}
+async function loadFromDisk(path) {
+    try {
+        const raw = await readFile(path, "utf-8");
+        const parsed = JSON.parse(raw);
+        if (!parsed || typeof parsed !== "object")
+            throw new Error("invalid");
+        const file = parsed;
+        if (file.version !== CACHE_VERSION || typeof file.tokens !== "object" || !file.tokens) {
+            throw new Error("invalid");
+        }
+        return {
+            version: CACHE_VERSION,
+            updatedAt: typeof file.updatedAt === "number" ? file.updatedAt : Date.now(),
+            tokens: file.tokens,
+        };
+    }
+    catch {
+        return { version: CACHE_VERSION, updatedAt: Date.now(), tokens: {} };
+    }
+}
+async function ensureLoaded() {
+    if (memCache)
+        return memCache;
+    if (loadPromise)
+        return loadPromise;
+    const path = getGoogleTokenCachePath();
+    loadPromise = (async () => {
+        const file = await loadFromDisk(path);
+        memCache = file;
+        loadPromise = null;
+        return file;
+    })();
+    return loadPromise;
+}
+async function persist() {
+    if (!memCache)
+        return;
+    const path = getGoogleTokenCachePath();
+    const dir = dirname(path);
+    await mkdir(dir, { recursive: true });
+    await writeFile(path, JSON.stringify(memCache, null, 2), "utf-8");
+}
+export async function getCachedAccessToken(params) {
+    const cache = await ensureLoaded();
+    const entry = cache.tokens[params.key];
+    if (!entry)
+        return null;
+    if (typeof entry.expiresAt !== "number")
+        return null;
+    if (entry.expiresAt <= Date.now() + params.skewMs)
+        return null;
+    return entry;
+}
+export async function setCachedAccessToken(params) {
+    const cache = await ensureLoaded();
+    cache.tokens[params.key] = params.entry;
+    cache.updatedAt = Date.now();
+    await persist();
+}
+export async function clearGoogleTokenCache() {
+    memCache = { version: CACHE_VERSION, updatedAt: Date.now(), tokens: {} };
+    await persist();
+}
+//# sourceMappingURL=google-token-cache.js.map

package/dist/lib/google-token-cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-token-cache.js","sourceRoot":"","sources":["../../src/lib/google-token-cache.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAC7B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAepC,MAAM,aAAa,GAAG,CAAU,CAAC;AAEjC,IAAI,QAAQ,GAAsC,IAAI,CAAC;AACvD,IAAI,WAAW,GAA+C,IAAI,CAAC;AAEnE,SAAS,eAAe;IACtB,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACjC,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,uBAAuB;IACrC,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,gBAAgB,EAAE,2BAA2B,CAAC,CAAC;AAChF,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAInC;IACC,MAAM,SAAS,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC5D,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC;SAC9B,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC;SAC3B,MAAM,CAAC,IAAI,CAAC;SACZ,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC;SACxB,MAAM,CAAC,KAAK,CAAC;SACb,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAChB,iDAAiD;IACjD,OAAO,GAAG,SAAS,KAAK,MAAM,CAAC,SAAS,KAAK,IAAI,EAAE,CAAC;AACtD,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,IAAY;IACtC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAY,CAAC;QAC1C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;QACtE,MAAM,IAAI,GAAG,MAAa,CAAC;QAC3B,IAAI,IAAI,CAAC,OAAO,KAAK,aAAa,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACtF,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO;YACL,OAAO,EAAE,aAAa;YACtB,SAAS,EAAE,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;YAC3E,MAAM,EAAE,IAAI,CAAC,MAAqD;SACnE,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACvE,CAAC;AACH,CAAC;AAED,KAAK,UAAU,YAAY;IACzB,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC9B,IAAI,WAAW;QAAE,OAAO,WAAW,CAAC;IAEpC,MAAM,IAAI,GAAG,uBAAuB,EAAE,CAAC;IACvC,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;QACxB,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,CAAC;QACtC,QAAQ,GAAG,IAAI,CAAC;QAChB,WAAW,GAAG,IAAI,CAAC;QACnB,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,OAAO;IACpB,IAAI,CAAC,QAAQ;QAAE,OAAO;IACtB,MAAM,IAAI,GAAG,uBAAuB,EAAE,CAAC;IACvC,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1B,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACtC,MAAM,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;AACpE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAG1C;IACC,MAAM,KAAK,GAAG,MAAM,YAAY,EAAE,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACvC,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACrD,IAAI,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IAC/D,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAG1C;IACC,MAAM,KAAK,GAAG,MAAM,YAAY,EAAE,CAAC;IACnC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;IACxC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,OAAO,EAAE,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB;IACzC,QAAQ,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACzE,MAAM,OAAO,EAAE,CAAC;AAClB,CAAC"}

package/dist/lib/google.d.ts

@@ -0,0 +1,43 @@
+/**
+ * Google Antigravity quota fetcher
+ *
+ * Uses OpenCode's antigravity-accounts.json at ~/.config/opencode/antigravity-accounts.json.
+ * Requires the user to have opencode-antigravity-auth installed and logged in.
+ */
+import type { AntigravityAccount, GoogleModelId, GoogleResult } from "./types.js";
+export declare function getAntigravityAccountsCandidatePaths(): string[];
+export declare function pickAntigravityAccountsPath(): string;
+/**
+ * Read Antigravity accounts from storage
+ */
+export declare function readAntigravityAccounts(): Promise<AntigravityAccount[] | null>;
+export declare function hasAntigravityAccountsConfigured(): Promise<boolean>;
+export declare function refreshGoogleTokensForAllAccounts(params?: {
+    skewMs?: number;
+    force?: boolean;
+}): Promise<null | {
+    total: number;
+    successCount: number;
+    failures: Array<{
+        email?: string;
+        error: string;
+    }>;
+}>;
+/**
+ * Query Google Antigravity quota for ALL accounts
+ *
+ * Reads accounts from ~/.config/opencode/antigravity-accounts.json.
+ * Refreshes access tokens and fetches quota for all accounts in parallel.
+ *
+ * @param modelIds - Model IDs to fetch quota for
+ * @returns Quota result with all models and any errors, or null if not configured
+ */
+export declare function queryGoogleQuota(modelIds: GoogleModelId[]): Promise<GoogleResult>;
+/**
+ * Format Google quota for toast display
+ *
+ * @param result - Google quota result
+ * @returns Formatted string like "G3Pro 100% * G3Flash 100% * Claude 0%" or null
+ */
+export declare function formatGoogleQuota(result: GoogleResult): string | null;
+//# sourceMappingURL=google.d.ts.map

package/dist/lib/google.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../src/lib/google.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAYH,OAAO,KAAK,EACV,kBAAkB,EAKlB,aAAa,EAEb,YAAY,EACb,MAAM,YAAY,CAAC;AA4BpB,wBAAgB,oCAAoC,IAAI,MAAM,EAAE,CAwB/D;AAED,wBAAgB,2BAA2B,IAAI,MAAM,CAMpD;AAED;;GAEG;AACH,wBAAsB,uBAAuB,IAAI,OAAO,CAAC,kBAAkB,EAAE,GAAG,IAAI,CAAC,CAmBpF;AAED,wBAAsB,gCAAgC,IAAI,OAAO,CAAC,OAAO,CAAC,CAGzE;AAsHD,wBAAsB,iCAAiC,CAAC,MAAM,CAAC,EAAE;IAC/D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB,GAAG,OAAO,CACP,IAAI,GACJ;IACE,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,KAAK,CAAC;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACpD,CACJ,CAmCA;AA+KD;;;;;;;;GAQG;AACH,wBAAsB,gBAAgB,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC,CAsCvF;AAED;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,YAAY,GAAG,MAAM,GAAG,IAAI,CAcrE"}

package/dist/lib/google.js

@@ -0,0 +1,399 @@
+/**
+ * Google Antigravity quota fetcher
+ *
+ * Uses OpenCode's antigravity-accounts.json at ~/.config/opencode/antigravity-accounts.json.
+ * Requires the user to have opencode-antigravity-auth installed and logged in.
+ */
+import { readFile } from "fs/promises";
+import { homedir } from "os";
+import { join } from "path";
+import { existsSync } from "fs";
+import { ANTIGRAVITY_CLIENT_ID as GOOGLE_CLIENT_ID, ANTIGRAVITY_CLIENT_SECRET as GOOGLE_CLIENT_SECRET, } from "opencode-antigravity-auth/dist/src/constants.js";
+import { REQUEST_TIMEOUT_MS, GOOGLE_MODEL_KEYS } from "./types.js";
+import { getCachedAccessToken, makeAccountCacheKey, setCachedAccessToken, } from "./google-token-cache.js";
+// =============================================================================
+// Constants
+// =============================================================================
+const GOOGLE_QUOTA_API_URL = "https://cloudcode-pa.googleapis.com/v1internal:fetchAvailableModels";
+const USER_AGENT = "antigravity/1.11.9 darwin/arm64";
+const GOOGLE_TOKEN_REFRESH_URL = "https://oauth2.googleapis.com/token";
+// Network timeouts tuned for reliability.
+const GOOGLE_TOKEN_TIMEOUT_MS = 8000;
+const GOOGLE_QUOTA_TIMEOUT_MS = 6000;
+// Multi-account fetching concurrency (reliability > speed).
+const GOOGLE_ACCOUNTS_CONCURRENCY = 3;
+// =============================================================================
+// Helpers
+// =============================================================================
+export function getAntigravityAccountsCandidatePaths() {
+    const home = homedir();
+    const isWindows = process.platform === "win32";
+    // Match common storage resolution used by opencode-antigravity-auth variants.
+    // - Config: $XDG_CONFIG_HOME/opencode (or ~/.config/opencode)
+    // - Data: $XDG_DATA_HOME/opencode (or ~/.local/share/opencode)
+    // - Windows: %APPDATA%/opencode (configBase), dataBase==configBase
+    const configBaseDir = isWindows
+        ? process.env.APPDATA || join(home, "AppData", "Roaming")
+        : process.env.XDG_CONFIG_HOME || join(home, ".config");
+    const xdgDataBase = isWindows
+        ? configBaseDir
+        : process.env.XDG_DATA_HOME || join(home, ".local", "share");
+    const candidates = [
+        join(configBaseDir, "opencode", "antigravity-accounts.json"),
+        join(xdgDataBase, "opencode", "antigravity-accounts.json"),
+    ];
+    // Unique + stable order.
+    return Array.from(new Set(candidates));
+}
+export function pickAntigravityAccountsPath() {
+    for (const p of getAntigravityAccountsCandidatePaths()) {
+        if (existsSync(p))
+            return p;
+    }
+    // Default to the first candidate for error/debug messaging.
+    return getAntigravityAccountsCandidatePaths()[0];
+}
+/**
+ * Read Antigravity accounts from storage
+ */
+export async function readAntigravityAccounts() {
+    for (const path of getAntigravityAccountsCandidatePaths()) {
+        try {
+            const content = await readFile(path, "utf-8");
+            const file = JSON.parse(content);
+            if (!file.accounts || file.accounts.length === 0) {
+                continue;
+            }
+            // Filter to accounts with refresh tokens
+            const validAccounts = file.accounts.filter((account) => account.refreshToken);
+            return validAccounts.length > 0 ? validAccounts : null;
+        }
+        catch {
+            // try next candidate
+        }
+    }
+    return null;
+}
+export async function hasAntigravityAccountsConfigured() {
+    const accounts = await readAntigravityAccounts();
+    return !!accounts && accounts.length > 0;
+}
+async function mapWithConcurrency(params) {
+    const n = Math.max(1, Math.trunc(params.concurrency));
+    const results = new Array(params.items.length);
+    let nextIndex = 0;
+    const workers = Array.from({ length: Math.min(n, params.items.length) }, async () => {
+        while (true) {
+            const idx = nextIndex++;
+            if (idx >= params.items.length)
+                return;
+            results[idx] = await params.fn(params.items[idx], idx);
+        }
+    });
+    await Promise.all(workers);
+    return results;
+}
+/**
+ * Refresh Google access token
+ */
+async function refreshAccessToken(refreshToken, timeoutMs = GOOGLE_TOKEN_TIMEOUT_MS) {
+    try {
+        const params = new URLSearchParams({
+            client_id: GOOGLE_CLIENT_ID,
+            client_secret: GOOGLE_CLIENT_SECRET,
+            refresh_token: refreshToken,
+            grant_type: "refresh_token",
+        });
+        const response = await fetchWithTimeout(GOOGLE_TOKEN_REFRESH_URL, {
+            method: "POST",
+            headers: { "Content-Type": "application/x-www-form-urlencoded" },
+            body: params,
+        }, timeoutMs);
+        if (!response.ok) {
+            // Try to extract error code from response
+            try {
+                const errorData = (await response.json());
+                if (errorData.error === "invalid_grant") {
+                    return { error: "Token revoked" };
+                }
+                return {
+                    error: errorData.error_description || `HTTP ${response.status}`,
+                };
+            }
+            catch {
+                return { error: `HTTP ${response.status}` };
+            }
+        }
+        const data = (await response.json());
+        return {
+            accessToken: data.access_token,
+            expiresIn: data.expires_in,
+        };
+    }
+    catch (err) {
+        if (err instanceof Error && err.message.includes("timeout")) {
+            return { error: "Token refresh timeout" };
+        }
+        return { error: "Token refresh failed" };
+    }
+}
+async function refreshAccessTokenWithCache(params) {
+    const skewMs = params.skewMs ?? 2 * 60_000;
+    const key = makeAccountCacheKey({
+        refreshToken: params.refreshToken,
+        projectId: params.projectId,
+        email: params.email,
+    });
+    if (!params.force) {
+        const cached = await getCachedAccessToken({ key, skewMs });
+        if (cached)
+            return { accessToken: cached.accessToken };
+    }
+    const refreshed = await refreshAccessToken(params.refreshToken);
+    if ("error" in refreshed)
+        return refreshed;
+    await setCachedAccessToken({
+        key,
+        entry: {
+            accessToken: refreshed.accessToken,
+            expiresAt: Date.now() + Math.max(1, refreshed.expiresIn) * 1000,
+            projectId: params.projectId,
+            email: params.email,
+        },
+    });
+    return { accessToken: refreshed.accessToken };
+}
+export async function refreshGoogleTokensForAllAccounts(params) {
+    const accounts = await readAntigravityAccounts();
+    if (!accounts || accounts.length === 0)
+        return null;
+    const valid = accounts.filter((a) => !!a.refreshToken);
+    if (valid.length === 0)
+        return null;
+    const results = await mapWithConcurrency({
+        items: valid,
+        concurrency: GOOGLE_ACCOUNTS_CONCURRENCY,
+        fn: async (account) => {
+            const email = account.email;
+            const projectId = getProjectId(account);
+            if (!projectId)
+                return { ok: false, email, error: "No projectId" };
+            const token = await refreshAccessTokenWithCache({
+                refreshToken: account.refreshToken,
+                projectId,
+                email,
+                skewMs: params?.skewMs,
+                force: params?.force,
+            });
+            if ("error" in token)
+                return { ok: false, email, error: token.error };
+            return { ok: true, email };
+        },
+    });
+    const failures = results.filter((r) => !r.ok).map((r) => ({ email: r.email, error: r.error }));
+    const successCount = results.filter((r) => r.ok).length;
+    return {
+        total: valid.length,
+        successCount,
+        failures,
+    };
+}
+/**
+ * Fetch with timeout
+ */
+async function fetchWithTimeout(url, options, timeoutMs = REQUEST_TIMEOUT_MS) {
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
+    try {
+        const response = await fetch(url, {
+            ...options,
+            signal: controller.signal,
+        });
+        return response;
+    }
+    catch (err) {
+        if (err instanceof Error && err.name === "AbortError") {
+            throw new Error(`Request timeout after ${Math.round(timeoutMs / 1000)}s`);
+        }
+        throw err;
+    }
+    finally {
+        clearTimeout(timeoutId);
+    }
+}
+/**
+ * Fetch quota from Google API
+ */
+async function fetchGoogleQuota(accessToken, projectId, timeoutMs = GOOGLE_QUOTA_TIMEOUT_MS) {
+    const response = await fetchWithTimeout(GOOGLE_QUOTA_API_URL, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${accessToken}`,
+            "User-Agent": USER_AGENT,
+        },
+        body: JSON.stringify({ project: projectId }),
+    }, timeoutMs);
+    if (!response.ok) {
+        if (response.status === 401 || response.status === 403) {
+            throw new Error(`Google API auth error: ${response.status}`);
+        }
+        throw new Error(`Google API error: ${response.status}`);
+    }
+    return response.json();
+}
+/**
+ * Extract model quotas from API response
+ */
+function extractModelQuotas(data, modelIds, accountEmail) {
+    const quotas = [];
+    for (const modelId of modelIds) {
+        const modelConfig = GOOGLE_MODEL_KEYS[modelId];
+        if (!modelConfig)
+            continue;
+        let modelInfo = data.models[modelConfig.key];
+        // Try alternate key if primary not found
+        if (!modelInfo && modelConfig.altKey) {
+            modelInfo = data.models[modelConfig.altKey];
+        }
+        if (modelInfo) {
+            const remainingFraction = modelInfo.quotaInfo?.remainingFraction ?? 0;
+            quotas.push({
+                modelId,
+                displayName: modelConfig.display,
+                percentRemaining: Math.round(remainingFraction * 100),
+                resetTimeIso: modelInfo.quotaInfo?.resetTime,
+                accountEmail,
+            });
+        }
+    }
+    return quotas;
+}
+/**
+ * Fetch quota for a single account
+ */
+function getProjectId(account) {
+    return account.projectId || account.projectID || account.managedProjectId;
+}
+// NOTE: This plugin treats Google Antigravity as truly multi-account.
+// Each account gets its own access token derived from its refresh token.
+async function fetchAccountQuotaWithAntigravityRefresh(params) {
+    const email = params.account.email || "Unknown";
+    const projectId = getProjectId(params.account);
+    if (!projectId) {
+        return { success: false, error: "No projectId", accountEmail: email };
+    }
+    try {
+        const tokenResult = await refreshAccessTokenWithCache({
+            refreshToken: params.account.refreshToken,
+            projectId,
+            email,
+        });
+        if ("error" in tokenResult)
+            return { success: false, error: tokenResult.error, accountEmail: email };
+        let data;
+        try {
+            data = await fetchGoogleQuota(tokenResult.accessToken, projectId);
+        }
+        catch (err) {
+            // One auth retry: refresh token then retry quota call.
+            if (err instanceof Error && err.message.includes("auth error")) {
+                const retryToken = await refreshAccessToken(params.account.refreshToken);
+                if ("error" in retryToken) {
+                    return { success: false, error: retryToken.error, accountEmail: email };
+                }
+                await setCachedAccessToken({
+                    key: makeAccountCacheKey({ refreshToken: params.account.refreshToken, projectId, email }),
+                    entry: {
+                        accessToken: retryToken.accessToken,
+                        expiresAt: Date.now() + Math.max(1, retryToken.expiresIn) * 1000,
+                        projectId,
+                        email,
+                    },
+                });
+                data = await fetchGoogleQuota(retryToken.accessToken, projectId);
+            }
+            else {
+                throw err;
+            }
+        }
+        const models = extractModelQuotas(data, params.modelIds, email);
+        return { success: true, models, accountEmail: email };
+    }
+    catch (err) {
+        if (err instanceof Error && err.message.includes("timeout")) {
+            return { success: false, error: "API timeout", accountEmail: email };
+        }
+        return {
+            success: false,
+            error: err instanceof Error ? err.message : String(err),
+            accountEmail: email,
+        };
+    }
+}
+// =============================================================================
+// Export
+// =============================================================================
+/**
+ * Query Google Antigravity quota for ALL accounts
+ *
+ * Reads accounts from ~/.config/opencode/antigravity-accounts.json.
+ * Refreshes access tokens and fetches quota for all accounts in parallel.
+ *
+ * @param modelIds - Model IDs to fetch quota for
+ * @returns Quota result with all models and any errors, or null if not configured
+ */
+export async function queryGoogleQuota(modelIds) {
+    const accounts = await readAntigravityAccounts();
+    if (!accounts || accounts.length === 0) {
+        return null;
+    }
+    // Query accounts with bounded concurrency (reliability > speed).
+    const results = await mapWithConcurrency({
+        items: accounts,
+        concurrency: GOOGLE_ACCOUNTS_CONCURRENCY,
+        fn: async (account) => fetchAccountQuotaWithAntigravityRefresh({ account, modelIds }),
+    });
+    // Collect all successful models and errors
+    const allModels = [];
+    const errors = [];
+    for (const result of results) {
+        if (result.success && result.models && result.models.length > 0) {
+            allModels.push(...result.models);
+        }
+        else if (!result.success && result.error && result.accountEmail) {
+            errors.push({ email: result.accountEmail, error: result.error });
+        }
+    }
+    // Return combined result
+    if (allModels.length === 0 && errors.length === 0) {
+        return {
+            success: false,
+            error: "No quota data available",
+        };
+    }
+    return {
+        success: true,
+        models: allModels,
+        errors: errors.length > 0 ? errors : undefined,
+    };
+}
+/**
+ * Format Google quota for toast display
+ *
+ * @param result - Google quota result
+ * @returns Formatted string like "G3Pro 100% * G3Flash 100% * Claude 0%" or null
+ */
+export function formatGoogleQuota(result) {
+    if (!result) {
+        return null;
+    }
+    if (!result.success) {
+        return null;
+    }
+    if (result.models.length === 0) {
+        return null;
+    }
+    return result.models.map((m) => `${m.displayName} ${m.percentRemaining}%`).join(" \u2022 ");
+}
+//# sourceMappingURL=google.js.map

package/dist/lib/google.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/lib/google.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAEhC,OAAO,EACL,qBAAqB,IAAI,gBAAgB,EACzC,yBAAyB,IAAI,oBAAoB,GAClD,MAAM,iDAAiD,CAAC;AAYzD,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACnE,OAAO,EACL,oBAAoB,EACpB,mBAAmB,EACnB,oBAAoB,GACrB,MAAM,yBAAyB,CAAC;AAEjC,gFAAgF;AAChF,YAAY;AACZ,gFAAgF;AAEhF,MAAM,oBAAoB,GAAG,qEAAqE,CAAC;AACnG,MAAM,UAAU,GAAG,iCAAiC,CAAC;AAErD,MAAM,wBAAwB,GAAG,qCAAqC,CAAC;AAEvE,0CAA0C;AAC1C,MAAM,uBAAuB,GAAG,IAAI,CAAC;AACrC,MAAM,uBAAuB,GAAG,IAAI,CAAC;AAErC,4DAA4D;AAC5D,MAAM,2BAA2B,GAAG,CAAC,CAAC;AAEtC,gFAAgF;AAChF,UAAU;AACV,gFAAgF;AAEhF,MAAM,UAAU,oCAAoC;IAClD,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC;IAE/C,8EAA8E;IAC9E,8DAA8D;IAC9D,+DAA+D;IAC/D,mEAAmE;IAEnE,MAAM,aAAa,GAAG,SAAS;QAC7B,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,SAAS,CAAC;QACzD,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IAEzD,MAAM,WAAW,GAAG,SAAS;QAC3B,CAAC,CAAC,aAAa;QACf,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IAE/D,MAAM,UAAU,GAAG;QACjB,IAAI,CAAC,aAAa,EAAE,UAAU,EAAE,2BAA2B,CAAC;QAC5D,IAAI,CAAC,WAAW,EAAE,UAAU,EAAE,2BAA2B,CAAC;KAC3D,CAAC;IAEF,yBAAyB;IACzB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,2BAA2B;IACzC,KAAK,MAAM,CAAC,IAAI,oCAAoC,EAAE,EAAE,CAAC;QACvD,IAAI,UAAU,CAAC,CAAC,CAAC;YAAE,OAAO,CAAC,CAAC;IAC9B,CAAC;IACD,4DAA4D;IAC5D,OAAO,oCAAoC,EAAE,CAAC,CAAC,CAAE,CAAC;AACpD,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB;IAC3C,KAAK,MAAM,IAAI,IAAI,oCAAoC,EAAE,EAAE,CAAC;QAC1D,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA4B,CAAC;YAE5D,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjD,SAAS;YACX,CAAC;YAED,yCAAyC;YACzC,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAC9E,OAAO,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC;QACzD,CAAC;QAAC,MAAM,CAAC;YACP,qBAAqB;QACvB,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gCAAgC;IACpD,MAAM,QAAQ,GAAG,MAAM,uBAAuB,EAAE,CAAC;IACjD,OAAO,CAAC,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;AAC3C,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAO,MAIvC;IACC,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;IACtD,MAAM,OAAO,GAAG,IAAI,KAAK,CAAI,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAClD,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;QAClF,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;YACxB,IAAI,GAAG,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM;gBAAE,OAAO;YACvC,OAAO,CAAC,GAAG,CAAC,GAAG,MAAM,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAE,EAAE,GAAG,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC3B,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,kBAAkB,CAC/B,YAAoB,EACpB,YAAoB,uBAAuB;IAE3C,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;YACjC,SAAS,EAAE,gBAAgB;YAC3B,aAAa,EAAE,oBAAoB;YACnC,aAAa,EAAE,YAAY;YAC3B,UAAU,EAAE,eAAe;SAC5B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CACrC,wBAAwB,EACxB;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,mCAAmC,EAAE;YAChE,IAAI,EAAE,MAAM;SACb,EACD,SAAS,CACV,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,0CAA0C;YAC1C,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAGvC,CAAC;gBACF,IAAI,SAAS,CAAC,KAAK,KAAK,eAAe,EAAE,CAAC;oBACxC,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;gBACpC,CAAC;gBACD,OAAO;oBACL,KAAK,EAAE,SAAS,CAAC,iBAAiB,IAAI,QAAQ,QAAQ,CAAC,MAAM,EAAE;iBAChE,CAAC;YACJ,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,KAAK,EAAE,QAAQ,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAGlC,CAAC;QAEF,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,YAAY;YAC9B,SAAS,EAAE,IAAI,CAAC,UAAU;SAC3B,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,KAAK,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5D,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;QAC5C,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,sBAAsB,EAAE,CAAC;IAC3C,CAAC;AACH,CAAC;AAED,KAAK,UAAU,2BAA2B,CAAC,MAM1C;IACC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,CAAC,GAAG,MAAM,CAAC;IAC3C,MAAM,GAAG,GAAG,mBAAmB,CAAC;QAC9B,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;KACpB,CAAC,CAAC;IAEH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAClB,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3D,IAAI,MAAM;YAAE,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC;IACzD,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,kBAAkB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IAChE,IAAI,OAAO,IAAI,SAAS;QAAE,OAAO,SAAS,CAAC;IAE3C,MAAM,oBAAoB,CAAC;QACzB,GAAG;QACH,KAAK,EAAE;YACL,WAAW,EAAE,SAAS,CAAC,WAAW;YAClC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,GAAG,IAAI;YAC/D,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;SACpB;KACF,CAAC,CAAC;IAEH,OAAO,EAAE,WAAW,EAAE,SAAS,CAAC,WAAW,EAAE,CAAC;AAChD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iCAAiC,CAAC,MAGvD;IAQC,MAAM,QAAQ,GAAG,MAAM,uBAAuB,EAAE,CAAC;IACjD,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAEpD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;IACvD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAEpC,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC;QACvC,KAAK,EAAE,KAAK;QACZ,WAAW,EAAE,2BAA2B;QACxC,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACpB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;YAC5B,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;YACxC,IAAI,CAAC,SAAS;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAc,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;YAE5E,MAAM,KAAK,GAAG,MAAM,2BAA2B,CAAC;gBAC9C,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,SAAS;gBACT,KAAK;gBACL,MAAM,EAAE,MAAM,EAAE,MAAM;gBACtB,KAAK,EAAE,MAAM,EAAE,KAAK;aACrB,CAAC,CAAC;YACH,IAAI,OAAO,IAAI,KAAK;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAc,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;YAC/E,OAAO,EAAE,EAAE,EAAE,IAAa,EAAE,KAAK,EAAE,CAAC;QACtC,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC/F,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC;IAExD,OAAO;QACL,KAAK,EAAE,KAAK,CAAC,MAAM;QACnB,YAAY;QACZ,QAAQ;KACT,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,gBAAgB,CAC7B,GAAW,EACX,OAAoB,EACpB,YAAoB,kBAAkB;IAEtC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;IAElE,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,GAAG,OAAO;YACV,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,KAAK,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACtD,MAAM,IAAI,KAAK,CAAC,yBAAyB,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5E,CAAC;QACD,MAAM,GAAG,CAAC;IACZ,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,gBAAgB,CAC7B,WAAmB,EACnB,SAAiB,EACjB,YAAoB,uBAAuB;IAE3C,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CACrC,oBAAoB,EACpB;QACE,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,WAAW,EAAE;YACtC,YAAY,EAAE,UAAU;SACzB;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;KAC7C,EACD,SAAS,CACV,CAAC;IAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/D,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,OAAO,QAAQ,CAAC,IAAI,EAAkC,CAAC;AACzD,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB,CACzB,IAAyB,EACzB,QAAyB,EACzB,YAAqB;IAErB,MAAM,MAAM,GAAuB,EAAE,CAAC;IAEtC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,WAAW,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC,WAAW;YAAE,SAAS;QAE3B,IAAI,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAE7C,yCAAyC;QACzC,IAAI,CAAC,SAAS,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;YACrC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,iBAAiB,GAAG,SAAS,CAAC,SAAS,EAAE,iBAAiB,IAAI,CAAC,CAAC;YACtE,MAAM,CAAC,IAAI,CAAC;gBACV,OAAO;gBACP,WAAW,EAAE,WAAW,CAAC,OAAO;gBAChC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB,GAAG,GAAG,CAAC;gBACrD,YAAY,EAAE,SAAS,CAAC,SAAS,EAAE,SAAS;gBAC5C,YAAY;aACb,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,OAA2B;IAC/C,OAAO,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,gBAAgB,CAAC;AAC5E,CAAC;AAED,sEAAsE;AACtE,yEAAyE;AAEzE,KAAK,UAAU,uCAAuC,CAAC,MAGtD;IAMC,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,IAAI,SAAS,CAAC;IAChD,MAAM,SAAS,GAAG,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAE/C,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;IACxE,CAAC;IAED,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,2BAA2B,CAAC;YACpD,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,YAAY;YACzC,SAAS;YACT,KAAK;SACN,CAAC,CAAC;QAEH,IAAI,OAAO,IAAI,WAAW;YAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;QAErG,IAAI,IAAyB,CAAC;QAC9B,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,gBAAgB,CAAC,WAAW,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACpE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,uDAAuD;YACvD,IAAI,GAAG,YAAY,KAAK,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;gBAC/D,MAAM,UAAU,GAAG,MAAM,kBAAkB,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;gBACzE,IAAI,OAAO,IAAI,UAAU,EAAE,CAAC;oBAC1B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;gBAC1E,CAAC;gBACD,MAAM,oBAAoB,CAAC;oBACzB,GAAG,EAAE,mBAAmB,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;oBACzF,KAAK,EAAE;wBACL,WAAW,EAAE,UAAU,CAAC,WAAW;wBACnC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,GAAG,IAAI;wBAChE,SAAS;wBACT,KAAK;qBACN;iBACF,CAAC,CAAC;gBACH,IAAI,GAAG,MAAM,gBAAgB,CAAC,UAAU,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YACnE,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAG,kBAAkB,CAAC,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAEhE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;IACxD,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,KAAK,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;QACvE,CAAC;QACD,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;YACvD,YAAY,EAAE,KAAK;SACpB,CAAC;IACJ,CAAC;AACH,CAAC;AAED,gFAAgF;AAChF,SAAS;AACT,gFAAgF;AAEhF;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,QAAyB;IAC9D,MAAM,QAAQ,GAAG,MAAM,uBAAuB,EAAE,CAAC;IACjD,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,iEAAiE;IACjE,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC;QACvC,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,2BAA2B;QACxC,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,uCAAuC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;KACtF,CAAC,CAAC;IAEH,2CAA2C;IAC3C,MAAM,SAAS,GAAuB,EAAE,CAAC;IACzC,MAAM,MAAM,GAAyB,EAAE,CAAC;IAExC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChE,SAAS,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;aAAM,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YAClE,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAClD,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,yBAAyB;SACjC,CAAC;IACJ,CAAC;IAED,OAAO;QACL,OAAO,EAAE,IAAI;QACb,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;KAC1B,CAAC;AACzB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAAoB;IACpD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,gBAAgB,GAAG,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;AAC9F,CAAC"}

package/dist/lib/markdown-table.d.ts

@@ -0,0 +1,8 @@
+type Align = "left" | "right" | "center";
+export declare function renderMarkdownTable(params: {
+    headers: string[];
+    rows: string[][];
+    aligns?: Align[];
+}): string;
+export {};
+//# sourceMappingURL=markdown-table.d.ts.map

package/dist/lib/markdown-table.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"markdown-table.d.ts","sourceRoot":"","sources":["../../src/lib/markdown-table.ts"],"names":[],"mappings":"AAAA,KAAK,KAAK,GAAG,MAAM,GAAG,OAAO,GAAG,QAAQ,CAAC;AAqBzC,wBAAgB,mBAAmB,CAAC,MAAM,EAAE;IAC1C,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC;IACjB,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;CAClB,GAAG,MAAM,CAsCT"}

package/dist/lib/markdown-table.js

@@ -0,0 +1,56 @@
+function padCell(text, width, align) {
+    const s = text ?? "";
+    const len = s.length;
+    if (len >= width)
+        return s;
+    const pad = width - len;
+    if (align === "right")
+        return " ".repeat(pad) + s;
+    if (align === "center") {
+        const left = Math.floor(pad / 2);
+        const right = pad - left;
+        return " ".repeat(left) + s + " ".repeat(right);
+    }
+    return s + " ".repeat(pad);
+}
+function escapeCell(text) {
+    // Keep this conservative: tool output should be plain ascii.
+    return (text ?? "").replace(/\|/g, "\\|").replace(/\r?\n/g, " ");
+}
+export function renderMarkdownTable(params) {
+    const aligns = params.aligns ?? params.headers.map(() => "left");
+    const colCount = params.headers.length;
+    const safeRows = params.rows.map((r) => {
+        const out = [];
+        for (let i = 0; i < colCount; i++)
+            out.push(escapeCell(r[i] ?? ""));
+        return out;
+    });
+    const headerCells = params.headers.map((h) => escapeCell(h));
+    const widths = headerCells.map((h) => Math.max(3, h.length));
+    for (const row of safeRows) {
+        for (let i = 0; i < colCount; i++) {
+            widths[i] = Math.max(widths[i], (row[i] ?? "").length);
+        }
+    }
+    const fmtRow = (cells) => `| ${cells
+        .map((c, i) => padCell(c, widths[i], aligns[i] ?? "left"))
+        .join(" | ")} |`;
+    const sep = `| ${widths
+        .map((w, i) => {
+        const a = aligns[i] ?? "left";
+        if (a === "right")
+            return "-".repeat(Math.max(3, w - 1)) + ":";
+        if (a === "center")
+            return ":" + "-".repeat(Math.max(3, w - 2)) + ":";
+        return "-".repeat(Math.max(3, w));
+    })
+        .join(" | ")} |`;
+    const lines = [];
+    lines.push(fmtRow(headerCells));
+    lines.push(sep);
+    for (const row of safeRows)
+        lines.push(fmtRow(row));
+    return lines.join("\n");
+}
+//# sourceMappingURL=markdown-table.js.map

package/dist/lib/markdown-table.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"markdown-table.js","sourceRoot":"","sources":["../../src/lib/markdown-table.ts"],"names":[],"mappings":"AAEA,SAAS,OAAO,CAAC,IAAY,EAAE,KAAa,EAAE,KAAY;IACxD,MAAM,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC;IACrB,MAAM,GAAG,GAAG,CAAC,CAAC,MAAM,CAAC;IACrB,IAAI,GAAG,IAAI,KAAK;QAAE,OAAO,CAAC,CAAC;IAC3B,MAAM,GAAG,GAAG,KAAK,GAAG,GAAG,CAAC;IACxB,IAAI,KAAK,KAAK,OAAO;QAAE,OAAO,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAClD,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;QACjC,MAAM,KAAK,GAAG,GAAG,GAAG,IAAI,CAAC;QACzB,OAAO,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC9B,6DAA6D;IAC7D,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;AACnE,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAInC;IACC,MAAM,MAAM,GAAY,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC;IAC1E,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;IAEvC,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACrC,MAAM,GAAG,GAAa,EAAE,CAAC;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE;YAAE,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACpE,OAAO,GAAG,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IAE7D,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7D,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,MAAM,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,CAAC,KAAe,EAAE,EAAE,CACjC,KAAK,KAAK;SACP,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC;SACzD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IAErB,MAAM,GAAG,GAAG,KAAK,MAAM;SACpB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACZ,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC;QAC9B,IAAI,CAAC,KAAK,OAAO;YAAE,OAAO,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;QAC/D,IAAI,CAAC,KAAK,QAAQ;YAAE,OAAO,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC;QACtE,OAAO,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC,CAAC;SACD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IAEnB,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;IAChC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAChB,KAAK,MAAM,GAAG,IAAI,QAAQ;QAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;IACpD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}