opencode-quotas 0.0.1

package/dist/src/providers/antigravity/auth.js

@@ -0,0 +1,109 @@
+import { readFile } from "node:fs/promises";
+import { ANTIGRAVITY_ACCOUNTS_FILE } from "../../utils/paths";
+/**
+ * PUBLIC OAUTH CREDENTIALS - INTENTIONALLY COMMITTED
+ *
+ * These are "Installed Application" credentials for Google's Native App OAuth flow.
+ * Per Google's documentation, the client_secret for native applications is NOT
+ * considered confidential. Security relies solely on the user's refresh_token
+ * stored locally in ~/.config/opencode/antigravity-accounts.json.
+ *
+ * See: https://developers.google.com/identity/protocols/oauth2/native-app
+ */
+const ANTIGRAVITY_CLIENT_ID = "1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com"; // gitleaks:allow
+const ANTIGRAVITY_CLIENT_SECRET = "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"; // gitleaks:allow
+const TOKEN_URL = "https://oauth2.googleapis.com/token";
+let cachedCredential = null;
+function getAccountsFilePath() {
+    return ANTIGRAVITY_ACCOUNTS_FILE();
+}
+async function loadAccounts() {
+    const accountsPath = getAccountsFilePath();
+    try {
+        const content = await readFile(accountsPath, "utf-8");
+        const data = JSON.parse(content);
+        if (!data.accounts || data.accounts.length === 0) {
+            throw new Error("No accounts found in antigravity-accounts.json");
+        }
+        return data;
+    }
+    catch (error) {
+        if (error.code === "ENOENT") {
+            throw new Error("Antigravity accounts file not found.\n" +
+                "Run 'opencode auth login' first to authenticate with Google.");
+        }
+        throw error;
+    }
+}
+export async function hasCloudCredentials() {
+    try {
+        await loadAccounts();
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+async function refreshAccessToken(refreshToken) {
+    const response = await fetch(TOKEN_URL, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/x-www-form-urlencoded",
+        },
+        body: new URLSearchParams({
+            client_id: ANTIGRAVITY_CLIENT_ID,
+            client_secret: ANTIGRAVITY_CLIENT_SECRET,
+            refresh_token: refreshToken,
+            grant_type: "refresh_token",
+        }).toString(),
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        if (errorText.toLowerCase().includes("invalid_grant")) {
+            throw new Error("Refresh token is invalid or expired. Run 'opencode auth login' to re-authenticate.");
+        }
+        throw new Error(`Token refresh failed: ${response.status} - ${errorText}`);
+    }
+    const data = (await response.json());
+    return {
+        accessToken: data.access_token,
+        expiresAt: Date.now() + data.expires_in * 1000,
+    };
+}
+export async function getCloudCredentials() {
+    const accountsFile = await loadAccounts();
+    const activeAccount = accountsFile.accounts[accountsFile.activeIndex] ?? accountsFile.accounts[0];
+    if (!activeAccount) {
+        throw new Error("No active account found in antigravity-accounts.json");
+    }
+    // Check cache (5 min buffer)
+    const fiveMinutesInMs = 5 * 60 * 1000;
+    if (cachedCredential &&
+        cachedCredential.email === activeAccount.email &&
+        cachedCredential.expiresAt > Date.now() + fiveMinutesInMs) {
+        return {
+            accessToken: cachedCredential.accessToken,
+            projectId: cachedCredential.projectId,
+            email: cachedCredential.email,
+        };
+    }
+    const { accessToken, expiresAt } = await refreshAccessToken(activeAccount.refreshToken);
+    cachedCredential = {
+        accessToken,
+        projectId: activeAccount.projectId,
+        email: activeAccount.email,
+        expiresAt,
+    };
+    return {
+        accessToken,
+        projectId: activeAccount.projectId,
+        email: activeAccount.email,
+    };
+}
+/**
+ * Reset the credential cache. Internal use only (primarily for tests).
+ */
+export function resetCredentialCache() {
+    cachedCredential = null;
+}
+//# sourceMappingURL=auth.js.map

package/dist/src/providers/antigravity/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","sourceRoot":"","sources":["../../../../src/providers/antigravity/auth.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAE9D;;;;;;;;;GASG;AACH,MAAM,qBAAqB,GACzB,2EAA2E,CAAC,CAAC,iBAAiB;AAChG,MAAM,yBAAyB,GAAG,qCAAqC,CAAC,CAAC,iBAAiB;AAC1F,MAAM,SAAS,GAAG,qCAAqC,CAAC;AAiCxD,IAAI,gBAAgB,GAA4B,IAAI,CAAC;AAErD,SAAS,mBAAmB;IAC1B,OAAO,yBAAyB,EAAE,CAAC;AACrC,CAAC;AAED,KAAK,UAAU,YAAY;IACzB,MAAM,YAAY,GAAG,mBAAmB,EAAE,CAAC;IAE3C,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAiB,CAAC;QAEjD,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACvD,MAAM,IAAI,KAAK,CACb,wCAAwC;gBACtC,8DAA8D,CACjE,CAAC;QACJ,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB;IACvC,IAAI,CAAC;QACH,MAAM,YAAY,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,YAAoB;IACpD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE;QACtC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,mCAAmC;SACpD;QACD,IAAI,EAAE,IAAI,eAAe,CAAC;YACxB,SAAS,EAAE,qBAAqB;YAChC,aAAa,EAAE,yBAAyB;YACxC,aAAa,EAAE,YAAY;YAC3B,UAAU,EAAE,eAAe;SAC5B,CAAC,CAAC,QAAQ,EAAE;KACd,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxC,IAAI,SAAS,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACtD,MAAM,IAAI,KAAK,CACb,oFAAoF,CACrF,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAkB,CAAC;IACtD,OAAO;QACL,WAAW,EAAE,IAAI,CAAC,YAAY;QAC9B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,UAAU,GAAG,IAAI;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB;IACvC,MAAM,YAAY,GAAG,MAAM,YAAY,EAAE,CAAC;IAC1C,MAAM,aAAa,GACjB,YAAY,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAE9E,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;IAC1E,CAAC;IAED,6BAA6B;IAC7B,MAAM,eAAe,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;IACtC,IACE,gBAAgB;QAChB,gBAAgB,CAAC,KAAK,KAAK,aAAa,CAAC,KAAK;QAC9C,gBAAgB,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,eAAe,EACzD,CAAC;QACD,OAAO;YACL,WAAW,EAAE,gBAAgB,CAAC,WAAW;YACzC,SAAS,EAAE,gBAAgB,CAAC,SAAS;YACrC,KAAK,EAAE,gBAAgB,CAAC,KAAK;SAC9B,CAAC;IACJ,CAAC;IAED,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,GAAG,MAAM,kBAAkB,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;IAExF,gBAAgB,GAAG;QACjB,WAAW;QACX,SAAS,EAAE,aAAa,CAAC,SAAS;QAClC,KAAK,EAAE,aAAa,CAAC,KAAK;QAC1B,SAAS;KACV,CAAC;IAEF,OAAO;QACL,WAAW;QACX,SAAS,EAAE,aAAa,CAAC,SAAS;QAClC,KAAK,EAAE,aAAa,CAAC,KAAK;KAC3B,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAChC,gBAAgB,GAAG,IAAI,CAAC;AAC5B,CAAC"}

package/dist/src/providers/antigravity/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./provider";
+//# sourceMappingURL=index.d.ts.map

package/dist/src/providers/antigravity/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/providers/antigravity/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC"}

package/dist/src/providers/antigravity/index.js

@@ -0,0 +1,2 @@
+export * from "./provider";
+//# sourceMappingURL=index.js.map

package/dist/src/providers/antigravity/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/providers/antigravity/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC"}

package/dist/src/providers/antigravity/provider.d.ts

@@ -0,0 +1,34 @@
+import { type IQuotaProvider } from "../../interfaces";
+export interface QuotaIndicator {
+    threshold: number;
+    symbol: string;
+}
+export interface AntigravityConfig {
+    indicators?: QuotaIndicator[];
+    debug?: boolean;
+}
+export interface QuotaInfo {
+    remainingFraction: number;
+    resetTime?: string;
+}
+export interface ModelConfig {
+    modelName: string;
+    label?: string;
+    quotaInfo?: QuotaInfo;
+}
+export interface CloudAccountInfo {
+    email?: string;
+    projectId?: string;
+}
+export interface CloudQuotaResult {
+    account: CloudAccountInfo;
+    models: ModelConfig[];
+    timestamp: number;
+}
+export declare function fetchCloudQuota(accessToken: string, projectId?: string, debugEnabled?: boolean): Promise<CloudQuotaResult>;
+/**
+ * Creates the Antigravity provider that returns flat, raw quota data.
+ * Grouping and aggregation is handled by the service layer via AggregatedGroups.
+ */
+export declare function createAntigravityProvider(config?: AntigravityConfig): IQuotaProvider;
+//# sourceMappingURL=provider.d.ts.map

package/dist/src/providers/antigravity/provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../../../src/providers/antigravity/provider.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,cAAc,EAAkB,MAAM,kBAAkB,CAAC;AAiBvE,MAAM,WAAW,cAAc;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,iBAAiB;IAC9B,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;IAC9B,KAAK,CAAC,EAAE,OAAO,CAAC;CACnB;AAoBD,MAAM,WAAW,SAAS;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,WAAW;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,SAAS,CAAC;CACzB;AAED,MAAM,WAAW,gBAAgB;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC7B,OAAO,EAAE,gBAAgB,CAAC;IAC1B,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACrB;AAsID,wBAAsB,eAAe,CACjC,WAAW,EAAE,MAAM,EACnB,SAAS,CAAC,EAAE,MAAM,EAClB,YAAY,GAAE,OAAe,GAC9B,OAAO,CAAC,gBAAgB,CAAC,CAmC3B;AAED;;;GAGG;AACH,wBAAgB,yBAAyB,CACrC,MAAM,GAAE,iBAAsB,GAC/B,cAAc,CAwFhB"}

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

@@ -0,0 +1,216 @@
+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 || "";
+                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/src/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;gBACnD,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/src/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/src/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;AAyPpE,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,SAAS,EAAE,CAoC/D;AAED,wBAAgB,mBAAmB,IAAI,cAAc,CAqCpD"}

package/dist/src/providers/codex.js

@@ -0,0 +1,242 @@
+import { readFile } from "node:fs/promises";
+import { AUTH_FILE } from "../utils/paths";
+import { logger } from "../logger";
+const AUTH_PATH = AUTH_FILE();
+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 raw = await readFile(AUTH_PATH, "utf8");
+        const parsed = JSON.parse(raw);
+        return parsed;
+    }
+    catch (e) {
+        logger.debug("provider:codex:auth_read_failed", { authPath: AUTH_PATH, error: e });
+        // Fallback: try the config directory location
+        try {
+            const configPath = AUTH_PATH.replace("auth.json", "antigravity-accounts.json");
+            // If the replacement is not valid, this will likely fail and we return null
+            const rawConfig = await readFile(configPath, "utf8");
+            return JSON.parse(rawConfig);
+        }
+        catch {
+            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_PATH });
+            const auth = await readAuthFile();
+            if (!auth) {
+                logger.debug("provider:codex:no_auth", { authPath: AUTH_PATH });
+                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/src/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;AAEnC,MAAM,SAAS,GAAG,SAAS,EAAE,CAAC;AAC9B,MAAM,gBAAgB,GAAG,iCAAiC,CAAC;AAC3D,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAClC,MAAM,oBAAoB,GAAG,KAAK,CAAC;AAsDnC,KAAK,UAAU,YAAY;IACzB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAa,CAAC;QAC3C,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACnF,8CAA8C;QAC9C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,2BAA2B,CAAC,CAAC;YAC/E,4EAA4E;YAC5E,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YACrD,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAa,CAAC;QAC3C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;AACH,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,CAAC,CAAC;YAEpE,MAAM,IAAI,GAAG,MAAM,YAAY,EAAE,CAAC;YAClC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;gBAChE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;YAED,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"}