@ereinha/opencode-enhanced-quotas 1.0.0

package/dist/providers/antigravity/provider.js

@@ -0,0 +1,221 @@
+import { formatRelativeTime } from "../../utils/time";
+import { getCloudCredentials } from "./auth";
+import { logger } from "../../logger";
+const CLOUDCODE_ENDPOINTS = [
+    "https://daily-cloudcode-pa.sandbox.googleapis.com",
+    "https://autopush-cloudcode-pa.sandbox.googleapis.com",
+    "https://cloudcode-pa.googleapis.com",
+];
+const CLOUDCODE_HEADERS = {
+    "Content-Type": "application/json",
+    "User-Agent": "antigravity/1.11.5 windows/amd64",
+    "X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
+    "Client-Metadata": '{"ideType":"IDE_UNSPECIFIED","platform":"PLATFORM_UNSPECIFIED","pluginType":"GEMINI"}',
+};
+const DEFAULT_INDICATORS = [
+    { threshold: 0.2, symbol: "!" },
+    { threshold: 0.05, symbol: "!!" },
+];
+function getIndicatorSymbol(fraction, indicators = DEFAULT_INDICATORS) {
+    if (indicators.length === 0)
+        return "";
+    const sorted = [...indicators].sort((a, b) => a.threshold - b.threshold);
+    for (const indicator of sorted) {
+        if (fraction <= indicator.threshold) {
+            return ` ${indicator.symbol}`;
+        }
+    }
+    return "";
+}
+async function fetchAvailableModels(accessToken, projectId, debugEnabled) {
+    const payload = projectId ? { project: projectId } : {};
+    let lastError = null;
+    const headers = {
+        ...CLOUDCODE_HEADERS,
+        Authorization: `Bearer ${accessToken}`,
+    };
+    for (const endpoint of CLOUDCODE_ENDPOINTS) {
+        try {
+            const url = `${endpoint}/v1internal:fetchAvailableModels`;
+            const controller = new AbortController();
+            const timeoutId = setTimeout(() => controller.abort(), 10000);
+            try {
+                if (debugEnabled) {
+                    logger.debug("antigravity:request", {
+                        endpoint,
+                        url,
+                        hasProjectId: !!projectId,
+                    });
+                }
+                const response = await fetch(url, {
+                    method: "POST",
+                    headers,
+                    body: JSON.stringify(payload),
+                    signal: controller.signal,
+                });
+                if (debugEnabled) {
+                    logger.debug("antigravity:response_meta", {
+                        endpoint,
+                        status: response.status,
+                        ok: response.ok,
+                        contentType: response.headers.get("content-type"),
+                    });
+                }
+                if (response.status === 401) {
+                    throw new Error("Authorization expired or invalid.");
+                }
+                if (response.status === 403) {
+                    throw new Error("Access forbidden (403). Check your account permissions.");
+                }
+                if (!response.ok) {
+                    const text = await response.text();
+                    if (debugEnabled) {
+                        logger.debug("antigravity:error_body", {
+                            endpoint,
+                            status: response.status,
+                            bodyPreview: text.slice(0, 2000),
+                        });
+                    }
+                    throw new Error(`Cloud Code API error ${response.status}: ${text.slice(0, 200)}`);
+                }
+                const json = (await response.json());
+                const modelCount = Object.keys(json.models || {}).length;
+                if (debugEnabled) {
+                    logger.debug("antigravity:fetch_success", { modelCount });
+                }
+                if (debugEnabled) {
+                    const sampleKeys = Object.keys(json.models || {}).slice(0, 8);
+                    const sanitizedSample = {};
+                    for (const key of sampleKeys) {
+                        sanitizedSample[key] = json.models?.[key];
+                    }
+                    logger.debug("antigravity:raw_response_sample", {
+                        modelCount,
+                        sampleKeys,
+                        sample: sanitizedSample,
+                    });
+                }
+                return json;
+            }
+            finally {
+                clearTimeout(timeoutId);
+            }
+        }
+        catch (error) {
+            lastError =
+                error instanceof Error ? error : new Error(String(error));
+            if (lastError.message.includes("Authorization") ||
+                lastError.message.includes("forbidden") ||
+                lastError.message.includes("invalid_grant")) {
+                throw lastError;
+            }
+        }
+    }
+    throw lastError || new Error("All Cloud Code API endpoints failed");
+}
+export async function fetchCloudQuota(accessToken, projectId, debugEnabled = false) {
+    if (!accessToken) {
+        throw new Error("Access token is required for cloud quota fetching");
+    }
+    const response = await fetchAvailableModels(accessToken, projectId, debugEnabled);
+    const models = [];
+    if (response.models) {
+        for (const [modelKey, info] of Object.entries(response.models)) {
+            if (!info.quotaInfo)
+                continue;
+            models.push({
+                modelName: info.model || modelKey,
+                label: info.displayName || modelKey,
+                quotaInfo: {
+                    remainingFraction: info.quotaInfo.remainingFraction ?? 0,
+                    resetTime: info.quotaInfo.resetTime,
+                },
+            });
+        }
+    }
+    return {
+        account: {
+            projectId,
+        },
+        models,
+        timestamp: Date.now(),
+    };
+}
+/**
+ * Creates the Antigravity provider that returns flat, raw quota data.
+ * Grouping and aggregation is handled by the service layer via AggregatedGroups.
+ */
+export function createAntigravityProvider(config = {}) {
+    return {
+        id: "antigravity",
+        async fetchQuota() {
+            const debugEnabled = !!config.debug;
+            logger.debug("provider:antigravity:fetch_start", {
+                configDebug: config.debug,
+            });
+            // Fetch cloud credentials (Google OAuth)
+            let credentials;
+            try {
+                credentials = await getCloudCredentials();
+                if (debugEnabled) {
+                    logger.debug("provider:antigravity:auth_ok", {
+                        projectId: credentials.projectId,
+                    });
+                }
+            }
+            catch (e) {
+                logger.error("provider:antigravity:auth_failed", e);
+                throw e;
+            }
+            // Fetch live quota from Antigravity Cloud API
+            const cloudResult = await fetchCloudQuota(credentials.accessToken, credentials.projectId, debugEnabled);
+            if (debugEnabled) {
+                logger.debug("provider:antigravity:cloud_ok", {
+                    modelCount: cloudResult.models.length,
+                });
+            }
+            // Return flat list of all models with quota info
+            const entries = [];
+            for (const model of cloudResult.models) {
+                if (!model.quotaInfo ||
+                    typeof model.quotaInfo.remainingFraction !== "number") {
+                    continue;
+                }
+                const label = model.label || model.modelName || "";
+                // Filter out internal/miscellaneous quotas (e.g. "chat 12345", "rev123")
+                const lowerLabel = label.toLowerCase();
+                if (/^chat \d+/.test(lowerLabel) || lowerLabel.startsWith("rev")) {
+                    continue;
+                }
+                const remainingFraction = model.quotaInfo.remainingFraction;
+                const usedPercent = Math.max(0, Math.min(100, (1 - remainingFraction) * 100));
+                // Generate stable raw ID from model name
+                const rawId = `ag-raw-${label
+                    .toLowerCase()
+                    .replace(/[^a-z0-9]+/g, "-")
+                    .replace(/^-|-$/g, "")}`;
+                const indicator = getIndicatorSymbol(remainingFraction, config.indicators);
+                let reset;
+                if (model.quotaInfo.resetTime) {
+                    reset = `resets in ${formatRelativeTime(new Date(model.quotaInfo.resetTime))}`;
+                }
+                entries.push({
+                    id: rawId,
+                    providerName: `Antigravity ${label}`,
+                    used: usedPercent,
+                    limit: 100,
+                    unit: "%",
+                    reset,
+                    info: indicator.trim() || undefined,
+                });
+            }
+            if (debugEnabled) {
+                logger.debug("provider:antigravity:fetch_ok", {
+                    count: entries.length,
+                });
+            }
+            return entries;
+        },
+    };
+}
+//# sourceMappingURL=provider.js.map

package/dist/providers/antigravity/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../../src/providers/antigravity/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,QAAQ,CAAC;AAE7C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,MAAM,mBAAmB,GAAG;IACxB,mDAAmD;IACnD,sDAAsD;IACtD,qCAAqC;CAC/B,CAAC;AAEX,MAAM,iBAAiB,GAAG;IACtB,cAAc,EAAE,kBAAkB;IAClC,YAAY,EAAE,kCAAkC;IAChD,mBAAmB,EAAE,8CAA8C;IACnE,iBAAiB,EACb,uFAAuF;CACrF,CAAC;AAoDX,MAAM,kBAAkB,GAAqB;IACzC,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;IAC/B,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;CACpC,CAAC;AAEF,SAAS,kBAAkB,CACvB,QAAgB,EAChB,aAA+B,kBAAkB;IAEjD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IACvC,MAAM,MAAM,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IACzE,KAAK,MAAM,SAAS,IAAI,MAAM,EAAE,CAAC;QAC7B,IAAI,QAAQ,IAAI,SAAS,CAAC,SAAS,EAAE,CAAC;YAClC,OAAO,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QAClC,CAAC;IACL,CAAC;IACD,OAAO,EAAE,CAAC;AACd,CAAC;AAED,KAAK,UAAU,oBAAoB,CAC/B,WAAmB,EACnB,SAA6B,EAC7B,YAAqB;IAErB,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACxD,IAAI,SAAS,GAAiB,IAAI,CAAC;IAEnC,MAAM,OAAO,GAA2B;QACpC,GAAG,iBAAiB;QACpB,aAAa,EAAE,UAAU,WAAW,EAAE;KACzC,CAAC;IAEF,KAAK,MAAM,QAAQ,IAAI,mBAAmB,EAAE,CAAC;QACzC,IAAI,CAAC;YACD,MAAM,GAAG,GAAG,GAAG,QAAQ,kCAAkC,CAAC;YAC1D,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,KAAK,CAAC,CAAC;YAE9D,IAAI,CAAC;gBACD,IAAI,YAAY,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE;wBAChC,QAAQ;wBACR,GAAG;wBACH,YAAY,EAAE,CAAC,CAAC,SAAS;qBAC5B,CAAC,CAAC;gBACP,CAAC;gBAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;oBAC9B,MAAM,EAAE,MAAM;oBACd,OAAO;oBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;oBAC7B,MAAM,EAAE,UAAU,CAAC,MAAM;iBAC5B,CAAC,CAAC;gBAEH,IAAI,YAAY,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE;wBACtC,QAAQ;wBACR,MAAM,EAAE,QAAQ,CAAC,MAAM;wBACvB,EAAE,EAAE,QAAQ,CAAC,EAAE;wBACf,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;qBACpD,CAAC,CAAC;gBACP,CAAC;gBAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;gBACzD,CAAC;gBAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CACX,yDAAyD,CAC5D,CAAC;gBACN,CAAC;gBAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACf,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACnC,IAAI,YAAY,EAAE,CAAC;wBACf,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;4BACnC,QAAQ;4BACR,MAAM,EAAE,QAAQ,CAAC,MAAM;4BACvB,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;yBACnC,CAAC,CAAC;oBACP,CAAC;oBAED,MAAM,IAAI,KAAK,CACX,wBAAwB,QAAQ,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CACnE,CAAC;gBACN,CAAC;gBAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAwB,CAAC;gBAE5D,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;gBACzD,IAAI,YAAY,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC9D,CAAC;gBAED,IAAI,YAAY,EAAE,CAAC;oBACf,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,KAAK,CACnD,CAAC,EACD,CAAC,CACJ,CAAC;oBACF,MAAM,eAAe,GAA4B,EAAE,CAAC;oBACpD,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;wBAC3B,eAAe,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC;oBAC9C,CAAC;oBAED,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;wBAC5C,UAAU;wBACV,UAAU;wBACV,MAAM,EAAE,eAAe;qBAC1B,CAAC,CAAC;gBACP,CAAC;gBAED,OAAO,IAAI,CAAC;YAChB,CAAC;oBAAS,CAAC;gBACP,YAAY,CAAC,SAAS,CAAC,CAAC;YAC5B,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,SAAS;gBACL,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC9D,IACI,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC;gBAC3C,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;gBACvC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAC7C,CAAC;gBACC,MAAM,SAAS,CAAC;YACpB,CAAC;QACL,CAAC;IACL,CAAC;IAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACjC,WAAmB,EACnB,SAAkB,EAClB,eAAwB,KAAK;IAE7B,IAAI,CAAC,WAAW,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;IACzE,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CACvC,WAAW,EACX,SAAS,EACT,YAAY,CACf,CAAC;IAEF,MAAM,MAAM,GAAkB,EAAE,CAAC;IAEjC,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QAClB,KAAK,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7D,IAAI,CAAC,IAAI,CAAC,SAAS;gBAAE,SAAS;YAE9B,MAAM,CAAC,IAAI,CAAC;gBACR,SAAS,EAAE,IAAI,CAAC,KAAK,IAAI,QAAQ;gBACjC,KAAK,EAAE,IAAI,CAAC,WAAW,IAAI,QAAQ;gBACnC,SAAS,EAAE;oBACP,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,IAAI,CAAC;oBACxD,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS;iBACtC;aACJ,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAED,OAAO;QACH,OAAO,EAAE;YACL,SAAS;SACZ;QACD,MAAM;QACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACxB,CAAC;AACN,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,yBAAyB,CACrC,SAA4B,EAAE;IAE9B,OAAO;QACH,EAAE,EAAE,aAAa;QACjB,KAAK,CAAC,UAAU;YACZ,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YACpC,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE;gBAC7C,WAAW,EAAE,MAAM,CAAC,KAAK;aAC5B,CAAC,CAAC;YAEH,yCAAyC;YACzC,IAAI,WAAW,CAAC;YAChB,IAAI,CAAC;gBACD,WAAW,GAAG,MAAM,mBAAmB,EAAE,CAAC;gBAC1C,IAAI,YAAY,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;wBACzC,SAAS,EAAE,WAAW,CAAC,SAAS;qBACnC,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAC;gBACpD,MAAM,CAAC,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,WAAW,GAAG,MAAM,eAAe,CACrC,WAAW,CAAC,WAAW,EACvB,WAAW,CAAC,SAAS,EACrB,YAAY,CACf,CAAC;YACF,IAAI,YAAY,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE;oBAC1C,UAAU,EAAE,WAAW,CAAC,MAAM,CAAC,MAAM;iBACxC,CAAC,CAAC;YACP,CAAC;YAED,iDAAiD;YACjD,MAAM,OAAO,GAAgB,EAAE,CAAC;YAEhC,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;gBACrC,IACI,CAAC,KAAK,CAAC,SAAS;oBAChB,OAAO,KAAK,CAAC,SAAS,CAAC,iBAAiB,KAAK,QAAQ,EACvD,CAAC;oBACC,SAAS;gBACb,CAAC;gBAED,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC;gBAEnD,yEAAyE;gBACzE,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;gBACvC,IAAI,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC/D,SAAS;gBACb,CAAC;gBAED,MAAM,iBAAiB,GAAG,KAAK,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBAC5D,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CACxB,CAAC,EACD,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,iBAAiB,CAAC,GAAG,GAAG,CAAC,CAC/C,CAAC;gBAEF,yCAAyC;gBACzC,MAAM,KAAK,GAAG,UAAU,KAAK;qBACxB,WAAW,EAAE;qBACb,OAAO,CAAC,aAAa,EAAE,GAAG,CAAC;qBAC3B,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC;gBAE7B,MAAM,SAAS,GAAG,kBAAkB,CAChC,iBAAiB,EACjB,MAAM,CAAC,UAAU,CACpB,CAAC;gBAEF,IAAI,KAAyB,CAAC;gBAC9B,IAAI,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;oBAC5B,KAAK,GAAG,aAAa,kBAAkB,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;gBACnF,CAAC;gBAED,OAAO,CAAC,IAAI,CAAC;oBACT,EAAE,EAAE,KAAK;oBACT,YAAY,EAAE,eAAe,KAAK,EAAE;oBACpC,IAAI,EAAE,WAAW;oBACjB,KAAK,EAAE,GAAG;oBACV,IAAI,EAAE,GAAG;oBACT,KAAK;oBACL,IAAI,EAAE,SAAS,CAAC,IAAI,EAAE,IAAI,SAAS;iBACtC,CAAC,CAAC;YACP,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE;oBAC1C,KAAK,EAAE,OAAO,CAAC,MAAM;iBACxB,CAAC,CAAC;YACP,CAAC;YACD,OAAO,OAAO,CAAC;QACnB,CAAC;KACJ,CAAC;AACN,CAAC"}

package/dist/providers/codex.d.ts

@@ -0,0 +1,4 @@
+import { type IQuotaProvider, type QuotaData } from "../interfaces";
+export declare function extractCodexQuota(payload: unknown): QuotaData[];
+export declare function createCodexProvider(): IQuotaProvider;
+//# sourceMappingURL=codex.d.ts.map

package/dist/providers/codex.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"codex.d.ts","sourceRoot":"","sources":["../../src/providers/codex.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,cAAc,EAAE,KAAK,SAAS,EAAE,MAAM,eAAe,CAAC;AAkPpE,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,SAAS,EAAE,CAoC/D;AAED,wBAAgB,mBAAmB,IAAI,cAAc,CAqCpD"}

package/dist/providers/codex.js

@@ -0,0 +1,233 @@
+import { readFile } from "node:fs/promises";
+import { AUTH_FILE } from "../utils/paths";
+import { logger } from "../logger";
+const DEFAULT_BASE_URL = "https://chatgpt.com/backend-api";
+const REQUEST_TIMEOUT_MS = 15_000;
+const MAX_ERROR_BODY_CHARS = 2_000;
+async function readAuthFile() {
+    try {
+        const authPath = AUTH_FILE();
+        const raw = await readFile(authPath, "utf8");
+        const parsed = JSON.parse(raw);
+        return parsed;
+    }
+    catch (e) {
+        logger.debug("provider:codex:auth_read_failed", { authPath: AUTH_FILE(), error: e });
+        return null;
+    }
+}
+function pickOauthAuth(auth) {
+    const preferred = ["opencode", "codex", "openai"];
+    for (const providerID of preferred) {
+        const info = auth[providerID];
+        if (info?.type === "oauth") {
+            return {
+                providerID,
+                access: info.access,
+                enterpriseUrl: info.enterpriseUrl,
+            };
+        }
+    }
+    for (const [providerID, info] of Object.entries(auth)) {
+        if (info.type === "oauth") {
+            return {
+                providerID,
+                access: info.access,
+                enterpriseUrl: info.enterpriseUrl,
+            };
+        }
+    }
+    return null;
+}
+function buildUsageUrl(baseUrl) {
+    const trimmed = baseUrl.replace(/\/+$/, "");
+    if (trimmed.includes("/backend-api")) {
+        return `${trimmed}/wham/usage`;
+    }
+    return `${trimmed}/api/codex/usage`;
+}
+async function fetchQuotaPayload(accessToken, baseUrl) {
+    const url = buildUsageUrl(baseUrl);
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), REQUEST_TIMEOUT_MS);
+    try {
+        const response = await fetch(url, {
+            headers: {
+                Authorization: `Bearer ${accessToken}`,
+            },
+            signal: controller.signal,
+        });
+        const bodyText = await response.text();
+        let payload = null;
+        try {
+            payload = JSON.parse(bodyText);
+        }
+        catch {
+            payload = bodyText;
+        }
+        if (!response.ok) {
+            const error = new Error(`quota request failed (${response.status})`);
+            error.cause = {
+                status: response.status,
+                bodyText: bodyText.slice(0, MAX_ERROR_BODY_CHARS),
+            };
+            throw error;
+        }
+        return payload;
+    }
+    finally {
+        clearTimeout(timeout);
+    }
+}
+function isObject(value) {
+    return Boolean(value) && typeof value === "object" && !Array.isArray(value);
+}
+function toNumber(value) {
+    if (typeof value === "number" && Number.isFinite(value)) {
+        return value;
+    }
+    if (typeof value === "string") {
+        const parsed = Number.parseFloat(value);
+        if (Number.isFinite(parsed))
+            return parsed;
+    }
+    return null;
+}
+function describeWindow(windowSeconds) {
+    if (!windowSeconds || windowSeconds <= 0)
+        return null;
+    const minutes = Math.round(windowSeconds / 60);
+    if (minutes >= 60 && minutes % 60 === 0) {
+        const hours = minutes / 60;
+        return `${hours}h window`;
+    }
+    return `${minutes}m window`;
+}
+function formatRelativeSeconds(seconds) {
+    if (seconds <= 0)
+        return "now";
+    const minutes = Math.floor(seconds / 60);
+    const hours = Math.floor(minutes / 60);
+    const remainingMinutes = minutes % 60;
+    if (hours > 0) {
+        return `${hours}h ${remainingMinutes}m`;
+    }
+    return `${minutes}m`;
+}
+function parseRateLimitWindow(id, label, snapshot) {
+    const usedPercent = toNumber(snapshot.used_percent);
+    if (usedPercent === null)
+        return null;
+    // Window info
+    let window;
+    const windowSeconds = toNumber(snapshot.limit_window_seconds);
+    const windowLabel = describeWindow(windowSeconds);
+    if (windowLabel)
+        window = windowLabel;
+    // Reset info
+    let reset;
+    const resetAfter = toNumber(snapshot.reset_after_seconds);
+    const resetAt = toNumber(snapshot.reset_at);
+    if (resetAfter !== null) {
+        reset = `resets in ${formatRelativeSeconds(resetAfter)}`;
+    }
+    else if (resetAt !== null) {
+        reset = `resets at ${new Date(resetAt * 1000).toLocaleTimeString()}`;
+    }
+    return {
+        id: `codex-${id}`,
+        providerName: `Codex ${label}`,
+        used: Math.max(0, Math.min(100, usedPercent)),
+        limit: 100,
+        unit: "%",
+        window,
+        reset,
+    };
+}
+function parseCredits(credits) {
+    const base = {
+        id: "codex-credits",
+        providerName: "Codex Credits",
+        unit: "credits",
+    };
+    if (credits.unlimited) {
+        return {
+            ...base,
+            used: 0,
+            limit: null,
+            info: "unlimited",
+        };
+    }
+    const balance = toNumber(credits.balance ?? null);
+    if (balance === null)
+        return null;
+    return {
+        ...base,
+        used: balance,
+        limit: null,
+        info: "balance",
+    };
+}
+export function extractCodexQuota(payload) {
+    if (!isObject(payload))
+        return [];
+    const rateLimitCandidate = payload["rate_limit"];
+    const rateLimit = isObject(rateLimitCandidate) ? rateLimitCandidate : null;
+    const entries = [];
+    if (rateLimit) {
+        const primary = isObject(rateLimit.primary_window)
+            ? rateLimit.primary_window
+            : null;
+        const secondary = isObject(rateLimit.secondary_window)
+            ? rateLimit.secondary_window
+            : null;
+        if (primary) {
+            const entry = parseRateLimitWindow("primary", "Primary", primary);
+            if (entry)
+                entries.push(entry);
+        }
+        if (secondary) {
+            const entry = parseRateLimitWindow("secondary", "Secondary", secondary);
+            if (entry)
+                entries.push(entry);
+        }
+    }
+    const creditsCandidate = payload["credits"];
+    const credits = isObject(creditsCandidate) ? creditsCandidate : null;
+    if (credits) {
+        const creditEntry = parseCredits(credits);
+        if (creditEntry)
+            entries.push(creditEntry);
+    }
+    return entries;
+}
+export function createCodexProvider() {
+    return {
+        id: "codex",
+        async fetchQuota() {
+            logger.debug("provider:codex:fetch_start", { authPath: AUTH_FILE() });
+            const auth = await readAuthFile();
+            if (!auth) {
+                logger.debug("provider:codex:no_auth", { authPath: AUTH_FILE() });
+                throw new Error("Codex auth.json not found");
+            }
+            const oauth = pickOauthAuth(auth);
+            if (!oauth) {
+                logger.debug("provider:codex:no_oauth", { availableProviders: Object.keys(auth) });
+                throw new Error("Codex OAuth credentials missing");
+            }
+            const baseUrl = process.env.OPENCODE_CODEX_BASE_URL ??
+                oauth.enterpriseUrl ??
+                DEFAULT_BASE_URL;
+            logger.debug("provider:codex:request", { providerID: oauth.providerID, baseUrl, url: buildUsageUrl(baseUrl) });
+            const payload = await fetchQuotaPayload(oauth.access, baseUrl);
+            const entries = extractCodexQuota(payload);
+            logger.debug("provider:codex:parse", { count: entries.length });
+            if (entries.length === 0) {
+                throw new Error("Codex quota payload did not include rate limits");
+            }
+            return entries;
+        },
+    };
+}
+//# sourceMappingURL=codex.js.map

package/dist/providers/codex.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codex.js","sourceRoot":"","sources":["../../src/providers/codex.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAE3C,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAGnC,MAAM,gBAAgB,GAAG,iCAAiC,CAAC;AAC3D,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAClC,MAAM,oBAAoB,GAAG,KAAK,CAAC;AAsDnC,KAAK,UAAU,YAAY;IACvB,IAAI,CAAC;QACD,MAAM,QAAQ,GAAG,SAAS,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAa,CAAC;QAC3C,OAAO,MAAM,CAAC;IAClB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACT,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACrF,OAAO,IAAI,CAAC;IAChB,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,IAAc;IACnC,MAAM,SAAS,GAAG,CAAC,UAAU,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;IAClD,KAAK,MAAM,UAAU,IAAI,SAAS,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9B,IAAI,IAAI,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;YAC3B,OAAO;gBACL,UAAU;gBACV,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,aAAa,EAAE,IAAI,CAAC,aAAa;aAClC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACtD,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC1B,OAAO;gBACL,UAAU;gBACV,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,aAAa,EAAE,IAAI,CAAC,aAAa;aAClC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,aAAa,CAAC,OAAe;IACpC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC5C,IAAI,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QACrC,OAAO,GAAG,OAAO,aAAa,CAAC;IACjC,CAAC;IACD,OAAO,GAAG,OAAO,kBAAkB,CAAC;AACtC,CAAC;AAED,KAAK,UAAU,iBAAiB,CAC9B,WAAmB,EACnB,OAAe;IAEf,MAAM,GAAG,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACnC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,kBAAkB,CAAC,CAAC;IACzE,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,WAAW,EAAE;aACvC;YACD,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACvC,IAAI,OAAO,GAAY,IAAI,CAAC;QAC5B,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,GAAG,QAAQ,CAAC;QACrB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YACrE,KAAK,CAAC,KAAK,GAAG;gBACZ,MAAM,EAAE,QAAQ,CAAC,MAAM;gBACvB,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,oBAAoB,CAAC;aAClD,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;AACH,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACxD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,MAAM,CAAC;IAC7C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,cAAc,CAAC,aAA4B;IAClD,IAAI,CAAC,aAAa,IAAI,aAAa,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACtD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,EAAE,CAAC,CAAC;IAC/C,IAAI,OAAO,IAAI,EAAE,IAAI,OAAO,GAAG,EAAE,KAAK,CAAC,EAAE,CAAC;QACxC,MAAM,KAAK,GAAG,OAAO,GAAG,EAAE,CAAC;QAC3B,OAAO,GAAG,KAAK,UAAU,CAAC;IAC5B,CAAC;IACD,OAAO,GAAG,OAAO,UAAU,CAAC;AAC9B,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAe;IAC5C,IAAI,OAAO,IAAI,CAAC;QAAE,OAAO,KAAK,CAAC;IAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;IACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;IACvC,MAAM,gBAAgB,GAAG,OAAO,GAAG,EAAE,CAAC;IACtC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QACd,OAAO,GAAG,KAAK,KAAK,gBAAgB,GAAG,CAAC;IAC1C,CAAC;IACD,OAAO,GAAG,OAAO,GAAG,CAAC;AACvB,CAAC;AAED,SAAS,oBAAoB,CAC3B,EAAU,EACV,KAAa,EACb,QAAiC;IAEjC,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IACpD,IAAI,WAAW,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAEtC,cAAc;IACd,IAAI,MAA0B,CAAC;IAC/B,MAAM,aAAa,GAAG,QAAQ,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC;IAC9D,MAAM,WAAW,GAAG,cAAc,CAAC,aAAa,CAAC,CAAC;IAClD,IAAI,WAAW;QAAE,MAAM,GAAG,WAAW,CAAC;IAEtC,aAAa;IACb,IAAI,KAAyB,CAAC;IAC9B,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;IAC1D,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC5C,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;QACxB,KAAK,GAAG,aAAa,qBAAqB,CAAC,UAAU,CAAC,EAAE,CAAC;IAC3D,CAAC;SAAM,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QAC5B,KAAK,GAAG,aAAa,IAAI,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,kBAAkB,EAAE,EAAE,CAAC;IACvE,CAAC;IAED,OAAO;QACL,EAAE,EAAE,SAAS,EAAE,EAAE;QACjB,YAAY,EAAE,SAAS,KAAK,EAAE;QAC9B,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;QAC7C,KAAK,EAAE,GAAG;QACV,IAAI,EAAE,GAAG;QACT,MAAM;QACN,KAAK;KACN,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,OAA4B;IAChD,MAAM,IAAI,GAAG;QACX,EAAE,EAAE,eAAe;QACnB,YAAY,EAAE,eAAe;QAC7B,IAAI,EAAE,SAAS;KAChB,CAAC;IAEF,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QACtB,OAAO;YACL,GAAG,IAAI;YACP,IAAI,EAAE,CAAC;YACP,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,WAAW;SAClB,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC;IAClD,IAAI,OAAO,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAElC,OAAO;QACL,GAAG,IAAI;QACP,IAAI,EAAE,OAAO;QACb,KAAK,EAAE,IAAI;QACX,IAAI,EAAE,SAAS;KAChB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,OAAgB;IAChD,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IAElC,MAAM,kBAAkB,GAAI,OAAmC,CAAC,YAAY,CAAC,CAAC;IAC9E,MAAM,SAAS,GAAG,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC;IAE3E,MAAM,OAAO,GAAgB,EAAE,CAAC;IAEhC,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,cAAc,CAAC;YAChD,CAAC,CAAE,SAAS,CAAC,cAA0C;YACvD,CAAC,CAAC,IAAI,CAAC;QACT,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC,gBAAgB,CAAC;YACpD,CAAC,CAAE,SAAS,CAAC,gBAA4C;YACzD,CAAC,CAAC,IAAI,CAAC;QAET,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,KAAK,GAAG,oBAAoB,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YAClE,IAAI,KAAK;gBAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjC,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,KAAK,GAAG,oBAAoB,CAAC,WAAW,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;YACxE,IAAI,KAAK;gBAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED,MAAM,gBAAgB,GAAI,OAAmC,CAAC,SAAS,CAAC,CAAC;IACzE,MAAM,OAAO,GAAG,QAAQ,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC;IAErE,IAAI,OAAO,EAAE,CAAC;QACZ,MAAM,WAAW,GAAG,YAAY,CAAC,OAA8B,CAAC,CAAC;QACjE,IAAI,WAAW;YAAE,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC7C,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,OAAO;QACL,EAAE,EAAE,OAAO;QACX,KAAK,CAAC,UAAU;YACd,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;YAEtE,MAAM,IAAI,GAAG,MAAM,YAAY,EAAE,CAAC;YAClC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACN,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;gBAClE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACjD,CAAC;YAEH,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,EAAE,kBAAkB,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACnF,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACrD,CAAC;YAED,MAAM,OAAO,GACX,OAAO,CAAC,GAAG,CAAC,uBAAuB;gBACnC,KAAK,CAAC,aAAa;gBACnB,gBAAgB,CAAC;YAEjB,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,OAAO,EAAE,GAAG,EAAE,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAEjH,MAAM,OAAO,GAAG,MAAM,iBAAiB,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC/D,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAE3C,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAEhE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACrE,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/providers/github.d.ts

@@ -0,0 +1,4 @@
+import { type IQuotaProvider, type QuotaData } from "../interfaces";
+export declare function parseGithubUsage(data: unknown, sku: string | null, apiWarning?: string | null): QuotaData[];
+export declare function createGithubProvider(): IQuotaProvider;
+//# sourceMappingURL=github.d.ts.map

package/dist/providers/github.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"github.d.ts","sourceRoot":"","sources":["../../src/providers/github.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,cAAc,EAAE,KAAK,SAAS,EAAE,MAAM,eAAe,CAAC;AAsDpE,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,GAAG,IAAI,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,EAAE,CAoD3G;AAED,wBAAgB,oBAAoB,IAAI,cAAc,CAiDrD"}

package/dist/providers/github.js

@@ -0,0 +1,139 @@
+import { readFile } from "node:fs/promises";
+import { AUTH_FILE, getConfigDirectory } from "../utils/paths";
+import { join } from "node:path";
+const AUTH_PATH_LOCAL = AUTH_FILE();
+const AUTH_PATH_CONFIG = join(getConfigDirectory(), "auth.json");
+async function readAuthFile() {
+    for (const path of [AUTH_PATH_LOCAL, AUTH_PATH_CONFIG]) {
+        try {
+            const raw = await readFile(path, "utf8");
+            return JSON.parse(raw);
+        }
+        catch {
+            continue;
+        }
+    }
+    return null;
+}
+function parseTokenSku(token) {
+    const parts = token.split(";");
+    for (const part of parts) {
+        if (part.startsWith("sku=")) {
+            return part.split("=")[1];
+        }
+    }
+    return null;
+}
+function getNextMonthStart() {
+    const now = new Date();
+    const nextMonth = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth() + 1, 1, 0, 0, 0));
+    return nextMonth;
+}
+function formatTimeUntil(target) {
+    const diff = target.getTime() - Date.now();
+    if (diff <= 0)
+        return "soon";
+    const days = Math.floor(diff / (1000 * 60 * 60 * 24));
+    const hours = Math.floor((diff % (1000 * 60 * 60 * 24)) / (1000 * 60 * 60));
+    if (days > 0)
+        return `${days}d ${hours}h`;
+    return `${hours}h`;
+}
+export function parseGithubUsage(data, sku, apiWarning) {
+    // free_engaged_oss_quota is actually a "pro" equivalent for OSS maintainers
+    const isFreeLimited = sku?.includes("free") && !sku?.includes("oss");
+    const now = new Date();
+    const resetTime = getNextMonthStart();
+    const resetStr = `resets in ${formatTimeUntil(resetTime)}`;
+    let usedSuggestions = 0;
+    let limit = isFreeLimited ? 2000 : null;
+    let unit = "suggestions";
+    if (Array.isArray(data) && data.length > 0) {
+        const currentMonthUsage = data.filter((day) => {
+            if (!day || typeof day !== "object")
+                return false;
+            const dayRecord = day;
+            const dayValue = dayRecord["day"];
+            if (typeof dayValue !== "string")
+                return false;
+            const dayDate = new Date(dayValue);
+            return dayDate.getUTCMonth() === now.getUTCMonth() && dayDate.getUTCFullYear() === now.getUTCFullYear();
+        });
+        usedSuggestions = currentMonthUsage.reduce((acc, day) => {
+            if (!day || typeof day !== "object")
+                return acc;
+            const dayRecord = day;
+            const suggestions = typeof dayRecord["total_suggestions_count"] === "number" ? dayRecord["total_suggestions_count"] : 0;
+            const chat = typeof dayRecord["total_chat_count"] === "number" ? dayRecord["total_chat_count"] : 0;
+            return acc + suggestions + chat;
+        }, 0);
+    }
+    const infoParts = [];
+    if (isFreeLimited)
+        infoParts.push("Free Plan");
+    else if (sku)
+        infoParts.push("Pro Plan");
+    if (apiWarning) {
+        infoParts.push("Service Currently Unavailable (API Deprecated)");
+    }
+    return [
+        {
+            id: "github-copilot",
+            providerName: "GitHub Copilot",
+            used: usedSuggestions,
+            limit: limit,
+            unit: unit,
+            reset: resetStr,
+            window: "Monthly",
+            info: infoParts.join(" | "),
+        }
+    ];
+}
+export function createGithubProvider() {
+    return {
+        id: "github-copilot",
+        async fetchQuota() {
+            const auth = await readAuthFile();
+            if (!auth) {
+                throw new Error("Opencode auth.json not found");
+            }
+            const info = auth["github-copilot"] || auth["github"];
+            if (!info || !info.access) {
+                throw new Error("GitHub Copilot credentials missing");
+            }
+            const sku = parseTokenSku(info.access);
+            // Note: GitHub does not currently support a user-level Copilot usage
+            // endpoint for individual accounts. The legacy/beta endpoints were
+            // deprecated (and may return 404). We still attempt the call, but we
+            // surface failures as provider metadata so users can diagnose.
+            let data = null;
+            let apiWarning = null;
+            try {
+                const response = await fetch("https://api.github.com/user/copilot/usage", {
+                    headers: {
+                        Authorization: `Bearer ${info.access}`,
+                        "X-GitHub-Api-Version": "2022-11-28",
+                        Accept: "application/vnd.github+json",
+                    },
+                });
+                if (!response.ok) {
+                    if (response.status === 404) {
+                        apiWarning = "404";
+                    }
+                    else {
+                        const body = await response.text();
+                        apiWarning = `${response.status}: ${body.slice(0, 50)}`;
+                    }
+                }
+                else {
+                    data = (await response.json());
+                }
+            }
+            catch (error) {
+                apiWarning = "Request Failed";
+            }
+            return parseGithubUsage(data, sku, apiWarning);
+        },
+    };
+}
+//# sourceMappingURL=github.js.map

package/dist/providers/github.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"github.js","sourceRoot":"","sources":["../../src/providers/github.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAC/D,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAGjC,MAAM,eAAe,GAAG,SAAS,EAAE,CAAC;AACpC,MAAM,gBAAgB,GAAG,IAAI,CAAC,kBAAkB,EAAE,EAAE,WAAW,CAAC,CAAC;AAWjE,KAAK,UAAU,YAAY;IACvB,KAAK,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,gBAAgB,CAAC,EAAE,CAAC;QACrD,IAAI,CAAC;YACD,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACzC,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAa,CAAC;QACvC,CAAC;QAAC,MAAM,CAAC;YACL,SAAS;QACb,CAAC;IACL,CAAC;IAED,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAa;IAChC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC;IACL,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,iBAAiB;IACtB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,GAAG,CAAC,WAAW,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC9F,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,eAAe,CAAC,MAAY;IACjC,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC3C,IAAI,IAAI,IAAI,CAAC;QAAE,OAAO,MAAM,CAAC;IAE7B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;IACtD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;IAE5E,IAAI,IAAI,GAAG,CAAC;QAAE,OAAO,GAAG,IAAI,KAAK,KAAK,GAAG,CAAC;IAC1C,OAAO,GAAG,KAAK,GAAG,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,IAAa,EAAE,GAAkB,EAAE,UAA0B;IAC1F,4EAA4E;IAC5E,MAAM,aAAa,GAAG,GAAG,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;IAErE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,MAAM,SAAS,GAAG,iBAAiB,EAAE,CAAC;IACtC,MAAM,QAAQ,GAAG,aAAa,eAAe,CAAC,SAAS,CAAC,EAAE,CAAC;IAE3D,IAAI,eAAe,GAAG,CAAC,CAAC;IACxB,IAAI,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IACxC,IAAI,IAAI,GAAG,aAAa,CAAC;IAEzB,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,MAAM,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAY,EAAE,EAAE;YACnD,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;gBAAE,OAAO,KAAK,CAAC;YAClD,MAAM,SAAS,GAAG,GAA8B,CAAC;YACjD,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,OAAO,QAAQ,KAAK,QAAQ;gBAAE,OAAO,KAAK,CAAC;YAE/C,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC;YACnC,OAAO,OAAO,CAAC,WAAW,EAAE,KAAK,GAAG,CAAC,WAAW,EAAE,IAAI,OAAO,CAAC,cAAc,EAAE,KAAK,GAAG,CAAC,cAAc,EAAE,CAAC;QAC5G,CAAC,CAAC,CAAC;QAEH,eAAe,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC,GAAW,EAAE,GAAY,EAAE,EAAE;YACrE,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;gBAAE,OAAO,GAAG,CAAC;YAChD,MAAM,SAAS,GAAG,GAA8B,CAAC;YACjD,MAAM,WAAW,GAAG,OAAO,SAAS,CAAC,yBAAyB,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACxH,MAAM,IAAI,GAAG,OAAO,SAAS,CAAC,kBAAkB,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACnG,OAAO,GAAG,GAAG,WAAW,GAAG,IAAI,CAAC;QACpC,CAAC,EAAE,CAAC,CAAC,CAAC;IACV,CAAC;IAED,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,IAAI,aAAa;QAAE,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;SAC1C,IAAI,GAAG;QAAE,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAEzC,IAAI,UAAU,EAAE,CAAC;QACb,SAAS,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;IACrE,CAAC;IAED,OAAO;QACH;YACI,EAAE,EAAE,gBAAgB;YACpB,YAAY,EAAE,gBAAgB;YAC9B,IAAI,EAAE,eAAe;YACrB,KAAK,EAAE,KAAK;YACZ,IAAI,EAAE,IAAI;YACV,KAAK,EAAE,QAAQ;YACf,MAAM,EAAE,SAAS;YACjB,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC;SAC9B;KACJ,CAAC;AACN,CAAC;AAED,MAAM,UAAU,oBAAoB;IAChC,OAAO;QACH,EAAE,EAAE,gBAAgB;QACpB,KAAK,CAAC,UAAU;YACZ,MAAM,IAAI,GAAG,MAAM,YAAY,EAAE,CAAC;YAClC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;YACpD,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC;YACtD,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;YAC1D,CAAC;YAED,MAAM,GAAG,GAAG,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEvC,qEAAqE;YACrE,mEAAmE;YACnE,qEAAqE;YACrE,+DAA+D;YAC/D,IAAI,IAAI,GAAY,IAAI,CAAC;YACzB,IAAI,UAAU,GAAkB,IAAI,CAAC;YAErC,IAAI,CAAC;gBACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,2CAA2C,EAAE;oBACtE,OAAO,EAAE;wBACL,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;wBACtC,sBAAsB,EAAE,YAAY;wBACpC,MAAM,EAAE,6BAA6B;qBACxC;iBACJ,CAAC,CAAC;gBAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACf,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;wBAC1B,UAAU,GAAG,KAAK,CAAC;oBACvB,CAAC;yBAAM,CAAC;wBACJ,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;wBACnC,UAAU,GAAG,GAAG,QAAQ,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;oBAC5D,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACJ,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAY,CAAC;gBAC9C,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,UAAU,GAAG,gBAAgB,CAAC;YAClC,CAAC;YAED,OAAO,gBAAgB,CAAC,IAAI,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QACnD,CAAC;KACJ,CAAC;AACN,CAAC"}