ag-quota 0.0.2

package/package.json

@@ -0,0 +1,57 @@
+{
+  "name": "ag-quota",
+  "version": "0.0.2",
+  "description": "Antigravity quota fetching library and CLI",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "bin": {
+    "ag-quota": "./bin/ag-quota.js"
+  },
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "bin",
+    "src",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "bun run build:lib && bun run build:cli",
+    "build:lib": "tsc",
+    "build:cli": "bun build src/cli.ts --outfile bin/ag-quota.js --target node",
+    "prepublishOnly": "bun run build",
+    "typecheck": "tsc --noEmit",
+    "ag-quota": "bun src/cli.ts"
+  },
+  "keywords": [
+    "antigravity",
+    "quota",
+    "windsurf",
+    "codeium",
+    "cli"
+  ],
+  "author": "Philipp",
+  "license": "ISC",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/phibkro/opencode-quota-display.git",
+    "directory": "packages/ag-quota"
+  },
+  "bugs": {
+    "url": "https://github.com/phibkro/opencode-quota-display/issues"
+  },
+  "homepage": "https://github.com/phibkro/opencode-quota-display/tree/main/packages/ag-quota#readme",
+  "engines": {
+    "node": ">=18"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "typescript": "^5.9.3"
+  }
+}

package/src/cli-auth.ts

@@ -0,0 +1,136 @@
+/*
+ * Authentication helper for the CLI.
+ * 
+ * This logic is duplicated here specifically for the CLI to be standalone 
+ * and user-friendly, without polluting the core library exports which 
+ * should remain pure and environment-agnostic.
+ */
+
+import { readFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+
+// ============================================================================
+// Constants
+// ============================================================================
+
+const ANTIGRAVITY_CLIENT_ID =
+    "1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com";
+const ANTIGRAVITY_CLIENT_SECRET = "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";
+const TOKEN_URL = "https://oauth2.googleapis.com/token";
+
+// ============================================================================
+// Types
+// ============================================================================
+
+interface StoredAccount {
+    email: string;
+    refreshToken: string;
+    projectId?: string;
+    managedProjectId?: string;
+    addedAt: number;
+    lastUsed: number;
+}
+
+interface AccountsFile {
+    version: number;
+    accounts: StoredAccount[];
+    activeIndex: number;
+}
+
+interface TokenResponse {
+    access_token: string;
+    expires_in: number;
+    token_type: string;
+}
+
+export interface CLICloudCredentials {
+    accessToken: string;
+    projectId?: string;
+    email: string;
+}
+
+// ============================================================================
+// Logic
+// ============================================================================
+
+function getAccountsFilePath(): string {
+    return join(homedir(), ".config", "opencode", "antigravity-accounts.json");
+}
+
+function loadAccounts(): AccountsFile {
+    const accountsPath = getAccountsFilePath();
+
+    try {
+        const content = readFileSync(accountsPath, "utf-8");
+        const data = JSON.parse(content) as AccountsFile;
+
+        if (!data.accounts || data.accounts.length === 0) {
+            throw new Error("No accounts found in antigravity-accounts.json");
+        }
+
+        return data;
+    } catch (error) {
+        if ((error as NodeJS.ErrnoException).code === "ENOENT") {
+            throw new Error("Antigravity accounts file not found.");
+        }
+        throw error;
+    }
+}
+
+async function refreshAccessToken(refreshToken: string): Promise<string> {
+    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();
+        throw new Error(`Token refresh failed: ${response.status} - ${errorText}`);
+    }
+
+    const data = (await response.json()) as TokenResponse;
+    return data.access_token;
+}
+
+/**
+ * Attempt to get cloud credentials from the local environment.
+ * Returns null if no credentials found or file missing.
+ * Throws if file exists but is invalid or refresh fails.
+ */
+export async function getCLICloudCredentials(): Promise<CLICloudCredentials | null> {
+    try {
+        // Load accounts
+        const accountsFile = loadAccounts();
+        const activeAccount =
+            accountsFile.accounts[accountsFile.activeIndex] ?? accountsFile.accounts[0];
+
+        if (!activeAccount) {
+            return null;
+        }
+
+        // Get access token
+        const accessToken = await refreshAccessToken(activeAccount.refreshToken);
+
+        return {
+            accessToken,
+            projectId: activeAccount.projectId,
+            email: activeAccount.email,
+        };
+    } catch (error) {
+        // If file not found, just return null (not available)
+        if (error instanceof Error && error.message.includes("not found")) {
+            return null;
+        }
+        // If other error (parsing, network), throw it
+        throw error;
+    }
+}

package/src/cli.ts

@@ -0,0 +1,154 @@
+#!/usr/bin/env node
+/*
+ * ISC License
+ * Copyright (c) 2026 Philipp
+ */
+
+import { execSync } from "node:child_process";
+import {
+    fetchQuota,
+    formatRelativeTime,
+    type ShellRunner,
+    type QuotaSource,
+    type CloudAuthCredentials,
+} from "./index";
+import { getCLICloudCredentials } from "./cli-auth";
+
+const shellRunner: ShellRunner = async (cmd: string) =>
+    execSync(cmd).toString();
+
+function parseArgs(): { source: QuotaSource; json: boolean; help: boolean; token: string; projectId: string } {
+    const args = process.argv.slice(2);
+    let source: QuotaSource = "auto";
+    let json = false;
+    let help = false;
+    let token = "";
+    let projectId = "";
+
+    for (const arg of args) {
+        if (arg === "--json") {
+            json = true;
+        } else if (arg === "--help" || arg === "-h") {
+            help = true;
+        } else if (arg === "--source=cloud" || arg === "-s=cloud") {
+            source = "cloud";
+        } else if (arg === "--source=local" || arg === "-s=local") {
+            source = "local";
+        } else if (arg === "--source=auto" || arg === "-s=auto") {
+            source = "auto";
+        } else if (arg.startsWith("--token=")) {
+            token = arg.split("=")[1];
+        } else if (arg.startsWith("--project-id=")) {
+            projectId = arg.split("=")[1];
+        } else if (arg.startsWith("--source=") || arg.startsWith("-s=")) {
+            const value = arg.split("=")[1];
+            console.error(`Invalid source: ${value}. Use 'cloud', 'local', or 'auto'.`);
+            process.exit(1);
+        }
+    }
+
+    return { source, json, help, token, projectId };
+}
+
+async function run() {
+    const { source, json: isJson, help: isHelp, token, projectId } = parseArgs();
+
+    if (isHelp) {
+        console.log(`
+Usage: ag-quota [options]
+
+Options:
+  --source=<cloud|local|auto>  Quota source (default: auto)
+  -s=<cloud|local|auto>        Alias for --source
+  --token=<token>              Access token (override auto-discovery)
+  --project-id=<id>            Google Cloud Project ID (optional)
+  --json                       Output result as JSON
+  -h, --help                   Show this help message
+
+Sources:
+  cloud   Fetch from Cloud Code API (uses auto-discovery or --token)
+  local   Fetch from local language server process
+  auto    Try cloud first, fallback to local (default)
+
+Examples:
+  ag-quota                     # Auto-detect (tries cloud then local)
+  ag-quota --source=cloud      # Force cloud (auto-discover token)
+  ag-quota --token=...         # Force cloud with specific token
+  ag-quota --source=local      # Force local source
+  ag-quota --json              # Output as JSON
+`);
+        process.exit(0);
+    }
+
+    try {
+        // Resolve credentials
+        let cloudAuth: CloudAuthCredentials | undefined = token ? { accessToken: token, projectId } : undefined;
+
+        // If no token provided, try to auto-discover
+        if (!cloudAuth && (source === "cloud" || source === "auto")) {
+            const creds = await getCLICloudCredentials();
+            if (creds) {
+                cloudAuth = { accessToken: creds.accessToken, projectId: creds.projectId };
+            }
+        }
+
+        if (source === "cloud" && !cloudAuth) {
+            throw new Error("Cloud credentials not found. Run 'opencode auth login' or provide --token.");
+        }
+
+        const result = await fetchQuota(source, shellRunner, cloudAuth);
+
+        if (isJson) {
+            console.log(
+                JSON.stringify(
+                    {
+                        source: result.source,
+                        timestamp: result.timestamp,
+                        categories: result.categories.map((cat) => ({
+                            name: cat.category,
+                            remainingFraction: cat.remainingFraction,
+                            remainingPercentage: parseFloat(
+                                (cat.remainingFraction * 100).toFixed(1),
+                            ),
+                            resetTime: cat.resetTime?.toISOString() ?? null,
+                            resetsIn: cat.resetTime
+                                ? formatRelativeTime(cat.resetTime)
+                                : null,
+                        })),
+                    },
+                    null,
+                    2,
+                ),
+            );
+            return;
+        }
+
+        const sourceLabel = result.source === "cloud" ? "Cloud API" : "Local Server";
+        console.log(
+            `\nAntigravity Quotas (Source: ${sourceLabel}, ${new Date(result.timestamp).toLocaleTimeString()}):`,
+        );
+        console.log(
+            "------------------------------------------------------------",
+        );
+
+        for (const cat of result.categories) {
+            const remaining = (cat.remainingFraction * 100).toFixed(1);
+            let output = `${cat.category.padEnd(20)}: ${remaining.padStart(5)}% remaining`;
+            if (cat.resetTime) {
+                output += ` (Resets in: ${formatRelativeTime(cat.resetTime)})`;
+            }
+            console.log(output);
+        }
+        console.log("");
+    } catch (error: unknown) {
+        const message = error instanceof Error ? error.message : String(error);
+        if (isJson) {
+            console.log(JSON.stringify({ error: message }, null, 2));
+        } else {
+            console.error("Error:", message);
+        }
+        process.exit(1);
+    }
+}
+
+run();

package/src/cloud.ts

@@ -0,0 +1,195 @@
+/*
+ * 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",
+] as const;
+
+// 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"}',
+} as const;
+
+// ============================================================================
+// Types
+// ============================================================================
+
+interface CloudQuotaInfo {
+    remainingFraction?: number;
+    resetTime?: string;
+}
+
+interface CloudModelInfo {
+    displayName?: string;
+    model?: string;
+    quotaInfo?: CloudQuotaInfo;
+    supportsImages?: boolean;
+    supportsThinking?: boolean;
+    recommended?: boolean;
+}
+
+interface FetchModelsResponse {
+    models?: Record<string, CloudModelInfo>;
+}
+
+/**
+ * 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;
+}
+
+// ============================================================================
+// API Fetching
+// ============================================================================
+
+/**
+ * Fetch available models with quota from the Cloud Code API.
+ */
+async function fetchAvailableModels(
+    accessToken: string,
+    projectId?: string,
+): Promise<FetchModelsResponse> {
+    const payload = projectId ? { project: projectId } : {};
+    let lastError: Error | null = null;
+
+    const headers: Record<string, string> = {
+        ...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()) as FetchModelsResponse;
+        } 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: string,
+    projectId?: string,
+): Promise<CloudQuotaResult> {
+    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: ModelConfig[] = [];
+
+    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(),
+    };
+}