@juspay/neurolink 10.11.3 → 10.12.1

package/dist/cli/commands/auth.d.ts

@@ -17,7 +17,7 @@
  * Currently supports:
  * - Anthropic (API key + OAuth)
  */
-import type { AuthCommandArgs } from "../../lib/types/index.js";
+import type { AccountQuota, AuthCommandArgs } from "../../lib/types/index.js";
 /**
  * Handle the login subcommand
  * `neurolink auth login <provider>`
@@ -26,6 +26,13 @@ import type { AuthCommandArgs } from "../../lib/types/index.js";
  * (e.g., "anthropic:alice") to support multi-account pools.
  */
 export declare function handleLogin(argv: AuthCommandArgs): Promise<void>;
+/**
+ * Format the dynamic per-plan limit windows (model-scoped weeklies such as
+ * Fable, plus any future kinds) as extra display lines. `session` and
+ * `weekly_all` are omitted — they already render as the SESSION / WEEKLY
+ * columns. Exported for the continuous test suite.
+ */
+export declare function formatQuotaWindowRows(quota: AccountQuota): string[];
 /**
  * Handle the list subcommand
  * `neurolink auth list`

package/dist/cli/commands/auth.js

@@ -27,7 +27,8 @@ import ora from "ora";
 import { logger } from "../../lib/utils/logger.js";
 import { defaultTokenStore } from "../../lib/auth/tokenStore.js";
 import { CLAUDE_CODE_CLIENT_ID, ANTHROPIC_AUTH_URL, ANTHROPIC_TOKEN_URL, ANTHROPIC_REDIRECT_URI, CLAUDE_CLI_USER_AGENT, OAUTH_BETA_HEADERS, } from "../../lib/auth/anthropicOAuth.js";
-import { loadAccountQuotas } from "../../lib/proxy/accountQuota.js";
+import { flushAccountQuotas, loadAccountQuotas, saveAccountQuota, } from "../../lib/proxy/accountQuota.js";
+import { fetchAccountUsage, listAnthropicAccountsForUsage, usageToQuota, } from "../../lib/proxy/accountUsage.js";
 // =============================================================================
 // CONSTANTS
 // =============================================================================
@@ -161,6 +162,137 @@ function formatQuotaColumns(quota) {
             : "",
     };
 }
+/**
+ * Format the dynamic per-plan limit windows (model-scoped weeklies such as
+ * Fable, plus any future kinds) as extra display lines. `session` and
+ * `weekly_all` are omitted — they already render as the SESSION / WEEKLY
+ * columns. Exported for the continuous test suite.
+ */
+export function formatQuotaWindowRows(quota) {
+    if (!quota.windows?.length) {
+        return [];
+    }
+    const colorize = (pct, text) => {
+        if (pct <= 10) {
+            return chalk.red(text);
+        }
+        if (pct <= 30) {
+            return chalk.yellow(text);
+        }
+        return chalk.green(text);
+    };
+    const rows = [];
+    for (const window of quota.windows) {
+        if (window.kind === "session" || window.kind === "weekly_all") {
+            continue;
+        }
+        const remaining = Math.round((1 - window.used) * 100);
+        const label = window.scopeModel
+            ? `${window.group ?? window.kind} (${window.scopeModel})`
+            : window.kind;
+        const reset = window.resetsAt > 0
+            ? chalk.gray(`  resets ${formatTimeUntil(window.resetsAt)}`)
+            : "";
+        rows.push(`${chalk.gray(`${label}:`)} ${colorize(remaining, `${remaining}% left`)}${reset}`);
+    }
+    return rows;
+}
+/**
+ * Fetch fresh limits for `auth list --refresh`.
+ *
+ * Prefers the running proxy's GET /limits endpoint so the proxy's in-memory
+ * routing state is refreshed as a side effect; falls back to fetching the
+ * usage endpoint directly from this process (persisting through the same
+ * quota store) when no proxy is running or the call fails.
+ */
+async function refreshAccountLimitsForList() {
+    const errors = [];
+    const proxyState = detectRunningProxyState();
+    if (proxyState?.port) {
+        const host = proxyState.host && proxyState.host !== "0.0.0.0"
+            ? proxyState.host
+            : "127.0.0.1";
+        try {
+            const response = await fetch(`http://${host}:${proxyState.port}/limits`, {
+                signal: AbortSignal.timeout(45_000),
+            });
+            if (response.ok) {
+                const payload = (await response.json());
+                const quotas = {};
+                for (const result of payload.results) {
+                    if (result.quota) {
+                        quotas[result.account] = result.quota;
+                    }
+                    if (result.status === "error" && result.error) {
+                        errors.push(`${result.account}: ${result.error}`);
+                    }
+                }
+                return { via: "proxy", quotas, errors };
+            }
+            errors.push(`running proxy /limits returned HTTP ${response.status}; fetching directly`);
+        }
+        catch {
+            // Keep the message generic: a raw fetch error can echo the requested
+            // URL, and this string reaches the text and JSON CLI output.
+            errors.push("running proxy /limits unreachable; fetching directly");
+        }
+    }
+    try {
+        const accounts = await listAnthropicAccountsForUsage();
+        const prior = await loadAccountQuotas().catch(() => ({}));
+        const quotas = {};
+        const CONCURRENCY = 3;
+        let nextIndex = 0;
+        const worker = async () => {
+            for (;;) {
+                const index = nextIndex++;
+                if (index >= accounts.length) {
+                    return;
+                }
+                const account = accounts[index];
+                if (account.type !== "oauth") {
+                    continue; // api_key accounts have no subscription windows
+                }
+                // Isolate failures per account: one rejection must not abort the
+                // Promise.all sweep or discard the other accounts' refreshed quotas.
+                try {
+                    const result = await fetchAccountUsage(account);
+                    if (!result.ok) {
+                        errors.push(`${account.label}: ${result.error}`);
+                        continue;
+                    }
+                    const quota = usageToQuota(result.usage, {
+                        now: Date.now(),
+                        prior: prior[account.label] ?? null,
+                    });
+                    if (!quota) {
+                        errors.push(`${account.label}: usage payload had no recognizable limit windows`);
+                        continue;
+                    }
+                    await saveAccountQuota(account.label, quota);
+                    quotas[account.label] = quota;
+                }
+                catch (err) {
+                    errors.push(`${account.label}: ${err instanceof Error ? err.message : String(err)}`);
+                }
+            }
+        };
+        try {
+            await Promise.all(Array.from({ length: Math.min(CONCURRENCY, accounts.length || 1) }, () => worker()));
+        }
+        finally {
+            // The quota store's debounced flush timer is unref()'d and this process
+            // is short-lived — flush now (even on a partial sweep) or the completed
+            // saves never reach disk.
+            await flushAccountQuotas().catch(() => undefined);
+        }
+        return { via: "direct", quotas, errors };
+    }
+    catch (err) {
+        errors.push(`direct limit fetch failed (${err instanceof Error ? err.message : String(err)})`);
+        return { via: "none", quotas: null, errors };
+    }
+}
 /**
  * Handle the list subcommand
  * `neurolink auth list`
@@ -189,6 +321,7 @@ export async function handleList(argv) {
             let email;
             let tokenStatus = "unknown";
             let expiresAt;
+            let tokenType;
             // Derive email from the compound key label when it looks like an email.
             // The credentials file is a shared singleton that gets overwritten on
             // every login — reading email from it would show the LATEST login's
@@ -220,6 +353,7 @@ export async function handleList(argv) {
                 const tokens = await defaultTokenStore.loadTokens(key);
                 if (tokens) {
                     expiresAt = tokens.expiresAt;
+                    tokenType = tokens.tokenType;
                     const isExpired = defaultTokenStore.isTokenExpired(tokens, 0);
                     tokenStatus = isExpired ? "expired" : "valid";
                     // Extract per-account metadata from scope (e.g. "tier:pro email:user@example.com")
@@ -242,9 +376,24 @@ export async function handleList(argv) {
             catch {
                 // Token load failed — show as unknown
             }
-            return { key, provider, label, email, tier, tokenStatus, expiresAt };
+            return {
+                key,
+                provider,
+                label,
+                email,
+                tier,
+                tokenStatus,
+                expiresAt,
+                tokenType,
+            };
         }));
-        // Load persisted quota data (captured from proxy responses).
+        // Optionally fetch FRESH limits from Anthropic before rendering.
+        let refreshOutcome;
+        if (argv.refresh) {
+            refreshOutcome = await refreshAccountLimitsForList();
+        }
+        // Load persisted quota data (captured from proxy responses), then overlay
+        // anything just refreshed — freshly fetched values win over the snapshot.
         let quotas = {};
         try {
             quotas = await loadAccountQuotas();
@@ -252,6 +401,9 @@ export async function handleList(argv) {
         catch {
             // Non-fatal — quota display is best-effort
         }
+        if (refreshOutcome?.quotas) {
+            quotas = { ...quotas, ...refreshOutcome.quotas };
+        }
         if (argv.format === "json") {
             // Merge quota data into each account object for JSON output
             const withQuota = enrichedAccounts.map((acct) => {
@@ -259,9 +411,29 @@ export async function handleList(argv) {
                 const quota = quotas[quotaKey] ?? null;
                 return { ...acct, quota };
             });
-            logger.always(JSON.stringify(withQuota, null, 2));
+            if (refreshOutcome) {
+                // --refresh envelopes the array so the fetch outcome travels with it.
+                logger.always(JSON.stringify({
+                    refresh: {
+                        via: refreshOutcome.via,
+                        errors: refreshOutcome.errors,
+                    },
+                    accounts: withQuota,
+                }, null, 2));
+            }
+            else {
+                logger.always(JSON.stringify(withQuota, null, 2));
+            }
         }
         else {
+            if (refreshOutcome) {
+                if (refreshOutcome.via !== "none") {
+                    logger.always(chalk.gray(`\nFetched fresh limits from Anthropic (${refreshOutcome.via === "proxy" ? "via running proxy" : "direct"}).`));
+                }
+                for (const refreshError of refreshOutcome.errors) {
+                    logger.always(chalk.yellow(`⚠ ${refreshError}`));
+                }
+            }
             logger.always(chalk.bold("\nAuthenticated Accounts:\n"));
             // Check if any account has quota data to decide column layout
             const hasQuota = enrichedAccounts.some((acct) => {
@@ -296,14 +468,21 @@ export async function handleList(argv) {
                 if (hasQuota && quota) {
                     const qc = formatQuotaColumns(quota);
                     logger.always(`  ${chalk.cyan(displayLabel)} ${displayProvider} ${displayEmail} ${statusText} ${qc.sessionText.padEnd(10)} ${qc.weeklyText.padEnd(10)}`);
+                    const indent = " ".repeat(2 + 20 + 1 + 12 + 1 + 28 + 1 + 14 + 1);
                     // Second line: reset times (indented under session/weekly columns)
                     if (qc.sessionReset || qc.weeklyReset) {
-                        const indent = " ".repeat(2 + 20 + 1 + 12 + 1 + 28 + 1 + 14 + 1);
                         logger.always(`${indent}${(qc.sessionReset || "").padEnd(10)} ${qc.weeklyReset || ""}`);
                     }
+                    // Dynamic per-plan windows (e.g. the Fable-only weekly limit)
+                    for (const windowRow of formatQuotaWindowRows(quota)) {
+                        logger.always(`${indent}${windowRow}`);
+                    }
                 }
                 else {
-                    logger.always(`  ${chalk.cyan(displayLabel)} ${displayProvider} ${displayEmail} ${statusText}${hasQuota ? "  -          -" : ""}`);
+                    const apiKeyNote = refreshOutcome && acct.tokenType && acct.tokenType !== "Bearer"
+                        ? chalk.gray("  (api key — not refreshed)")
+                        : "";
+                    logger.always(`  ${chalk.cyan(displayLabel)} ${displayProvider} ${displayEmail} ${statusText}${hasQuota ? "  -          -" : ""}${apiKeyNote}`);
                 }
             }
             logger.always("");

package/dist/cli/factories/authCommandFactory.js

@@ -214,8 +214,14 @@ export class AuthCommandFactory {
      */
     static buildListOptions(yargs) {
         return yargs
+            .option("refresh", {
+            type: "boolean",
+            default: false,
+            description: "Fetch fresh limits from Anthropic for all OAuth accounts before listing (via the running proxy when available)",
+        })
             .example("$0 auth list", "List all authenticated accounts")
-            .example("$0 auth list --format json", "List accounts in JSON format");
+            .example("$0 auth list --format json", "List accounts in JSON format")
+            .example("$0 auth list --refresh", "Fetch fresh session/weekly/model-scoped limits before listing");
     }
     /**
      * Build options for remove subcommand

package/dist/lib/processors/archive/ArchiveProcessor.js

@@ -162,6 +162,17 @@ const SINGLE_STREAM_TOOLS = {
     xz: "xz",
     zst: "zstd",
 };
+/**
+ * Whether a zlib rejection is the output bound firing rather than bad input.
+ *
+ * `maxOutputLength` aborts an inflate the moment its output would pass the cap,
+ * which is the whole point — but it surfaces as a plain `RangeError`, and a
+ * bomb reported as "failed to decompress" reads as a corrupt upload and invites
+ * the user to send it again. It will fail identically every time.
+ *
+ * Keyed on `code`, not the message: the message embeds a byte count.
+ */
+const isDecompressionBoundExceeded = (error) => error?.code === "ERR_BUFFER_TOO_LARGE";
 /** File extensions recognized as archive formats */
 const SUPPORTED_ARCHIVE_EXTENSIONS = [".zip", ".tar", ".gz", ".tgz", ".bz2", ".tbz2", ".jar", ".xz", ".txz", ".zst", ".tzst"];
 // =============================================================================
@@ -761,19 +772,16 @@ export class ArchiveProcessor extends BaseFileProcessor {
             const zlib = await import("zlib");
             const { promisify } = await import("util");
             const gunzip = promisify(zlib.gunzip);
-            const decompressed = await gunzip(buffer);
+            // Bounded at the decoder, matching the zstd path. Checking the length
+            // afterwards only reports a bomb once it has already been paid for: 40KB
+            // of gzip inflates to 40MB, and the allocation is the damage, not the
+            // number. `maxOutputLength` abandons the inflate at the cap instead, so
+            // the ceiling on memory is the limit rather than whatever the attacker
+            // chose. The overflow is classified in the catch below.
+            const decompressed = await gunzip(buffer, {
+                maxOutputLength: ARCHIVE_SECURITY.MAX_DECOMPRESSED_SIZE,
+            });
             const tarBuffer = Buffer.from(decompressed);
-            // Security: check decompressed size
-            if (tarBuffer.length > ARCHIVE_SECURITY.MAX_DECOMPRESSED_SIZE) {
-                return {
-                    success: false,
-                    entries: [],
-                    securityWarnings: [],
-                    error: this.createError(FileErrorCode.SECURITY_VALIDATION_FAILED, {
-                        reason: `Decompressed TAR size (${this.formatSizeMB(tarBuffer.length)} MB) exceeds limit (${this.formatSizeMB(ARCHIVE_SECURITY.MAX_DECOMPRESSED_SIZE)} MB)`,
-                    }),
-                };
-            }
             // Security: check compression ratio
             if (buffer.length > 0) {
                 const ratio = tarBuffer.length / buffer.length;
@@ -793,6 +801,16 @@ export class ArchiveProcessor extends BaseFileProcessor {
             return await this.parseTarStream(tarStream, tarBuffer);
         }
         catch (error) {
+            if (isDecompressionBoundExceeded(error)) {
+                return {
+                    success: false,
+                    entries: [],
+                    securityWarnings: [],
+                    error: this.createError(FileErrorCode.SECURITY_VALIDATION_FAILED, {
+                        reason: `Decompressed TAR size exceeds limit (${this.formatSizeMB(ARCHIVE_SECURITY.MAX_DECOMPRESSED_SIZE)} MB)`,
+                    }),
+                };
+            }
             // Check if the error is one we already created (security validation)
             if (error &&
                 typeof error === "object" &&
@@ -976,18 +994,11 @@ export class ArchiveProcessor extends BaseFileProcessor {
             const zlib = await import("zlib");
             const { promisify } = await import("util");
             const gunzip = promisify(zlib.gunzip);
-            const decompressed = await gunzip(buffer);
-            // Security: check decompressed size
-            if (decompressed.length > ARCHIVE_SECURITY.MAX_DECOMPRESSED_SIZE) {
-                return {
-                    success: false,
-                    entries: [],
-                    securityWarnings: [],
-                    error: this.createError(FileErrorCode.SECURITY_VALIDATION_FAILED, {
-                        reason: `Decompressed size (${this.formatSizeMB(decompressed.length)} MB) exceeds limit (${this.formatSizeMB(ARCHIVE_SECURITY.MAX_DECOMPRESSED_SIZE)} MB)`,
-                    }),
-                };
-            }
+            // Bounded at the decoder — see the matching call in extractTarGzEntries.
+            // The overflow is classified in the catch below.
+            const decompressed = await gunzip(buffer, {
+                maxOutputLength: ARCHIVE_SECURITY.MAX_DECOMPRESSED_SIZE,
+            });
             // Security: compression ratio
             if (buffer.length > 0) {
                 const ratio = decompressed.length / buffer.length;
@@ -1031,6 +1042,16 @@ export class ArchiveProcessor extends BaseFileProcessor {
             return { success: true, entries, securityWarnings, contents };
         }
         catch (error) {
+            if (isDecompressionBoundExceeded(error)) {
+                return {
+                    success: false,
+                    entries: [],
+                    securityWarnings: [],
+                    error: this.createError(FileErrorCode.SECURITY_VALIDATION_FAILED, {
+                        reason: `Decompressed size exceeds limit (${this.formatSizeMB(ARCHIVE_SECURITY.MAX_DECOMPRESSED_SIZE)} MB)`,
+                    }),
+                };
+            }
             return {
                 success: false,
                 entries: [],

package/dist/lib/proxy/accountQuota.d.ts

@@ -48,4 +48,10 @@ export declare function loadAccountQuota(accountKey: string): Promise<AccountQuo
  * other accounts' snapshots and blinded quota-aware routing to them).
  */
 export declare function saveAccountQuota(accountKey: string, quota: AccountQuota): Promise<void>;
+/**
+ * Cancel any pending debounced write and flush the cache to disk now.
+ * Short-lived processes (CLI refresh path) must call this before exit —
+ * the debounce timer is unref()'d and will not keep the process alive.
+ */
+export declare function flushAccountQuotas(): Promise<void>;
 export declare function flushAccountQuotaStateForTests(): Promise<void>;

package/dist/lib/proxy/accountQuota.js

@@ -94,6 +94,7 @@ export function parseQuotaHeaders(headers) {
         upgradePaths: getHeader(headers, `${P}unified-upgrade-paths`),
         overageStatus: getHeader(headers, `${P}unified-overage-status`) ?? "unknown",
         lastUpdated: Date.now(),
+        source: "headers",
     };
 }
 // ---------------------------------------------------------------------------
@@ -229,17 +230,33 @@ export async function loadAccountQuota(accountKey) {
 export async function saveAccountQuota(accountKey, quota) {
     await stateMutex.runExclusive(async () => {
         await ensureAccountQuotasLoaded();
-        memoryCache[accountKey] = { ...quota };
+        const next = { ...quota };
+        // Header-sourced saves carry no dynamic windows; a passive capture right
+        // after a usage-API refresh must not erase the refreshed buckets.
+        const existing = memoryCache[accountKey];
+        if (next.windows === undefined && existing?.windows !== undefined) {
+            next.windows = existing.windows;
+            next.windowsUpdatedAt = existing.windowsUpdatedAt;
+        }
+        memoryCache[accountKey] = next;
         dirty = true;
         cacheVersion += 1;
     });
     scheduleFlush();
 }
-export async function flushAccountQuotaStateForTests() {
+/**
+ * Cancel any pending debounced write and flush the cache to disk now.
+ * Short-lived processes (CLI refresh path) must call this before exit —
+ * the debounce timer is unref()'d and will not keep the process alive.
+ */
+export async function flushAccountQuotas() {
     if (flushTimer) {
         clearTimeout(flushTimer);
         flushTimer = null;
     }
     await flushToDisk();
 }
+export async function flushAccountQuotaStateForTests() {
+    await flushAccountQuotas();
+}
 //# sourceMappingURL=accountQuota.js.map

package/dist/lib/proxy/accountUsage.d.ts

@@ -0,0 +1,45 @@
+/**
+ * On-Demand Account Usage Fetch
+ *
+ * Manual refetch path for account limits: queries Anthropic's OAuth usage
+ * endpoint (the same call Claude Code's /usage makes) to get FRESH session /
+ * weekly / model-scoped windows for an account without consuming any tokens
+ * and without starting a 5h session window.
+ *
+ * This complements — never replaces — the passive header capture in
+ * accountQuota.ts: `usageToQuota` normalizes the endpoint payload into the
+ * same `AccountQuota` shape so refreshed data flows through the existing
+ * save/merge/cooldown chain.
+ *
+ * Only OAuth (Bearer) accounts have subscription windows; api_key accounts
+ * are skipped and keep their header-derived absolute limits.
+ */
+import type { AccountAllowlist, AccountQuota, AccountUsageFetchResult, AnthropicUsageResponse, ProxyPassthroughAccount } from "../types/index.js";
+export declare const ANTHROPIC_USAGE_URL = "https://api.anthropic.com/api/oauth/usage";
+/**
+ * Enumerate stored Anthropic accounts for a usage refresh.
+ * Read-only: no token refresh, no disable side effects — those decisions
+ * belong to the caller (fetchAccountUsage / the routing path respectively).
+ */
+export declare function listAnthropicAccountsForUsage(allowlist?: AccountAllowlist): Promise<ProxyPassthroughAccount[]>;
+/**
+ * Fetch fresh usage windows for one account. Return-not-throw.
+ * Refreshes an expiring/expired token first (de-duped, rotation-safe), and
+ * retries once through a forced refresh on a 401.
+ */
+export declare function fetchAccountUsage(account: ProxyPassthroughAccount): Promise<AccountUsageFetchResult>;
+/**
+ * Normalize a usage-endpoint payload into the `AccountQuota` shape used by
+ * the passive header path, so a refresh writes through the exact same
+ * save/merge/cooldown chain.
+ *
+ * `prior` supplies fields the usage payload cannot express (fallback and
+ * upgrade-path signals) so `isQuotaOverageAvailable` behaves identically
+ * before and after a refresh. `unifiedStatus` is deliberately never set:
+ * fabricating it would let the reconcile path treat a refresh like a
+ * unified-rejected 429.
+ */
+export declare function usageToQuota(usage: AnthropicUsageResponse, opts: {
+    now: number;
+    prior?: AccountQuota | null;
+}): AccountQuota | null;