@jsonstudio/rcc 0.89.1189 → 0.89.1348

package/dist/manager/modules/quota/index.js

@@ -1,1094 +1,3 @@
-import fs from 'node:fs';
-import path from 'node:path';
-import { homedir } from 'node:os';
-import { fetchAntigravityQuotaSnapshot, loadAntigravityAccessToken } from '../../../providers/core/runtime/antigravity-quota-client.js';
-import { scanProviderTokenFiles } from '../../../providers/auth/token-scanner/index.js';
-import { resolveAntigravityApiBase } from '../../../providers/auth/antigravity-userinfo-helper.js';
-import { getProviderErrorCenter } from '../../../modules/llmswitch/bridge.js';
-import { readTokenFile, evaluateTokenState } from '../../../token-daemon/token-utils.js';
-import { applyErrorEvent as applyQuotaErrorEvent, applySuccessEvent as applyQuotaSuccessEvent, applyUsageEvent as applyQuotaUsageEvent, createInitialQuotaState, tickQuotaStateTime } from '../../quota/provider-quota-center.js';
-import { appendProviderErrorEvent, loadProviderQuotaSnapshot, saveProviderQuotaSnapshot } from '../../quota/provider-quota-store.js';
-export class QuotaManagerModule {
-    id = 'quota';
-    snapshot = {};
-    antigravityTokens = new Map();
-    refreshTimer = null;
-    quotaRoutingEnabled = true;
-    providerErrorCenter = null;
-    async init(context) {
-        this.snapshot = this.loadSnapshotFromDisk();
-        this.quotaRoutingEnabled = context.quotaRoutingEnabled !== false;
-    }
-    async start() {
-        // 启动时立即做一次最佳努力刷新，然后根据 token 过期时间和 15 分钟基准动态调度后续刷新。
-        try {
-            await this.refreshAllAntigravityQuotas();
-        }
-        catch {
-            // ignore startup refresh failures
-        }
-        void this.scheduleNextRefresh().catch(() => {
-            // ignore scheduling failures
-        });
-    }
-    async stop() {
-        if (this.refreshTimer) {
-            clearTimeout(this.refreshTimer);
-            this.refreshTimer = null;
-        }
-        this.saveSnapshotToDisk();
-    }
-    /**
-     * 用于 antigravity：注册需要追踪配额的 alias/token。
-     * 多次调用同一 alias 会覆盖最新配置。
-     */
-    registerAntigravityToken(alias, tokenFile, apiBase) {
-        const cleanAlias = alias.trim();
-        const cleanToken = tokenFile.trim();
-        const cleanBase = apiBase.trim();
-        if (!cleanAlias || !cleanToken || !cleanBase) {
-            return;
-        }
-        this.antigravityTokens.set(cleanAlias, {
-            alias: cleanAlias,
-            tokenFile: cleanToken,
-            apiBase: cleanBase
-        });
-    }
-    /**
-     * 用于 antigravity：根据 alias+model 更新配额快照。
-     */
-    updateAntigravityQuota(alias, quota) {
-        const aliasId = alias.trim();
-        if (!aliasId) {
-            return;
-        }
-        const now = Date.now();
-        const next = { ...this.snapshot };
-        for (const [modelId, info] of Object.entries(quota.models)) {
-            const key = this.buildAntigravityKey(aliasId, modelId);
-            const record = {
-                remainingFraction: Number.isFinite(info.remainingFraction) ? info.remainingFraction : null,
-                fetchedAt: quota.fetchedAt
-            };
-            const resetAt = this.computeResetAt(info.resetTimeRaw);
-            if (resetAt && resetAt > now) {
-                record.resetAt = resetAt;
-            }
-            next[key] = record;
-            const providerKey = `antigravity.${aliasId}.${modelId}`;
-            if (record.remainingFraction !== null && record.remainingFraction > 0) {
-                void this.emitQuotaRecoveryEvent(providerKey, modelId);
-            }
-            else {
-                const cooldownHint = record.resetAt ? Math.max(0, record.resetAt - now) : undefined;
-                void this.emitQuotaDepletedEvent(providerKey, modelId, cooldownHint);
-            }
-        }
-        this.snapshot = next;
-        this.saveSnapshotToDisk();
-    }
-    /**
-     * 判断给定 providerKey+model 是否有可用配额（仅针对 antigravity 语义）。
-     */
-    hasQuotaForAntigravity(providerKey, modelId) {
-        const alias = this.extractAntigravityAlias(providerKey);
-        if (!alias || !modelId) {
-            return true;
-        }
-        const key = this.buildAntigravityKey(alias, modelId);
-        const record = this.snapshot[key];
-        if (!record) {
-            // 没有任何配额记录时视为“无配额”，禁止进入路由池。
-            return false;
-        }
-        const now = Date.now();
-        // 如果已经超过 resetAt，但尚未刷新到新一轮配额，视为配额状态未知，同样禁止。
-        if (record.resetAt && record.resetAt <= now) {
-            return false;
-        }
-        if (record.remainingFraction === null) {
-            return false;
-        }
-        return record.remainingFraction > 0;
-    }
-    getRawSnapshot() {
-        return { ...this.snapshot };
-    }
-    buildAntigravityKey(alias, modelId) {
-        return `antigravity://${alias}/${modelId}`;
-    }
-    extractAntigravityAlias(providerKey) {
-        if (!providerKey || typeof providerKey !== 'string') {
-            return null;
-        }
-        const trimmed = providerKey.trim();
-        if (!trimmed.toLowerCase().startsWith('antigravity.')) {
-            return null;
-        }
-        const segments = trimmed.split('.');
-        if (segments.length < 2) {
-            return null;
-        }
-        return segments[1];
-    }
-    computeResetAt(raw) {
-        if (!raw || typeof raw !== 'string' || !raw.trim()) {
-            return undefined;
-        }
-        const value = raw.trim();
-        try {
-            const normalized = value.endsWith('Z') ? value.replace(/Z$/, '+00:00') : value;
-            const parsed = Date.parse(normalized);
-            if (!Number.isFinite(parsed) || parsed <= 0) {
-                return undefined;
-            }
-            return parsed;
-        }
-        catch {
-            return undefined;
-        }
-    }
-    async refreshAllAntigravityQuotas() {
-        await this.syncAntigravityTokensFromDisk();
-        if (this.antigravityTokens.size === 0) {
-            return;
-        }
-        for (const { alias, tokenFile, apiBase } of this.antigravityTokens.values()) {
-            try {
-                const accessToken = await loadAntigravityAccessToken(tokenFile);
-                if (!accessToken) {
-                    continue;
-                }
-                const snapshot = await fetchAntigravityQuotaSnapshot(apiBase, accessToken);
-                if (!snapshot) {
-                    continue;
-                }
-                this.updateAntigravityQuota(alias, snapshot);
-            }
-            catch {
-                // 单个 alias 失败不影响其他 alias 的刷新
-            }
-        }
-    }
-    /**
-     * 根据当前 token 池的过期时间和固定 15 分钟基准，动态安排下一次 quota 刷新：
-     * - 如有 token 会在 15 分钟内到期，则在该 token 到期时间附近刷新；
-     * - 否则按固定 15 分钟间隔刷新。
-     */
-    async scheduleNextRefresh() {
-        if (this.refreshTimer) {
-            clearTimeout(this.refreshTimer);
-            this.refreshTimer = null;
-        }
-        const baseIntervalMs = 15 * 60 * 1000;
-        let delayMs = baseIntervalMs;
-        try {
-            const nextExpiryDelay = await this.computeNextTokenExpiryDelayMs();
-            if (nextExpiryDelay !== null && nextExpiryDelay > 0 && nextExpiryDelay < baseIntervalMs) {
-                delayMs = nextExpiryDelay;
-            }
-        }
-        catch {
-            // 如果计算失败，退回到固定 15 分钟间隔
-            delayMs = baseIntervalMs;
-        }
-        this.refreshTimer = setTimeout(() => {
-            void this.refreshAllAntigravityQuotas()
-                .catch(() => {
-                // ignore refresh failure
-            })
-                .finally(() => {
-                void this.scheduleNextRefresh().catch(() => {
-                    // ignore reschedule failure
-                });
-            });
-        }, delayMs);
-    }
-    /**
-     * 扫描 antigravity token 文件，计算距离最近一次 token 过期还剩多少毫秒。
-     * 若所有 token 都无过期时间或已过期，则返回 null。
-     */
-    async computeNextTokenExpiryDelayMs() {
-        let matches = [];
-        try {
-            const raw = await scanProviderTokenFiles('antigravity');
-            matches = raw.map((m) => ({ filePath: m.filePath }));
-        }
-        catch {
-            matches = [];
-        }
-        if (!matches.length) {
-            return null;
-        }
-        const now = Date.now();
-        let minDelay = null;
-        for (const match of matches) {
-            try {
-                const token = await readTokenFile(match.filePath);
-                const state = evaluateTokenState(token, now);
-                const msLeft = state.msUntilExpiry;
-                if (msLeft === null || msLeft <= 0) {
-                    continue;
-                }
-                if (minDelay === null || msLeft < minDelay) {
-                    minDelay = msLeft;
-                }
-            }
-            catch {
-                // ignore single token file errors
-            }
-        }
-        return minDelay;
-    }
-    /**
-     * 自动从本地 auth 目录扫描 antigravity OAuth token，并同步到内存注册表。
-     * 这确保「每个 token」都能定期刷新 quota，而不依赖额外的显式注册流程。
-     */
-    async syncAntigravityTokensFromDisk() {
-        let matches = [];
-        try {
-            matches = await scanProviderTokenFiles('antigravity');
-        }
-        catch {
-            matches = [];
-        }
-        if (!matches.length) {
-            this.antigravityTokens.clear();
-            return;
-        }
-        const base = resolveAntigravityApiBase();
-        const next = new Map();
-        for (const match of matches) {
-            const label = match.alias && match.alias !== 'default'
-                ? `${match.sequence}-${match.alias}`
-                : String(match.sequence);
-            const alias = label.trim();
-            if (!alias) {
-                continue;
-            }
-            next.set(alias, {
-                alias,
-                tokenFile: match.filePath,
-                apiBase: base
-            });
-        }
-        // 若已有显式注册的 alias，保留其覆盖权
-        for (const [alias, reg] of this.antigravityTokens.entries()) {
-            if (!next.has(alias)) {
-                next.set(alias, reg);
-            }
-        }
-        this.antigravityTokens = next;
-    }
-    resolveStatePath() {
-        const baseDir = path.join(homedir(), '.routecodex', 'state', 'quota');
-        try {
-            fs.mkdirSync(baseDir, { recursive: true });
-        }
-        catch {
-            // best effort
-        }
-        return path.join(baseDir, 'antigravity.json');
-    }
-    loadSnapshotFromDisk() {
-        const filePath = this.resolveStatePath();
-        try {
-            if (!fs.existsSync(filePath)) {
-                return {};
-            }
-            const content = fs.readFileSync(filePath, 'utf8');
-            const parsed = content.trim() ? JSON.parse(content) : {};
-            if (!parsed || typeof parsed !== 'object' || Array.isArray(parsed)) {
-                return {};
-            }
-            const raw = parsed;
-            const result = {};
-            for (const [key, value] of Object.entries(raw)) {
-                if (!value || typeof value !== 'object') {
-                    continue;
-                }
-                let remainingFraction = null;
-                if (typeof value.remainingFraction === 'number') {
-                    remainingFraction = value.remainingFraction ?? null;
-                }
-                let resetAt;
-                if (typeof value.resetAt === 'number') {
-                    resetAt = value.resetAt;
-                }
-                const fetchedAt = typeof value.fetchedAt === 'number'
-                    ? value.fetchedAt
-                    : Date.now();
-                result[key] = { remainingFraction, resetAt, fetchedAt };
-            }
-            return result;
-        }
-        catch {
-            return {};
-        }
-    }
-    saveSnapshotToDisk() {
-        const filePath = this.resolveStatePath();
-        try {
-            fs.writeFileSync(filePath, `${JSON.stringify(this.snapshot, null, 2)}\n`, 'utf8');
-        }
-        catch {
-            // best effort
-        }
-    }
-    async getProviderErrorCenterInstance() {
-        if (this.providerErrorCenter) {
-            return this.providerErrorCenter;
-        }
-        try {
-            const center = await getProviderErrorCenter();
-            if (center && typeof center.emit === 'function') {
-                this.providerErrorCenter = center;
-            }
-            else {
-                this.providerErrorCenter = null;
-            }
-        }
-        catch {
-            this.providerErrorCenter = null;
-        }
-        return this.providerErrorCenter;
-    }
-    async emitQuotaRecoveryEvent(providerKey, modelId) {
-        if (!providerKey || !modelId) {
-            return;
-        }
-        if (!this.quotaRoutingEnabled) {
-            return;
-        }
-        const center = await this.getProviderErrorCenterInstance();
-        if (!center) {
-            return;
-        }
-        const now = Date.now();
-        const event = {
-            code: 'QUOTA_RECOVERY',
-            message: 'Quota manager: provider quota refreshed',
-            stage: 'quota',
-            status: 200,
-            recoverable: true,
-            runtime: {
-                requestId: `quota_${now}`,
-                providerKey,
-                providerId: 'antigravity'
-            },
-            timestamp: now,
-            details: {
-                virtualRouterQuotaRecovery: {
-                    providerKey,
-                    reason: `quota>0 for model ${modelId}`,
-                    source: 'quota-manager'
-                }
-            }
-        };
-        try {
-            center.emit(event);
-        }
-        catch {
-            // 忽略 error center 失败，避免影响配额刷新流程
-        }
-    }
-    async emitQuotaDepletedEvent(providerKey, modelId, cooldownMs) {
-        if (!providerKey || !modelId) {
-            return;
-        }
-        if (!this.quotaRoutingEnabled) {
-            return;
-        }
-        const center = await this.getProviderErrorCenterInstance();
-        if (!center) {
-            return;
-        }
-        const now = Date.now();
-        const detail = {
-            virtualRouterQuotaDepleted: {
-                providerKey,
-                reason: `quota<=0 for model ${modelId}`,
-                ...(typeof cooldownMs === 'number' && cooldownMs > 0 ? { cooldownMs } : {})
-            }
-        };
-        const event = {
-            code: 'QUOTA_DEPLETED',
-            message: 'Quota manager: provider quota exhausted',
-            stage: 'quota',
-            status: 429,
-            recoverable: false,
-            runtime: {
-                requestId: `quota_${now}`,
-                providerKey,
-                providerId: 'antigravity'
-            },
-            timestamp: now,
-            details: detail
-        };
-        try {
-            center.emit(event);
-        }
-        catch {
-            // ignore emit errors
-        }
-    }
-}
-export class ProviderQuotaDaemonModule {
-    id = 'provider-quota';
-    quotaStates = new Map();
-    staticConfigs = new Map();
-    unsubscribe = null;
-    maintenanceTimer = null;
-    persistTimer = null;
-    quotaRoutingEnabled = true;
-    async loadSnapshotIntoMemory() {
-        // Always clear in-memory state first so operator actions like deleting provider-quota.json
-        // take effect immediately after a reload.
-        this.quotaStates = new Map();
-        const snapshot = await loadProviderQuotaSnapshot();
-        if (snapshot && snapshot.providers && typeof snapshot.providers === 'object') {
-            for (const [providerKey, state] of Object.entries(snapshot.providers)) {
-                if (state && typeof state === 'object') {
-                    this.quotaStates.set(providerKey, normalizeLoadedQuotaState(providerKey, state));
-                }
-            }
-        }
-        if (this.quotaStates.size) {
-            const nowMs = Date.now();
-            let changed = false;
-            for (const [providerKey, state] of this.quotaStates.entries()) {
-                const next = tickQuotaStateTime(state, nowMs);
-                if (next !== state) {
-                    this.quotaStates.set(providerKey, next);
-                    changed = true;
-                }
-            }
-            if (changed) {
-                this.schedulePersist(nowMs);
-            }
-        }
-        // Ensure we always have default entries for known providers (seeded via static configs),
-        // so deleting the snapshot file doesn't permanently "hide" providers from admin views.
-        if (this.staticConfigs.size) {
-            const nowMs = Date.now();
-            let seeded = false;
-            for (const [providerKey, cfg] of this.staticConfigs.entries()) {
-                if (this.quotaStates.has(providerKey)) {
-                    continue;
-                }
-                this.quotaStates.set(providerKey, createInitialQuotaState(providerKey, cfg, nowMs));
-                seeded = true;
-            }
-            if (seeded) {
-                this.schedulePersist(nowMs);
-            }
-        }
-    }
-    async init(context) {
-        this.quotaRoutingEnabled = context.quotaRoutingEnabled !== false;
-        try {
-            await this.loadSnapshotIntoMemory();
-        }
-        catch {
-            this.quotaStates = new Map();
-        }
-    }
-    async reloadFromDisk() {
-        await this.loadSnapshotIntoMemory();
-        return { loadedAt: Date.now(), providerCount: this.quotaStates.size };
-    }
-    async reset(options = {}) {
-        const nowMs = Date.now();
-        this.quotaStates = new Map();
-        // Rebuild default quota entries for known providers so routing can recover immediately.
-        if (this.staticConfigs.size) {
-            for (const [providerKey, cfg] of this.staticConfigs.entries()) {
-                this.quotaStates.set(providerKey, createInitialQuotaState(providerKey, cfg, nowMs));
-            }
-        }
-        const persisted = options.persist !== false;
-        if (persisted) {
-            try {
-                await saveProviderQuotaSnapshot(this.toSnapshotObject(), new Date(nowMs));
-            }
-            catch {
-                // ignore persistence failure
-            }
-        }
-        return { resetAt: nowMs, persisted };
-    }
-    async resetProvider(providerKey) {
-        const key = typeof providerKey === 'string' ? providerKey.trim() : '';
-        if (!key) {
-            return null;
-        }
-        const nowMs = Date.now();
-        const next = createInitialQuotaState(key, this.staticConfigs.get(key), nowMs);
-        this.quotaStates.set(key, next);
-        try {
-            await saveProviderQuotaSnapshot(this.toSnapshotObject(), new Date(nowMs));
-        }
-        catch {
-            // ignore persistence failure
-        }
-        return { providerKey: key, state: next };
-    }
-    async recoverProvider(providerKey) {
-        const key = typeof providerKey === 'string' ? providerKey.trim() : '';
-        if (!key) {
-            return null;
-        }
-        const nowMs = Date.now();
-        const previous = this.quotaStates.get(key) ??
-            createInitialQuotaState(key, this.staticConfigs.get(key), nowMs);
-        const next = {
-            ...previous,
-            inPool: true,
-            reason: 'ok',
-            cooldownUntil: null,
-            blacklistUntil: null,
-            lastErrorSeries: null,
-            consecutiveErrorCount: 0
-        };
-        this.quotaStates.set(key, next);
-        try {
-            await saveProviderQuotaSnapshot(this.toSnapshotObject(), new Date(nowMs));
-        }
-        catch {
-            // ignore persistence failure
-        }
-        return { providerKey: key, state: next };
-    }
-    async disableProvider(options) {
-        const key = typeof options?.providerKey === 'string' ? options.providerKey.trim() : '';
-        if (!key) {
-            return null;
-        }
-        const durationMs = typeof options.durationMs === 'number' && Number.isFinite(options.durationMs) && options.durationMs > 0
-            ? Math.floor(options.durationMs)
-            : 0;
-        if (!durationMs) {
-            return null;
-        }
-        const mode = options.mode === 'blacklist' ? 'blacklist' : 'cooldown';
-        const nowMs = Date.now();
-        const previous = this.quotaStates.get(key) ??
-            createInitialQuotaState(key, this.staticConfigs.get(key), nowMs);
-        const next = mode === 'blacklist'
-            ? {
-                ...previous,
-                inPool: false,
-                reason: 'blacklist',
-                blacklistUntil: nowMs + durationMs,
-                cooldownUntil: null
-            }
-            : {
-                ...previous,
-                inPool: false,
-                reason: 'cooldown',
-                cooldownUntil: nowMs + durationMs
-            };
-        this.quotaStates.set(key, next);
-        try {
-            await saveProviderQuotaSnapshot(this.toSnapshotObject(), new Date(nowMs));
-        }
-        catch {
-            // ignore persistence failure
-        }
-        return { providerKey: key, state: next };
-    }
-    getAdminSnapshot() {
-        return this.toSnapshotObject();
-    }
-    async start() {
-        if (!this.quotaRoutingEnabled) {
-            return;
-        }
-        let center = null;
-        try {
-            center = await getProviderErrorCenter();
-        }
-        catch {
-            center = null;
-        }
-        if (center && typeof center.subscribe === 'function') {
-            this.unsubscribe = center.subscribe((event) => {
-                void this.handleProviderErrorEvent(event).catch(() => {
-                    // swallow handler errors to avoid unhandled rejection noise; quota updates are best-effort
-                });
-            });
-        }
-        const intervalMs = readPositiveNumberFromEnv('ROUTECODEX_QUOTA_DAEMON_INTERVAL_MS', 60_000);
-        if (intervalMs > 0) {
-            this.maintenanceTimer = setInterval(() => {
-                void this.runMaintenanceTick().catch(() => {
-                    // ignore maintenance failures
-                });
-            }, intervalMs);
-        }
-        // Run a one-off maintenance tick immediately so expired cooldown/blacklist entries
-        // are cleared even before the first request hits quotaView.
-        void this.runMaintenanceTick().catch(() => {
-            // ignore immediate tick failures
-        });
-    }
-    async stop() {
-        if (this.unsubscribe) {
-            try {
-                this.unsubscribe();
-            }
-            catch {
-                // ignore unsubscribe failures
-            }
-            this.unsubscribe = null;
-        }
-        if (this.maintenanceTimer) {
-            clearInterval(this.maintenanceTimer);
-            this.maintenanceTimer = null;
-        }
-        if (this.persistTimer) {
-            clearTimeout(this.persistTimer);
-            this.persistTimer = null;
-        }
-        if (!this.quotaRoutingEnabled) {
-            return;
-        }
-        try {
-            await saveProviderQuotaSnapshot(this.toSnapshotObject(), new Date());
-        }
-        catch {
-            // best-effort persistence
-        }
-    }
-    recordProviderUsage(event) {
-        if (!this.quotaRoutingEnabled) {
-            return;
-        }
-        const providerKey = typeof event?.providerKey === 'string' ? event.providerKey.trim() : '';
-        if (!providerKey) {
-            return;
-        }
-        const nowMs = typeof event.timestampMs === 'number' && Number.isFinite(event.timestampMs) && event.timestampMs > 0
-            ? event.timestampMs
-            : Date.now();
-        const requestedTokens = typeof event.requestedTokens === 'number' && Number.isFinite(event.requestedTokens) && event.requestedTokens > 0
-            ? event.requestedTokens
-            : 0;
-        const previous = this.quotaStates.get(providerKey) ??
-            createInitialQuotaState(providerKey, this.staticConfigs.get(providerKey), nowMs);
-        const nextState = applyQuotaUsageEvent(previous, { providerKey, requestedTokens, timestampMs: nowMs }, nowMs);
-        this.quotaStates.set(providerKey, nextState);
-        this.schedulePersist(nowMs);
-    }
-    recordProviderSuccess(event) {
-        if (!this.quotaRoutingEnabled) {
-            return;
-        }
-        const providerKey = typeof event?.providerKey === 'string' ? event.providerKey.trim() : '';
-        if (!providerKey) {
-            return;
-        }
-        const nowMs = typeof event.timestampMs === 'number' && Number.isFinite(event.timestampMs) && event.timestampMs > 0
-            ? event.timestampMs
-            : Date.now();
-        const usedTokens = typeof event.usedTokens === 'number' && Number.isFinite(event.usedTokens) && event.usedTokens > 0
-            ? event.usedTokens
-            : 0;
-        const previous = this.quotaStates.get(providerKey) ??
-            createInitialQuotaState(providerKey, this.staticConfigs.get(providerKey), nowMs);
-        const nextState = applyQuotaSuccessEvent(previous, { providerKey, usedTokens, timestampMs: nowMs }, nowMs);
-        this.quotaStates.set(providerKey, nextState);
-        this.schedulePersist(nowMs);
-    }
-    registerProviderStaticConfig(providerKey, config = {}) {
-        if (!this.quotaRoutingEnabled) {
-            return;
-        }
-        const key = typeof providerKey === 'string' ? providerKey.trim() : '';
-        if (!key) {
-            return;
-        }
-        const authTypeRaw = typeof config.authType === 'string' ? config.authType.trim().toLowerCase() : '';
-        const authType = authTypeRaw === 'apikey' ? 'apikey' : authTypeRaw === 'oauth' ? 'oauth' : 'unknown';
-        const staticConfig = {
-            ...(typeof config.priorityTier === 'number' && Number.isFinite(config.priorityTier)
-                ? { priorityTier: config.priorityTier }
-                : {}),
-            authType
-        };
-        this.staticConfigs.set(key, staticConfig);
-        const nowMs = Date.now();
-        const existing = this.quotaStates.get(key);
-        if (existing) {
-            this.quotaStates.set(key, {
-                ...existing,
-                authType,
-                ...(typeof staticConfig.priorityTier === 'number' ? { priorityTier: staticConfig.priorityTier } : {})
-            });
-            return;
-        }
-        // If snapshot is missing (or operator cleared it), we still want a predictable default:
-        // API-key providers start "inPool: true" with unlimited quota until errors are observed.
-        // We persist (debounced) so admin tools can inspect the pool quickly after startup.
-        const initial = createInitialQuotaState(key, staticConfig, nowMs);
-        // Antigravity OAuth providers require an explicit quota snapshot before being routed:
-        // - If we haven't observed quota for a model yet, treat it as "no quota" and keep it out of pool.
-        // - QuotaManagerModule will emit QUOTA_RECOVERY / QUOTA_DEPLETED to flip this state.
-        // This prevents sticky/prefer targets from repeatedly hitting an untracked model and returning 429s.
-        const isAntigravity = key.toLowerCase().startsWith('antigravity.');
-        if (isAntigravity && authType === 'oauth') {
-            this.quotaStates.set(key, {
-                ...initial,
-                inPool: false,
-                reason: 'cooldown',
-                cooldownUntil: null
-            });
-        }
-        else {
-            this.quotaStates.set(key, initial);
-        }
-        this.schedulePersist(nowMs);
-    }
-    async handleProviderErrorEvent(event) {
-        if (!event) {
-            return;
-        }
-        const code = typeof event.code === 'string' ? event.code : '';
-        const providerKey = this.extractProviderKey(event);
-        if (!providerKey) {
-            return;
-        }
-        const nowMs = typeof event.timestamp === 'number' && Number.isFinite(event.timestamp) && event.timestamp > 0
-            ? event.timestamp
-            : Date.now();
-        const previous = this.quotaStates.get(providerKey) ??
-            createInitialQuotaState(providerKey, this.staticConfigs.get(providerKey), nowMs);
-        // fatal 黑名单在锁定期内，不应被其它事件（包括 429/配额信号）改变。
-        if (previous.reason === 'fatal' && previous.blacklistUntil && nowMs < previous.blacklistUntil) {
-            return;
-        }
-        // QUOTA_* 属于“确定性配额信号”，不进入错误 series 统计。
-        if (code === 'QUOTA_DEPLETED') {
-            const detailCarrier = (event.details && typeof event.details === 'object') ? event.details : {};
-            const raw = detailCarrier.virtualRouterQuotaDepleted;
-            const cooldownMs = raw && typeof raw === 'object' && typeof raw.cooldownMs === 'number'
-                ? raw.cooldownMs
-                : undefined;
-            const ttl = typeof cooldownMs === 'number' && Number.isFinite(cooldownMs) && cooldownMs > 0
-                ? cooldownMs
-                : undefined;
-            const nextState = {
-                ...previous,
-                inPool: false,
-                reason: 'quotaDepleted',
-                cooldownUntil: ttl ? nowMs + ttl : previous.cooldownUntil
-            };
-            this.quotaStates.set(providerKey, nextState);
-            this.schedulePersist(nowMs);
-            return;
-        }
-        if (code === 'QUOTA_RECOVERY') {
-            const withinBlacklist = previous.blacklistUntil !== null && nowMs < previous.blacklistUntil;
-            const withinFatalBlacklist = previous.reason === 'fatal' && previous.blacklistUntil !== null && nowMs < previous.blacklistUntil;
-            if (!withinBlacklist && !withinFatalBlacklist) {
-                const nextState = {
-                    ...previous,
-                    inPool: true,
-                    reason: 'ok',
-                    cooldownUntil: null
-                };
-                this.quotaStates.set(providerKey, nextState);
-                this.schedulePersist(nowMs);
-            }
-            return;
-        }
-        // Gemini-family quota exhausted errors often carry quota reset delay.
-        // When present, treat as deterministic quota depletion signal rather than generic 429 backoff/blacklist.
-        if (typeof event.status === 'number' && event.status === 429) {
-            const seriesCooldownUntil = extractVirtualRouterSeriesCooldownUntil(event, nowMs);
-            if (seriesCooldownUntil) {
-                const nextState = {
-                    ...previous,
-                    inPool: false,
-                    reason: 'quotaDepleted',
-                    cooldownUntil: seriesCooldownUntil,
-                    blacklistUntil: null,
-                    lastErrorSeries: null,
-                    consecutiveErrorCount: 0
-                };
-                this.quotaStates.set(providerKey, nextState);
-                this.schedulePersist(nowMs);
-                return;
-            }
-            const runtime = event.runtime;
-            const providerIdRaw = runtime && typeof runtime.providerId === 'string' ? runtime.providerId.trim().toLowerCase() : '';
-            const isQuotaProvider = providerIdRaw === 'antigravity' || providerIdRaw === 'gemini-cli';
-            if (isQuotaProvider) {
-                const ttl = parseQuotaResetDelayMs(event);
-                if (ttl && ttl > 0) {
-                    const nextState = {
-                        ...previous,
-                        inPool: false,
-                        reason: 'quotaDepleted',
-                        cooldownUntil: nowMs + ttl,
-                        blacklistUntil: null,
-                        lastErrorSeries: null,
-                        consecutiveErrorCount: 0
-                    };
-                    this.quotaStates.set(providerKey, nextState);
-                    this.schedulePersist(nowMs);
-                    return;
-                }
-            }
-        }
-        const errorForQuota = {
-            providerKey,
-            httpStatus: typeof event.status === 'number' ? event.status : undefined,
-            code: typeof event.code === 'string' ? event.code : undefined,
-            fatal: this.isFatalForQuota(event),
-            timestampMs: nowMs
-        };
-        const nextState = applyQuotaErrorEvent(previous, errorForQuota, nowMs);
-        this.quotaStates.set(providerKey, nextState);
-        const tsIso = new Date(nowMs).toISOString();
-        try {
-            await appendProviderErrorEvent({
-                ts: tsIso,
-                providerKey,
-                code: typeof errorForQuota.code === 'string' ? errorForQuota.code : undefined,
-                httpStatus: typeof errorForQuota.httpStatus === 'number' ? errorForQuota.httpStatus : undefined,
-                message: event.message,
-                details: {
-                    stage: event.stage,
-                    routeName: event.runtime.routeName,
-                    entryEndpoint: event.runtime.entryEndpoint
-                }
-            });
-        }
-        catch {
-            // logging failure is non-fatal
-        }
-        try {
-            await saveProviderQuotaSnapshot(this.toSnapshotObject(), new Date(nowMs));
-        }
-        catch {
-            // best-effort persistence only
-        }
-    }
-    async runMaintenanceTick() {
-        if (!this.quotaStates.size) {
-            return;
-        }
-        const nowMs = Date.now();
-        const updated = new Map();
-        for (const [providerKey, state] of this.quotaStates.entries()) {
-            const next = tickQuotaStateTime(state, nowMs);
-            updated.set(providerKey, next);
-        }
-        this.quotaStates = updated;
-        try {
-            await saveProviderQuotaSnapshot(this.toSnapshotObject(), new Date(nowMs));
-        }
-        catch {
-            // ignore persistence errors
-        }
-    }
-    schedulePersist(_nowMs) {
-        if (this.persistTimer) {
-            return;
-        }
-        const debounceMs = readPositiveNumberFromEnv('ROUTECODEX_QUOTA_PERSIST_DEBOUNCE_MS', 5_000);
-        this.persistTimer = setTimeout(() => {
-            this.persistTimer = null;
-            void saveProviderQuotaSnapshot(this.toSnapshotObject(), new Date()).catch(() => {
-                // ignore persistence errors
-            });
-        }, debounceMs);
-    }
-    toSnapshotObject() {
-        const result = {};
-        for (const [key, state] of this.quotaStates.entries()) {
-            result[key] = state;
-        }
-        return result;
-    }
-    extractProviderKey(event) {
-        const runtime = event.runtime;
-        const direct = runtime && typeof runtime.providerKey === 'string' && runtime.providerKey.trim()
-            ? runtime.providerKey.trim()
-            : null;
-        if (direct) {
-            return direct;
-        }
-        const target = runtime && runtime.target;
-        if (target && typeof target === 'object') {
-            const targetKey = target.providerKey;
-            if (typeof targetKey === 'string' && targetKey.trim()) {
-                return targetKey.trim();
-            }
-        }
-        return null;
-    }
-    getQuotaView() {
-        if (!this.quotaRoutingEnabled) {
-            return () => null;
-        }
-        return (providerKey) => {
-            const key = typeof providerKey === 'string' ? providerKey.trim() : '';
-            if (!key) {
-                return null;
-            }
-            const state = this.quotaStates.get(key);
-            if (!state) {
-                return null;
-            }
-            // 视图层做一次“即时修复”，确保即使 maintenance tick 未运行，
-            // 冷却/黑名单到期也能立刻恢复可用状态，避免路由长期卡死。
-            const nowMs = Date.now();
-            const normalized = tickQuotaStateTime(state, nowMs);
-            if (normalized !== state) {
-                this.quotaStates.set(key, normalized);
-                this.schedulePersist(nowMs);
-            }
-            const effective = normalized;
-            return {
-                providerKey: effective.providerKey,
-                inPool: effective.inPool,
-                reason: effective.reason,
-                priorityTier: effective.priorityTier,
-                cooldownUntil: effective.cooldownUntil ?? null,
-                blacklistUntil: effective.blacklistUntil ?? null
-            };
-        };
-    }
-    isFatalForQuota(event) {
-        const status = typeof event.status === 'number' ? event.status : undefined;
-        const code = typeof event.code === 'string' ? event.code.toUpperCase() : '';
-        const stage = typeof event.stage === 'string' ? event.stage.toLowerCase() : '';
-        if (status === 401 || status === 402 || status === 403) {
-            return true;
-        }
-        if (code.includes('AUTH') || code.includes('UNAUTHORIZED')) {
-            return true;
-        }
-        if (code.includes('CONFIG')) {
-            return true;
-        }
-        if (stage.includes('compat')) {
-            return true;
-        }
-        if (event.recoverable === false && status !== undefined && status >= 500) {
-            return true;
-        }
-        return false;
-    }
-}
-function extractVirtualRouterSeriesCooldownUntil(event, nowMs) {
-    if (!event || !event.details || typeof event.details !== 'object') {
-        return null;
-    }
-    const raw = event.details.virtualRouterSeriesCooldown;
-    if (!raw || typeof raw !== 'object') {
-        return null;
-    }
-    const record = raw;
-    const cooldownMsRaw = record.cooldownMs;
-    const expiresAtRaw = record.expiresAt;
-    const expiresAt = typeof expiresAtRaw === 'number' && Number.isFinite(expiresAtRaw) && expiresAtRaw > nowMs
-        ? expiresAtRaw
-        : null;
-    if (expiresAt) {
-        return expiresAt;
-    }
-    const cooldownMs = typeof cooldownMsRaw === 'number'
-        ? cooldownMsRaw
-        : typeof cooldownMsRaw === 'string'
-            ? Number.parseFloat(cooldownMsRaw)
-            : Number.NaN;
-    if (!Number.isFinite(cooldownMs) || cooldownMs <= 0) {
-        return null;
-    }
-    return nowMs + cooldownMs;
-}
-function parseQuotaResetDelayMs(event) {
-    const message = typeof event.message === 'string' ? event.message : '';
-    const raw = message.toLowerCase();
-    // Common shape: "reset after 3h22m41s" (Gemini quota exhausted)
-    const afterMatch = raw.match(/reset after\s+([0-9a-z.\s]+)\.?/i);
-    if (afterMatch && afterMatch[1]) {
-        const parsed = parseDurationToMs(afterMatch[1]);
-        if (parsed && parsed > 0) {
-            return parsed;
-        }
-    }
-    // Sometimes the upstream JSON is embedded; try extracting quotaResetDelay.
-    const embeddedDelayMatch = raw.match(/quotaresetdelay"\s*:\s*"([^"]+)"/i);
-    if (embeddedDelayMatch && embeddedDelayMatch[1]) {
-        const parsed = parseDurationToMs(embeddedDelayMatch[1]);
-        if (parsed && parsed > 0) {
-            return parsed;
-        }
-    }
-    return null;
-}
-function parseDurationToMs(value) {
-    if (!value || typeof value !== 'string') {
-        return null;
-    }
-    const pattern = /(\d+(?:\.\d+)?)(ms|s|m|h)/gi;
-    let totalMs = 0;
-    let matched = false;
-    let match;
-    while ((match = pattern.exec(value)) !== null) {
-        matched = true;
-        const amount = Number.parseFloat(match[1]);
-        if (!Number.isFinite(amount)) {
-            continue;
-        }
-        const unit = match[2].toLowerCase();
-        if (unit === 'ms') {
-            totalMs += amount;
-        }
-        else if (unit === 'h') {
-            totalMs += amount * 3_600_000;
-        }
-        else if (unit === 'm') {
-            totalMs += amount * 60_000;
-        }
-        else if (unit === 's') {
-            totalMs += amount * 1_000;
-        }
-    }
-    if (!matched) {
-        return null;
-    }
-    if (totalMs <= 0) {
-        return null;
-    }
-    return Math.round(totalMs);
-}
-function normalizeLoadedQuotaState(providerKey, state) {
-    const key = typeof providerKey === 'string' && providerKey.trim() ? providerKey.trim() : state.providerKey;
-    const rawAuth = typeof state.authType === 'string'
-        ? String(state.authType).trim().toLowerCase()
-        : '';
-    const authType = rawAuth === 'apikey' ? 'apikey' : rawAuth === 'oauth' ? 'oauth' : 'unknown';
-    return {
-        ...state,
-        providerKey: key,
-        authType
-    };
-}
-function readPositiveNumberFromEnv(name, fallback) {
-    const raw = process.env[name];
-    if (!raw) {
-        return fallback;
-    }
-    const parsed = Number(raw);
-    if (!Number.isFinite(parsed) || parsed <= 0) {
-        return fallback;
-    }
-    return parsed;
-}
+export { QuotaManagerModule } from './antigravity-quota-manager.js';
+export { ProviderQuotaDaemonModule } from './provider-quota-daemon.js';
 //# sourceMappingURL=index.js.map