@metyatech/ai-quota 0.2.0

package/dist/gemini.js

@@ -0,0 +1,235 @@
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
+const ENV_GEMINI_OAUTH_CLIENT_ID = "AGENT_RUNNER_GEMINI_OAUTH_CLIENT_ID";
+const ENV_GEMINI_OAUTH_CLIENT_SECRET = "AGENT_RUNNER_GEMINI_OAUTH_CLIENT_SECRET";
+let cachedGeminiOauthClientInfo = null;
+function getClientIdFromIdToken(idToken) {
+    if (typeof idToken !== "string")
+        return undefined;
+    const parts = idToken.split(".");
+    if (parts.length !== 3)
+        return undefined;
+    try {
+        const payloadJson = Buffer.from(parts[1], "base64url").toString("utf8");
+        const payload = JSON.parse(payloadJson);
+        if (typeof payload?.aud === "string" && payload.aud.length > 0)
+            return payload.aud;
+        if (typeof payload?.azp === "string" && payload.azp.length > 0)
+            return payload.azp;
+        return undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+function readGeminiOauthClientInfoFromEnv() {
+    const clientId = process.env[ENV_GEMINI_OAUTH_CLIENT_ID];
+    const clientSecret = process.env[ENV_GEMINI_OAUTH_CLIENT_SECRET];
+    if (!clientId && !clientSecret)
+        return null;
+    return {
+        clientId: typeof clientId === "string" && clientId.length > 0 ? clientId : undefined,
+        clientSecret: typeof clientSecret === "string" && clientSecret.length > 0 ? clientSecret : undefined,
+        source: "env"
+    };
+}
+function extractOauthConstantsFromJs(content) {
+    const clientIdMatch = content.match(/const\s+OAUTH_CLIENT_ID\s*=\s*['"]([^'"]+)['"]/);
+    const clientSecretMatch = content.match(/const\s+OAUTH_CLIENT_SECRET\s*=\s*['"]([^'"]+)['"]/);
+    const clientId = clientIdMatch?.[1];
+    const clientSecret = clientSecretMatch?.[1];
+    if (!clientId && !clientSecret)
+        return null;
+    return {
+        clientId,
+        clientSecret,
+        source: "gemini-cli"
+    };
+}
+function tryReadGeminiCliOauthClientInfoFromWellKnownPaths() {
+    const candidates = [];
+    if (process.platform === "win32") {
+        const appData = process.env.APPDATA;
+        if (appData) {
+            const npmGlobal = path.join(appData, "npm", "node_modules");
+            candidates.push(path.join(npmGlobal, "@google", "gemini-cli", "node_modules", "@google", "gemini-cli-core", "dist", "src", "code_assist", "oauth2.js"), path.join(npmGlobal, "@google", "gemini-cli-core", "dist", "src", "code_assist", "oauth2.js"));
+        }
+    }
+    for (const filePath of candidates) {
+        try {
+            if (!fs.existsSync(filePath))
+                continue;
+            const content = fs.readFileSync(filePath, "utf8");
+            const extracted = extractOauthConstantsFromJs(content);
+            if (extracted)
+                return extracted;
+        }
+        catch {
+            // ignore and continue
+        }
+    }
+    return null;
+}
+function getGeminiOauthClientInfo() {
+    if (cachedGeminiOauthClientInfo)
+        return cachedGeminiOauthClientInfo;
+    cachedGeminiOauthClientInfo =
+        readGeminiOauthClientInfoFromEnv() ?? tryReadGeminiCliOauthClientInfoFromWellKnownPaths();
+    return cachedGeminiOauthClientInfo;
+}
+async function refreshAccessToken(options) {
+    const params = new URLSearchParams();
+    params.set("client_id", options.clientId);
+    params.set("client_secret", options.clientSecret);
+    params.set("refresh_token", options.refreshToken);
+    params.set("grant_type", "refresh_token");
+    const res = await fetch("https://oauth2.googleapis.com/token", {
+        method: "POST",
+        headers: { "Content-Type": "application/x-www-form-urlencoded" },
+        body: params.toString()
+    });
+    if (!res.ok) {
+        throw new Error(`Failed to refresh Google access token: ${res.status} ${await res.text()}`);
+    }
+    const data = (await res.json());
+    if (typeof data?.access_token !== "string" || data.access_token.length === 0) {
+        throw new Error("Google token refresh response missing access_token.");
+    }
+    const expiryDate = typeof data?.expires_in === "number"
+        ? Date.now() + Math.max(0, data.expires_in) * 1000
+        : undefined;
+    return { accessToken: data.access_token, expiryDate };
+}
+async function getCredentials() {
+    const credsPath = path.join(os.homedir(), ".gemini", "oauth_creds.json");
+    if (!fs.existsSync(credsPath)) {
+        throw new Error(`Gemini OAuth credentials not found at ${credsPath}`);
+    }
+    const raw = fs.readFileSync(credsPath, "utf8");
+    const creds = JSON.parse(raw);
+    let accessToken = creds.access_token;
+    const now = Date.now();
+    // Buffer of 5 minutes
+    if (!accessToken ||
+        (typeof creds.expiry_date === "number" && creds.expiry_date < now + 300000)) {
+        if (creds.refresh_token) {
+            const discovered = getGeminiOauthClientInfo();
+            const clientId = getClientIdFromIdToken(creds.id_token) ??
+                discovered?.clientId ??
+                (typeof creds.client_id === "string" && creds.client_id.length > 0
+                    ? creds.client_id
+                    : undefined);
+            const clientSecret = discovered?.clientSecret ??
+                (typeof creds.client_secret === "string" && creds.client_secret.length > 0
+                    ? creds.client_secret
+                    : undefined);
+            if (!clientId) {
+                throw new Error(`Gemini OAuth refresh requires a client ID; set ${ENV_GEMINI_OAUTH_CLIENT_ID} or install Gemini CLI.`);
+            }
+            if (!clientSecret) {
+                throw new Error(`Gemini OAuth refresh requires a client secret; set ${ENV_GEMINI_OAUTH_CLIENT_SECRET} or install Gemini CLI.`);
+            }
+            const refreshed = await refreshAccessToken({
+                refreshToken: creds.refresh_token,
+                clientId,
+                clientSecret
+            });
+            accessToken = refreshed.accessToken;
+            try {
+                creds.access_token = accessToken;
+                if (typeof refreshed.expiryDate === "number") {
+                    creds.expiry_date = refreshed.expiryDate;
+                }
+                fs.writeFileSync(credsPath, JSON.stringify(creds, null, 2), "utf8");
+            }
+            catch {
+                // Best-effort: keep the refreshed token in memory even if persisting fails.
+            }
+        }
+        else {
+            throw new Error("Gemini access token expired and no refresh token available.");
+        }
+    }
+    return { accessToken: accessToken };
+}
+/**
+ * Fetches Gemini quota usage from the Cloud Code Assist API.
+ *
+ * Reads OAuth credentials from `~/.gemini/oauth_creds.json` and automatically
+ * refreshes the access token when needed. Returns null on any error.
+ *
+ * The environment variables `AGENT_RUNNER_GEMINI_OAUTH_CLIENT_ID` and
+ * `AGENT_RUNNER_GEMINI_OAUTH_CLIENT_SECRET` can override the OAuth client
+ * credentials when the Gemini CLI is not installed.
+ */
+export async function fetchGeminiRateLimits() {
+    try {
+        const { accessToken } = await getCredentials();
+        // 1. Load Code Assist to get the project ID
+        const loadRes = await fetch("https://cloudcode-pa.googleapis.com/v1internal:loadCodeAssist", {
+            method: "POST",
+            headers: {
+                Authorization: `Bearer ${accessToken}`,
+                "Content-Type": "application/json",
+                "User-Agent": "ai-quota"
+            },
+            body: JSON.stringify({
+                metadata: {
+                    ideType: "GEMINI_CLI",
+                    platform: process.platform === "win32" ? "WINDOWS_AMD64" : "LINUX_AMD64"
+                }
+            })
+        });
+        if (!loadRes.ok) {
+            throw new Error(`loadCodeAssist failed: ${loadRes.status} ${await loadRes.text()}`);
+        }
+        const loadData = (await loadRes.json());
+        const projectId = loadData.cloudaicompanionProject;
+        if (!projectId) {
+            throw new Error("No cloudaicompanionProject found in loadCodeAssist response.");
+        }
+        // 2. Retrieve User Quota
+        const quotaRes = await fetch("https://cloudcode-pa.googleapis.com/v1internal:retrieveUserQuota", {
+            method: "POST",
+            headers: {
+                Authorization: `Bearer ${accessToken}`,
+                "Content-Type": "application/json",
+                "User-Agent": "ai-quota"
+            },
+            body: JSON.stringify({ project: projectId })
+        });
+        if (!quotaRes.ok) {
+            throw new Error(`retrieveUserQuota failed: ${quotaRes.status} ${await quotaRes.text()}`);
+        }
+        const quotaData = (await quotaRes.json());
+        const usage = {};
+        if (Array.isArray(quotaData.buckets)) {
+            for (const bucket of quotaData.buckets) {
+                const modelId = bucket.modelId;
+                if (modelId === "gemini-3-pro-preview" || modelId === "gemini-3-flash-preview") {
+                    // remainingFraction is usually between 0.0 and 1.0.
+                    // Keep fractional precision so tiny consumption (<0.5%) does not get rounded away.
+                    const remainingFraction = typeof bucket.remainingFraction === "number" &&
+                        Number.isFinite(bucket.remainingFraction)
+                        ? Math.min(Math.max(bucket.remainingFraction, 0), 1)
+                        : 1.0;
+                    const limit = 100; // percentage scale
+                    const usedRaw = (1.0 - remainingFraction) * 100;
+                    const used = Math.round(usedRaw * 1_000_000) / 1_000_000;
+                    usage[modelId] = {
+                        limit,
+                        usage: used,
+                        resetAt: bucket.resetTime ? new Date(bucket.resetTime) : new Date(Date.now() + 3600000)
+                    };
+                }
+            }
+        }
+        return usage;
+    }
+    catch (error) {
+        console.error("Error fetching Gemini usage:", error);
+        return null;
+    }
+}
+//# sourceMappingURL=gemini.js.map

package/dist/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../src/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AAKzB,MAAM,0BAA0B,GAAG,qCAAqC,CAAC;AACzE,MAAM,8BAA8B,GAAG,yCAAyC,CAAC;AAajF,IAAI,2BAA2B,GAAiC,IAAI,CAAC;AAErE,SAAS,sBAAsB,CAAC,OAAgB;IAC9C,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAClD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAEzC,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAA4B,CAAC;QACnE,IAAI,OAAO,OAAO,EAAE,GAAG,KAAK,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC;QACnF,IAAI,OAAO,OAAO,EAAE,GAAG,KAAK,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC;QACnF,OAAO,SAAS,CAAC;IACnB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,gCAAgC;IACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IACzD,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;IACjE,IAAI,CAAC,QAAQ,IAAI,CAAC,YAAY;QAAE,OAAO,IAAI,CAAC;IAE5C,OAAO;QACL,QAAQ,EAAE,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;QACpF,YAAY,EACV,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;QACxF,MAAM,EAAE,KAAK;KACd,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAe;IAClD,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACtF,MAAM,iBAAiB,GAAG,OAAO,CAAC,KAAK,CAAC,oDAAoD,CAAC,CAAC;IAC9F,MAAM,QAAQ,GAAG,aAAa,EAAE,CAAC,CAAC,CAAC,CAAC;IACpC,MAAM,YAAY,GAAG,iBAAiB,EAAE,CAAC,CAAC,CAAC,CAAC;IAE5C,IAAI,CAAC,QAAQ,IAAI,CAAC,YAAY;QAAE,OAAO,IAAI,CAAC;IAE5C,OAAO;QACL,QAAQ;QACR,YAAY;QACZ,MAAM,EAAE,YAAY;KACrB,CAAC;AACJ,CAAC;AAED,SAAS,iDAAiD;IACxD,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;QACpC,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC;YAC5D,UAAU,CAAC,IAAI,CACb,IAAI,CAAC,IAAI,CACP,SAAS,EACT,SAAS,EACT,YAAY,EACZ,cAAc,EACd,SAAS,EACT,iBAAiB,EACjB,MAAM,EACN,KAAK,EACL,aAAa,EACb,WAAW,CACZ,EACD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,WAAW,CAAC,CAC9F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,MAAM,QAAQ,IAAI,UAAU,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC;gBAAE,SAAS;YACvC,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAClD,MAAM,SAAS,GAAG,2BAA2B,CAAC,OAAO,CAAC,CAAC;YACvD,IAAI,SAAS;gBAAE,OAAO,SAAS,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACP,sBAAsB;QACxB,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,wBAAwB;IAC/B,IAAI,2BAA2B;QAAE,OAAO,2BAA2B,CAAC;IAEpE,2BAA2B;QACzB,gCAAgC,EAAE,IAAI,iDAAiD,EAAE,CAAC;IAE5F,OAAO,2BAA2B,CAAC;AACrC,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,OAIjC;IACC,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;IACrC,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC1C,MAAM,CAAC,GAAG,CAAC,eAAe,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;IAClD,MAAM,CAAC,GAAG,CAAC,eAAe,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;IAClD,MAAM,CAAC,GAAG,CAAC,YAAY,EAAE,eAAe,CAAC,CAAC;IAE1C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,qCAAqC,EAAE;QAC7D,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,cAAc,EAAE,mCAAmC,EAAE;QAChE,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;KACxB,CAAC,CAAC;IAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,0CAA0C,GAAG,CAAC,MAAM,IAAI,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC9F,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA4B,CAAC;IAC3D,IAAI,OAAO,IAAI,EAAE,YAAY,KAAK,QAAQ,IAAK,IAAI,CAAC,YAAuB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzF,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACzE,CAAC;IAED,MAAM,UAAU,GACd,OAAO,IAAI,EAAE,UAAU,KAAK,QAAQ;QAClC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,UAAoB,CAAC,GAAG,IAAI;QAC5D,CAAC,CAAC,SAAS,CAAC;IAEhB,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,YAAsB,EAAE,UAAU,EAAE,CAAC;AAClE,CAAC;AAED,KAAK,UAAU,cAAc;IAC3B,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,kBAAkB,CAAC,CAAC;IACzE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9B,MAAM,IAAI,KAAK,CAAC,yCAAyC,SAAS,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA4B,CAAC;IAEzD,IAAI,WAAW,GAAG,KAAK,CAAC,YAAY,CAAC;IACrC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,sBAAsB;IACtB,IACE,CAAC,WAAW;QACZ,CAAC,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ,IAAI,KAAK,CAAC,WAAW,GAAG,GAAG,GAAG,MAAM,CAAC,EAC3E,CAAC;QACD,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,UAAU,GAAG,wBAAwB,EAAE,CAAC;YAC9C,MAAM,QAAQ,GACZ,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC;gBACtC,UAAU,EAAE,QAAQ;gBACpB,CAAC,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC;oBAChE,CAAC,CAAC,KAAK,CAAC,SAAS;oBACjB,CAAC,CAAC,SAAS,CAAC,CAAC;YACjB,MAAM,YAAY,GAChB,UAAU,EAAE,YAAY;gBACxB,CAAC,OAAO,KAAK,CAAC,aAAa,KAAK,QAAQ,IAAI,KAAK,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC;oBACxE,CAAC,CAAC,KAAK,CAAC,aAAa;oBACrB,CAAC,CAAC,SAAS,CAAC,CAAC;YAEjB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CACb,kDAAkD,0BAA0B,yBAAyB,CACtG,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CACb,sDAAsD,8BAA8B,yBAAyB,CAC9G,CAAC;YACJ,CAAC;YAED,MAAM,SAAS,GAAG,MAAM,kBAAkB,CAAC;gBACzC,YAAY,EAAE,KAAK,CAAC,aAAuB;gBAC3C,QAAQ;gBACR,YAAY;aACb,CAAC,CAAC;YACH,WAAW,GAAG,SAAS,CAAC,WAAW,CAAC;YAEpC,IAAI,CAAC;gBACH,KAAK,CAAC,YAAY,GAAG,WAAW,CAAC;gBACjC,IAAI,OAAO,SAAS,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;oBAC7C,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC,UAAU,CAAC;gBAC3C,CAAC;gBACD,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YACtE,CAAC;YAAC,MAAM,CAAC;gBACP,4EAA4E;YAC9E,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;IAED,OAAO,EAAE,WAAW,EAAE,WAAqB,EAAE,CAAC;AAChD,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB;IACzC,IAAI,CAAC;QACH,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,cAAc,EAAE,CAAC;QAE/C,4CAA4C;QAC5C,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,+DAA+D,EAAE;YAC3F,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,WAAW,EAAE;gBACtC,cAAc,EAAE,kBAAkB;gBAClC,YAAY,EAAE,UAAU;aACzB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,QAAQ,EAAE;oBACR,OAAO,EAAE,YAAY;oBACrB,QAAQ,EAAE,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,aAAa;iBACzE;aACF,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,0BAA0B,OAAO,CAAC,MAAM,IAAI,MAAM,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACtF,CAAC;QAED,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,IAAI,EAAE,CAA4B,CAAC;QACnE,MAAM,SAAS,GAAG,QAAQ,CAAC,uBAAuB,CAAC;QAEnD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;QAClF,CAAC;QAED,yBAAyB;QACzB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,kEAAkE,EAClE;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,WAAW,EAAE;gBACtC,cAAc,EAAE,kBAAkB;gBAClC,YAAY,EAAE,UAAU;aACzB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;SAC7C,CACF,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,CAAC,MAAM,IAAI,MAAM,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC3F,CAAC;QAED,MAAM,SAAS,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA4B,CAAC;QACrE,MAAM,KAAK,GAAgB,EAAE,CAAC;QAE9B,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;YACrC,KAAK,MAAM,MAAM,IAAI,SAAS,CAAC,OAAoC,EAAE,CAAC;gBACpE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;gBAC/B,IAAI,OAAO,KAAK,sBAAsB,IAAI,OAAO,KAAK,wBAAwB,EAAE,CAAC;oBAC/E,oDAAoD;oBACpD,mFAAmF;oBACnF,MAAM,iBAAiB,GACrB,OAAO,MAAM,CAAC,iBAAiB,KAAK,QAAQ;wBAC5C,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC;wBACvC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;wBACpD,CAAC,CAAC,GAAG,CAAC;oBACV,MAAM,KAAK,GAAG,GAAG,CAAC,CAAC,mBAAmB;oBACtC,MAAM,OAAO,GAAG,CAAC,GAAG,GAAG,iBAAiB,CAAC,GAAG,GAAG,CAAC;oBAChD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,SAAS,CAAC,GAAG,SAAS,CAAC;oBACzD,KAAK,CAAC,OAA4B,CAAC,GAAG;wBACpC,KAAK;wBACL,KAAK,EAAE,IAAI;wBACX,OAAO,EACL,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAmB,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;qBAC3F,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @metyatech/ai-quota
+ *
+ * Quota / rate-limit fetching for Claude, Gemini, Copilot, Amazon Q, and Codex.
+ * Only fetching is provided here — gate/ramp evaluation logic stays in the caller.
+ */
+export type { RateLimitWindow, RateLimitSnapshot } from "./types.js";
+export type { ClaudeUsageData, ClaudeUsageBucket } from "./types.js";
+export type { GeminiUsage, GeminiModelUsage } from "./types.js";
+export type { CopilotUsage } from "./types.js";
+export type { AmazonQUsageSnapshot } from "./types.js";
+export { fetchClaudeRateLimits } from "./claude.js";
+export { fetchGeminiRateLimits } from "./gemini.js";
+export { fetchCopilotRateLimits, parseCopilotUserInfo, parseCopilotQuotaHeader } from "./copilot.js";
+export type { FetchCopilotRateLimitsOptions } from "./copilot.js";
+export { fetchAmazonQRateLimits, recordAmazonQUsage, loadAmazonQUsageState, saveAmazonQUsageState, resolveAmazonQUsageStatePath } from "./amazon-q.js";
+export { fetchCodexRateLimits, rateLimitSnapshotToStatus } from "./codex.js";
+export type { CodexStatus, UsageWindow, UsageWindowKey, FetchCodexRateLimitsOptions } from "./codex.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,YAAY,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACrE,YAAY,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACrE,YAAY,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAChE,YAAY,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC/C,YAAY,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAGvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAGpD,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAGpD,OAAO,EACL,sBAAsB,EACtB,oBAAoB,EACpB,uBAAuB,EACxB,MAAM,cAAc,CAAC;AACtB,YAAY,EAAE,6BAA6B,EAAE,MAAM,cAAc,CAAC;AAGlE,OAAO,EACL,sBAAsB,EACtB,kBAAkB,EAClB,qBAAqB,EACrB,qBAAqB,EACrB,4BAA4B,EAC7B,MAAM,eAAe,CAAC;AAGvB,OAAO,EACL,oBAAoB,EACpB,yBAAyB,EAC1B,MAAM,YAAY,CAAC;AACpB,YAAY,EACV,WAAW,EACX,WAAW,EACX,cAAc,EACd,2BAA2B,EAC5B,MAAM,YAAY,CAAC"}

package/dist/index.js

@@ -0,0 +1,17 @@
+/**
+ * @metyatech/ai-quota
+ *
+ * Quota / rate-limit fetching for Claude, Gemini, Copilot, Amazon Q, and Codex.
+ * Only fetching is provided here — gate/ramp evaluation logic stays in the caller.
+ */
+// Claude
+export { fetchClaudeRateLimits } from "./claude.js";
+// Gemini
+export { fetchGeminiRateLimits } from "./gemini.js";
+// Copilot
+export { fetchCopilotRateLimits, parseCopilotUserInfo, parseCopilotQuotaHeader } from "./copilot.js";
+// Amazon Q
+export { fetchAmazonQRateLimits, recordAmazonQUsage, loadAmazonQUsageState, saveAmazonQUsageState, resolveAmazonQUsageStatePath } from "./amazon-q.js";
+// Codex
+export { fetchCodexRateLimits, rateLimitSnapshotToStatus } from "./codex.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AASH,SAAS;AACT,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAEpD,SAAS;AACT,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAEpD,UAAU;AACV,OAAO,EACL,sBAAsB,EACtB,oBAAoB,EACpB,uBAAuB,EACxB,MAAM,cAAc,CAAC;AAGtB,WAAW;AACX,OAAO,EACL,sBAAsB,EACtB,kBAAkB,EAClB,qBAAqB,EACrB,qBAAqB,EACrB,4BAA4B,EAC7B,MAAM,eAAe,CAAC;AAEvB,QAAQ;AACR,OAAO,EACL,oBAAoB,EACpB,yBAAyB,EAC1B,MAAM,YAAY,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,64 @@
+/**
+ * Shared types for @metyatech/ai-quota.
+ *
+ * These types are intentionally kept compatible with the agent-runner
+ * usage types so that migration from local files to this package is
+ * a drop-in replacement.
+ */
+export type RateLimitWindow = {
+    usedPercent?: number;
+    used_percent?: number;
+    windowDurationMins?: number | null;
+    window_minutes?: number | null;
+    windowMinutes?: number | null;
+    resetsAt?: number | null;
+    resets_at?: number | null;
+};
+export type RateLimitSnapshot = {
+    primary?: RateLimitWindow | null;
+    secondary?: RateLimitWindow | null;
+    credits?: unknown;
+    planType?: string | null;
+    plan_type?: string | null;
+};
+export type ClaudeUsageBucket = {
+    utilization: number;
+    resets_at: string;
+};
+export type ClaudeUsageData = {
+    five_hour: ClaudeUsageBucket | null;
+    seven_day: ClaudeUsageBucket | null;
+    seven_day_sonnet: ClaudeUsageBucket | null;
+    extra_usage: {
+        is_enabled: boolean;
+        monthly_limit: number | null;
+        used_credits: number;
+        utilization: number;
+    } | null;
+};
+export type GeminiModelUsage = {
+    limit: number;
+    usage: number;
+    resetAt: Date;
+};
+export type GeminiUsage = {
+    "gemini-3-pro-preview"?: GeminiModelUsage;
+    "gemini-3-flash-preview"?: GeminiModelUsage;
+};
+export type CopilotUsage = {
+    percentRemaining: number;
+    resetAt: Date;
+    entitlement: number;
+    overageUsed: number;
+    overageEnabled: boolean;
+    source: "user" | "header";
+    raw: unknown;
+};
+export type AmazonQUsageSnapshot = {
+    used: number;
+    limit: number;
+    percentRemaining: number;
+    resetAt: Date;
+    periodKey: string;
+};
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAMH,MAAM,MAAM,eAAe,GAAG;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,kBAAkB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,OAAO,CAAC,EAAE,eAAe,GAAG,IAAI,CAAC;IACjC,SAAS,CAAC,EAAE,eAAe,GAAG,IAAI,CAAC;IACnC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC3B,CAAC;AAMF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,SAAS,EAAE,iBAAiB,GAAG,IAAI,CAAC;IACpC,SAAS,EAAE,iBAAiB,GAAG,IAAI,CAAC;IACpC,gBAAgB,EAAE,iBAAiB,GAAG,IAAI,CAAC;IAC3C,WAAW,EAAE;QACX,UAAU,EAAE,OAAO,CAAC;QACpB,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;QAC7B,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;KACrB,GAAG,IAAI,CAAC;CACV,CAAC;AAMF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,IAAI,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,sBAAsB,CAAC,EAAE,gBAAgB,CAAC;IAC1C,wBAAwB,CAAC,EAAE,gBAAgB,CAAC;CAC7C,CAAC;AAMF,MAAM,MAAM,YAAY,GAAG;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,IAAI,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,OAAO,CAAC;IACxB,MAAM,EAAE,MAAM,GAAG,QAAQ,CAAC;IAC1B,GAAG,EAAE,OAAO,CAAC;CACd,CAAC;AAMF,MAAM,MAAM,oBAAoB,GAAG;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,IAAI,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC"}

package/dist/types.js

@@ -0,0 +1,9 @@
+/**
+ * Shared types for @metyatech/ai-quota.
+ *
+ * These types are intentionally kept compatible with the agent-runner
+ * usage types so that migration from local files to this package is
+ * a drop-in replacement.
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG"}

package/package.json

@@ -0,0 +1,53 @@
+{
+  "name": "@metyatech/ai-quota",
+  "version": "0.2.0",
+  "description": "AI agent quota/rate-limit fetching library and CLI for Claude, Gemini, Copilot, Amazon Q, and Codex",
+  "license": "MIT",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "bin": {
+    "ai-quota": "./dist/cli.js"
+  },
+  "exports": {
+    ".": {
+      "import": "./dist/index.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/metyatech/ai-quota.git"
+  },
+  "bugs": {
+    "url": "https://github.com/metyatech/ai-quota/issues"
+  },
+  "homepage": "https://github.com/metyatech/ai-quota",
+  "engines": {
+    "node": ">=18"
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.build.json && node scripts/add-shebang.mjs",
+    "test": "vitest run",
+    "lint": "eslint . && tsc -p tsconfig.json --noEmit",
+    "format": "prettier --write .",
+    "format:check": "prettier --check .",
+    "verify": "npm run lint && npm test && npm run build"
+  },
+  "devDependencies": {
+    "@eslint/js": "^9.13.0",
+    "@types/node": "^22.10.0",
+    "@typescript-eslint/eslint-plugin": "^8.17.0",
+    "@typescript-eslint/parser": "^8.17.0",
+    "eslint": "^9.13.0",
+    "prettier": "^3.4.0",
+    "typescript": "^5.7.0",
+    "vitest": "^1.6.0"
+  }
+}