oc-chatgpt-multi-auth 4.9.0

package/dist/lib/rotation.js

@@ -0,0 +1,248 @@
+/**
+ * Rotation Strategy Module
+ *
+ * Implements health-based account selection with token bucket rate limiting.
+ * Ported from antigravity-auth rotation logic for optimal account rotation
+ * when rate limits are encountered.
+ */
+export const DEFAULT_HEALTH_SCORE_CONFIG = {
+    successDelta: 1,
+    rateLimitDelta: -10,
+    failureDelta: -20,
+    maxScore: 100,
+    minScore: 0,
+    passiveRecoveryPerHour: 2,
+};
+/**
+ * Tracks health scores for accounts to prioritize healthy accounts.
+ * Accounts with higher health scores are preferred for selection.
+ */
+export class HealthScoreTracker {
+    entries = new Map();
+    config;
+    constructor(config = {}) {
+        this.config = { ...DEFAULT_HEALTH_SCORE_CONFIG, ...config };
+    }
+    getKey(accountIndex, quotaKey) {
+        return quotaKey ? `${accountIndex}:${quotaKey}` : `${accountIndex}`;
+    }
+    applyPassiveRecovery(entry) {
+        const now = Date.now();
+        const hoursSinceUpdate = (now - entry.lastUpdated) / (1000 * 60 * 60);
+        const recovery = hoursSinceUpdate * this.config.passiveRecoveryPerHour;
+        return Math.min(entry.score + recovery, this.config.maxScore);
+    }
+    getScore(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        const entry = this.entries.get(key);
+        if (!entry)
+            return this.config.maxScore;
+        return this.applyPassiveRecovery(entry);
+    }
+    getConsecutiveFailures(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        const entry = this.entries.get(key);
+        return entry?.consecutiveFailures ?? 0;
+    }
+    recordSuccess(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        const entry = this.entries.get(key);
+        const baseScore = entry ? this.applyPassiveRecovery(entry) : this.config.maxScore;
+        const newScore = Math.min(baseScore + this.config.successDelta, this.config.maxScore);
+        this.entries.set(key, {
+            score: newScore,
+            lastUpdated: Date.now(),
+            consecutiveFailures: 0,
+        });
+    }
+    recordRateLimit(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        const entry = this.entries.get(key);
+        const baseScore = entry ? this.applyPassiveRecovery(entry) : this.config.maxScore;
+        const newScore = Math.max(baseScore + this.config.rateLimitDelta, this.config.minScore);
+        this.entries.set(key, {
+            score: newScore,
+            lastUpdated: Date.now(),
+            consecutiveFailures: (entry?.consecutiveFailures ?? 0) + 1,
+        });
+    }
+    recordFailure(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        const entry = this.entries.get(key);
+        const baseScore = entry ? this.applyPassiveRecovery(entry) : this.config.maxScore;
+        const newScore = Math.max(baseScore + this.config.failureDelta, this.config.minScore);
+        this.entries.set(key, {
+            score: newScore,
+            lastUpdated: Date.now(),
+            consecutiveFailures: (entry?.consecutiveFailures ?? 0) + 1,
+        });
+    }
+    reset(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        this.entries.delete(key);
+    }
+    clear() {
+        this.entries.clear();
+    }
+}
+export const DEFAULT_TOKEN_BUCKET_CONFIG = {
+    maxTokens: 50,
+    tokensPerMinute: 6,
+};
+/**
+ * Client-side token bucket for rate limiting requests per account.
+ * Prevents sending requests to accounts that are likely to be rate-limited.
+ */
+export class TokenBucketTracker {
+    buckets = new Map();
+    config;
+    constructor(config = {}) {
+        this.config = { ...DEFAULT_TOKEN_BUCKET_CONFIG, ...config };
+    }
+    getKey(accountIndex, quotaKey) {
+        return quotaKey ? `${accountIndex}:${quotaKey}` : `${accountIndex}`;
+    }
+    refillTokens(entry) {
+        const now = Date.now();
+        const minutesSinceRefill = (now - entry.lastRefill) / (1000 * 60);
+        const tokensToAdd = minutesSinceRefill * this.config.tokensPerMinute;
+        return Math.min(entry.tokens + tokensToAdd, this.config.maxTokens);
+    }
+    getTokens(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        const entry = this.buckets.get(key);
+        if (!entry)
+            return this.config.maxTokens;
+        return this.refillTokens(entry);
+    }
+    /**
+     * Attempt to consume a token. Returns true if successful, false if bucket is empty.
+     */
+    tryConsume(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        const entry = this.buckets.get(key);
+        const currentTokens = entry ? this.refillTokens(entry) : this.config.maxTokens;
+        if (currentTokens < 1) {
+            return false;
+        }
+        this.buckets.set(key, {
+            tokens: currentTokens - 1,
+            lastRefill: Date.now(),
+        });
+        return true;
+    }
+    /**
+     * Drain tokens on rate limit to prevent immediate retries.
+     */
+    drain(accountIndex, quotaKey, drainAmount = 10) {
+        const key = this.getKey(accountIndex, quotaKey);
+        const entry = this.buckets.get(key);
+        const currentTokens = entry ? this.refillTokens(entry) : this.config.maxTokens;
+        this.buckets.set(key, {
+            tokens: Math.max(0, currentTokens - drainAmount),
+            lastRefill: Date.now(),
+        });
+    }
+    reset(accountIndex, quotaKey) {
+        const key = this.getKey(accountIndex, quotaKey);
+        this.buckets.delete(key);
+    }
+    clear() {
+        this.buckets.clear();
+    }
+}
+export const DEFAULT_HYBRID_SELECTION_CONFIG = {
+    healthWeight: 2,
+    tokenWeight: 5,
+    freshnessWeight: 0.1,
+};
+/**
+ * Selects the best account using a hybrid scoring strategy.
+ *
+ * Score = (health * healthWeight) + (tokens * tokenWeight) + (freshness * freshnessWeight)
+ *
+ * Where:
+ * - health: Account health score (0-100)
+ * - tokens: Available tokens in bucket (0-maxTokens)
+ * - freshness: Hours since last used (higher = more fresh for rotation)
+ */
+export function selectHybridAccount(accounts, healthTracker, tokenTracker, quotaKey, config = {}) {
+    const cfg = { ...DEFAULT_HYBRID_SELECTION_CONFIG, ...config };
+    const available = accounts.filter((a) => a.isAvailable);
+    if (available.length === 0)
+        return null;
+    if (available.length === 1)
+        return available[0];
+    const now = Date.now();
+    let bestAccount = null;
+    let bestScore = -Infinity;
+    for (const account of available) {
+        const health = healthTracker.getScore(account.index, quotaKey);
+        const tokens = tokenTracker.getTokens(account.index, quotaKey);
+        const hoursSinceUsed = (now - account.lastUsed) / (1000 * 60 * 60);
+        const score = health * cfg.healthWeight +
+            tokens * cfg.tokenWeight +
+            hoursSinceUsed * cfg.freshnessWeight;
+        if (score > bestScore) {
+            bestScore = score;
+            bestAccount = account;
+        }
+    }
+    return bestAccount;
+}
+// ============================================================================
+// Utility Functions
+// ============================================================================
+/**
+ * Adds random jitter to a delay value.
+ * @param baseMs - Base delay in milliseconds
+ * @param jitterFactor - Jitter factor (0-1), default 0.1 (10%)
+ * @returns Delay with jitter applied
+ */
+export function addJitter(baseMs, jitterFactor = 0.1) {
+    const jitter = baseMs * jitterFactor * (Math.random() * 2 - 1);
+    return Math.max(0, Math.floor(baseMs + jitter));
+}
+/**
+ * Returns a random delay within a range.
+ * @param minMs - Minimum delay in milliseconds
+ * @param maxMs - Maximum delay in milliseconds
+ * @returns Random delay within range
+ */
+export function randomDelay(minMs, maxMs) {
+    return Math.floor(minMs + Math.random() * (maxMs - minMs));
+}
+/**
+ * Calculates exponential backoff with jitter.
+ * @param attempt - Attempt number (1-based)
+ * @param baseMs - Base delay in milliseconds
+ * @param maxMs - Maximum delay in milliseconds
+ * @param jitterFactor - Jitter factor (0-1)
+ * @returns Backoff delay with jitter
+ */
+export function exponentialBackoff(attempt, baseMs = 1000, maxMs = 60000, jitterFactor = 0.1) {
+    const delay = Math.min(baseMs * Math.pow(2, attempt - 1), maxMs);
+    return addJitter(delay, jitterFactor);
+}
+// ============================================================================
+// Singleton Instances
+// ============================================================================
+let healthTrackerInstance = null;
+let tokenTrackerInstance = null;
+export function getHealthTracker(config) {
+    if (!healthTrackerInstance) {
+        healthTrackerInstance = new HealthScoreTracker(config);
+    }
+    return healthTrackerInstance;
+}
+export function getTokenTracker(config) {
+    if (!tokenTrackerInstance) {
+        tokenTrackerInstance = new TokenBucketTracker(config);
+    }
+    return tokenTrackerInstance;
+}
+export function resetTrackers() {
+    healthTrackerInstance?.clear();
+    tokenTrackerInstance?.clear();
+}
+//# sourceMappingURL=rotation.js.map

package/dist/lib/rotation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rotation.js","sourceRoot":"","sources":["../../lib/rotation.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAqBH,MAAM,CAAC,MAAM,2BAA2B,GAAsB;IAC5D,YAAY,EAAE,CAAC;IACf,cAAc,EAAE,CAAC,EAAE;IACnB,YAAY,EAAE,CAAC,EAAE;IACjB,QAAQ,EAAE,GAAG;IACb,QAAQ,EAAE,CAAC;IACX,sBAAsB,EAAE,CAAC;CAC1B,CAAC;AAQF;;;GAGG;AACH,MAAM,OAAO,kBAAkB;IACrB,OAAO,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC9C,MAAM,CAAoB;IAElC,YAAY,SAAqC,EAAE;QACjD,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,2BAA2B,EAAE,GAAG,MAAM,EAAE,CAAC;IAC9D,CAAC;IAEO,MAAM,CAAC,YAAoB,EAAE,QAAiB;QACpD,OAAO,QAAQ,CAAC,CAAC,CAAC,GAAG,YAAY,IAAI,QAAQ,EAAE,CAAC,CAAC,CAAC,GAAG,YAAY,EAAE,CAAC;IACtE,CAAC;IAEO,oBAAoB,CAAC,KAAkB;QAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,gBAAgB,GAAG,CAAC,GAAG,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;QACtE,MAAM,QAAQ,GAAG,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC;QACvE,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,GAAG,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAChE,CAAC;IAED,QAAQ,CAAC,YAAoB,EAAE,QAAiB;QAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;QACxC,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC;IAED,sBAAsB,CAAC,YAAoB,EAAE,QAAiB;QAC5D,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,OAAO,KAAK,EAAE,mBAAmB,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,aAAa,CAAC,YAAoB,EAAE,QAAiB;QACnD,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;QAClF,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;YACpB,KAAK,EAAE,QAAQ;YACf,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,mBAAmB,EAAE,CAAC;SACvB,CAAC,CAAC;IACL,CAAC;IAED,eAAe,CAAC,YAAoB,EAAE,QAAiB;QACrD,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;QAClF,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACxF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;YACpB,KAAK,EAAE,QAAQ;YACf,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,mBAAmB,EAAE,CAAC,KAAK,EAAE,mBAAmB,IAAI,CAAC,CAAC,GAAG,CAAC;SAC3D,CAAC,CAAC;IACL,CAAC;IAED,aAAa,CAAC,YAAoB,EAAE,QAAiB;QACnD,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;QAClF,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;YACpB,KAAK,EAAE,QAAQ;YACf,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,mBAAmB,EAAE,CAAC,KAAK,EAAE,mBAAmB,IAAI,CAAC,CAAC,GAAG,CAAC;SAC3D,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAoB,EAAE,QAAiB;QAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;CACF;AAaD,MAAM,CAAC,MAAM,2BAA2B,GAAsB;IAC5D,SAAS,EAAE,EAAE;IACb,eAAe,EAAE,CAAC;CACnB,CAAC;AAOF;;;GAGG;AACH,MAAM,OAAO,kBAAkB;IACrB,OAAO,GAAkC,IAAI,GAAG,EAAE,CAAC;IACnD,MAAM,CAAoB;IAElC,YAAY,SAAqC,EAAE;QACjD,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,2BAA2B,EAAE,GAAG,MAAM,EAAE,CAAC;IAC9D,CAAC;IAEO,MAAM,CAAC,YAAoB,EAAE,QAAiB;QACpD,OAAO,QAAQ,CAAC,CAAC,CAAC,GAAG,YAAY,IAAI,QAAQ,EAAE,CAAC,CAAC,CAAC,GAAG,YAAY,EAAE,CAAC;IACtE,CAAC;IAEO,YAAY,CAAC,KAAuB;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,kBAAkB,GAAG,CAAC,GAAG,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;QAClE,MAAM,WAAW,GAAG,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC;QACrE,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACrE,CAAC;IAED,SAAS,CAAC,YAAoB,EAAE,QAAiB;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;QACzC,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,YAAoB,EAAE,QAAiB;QAChD,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,aAAa,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;QAE/E,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;YACtB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;YACpB,MAAM,EAAE,aAAa,GAAG,CAAC;YACzB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;SACvB,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAoB,EAAE,QAAiB,EAAE,cAAsB,EAAE;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,aAAa,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;QAC/E,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;YACpB,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,aAAa,GAAG,WAAW,CAAC;YAChD,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;SACvB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAoB,EAAE,QAAiB;QAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;CACF;AAqBD,MAAM,CAAC,MAAM,+BAA+B,GAA0B;IACpE,YAAY,EAAE,CAAC;IACf,WAAW,EAAE,CAAC;IACd,eAAe,EAAE,GAAG;CACrB,CAAC;AAEF;;;;;;;;;GASG;AACH,MAAM,UAAU,mBAAmB,CACjC,QAA8B,EAC9B,aAAiC,EACjC,YAAgC,EAChC,QAAiB,EACjB,SAAyC,EAAE;IAE3C,MAAM,GAAG,GAAG,EAAE,GAAG,+BAA+B,EAAE,GAAG,MAAM,EAAE,CAAC;IAC9D,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IAExD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACxC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC;IAEhD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,IAAI,WAAW,GAA8B,IAAI,CAAC;IAClD,IAAI,SAAS,GAAG,CAAC,QAAQ,CAAC;IAE1B,KAAK,MAAM,OAAO,IAAI,SAAS,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC/D,MAAM,cAAc,GAAG,CAAC,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;QAEnE,MAAM,KAAK,GACT,MAAM,GAAG,GAAG,CAAC,YAAY;YACzB,MAAM,GAAG,GAAG,CAAC,WAAW;YACxB,cAAc,GAAG,GAAG,CAAC,eAAe,CAAC;QAEvC,IAAI,KAAK,GAAG,SAAS,EAAE,CAAC;YACtB,SAAS,GAAG,KAAK,CAAC;YAClB,WAAW,GAAG,OAAO,CAAC;QACxB,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,+EAA+E;AAC/E,oBAAoB;AACpB,+EAA+E;AAE/E;;;;;GAKG;AACH,MAAM,UAAU,SAAS,CAAC,MAAc,EAAE,eAAuB,GAAG;IAClE,MAAM,MAAM,GAAG,MAAM,GAAG,YAAY,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IAC/D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC;AAClD,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,WAAW,CAAC,KAAa,EAAE,KAAa;IACtD,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,kBAAkB,CAChC,OAAe,EACf,SAAiB,IAAI,EACrB,QAAgB,KAAK,EACrB,eAAuB,GAAG;IAE1B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjE,OAAO,SAAS,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;AACxC,CAAC;AAED,+EAA+E;AAC/E,sBAAsB;AACtB,+EAA+E;AAE/E,IAAI,qBAAqB,GAA8B,IAAI,CAAC;AAC5D,IAAI,oBAAoB,GAA8B,IAAI,CAAC;AAE3D,MAAM,UAAU,gBAAgB,CAAC,MAAmC;IAClE,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC3B,qBAAqB,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC;IACD,OAAO,qBAAqB,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,MAAmC;IACjE,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,oBAAoB,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;IACxD,CAAC;IACD,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,aAAa;IAC3B,qBAAqB,EAAE,KAAK,EAAE,CAAC;IAC/B,oBAAoB,EAAE,KAAK,EAAE,CAAC;AAChC,CAAC"}

package/dist/lib/storage.d.ts

@@ -0,0 +1,91 @@
+import { type ModelFamily } from "./prompts/codex.js";
+export type CooldownReason = "auth-failure" | "network-error";
+export interface RateLimitStateV3 {
+    [key: string]: number | undefined;
+}
+export interface AccountMetadataV1 {
+    accountId?: string;
+    email?: string;
+    refreshToken: string;
+    addedAt: number;
+    lastUsed: number;
+    lastSwitchReason?: "rate-limit" | "initial" | "rotation";
+    rateLimitResetTime?: number;
+    coolingDownUntil?: number;
+    cooldownReason?: CooldownReason;
+}
+export interface AccountStorageV1 {
+    version: 1;
+    accounts: AccountMetadataV1[];
+    activeIndex: number;
+}
+export interface AccountMetadataV3 {
+    accountId?: string;
+    email?: string;
+    refreshToken: string;
+    addedAt: number;
+    lastUsed: number;
+    lastSwitchReason?: "rate-limit" | "initial" | "rotation";
+    rateLimitResetTimes?: RateLimitStateV3;
+    coolingDownUntil?: number;
+    cooldownReason?: CooldownReason;
+}
+export interface AccountStorageV3 {
+    version: 3;
+    accounts: AccountMetadataV3[];
+    activeIndex: number;
+    activeIndexByFamily?: Partial<Record<ModelFamily, number>>;
+}
+/**
+ * Returns the file path for the account storage JSON file.
+ * @returns Absolute path to ~/.opencode/openai-codex-accounts.json
+ */
+export declare function getStoragePath(): string;
+/**
+ * Removes duplicate accounts, keeping the most recently used entry for each unique key.
+ * Deduplication is based on accountId or refreshToken.
+ * @param accounts - Array of accounts to deduplicate
+ * @returns New array with duplicates removed
+ */
+export declare function deduplicateAccounts<T extends {
+    accountId?: string;
+    refreshToken: string;
+    lastUsed?: number;
+    addedAt?: number;
+}>(accounts: T[]): T[];
+/**
+ * Removes duplicate accounts by email, keeping the most recently used entry.
+ * Accounts without email are always preserved.
+ * @param accounts - Array of accounts to deduplicate
+ * @returns New array with email duplicates removed
+ */
+export declare function deduplicateAccountsByEmail<T extends {
+    email?: string;
+    lastUsed?: number;
+    addedAt?: number;
+}>(accounts: T[]): T[];
+/**
+ * Normalizes and validates account storage data, migrating from v1 to v3 if needed.
+ * Handles deduplication, index clamping, and per-family active index mapping.
+ * @param data - Raw storage data (unknown format)
+ * @returns Normalized AccountStorageV3 or null if invalid
+ */
+export declare function normalizeAccountStorage(data: unknown): AccountStorageV3 | null;
+/**
+ * Loads OAuth accounts from disk storage.
+ * Automatically migrates v1 storage to v3 format if needed.
+ * @returns AccountStorageV3 if file exists and is valid, null otherwise
+ */
+export declare function loadAccounts(): Promise<AccountStorageV3 | null>;
+/**
+ * Persists account storage to disk.
+ * Creates the .opencode directory if it doesn't exist.
+ * @param storage - Account storage data to save
+ */
+export declare function saveAccounts(storage: AccountStorageV3): Promise<void>;
+/**
+ * Deletes the account storage file from disk.
+ * Silently ignores if file doesn't exist.
+ */
+export declare function clearAccounts(): Promise<void>;
+//# sourceMappingURL=storage.d.ts.map

package/dist/lib/storage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../lib/storage.ts"],"names":[],"mappings":"AAIA,OAAO,EAAkB,KAAK,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAetE,MAAM,MAAM,cAAc,GAAG,cAAc,GAAG,eAAe,CAAC;AAE9D,MAAM,WAAW,gBAAgB;IAC/B,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;CACnC;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,YAAY,GAAG,SAAS,GAAG,UAAU,CAAC;IACzD,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,CAAC,CAAC;IACX,QAAQ,EAAE,iBAAiB,EAAE,CAAC;IAC9B,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,YAAY,GAAG,SAAS,GAAG,UAAU,CAAC;IACzD,mBAAmB,CAAC,EAAE,gBAAgB,CAAC;IACvC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,CAAC,CAAC;IACX,QAAQ,EAAE,iBAAiB,EAAE,CAAC;IAC9B,WAAW,EAAE,MAAM,CAAC;IACpB,mBAAmB,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;CAC5D;AAgBD;;;GAGG;AACH,wBAAgB,cAAc,IAAI,MAAM,CAEvC;AAuDD;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,CAAC,SAAS;IAAE,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAA;CAAE,EAC7H,QAAQ,EAAE,CAAC,EAAE,GACZ,CAAC,EAAE,CAEL;AAED;;;;;GAKG;AACH,wBAAgB,0BAA0B,CAAC,CAAC,SAAS;IAAE,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAA;CAAE,EAC1G,QAAQ,EAAE,CAAC,EAAE,GACZ,CAAC,EAAE,CAoDL;AAiED;;;;;GAKG;AACH,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,OAAO,GAAG,gBAAgB,GAAG,IAAI,CAyF9E;AAED;;;;GAIG;AACH,wBAAsB,YAAY,IAAI,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,CA2BrE;AAED;;;;GAIG;AACH,wBAAsB,YAAY,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC,CAO3E;AAED;;;GAGG;AACH,wBAAsB,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC,CAYnD"}

package/dist/lib/storage.js

@@ -0,0 +1,323 @@
+import { promises as fs } from "node:fs";
+import { dirname, join } from "node:path";
+import { homedir } from "node:os";
+import { createLogger } from "./logger.js";
+import { MODEL_FAMILIES } from "./prompts/codex.js";
+const log = createLogger("storage");
+let storageMutex = Promise.resolve();
+function withStorageLock(fn) {
+    const previousMutex = storageMutex;
+    let releaseLock;
+    storageMutex = new Promise((resolve) => {
+        releaseLock = resolve;
+    });
+    return previousMutex.then(fn).finally(() => releaseLock());
+}
+function getConfigDir() {
+    return join(homedir(), ".opencode");
+}
+/**
+ * Returns the file path for the account storage JSON file.
+ * @returns Absolute path to ~/.opencode/openai-codex-accounts.json
+ */
+export function getStoragePath() {
+    return join(getConfigDir(), "openai-codex-accounts.json");
+}
+function nowMs() {
+    return Date.now();
+}
+function selectNewestAccount(current, candidate) {
+    if (!current)
+        return candidate;
+    const currentLastUsed = current.lastUsed || 0;
+    const candidateLastUsed = candidate.lastUsed || 0;
+    if (candidateLastUsed > currentLastUsed)
+        return candidate;
+    if (candidateLastUsed < currentLastUsed)
+        return current;
+    const currentAddedAt = current.addedAt || 0;
+    const candidateAddedAt = candidate.addedAt || 0;
+    return candidateAddedAt >= currentAddedAt ? candidate : current;
+}
+function deduplicateAccountsByKey(accounts) {
+    const keyToIndex = new Map();
+    const indicesToKeep = new Set();
+    for (let i = 0; i < accounts.length; i += 1) {
+        const account = accounts[i];
+        if (!account)
+            continue;
+        const key = account.accountId || account.refreshToken;
+        if (!key)
+            continue;
+        const existingIndex = keyToIndex.get(key);
+        if (existingIndex === undefined) {
+            keyToIndex.set(key, i);
+            continue;
+        }
+        const existing = accounts[existingIndex];
+        const newest = selectNewestAccount(existing, account);
+        keyToIndex.set(key, newest === account ? i : existingIndex);
+    }
+    for (const idx of keyToIndex.values()) {
+        indicesToKeep.add(idx);
+    }
+    const result = [];
+    for (let i = 0; i < accounts.length; i += 1) {
+        if (indicesToKeep.has(i)) {
+            const account = accounts[i];
+            if (account)
+                result.push(account);
+        }
+    }
+    return result;
+}
+/**
+ * Removes duplicate accounts, keeping the most recently used entry for each unique key.
+ * Deduplication is based on accountId or refreshToken.
+ * @param accounts - Array of accounts to deduplicate
+ * @returns New array with duplicates removed
+ */
+export function deduplicateAccounts(accounts) {
+    return deduplicateAccountsByKey(accounts);
+}
+/**
+ * Removes duplicate accounts by email, keeping the most recently used entry.
+ * Accounts without email are always preserved.
+ * @param accounts - Array of accounts to deduplicate
+ * @returns New array with email duplicates removed
+ */
+export function deduplicateAccountsByEmail(accounts) {
+    const emailToNewestIndex = new Map();
+    const indicesToKeep = new Set();
+    for (let i = 0; i < accounts.length; i += 1) {
+        const account = accounts[i];
+        if (!account)
+            continue;
+        const email = account.email?.trim();
+        if (!email) {
+            indicesToKeep.add(i);
+            continue;
+        }
+        const existingIndex = emailToNewestIndex.get(email);
+        if (existingIndex === undefined) {
+            emailToNewestIndex.set(email, i);
+            continue;
+        }
+        const existing = accounts[existingIndex];
+        if (!existing) {
+            emailToNewestIndex.set(email, i);
+            continue;
+        }
+        const existingLastUsed = existing.lastUsed || 0;
+        const candidateLastUsed = account.lastUsed || 0;
+        const existingAddedAt = existing.addedAt || 0;
+        const candidateAddedAt = account.addedAt || 0;
+        const isNewer = candidateLastUsed > existingLastUsed ||
+            (candidateLastUsed === existingLastUsed && candidateAddedAt > existingAddedAt);
+        if (isNewer) {
+            emailToNewestIndex.set(email, i);
+        }
+    }
+    for (const idx of emailToNewestIndex.values()) {
+        indicesToKeep.add(idx);
+    }
+    const result = [];
+    for (let i = 0; i < accounts.length; i += 1) {
+        if (indicesToKeep.has(i)) {
+            const account = accounts[i];
+            if (account)
+                result.push(account);
+        }
+    }
+    return result;
+}
+function isRecord(value) {
+    return !!value && typeof value === "object" && !Array.isArray(value);
+}
+function clampIndex(index, length) {
+    if (length <= 0)
+        return 0;
+    return Math.max(0, Math.min(index, length - 1));
+}
+function toAccountKey(account) {
+    return account.accountId || account.refreshToken;
+}
+function extractActiveKey(accounts, activeIndex) {
+    const candidate = accounts[activeIndex];
+    if (!isRecord(candidate))
+        return undefined;
+    const accountId = typeof candidate.accountId === "string" && candidate.accountId.trim()
+        ? candidate.accountId
+        : undefined;
+    const refreshToken = typeof candidate.refreshToken === "string" && candidate.refreshToken.trim()
+        ? candidate.refreshToken
+        : undefined;
+    return accountId || refreshToken;
+}
+function migrateV1ToV3(v1) {
+    const now = nowMs();
+    return {
+        version: 3,
+        accounts: v1.accounts.map((account) => {
+            const rateLimitResetTimes = {};
+            if (typeof account.rateLimitResetTime === "number" && account.rateLimitResetTime > now) {
+                for (const family of MODEL_FAMILIES) {
+                    rateLimitResetTimes[family] = account.rateLimitResetTime;
+                }
+            }
+            return {
+                accountId: account.accountId,
+                email: account.email,
+                refreshToken: account.refreshToken,
+                addedAt: account.addedAt,
+                lastUsed: account.lastUsed,
+                lastSwitchReason: account.lastSwitchReason,
+                rateLimitResetTimes: Object.keys(rateLimitResetTimes).length > 0 ? rateLimitResetTimes : undefined,
+                coolingDownUntil: account.coolingDownUntil,
+                cooldownReason: account.cooldownReason,
+            };
+        }),
+        activeIndex: v1.activeIndex,
+        activeIndexByFamily: {
+            "gpt-5.2-codex": v1.activeIndex,
+            "codex-max": v1.activeIndex,
+            codex: v1.activeIndex,
+            "gpt-5.2": v1.activeIndex,
+            "gpt-5.1": v1.activeIndex,
+        },
+    };
+}
+/**
+ * Normalizes and validates account storage data, migrating from v1 to v3 if needed.
+ * Handles deduplication, index clamping, and per-family active index mapping.
+ * @param data - Raw storage data (unknown format)
+ * @returns Normalized AccountStorageV3 or null if invalid
+ */
+export function normalizeAccountStorage(data) {
+    if (!isRecord(data)) {
+        log.warn("Invalid storage format, ignoring");
+        return null;
+    }
+    if (data.version !== 1 && data.version !== 3) {
+        log.warn("Unknown storage version, ignoring", {
+            version: data.version,
+        });
+        return null;
+    }
+    const rawAccounts = data.accounts;
+    if (!Array.isArray(rawAccounts)) {
+        log.warn("Invalid storage format, ignoring");
+        return null;
+    }
+    const activeIndexValue = typeof data.activeIndex === "number" && Number.isFinite(data.activeIndex)
+        ? data.activeIndex
+        : 0;
+    const rawActiveIndex = clampIndex(activeIndexValue, rawAccounts.length);
+    const activeKey = extractActiveKey(rawAccounts, rawActiveIndex);
+    const fromVersion = data.version;
+    const baseStorage = fromVersion === 1
+        ? migrateV1ToV3(data)
+        : data;
+    const validAccounts = rawAccounts.filter((account) => isRecord(account) && typeof account.refreshToken === "string" && !!account.refreshToken.trim());
+    const deduplicatedAccounts = deduplicateAccountsByEmail(deduplicateAccountsByKey(validAccounts));
+    const activeIndex = (() => {
+        if (deduplicatedAccounts.length === 0)
+            return 0;
+        if (activeKey) {
+            const mappedIndex = deduplicatedAccounts.findIndex((account) => toAccountKey(account) === activeKey);
+            if (mappedIndex >= 0)
+                return mappedIndex;
+        }
+        return clampIndex(rawActiveIndex, deduplicatedAccounts.length);
+    })();
+    const activeIndexByFamily = {};
+    const rawFamilyIndices = isRecord(baseStorage.activeIndexByFamily)
+        ? baseStorage.activeIndexByFamily
+        : {};
+    for (const family of MODEL_FAMILIES) {
+        const rawIndexValue = rawFamilyIndices[family];
+        const rawIndex = typeof rawIndexValue === "number" && Number.isFinite(rawIndexValue)
+            ? rawIndexValue
+            : rawActiveIndex;
+        const clampedRawIndex = clampIndex(rawIndex, rawAccounts.length);
+        const familyKey = extractActiveKey(rawAccounts, clampedRawIndex);
+        let mappedIndex = clampIndex(rawIndex, deduplicatedAccounts.length);
+        if (familyKey && deduplicatedAccounts.length > 0) {
+            const idx = deduplicatedAccounts.findIndex((account) => toAccountKey(account) === familyKey);
+            if (idx >= 0) {
+                mappedIndex = idx;
+            }
+        }
+        activeIndexByFamily[family] = mappedIndex;
+    }
+    return {
+        version: 3,
+        accounts: deduplicatedAccounts,
+        activeIndex,
+        activeIndexByFamily,
+    };
+}
+/**
+ * Loads OAuth accounts from disk storage.
+ * Automatically migrates v1 storage to v3 format if needed.
+ * @returns AccountStorageV3 if file exists and is valid, null otherwise
+ */
+export async function loadAccounts() {
+    try {
+        const path = getStoragePath();
+        const content = await fs.readFile(path, "utf-8");
+        const data = JSON.parse(content);
+        const normalized = normalizeAccountStorage(data);
+        const storedVersion = isRecord(data) ? data.version : undefined;
+        if (normalized && storedVersion !== normalized.version) {
+            log.info("Migrating account storage to v3", { from: storedVersion, to: normalized.version });
+            try {
+                await saveAccounts(normalized);
+            }
+            catch (saveError) {
+                log.warn("Failed to persist migrated storage", { error: String(saveError) });
+            }
+        }
+        return normalized;
+    }
+    catch (error) {
+        const code = error.code;
+        if (code === "ENOENT") {
+            return null;
+        }
+        log.error("Failed to load account storage", { error: String(error) });
+        return null;
+    }
+}
+/**
+ * Persists account storage to disk.
+ * Creates the .opencode directory if it doesn't exist.
+ * @param storage - Account storage data to save
+ */
+export async function saveAccounts(storage) {
+    return withStorageLock(async () => {
+        const path = getStoragePath();
+        await fs.mkdir(dirname(path), { recursive: true });
+        const content = JSON.stringify(storage, null, 2);
+        await fs.writeFile(path, content, "utf-8");
+    });
+}
+/**
+ * Deletes the account storage file from disk.
+ * Silently ignores if file doesn't exist.
+ */
+export async function clearAccounts() {
+    return withStorageLock(async () => {
+        try {
+            const path = getStoragePath();
+            await fs.unlink(path);
+        }
+        catch (error) {
+            const code = error.code;
+            if (code !== "ENOENT") {
+                log.error("Failed to clear account storage", { error: String(error) });
+            }
+        }
+    });
+}
+//# sourceMappingURL=storage.js.map