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

package/dist/accounts-FH6AKZT7.js

@@ -0,0 +1,27 @@
+#!/usr/bin/env node
+import "./chunk-H7BEO7IF.js";
+import {
+  completeManualLogin,
+  getAccountTokens,
+  getActiveAccountTokens,
+  getAllAccountTokens,
+  getManualLoginUrl,
+  listAccounts,
+  loginAccount,
+  removeAccount,
+  saveAccountTokens,
+  setActiveAccount
+} from "./chunk-VLP22VAR.js";
+export {
+  completeManualLogin,
+  getAccountTokens,
+  getActiveAccountTokens,
+  getAllAccountTokens,
+  getManualLoginUrl,
+  listAccounts,
+  loginAccount,
+  removeAccount,
+  saveAccountTokens,
+  setActiveAccount
+};
+//# sourceMappingURL=accounts-FH6AKZT7.js.map

package/dist/{auth-K5ETZYGY.js → auth-GX4ZIG5O.js}

@@ -3,10 +3,10 @@ import {
   completeManualLogin,
   getManualLoginUrl,
   loginAccount
-} from "./chunk-24ILGZJ2.js";
+} from "./chunk-VLP22VAR.js";
 export {
   completeManualLogin,
   getManualLoginUrl,
   loginAccount
 };
-//# sourceMappingURL=auth-K5ETZYGY.js.map
+//# sourceMappingURL=auth-GX4ZIG5O.js.map

package/dist/auth-GX4ZIG5O.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/chunk-H7BEO7IF.js

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+//# sourceMappingURL=chunk-H7BEO7IF.js.map

package/dist/chunk-H7BEO7IF.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/{chunk-24ILGZJ2.js → chunk-VLP22VAR.js}

@@ -4,6 +4,21 @@
 import https from "https";
 import { URL } from "url";
 
+// src/core/logger.ts
+var _verbose = false;
+function setVerbose(verbose) {
+  _verbose = verbose;
+}
+function debug(module, msg, data) {
+  if (!_verbose) return;
+  const prefix = `[${module}]`;
+  if (data !== void 0) {
+    console.error(`${prefix} ${msg}`, data);
+  } else {
+    console.error(`${prefix} ${msg}`);
+  }
+}
+
 // src/sources/cloud-code-api.ts
 var CLOUD_CODE_ENDPOINTS = {
   /** Load code assist configuration */
@@ -30,6 +45,163 @@ var OAUTH_CLIENT_SECRET = "d-FL95Q19q7MQmFpd7hHD0Ty";
 var OAUTH_SCOPES = [
   "https://www.googleapis.com/auth/cloud-platform"
 ].join(" ");
+var OAUTH_REDIRECT_URI = "http://127.0.0.1:0/callback";
+
+// src/sdk/accounts/storage.ts
+import { readFileSync as readFileSync2, writeFileSync as writeFileSync2, existsSync as existsSync2, mkdirSync as mkdirSync2 } from "fs";
+import { join as join2, dirname } from "path";
+
+// src/core/config.ts
+import { readFileSync, writeFileSync, mkdirSync, existsSync } from "fs";
+import { join } from "path";
+import { homedir } from "os";
+var CONFIG_DIR_NAME = "when-does-my-quota-refresh";
+var DEFAULT_CONFIG = {
+  defaultMode: "local",
+  cacheTtlMs: 5 * 60 * 1e3,
+  // 5 minutes
+  refreshIntervalMs: 5 * 60 * 1e3,
+  wakeupModels: ["claude-sonnet-4-5", "gemini-3-flash", "gemini-3-pro-low"],
+  wakeupAccounts: [],
+  wakeupMode: "smart",
+  wakeupIntervalHours: 6,
+  wakeupDailyTimes: "9,17",
+  notifications: true,
+  plugins: [],
+  daemon: {
+    pollIntervalMs: 60 * 1e3,
+    enabled: false
+  }
+};
+function getConfigDir() {
+  const home = homedir();
+  switch (process.platform) {
+    case "darwin":
+      return join(home, "Library", "Application Support", CONFIG_DIR_NAME);
+    case "win32":
+      return join(process.env.APPDATA || join(home, "AppData", "Roaming"), CONFIG_DIR_NAME);
+    default:
+      return join(home, ".config", CONFIG_DIR_NAME);
+  }
+}
+function getConfigPath() {
+  return join(getConfigDir(), "config.json");
+}
+function getDataDir() {
+  return join(getConfigDir(), "data");
+}
+function ensureDirs() {
+  const configDir = getConfigDir();
+  const dataDir = getDataDir();
+  if (!existsSync(configDir)) mkdirSync(configDir, { recursive: true });
+  if (!existsSync(dataDir)) mkdirSync(dataDir, { recursive: true });
+}
+function loadConfig() {
+  try {
+    const path = getConfigPath();
+    if (!existsSync(path)) return DEFAULT_CONFIG;
+    const raw = readFileSync(path, "utf-8");
+    const parsed = JSON.parse(raw);
+    debug("config", `Loaded config from ${path}`);
+    return { ...DEFAULT_CONFIG, ...parsed };
+  } catch (err) {
+    debug("config", "Failed to load config, using defaults", err);
+    return DEFAULT_CONFIG;
+  }
+}
+function saveConfig(config) {
+  ensureDirs();
+  const path = getConfigPath();
+  writeFileSync(path, JSON.stringify(config, null, 2));
+  debug("config", `Saved config to ${path}`);
+}
+function getConfigDirPath() {
+  return getConfigDir();
+}
+function getDataDirPath() {
+  return getDataDir();
+}
+
+// src/sdk/accounts/storage.ts
+var TOKENS_DIR = "accounts";
+function getStorePath() {
+  return join2(getConfigDirPath(), TOKENS_DIR, "tokens.json");
+}
+function loadStore() {
+  const path = getStorePath();
+  if (!existsSync2(path)) return { accounts: [] };
+  try {
+    return JSON.parse(readFileSync2(path, "utf-8"));
+  } catch {
+    return { accounts: [] };
+  }
+}
+function saveStore(store) {
+  ensureDirs();
+  const storePath = getStorePath();
+  mkdirSync2(dirname(storePath), { recursive: true });
+  writeFileSync2(storePath, JSON.stringify(store, null, 2));
+  debug("account-storage", `Saved ${store.accounts.length} account(s)`);
+}
+function saveAccountTokens(email, tokens) {
+  const store = loadStore();
+  const existing = store.accounts.findIndex((a) => a.email === email);
+  const account = {
+    email,
+    tokens,
+    addedAt: (/* @__PURE__ */ new Date()).toISOString()
+  };
+  if (existing >= 0) {
+    store.accounts[existing] = account;
+  } else {
+    store.accounts.push(account);
+  }
+  if (store.accounts.length === 1) {
+    store.activeEmail = email;
+  }
+  saveStore(store);
+}
+function getAccountTokens(email) {
+  const store = loadStore();
+  const account = store.accounts.find((a) => a.email === email);
+  return account?.tokens || null;
+}
+function getActiveAccountTokens() {
+  const store = loadStore();
+  if (!store.activeEmail) return null;
+  const account = store.accounts.find((a) => a.email === store.activeEmail);
+  return account?.tokens || null;
+}
+function setActiveAccount(email) {
+  const store = loadStore();
+  if (!store.accounts.find((a) => a.email === email)) return false;
+  store.activeEmail = email;
+  saveStore(store);
+  return true;
+}
+function removeAccount(email) {
+  const store = loadStore();
+  const idx = store.accounts.findIndex((a) => a.email === email);
+  if (idx < 0) return false;
+  store.accounts.splice(idx, 1);
+  if (store.activeEmail === email) {
+    store.activeEmail = store.accounts[0]?.email;
+  }
+  saveStore(store);
+  return true;
+}
+function listAccounts() {
+  const store = loadStore();
+  return store.accounts.map((a) => ({
+    email: a.email,
+    isActive: a.email === store.activeEmail,
+    addedAt: a.addedAt
+  }));
+}
+function getAllAccountTokens() {
+  const store = loadStore();
+  return store.accounts.map((a) => a.tokens);
+}
 
 // src/sources/oauth/client.ts
 var OAuthClient = class {
@@ -40,10 +212,10 @@ var OAuthClient = class {
   /**
    * Generate the OAuth authorization URL for user login
    */
-  getAuthUrl() {
+  getAuthUrl(redirectUri) {
     const params = new URLSearchParams({
       client_id: OAUTH_CLIENT_ID,
-      redirect_uri: "http://127.0.0.1:0/callback",
+      redirect_uri: redirectUri || OAUTH_REDIRECT_URI,
       response_type: "code",
       scope: OAUTH_SCOPES,
       access_type: "offline",
@@ -54,12 +226,12 @@ var OAuthClient = class {
   /**
    * Exchange an authorization code for tokens
    */
-  async exchangeCode(code) {
+  async exchangeCode(code, redirectUri) {
     const body = new URLSearchParams({
       code,
       client_id: OAUTH_CLIENT_ID,
       client_secret: OAUTH_CLIENT_SECRET,
-      redirect_uri: "http://127.0.0.1:0/callback",
+      redirect_uri: redirectUri || OAUTH_REDIRECT_URI,
       grant_type: "authorization_code"
     });
     const response = await this.postUrlEncoded("https://oauth2.googleapis.com/token", body);
@@ -91,6 +263,14 @@ var OAuthClient = class {
       accessToken: data.access_token,
       expiresAt: Date.now() + data.expires_in * 1e3
     };
+    if (this.tokens.email) {
+      try {
+        saveAccountTokens(this.tokens.email, this.tokens);
+        debug("oauth", `Persisted refreshed token for ${this.tokens.email}`);
+      } catch (err) {
+        debug("oauth", `Failed to persist refreshed token: ${err}`);
+      }
+    }
     return this.tokens;
   }
   /**
@@ -136,6 +316,43 @@ var OAuthClient = class {
       token
     );
   }
+  /**
+   * Fetch the authenticated user's profile (email, name, etc.)
+   */
+  async fetchUserInfo() {
+    const token = await this.getAccessToken();
+    return new Promise((resolve, reject) => {
+      const options = {
+        hostname: "www.googleapis.com",
+        path: "/oauth2/v3/userinfo",
+        method: "GET",
+        headers: {
+          "Authorization": `Bearer ${token}`
+        },
+        timeout: 1e4
+      };
+      const req = https.request(options, (res) => {
+        let data = "";
+        res.on("data", (chunk) => {
+          data += chunk;
+        });
+        res.on("end", () => {
+          try {
+            const info = JSON.parse(data);
+            resolve({ email: info.email, name: info.name });
+          } catch {
+            resolve({});
+          }
+        });
+      });
+      req.on("error", () => resolve({}));
+      req.on("timeout", () => {
+        req.destroy();
+        resolve({});
+      });
+      req.end();
+    });
+  }
   // ── HTTP Helpers ────────────────────────────────────────────────
   authenticatedPost(url, body, token) {
     return new Promise((resolve, reject) => {
@@ -158,6 +375,10 @@ var OAuthClient = class {
           responseData += chunk;
         });
         res.on("end", () => {
+          if (res.statusCode && (res.statusCode < 200 || res.statusCode >= 300)) {
+            reject(new Error(`HTTP ${res.statusCode}: ${responseData.slice(0, 200)}`));
+            return;
+          }
           try {
             resolve(JSON.parse(responseData));
           } catch {
@@ -194,6 +415,10 @@ var OAuthClient = class {
           responseData += chunk;
         });
         res.on("end", () => {
+          if (res.statusCode && (res.statusCode < 200 || res.statusCode >= 300)) {
+            reject(new Error(`HTTP ${res.statusCode}: ${responseData.slice(0, 200)}`));
+            return;
+          }
           try {
             resolve(JSON.parse(responseData));
           } catch {
@@ -212,169 +437,6 @@ var OAuthClient = class {
   }
 };
 
-// src/sdk/accounts/storage.ts
-import { readFileSync as readFileSync2, writeFileSync as writeFileSync2, existsSync as existsSync2 } from "fs";
-import { join as join2 } from "path";
-
-// src/core/config.ts
-import { readFileSync, writeFileSync, mkdirSync, existsSync } from "fs";
-import { join } from "path";
-import { homedir } from "os";
-
-// src/core/logger.ts
-var _verbose = false;
-function setVerbose(verbose) {
-  _verbose = verbose;
-}
-function debug(module, msg, data) {
-  if (!_verbose) return;
-  const prefix = `[${module}]`;
-  if (data !== void 0) {
-    console.error(`${prefix} ${msg}`, data);
-  } else {
-    console.error(`${prefix} ${msg}`);
-  }
-}
-
-// src/core/config.ts
-var CONFIG_DIR_NAME = "when-does-my-quota-refresh";
-var DEFAULT_CONFIG = {
-  defaultMode: "local",
-  cacheTtlMs: 5 * 60 * 1e3,
-  // 5 minutes
-  refreshIntervalMs: 5 * 60 * 1e3,
-  wakeupModels: ["claude-sonnet-4-5", "gemini-3-flash", "gemini-3-pro-low"],
-  wakeupAccounts: [],
-  notifications: true,
-  plugins: [],
-  daemon: {
-    pollIntervalMs: 60 * 1e3,
-    enabled: false
-  }
-};
-function getConfigDir() {
-  const home = homedir();
-  switch (process.platform) {
-    case "darwin":
-      return join(home, "Library", "Application Support", CONFIG_DIR_NAME);
-    case "win32":
-      return join(process.env.APPDATA || join(home, "AppData", "Roaming"), CONFIG_DIR_NAME);
-    default:
-      return join(home, ".config", CONFIG_DIR_NAME);
-  }
-}
-function getConfigPath() {
-  return join(getConfigDir(), "config.json");
-}
-function getDataDir() {
-  return join(getConfigDir(), "data");
-}
-function ensureDirs() {
-  const configDir = getConfigDir();
-  const dataDir = getDataDir();
-  if (!existsSync(configDir)) mkdirSync(configDir, { recursive: true });
-  if (!existsSync(dataDir)) mkdirSync(dataDir, { recursive: true });
-}
-function loadConfig() {
-  try {
-    const path = getConfigPath();
-    if (!existsSync(path)) return DEFAULT_CONFIG;
-    const raw = readFileSync(path, "utf-8");
-    const parsed = JSON.parse(raw);
-    debug("config", `Loaded config from ${path}`);
-    return { ...DEFAULT_CONFIG, ...parsed };
-  } catch (err) {
-    debug("config", "Failed to load config, using defaults", err);
-    return DEFAULT_CONFIG;
-  }
-}
-function saveConfig(config) {
-  ensureDirs();
-  const path = getConfigPath();
-  writeFileSync(path, JSON.stringify(config, null, 2));
-  debug("config", `Saved config to ${path}`);
-}
-function getConfigDirPath() {
-  return getConfigDir();
-}
-function getDataDirPath() {
-  return getDataDir();
-}
-
-// src/sdk/accounts/storage.ts
-var TOKENS_DIR = "accounts";
-function getStorePath() {
-  return join2(getConfigDirPath(), TOKENS_DIR, "tokens.json");
-}
-function loadStore() {
-  const path = getStorePath();
-  if (!existsSync2(path)) return { accounts: [] };
-  try {
-    return JSON.parse(readFileSync2(path, "utf-8"));
-  } catch {
-    return { accounts: [] };
-  }
-}
-function saveStore(store) {
-  ensureDirs();
-  writeFileSync2(getStorePath(), JSON.stringify(store, null, 2));
-  debug("account-storage", `Saved ${store.accounts.length} account(s)`);
-}
-function saveAccountTokens(email, tokens) {
-  const store = loadStore();
-  const existing = store.accounts.findIndex((a) => a.email === email);
-  const account = {
-    email,
-    tokens,
-    addedAt: (/* @__PURE__ */ new Date()).toISOString()
-  };
-  if (existing >= 0) {
-    store.accounts[existing] = account;
-  } else {
-    store.accounts.push(account);
-  }
-  if (store.accounts.length === 1) {
-    store.activeEmail = email;
-  }
-  saveStore(store);
-}
-function getActiveAccountTokens() {
-  const store = loadStore();
-  if (!store.activeEmail) return null;
-  const account = store.accounts.find((a) => a.email === store.activeEmail);
-  return account?.tokens || null;
-}
-function setActiveAccount(email) {
-  const store = loadStore();
-  if (!store.accounts.find((a) => a.email === email)) return false;
-  store.activeEmail = email;
-  saveStore(store);
-  return true;
-}
-function removeAccount(email) {
-  const store = loadStore();
-  const idx = store.accounts.findIndex((a) => a.email === email);
-  if (idx < 0) return false;
-  store.accounts.splice(idx, 1);
-  if (store.activeEmail === email) {
-    store.activeEmail = store.accounts[0]?.email;
-  }
-  saveStore(store);
-  return true;
-}
-function listAccounts() {
-  const store = loadStore();
-  return store.accounts.map((a) => ({
-    email: a.email,
-    isActive: a.email === store.activeEmail,
-    addedAt: a.addedAt
-  }));
-}
-function getAllAccountTokens() {
-  const store = loadStore();
-  return store.accounts.map((a) => a.tokens);
-}
-
 // src/core/errors.ts
 var SourceUnavailableError = class extends Error {
   constructor(source, reason) {
@@ -395,14 +457,14 @@ import http from "http";
 import { URL as URL2 } from "url";
 async function loginAccount() {
   const client = new OAuthClient();
-  const authUrl = client.getAuthUrl();
   debug("auth", "Starting OAuth login flow");
+  let capturedRedirectUri = "http://127.0.0.1:0/callback";
   const code = await new Promise((resolve, reject) => {
     const server = http.createServer(async (req, res) => {
       const url = new URL2(req.url || "/", `http://127.0.0.1`);
       const code2 = url.searchParams.get("code");
       if (code2) {
-        res.writeHead(200, { "Content-Type": "text/html" });
+        res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
         res.end(`
           <html><body style="font-family: sans-serif; text-align: center; padding: 50px;">
             <h1>\u2705 Login successful!</h1>
@@ -414,7 +476,7 @@ async function loginAccount() {
       } else {
         const error = url.searchParams.get("error");
         if (error) {
-          res.writeHead(400, { "Content-Type": "text/html" });
+          res.writeHead(400, { "Content-Type": "text/html; charset=utf-8" });
           res.end(`<h1>Login failed: ${error}</h1>`);
           server.close();
           reject(new AuthError(`OAuth error: ${error}`));
@@ -427,7 +489,9 @@ async function loginAccount() {
     server.listen(0, "127.0.0.1", () => {
       const addr = server.address();
       if (typeof addr === "object" && addr) {
+        capturedRedirectUri = `http://127.0.0.1:${addr.port}/callback`;
         debug("auth", `Callback server on port ${addr.port}`);
+        const authUrl = client.getAuthUrl(capturedRedirectUri);
         open(authUrl).catch(() => {
           console.log(`
 Open this URL in your browser:
@@ -442,8 +506,19 @@ ${authUrl}
       reject(new AuthError("Login timed out after 5 minutes"));
     }, 5 * 60 * 1e3);
   });
-  const tokens = await client.exchangeCode(code);
-  const email = tokens.email || "unknown@gmail.com";
+  const tokens = await client.exchangeCode(code, capturedRedirectUri);
+  let email = tokens.email;
+  if (!email) {
+    try {
+      const userInfo = await client.fetchUserInfo();
+      email = userInfo.email;
+    } catch (err) {
+      debug("auth", `Failed to fetch user info: ${err}`);
+    }
+  }
+  if (!email) {
+    throw new AuthError("Could not determine user email from Google account");
+  }
   tokens.email = email;
   saveAccountTokens(email, tokens);
   setActiveAccount(email);
@@ -457,7 +532,18 @@ function getManualLoginUrl() {
 async function completeManualLogin(code) {
   const client = new OAuthClient();
   const tokens = await client.exchangeCode(code);
-  const email = tokens.email || "unknown@gmail.com";
+  let email = tokens.email;
+  if (!email) {
+    try {
+      const userInfo = await client.fetchUserInfo();
+      email = userInfo.email;
+    } catch (err) {
+      debug("auth", `Failed to fetch user info: ${err}`);
+    }
+  }
+  if (!email) {
+    throw new AuthError("Could not determine user email from Google account");
+  }
   tokens.email = email;
   saveAccountTokens(email, tokens);
   setActiveAccount(email);
@@ -468,19 +554,21 @@ export {
   setVerbose,
   debug,
   SourceUnavailableError,
-  OAuthClient,
   ensureDirs,
   loadConfig,
   saveConfig,
   getConfigDirPath,
   getDataDirPath,
+  saveAccountTokens,
+  getAccountTokens,
   getActiveAccountTokens,
   setActiveAccount,
   removeAccount,
   listAccounts,
   getAllAccountTokens,
+  OAuthClient,
   loginAccount,
   getManualLoginUrl,
   completeManualLogin
 };
-//# sourceMappingURL=chunk-24ILGZJ2.js.map
+//# sourceMappingURL=chunk-VLP22VAR.js.map