opencode-ag-quota 0.0.2

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Philipp
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,54 @@
+# opencode-ag-quota
+
+Opencode plugin that shows your **Antigravity quota** inside the Opencode TUI.
+
+## TL;DR
+
+- Installs as an Opencode plugin (`"opencode-ag-quota"`).
+- **Antigravity-only for now** (quota is shown only when the model ID contains `antigravity`).
+- Cloud mode uses `opencode auth login` credentials (auth/API approach based on [`opencode-antigravity-auth`](https://github.com/NoeFabris/opencode-antigravity-auth)).
+
+## Installation
+
+Add the plugin to your Opencode config (`opencode.json`):
+
+```json
+{
+  "plugin": ["opencode-ag-quota"]
+}
+```
+
+## Configuration
+
+Zero-config by default. Customize via:
+
+- Project: `.opencode/ag-quota.json`
+- Global: `~/.config/opencode/ag-quota.json`
+
+```json
+{
+  "quotaSource": "auto",
+  "format": "{category}: {percent}% ({resetIn})",
+  "separator": " | ",
+  "displayMode": "all",
+  "alwaysAppend": true
+}
+```
+
+### Options
+
+| Option | Type | Default | Description |
+|--------|------|---------|-------------|
+| `quotaSource` | `"auto" \| "cloud" \| "local"` | `"auto"` | `auto` tries cloud first, falls back to local. |
+| `format` | `string` | `"{category}: {percent}% ({resetIn})"` | Placeholders: `{category}`, `{percent}`, `{resetIn}`, `{resetAt}`, `{model}`. |
+| `separator` | `string` | `" | "` | Separator when `displayMode` is `all`. |
+| `displayMode` | `"all" \| "current"` | `"all"` | Show all quotas or only the current model's quota. |
+| `alwaysAppend` | `boolean` | `true` | Show an "Unavailable" hint when quota can't be read. |
+
+## Disclaimer
+
+This is an independent, third-party plugin for Opencode. It is not affiliated with, endorsed by, or maintained by the Opencode developers.
+
+## License
+
+MIT

package/dist/auth.d.ts

@@ -0,0 +1,14 @@
+export interface CloudAuthCredentials {
+    accessToken: string;
+    projectId?: string;
+    email: string;
+}
+/**
+ * Check if cloud credentials are available.
+ */
+export declare function hasCloudCredentials(): Promise<boolean>;
+/**
+ * Get a valid access token and project ID from the stored accounts.
+ */
+export declare function getCloudCredentials(): Promise<CloudAuthCredentials>;
+//# sourceMappingURL=auth.d.ts.map

package/dist/auth.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.d.ts","sourceRoot":"","sources":["../src/auth.ts"],"names":[],"mappings":"AAsCA,MAAM,WAAW,oBAAoB;IACjC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACjB;AAwCD;;GAEG;AACH,wBAAsB,mBAAmB,IAAI,OAAO,CAAC,OAAO,CAAC,CAO5D;AAqCD;;GAEG;AACH,wBAAsB,mBAAmB,IAAI,OAAO,CAAC,oBAAoB,CAAC,CAkBzE"}

package/dist/auth.js

@@ -0,0 +1,99 @@
+import { readFile } from "node:fs/promises";
+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";
+// ============================================================================
+// Account Loading
+// ============================================================================
+/**
+ * Get the path to the antigravity accounts file.
+ */
+function getAccountsFilePath() {
+    return join(homedir(), ".config", "opencode", "antigravity-accounts.json");
+}
+/**
+ * Load accounts from the opencode-antigravity-auth plugin.
+ * @throws Error if no accounts file exists or no accounts are configured
+ */
+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;
+    }
+}
+/**
+ * Check if cloud credentials are available.
+ */
+export async function hasCloudCredentials() {
+    try {
+        await loadAccounts();
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+// ============================================================================
+// Token Refresh
+// ============================================================================
+/**
+ * Exchange a refresh token for an access token.
+ */
+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 data.access_token;
+}
+/**
+ * Get a valid access token and project ID from the stored accounts.
+ */
+export async function getCloudCredentials() {
+    // Load accounts
+    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");
+    }
+    // Get access token
+    const accessToken = await refreshAccessToken(activeAccount.refreshToken);
+    return {
+        accessToken,
+        projectId: activeAccount.projectId,
+        email: activeAccount.email,
+    };
+}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+import QuotaPlugin from "./plugin";
+export { QuotaPlugin };
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,WAAW,MAAM,UAAU,CAAC;AAEnC,OAAO,EAAE,WAAW,EAAE,CAAC"}

package/dist/index.js

@@ -0,0 +1,2 @@
+import QuotaPlugin from "./plugin";
+export { QuotaPlugin };

package/dist/plugin.d.ts

@@ -0,0 +1,4 @@
+import { type Plugin } from "@opencode-ai/plugin";
+export declare const QuotaPlugin: Plugin;
+export default QuotaPlugin;
+//# sourceMappingURL=plugin.d.ts.map

package/dist/plugin.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin.d.ts","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":"AAyBA,OAAO,EAAE,KAAK,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAYlD,eAAO,MAAM,WAAW,EAAE,MA+QzB,CAAC;AAEF,eAAe,WAAW,CAAC"}

package/dist/plugin.js

@@ -0,0 +1,263 @@
+/*
+ * MIT License
+ *
+ * Copyright (c) 2026 Philipp
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
+ * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ */
+import { fetchQuota, formatRelativeTime, formatAbsoluteTime, loadConfig, formatQuotaEntry, } from "ag-quota";
+import { getCloudCredentials } from "./auth";
+export const QuotaPlugin = async ({ client, directory, $ }) => {
+    // Load configuration (loadConfig always returns Required<QuotaConfig> with defaults)
+    const config = await loadConfig(directory);
+    // Use Bun's $ shell helper from opencode
+    const shellRunner = async (cmd) => {
+        try {
+            const result = await $ `sh -c ${cmd}`.quiet().text();
+            return result;
+        }
+        catch {
+            return "";
+        }
+    };
+    let quotaState = {
+        data: null,
+        lastAlertLevel: 1.0,
+        isConnected: false,
+        currentSource: null,
+    };
+    /**
+     * Check if a model ID is an Antigravity model (quota tracking applies).
+     */
+    const isAntigravityModel = (modelID) => {
+        return modelID.toLowerCase().includes("antigravity");
+    };
+    /**
+     * Get the last assistant message's model ID from a session
+     */
+    const getLastAssistantModelID = async (sessionID) => {
+        try {
+            const response = await client.session.messages({
+                path: { id: sessionID },
+                query: { limit: 10 }, // Get recent messages, we just need the last assistant one
+            });
+            if (!response.data)
+                return null;
+            // Find the last assistant message (iterate in reverse)
+            for (let i = response.data.length - 1; i >= 0; i--) {
+                const msg = response.data[i];
+                if (msg.info.role === "assistant" && msg.info.modelID) {
+                    return msg.info.modelID;
+                }
+            }
+            return null;
+        }
+        catch {
+            return null;
+        }
+    };
+    /**
+     * Get the indicator symbol for a given remaining fraction based on config.
+     */
+    const getIndicatorSymbol = (fraction) => {
+        if (!config.indicators || config.indicators.length === 0)
+            return "";
+        // Sort by threshold ascending to find the most severe (lowest threshold met)
+        const sorted = [...config.indicators].sort((a, b) => a.threshold - b.threshold);
+        for (const indicator of sorted) {
+            if (fraction <= indicator.threshold) {
+                return ` ${indicator.symbol}`;
+            }
+        }
+        return "";
+    };
+    const buildQuotaSummary = (quotaResult, modelID, separator) => {
+        if (config.displayMode === "current" && modelID) {
+            // Find the model that matches the last used model
+            const currentModel = quotaResult.models.find((m) => m.modelName === modelID ||
+                modelID.includes(m.modelName) ||
+                (m.label && modelID.toLowerCase().includes(m.label.toLowerCase())) ||
+                (m.label && m.label.toLowerCase().includes(modelID.toLowerCase())));
+            if (currentModel?.quotaInfo) {
+                const fraction = currentModel.quotaInfo.remainingFraction;
+                const percent = (typeof fraction === "number" && Number.isFinite(fraction)
+                    ? fraction * 100
+                    : 0).toFixed(1);
+                const resetDate = currentModel.quotaInfo.resetTime
+                    ? new Date(currentModel.quotaInfo.resetTime)
+                    : null;
+                const displayPercent = `${percent}%${getIndicatorSymbol(fraction)}`;
+                return formatQuotaEntry(config.format, {
+                    category: "Current",
+                    percent: displayPercent,
+                    resetIn: resetDate ? formatRelativeTime(resetDate) : null,
+                    resetAt: resetDate ? formatAbsoluteTime(resetDate) : null,
+                    model: currentModel.modelName,
+                });
+            }
+        }
+        const parts = [];
+        for (const cat of quotaResult.categories) {
+            const percent = (cat.remainingFraction * 100).toFixed(0);
+            const displayPercent = `${percent}%${getIndicatorSymbol(cat.remainingFraction)}`;
+            const formatted = formatQuotaEntry(config.format, {
+                category: cat.category,
+                percent: displayPercent,
+                resetIn: cat.resetTime ? formatRelativeTime(cat.resetTime) : null,
+                resetAt: cat.resetTime ? formatAbsoluteTime(cat.resetTime) : null,
+                model: modelID ?? "current",
+            });
+            parts.push(formatted);
+        }
+        return parts.join(separator ?? config.separator);
+    };
+    // Try to connect with the configured source
+    const tryConnect = async () => {
+        try {
+            let cloudAuth;
+            if (config.quotaSource !== "local") {
+                try {
+                    cloudAuth = await getCloudCredentials();
+                }
+                catch {
+                    // If cloud auth fails but source is auto, we'll continue to local
+                    if (config.quotaSource === "cloud")
+                        throw new Error("Cloud auth failed");
+                }
+            }
+            const result = await fetchQuota(config.quotaSource, shellRunner, cloudAuth);
+            quotaState.currentSource = result.source;
+            return result;
+        }
+        catch {
+            return null;
+        }
+    };
+    /**
+     * Check if we need to alert the user about low quota
+     */
+    const checkAlerts = () => {
+        if (!quotaState.data)
+            return;
+        // Find lowest category fraction
+        let minFraction = 1.0;
+        for (const cat of quotaState.data.categories) {
+            if (cat.remainingFraction < minFraction) {
+                minFraction = cat.remainingFraction;
+            }
+        }
+        // Check against thresholds (ascending order).
+        // We want to show the most severe (lowest) warning once when crossing multiple levels.
+        const thresholds = [...config.alertThresholds].sort((a, b) => a - b);
+        for (const threshold of thresholds) {
+            // If we dropped below this threshold AND haven't alerted for it yet
+            if (minFraction <= threshold && quotaState.lastAlertLevel > threshold) {
+                // Use buildQuotaSummary for consistent formatting across all toasts
+                const summary = buildQuotaSummary(quotaState.data, undefined, "\n");
+                client.tui.showToast({
+                    body: {
+                        title: "Low Quota Warning",
+                        message: `Below ${(threshold * 100).toFixed(0)}% threshold.\n\n${summary}`,
+                        variant: "warning",
+                    },
+                });
+                quotaState.lastAlertLevel = threshold;
+                break; // Only one alert per drop
+            }
+        }
+        // If quota went back up (reset), reset alert level
+        if (minFraction > quotaState.lastAlertLevel) {
+            quotaState.lastAlertLevel = 1.0;
+        }
+    };
+    /**
+     * Start background polling loop
+     */
+    const startPolling = async () => {
+        // Fetch quota
+        const result = await tryConnect();
+        if (result) {
+            if (!quotaState.isConnected) {
+                // Just became connected
+                quotaState.isConnected = true;
+                const sourceLabel = quotaState.currentSource === "cloud" ? "Cloud API" : "Language Server";
+                const summary = buildQuotaSummary(result, undefined, "\n") || "unknown";
+                client.tui.showToast({
+                    body: {
+                        title: "AG Quota",
+                        message: `Connected via ${sourceLabel}.\n${summary}`,
+                        variant: "success",
+                    },
+                });
+            }
+            quotaState.data = result;
+            quotaState.currentSource = result.source;
+            checkAlerts();
+        }
+        else {
+            if (quotaState.isConnected) {
+                // Just became disconnected
+                quotaState.isConnected = false;
+                quotaState.currentSource = null;
+                client.tui.showToast({
+                    body: {
+                        title: "Quota Disconnected",
+                        message: "Quota connection lost",
+                        variant: "warning",
+                    },
+                });
+            }
+        }
+        // Schedule next poll
+        setTimeout(startPolling, config.pollingInterval);
+    };
+    // Start polling immediately
+    startPolling();
+    return {
+        event: async ({ event }) => {
+            // On session.idle, show quota toast if an Antigravity model was used
+            if (event.type === "session.idle") {
+                const { sessionID } = event.properties;
+                // Get the last assistant message's model ID from the session
+                const lastModelID = await getLastAssistantModelID(sessionID);
+                // Only show quota for Antigravity models (model ID contains "antigravity")
+                if (!lastModelID || !isAntigravityModel(lastModelID)) {
+                    return;
+                }
+                // Check if we have quota data
+                if (!quotaState.isConnected || !quotaState.data) {
+                    return;
+                }
+                // Build and show quota summary
+                const summary = buildQuotaSummary(quotaState.data, lastModelID, "\n");
+                if (summary) {
+                    client.tui.showToast({
+                        body: {
+                            title: "AG Quota",
+                            message: summary,
+                            variant: "info",
+                        },
+                    });
+                }
+            }
+        },
+    };
+};
+export default QuotaPlugin;

package/package.json

@@ -0,0 +1,59 @@
+{
+  "name": "opencode-ag-quota",
+  "version": "0.0.2",
+  "description": "Opencode plugin to display Antigravity/Windsurf quotas",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "src",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "prepublishOnly": "bun run build",
+    "typecheck": "tsc --noEmit"
+  },
+  "keywords": [
+    "opencode",
+    "plugin",
+    "antigravity",
+    "windsurf",
+    "quota",
+    "codeium"
+  ],
+  "author": "Philipp",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/phibkro/opencode-quota-display.git",
+    "directory": "packages/opencode-ag-quota"
+  },
+  "bugs": {
+    "url": "https://github.com/phibkro/opencode-quota-display/issues"
+  },
+  "homepage": "https://github.com/phibkro/opencode-quota-display/tree/main/packages/opencode-ag-quota#readme",
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "ag-quota": "^0.0.2"
+  },
+  "devDependencies": {
+    "@opencode-ai/plugin": "^1.1.3",
+    "@opencode-ai/sdk": "^1.1.3",
+    "@types/node": "^22.0.0",
+    "typescript": "^5.9.3"
+  },
+  "peerDependencies": {
+    "@opencode-ai/plugin": "^1.0.0"
+  }
+}

package/src/auth.ts

@@ -0,0 +1,151 @@
+import { readFile } from "node:fs/promises";
+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 CloudAuthCredentials {
+    accessToken: string;
+    projectId?: string;
+    email: string;
+}
+
+// ============================================================================
+// Account Loading
+// ============================================================================
+
+/**
+ * Get the path to the antigravity accounts file.
+ */
+function getAccountsFilePath(): string {
+    return join(homedir(), ".config", "opencode", "antigravity-accounts.json");
+}
+
+/**
+ * Load accounts from the opencode-antigravity-auth plugin.
+ * @throws Error if no accounts file exists or no accounts are configured
+ */
+async function loadAccounts(): Promise<AccountsFile> {
+    const accountsPath = getAccountsFilePath();
+
+    try {
+        const content = await readFile(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.\n` +
+                    `Run 'opencode auth login' first to authenticate with Google.`,
+            );
+        }
+        throw error;
+    }
+}
+
+/**
+ * Check if cloud credentials are available.
+ */
+export async function hasCloudCredentials(): Promise<boolean> {
+    try {
+        await loadAccounts();
+        return true;
+    } catch {
+        return false;
+    }
+}
+
+// ============================================================================
+// Token Refresh
+// ============================================================================
+
+/**
+ * Exchange a refresh token for an access token.
+ */
+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();
+        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()) as TokenResponse;
+    return data.access_token;
+}
+
+/**
+ * Get a valid access token and project ID from the stored accounts.
+ */
+export async function getCloudCredentials(): Promise<CloudAuthCredentials> {
+    // Load accounts
+    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");
+    }
+
+    // Get access token
+    const accessToken = await refreshAccessToken(activeAccount.refreshToken);
+
+    return {
+        accessToken,
+        projectId: activeAccount.projectId,
+        email: activeAccount.email,
+    };
+}

package/src/index.ts

@@ -0,0 +1,3 @@
+import QuotaPlugin from "./plugin";
+
+export { QuotaPlugin };

package/src/plugin.ts

@@ -0,0 +1,311 @@
+/*
+ * MIT License
+ *
+ * Copyright (c) 2026 Philipp
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
+ * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ */
+
+import { type Plugin } from "@opencode-ai/plugin";
+import {
+    fetchQuota,
+    formatRelativeTime,
+    formatAbsoluteTime,
+    loadConfig,
+    formatQuotaEntry,
+    type ShellRunner,
+    type UnifiedQuotaResult,
+} from "ag-quota";
+import { getCloudCredentials } from "./auth";
+
+export const QuotaPlugin: Plugin = async ({ client, directory, $ }) => {
+    // Load configuration (loadConfig always returns Required<QuotaConfig> with defaults)
+    const config = await loadConfig(directory);
+
+    // Use Bun's $ shell helper from opencode
+    const shellRunner: ShellRunner = async (cmd: string) => {
+        try {
+            const result = await $`sh -c ${cmd}`.quiet().text();
+            return result;
+        } catch {
+            return "";
+        }
+    };
+
+    // Connection state
+    interface PluginState {
+        data: UnifiedQuotaResult | null;
+        lastAlertLevel: number; // Start at 1.0 (100%)
+        isConnected: boolean;
+        currentSource: "cloud" | "local" | null;
+    }
+
+    let quotaState: PluginState = {
+        data: null,
+        lastAlertLevel: 1.0,
+        isConnected: false,
+        currentSource: null,
+    };
+
+    /**
+     * Check if a model ID is an Antigravity model (quota tracking applies).
+     */
+    const isAntigravityModel = (modelID: string): boolean => {
+        return modelID.toLowerCase().includes("antigravity");
+    };
+
+    /**
+     * Get the last assistant message's model ID from a session
+     */
+    const getLastAssistantModelID = async (sessionID: string): Promise<string | null> => {
+        try {
+            const response = await client.session.messages({
+                path: { id: sessionID },
+                query: { limit: 10 }, // Get recent messages, we just need the last assistant one
+            });
+
+            if (!response.data) return null;
+
+            // Find the last assistant message (iterate in reverse)
+            for (let i = response.data.length - 1; i >= 0; i--) {
+                const msg = response.data[i];
+                if (msg.info.role === "assistant" && msg.info.modelID) {
+                    return msg.info.modelID;
+                }
+            }
+
+            return null;
+        } catch {
+            return null;
+        }
+    };
+
+    /**
+     * Get the indicator symbol for a given remaining fraction based on config.
+     */
+    const getIndicatorSymbol = (fraction: number): string => {
+        if (!config.indicators || config.indicators.length === 0) return "";
+        // Sort by threshold ascending to find the most severe (lowest threshold met)
+        const sorted = [...config.indicators].sort((a, b) => a.threshold - b.threshold);
+        for (const indicator of sorted) {
+            if (fraction <= indicator.threshold) {
+                return ` ${indicator.symbol}`;
+            }
+        }
+        return "";
+    };
+
+    const buildQuotaSummary = (quotaResult: UnifiedQuotaResult, modelID?: string, separator?: string): string => {
+        if (config.displayMode === "current" && modelID) {
+            // Find the model that matches the last used model
+            const currentModel = quotaResult.models.find(
+                (m) =>
+                    m.modelName === modelID ||
+                    modelID.includes(m.modelName) ||
+                    (m.label && modelID.toLowerCase().includes(m.label.toLowerCase())) ||
+                    (m.label && m.label.toLowerCase().includes(modelID.toLowerCase())),
+            );
+
+            if (currentModel?.quotaInfo) {
+                const fraction = currentModel.quotaInfo.remainingFraction;
+                const percent = (
+                    typeof fraction === "number" && Number.isFinite(fraction)
+                        ? fraction * 100
+                        : 0
+                ).toFixed(1);
+                const resetDate = currentModel.quotaInfo.resetTime
+                    ? new Date(currentModel.quotaInfo.resetTime)
+                    : null;
+
+                const displayPercent = `${percent}%${getIndicatorSymbol(fraction)}`;
+
+                return formatQuotaEntry(config.format, {
+                    category: "Current",
+                    percent: displayPercent,
+                    resetIn: resetDate ? formatRelativeTime(resetDate) : null,
+                    resetAt: resetDate ? formatAbsoluteTime(resetDate) : null,
+                    model: currentModel.modelName,
+                });
+            }
+        }
+
+        const parts: string[] = [];
+        for (const cat of quotaResult.categories) {
+            const percent = (cat.remainingFraction * 100).toFixed(0);
+            const displayPercent = `${percent}%${getIndicatorSymbol(cat.remainingFraction)}`;
+
+            const formatted = formatQuotaEntry(config.format, {
+                category: cat.category,
+                percent: displayPercent,
+                resetIn: cat.resetTime ? formatRelativeTime(cat.resetTime) : null,
+                resetAt: cat.resetTime ? formatAbsoluteTime(cat.resetTime) : null,
+                model: modelID ?? "current",
+            });
+            parts.push(formatted);
+        }
+        return parts.join(separator ?? config.separator);
+    };
+
+    // Try to connect with the configured source
+    const tryConnect = async (): Promise<UnifiedQuotaResult | null> => {
+        try {
+            let cloudAuth;
+            if (config.quotaSource !== "local") {
+                try {
+                    cloudAuth = await getCloudCredentials();
+                } catch {
+                    // If cloud auth fails but source is auto, we'll continue to local
+                    if (config.quotaSource === "cloud") throw new Error("Cloud auth failed");
+                }
+            }
+
+            const result = await fetchQuota(config.quotaSource, shellRunner, cloudAuth);
+            quotaState.currentSource = result.source;
+            return result;
+        } catch {
+            return null;
+        }
+    };
+
+    /**
+     * Check if we need to alert the user about low quota
+     */
+    const checkAlerts = () => {
+        if (!quotaState.data) return;
+
+        // Find lowest category fraction
+        let minFraction = 1.0;
+        for (const cat of quotaState.data.categories) {
+            if (cat.remainingFraction < minFraction) {
+                minFraction = cat.remainingFraction;
+            }
+        }
+
+        // Check against thresholds (ascending order).
+        // We want to show the most severe (lowest) warning once when crossing multiple levels.
+        const thresholds = [...config.alertThresholds].sort((a, b) => a - b);
+
+        for (const threshold of thresholds) {
+            // If we dropped below this threshold AND haven't alerted for it yet
+            if (minFraction <= threshold && quotaState.lastAlertLevel > threshold) {
+                // Use buildQuotaSummary for consistent formatting across all toasts
+                const summary = buildQuotaSummary(quotaState.data, undefined, "\n");
+
+                client.tui.showToast({
+                    body: {
+                        title: "Low Quota Warning",
+                        message: `Below ${(threshold * 100).toFixed(0)}% threshold.\n\n${summary}`,
+                        variant: "warning",
+                    },
+                });
+                quotaState.lastAlertLevel = threshold;
+                break; // Only one alert per drop
+            }
+        }
+
+        // If quota went back up (reset), reset alert level
+        if (minFraction > quotaState.lastAlertLevel) {
+            quotaState.lastAlertLevel = 1.0;
+        }
+    };
+
+    /**
+     * Start background polling loop
+     */
+    const startPolling = async () => {
+        // Fetch quota
+        const result = await tryConnect();
+
+        if (result) {
+            if (!quotaState.isConnected) {
+                // Just became connected
+                quotaState.isConnected = true;
+                const sourceLabel = quotaState.currentSource === "cloud" ? "Cloud API" : "Language Server";
+                const summary = buildQuotaSummary(result, undefined, "\n") || "unknown";
+                client.tui.showToast({
+                    body: {
+                        title: "AG Quota",
+                        message: `Connected via ${sourceLabel}.\n${summary}`,
+                        variant: "success",
+                    },
+                });
+            }
+
+            quotaState.data = result;
+            quotaState.currentSource = result.source;
+            checkAlerts();
+        } else {
+            if (quotaState.isConnected) {
+                // Just became disconnected
+                quotaState.isConnected = false;
+                quotaState.currentSource = null;
+                client.tui.showToast({
+                    body: {
+                        title: "Quota Disconnected",
+                        message: "Quota connection lost",
+                        variant: "warning",
+                    },
+                });
+            }
+        }
+
+        // Schedule next poll
+        setTimeout(startPolling, config.pollingInterval);
+    };
+
+    // Start polling immediately
+    startPolling();
+
+    return {
+        event: async ({ event }) => {
+            // On session.idle, show quota toast if an Antigravity model was used
+            if (event.type === "session.idle") {
+                const { sessionID } = event.properties;
+
+                // Get the last assistant message's model ID from the session
+                const lastModelID = await getLastAssistantModelID(sessionID);
+
+                // Only show quota for Antigravity models (model ID contains "antigravity")
+                if (!lastModelID || !isAntigravityModel(lastModelID)) {
+                    return;
+                }
+
+                // Check if we have quota data
+                if (!quotaState.isConnected || !quotaState.data) {
+                    return;
+                }
+
+                // Build and show quota summary
+                const summary = buildQuotaSummary(quotaState.data, lastModelID, "\n");
+                if (summary) {
+                    client.tui.showToast({
+                        body: {
+                            title: "AG Quota",
+                            message: summary,
+                            variant: "info",
+                        },
+                    });
+                }
+            }
+        },
+    };
+};
+
+export default QuotaPlugin;