oc-codex-multi-auth 6.0.0 → 6.1.1

package/dist/lib/accounts/rotation.js

@@ -0,0 +1,204 @@
+/**
+ * Rotation + rate-limit + cooldown surface for {@link AccountManager}.
+ *
+ * Pure logic module: all state mutations land on a shared {@link AccountState}
+ * reference. Integrates with the health and token-bucket trackers in
+ * `lib/rotation.ts` (note: that file is the hybrid selection algorithm; this
+ * module is the manager-facing wrapper that wires it to `AccountState`).
+ */
+import { getHealthTracker, getTokenTracker, selectHybridAccount, } from "../rotation.js";
+import { nowMs } from "../utils.js";
+import { clearExpiredRateLimits, getQuotaKey, isRateLimitedForFamily, } from "./rate-limits.js";
+export class AccountRotation {
+    state;
+    constructor(state) {
+        this.state = state;
+    }
+    getCurrentOrNextForFamily(family, model) {
+        const count = this.state.accounts.length;
+        if (count === 0)
+            return null;
+        const cursor = this.state.cursorByFamily[family];
+        for (let i = 0; i < count; i++) {
+            const idx = (cursor + i) % count;
+            const account = this.state.accounts[idx];
+            if (!account)
+                continue;
+            if (account.enabled === false)
+                continue;
+            clearExpiredRateLimits(account);
+            if (isRateLimitedForFamily(account, family, model) ||
+                this.state.isAccountCoolingDown(account)) {
+                continue;
+            }
+            this.state.cursorByFamily[family] = (idx + 1) % count;
+            this.state.currentAccountIndexByFamily[family] = idx;
+            account.lastUsed = nowMs();
+            return account;
+        }
+        return null;
+    }
+    getNextForFamily(family, model) {
+        const count = this.state.accounts.length;
+        if (count === 0)
+            return null;
+        const cursor = this.state.cursorByFamily[family];
+        for (let i = 0; i < count; i++) {
+            const idx = (cursor + i) % count;
+            const account = this.state.accounts[idx];
+            if (!account)
+                continue;
+            if (account.enabled === false)
+                continue;
+            clearExpiredRateLimits(account);
+            if (isRateLimitedForFamily(account, family, model) ||
+                this.state.isAccountCoolingDown(account)) {
+                continue;
+            }
+            this.state.cursorByFamily[family] = (idx + 1) % count;
+            account.lastUsed = nowMs();
+            return account;
+        }
+        return null;
+    }
+    getCurrentOrNextForFamilyHybrid(family, model, options) {
+        const count = this.state.accounts.length;
+        if (count === 0)
+            return null;
+        const currentIndex = this.state.currentAccountIndexByFamily[family];
+        if (currentIndex >= 0 && currentIndex < count) {
+            const currentAccount = this.state.accounts[currentIndex];
+            if (currentAccount) {
+                if (currentAccount.enabled === false) {
+                    // Fall through to hybrid selection.
+                }
+                else {
+                    clearExpiredRateLimits(currentAccount);
+                    if (!isRateLimitedForFamily(currentAccount, family, model) &&
+                        !this.state.isAccountCoolingDown(currentAccount)) {
+                        currentAccount.lastUsed = nowMs();
+                        return currentAccount;
+                    }
+                }
+            }
+        }
+        const quotaKey = model ? `${family}:${model}` : family;
+        const healthTracker = getHealthTracker();
+        const tokenTracker = getTokenTracker();
+        const accountsWithMetrics = this.state.accounts
+            .map((account) => {
+            if (!account)
+                return null;
+            if (account.enabled === false)
+                return null;
+            clearExpiredRateLimits(account);
+            const isAvailable = !isRateLimitedForFamily(account, family, model) &&
+                !this.state.isAccountCoolingDown(account);
+            return {
+                index: account.index,
+                isAvailable,
+                lastUsed: account.lastUsed,
+            };
+        })
+            .filter((a) => a !== null);
+        const selected = selectHybridAccount(accountsWithMetrics, healthTracker, tokenTracker, quotaKey, {}, options);
+        if (!selected)
+            return null;
+        const account = this.state.accounts[selected.index];
+        if (!account)
+            return null;
+        this.state.currentAccountIndexByFamily[family] = account.index;
+        this.state.cursorByFamily[family] = (account.index + 1) % count;
+        account.lastUsed = nowMs();
+        return account;
+    }
+    recordSuccess(account, family, model) {
+        const quotaKey = model ? `${family}:${model}` : family;
+        getHealthTracker().recordSuccess(account.index, quotaKey);
+    }
+    recordRateLimit(account, family, model) {
+        const quotaKey = model ? `${family}:${model}` : family;
+        getHealthTracker().recordRateLimit(account.index, quotaKey);
+        getTokenTracker().drain(account.index, quotaKey);
+    }
+    recordFailure(account, family, model) {
+        const quotaKey = model ? `${family}:${model}` : family;
+        getHealthTracker().recordFailure(account.index, quotaKey);
+    }
+    consumeToken(account, family, model) {
+        const quotaKey = model ? `${family}:${model}` : family;
+        return getTokenTracker().tryConsume(account.index, quotaKey);
+    }
+    /**
+     * Refund a token consumed within the refund window (30 seconds).
+     * Use this when a request fails due to network errors (not rate limits).
+     * @returns true if refund was successful, false if no valid consumption found
+     */
+    refundToken(account, family, model) {
+        const quotaKey = model ? `${family}:${model}` : family;
+        return getTokenTracker().refundToken(account.index, quotaKey);
+    }
+    markRateLimited(account, retryAfterMs, family, model) {
+        this.markRateLimitedWithReason(account, retryAfterMs, family, "unknown", model);
+    }
+    markRateLimitedWithReason(account, retryAfterMs, family, reason, model) {
+        const retryMs = Math.max(0, Math.floor(retryAfterMs));
+        const resetAt = nowMs() + retryMs;
+        const baseKey = getQuotaKey(family);
+        account.rateLimitResetTimes[baseKey] = resetAt;
+        if (model) {
+            const modelKey = getQuotaKey(family, model);
+            account.rateLimitResetTimes[modelKey] = resetAt;
+        }
+        account.lastRateLimitReason = reason;
+    }
+    markAccountCoolingDown(account, cooldownMs, reason) {
+        const ms = Math.max(0, Math.floor(cooldownMs));
+        account.coolingDownUntil = nowMs() + ms;
+        account.cooldownReason = reason;
+    }
+    /**
+     * Mark every in-memory account sharing a refresh token as cooling down.
+     * @returns Number of live accounts updated.
+     */
+    markAccountsWithRefreshTokenCoolingDown(refreshToken, cooldownMs, reason) {
+        const matches = this.state.accounts.filter((account) => account.refreshToken === refreshToken);
+        for (const account of matches) {
+            this.markAccountCoolingDown(account, cooldownMs, reason);
+        }
+        return matches.length;
+    }
+    getMinWaitTimeForFamily(family, model) {
+        const now = nowMs();
+        const enabledAccounts = this.state.accounts.filter((account) => account.enabled !== false);
+        const available = enabledAccounts.filter((account) => {
+            clearExpiredRateLimits(account);
+            return (!isRateLimitedForFamily(account, family, model) &&
+                !this.state.isAccountCoolingDown(account));
+        });
+        if (available.length > 0)
+            return 0;
+        if (enabledAccounts.length === 0)
+            return 0;
+        const waitTimes = [];
+        const baseKey = getQuotaKey(family);
+        const modelKey = model ? getQuotaKey(family, model) : null;
+        for (const account of enabledAccounts) {
+            const baseResetAt = account.rateLimitResetTimes[baseKey];
+            if (typeof baseResetAt === "number") {
+                waitTimes.push(Math.max(0, baseResetAt - now));
+            }
+            if (modelKey) {
+                const modelResetAt = account.rateLimitResetTimes[modelKey];
+                if (typeof modelResetAt === "number") {
+                    waitTimes.push(Math.max(0, modelResetAt - now));
+                }
+            }
+            if (typeof account.coolingDownUntil === "number") {
+                waitTimes.push(Math.max(0, account.coolingDownUntil - now));
+            }
+        }
+        return waitTimes.length > 0 ? Math.min(...waitTimes) : 0;
+    }
+}
+//# sourceMappingURL=rotation.js.map

package/dist/lib/accounts/rotation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rotation.js","sourceRoot":"","sources":["../../../lib/accounts/rotation.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,EACN,gBAAgB,EAChB,eAAe,EACf,mBAAmB,GAGnB,MAAM,gBAAgB,CAAC;AAExB,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACpC,OAAO,EACN,sBAAsB,EACtB,WAAW,EACX,sBAAsB,GAEtB,MAAM,kBAAkB,CAAC;AAG1B,MAAM,OAAO,eAAe;IACE;IAA7B,YAA6B,KAAmB;QAAnB,UAAK,GAAL,KAAK,CAAc;IAAG,CAAC;IAEpD,yBAAyB,CACxB,MAAmB,EACnB,KAAqB;QAErB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;QACzC,IAAI,KAAK,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAE7B,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAEjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YACzC,IAAI,CAAC,OAAO;gBAAE,SAAS;YACvB,IAAI,OAAO,CAAC,OAAO,KAAK,KAAK;gBAAE,SAAS;YAExC,sBAAsB,CAAC,OAAO,CAAC,CAAC;YAChC,IACC,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC;gBAC9C,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC,EACvC,CAAC;gBACF,SAAS;YACV,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;YACtD,IAAI,CAAC,KAAK,CAAC,2BAA2B,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;YACrD,OAAO,CAAC,QAAQ,GAAG,KAAK,EAAE,CAAC;YAC3B,OAAO,OAAO,CAAC;QAChB,CAAC;QAED,OAAO,IAAI,CAAC;IACb,CAAC;IAED,gBAAgB,CAAC,MAAmB,EAAE,KAAqB;QAC1D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;QACzC,IAAI,KAAK,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAE7B,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAEjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YACzC,IAAI,CAAC,OAAO;gBAAE,SAAS;YACvB,IAAI,OAAO,CAAC,OAAO,KAAK,KAAK;gBAAE,SAAS;YAExC,sBAAsB,CAAC,OAAO,CAAC,CAAC;YAChC,IACC,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC;gBAC9C,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC,EACvC,CAAC;gBACF,SAAS;YACV,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;YACtD,OAAO,CAAC,QAAQ,GAAG,KAAK,EAAE,CAAC;YAC3B,OAAO,OAAO,CAAC;QAChB,CAAC;QAED,OAAO,IAAI,CAAC;IACb,CAAC;IAED,+BAA+B,CAC9B,MAAmB,EACnB,KAAqB,EACrB,OAAgC;QAEhC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;QACzC,IAAI,KAAK,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAE7B,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,2BAA2B,CAAC,MAAM,CAAC,CAAC;QACpE,IAAI,YAAY,IAAI,CAAC,IAAI,YAAY,GAAG,KAAK,EAAE,CAAC;YAC/C,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YACzD,IAAI,cAAc,EAAE,CAAC;gBACpB,IAAI,cAAc,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;oBACtC,oCAAoC;gBACrC,CAAC;qBAAM,CAAC;oBACP,sBAAsB,CAAC,cAAc,CAAC,CAAC;oBACvC,IACC,CAAC,sBAAsB,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,CAAC;wBACtD,CAAC,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,cAAc,CAAC,EAC/C,CAAC;wBACF,cAAc,CAAC,QAAQ,GAAG,KAAK,EAAE,CAAC;wBAClC,OAAO,cAAc,CAAC;oBACvB,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;QACvD,MAAM,aAAa,GAAG,gBAAgB,EAAE,CAAC;QACzC,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC;QAEvC,MAAM,mBAAmB,GAAyB,IAAI,CAAC,KAAK,CAAC,QAAQ;aACnE,GAAG,CAAC,CAAC,OAAO,EAA6B,EAAE;YAC3C,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC;YAC1B,IAAI,OAAO,CAAC,OAAO,KAAK,KAAK;gBAAE,OAAO,IAAI,CAAC;YAC3C,sBAAsB,CAAC,OAAO,CAAC,CAAC;YAChC,MAAM,WAAW,GAChB,CAAC,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC;gBAC/C,CAAC,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAC3C,OAAO;gBACN,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,WAAW;gBACX,QAAQ,EAAE,OAAO,CAAC,QAAQ;aAC1B,CAAC;QACH,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAA2B,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;QAErD,MAAM,QAAQ,GAAG,mBAAmB,CACnC,mBAAmB,EACnB,aAAa,EACb,YAAY,EACZ,QAAQ,EACR,EAAE,EACF,OAAO,CACP,CAAC;QACF,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAE3B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACpD,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAE1B,IAAI,CAAC,KAAK,CAAC,2BAA2B,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;QAC/D,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;QAChE,OAAO,CAAC,QAAQ,GAAG,KAAK,EAAE,CAAC;QAC3B,OAAO,OAAO,CAAC;IAChB,CAAC;IAED,aAAa,CACZ,OAAuB,EACvB,MAAmB,EACnB,KAAqB;QAErB,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;QACvD,gBAAgB,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC3D,CAAC;IAED,eAAe,CACd,OAAuB,EACvB,MAAmB,EACnB,KAAqB;QAErB,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;QACvD,gBAAgB,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC5D,eAAe,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAClD,CAAC;IAED,aAAa,CACZ,OAAuB,EACvB,MAAmB,EACnB,KAAqB;QAErB,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;QACvD,gBAAgB,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC3D,CAAC;IAED,YAAY,CACX,OAAuB,EACvB,MAAmB,EACnB,KAAqB;QAErB,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;QACvD,OAAO,eAAe,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;IAED;;;;OAIG;IACH,WAAW,CACV,OAAuB,EACvB,MAAmB,EACnB,KAAqB;QAErB,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;QACvD,OAAO,eAAe,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC/D,CAAC;IAED,eAAe,CACd,OAAuB,EACvB,YAAoB,EACpB,MAAmB,EACnB,KAAqB;QAErB,IAAI,CAAC,yBAAyB,CAAC,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;IACjF,CAAC;IAED,yBAAyB,CACxB,OAAuB,EACvB,YAAoB,EACpB,MAAmB,EACnB,MAAuB,EACvB,KAAqB;QAErB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;QAElC,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC;QAE/C,IAAI,KAAK,EAAE,CAAC;YACX,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC5C,OAAO,CAAC,mBAAmB,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC;QACjD,CAAC;QAED,OAAO,CAAC,mBAAmB,GAAG,MAAM,CAAC;IACtC,CAAC;IAED,sBAAsB,CACrB,OAAuB,EACvB,UAAkB,EAClB,MAAsB;QAEtB,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;QAC/C,OAAO,CAAC,gBAAgB,GAAG,KAAK,EAAE,GAAG,EAAE,CAAC;QACxC,OAAO,CAAC,cAAc,GAAG,MAAM,CAAC;IACjC,CAAC;IAED;;;OAGG;IACH,uCAAuC,CACtC,YAAoB,EACpB,UAAkB,EAClB,MAAsB;QAEtB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CACzC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,YAAY,KAAK,YAAY,CAClD,CAAC;QACF,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE,CAAC;YAC/B,IAAI,CAAC,sBAAsB,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC1D,CAAC;QACD,OAAO,OAAO,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,uBAAuB,CAAC,MAAmB,EAAE,KAAqB;QACjE,MAAM,GAAG,GAAG,KAAK,EAAE,CAAC;QACpB,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CACjD,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,KAAK,KAAK,CACtC,CAAC;QACF,MAAM,SAAS,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;YACpD,sBAAsB,CAAC,OAAO,CAAC,CAAC;YAChC,OAAO,CACN,CAAC,sBAAsB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC;gBAC/C,CAAC,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC,CACzC,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,CAAC,CAAC;QACnC,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAE3C,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAE3D,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;YACvC,MAAM,WAAW,GAAG,OAAO,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;YACzD,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,GAAG,GAAG,CAAC,CAAC,CAAC;YAChD,CAAC;YAED,IAAI,QAAQ,EAAE,CAAC;gBACd,MAAM,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;gBAC3D,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;oBACtC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,GAAG,GAAG,CAAC,CAAC,CAAC;gBACjD,CAAC;YACF,CAAC;YAED,IAAI,OAAO,OAAO,CAAC,gBAAgB,KAAK,QAAQ,EAAE,CAAC;gBAClD,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,gBAAgB,GAAG,GAAG,CAAC,CAAC,CAAC;YAC7D,CAAC;QACF,CAAC;QAED,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1D,CAAC;CACD"}

package/dist/lib/accounts/state.d.ts

@@ -0,0 +1,92 @@
+/**
+ * In-memory account registry for {@link AccountManager}.
+ *
+ * Owns the mutable account list, per-family cursors/active indices, and
+ * auxiliary in-memory state (toast debounce, auth-failure counters).
+ * Other account services (persistence, rotation, recovery) receive a reference
+ * to an `AccountState` instance and mutate it through this narrow surface.
+ */
+import type { Auth } from "@opencode-ai/sdk";
+import { type ModelFamily } from "../prompts/codex.js";
+import type { AccountIdSource, OAuthAuthDetails } from "../types.js";
+import { type RateLimitReason } from "./rate-limits.js";
+import type { AccountStorageV3, CooldownReason, RateLimitStateV3 } from "../storage.js";
+export interface ManagedAccount {
+    index: number;
+    accountId?: string;
+    organizationId?: string;
+    accountIdSource?: AccountIdSource;
+    accountLabel?: string;
+    accountTags?: string[];
+    accountNote?: string;
+    email?: string;
+    refreshToken: string;
+    enabled?: boolean;
+    access?: string;
+    expires?: number;
+    addedAt: number;
+    lastUsed: number;
+    lastSwitchReason?: "rate-limit" | "initial" | "rotation";
+    lastRateLimitReason?: RateLimitReason;
+    rateLimitResetTimes: RateLimitStateV3;
+    coolingDownUntil?: number;
+    cooldownReason?: CooldownReason;
+}
+export interface AccountSelectionExplainability {
+    index: number;
+    enabled: boolean;
+    isCurrentForFamily: boolean;
+    eligible: boolean;
+    reasons: string[];
+    healthScore: number;
+    tokensAvailable: number;
+    rateLimitedUntil?: number;
+    coolingDownUntil?: number;
+    cooldownReason?: CooldownReason;
+    lastUsed: number;
+}
+export declare class AccountState {
+    accounts: ManagedAccount[];
+    cursorByFamily: Record<ModelFamily, number>;
+    currentAccountIndexByFamily: Record<ModelFamily, number>;
+    lastToastAccountIndex: number;
+    lastToastTime: number;
+    authFailuresByRefreshToken: Map<string, number>;
+    /**
+     * Per-refresh-token promise chain used to serialize concurrent
+     * `incrementAuthFailures` calls. Prevents lost updates when two org-variant
+     * accounts that share a refresh token observe an auth failure at once — see
+     * audit finding `docs/audits/03-critical-issues.md` (ledger id `47`).
+     */
+    incrementAuthFailuresChain: Map<string, Promise<number>>;
+    initializeFromStorage(authFallback: OAuthAuthDetails | undefined, stored: AccountStorageV3 | null | undefined): void;
+    hasRefreshToken(refreshToken: string): boolean;
+    getAccountCount(): number;
+    getActiveIndexForFamily(family: ModelFamily): number;
+    getAccountsSnapshot(): ManagedAccount[];
+    getSelectionExplainability(family: ModelFamily, model?: string | null, now?: number): AccountSelectionExplainability[];
+    setActiveIndex(index: number): ManagedAccount | null;
+    getCurrentAccountForFamily(family: ModelFamily): ManagedAccount | null;
+    shouldShowAccountToast(accountIndex: number, debounceMs?: number): boolean;
+    markToastShown(accountIndex: number): void;
+    updateFromAuth(account: ManagedAccount, auth: OAuthAuthDetails): void;
+    toAuthDetails(account: ManagedAccount): Auth;
+    markSwitched(account: ManagedAccount, reason: "rate-limit" | "initial" | "rotation", family: ModelFamily): void;
+    removeAccount(account: ManagedAccount): boolean;
+    removeAccountByIndex(index: number): boolean;
+    setAccountEnabled(index: number, enabled: boolean): ManagedAccount | null;
+    /**
+     * Check whether a cooldown window is still active for the given account.
+     * Clears expired cooldowns as a side-effect so that stale `coolingDownUntil`
+     * timestamps do not leak into snapshots or persistence.
+     */
+    isAccountCoolingDown(account: ManagedAccount): boolean;
+    clearAccountCooldown(account: ManagedAccount): void;
+    /**
+     * Shared predicate used by rotation and diagnostic paths: is this account
+     * usable for the given family/model right now? Combines enabled flag,
+     * rate-limit expiry, and cooldown.
+     */
+    isEligibleForFamily(account: ManagedAccount, family: ModelFamily, model?: string | null): boolean;
+}
+//# sourceMappingURL=state.d.ts.map

package/dist/lib/accounts/state.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"state.d.ts","sourceRoot":"","sources":["../../../lib/accounts/state.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAkB,KAAK,WAAW,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,KAAK,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAErE,OAAO,EAKN,KAAK,eAAe,EACpB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,gBAAgB,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AASxF,MAAM,WAAW,cAAc;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,YAAY,GAAG,SAAS,GAAG,UAAU,CAAC;IACzD,mBAAmB,CAAC,EAAE,eAAe,CAAC;IACtC,mBAAmB,EAAE,gBAAgB,CAAC;IACtC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,cAAc,CAAC,EAAE,cAAc,CAAC;CAChC;AAED,MAAM,WAAW,8BAA8B;IAC9C,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,OAAO,CAAC;IACjB,kBAAkB,EAAE,OAAO,CAAC;IAC5B,QAAQ,EAAE,OAAO,CAAC;IAClB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,QAAQ,EAAE,MAAM,CAAC;CACjB;AAQD,qBAAa,YAAY;IACxB,QAAQ,EAAE,cAAc,EAAE,CAAM;IAChC,cAAc,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAsB;IACjE,2BAA2B,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAuB;IAC/E,qBAAqB,SAAM;IAC3B,aAAa,SAAK;IAClB,0BAA0B,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAa;IAC5D;;;;;OAKG;IACH,0BAA0B,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAa;IAErE,qBAAqB,CACpB,YAAY,EAAE,gBAAgB,GAAG,SAAS,EAC1C,MAAM,EAAE,gBAAgB,GAAG,IAAI,GAAG,SAAS,GACzC,IAAI;IAwHP,eAAe,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO;IAI9C,eAAe,IAAI,MAAM;IAIzB,uBAAuB,CAAC,MAAM,EAAE,WAAW,GAAG,MAAM;IAQpD,mBAAmB,IAAI,cAAc,EAAE;IAOvC,0BAA0B,CACzB,MAAM,EAAE,WAAW,EACnB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,EACrB,GAAG,SAAU,GACX,8BAA8B,EAAE;IAmEnC,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,cAAc,GAAG,IAAI;IAiBpD,0BAA0B,CAAC,MAAM,EAAE,WAAW,GAAG,cAAc,GAAG,IAAI;IAYtE,sBAAsB,CAAC,YAAY,EAAE,MAAM,EAAE,UAAU,SAAQ,GAAG,OAAO;IAWzE,cAAc,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI;IAK1C,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,GAAG,IAAI;IAmBrE,aAAa,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAS5C,YAAY,CACX,OAAO,EAAE,cAAc,EACvB,MAAM,EAAE,YAAY,GAAG,SAAS,GAAG,UAAU,EAC7C,MAAM,EAAE,WAAW,GACjB,IAAI;IAKP,aAAa,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO;IAwC/C,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAQ5C,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,cAAc,GAAG,IAAI;IASzE;;;;OAIG;IACH,oBAAoB,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO;IAStD,oBAAoB,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAKnD;;;;OAIG;IACH,mBAAmB,CAClB,OAAO,EAAE,cAAc,EACvB,MAAM,EAAE,WAAW,EACnB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,GACnB,OAAO;CAOV"}