when-does-my-quota-refresh 1.1.0 → 1.2.0

package/dist/cli/index.js

@@ -1,4 +1,5 @@
 #!/usr/bin/env node
+import "../chunk-H7BEO7IF.js";
 import {
   OAuthClient,
   SourceUnavailableError,
@@ -15,7 +16,7 @@ import {
   saveConfig,
   setActiveAccount,
   setVerbose
-} from "../chunk-ST6OOHQI.js";
+} from "../chunk-VLP22VAR.js";
 
 // src/cli/index.ts
 import { Command } from "commander";
@@ -570,11 +571,11 @@ var ConnectClient = class {
 };
 
 // src/sources/local/parser.ts
-function parseQuotaSummary(summary, email) {
+function parseQuotaSummary(summary, email, method = "local") {
   debug("parser", `Parsing quota summary: ${summary.groups.length} groups`);
   return {
     timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-    method: "local",
+    method,
     email,
     groups: summary.groups.map((g) => ({
       displayName: g.displayName,
@@ -589,11 +590,11 @@ function parseQuotaSummary(summary, email) {
     }))
   };
 }
-function parseUserStatus(userStatus) {
+function parseUserStatus(userStatus, method = "local") {
   debug("parser", "Parsing GetUserStatus fallback");
   const snapshot = {
     timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-    method: "local",
+    method,
     email: userStatus.email,
     groups: []
   };
@@ -667,7 +668,7 @@ function isGeminiModel(modelId) {
 }
 function detectAutocomplete(modelId, label) {
   const text = `${modelId} ${label || ""}`.toLowerCase();
-  return text.includes("autocomplete") || text.includes("gemini-2.5") || text.includes("gemini 2.5");
+  return text.includes("autocomplete");
 }
 function clampFraction(value) {
   if (value === void 0 || value === null) return 1;
@@ -841,10 +842,10 @@ var CliSource = class {
         const summary = await client.fetchQuotaSummary();
         if (summary && summary.groups.length > 0) {
           const email = (await client.fetchUserStatus())?.email;
-          return parseQuotaSummary(summary, email);
+          return parseQuotaSummary(summary, email, "cli");
         }
         const userStatus = await client.fetchUserStatus();
-        if (userStatus) return parseUserStatus(userStatus);
+        if (userStatus) return parseUserStatus(userStatus, "cli");
       } catch {
       }
     }
@@ -903,20 +904,25 @@ var OAuthSource = class {
   }
   async fetchQuota() {
     debug("oauth", "Fetching quota via Cloud Code API");
+    let email;
+    try {
+      const userInfo = await this.client.fetchUserInfo();
+      email = userInfo.email;
+    } catch {
+      debug("oauth", "Could not fetch user info for email");
+    }
     try {
       const summaryRaw = await this.client.fetchQuotaSummary();
       const summary = this.parseRemoteSummary(summaryRaw);
       if (summary && summary.groups.length > 0) {
-        const statusRaw2 = await this.client.fetchUserStatus();
-        const email = this.extractEmail(statusRaw2);
         debug("oauth", "Got quota summary via OAuth");
-        return parseQuotaSummary(summary, email);
+        return parseQuotaSummary(summary, email, "oauth");
       }
       const statusRaw = await this.client.fetchUserStatus();
       const status = this.parseRemoteStatus(statusRaw);
       if (status) {
         debug("oauth", "Got user status via OAuth");
-        return parseUserStatus(status);
+        return parseUserStatus(status, "oauth");
       }
     } catch (err) {
       debug("oauth", `Fetch failed: ${err}`);
@@ -950,13 +956,12 @@ var OAuthSource = class {
    */
   parseRemoteStatus(raw) {
     if (typeof raw !== "object" || raw === null) return null;
-    return { raw };
-  }
-  extractEmail(raw) {
-    if (typeof raw !== "object" || raw === null) return void 0;
     const data = raw;
-    const userStatus = data.userStatus;
-    return userStatus?.email || data.email;
+    const userStatus = data.userStatus || data;
+    return {
+      email: userStatus.email,
+      raw
+    };
   }
 };
 
@@ -1230,17 +1235,160 @@ function closeStore() {
   }
 }
 
+// src/plugins/builtin/notifications.ts
+var NotificationPlugin = class {
+  name = "notifications";
+  version = "1.0.0";
+  notifier = null;
+  notifiedLow = /* @__PURE__ */ new Set();
+  async onInit(ctx) {
+    try {
+      const { default: notifier } = await import("node-notifier");
+      this.notifier = notifier;
+      debug("notifications", "Notification plugin initialized");
+    } catch {
+      debug("notifications", "node-notifier not available, notifications disabled");
+    }
+  }
+  async onQuotaUpdate(snapshot, ctx) {
+    for (const group of snapshot.groups) {
+      for (const bucket of group.buckets) {
+        if (bucket.remainingFraction > 0.8) {
+          this.notifiedLow.delete(`${group.displayName}:${bucket.displayName}`);
+        }
+      }
+    }
+  }
+  async onQuotaLow(group, bucket, remaining, ctx) {
+    const key = `${group}:${bucket}`;
+    if (this.notifiedLow.has(key)) return;
+    const pct = Math.round(remaining * 100);
+    const title = "when-does-my-quota-refresh";
+    const message = `\u26A0\uFE0F Low quota: ${group} \u2014 ${bucket} at ${pct}%`;
+    ctx.log(message);
+    this.notify(title, message);
+    this.notifiedLow.add(key);
+  }
+  async onReset(group, nextReset, ctx) {
+    const title = "when-does-my-quota-refresh";
+    const message = `\u2705 ${group} quota has reset!`;
+    ctx.log(message);
+    this.notify(title, message);
+  }
+  notify(title, message) {
+    if (!this.notifier) return;
+    try {
+      this.notifier.notify({ title, message, timeout: 5 });
+    } catch (err) {
+      debug("notifications", `Notification failed: ${err}`);
+    }
+  }
+};
+
+// src/plugins/index.ts
+var PluginRegistry = class {
+  plugins = [];
+  context;
+  constructor(context) {
+    this.context = {
+      notify: context?.notify || (() => {
+      }),
+      log: context?.log || ((msg) => console.log(msg))
+    };
+  }
+  /**
+   * Register a plugin
+   */
+  async register(plugin) {
+    this.plugins.push(plugin);
+    if (plugin.onInit) {
+      await plugin.onInit(this.context);
+    }
+    debug("plugins", `Registered: ${plugin.name}@${plugin.version}`);
+  }
+  /**
+   * Load built-in plugins
+   */
+  async loadBuiltins() {
+    await this.register(new NotificationPlugin());
+  }
+  /**
+   * Notify all plugins of a quota update
+   */
+  async notifyQuotaUpdate(snapshot) {
+    for (const plugin of this.plugins) {
+      if (plugin.onQuotaUpdate) {
+        try {
+          await plugin.onQuotaUpdate(snapshot, this.context);
+        } catch (err) {
+          debug("plugins", `Error in ${plugin.name}.onQuotaUpdate: ${err}`);
+        }
+      }
+    }
+    for (const group of snapshot.groups) {
+      for (const bucket of group.buckets) {
+        if (bucket.remainingFraction < 0.15) {
+          await this.notifyQuotaLow(group.displayName, bucket.displayName, bucket.remainingFraction);
+        }
+      }
+    }
+  }
+  /**
+   * Notify all plugins of low quota
+   */
+  async notifyQuotaLow(group, bucket, remaining) {
+    for (const plugin of this.plugins) {
+      if (plugin.onQuotaLow) {
+        try {
+          await plugin.onQuotaLow(group, bucket, remaining, this.context);
+        } catch (err) {
+          debug("plugins", `Error in ${plugin.name}.onQuotaLow: ${err}`);
+        }
+      }
+    }
+  }
+  /**
+   * Destroy all plugins
+   */
+  async destroy() {
+    for (const plugin of this.plugins) {
+      if (plugin.onDestroy) {
+        try {
+          await plugin.onDestroy();
+        } catch (err) {
+          debug("plugins", `Error in ${plugin.name}.onDestroy: ${err}`);
+        }
+      }
+    }
+    this.plugins = [];
+  }
+  /**
+   * Get registered plugin names
+   */
+  getNames() {
+    return this.plugins.map((p) => p.name);
+  }
+};
+
 // src/sdk/quota.ts
 var QuotaClient = class {
   registry;
   cache;
+  plugins;
   verbose;
   constructor(options) {
     this.verbose = options?.verbose || false;
-    setVerbose(this.verbose);
+    if (options?.verbose !== void 0) {
+      setVerbose(this.verbose);
+    }
     const config = loadConfig();
     this.registry = new SourceRegistry();
     this.cache = new QuotaCache(options?.cacheTtlMs || config.cacheTtlMs);
+    this.plugins = new PluginRegistry();
+    if (config.notifications) {
+      this.plugins.loadBuiltins().catch(() => {
+      });
+    }
     this.registerOAuthSources();
   }
   /**
@@ -1251,11 +1399,19 @@ var QuotaClient = class {
       return this.fetchAllAccounts(options);
     }
     if (!options?.refresh) {
+      const tokens = getAllAccountTokens();
+      for (const token of tokens) {
+        if (!token.email) continue;
+        const cached = this.cache.get(token.email, options?.source || "auto");
+        if (cached) return cached;
+      }
     }
     const snapshot = await this.registry.fetchQuota(options?.source);
-    if (snapshot.email) {
+    if (snapshot.email && snapshot.email !== "unknown") {
       this.cache.set(snapshot.email, snapshot.method, snapshot);
       storeSnapshot(snapshot);
+      this.plugins.notifyQuotaUpdate(snapshot).catch(() => {
+      });
     }
     return snapshot;
   }
@@ -1659,8 +1815,8 @@ function registerAccountsCommand(program2) {
   accounts.command("add").description("Login a new Google account").option("--manual", "Manual login for headless/SSH (get URL instead of browser)").action(async (opts) => {
     try {
       if (opts.manual) {
-        const { getManualLoginUrl: getManualLoginUrl2, completeManualLogin: completeManualLogin2 } = await import("../auth-XYJBJA2Q.js");
-        const url = getManualLoginUrl2();
+        const { getManualLoginUrl, completeManualLogin } = await import("../auth-GX4ZIG5O.js");
+        const url = getManualLoginUrl();
         console.log(chalk6.bold("\nOpen this URL in your browser:\n"));
         console.log(chalk6.cyan(url));
         console.log(chalk6.dim("\nPaste the authorization code here:"));
@@ -1669,7 +1825,7 @@ function registerAccountsCommand(program2) {
           name: "code",
           message: "Authorization code:"
         }]);
-        const email = await completeManualLogin2(code);
+        const email = await completeManualLogin(code);
         console.log(chalk6.green(`
 \u2713 Logged in as ${email}`));
       } else {
@@ -2086,7 +2242,15 @@ function registerWakeupCommand(program2) {
         default: true
       }
     ]);
-    config.wakeupAccounts = answers.allAccounts ? [] : [];
+    config.wakeupMode = answers.mode;
+    if (answers.intervalHours) config.wakeupIntervalHours = answers.intervalHours;
+    if (answers.dailyTimes) config.wakeupDailyTimes = answers.dailyTimes;
+    if (!answers.allAccounts) {
+      const { listAccounts: listAccounts2 } = await import("../accounts-FH6AKZT7.js");
+      config.wakeupAccounts = listAccounts2().map((a) => a.email);
+    } else {
+      config.wakeupAccounts = [];
+    }
     saveConfig(config);
     console.log(chalk11.green("\n  \u2713 Configuration saved"));
     console.log(chalk11.dim(`  Run \`aq wakeup install\` to activate`));
@@ -2120,8 +2284,11 @@ function registerWakeupCommand(program2) {
   wakeup.command("status").description("Check wakeup status").action(() => {
     const config = loadConfig();
     console.log(chalk11.bold("\n  Wakeup Status\n"));
+    console.log(`  Mode: ${config.wakeupMode}`);
     console.log(`  Models: ${config.wakeupModels.join(", ")}`);
     console.log(`  Accounts: ${config.wakeupAccounts.length > 0 ? config.wakeupAccounts.join(", ") : "all"}`);
+    if (config.wakeupMode === "interval") console.log(`  Interval: every ${config.wakeupIntervalHours}h`);
+    if (config.wakeupMode === "daily") console.log(`  Daily at: ${config.wakeupDailyTimes}`);
     if (isWindows()) {
       const installed = checkWindowsTask();
       console.log(`  Scheduler: ${installed ? chalk11.green("Installed") : chalk11.dim("Not installed")}`);
@@ -2142,12 +2309,32 @@ function registerWakeupCommand(program2) {
     }
   });
 }
+function escapeShellArg(arg) {
+  return arg.replace(/'/g, "'\\''");
+}
 function installCrontab() {
   const aqPath = process.argv[1];
+  const safeAqPath = escapeShellArg(aqPath);
   const config = loadConfig();
   uninstallCrontab();
-  const intervalHours = 6;
-  const cronLine = `0 */${intervalHours} * * * "${aqPath}" quota --refresh --json > /dev/null 2>&1 ${CRON_MARKER}`;
+  const intervalHours = config.wakeupIntervalHours || 6;
+  let cronLine;
+  if (config.wakeupMode === "daily" && config.wakeupDailyTimes) {
+    const hours = config.wakeupDailyTimes.split(",").map((h) => parseInt(h.trim(), 10)).filter((h) => !isNaN(h));
+    for (const hour of hours) {
+      cronLine = `0 ${hour} * * * "${safeAqPath}" quota --refresh --json > /dev/null 2>&1 ${CRON_MARKER}`;
+      try {
+        const existing = execSync("crontab -l 2>/dev/null", { encoding: "utf-8" }).trim();
+        const newCron = existing ? `${existing}
+${cronLine}` : cronLine;
+        execSync(`echo '${newCron}' | crontab -`, { encoding: "utf-8" });
+      } catch {
+        execSync(`echo '${cronLine}' | crontab -`, { encoding: "utf-8" });
+      }
+    }
+    return;
+  }
+  cronLine = `0 */${intervalHours} * * * "${safeAqPath}" quota --refresh --json > /dev/null 2>&1 ${CRON_MARKER}`;
   try {
     const existing = execSync("crontab -l 2>/dev/null", { encoding: "utf-8" }).trim();
     const newCron = existing ? `${existing}
@@ -2161,7 +2348,8 @@ function uninstallCrontab() {
   try {
     const existing = execSync("crontab -l 2>/dev/null", { encoding: "utf-8" });
     const filtered = existing.split("\n").filter((l) => !l.includes(CRON_MARKER)).join("\n").trim();
-    execSync(`echo '${filtered}' | crontab -`, { encoding: "utf-8" });
+    execSync(`printf '%s
+' '${filtered.replace(/'/g, "'\\''")}' | crontab -`, { encoding: "utf-8" });
   } catch {
   }
 }
@@ -2177,12 +2365,14 @@ function installWindowsTask() {
   const aqPath = process.argv[1];
   const xmlPath = join2(homedir(), "when-does-my-quota-refresh-wakeup.xml");
   uninstallWindowsTask();
+  const config = loadConfig();
+  const intervalHours = config.wakeupIntervalHours || 6;
   const xml = `<?xml version="1.0" encoding="UTF-16"?>
 <Task version="1.4" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">
   <Triggers>
     <TimeTrigger>
       <Repetition>
-        <Interval>PT6H</Interval>
+        <Interval>PT${intervalHours}H</Interval>
       </Repetition>
       <StartBoundary>2024-01-01T09:00:00</StartBoundary>
     </TimeTrigger>