antigravity-usage 0.2.6 → 0.2.7

package/README.md

@@ -49,6 +49,13 @@ Want to check quota for **multiple accounts** or when your IDE is closed?
 antigravity-usage login
 ```
 
+### 1a. Manual Login (Headless/SSH)
+If you are on a headless server or cannot open a browser locally:
+```bash
+antigravity-usage login --manual
+```
+Follow the on-screen instructions to paste the authentication URL into your local browser and copy the result back.
+
 ### 2. Add more accounts
 ```bash
 antigravity-usage accounts add
@@ -109,8 +116,14 @@ To keep the CLI snappy and avoid hitting API rate limits:
     antigravity-usage quota --refresh
     ```
 
-### 📱 Responsive UI
-Tables automatically adapt to your terminal size, switching between "Compact" and "Spacious" views to show you the most relevant data without wrapping.
+
+### 🎯 Focused Model View
+By default, `antigravity-usage` hides "autocomplete" models (like `gemini-2.5-flash-002`) to reduce clutter, as these typically share quota with their main counterparts or are less relevant for tracking.
+
+To see **ALL** available models, including autocomplete ones:
+```bash
+antigravity-usage quota --all-models
+```
 
 ---
 
@@ -124,6 +137,7 @@ antigravity-usage                   # Auto-detect (Local -> Cloud)
 antigravity-usage --all             # Fetch ALL accounts
 antigravity-usage --method local    # Force local IDE connection
 antigravity-usage --method google   # Force google IDE connection
+antigravity-usage --all-models      # Show ALL models (including autocomplete)
 antigravity-usage --json            # Output JSON for scripts
 antigravity-usage --version         # Show version number
 ```

package/dist/index.js

@@ -48,6 +48,7 @@ function success(message) {
 import { createServer } from "http";
 import { URL as URL2, URLSearchParams } from "url";
 import open from "open";
+import inquirer from "inquirer";
 
 // src/accounts/types.ts
 var DEFAULT_CONFIG = {
@@ -722,6 +723,33 @@ async function resolveProjectId(accessToken) {
     return { projectId: void 0, tierId: void 0 };
   }
 }
+async function completeLogin(code, redirectUri) {
+  const tokenResponse = await exchangeCodeForTokens(code, redirectUri);
+  const email = await getUserEmail(tokenResponse.access_token);
+  let projectId;
+  try {
+    const projectResult = await resolveProjectId(tokenResponse.access_token);
+    projectId = projectResult.projectId;
+    if (projectId) {
+      debug("oauth", `Project ID resolved: ${projectId}`);
+    } else {
+      debug("oauth", "No project ID obtained (will fetch on demand)");
+    }
+  } catch (err) {
+    debug("oauth", "Failed to resolve project ID during login (will fetch on demand)", err);
+  }
+  const tokens = {
+    accessToken: tokenResponse.access_token,
+    refreshToken: tokenResponse.refresh_token || "",
+    expiresAt: Date.now() + tokenResponse.expires_in * 1e3,
+    email,
+    projectId
+  };
+  if (email) {
+    getAccountManager().addAccount(tokens, email);
+  }
+  return { success: true, email };
+}
 async function startOAuthFlow(options = {}) {
   const port = await getAvailablePort(options.port);
   const redirectUri = `http://127.0.0.1:${port}/callback`;
@@ -737,6 +765,43 @@ async function startOAuthFlow(options = {}) {
     state
   });
   const authUrl = `${OAUTH_CONFIG.authUrl}?${authParams.toString()}`;
+  if (options.manual) {
+    info("");
+    info("MANUAL LOGIN MODE");
+    info("1. Copy this URL and open it in your browser:");
+    info(authUrl);
+    info("");
+    info("2. Login with your Google account.");
+    info("3. You will be redirected to a localhost URL (which may fail to load).");
+    info("4. Copy that ENTIRE localhost URL and paste it below.");
+    info("");
+    const { pastedUrl } = await inquirer.prompt([
+      {
+        type: "input",
+        name: "pastedUrl",
+        message: "Paste the full redirect URL here:",
+        validate: (input) => input.trim().length > 0 ? true : "Please paste the URL"
+      }
+    ]);
+    try {
+      const url = new URL2(pastedUrl.trim());
+      const code = url.searchParams.get("code");
+      const returnedState = url.searchParams.get("state");
+      const errorParam = url.searchParams.get("error");
+      if (errorParam) {
+        return { success: false, error: errorParam };
+      }
+      if (!code || returnedState !== state) {
+        return { success: false, error: "Invalid URL: Missing code or state mismatch" };
+      }
+      return await completeLogin(code, redirectUri);
+    } catch (err) {
+      if (err instanceof Error) {
+        return { success: false, error: err.message };
+      }
+      return { success: false, error: "Invalid URL format" };
+    }
+  }
   return new Promise((resolve) => {
     let resolved = false;
     const server = createServer(async (req, res) => {
@@ -763,43 +828,20 @@ async function startOAuthFlow(options = {}) {
           return;
         }
         try {
-          const tokenResponse = await exchangeCodeForTokens(code, redirectUri);
-          const email = await getUserEmail(tokenResponse.access_token);
-          let projectId;
-          try {
-            const projectResult = await resolveProjectId(tokenResponse.access_token);
-            projectId = projectResult.projectId;
-            if (projectId) {
-              debug("oauth", `Project ID resolved: ${projectId}`);
-            } else {
-              debug("oauth", "No project ID obtained (will fetch on demand)");
-            }
-          } catch (err) {
-            debug("oauth", "Failed to resolve project ID during login (will fetch on demand)", err);
-          }
-          const tokens = {
-            accessToken: tokenResponse.access_token,
-            refreshToken: tokenResponse.refresh_token || "",
-            expiresAt: Date.now() + tokenResponse.expires_in * 1e3,
-            email,
-            projectId
-          };
-          if (email) {
-            getAccountManager().addAccount(tokens, email);
-          }
+          const result = await completeLogin(code, redirectUri);
           res.writeHead(200, { "Content-Type": "text/html" });
           res.end(`
             <html>
               <body style="font-family: system-ui; padding: 40px; text-align: center;">
                 <h1>Login Successful!</h1>
-                <p>You are now logged in${email ? ` as <strong>${email}</strong>` : ""}.</p>
+                <p>You are now logged in${result.email ? ` as <strong>${result.email}</strong>` : ""}.</p>
                 <p>You can close this window and return to the terminal.</p>
               </body>
             </html>
           `);
           resolved = true;
           server.close();
-          resolve({ success: true, email });
+          resolve(result);
         } catch (err) {
           res.writeHead(500, { "Content-Type": "text/html" });
           res.end("<html><body><h1>Login Failed</h1><p>Token exchange failed.</p></body></html>");
@@ -1203,7 +1245,8 @@ async function loginCommand(options) {
   }
   const result = await startOAuthFlow({
     noBrowser: options.noBrowser,
-    port: options.port
+    port: options.port,
+    manual: options.manual
   });
   if (result.success) {
     resetTokenManager();
@@ -1730,7 +1773,8 @@ function parseModelInfo(modelId, model) {
     remainingPercentage: quotaInfo?.remainingFraction,
     isExhausted: quotaInfo?.isExhausted ?? quotaInfo?.remainingFraction === 0,
     resetTime: quotaInfo?.resetTime,
-    timeUntilResetMs: parseResetTime(quotaInfo?.resetTime)
+    timeUntilResetMs: parseResetTime(quotaInfo?.resetTime),
+    isAutocompleteOnly: modelId.includes("gemini-2.5") || (model.displayName || "").includes("Gemini 2.5")
   };
 }
 function parsePromptCredits(response) {
@@ -2366,7 +2410,8 @@ function parseModelQuota(model) {
     remainingPercentage: quota?.remainingPercentage,
     isExhausted: model.isExhausted ?? quota?.remainingPercentage === 0,
     resetTime: quota?.resetTime,
-    timeUntilResetMs: quota?.timeUntilResetMs
+    timeUntilResetMs: quota?.timeUntilResetMs,
+    isAutocompleteOnly: model.modelId.includes("gemini-2.5") || (model.label || "").includes("Gemini 2.5") || (model.displayName || "").includes("Gemini 2.5")
   };
 }
 
@@ -2465,7 +2510,7 @@ function formatRemaining(model) {
   if (pct >= 25) return `\u{1F7E0} ${pct}%`;
   return `\u{1F534} ${pct}%`;
 }
-function printQuotaTable(snapshot) {
+function printQuotaTable(snapshot, options = {}) {
   const timestamp = new Date(snapshot.timestamp).toLocaleString();
   console.log();
   console.log(`\u{1F4CA} Antigravity Quota Status (via ${snapshot.method.toUpperCase()})`);
@@ -2480,8 +2525,8 @@ function printQuotaTable(snapshot) {
     }
     console.log(`   ${userParts.join(" | ")}`);
   }
-  console.log();
-  if (snapshot.models.length > 0) {
+  const visibleModels = options.allModels ? snapshot.models : snapshot.models.filter((m) => !m.isAutocompleteOnly);
+  if (visibleModels.length > 0) {
     const table = new Table2({
       head: ["Model", "Remaining", "Resets In"],
       style: {
@@ -2489,7 +2534,7 @@ function printQuotaTable(snapshot) {
         border: ["gray"]
       }
     });
-    for (const model of snapshot.models) {
+    for (const model of visibleModels) {
       table.push([
         model.label,
         formatRemaining(model),
@@ -2499,6 +2544,9 @@ function printQuotaTable(snapshot) {
     console.log(table.toString());
   } else {
     console.log("No model quota information available.");
+    if (!options.allModels && snapshot.models.some((m) => m.isAutocompleteOnly)) {
+      console.log("Tip: Use --all-models to see autocomplete models.");
+    }
   }
   console.log();
 }
@@ -2575,13 +2623,16 @@ function renderAccountsTable(accounts) {
 }
 function formatQuotaRemainingBar(remainingPercentage) {
   const width = 10;
-  const filled = Math.round(remainingPercentage / 100 * width);
-  const empty = width - filled;
   const filledChar = "\u2588";
   const emptyChar = "\u2591";
+  if (remainingPercentage === void 0) {
+    return `${emptyChar.repeat(width)} N/A`;
+  }
+  const filled = Math.round(remainingPercentage / 100 * width);
+  const empty = width - filled;
   return `${filledChar.repeat(filled)}${emptyChar.repeat(empty)} ${Math.round(remainingPercentage)}%`;
 }
-function renderAllQuotaTable(results) {
+function renderAllQuotaTable(results, options = {}) {
   if (results.length === 0) {
     console.log("\n\u{1F4ED} No accounts found.");
     console.log("\n\u{1F4A1} Run `antigravity-usage login` to add an account.\n");
@@ -2592,7 +2643,8 @@ function renderAllQuotaTable(results) {
     if (a.status !== "error" && b.status === "error") return -1;
     if (a.status === "error" && b.status === "error") return 0;
     const getRemaining = (result) => {
-      const firstModel = result.snapshot?.models?.[0];
+      const models = options.allModels ? result.snapshot?.models : result.snapshot?.models?.filter((m) => !m.isAutocompleteOnly);
+      const firstModel = models?.[0];
       if (!firstModel) return -1;
       if (firstModel.isExhausted) return 0;
       return firstModel.remainingPercentage ?? -1;
@@ -2643,15 +2695,17 @@ function renderAllQuotaTable(results) {
         credits = `${pc.available} / ${pc.monthly}`;
       }
       let quotaRemaining = "-";
-      if (snapshot?.models && snapshot.models.length > 0) {
-        const minRemaining = Math.min(
-          ...snapshot.models.filter((m) => m.remainingPercentage !== void 0).map((m) => m.remainingPercentage)
-        );
-        if (isFinite(minRemaining)) {
-          const remainingPct = minRemaining * 100;
-          quotaRemaining = formatQuotaRemainingBar(remainingPct);
-        } else if (snapshot.models.some((m) => m.isExhausted)) {
+      const models = snapshot?.models || [];
+      const relevantModels = options.allModels ? models : models.filter((m) => !m.isAutocompleteOnly);
+      if (relevantModels.length > 0) {
+        const percentages = relevantModels.filter((m) => m.remainingPercentage !== void 0).map((m) => m.remainingPercentage);
+        if (percentages.length > 0) {
+          const minRemaining = Math.min(...percentages);
+          quotaRemaining = formatQuotaRemainingBar(minRemaining * 100);
+        } else if (relevantModels.some((m) => m.isExhausted)) {
           quotaRemaining = "\u274C EXHAUSTED";
+        } else {
+          quotaRemaining = formatQuotaRemainingBar(void 0);
         }
       }
       table.push([
@@ -2713,7 +2767,7 @@ async function fetchSingleAccountQuota(options) {
       if (options.json) {
         printQuotaJson(snapshot);
       } else {
-        printQuotaTable(snapshot);
+        printQuotaTable(snapshot, { allModels: options.allModels });
       }
     } finally {
       if (accountSwitched && originalActiveEmail) {
@@ -2787,7 +2841,7 @@ async function fetchAllAccountsQuota(options) {
   if (options.json) {
     console.log(JSON.stringify(results, null, 2));
   } else {
-    renderAllQuotaTable(results);
+    renderAllQuotaTable(results, { allModels: options.allModels });
   }
 }
 async function fetchQuotaForAccount(email, method) {
@@ -3137,7 +3191,7 @@ async function accountsCommand(subcommand, args, options) {
 }
 
 // src/commands/wakeup.ts
-import inquirer from "inquirer";
+import inquirer2 from "inquirer";
 import Table4 from "cli-table3";
 
 // src/wakeup/types.ts
@@ -3835,7 +3889,7 @@ async function configureWakeup() {
     console.log("   antigravity-usage login\n");
     return;
   }
-  const { enabled } = await inquirer.prompt([{
+  const { enabled } = await inquirer2.prompt([{
     type: "confirm",
     name: "enabled",
     message: "Enable auto wake-up?",
@@ -3847,7 +3901,7 @@ async function configureWakeup() {
     console.log("\n\u2705 Auto wake-up disabled");
     return;
   }
-  const { triggerMode } = await inquirer.prompt([{
+  const { triggerMode } = await inquirer2.prompt([{
     type: "list",
     name: "triggerMode",
     message: "Trigger mode:",
@@ -3859,7 +3913,7 @@ async function configureWakeup() {
   }]);
   config.wakeOnReset = triggerMode === "reset";
   if (!config.wakeOnReset) {
-    const { scheduleMode } = await inquirer.prompt([{
+    const { scheduleMode } = await inquirer2.prompt([{
       type: "list",
       name: "scheduleMode",
       message: "Schedule type:",
@@ -3872,7 +3926,7 @@ async function configureWakeup() {
     }]);
     config.scheduleMode = scheduleMode;
     if (scheduleMode === "interval") {
-      const { intervalHours } = await inquirer.prompt([{
+      const { intervalHours } = await inquirer2.prompt([{
         type: "number",
         name: "intervalHours",
         message: "Trigger every N hours:",
@@ -3881,7 +3935,7 @@ async function configureWakeup() {
       }]);
       config.intervalHours = intervalHours;
     } else if (scheduleMode === "daily") {
-      const { dailyTime } = await inquirer.prompt([{
+      const { dailyTime } = await inquirer2.prompt([{
         type: "input",
         name: "dailyTime",
         message: "Time to trigger (HH:MM):",
@@ -3890,7 +3944,7 @@ async function configureWakeup() {
       }]);
       config.dailyTimes = [dailyTime];
     } else if (scheduleMode === "custom") {
-      const { cronExpression } = await inquirer.prompt([{
+      const { cronExpression } = await inquirer2.prompt([{
         type: "input",
         name: "cronExpression",
         message: "Cron expression (min hour day month weekday):",
@@ -3899,7 +3953,7 @@ async function configureWakeup() {
       config.cronExpression = cronExpression;
     }
   } else {
-    const { resetCooldown } = await inquirer.prompt([{
+    const { resetCooldown } = await inquirer2.prompt([{
       type: "number",
       name: "resetCooldown",
       message: "Cooldown between triggers (minutes):",
@@ -3912,7 +3966,7 @@ async function configureWakeup() {
   console.log("\n   \u{1F4E6} Models: claude-sonnet-4-5, gemini-3-flash, gemini-3-pro-low");
   console.log("      (Triggers both Claude and Gemini families)");
   if (accounts.length > 1) {
-    const { selectedAccounts } = await inquirer.prompt([{
+    const { selectedAccounts } = await inquirer2.prompt([{
       type: "checkbox",
       name: "selectedAccounts",
       message: "Select accounts to use:",
@@ -3926,14 +3980,14 @@ async function configureWakeup() {
   } else {
     config.selectedAccounts = void 0;
   }
-  const { customPrompt } = await inquirer.prompt([{
+  const { customPrompt } = await inquirer2.prompt([{
     type: "input",
     name: "customPrompt",
     message: 'Custom wake-up prompt (leave empty for default "hi"):',
     default: config.customPrompt || ""
   }]);
   config.customPrompt = customPrompt || void 0;
-  const { maxTokens } = await inquirer.prompt([{
+  const { maxTokens } = await inquirer2.prompt([{
     type: "number",
     name: "maxTokens",
     message: "Max output tokens (0 = no limit):",
@@ -3947,7 +4001,7 @@ async function configureWakeup() {
   console.log(`   Models: ${config.selectedModels.join(", ")}`);
   console.log(`   Accounts: ${config.selectedAccounts?.join(", ") || "Active account"}`);
   if (!config.wakeOnReset && isCronSupported()) {
-    const { installNow } = await inquirer.prompt([{
+    const { installNow } = await inquirer2.prompt([{
       type: "confirm",
       name: "installNow",
       message: "Install to system cron now?",
@@ -4059,7 +4113,7 @@ async function runTestTrigger() {
   }
   let accountEmail = accounts[0];
   if (accounts.length > 1) {
-    const { selectedAccount } = await inquirer.prompt([{
+    const { selectedAccount } = await inquirer2.prompt([{
       type: "list",
       name: "selectedAccount",
       message: "Select account:",
@@ -4068,13 +4122,13 @@ async function runTestTrigger() {
     accountEmail = selectedAccount;
   }
   const config = loadWakeupConfig();
-  const { modelId } = await inquirer.prompt([{
+  const { modelId } = await inquirer2.prompt([{
     type: "input",
     name: "modelId",
     message: "Model ID to test:",
     default: config?.selectedModels[0] || "claude-sonnet-4-5"
   }]);
-  const { prompt } = await inquirer.prompt([{
+  const { prompt } = await inquirer2.prompt([{
     type: "input",
     name: "prompt",
     message: "Test prompt:",
@@ -4195,10 +4249,10 @@ program.name("antigravity-usage").description("CLI tool to check Antigravity mod
     setDebugMode(true);
   }
 });
-program.command("login").description("Authenticate with Google (adds a new account)").option("--no-browser", "Do not open browser, print URL instead").option("-p, --port <port>", "Port for OAuth callback server", parseInt).action(loginCommand);
+program.command("login").description("Authenticate with Google (adds a new account)").option("--no-browser", "Do not open browser, print URL instead").option("--manual", "Manual login flow (copy-paste URL)").option("-p, --port <port>", "Port for OAuth callback server", parseInt).action(loginCommand);
 program.command("logout [email]").description("Remove stored credentials").option("--all", "Logout from all accounts").action((email, options) => logoutCommand(options, email));
 program.command("status").description("Show current authentication status").option("--all", "Show status for all accounts").option("-a, --account <email>", "Show status for specific account").action(statusCommand);
-program.command("quota", { isDefault: true }).description("Fetch and display quota information").option("--json", "Output as JSON").option("-m, --method <method>", "Method to use: auto (default), local, or google", "auto").option("--all", "Show quota for all accounts").option("-a, --account <email>", "Show quota for specific account").option("--refresh", "Force refresh (ignore cache)").action(quotaCommand);
+program.command("quota", { isDefault: true }).description("Fetch and display quota information").option("--json", "Output as JSON").option("-m, --method <method>", "Method to use: auto (default), local, or google", "auto").option("--all", "Show quota for all accounts").option("-a, --account <email>", "Show quota for specific account").option("--refresh", "Force refresh (ignore cache)").option("--all-models", "Include autocomplete models (Gemini 2.5) in quota display").action(quotaCommand);
 var accountsCmd = program.command("accounts").description("Manage multiple accounts");
 accountsCmd.command("list").description("List all accounts").option("--refresh", "Show refresh tip").action((options) => accountsCommand("list", [], options));
 accountsCmd.command("add").description("Add a new account (triggers OAuth login)").action(() => accountsCommand("add", [], {}));