gusage 1.1.1 → 1.2.0

package/README.md

@@ -67,6 +67,7 @@ gusage --json --watch 5s | jq .
 - `-n, --notify [threshold]`: Send a desktop notification when a model drops below the threshold percent (default: `20`). Requires `--watch`.
 - `-j, --json`: Output raw JSON instead of a table. Can be combined with `--watch` for streaming data.
 - `--no-color`: Disable color output (also respects `NO_COLOR` env var).
+- `-v, --version`: Show version.
 
 ## Requirements
 

package/dist/index.js

@@ -7,6 +7,56 @@ import path from "path";
 import os from "os";
 import crypto from "crypto";
 import { parseArgs } from "util";
+import { execSync, spawn } from "child_process";
+// package.json
+var package_default = {
+  name: "gusage",
+  version: "1.2.0",
+  description: "A standalone CLI to export Gemini CLI quota and usage statistics",
+  module: "index.ts",
+  type: "module",
+  bin: {
+    gusage: "dist/index.js"
+  },
+  files: [
+    "dist/index.js",
+    "README.md",
+    "LICENSE"
+  ],
+  author: "Abdellah Hariti <haritiabdellah@gmail.com>",
+  license: "MIT",
+  private: false,
+  scripts: {
+    format: "prettier --write .",
+    prepare: "husky",
+    build: "bun build ./index.ts --outfile dist/index.js && chmod +x dist/index.js",
+    prepack: "bun run build"
+  },
+  "lint-staged": {
+    "*": "prettier --write --ignore-unknown",
+    "*.ts": "bash -c 'bun x tsc --noEmit'"
+  },
+  devDependencies: {
+    "@types/bun": "latest",
+    husky: "^9.1.7",
+    "lint-staged": "^16.2.7",
+    prettier: "^3.8.1"
+  },
+  peerDependencies: {
+    typescript: "^5"
+  },
+  keywords: [
+    "gemini-cli",
+    "gemini",
+    "google-gemini",
+    "quota",
+    "usage",
+    "stats",
+    "monitoring"
+  ]
+};
+
+// index.ts
 var OAUTH_CLIENT_ID = "681255809395-oo8ft2oprdrnp9e3aqf6av3hmdib135j.apps.googleusercontent.com";
 var OAUTH_CLIENT_SECRET = "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";
 var CODE_ASSIST_ENDPOINT = "https://cloudcode-pa.googleapis.com";
@@ -36,6 +86,56 @@ function formatRelativeTime(dateString) {
     return `${h}h ${m}m`;
   return `${m}m`;
 }
+function sendNotification(title, message) {
+  const spawnDetached = (command, args) => {
+    const child = spawn(command, args, { detached: true, stdio: "ignore" });
+    child.on("error", () => {});
+    child.unref();
+  };
+  if (process.platform === "darwin") {
+    const escapedMessage = message.replace(/"/g, "\\\"");
+    const escapedTitle = title.replace(/"/g, "\\\"");
+    const script = `display notification "${escapedMessage}" with title "${escapedTitle}"`;
+    spawnDetached("osascript", ["-e", script]);
+  } else if (process.platform === "linux") {
+    spawnDetached("notify-send", [title, message]);
+  } else if (process.platform === "win32") {
+    const command = `Add-Type -AssemblyName System.Windows.Forms; $g = [System.Windows.Forms.NotifyIcon]::new(); $g.Icon = [System.Drawing.Icon]::ExtractAssociatedIcon("$PSHOME\\powershell.exe"); $g.Visible = $true; $g.ShowBalloonTip(5000, "${title.replace(/"/g, '`"')}", "${message.replace(/"/g, '`"')}", [System.Windows.Forms.ToolTipIcon]::Info);`;
+    spawnDetached("powershell", ["-Command", command]);
+  }
+}
+function checkNotificationSupport() {
+  if (process.platform === "linux") {
+    try {
+      execSync("which notify-send", { stdio: "ignore" });
+      return true;
+    } catch {
+      return false;
+    }
+  }
+  return true;
+}
+function parseNotifyThresholdArg(raw) {
+  if (raw === undefined)
+    return null;
+  const thresholdPct = Number(raw);
+  if (!Number.isFinite(thresholdPct) || thresholdPct < 0 || thresholdPct > 100) {
+    throw new Error("Error: --notify threshold must be a number between 0 and 100.");
+  }
+  return thresholdPct / 100;
+}
+function validateNotifyRuntime(notifyThreshold, isWatching, hasNotificationSupport) {
+  if (notifyThreshold === null)
+    return null;
+  if (!isWatching)
+    return "Error: --notify requires --watch.";
+  if (!hasNotificationSupport)
+    return "Error: --notify is not supported on this system (missing notify-send).";
+  return null;
+}
+function shouldSendThresholdNotification(prevFraction, currentFraction, threshold) {
+  return prevFraction !== undefined && prevFraction > threshold && currentFraction <= threshold;
+}
 function parseVersion(modelId) {
   const match = modelId.match(/gemini-(\d+)(?:\.(\d+))?-(.*)/);
   if (match) {
@@ -89,18 +189,25 @@ function padVisual(str, width, side = "right") {
   return side === "right" ? str + pad : pad + str;
 }
 async function main() {
+  const packageVersion = package_default.version;
   const args = Bun.argv.slice(2);
   const watchIdx = args.findIndex((a) => a === "--watch" || a === "-w");
   if (watchIdx !== -1 && (watchIdx === args.length - 1 || args[watchIdx + 1].startsWith("-"))) {
     args.splice(watchIdx + 1, 0, "10s");
   }
+  const notifyIdx = args.findIndex((a) => a === "--notify" || a === "-n");
+  if (notifyIdx !== -1 && (notifyIdx === args.length - 1 || args[notifyIdx + 1].startsWith("-"))) {
+    args.splice(notifyIdx + 1, 0, "20");
+  }
   const { values } = parseArgs({
     args,
     options: {
       help: { type: "boolean", short: "h" },
-      "output-format": { type: "string", short: "o", default: "table" },
+      version: { type: "boolean", short: "v" },
+      json: { type: "boolean", short: "j" },
       "no-color": { type: "boolean" },
-      watch: { type: "string", short: "w" }
+      watch: { type: "string", short: "w" },
+      notify: { type: "string", short: "n" }
     },
     strict: true
   });
@@ -110,19 +217,22 @@ Usage: gusage [options]
 
 Options:
   -h, --help                Show this help message
+  -v, --version             Show version and exit
   -w, --watch [interval]    Update live every interval (default: 10s).
                             Supports combined units: 20s, 5m, 1m20s.
-  -o, --output-format <fmt> Output format: table (default), json
+  -n, --notify [threshold]  Show a critical OS notification if any model falls below
+                            the threshold (default: 20%; requires --watch).
+  -j, --json                Output raw JSON instead of a table
   --no-color                Disable color output
     `);
     return;
   }
-  const outputFormat = values["output-format"];
-  if (outputFormat !== "json" && outputFormat !== "table") {
-    console.error(`Error: Unsupported output format "${outputFormat}". Use "table" or "json".`);
-    process.exit(1);
+  if (values.version) {
+    console.log(packageVersion);
+    return;
   }
   const isWatching = values.watch !== undefined;
+  const isJson = values.json === true;
   let intervalMs = 1e4;
   let intervalStr = "10s";
   if (values.watch) {
@@ -146,6 +256,20 @@ Options:
     }
   }
   const useColor = !values["no-color"] && !process.env.NO_COLOR && process.stdout.isTTY;
+  let notifyThreshold;
+  const lastFractions = new Map;
+  try {
+    notifyThreshold = parseNotifyThresholdArg(values.notify);
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    console.error(message);
+    process.exit(1);
+  }
+  const notifyRuntimeError = validateNotifyRuntime(notifyThreshold, isWatching, checkNotificationSupport());
+  if (notifyRuntimeError) {
+    console.error(notifyRuntimeError);
+    process.exit(1);
+  }
   const colors = {
     reset: useColor ? "\x1B[0m" : "",
     dim: useColor ? "\x1B[2m" : "",
@@ -242,11 +366,22 @@ Options:
           return vB.minor - vA.minor;
         return vB.suffix.localeCompare(vA.suffix);
       });
+      if (notifyThreshold !== null) {
+        for (const b of quotaData.buckets) {
+          const fraction = b.remainingFraction ?? 0;
+          const modelId = b.modelId;
+          const prevFraction = lastFractions.get(modelId);
+          if (shouldSendThresholdNotification(prevFraction, fraction, notifyThreshold)) {
+            sendNotification("Gemini Quota Alert", `Model ${modelId} has dropped below ${Math.round(notifyThreshold * 100)}%.`);
+          }
+          lastFractions.set(modelId, fraction);
+        }
+      }
     }
-    if (isWatching)
+    if (isWatching && !isJson)
       process.stdout.write(colors.clear);
-    if (outputFormat === "json") {
-      console.log(JSON.stringify(quotaData.buckets || [], null, 2));
+    if (isJson) {
+      console.log(JSON.stringify(quotaData.buckets || []));
     } else {
       if (!quotaData.buckets || quotaData.buckets.length === 0) {
         console.log("No quota data available.");
@@ -286,7 +421,7 @@ Options:
           console.log("");
         }
       });
-      if (isWatching) {
+      if (isWatching && !isJson) {
         console.log(`
 ${colors.dim}Updating every ${intervalStr}, press q to quit${colors.reset}`);
       }
@@ -374,5 +509,8 @@ async function refreshAccessToken(refreshToken) {
   return response.json();
 }
 export {
+  validateNotifyRuntime,
+  shouldSendThresholdNotification,
+  parseNotifyThresholdArg,
   main as default
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gusage",
-  "version": "1.1.1",
+  "version": "1.2.0",
   "description": "A standalone CLI to export Gemini CLI quota and usage statistics",
   "module": "index.ts",
   "type": "module",
@@ -18,7 +18,8 @@
   "scripts": {
     "format": "prettier --write .",
     "prepare": "husky",
-    "build": "bun build ./index.ts --outfile dist/index.js && chmod +x dist/index.js"
+    "build": "bun build ./index.ts --outfile dist/index.js && chmod +x dist/index.js",
+    "prepack": "bun run build"
   },
   "lint-staged": {
     "*": "prettier --write --ignore-unknown",