ag-quota 0.0.2

package/dist/cloud.d.ts

@@ -0,0 +1,40 @@
+/**
+ * Unified quota info structure (shared between cloud and local)
+ */
+export interface QuotaInfo {
+    remainingFraction: number;
+    resetTime?: string;
+}
+/**
+ * Unified model config structure (shared between cloud and local)
+ */
+export interface ModelConfig {
+    modelName: string;
+    label?: string;
+    quotaInfo?: QuotaInfo;
+}
+/**
+ * Cloud-specific account info
+ */
+export interface CloudAccountInfo {
+    email?: string;
+    projectId?: string;
+}
+/**
+ * Result from cloud quota fetch
+ */
+export interface CloudQuotaResult {
+    account: CloudAccountInfo;
+    models: ModelConfig[];
+    timestamp: number;
+}
+/**
+ * Fetch quota information from the Cloud Code API.
+ *
+ * @param accessToken - Valid OAuth2 access token
+ * @param projectId - Optional Google Cloud project ID
+ * @returns CloudQuotaResult with account info and model quotas
+ * @throws Error if API fails
+ */
+export declare function fetchCloudQuota(accessToken: string, projectId?: string): Promise<CloudQuotaResult>;
+//# sourceMappingURL=cloud.d.ts.map

package/dist/cloud.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloud.d.ts","sourceRoot":"","sources":["../src/cloud.ts"],"names":[],"mappings":"AAkDA;;GAEG;AACH,MAAM,WAAW,SAAS;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,SAAS,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC7B,OAAO,EAAE,gBAAgB,CAAC;IAC1B,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACrB;AAoED;;;;;;;GAOG;AACH,wBAAsB,eAAe,CACjC,WAAW,EAAE,MAAM,EACnB,SAAS,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,gBAAgB,CAAC,CAiC3B"}

package/dist/cloud.js

@@ -0,0 +1,110 @@
+/*
+ * ISC License
+ * Copyright (c) 2025, Cristian Militaru
+ * Copyright (c) 2026, Philipp
+ *
+ * Cloud quota fetching via Google Cloud Code API.
+ */
+// ============================================================================
+// Constants from opencode-antigravity-auth
+// ============================================================================
+// Endpoint fallback order (daily → autopush → prod)
+const CLOUDCODE_ENDPOINTS = [
+    "https://daily-cloudcode-pa.sandbox.googleapis.com",
+    "https://autopush-cloudcode-pa.sandbox.googleapis.com",
+    "https://cloudcode-pa.googleapis.com",
+];
+// Headers matching opencode-antigravity-auth
+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"}',
+};
+// ============================================================================
+// API Fetching
+// ============================================================================
+/**
+ * Fetch available models with quota from the Cloud Code API.
+ */
+async function fetchAvailableModels(accessToken, projectId) {
+    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 response = await fetch(url, {
+                method: "POST",
+                headers,
+                body: JSON.stringify(payload),
+            });
+            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();
+                throw new Error(`Cloud Code API error ${response.status}: ${text.slice(0, 200)}`);
+            }
+            return (await response.json());
+        }
+        catch (error) {
+            lastError = error instanceof Error ? error : new Error(String(error));
+            // Auth errors should not fallback to other endpoints
+            if (lastError.message.includes("Authorization") ||
+                lastError.message.includes("forbidden") ||
+                lastError.message.includes("invalid_grant")) {
+                throw lastError;
+            }
+            // Try next endpoint
+        }
+    }
+    throw lastError || new Error("All Cloud Code API endpoints failed");
+}
+// ============================================================================
+// Main Export
+// ============================================================================
+/**
+ * Fetch quota information from the Cloud Code API.
+ *
+ * @param accessToken - Valid OAuth2 access token
+ * @param projectId - Optional Google Cloud project ID
+ * @returns CloudQuotaResult with account info and model quotas
+ * @throws Error if API fails
+ */
+export async function fetchCloudQuota(accessToken, projectId) {
+    if (!accessToken) {
+        throw new Error("Access token is required for cloud quota fetching");
+    }
+    // Fetch quota data
+    const response = await fetchAvailableModels(accessToken, projectId);
+    // Convert to unified format
+    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(),
+    };
+}

package/dist/config.d.ts

@@ -0,0 +1,101 @@
+/**
+ * Quota data source
+ * - "cloud": Fetch from Cloud Code API (requires opencode-antigravity-auth)
+ * - "local": Fetch from local language server process
+ * - "auto": Try cloud first, fallback to local
+ */
+export type QuotaSource = "cloud" | "local" | "auto";
+export interface QuotaIndicator {
+    threshold: number;
+    symbol: string;
+}
+/**
+ * Configuration options for the quota plugin
+ */
+export interface QuotaConfig {
+    /**
+     * Where to fetch quota data from.
+     * - "cloud": Use Cloud Code API (requires opencode-antigravity-auth)
+     * - "local": Use local language server process
+     * - "auto": Try cloud first, fallback to local (default)
+     * @default "auto"
+     */
+    quotaSource?: QuotaSource;
+    /**
+     * Format string for quota display.
+     * Available placeholders:
+     * - {category} - Category name (Flash, Pro, Claude/GPT)
+     * - {percent} - Quota percentage (e.g., "85.5")
+     * - {resetIn} - Relative time until reset (e.g., "2h 30m")
+     * - {resetAt} - Absolute reset time (e.g., "10:30 PM")
+     * - {model} - Current model ID
+     *
+     * For all quotas mode, the format is applied per category and joined with separator.
+     * @default "{category}: {percent}% ({resetIn})"
+     */
+    format?: string;
+    /**
+     * Separator between categories when showing all quotas
+     * @default " | "
+     */
+    separator?: string;
+    /**
+     * Show all quota categories or only the current model's quota
+     * @default "all"
+     */
+    displayMode?: "all" | "current";
+    /**
+     * Always append quota info, even when unavailable
+     * @default true
+     */
+    alwaysAppend?: boolean;
+    /**
+     * The marker string used to separate quota info from the message
+     * @default "> AG Quota:"
+     */
+    quotaMarker?: string;
+    /**
+     * Polling interval in milliseconds
+     * @default 30000 (30 seconds)
+     */
+    pollingInterval?: number;
+    /**
+     * Array of quota usage percentages (remaining) that trigger alerts
+     * @default [0.5, 0.1, 0.05] (50%, 10%, 5%)
+     */
+    alertThresholds?: number[];
+    /**
+     * Visual indicators appended to quota percentages when remaining fraction is low.
+     * The most severe matching indicator is chosen.
+     *
+     * Example: with indicators [{threshold: 0.1, symbol: "⚠️"}, {threshold: 0.05, symbol: "⛔"}]
+     * a remainingFraction of 0.04 will show "⛔".
+     *
+     * @default [{threshold: 0.1, symbol: "⚠️"}, {threshold: 0.05, symbol: "⛔"}]
+     */
+    indicators?: QuotaIndicator[];
+}
+declare const DEFAULT_CONFIG: Required<QuotaConfig>;
+/**
+ * Load configuration from file system.
+ * Searches in order:
+ * 1. .opencode/ag-quota.json (project-local)
+ * 2. ~/.config/opencode/ag-quota.json (user global)
+ *
+ * @param projectDir - The project directory to search from
+ * @returns Merged configuration with defaults
+ */
+export declare function loadConfig(projectDir?: string): Promise<Required<QuotaConfig>>;
+/**
+ * Format a quota entry using the format string.
+ * Placeholders are only replaced if they exist in the format string.
+ */
+export declare function formatQuotaEntry(format: string, data: {
+    category: string;
+    percent: string;
+    resetIn: string | null;
+    resetAt: string | null;
+    model: string;
+}): string;
+export { DEFAULT_CONFIG };
+//# sourceMappingURL=config.d.ts.map

package/dist/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AASA;;;;;GAKG;AACH,MAAM,MAAM,WAAW,GAAG,OAAO,GAAG,OAAO,GAAG,MAAM,CAAC;AAErD,MAAM,WAAW,cAAc;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IACxB;;;;;;OAMG;IACH,WAAW,CAAC,EAAE,WAAW,CAAC;IAE1B;;;;;;;;;;;OAWG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;OAGG;IACH,WAAW,CAAC,EAAE,KAAK,GAAG,SAAS,CAAC;IAEhC;;;OAGG;IACH,YAAY,CAAC,EAAE,OAAO,CAAC;IAEvB;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;IAEzB;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAE3B;;;;;;;;OAQG;IACH,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;CACjC;AAED,QAAA,MAAM,cAAc,EAAE,QAAQ,CAAC,WAAW,CAazC,CAAC;AAEF;;;;;;;;GAQG;AACH,wBAAsB,UAAU,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAyBpF;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAC5B,MAAM,EAAE,MAAM,EACd,IAAI,EAAE;IACF,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;CACjB,GACF,MAAM,CA6BR;AAED,OAAO,EAAE,cAAc,EAAE,CAAC"}

package/dist/config.js

@@ -0,0 +1,88 @@
+/*
+ * ISC License
+ * Copyright (c) 2026 Philipp
+ */
+import { readFile } from "node:fs/promises";
+import { homedir } from "node:os";
+import { join } from "node:path";
+const DEFAULT_CONFIG = {
+    quotaSource: "auto",
+    format: "{category}: {percent}% ({resetIn})",
+    separator: " | ",
+    displayMode: "all",
+    alwaysAppend: true,
+    quotaMarker: "> AG Quota:",
+    pollingInterval: 30000,
+    alertThresholds: [0.5, 0.1, 0.05],
+    indicators: [
+        { threshold: 0.2, symbol: "⚠️" },
+        { threshold: 0.05, symbol: "🛑" },
+    ],
+};
+/**
+ * Load configuration from file system.
+ * Searches in order:
+ * 1. .opencode/ag-quota.json (project-local)
+ * 2. ~/.config/opencode/ag-quota.json (user global)
+ *
+ * @param projectDir - The project directory to search from
+ * @returns Merged configuration with defaults
+ */
+export async function loadConfig(projectDir) {
+    const paths = [];
+    // Project-local config
+    if (projectDir) {
+        paths.push(join(projectDir, ".opencode", "ag-quota.json"));
+    }
+    else {
+        paths.push(join(process.cwd(), ".opencode", "ag-quota.json"));
+    }
+    // User global config
+    paths.push(join(homedir(), ".config", "opencode", "ag-quota.json"));
+    for (const configPath of paths) {
+        try {
+            const content = await readFile(configPath, "utf-8");
+            const userConfig = JSON.parse(content);
+            return { ...DEFAULT_CONFIG, ...userConfig };
+        }
+        catch {
+            // File doesn't exist or is invalid, try next
+            continue;
+        }
+    }
+    return DEFAULT_CONFIG;
+}
+/**
+ * Format a quota entry using the format string.
+ * Placeholders are only replaced if they exist in the format string.
+ */
+export function formatQuotaEntry(format, data) {
+    let result = format
+        .replace("{category}", data.category)
+        .replace("{percent}", data.percent)
+        .replace("{model}", data.model);
+    // Handle resetIn (relative time) - e.g., "2h 30m"
+    if (format.includes("{resetIn}")) {
+        if (data.resetIn) {
+            result = result.replace("{resetIn}", data.resetIn);
+        }
+        else {
+            result = result
+                .replace(/\s*\(\{resetIn}\)/, "")
+                .replace(/\s*\{resetIn}/, "");
+        }
+    }
+    // Handle resetAt (absolute time) - e.g., "10:30 PM"
+    if (format.includes("{resetAt}")) {
+        if (data.resetAt) {
+            result = result.replace("{resetAt}", data.resetAt);
+        }
+        else {
+            result = result
+                .replace(/\s*\(\{resetAt}\)/, "")
+                .replace(/\s*\{resetAt}/, "");
+        }
+    }
+    return result;
+}
+export { DEFAULT_CONFIG };

package/dist/index.d.ts

@@ -0,0 +1,75 @@
+export declare const API_ENDPOINTS: {
+    GET_USER_STATUS: string;
+};
+export interface QuotaInfo {
+    remainingFraction: number;
+    resetTime?: string;
+}
+export interface ModelConfig {
+    modelName: string;
+    label?: string;
+    quotaInfo?: QuotaInfo;
+}
+export interface UserStatus {
+    cascadeModelConfigData?: {
+        clientModelConfigs?: ModelConfig[];
+    };
+}
+export interface UserStatusResponse {
+    userStatus: UserStatus;
+    timestamp: number;
+}
+export type ShellRunner = (cmd: string) => Promise<string>;
+export declare function fetchAntigravityStatus(runShell: ShellRunner): Promise<UserStatusResponse>;
+/**
+ * Format relative time until target date (e.g., "2h 30m")
+ */
+export declare function formatRelativeTime(targetDate: Date): string;
+/**
+ * Format absolute time of target date (e.g., "10:30 PM" or "22:30")
+ */
+export declare function formatAbsoluteTime(targetDate: Date): string;
+export { loadConfig, formatQuotaEntry, DEFAULT_CONFIG, type QuotaConfig, type QuotaSource, } from "./config";
+export { fetchCloudQuota, type CloudQuotaResult, type CloudAccountInfo, } from "./cloud";
+/**
+ * Unified category quota info (for the three model groups)
+ */
+export interface CategoryQuota {
+    category: "Flash" | "Pro" | "Claude/GPT";
+    remainingFraction: number;
+    resetTime: Date | null;
+}
+/**
+ * Unified quota result from either source
+ */
+export interface UnifiedQuotaResult {
+    source: "cloud" | "local";
+    categories: CategoryQuota[];
+    models: ModelConfig[];
+    timestamp: number;
+}
+/**
+ * Credentials for cloud quota fetching
+ */
+export interface CloudAuthCredentials {
+    accessToken: string;
+    projectId?: string;
+}
+/**
+ * Categorize a model label into one of the three groups.
+ */
+export declare function categorizeModel(label: string): "Flash" | "Pro" | "Claude/GPT";
+/**
+ * Group models into the three categories, taking the minimum quota per category.
+ */
+export declare function groupModelsByCategory(models: ModelConfig[]): CategoryQuota[];
+/**
+ * Fetch quota from either cloud or local source.
+ *
+ * @param source - "cloud", "local", or "auto" (try cloud first, fallback to local)
+ * @param shellRunner - Required for local source
+ * @param cloudAuth - Required for cloud source
+ * @returns Unified quota result
+ */
+export declare function fetchQuota(source: "cloud" | "local" | "auto", shellRunner?: ShellRunner, cloudAuth?: CloudAuthCredentials): Promise<UnifiedQuotaResult>;
+//# 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":"AAqBA,eAAO,MAAM,aAAa;;CAGzB,CAAC;AAEF,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,UAAU;IACvB,sBAAsB,CAAC,EAAE;QACrB,kBAAkB,CAAC,EAAE,WAAW,EAAE,CAAC;KACtC,CAAC;CACL;AAED,MAAM,WAAW,kBAAkB;IAC/B,UAAU,EAAE,UAAU,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,MAAM,WAAW,GAAG,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;AAqD3D,wBAAsB,sBAAsB,CACxC,QAAQ,EAAE,WAAW,GACtB,OAAO,CAAC,kBAAkB,CAAC,CAqF7B;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,UAAU,EAAE,IAAI,GAAG,MAAM,CAa3D;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,UAAU,EAAE,IAAI,GAAG,MAAM,CAK3D;AAGD,OAAO,EACH,UAAU,EACV,gBAAgB,EAChB,cAAc,EACd,KAAK,WAAW,EAChB,KAAK,WAAW,GACnB,MAAM,UAAU,CAAC;AAGlB,OAAO,EACH,eAAe,EACf,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,GACxB,MAAM,SAAS,CAAC;AAMjB;;GAEG;AACH,MAAM,WAAW,aAAa;IAC1B,QAAQ,EAAE,OAAO,GAAG,KAAK,GAAG,YAAY,CAAC;IACzC,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,IAAI,GAAG,IAAI,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IAC/B,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC;IAC1B,UAAU,EAAE,aAAa,EAAE,CAAC;IAC5B,MAAM,EAAE,WAAW,EAAE,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACjC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,GAAG,KAAK,GAAG,YAAY,CAS7E;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAkC5E;AAED;;;;;;;GAOG;AACH,wBAAsB,UAAU,CAC5B,MAAM,EAAE,OAAO,GAAG,OAAO,GAAG,MAAM,EAClC,WAAW,CAAC,EAAE,WAAW,EACzB,SAAS,CAAC,EAAE,oBAAoB,GACjC,OAAO,CAAC,kBAAkB,CAAC,CAgD7B"}

package/dist/index.js

@@ -0,0 +1,248 @@
+/*
+ ISC License
+
+ Copyright (c) 2025, Cristian Militaru
+
+ Permission to use, copy, modify, and/or distribute this software for any
+ purpose with or without fee is hereby granted, provided that the above
+ copyright notice and this permission notice appear in all copies.
+
+ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+import * as http from "node:http";
+import * as https from "node:https";
+export const API_ENDPOINTS = {
+    GET_USER_STATUS: "/exa.language_server_pb.LanguageServerService/GetUserStatus",
+};
+function makeRequest(port, csrfToken, path, body) {
+    return new Promise((resolve, reject) => {
+        const payload = JSON.stringify(body);
+        const options = {
+            hostname: "127.0.0.1",
+            port,
+            path,
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                "Content-Length": Buffer.byteLength(payload),
+                "X-Codeium-Csrf-Token": csrfToken,
+                "Connect-Protocol-Version": "1",
+            },
+            timeout: 2000,
+        };
+        const handleResponse = (response) => {
+            let data = "";
+            response.on("data", (chunk) => {
+                data += chunk.toString();
+            });
+            response.on("end", () => {
+                try {
+                    resolve(JSON.parse(data));
+                }
+                catch {
+                    reject(new Error("JSON parse error"));
+                }
+            });
+        };
+        const req = https.request({ ...options, rejectUnauthorized: false }, handleResponse);
+        req.on("error", () => {
+            const reqHttp = http.request(options, handleResponse);
+            reqHttp.on("error", (err) => reject(err));
+            reqHttp.write(payload);
+            reqHttp.end();
+        });
+        req.write(payload);
+        req.end();
+    });
+}
+export async function fetchAntigravityStatus(runShell) {
+    let procOutput = "";
+    try {
+        procOutput = await runShell('ps aux | grep -E "csrf_token|language_server" | grep -v grep');
+    }
+    catch {
+        procOutput = "";
+    }
+    const lines = procOutput.split("\n");
+    let csrfToken = "";
+    let cmdLinePort = 0;
+    for (const line of lines) {
+        const csrfMatch = line.match(/--csrf_token[=\s]+([\w-]+)/i);
+        if (csrfMatch?.[1])
+            csrfToken = csrfMatch[1];
+        const portMatch = line.match(/--extension_server_port[=\s]+(\d+)/i);
+        if (portMatch?.[1])
+            cmdLinePort = parseInt(portMatch[1], 10);
+        if (csrfToken && cmdLinePort)
+            break;
+    }
+    if (!csrfToken) {
+        throw new Error("Antigravity CSRF token not found. Is the Language Server running?");
+    }
+    let netstatOutput = "";
+    try {
+        netstatOutput = await runShell('ss -tlnp | grep -E "language_server|opencode|node"');
+    }
+    catch {
+        netstatOutput = "";
+    }
+    const portMatches = netstatOutput.match(/:(\d+)/g);
+    let ports = portMatches
+        ? portMatches.map((p) => parseInt(p.replace(":", ""), 10))
+        : [];
+    if (cmdLinePort && !ports.includes(cmdLinePort)) {
+        ports.unshift(cmdLinePort);
+    }
+    ports = Array.from(new Set(ports));
+    if (ports.length === 0) {
+        throw new Error("No listening ports found for Antigravity. Check if the server is active.");
+    }
+    let userStatus = null;
+    let lastError = null;
+    for (const p of ports) {
+        try {
+            const resp = await makeRequest(p, csrfToken, API_ENDPOINTS.GET_USER_STATUS, { metadata: { ideName: "opencode" } });
+            if (resp?.userStatus) {
+                userStatus = resp.userStatus;
+                break;
+            }
+        }
+        catch (e) {
+            lastError = e instanceof Error ? e : new Error(String(e));
+            continue;
+        }
+    }
+    if (!userStatus) {
+        throw new Error(`Could not communicate with Antigravity API. ${lastError?.message ?? ""}`);
+    }
+    return {
+        userStatus,
+        timestamp: Date.now(),
+    };
+}
+/**
+ * Format relative time until target date (e.g., "2h 30m")
+ */
+export function formatRelativeTime(targetDate) {
+    const now = new Date();
+    const diffMs = targetDate.getTime() - now.getTime();
+    if (diffMs <= 0)
+        return "now";
+    const diffMins = Math.floor(diffMs / (1000 * 60));
+    const diffHours = Math.floor(diffMins / 60);
+    const remainingMins = diffMins % 60;
+    if (diffHours > 0) {
+        return `${diffHours}h ${remainingMins}m`;
+    }
+    return `${diffMins}m`;
+}
+/**
+ * Format absolute time of target date (e.g., "10:30 PM" or "22:30")
+ */
+export function formatAbsoluteTime(targetDate) {
+    return targetDate.toLocaleTimeString(undefined, {
+        hour: "2-digit",
+        minute: "2-digit",
+    });
+}
+// Re-export config utilities
+export { loadConfig, formatQuotaEntry, DEFAULT_CONFIG, } from "./config";
+// Re-export cloud utilities
+export { fetchCloudQuota, } from "./cloud";
+/**
+ * Categorize a model label into one of the three groups.
+ */
+export function categorizeModel(label) {
+    const lowerLabel = label.toLowerCase();
+    if (lowerLabel.includes("flash")) {
+        return "Flash";
+    }
+    if (lowerLabel.includes("gemini") || lowerLabel.includes("pro")) {
+        return "Pro";
+    }
+    return "Claude/GPT";
+}
+/**
+ * Group models into the three categories, taking the minimum quota per category.
+ */
+export function groupModelsByCategory(models) {
+    const categories = {};
+    for (const model of models) {
+        const label = model.label || model.modelName || "";
+        const category = categorizeModel(label);
+        const fraction = model.quotaInfo?.remainingFraction ?? 0;
+        const resetTime = model.quotaInfo?.resetTime
+            ? new Date(model.quotaInfo.resetTime)
+            : null;
+        if (!categories[category] ||
+            fraction < categories[category].remainingFraction) {
+            categories[category] = { remainingFraction: fraction, resetTime };
+        }
+    }
+    const result = [];
+    for (const cat of ["Flash", "Pro", "Claude/GPT"]) {
+        if (categories[cat]) {
+            result.push({
+                category: cat,
+                remainingFraction: categories[cat].remainingFraction,
+                resetTime: categories[cat].resetTime,
+            });
+        }
+    }
+    return result;
+}
+/**
+ * Fetch quota from either cloud or local source.
+ *
+ * @param source - "cloud", "local", or "auto" (try cloud first, fallback to local)
+ * @param shellRunner - Required for local source
+ * @param cloudAuth - Required for cloud source
+ * @returns Unified quota result
+ */
+export async function fetchQuota(source, shellRunner, cloudAuth) {
+    // Import cloud module dynamically to avoid issues if not available
+    const { fetchCloudQuota } = await import("./cloud");
+    if (source === "cloud" || source === "auto") {
+        // Try cloud first
+        if (cloudAuth) {
+            try {
+                const cloudResult = await fetchCloudQuota(cloudAuth.accessToken, cloudAuth.projectId);
+                const categories = groupModelsByCategory(cloudResult.models);
+                return {
+                    source: "cloud",
+                    categories,
+                    models: cloudResult.models,
+                    timestamp: cloudResult.timestamp,
+                };
+            }
+            catch (error) {
+                if (source === "cloud") {
+                    throw error; // Don't fallback if explicitly requested cloud
+                }
+                // Fall through to local if auto
+            }
+        }
+        else if (source === "cloud") {
+            throw new Error("Cloud access token not provided. Cannot fetch cloud quota.");
+        }
+    }
+    // Try local
+    if (!shellRunner) {
+        throw new Error("Shell runner required for local quota fetching");
+    }
+    const localResult = await fetchAntigravityStatus(shellRunner);
+    const models = localResult.userStatus.cascadeModelConfigData?.clientModelConfigs || [];
+    const categories = groupModelsByCategory(models);
+    return {
+        source: "local",
+        categories,
+        models,
+        timestamp: localResult.timestamp,
+    };
+}