opencode-quotas 0.0.1

package/dist/src/services/history-service.js

@@ -0,0 +1,99 @@
+import { writeFile, readFile } from "node:fs/promises";
+import { existsSync, mkdirSync } from "node:fs";
+import { join } from "node:path";
+import { HISTORY_FILE } from "../utils/paths";
+import { logger } from "../logger";
+export class HistoryService {
+    historyPath;
+    data = {};
+    maxWindowMs = 24 * 60 * 60 * 1000; // Keep 24 hours of history
+    saveTimeout = null;
+    constructor(customPath) {
+        this.historyPath = customPath || HISTORY_FILE();
+    }
+    async init() {
+        try {
+            const dir = join(this.historyPath, "..");
+            if (!existsSync(dir)) {
+                mkdirSync(dir, { recursive: true });
+            }
+            if (existsSync(this.historyPath)) {
+                const raw = await readFile(this.historyPath, "utf-8");
+                this.data = JSON.parse(raw);
+            }
+        }
+        catch (e) {
+            logger.error("history-service:init_failed", { path: this.historyPath, error: e });
+            this.data = {};
+        }
+    }
+    async append(snapshot) {
+        const timestamp = Date.now();
+        let changed = false;
+        for (const quota of snapshot) {
+            if (!this.data[quota.id]) {
+                this.data[quota.id] = [];
+            }
+            this.data[quota.id].push({
+                timestamp,
+                used: quota.used,
+                limit: quota.limit
+            });
+            // Prune old data for this quota
+            const cutoff = timestamp - this.maxWindowMs;
+            const originalLength = this.data[quota.id].length;
+            this.data[quota.id] = this.data[quota.id].filter(p => p.timestamp >= cutoff);
+            if (this.data[quota.id].length !== originalLength || originalLength > 0) {
+                changed = true;
+            }
+        }
+        if (changed || snapshot.length > 0) {
+            this.save();
+        }
+    }
+    getHistory(quotaId, windowMs) {
+        const now = Date.now();
+        const cutoff = now - windowMs;
+        const history = this.data[quotaId] || [];
+        return history.filter(p => p.timestamp >= cutoff);
+    }
+    setMaxAge(hours) {
+        this.maxWindowMs = hours * 60 * 60 * 1000;
+    }
+    async pruneAll() {
+        const now = Date.now();
+        const cutoff = now - this.maxWindowMs;
+        let changed = false;
+        for (const id in this.data) {
+            const originalLen = this.data[id].length;
+            this.data[id] = this.data[id].filter(p => p.timestamp >= cutoff);
+            if (this.data[id].length !== originalLen) {
+                changed = true;
+            }
+            // Remove key if empty to free memory
+            if (this.data[id].length === 0) {
+                delete this.data[id];
+                changed = true;
+            }
+        }
+        if (changed) {
+            this.save();
+        }
+    }
+    save() {
+        if (this.saveTimeout) {
+            clearTimeout(this.saveTimeout);
+        }
+        this.saveTimeout = setTimeout(async () => {
+            try {
+                await writeFile(this.historyPath, JSON.stringify(this.data, null, 2), "utf-8");
+                logger.debug("history-service:save_success", { path: this.historyPath });
+            }
+            catch (e) {
+                logger.error("history-service:save_failed", { path: this.historyPath, error: e });
+            }
+            this.saveTimeout = null;
+        }, 5000);
+    }
+}
+//# sourceMappingURL=history-service.js.map

package/dist/src/services/history-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"history-service.js","sourceRoot":"","sources":["../../../src/services/history-service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAChD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE9C,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAEnC,MAAM,OAAO,cAAc;IACf,WAAW,CAAS;IACpB,IAAI,GAAmC,EAAE,CAAC;IAC1C,WAAW,GAAW,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,2BAA2B;IACtE,WAAW,GAAyC,IAAI,CAAC;IAEjE,YAAY,UAAmB;QAC3B,IAAI,CAAC,WAAW,GAAG,UAAU,IAAI,YAAY,EAAE,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,IAAI;QACN,IAAI,CAAC;YACD,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;YACzC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACnB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACxC,CAAC;YAED,IAAI,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC/B,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;gBACtD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAChC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAClF,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;QACnB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAqB;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,OAAO,GAAG,KAAK,CAAC;QAEpB,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;gBACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC;YAC7B,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC;gBACrB,SAAS;gBACT,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,KAAK,EAAE,KAAK,CAAC,KAAK;aACrB,CAAC,CAAC;YAEH,gCAAgC;YAChC,MAAM,MAAM,GAAG,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC;YAC5C,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC;YAClD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,MAAM,CAAC,CAAC;YAE7E,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,MAAM,KAAK,cAAc,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;gBACtE,OAAO,GAAG,IAAI,CAAC;YACnB,CAAC;QACL,CAAC;QAED,IAAI,OAAO,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,IAAI,EAAE,CAAC;QAChB,CAAC;IACL,CAAC;IAED,UAAU,CAAC,OAAe,EAAE,QAAgB;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QACzC,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,SAAS,CAAC,KAAa;QACnB,IAAI,CAAC,WAAW,GAAG,KAAK,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC;QACtC,IAAI,OAAO,GAAG,KAAK,CAAC;QAEpB,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACzB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC;YACzC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,MAAM,CAAC,CAAC;YAEjE,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACvC,OAAO,GAAG,IAAI,CAAC;YACnB,CAAC;YAED,qCAAqC;YACrC,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACrB,OAAO,GAAG,IAAI,CAAC;YACnB,CAAC;QACL,CAAC;QAED,IAAI,OAAO,EAAE,CAAC;YACV,IAAI,CAAC,IAAI,EAAE,CAAC;QAChB,CAAC;IACL,CAAC;IAEO,IAAI;QACR,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;YACrC,IAAI,CAAC;gBACD,MAAM,SAAS,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBAC/E,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YAC7E,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACtF,CAAC;YACD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC5B,CAAC,EAAE,IAAI,CAAC,CAAC;IACb,CAAC;CACJ"}

package/dist/src/services/prediction-engine.d.ts

@@ -0,0 +1,47 @@
+import { type IPredictionEngine, type IHistoryService, type HistoryPoint } from "../interfaces";
+/**
+ * Configuration options for the prediction engine.
+ */
+export interface PredictionEngineConfig {
+    /**
+     * Default short time window for regression to capture spikes (minutes).
+     * Defaults to 5.
+     */
+    predictionShortWindowMinutes?: number;
+    /**
+     * Idle timeout in milliseconds. If the last data point is older than this,
+     * the prediction returns Infinity. Defaults to 5 minutes.
+     */
+    idleTimeoutMs?: number;
+}
+/**
+ * Prediction engine using dual-window linear regression.
+ *
+ * This implementation uses two time windows to calculate usage slopes:
+ * - Long window: Captures overall trend (default: 60 minutes)
+ * - Short window: Captures recent spikes (default: 5 minutes)
+ *
+ * The maximum of the two slopes is used for conservative estimation.
+ */
+export declare class LinearRegressionPredictionEngine implements IPredictionEngine {
+    private readonly historyService;
+    private readonly config;
+    constructor(historyService: IHistoryService, config?: PredictionEngineConfig);
+    /**
+     * Predicts time to limit in milliseconds using a dual-window linear regression approach.
+     * Returns Infinity if usage is decreasing, stable, or idle.
+     */
+    predictTimeToLimit(quotaId: string, windowMinutes?: number, shortWindowMinutes?: number): number;
+    /**
+     * Calculates the slope (usage per ms) using linear regression for the given history points.
+     */
+    calculateSlope(history: HistoryPoint[]): number;
+}
+/**
+ * A null prediction engine that always returns Infinity.
+ * Used when no history service is available.
+ */
+export declare class NullPredictionEngine implements IPredictionEngine {
+    predictTimeToLimit(_quotaId: string, _windowMinutes?: number, _shortWindowMinutes?: number): number;
+}
+//# sourceMappingURL=prediction-engine.d.ts.map

package/dist/src/services/prediction-engine.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prediction-engine.d.ts","sourceRoot":"","sources":["../../../src/services/prediction-engine.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,iBAAiB,EAAE,KAAK,eAAe,EAAE,KAAK,YAAY,EAAE,MAAM,eAAe,CAAC;AAEhG;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACnC;;;OAGG;IACH,4BAA4B,CAAC,EAAE,MAAM,CAAC;IAEtC;;;OAGG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED;;;;;;;;GAQG;AACH,qBAAa,gCAAiC,YAAW,iBAAiB;IACtE,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAkB;IACjD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAmC;gBAE9C,cAAc,EAAE,eAAe,EAAE,MAAM,CAAC,EAAE,sBAAsB;IAQ5E;;;OAGG;IACH,kBAAkB,CACd,OAAO,EAAE,MAAM,EACf,aAAa,GAAE,MAAW,EAC1B,kBAAkB,CAAC,EAAE,MAAM,GAC5B,MAAM;IA8CT;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,MAAM;CAqBlD;AAED;;;GAGG;AACH,qBAAa,oBAAqB,YAAW,iBAAiB;IAC1D,kBAAkB,CACd,QAAQ,EAAE,MAAM,EAChB,cAAc,GAAE,MAAW,EAC3B,mBAAmB,CAAC,EAAE,MAAM,GAC7B,MAAM;CAGZ"}

package/dist/src/services/prediction-engine.js

@@ -0,0 +1,94 @@
+/**
+ * Prediction engine using dual-window linear regression.
+ *
+ * This implementation uses two time windows to calculate usage slopes:
+ * - Long window: Captures overall trend (default: 60 minutes)
+ * - Short window: Captures recent spikes (default: 5 minutes)
+ *
+ * The maximum of the two slopes is used for conservative estimation.
+ */
+export class LinearRegressionPredictionEngine {
+    historyService;
+    config;
+    constructor(historyService, config) {
+        this.historyService = historyService;
+        this.config = {
+            predictionShortWindowMinutes: config?.predictionShortWindowMinutes ?? 5,
+            idleTimeoutMs: config?.idleTimeoutMs ?? 5 * 60 * 1000,
+        };
+    }
+    /**
+     * Predicts time to limit in milliseconds using a dual-window linear regression approach.
+     * Returns Infinity if usage is decreasing, stable, or idle.
+     */
+    predictTimeToLimit(quotaId, windowMinutes = 60, shortWindowMinutes) {
+        const longWindowMs = windowMinutes * 60 * 1000;
+        const shortWindowMin = shortWindowMinutes ?? this.config.predictionShortWindowMinutes;
+        const shortWindowMs = shortWindowMin * 60 * 1000;
+        const history = this.historyService.getHistory(quotaId, longWindowMs);
+        if (history.length < 2)
+            return Infinity;
+        // Idle Handling: If the last history point is older than the idle timeout, 
+        // assume usage has stopped.
+        const lastPoint = history[history.length - 1];
+        const now = Date.now();
+        if (now - lastPoint.timestamp > this.config.idleTimeoutMs) {
+            return Infinity;
+        }
+        // Long Slope
+        const mLong = this.calculateSlope(history);
+        // Short Slope: most recent data in short window or last 15% of points
+        const shortHistory = history.filter(p => p.timestamp > now - shortWindowMs);
+        // Ensure we have enough points in short history, or take the last 15%
+        let effectiveShortHistory = shortHistory;
+        if (effectiveShortHistory.length < 2) {
+            const fifteenPercentCount = Math.max(2, Math.ceil(history.length * 0.15));
+            effectiveShortHistory = history.slice(-fifteenPercentCount);
+        }
+        const mShort = this.calculateSlope(effectiveShortHistory);
+        // Conservative Estimation: use the maximum slope
+        const m = Math.max(mLong, mShort);
+        if (m <= 0)
+            return Infinity;
+        if (lastPoint.limit === null)
+            return Infinity;
+        const remaining = lastPoint.limit - lastPoint.used;
+        if (remaining <= 0)
+            return 0;
+        const msFromLastPoint = remaining / m;
+        const elapsedSinceLastPoint = now - lastPoint.timestamp;
+        return Math.max(0, msFromLastPoint - elapsedSinceLastPoint);
+    }
+    /**
+     * Calculates the slope (usage per ms) using linear regression for the given history points.
+     */
+    calculateSlope(history) {
+        if (history.length < 2)
+            return 0;
+        let sumX = 0, sumY = 0, sumXY = 0, sumX2 = 0;
+        const n = history.length;
+        const firstTimestamp = history[0].timestamp;
+        for (const p of history) {
+            const x = p.timestamp - firstTimestamp;
+            const y = p.used;
+            sumX += x;
+            sumY += y;
+            sumXY += x * y;
+            sumX2 += x * x;
+        }
+        const denominator = (n * sumX2 - sumX * sumX);
+        if (denominator === 0)
+            return 0;
+        return (n * sumXY - sumX * sumY) / denominator;
+    }
+}
+/**
+ * A null prediction engine that always returns Infinity.
+ * Used when no history service is available.
+ */
+export class NullPredictionEngine {
+    predictTimeToLimit(_quotaId, _windowMinutes = 60, _shortWindowMinutes) {
+        return Infinity;
+    }
+}
+//# sourceMappingURL=prediction-engine.js.map

package/dist/src/services/prediction-engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prediction-engine.js","sourceRoot":"","sources":["../../../src/services/prediction-engine.ts"],"names":[],"mappings":"AAmBA;;;;;;;;GAQG;AACH,MAAM,OAAO,gCAAgC;IACxB,cAAc,CAAkB;IAChC,MAAM,CAAmC;IAE1D,YAAY,cAA+B,EAAE,MAA+B;QACxE,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,MAAM,GAAG;YACV,4BAA4B,EAAE,MAAM,EAAE,4BAA4B,IAAI,CAAC;YACvE,aAAa,EAAE,MAAM,EAAE,aAAa,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI;SACxD,CAAC;IACN,CAAC;IAED;;;OAGG;IACH,kBAAkB,CACd,OAAe,EACf,gBAAwB,EAAE,EAC1B,kBAA2B;QAE3B,MAAM,YAAY,GAAG,aAAa,GAAG,EAAE,GAAG,IAAI,CAAC;QAC/C,MAAM,cAAc,GAAG,kBAAkB,IAAI,IAAI,CAAC,MAAM,CAAC,4BAA4B,CAAC;QACtF,MAAM,aAAa,GAAG,cAAc,GAAG,EAAE,GAAG,IAAI,CAAC;QAEjD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QACtE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,QAAQ,CAAC;QAExC,4EAA4E;QAC5E,4BAA4B;QAC5B,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,GAAG,GAAG,SAAS,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YACxD,OAAO,QAAQ,CAAC;QACpB,CAAC;QAED,aAAa;QACb,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAE3C,sEAAsE;QACtE,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,GAAG,GAAG,aAAa,CAAC,CAAC;QAE5E,sEAAsE;QACtE,IAAI,qBAAqB,GAAG,YAAY,CAAC;QACzC,IAAI,qBAAqB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,MAAM,mBAAmB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC;YAC1E,qBAAqB,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,mBAAmB,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,qBAAqB,CAAC,CAAC;QAE1D,iDAAiD;QACjD,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAElC,IAAI,CAAC,IAAI,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC5B,IAAI,SAAS,CAAC,KAAK,KAAK,IAAI;YAAE,OAAO,QAAQ,CAAC;QAE9C,MAAM,SAAS,GAAG,SAAS,CAAC,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC;QACnD,IAAI,SAAS,IAAI,CAAC;YAAE,OAAO,CAAC,CAAC;QAE7B,MAAM,eAAe,GAAG,SAAS,GAAG,CAAC,CAAC;QACtC,MAAM,qBAAqB,GAAG,GAAG,GAAG,SAAS,CAAC,SAAS,CAAC;QAExD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,eAAe,GAAG,qBAAqB,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,OAAuB;QAClC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,CAAC,CAAC;QAEjC,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;QAC7C,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QACzB,MAAM,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAE5C,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACtB,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,GAAG,cAAc,CAAC;YACvC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;YACjB,IAAI,IAAI,CAAC,CAAC;YACV,IAAI,IAAI,CAAC,CAAC;YACV,KAAK,IAAI,CAAC,GAAG,CAAC,CAAC;YACf,KAAK,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;QAED,MAAM,WAAW,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC;QAC9C,IAAI,WAAW,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAEhC,OAAO,CAAC,CAAC,GAAG,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC,GAAG,WAAW,CAAC;IACnD,CAAC;CACJ;AAED;;;GAGG;AACH,MAAM,OAAO,oBAAoB;IAC7B,kBAAkB,CACd,QAAgB,EAChB,iBAAyB,EAAE,EAC3B,mBAA4B;QAE5B,OAAO,QAAQ,CAAC;IACpB,CAAC;CACJ"}

package/dist/src/services/quota-service.d.ts

@@ -0,0 +1,41 @@
+import { type QuotaConfig, type QuotaData, type IQuotaProvider, type IHistoryService, type IPredictionEngine, type IAggregationService } from "../interfaces";
+export declare class QuotaService {
+    private config;
+    private initialized;
+    private initPromise;
+    private historyService?;
+    private predictionEngine;
+    private aggregationService;
+    constructor(initialConfig?: Partial<QuotaConfig>);
+    init(directory: string, historyService?: IHistoryService): Promise<void>;
+    private registerProviders;
+    getConfig(): QuotaConfig;
+    getProviders(): IQuotaProvider[];
+    /**
+     * Returns the prediction engine used by this service.
+     * Useful for testing or for other services that need prediction capabilities.
+     */
+    getPredictionEngine(): IPredictionEngine;
+    /**
+     * Returns the aggregation service used by this service.
+     * Useful for testing or for other services that need aggregation capabilities.
+     */
+    getAggregationService(): IAggregationService;
+    getQuotas(context?: {
+        providerId?: string;
+        modelId?: string;
+    }): Promise<QuotaData[]>;
+    processQuotas(data: QuotaData[], context?: {
+        providerId?: string;
+        modelId?: string;
+    }): QuotaData[];
+    private applyAggregation;
+    /**
+     * Resolves which quotas belong to an AggregatedGroup using explicit sources and patterns.
+     */
+    private resolveGroupSources;
+    private filterQuotas;
+    private filterByModel;
+    private sortQuotas;
+}
+//# sourceMappingURL=quota-service.d.ts.map

package/dist/src/services/quota-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-service.d.ts","sourceRoot":"","sources":["../../../src/services/quota-service.ts"],"names":[],"mappings":"AACA,OAAO,EACH,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,cAAc,EACnB,KAAK,eAAe,EACpB,KAAK,iBAAiB,EACtB,KAAK,mBAAmB,EAE3B,MAAM,eAAe,CAAC;AAUvB,qBAAa,YAAY;IACrB,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,WAAW,CAA8B;IACjD,OAAO,CAAC,cAAc,CAAC,CAAkB;IACzC,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,kBAAkB,CAAsB;gBAEpC,aAAa,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC;IAQ1C,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,cAAc,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;YAiChE,iBAAiB;IA0B/B,SAAS,IAAI,WAAW;IAIxB,YAAY,IAAI,cAAc,EAAE;IAIhC;;;OAGG;IACH,mBAAmB,IAAI,iBAAiB;IAIxC;;;OAGG;IACH,qBAAqB,IAAI,mBAAmB;IAItC,SAAS,CAAC,OAAO,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IA2C1F,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,OAAO,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,SAAS,EAAE;IA2BlG,OAAO,CAAC,gBAAgB;IA4DxB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IA6C3B,OAAO,CAAC,YAAY;IAepB,OAAO,CAAC,aAAa;IA4BrB,OAAO,CAAC,UAAU;CAGrB"}

package/dist/src/services/quota-service.js

@@ -0,0 +1,257 @@
+import { getQuotaRegistry } from "../registry";
+import { createAntigravityProvider } from "../providers/antigravity";
+import { createCodexProvider } from "../providers/codex";
+import { formatDurationMs } from "../utils/time";
+import { logger } from "../logger";
+import { LinearRegressionPredictionEngine, NullPredictionEngine } from "./prediction-engine";
+import { AggregationService } from "./aggregation-service";
+import { ConfigLoader } from "./config-loader";
+export class QuotaService {
+    config;
+    initialized = false;
+    initPromise = null;
+    historyService;
+    predictionEngine;
+    aggregationService;
+    constructor(initialConfig) {
+        this.config = ConfigLoader.createConfig(initialConfig);
+        // Initialize with null prediction engine until init() is called
+        this.predictionEngine = new NullPredictionEngine();
+        this.aggregationService = new AggregationService(this.predictionEngine);
+    }
+    async init(directory, historyService) {
+        if (this.initialized)
+            return;
+        if (this.initPromise)
+            return this.initPromise;
+        this.initPromise = (async () => {
+            this.historyService = historyService;
+            // Load config from disk
+            this.config = await ConfigLoader.loadFromDisk(directory, this.config);
+            if (this.historyService && this.config.historyMaxAgeHours !== undefined) {
+                this.historyService.setMaxAge(this.config.historyMaxAgeHours);
+            }
+            // Initialize prediction engine with history service
+            if (this.historyService) {
+                this.predictionEngine = new LinearRegressionPredictionEngine(this.historyService, { predictionShortWindowMinutes: this.config.predictionShortWindowMinutes });
+            }
+            // Re-initialize aggregation service with the new prediction engine
+            this.aggregationService = new AggregationService(this.predictionEngine);
+            // Register providers
+            await this.registerProviders();
+            this.initialized = true;
+        })();
+        return this.initPromise;
+    }
+    async registerProviders() {
+        const registry = getQuotaRegistry();
+        // Register Antigravity
+        try {
+            registry.register(createAntigravityProvider({
+                debug: !!this.config.debug,
+            }));
+            logger.debug("init:provider_registered", { id: "antigravity" });
+        }
+        catch (e) {
+            logger.error("init:provider_failed", { id: "antigravity", error: e });
+            console.warn("[QuotaService] Failed to initialize Antigravity provider:", e);
+        }
+        // Register Codex
+        try {
+            registry.register(createCodexProvider());
+            logger.debug("init:provider_registered", { id: "codex" });
+        }
+        catch (e) {
+            logger.error("init:provider_failed", { id: "codex", error: e });
+            console.warn("[QuotaService] Failed to initialize Codex provider:", e);
+        }
+    }
+    getConfig() {
+        return this.config;
+    }
+    getProviders() {
+        return getQuotaRegistry().getAll();
+    }
+    /**
+     * Returns the prediction engine used by this service.
+     * Useful for testing or for other services that need prediction capabilities.
+     */
+    getPredictionEngine() {
+        return this.predictionEngine;
+    }
+    /**
+     * Returns the aggregation service used by this service.
+     * Useful for testing or for other services that need aggregation capabilities.
+     */
+    getAggregationService() {
+        return this.aggregationService;
+    }
+    async getQuotas(context) {
+        const providers = this.getProviders();
+        logger.debug("quota_service:get_quotas_start", { providerCount: providers.length, ids: providers.map((p) => p.id) });
+        if (providers.length === 0)
+            return [];
+        const results = await Promise.all(providers.map(async (p) => {
+            const startedAt = Date.now();
+            try {
+                logger.debug("quota_service:provider_fetch_start", { id: p.id });
+                const result = await p.fetchQuota();
+                logger.debug("quota_service:provider_fetch_ok", { id: p.id, count: result.length, durationMs: Date.now() - startedAt });
+                return result;
+            }
+            catch (e) {
+                logger.error("quota_service:provider_fetch_error", { id: p.id, durationMs: Date.now() - startedAt, error: e });
+                console.error(`Provider ${p.id} failed:`, e);
+                return [];
+            }
+        }));
+        const processed = this.processQuotas(results.flat(), context);
+        logger.debug("quota_service:get_quotas_end", { totalCount: processed.length });
+        return processed;
+    }
+    processQuotas(data, context) {
+        let results = [...data];
+        // 1. Enrich with predictions (before aggregation so sources have it too)
+        results = results.map(q => {
+            const time = this.predictionEngine.predictTimeToLimit(q.id, 60);
+            if (time !== Infinity) {
+                return {
+                    ...q,
+                    predictedReset: `${formatDurationMs(time)} (predicted)`
+                };
+            }
+            return q;
+        });
+        // 2. Apply Aggregation
+        results = this.applyAggregation(results);
+        // 3. Filter (Disabled & Model Mapping). If requested, perform model-strict filtering.
+        results = this.filterQuotas(results, context);
+        // 4. Sort
+        results = this.sortQuotas(results);
+        return results;
+    }
+    applyAggregation(quotas) {
+        if (!this.config.aggregatedGroups || this.config.aggregatedGroups.length === 0) {
+            return quotas;
+        }
+        const aggregatedResults = [];
+        let remainingQuotas = [...quotas];
+        for (const group of this.config.aggregatedGroups) {
+            // Resolve source quotas from explicit sources and patterns
+            const sourceQuotas = this.resolveGroupSources(remainingQuotas, group);
+            if (sourceQuotas.length === 0)
+                continue;
+            const strategy = group.strategy || "most_critical";
+            let representative = null;
+            if (strategy === "most_critical") {
+                representative = this.aggregationService.aggregateMostCritical(sourceQuotas, group.predictionWindowMinutes, group.predictionShortWindowMinutes);
+            }
+            else if (strategy === "max") {
+                representative = this.aggregationService.aggregateMax(sourceQuotas);
+            }
+            else if (strategy === "min") {
+                representative = this.aggregationService.aggregateMin(sourceQuotas);
+            }
+            else if (strategy === "mean" || strategy === "median") {
+                representative = this.aggregationService.aggregateAverage(sourceQuotas, group.name, group.id, strategy);
+            }
+            if (representative) {
+                // Create a copy for display
+                const displayQuota = {
+                    ...representative,
+                    id: group.id,
+                    providerName: group.name
+                };
+                // Remove matched sources from pool to avoid double aggregation
+                const sourceIds = new Set(sourceQuotas.map(q => q.id));
+                remainingQuotas = remainingQuotas.filter(q => !sourceIds.has(q.id));
+                aggregatedResults.push(displayQuota);
+            }
+        }
+        // Return aggregated results. 
+        // If showUnaggregated is false, only return what matched a group.
+        if (this.config.showUnaggregated === false) {
+            return aggregatedResults;
+        }
+        return [...remainingQuotas, ...aggregatedResults];
+    }
+    /**
+     * Resolves which quotas belong to an AggregatedGroup using explicit sources and patterns.
+     */
+    resolveGroupSources(quotas, group) {
+        const matched = [];
+        const matchedIds = new Set();
+        // 1. Explicit sources (highest priority)
+        if (group.sources && group.sources.length > 0) {
+            for (const quota of quotas) {
+                if (group.sources.includes(quota.id)) {
+                    matched.push(quota);
+                    matchedIds.add(quota.id);
+                }
+            }
+        }
+        // 2. Pattern matching
+        if (group.patterns && group.patterns.length > 0) {
+            for (const quota of quotas) {
+                // Skip if already matched by explicit source
+                if (matchedIds.has(quota.id))
+                    continue;
+                // Filter by providerId if specified
+                if (group.providerId) {
+                    const providerMatch = quota.providerName.toLowerCase().includes(group.providerId.toLowerCase()) ||
+                        quota.id.toLowerCase().startsWith(group.providerId.toLowerCase());
+                    if (!providerMatch)
+                        continue;
+                }
+                // Check if any pattern matches
+                const matchTarget = `${quota.id} ${quota.providerName}`.toLowerCase();
+                const patternMatches = group.patterns.some(pattern => {
+                    const lowerPattern = pattern.toLowerCase();
+                    return matchTarget.includes(lowerPattern);
+                });
+                if (patternMatches) {
+                    matched.push(quota);
+                    matchedIds.add(quota.id);
+                }
+            }
+        }
+        return matched;
+    }
+    filterQuotas(quotas, context) {
+        let results = [...quotas];
+        // Filter out disabled quotas
+        const disabledIds = new Set(this.config.disabled || []);
+        results = results.filter((data) => !disabledIds.has(data.id));
+        // If requested, apply model-aware filtering
+        if (this.config.filterByCurrentModel && context && context.providerId && context.modelId) {
+            return this.filterByModel(results, context.providerId, context.modelId);
+        }
+        return results;
+    }
+    filterByModel(quotas, providerId, modelId) {
+        const providerLower = providerId.toLowerCase();
+        const modelIdLower = modelId.toLowerCase();
+        // Fuzzy token match with scoring
+        const tokens = modelIdLower.split(/[^a-z0-9]+/).filter(Boolean);
+        const scoredMatches = quotas
+            .map(q => {
+            const id = q.id.toLowerCase();
+            const name = q.providerName.toLowerCase();
+            const score = tokens.reduce((acc, t) => acc + (id.includes(t) || name.includes(t) ? 1 : 0), 0);
+            return { q, score };
+        })
+            .filter(m => m.score > 0)
+            .sort((a, b) => b.score - a.score);
+        if (scoredMatches.length > 0) {
+            const maxScore = scoredMatches[0].score;
+            return scoredMatches.filter(m => m.score === maxScore).map(m => m.q);
+        }
+        // 3) Provider fallback
+        const matchesProvider = quotas.filter(q => q.providerName.toLowerCase().includes(providerLower));
+        return matchesProvider.length > 0 ? matchesProvider : [];
+    }
+    sortQuotas(quotas) {
+        return quotas.sort((a, b) => a.providerName.localeCompare(b.providerName));
+    }
+}
+//# sourceMappingURL=quota-service.js.map

package/dist/src/services/quota-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-service.js","sourceRoot":"","sources":["../../../src/services/quota-service.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACnC,OAAO,EAAE,gCAAgC,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC7F,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,OAAO,YAAY;IACb,MAAM,CAAc;IACpB,WAAW,GAAY,KAAK,CAAC;IAC7B,WAAW,GAAyB,IAAI,CAAC;IACzC,cAAc,CAAmB;IACjC,gBAAgB,CAAoB;IACpC,kBAAkB,CAAsB;IAEhD,YAAY,aAAoC;QAC5C,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;QAEvD,gEAAgE;QAChE,IAAI,CAAC,gBAAgB,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACnD,IAAI,CAAC,kBAAkB,GAAG,IAAI,kBAAkB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC5E,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB,EAAE,cAAgC;QAC1D,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAC7B,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO,IAAI,CAAC,WAAW,CAAC;QAE9C,IAAI,CAAC,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;YAC3B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;YAErC,wBAAwB;YACxB,IAAI,CAAC,MAAM,GAAG,MAAM,YAAY,CAAC,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAEtE,IAAI,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACtE,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;YAClE,CAAC;YAED,oDAAoD;YACpD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,IAAI,CAAC,gBAAgB,GAAG,IAAI,gCAAgC,CACxD,IAAI,CAAC,cAAc,EACnB,EAAE,4BAA4B,EAAE,IAAI,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAC7E,CAAC;YACN,CAAC;YACD,mEAAmE;YACnE,IAAI,CAAC,kBAAkB,GAAG,IAAI,kBAAkB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAExE,qBAAqB;YACrB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAE/B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC5B,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC3B,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;QAEpC,uBAAuB;QACvB,IAAI,CAAC;YACD,QAAQ,CAAC,QAAQ,CACb,yBAAyB,CAAC;gBACtB,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK;aAC7B,CAAC,CACL,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,CAAC;QACpE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,2DAA2D,EAAE,CAAC,CAAC,CAAC;QACjF,CAAC;QAED,iBAAiB;QACjB,IAAI,CAAC;YACD,QAAQ,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,MAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAChE,OAAO,CAAC,IAAI,CAAC,qDAAqD,EAAE,CAAC,CAAC,CAAC;QAC3E,CAAC;IACL,CAAC;IAED,SAAS;QACL,OAAO,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,YAAY;QACR,OAAO,gBAAgB,EAAE,CAAC,MAAM,EAAE,CAAC;IACvC,CAAC;IAED;;;OAGG;IACH,mBAAmB;QACf,OAAO,IAAI,CAAC,gBAAgB,CAAC;IACjC,CAAC;IAED;;;OAGG;IACH,qBAAqB;QACjB,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,OAAmD;QAC/D,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAEtC,MAAM,CAAC,KAAK,CACR,gCAAgC,EAChC,EAAE,aAAa,EAAE,SAAS,CAAC,MAAM,EAAE,GAAG,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CACvE,CAAC;QAEF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEtC,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC7B,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,CAAiB,EAAE,EAAE;YACtC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACD,MAAM,CAAC,KAAK,CACR,oCAAoC,EACpC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CACf,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,UAAU,EAAE,CAAC;gBACpC,MAAM,CAAC,KAAK,CACR,iCAAiC,EACjC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,CACzE,CAAC;gBACF,OAAO,MAAM,CAAC;YAClB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,MAAM,CAAC,KAAK,CACR,oCAAoC,EACpC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,CAC7D,CAAC;gBACF,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;gBAC7C,OAAO,EAAE,CAAC;YACd,CAAC;QACL,CAAC,CAAC,CACL,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAC9D,MAAM,CAAC,KAAK,CACR,8BAA8B,EAC9B,EAAE,UAAU,EAAE,SAAS,CAAC,MAAM,EAAE,CACnC,CAAC;QACF,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,aAAa,CAAC,IAAiB,EAAE,OAAmD;QAChF,IAAI,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QAExB,yEAAyE;QACzE,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;YACtB,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YAChE,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpB,OAAO;oBACH,GAAG,CAAC;oBACJ,cAAc,EAAE,GAAG,gBAAgB,CAAC,IAAI,CAAC,cAAc;iBAC1D,CAAC;YACN,CAAC;YACD,OAAO,CAAC,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,uBAAuB;QACvB,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAEzC,sFAAsF;QACtF,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAE9C,UAAU;QACV,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAEnC,OAAO,OAAO,CAAC;IACnB,CAAC;IAEO,gBAAgB,CAAC,MAAmB;QACxC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7E,OAAO,MAAM,CAAC;QAClB,CAAC;QAED,MAAM,iBAAiB,GAAgB,EAAE,CAAC;QAC1C,IAAI,eAAe,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;QAElC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC/C,2DAA2D;YAC3D,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;YACtE,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAExC,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,IAAI,eAAe,CAAC;YACnD,IAAI,cAAc,GAAqB,IAAI,CAAC;YAE5C,IAAI,QAAQ,KAAK,eAAe,EAAE,CAAC;gBAC/B,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,qBAAqB,CAC1D,YAAY,EACZ,KAAK,CAAC,uBAAuB,EAC7B,KAAK,CAAC,4BAA4B,CACrC,CAAC;YACN,CAAC;iBAAM,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBAC5B,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;YACxE,CAAC;iBAAM,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBAC5B,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;YACxE,CAAC;iBAAM,IAAI,QAAQ,KAAK,MAAM,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACtD,cAAc,GAAG,IAAI,CAAC,kBAAkB,CAAC,gBAAgB,CACrD,YAAY,EACZ,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,EAAE,EACR,QAAQ,CACX,CAAC;YACN,CAAC;YAED,IAAI,cAAc,EAAE,CAAC;gBACjB,4BAA4B;gBAC5B,MAAM,YAAY,GAAG;oBACjB,GAAG,cAAc;oBACjB,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,YAAY,EAAE,KAAK,CAAC,IAAI;iBAC3B,CAAC;gBAEF,+DAA+D;gBAC/D,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACvD,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBAEpE,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzC,CAAC;QACL,CAAC;QAED,8BAA8B;QAC9B,kEAAkE;QAClE,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,KAAK,KAAK,EAAE,CAAC;YACzC,OAAO,iBAAiB,CAAC;QAC7B,CAAC;QAED,OAAO,CAAC,GAAG,eAAe,EAAE,GAAG,iBAAiB,CAAC,CAAC;IACtD,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,MAAmB,EAAE,KAAsB;QACnE,MAAM,OAAO,GAAgB,EAAE,CAAC;QAChC,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;QAErC,yCAAyC;QACzC,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACzB,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;oBACnC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACpB,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC7B,CAAC;YACL,CAAC;QACL,CAAC;QAED,sBAAsB;QACtB,IAAI,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACzB,6CAA6C;gBAC7C,IAAI,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;oBAAE,SAAS;gBAEvC,oCAAoC;gBACpC,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACnB,MAAM,aAAa,GACf,KAAK,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;wBACzE,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC;oBACtE,IAAI,CAAC,aAAa;wBAAE,SAAS;gBACjC,CAAC;gBAED,+BAA+B;gBAC/B,MAAM,WAAW,GAAG,GAAG,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC,WAAW,EAAE,CAAC;gBACtE,MAAM,cAAc,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;oBACjD,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;oBAC3C,OAAO,WAAW,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBAC9C,CAAC,CAAC,CAAC;gBAEH,IAAI,cAAc,EAAE,CAAC;oBACjB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACpB,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC7B,CAAC;YACL,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;IAEO,YAAY,CAAC,MAAmB,EAAE,OAAmD;QACzF,IAAI,OAAO,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;QAE1B,6BAA6B;QAC7B,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;QACxD,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAE9D,4CAA4C;QAC5C,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,OAAO,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACvF,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAC5E,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;IAEO,aAAa,CAAC,MAAmB,EAAE,UAAkB,EAAE,OAAe;QAC1E,MAAM,aAAa,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;QAC/C,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAE3C,iCAAiC;QACjC,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAChE,MAAM,aAAa,GAAG,MAAM;aACvB,GAAG,CAAC,CAAC,CAAC,EAAE;YACL,MAAM,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;YAC9B,MAAM,IAAI,GAAG,CAAC,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC;YAC1C,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/F,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC;QACxB,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC;aACxB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAEvC,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;YACxC,OAAO,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACzE,CAAC;QAED,uBAAuB;QACvB,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CACtC,CAAC,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CACvD,CAAC;QACF,OAAO,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC;IAC7D,CAAC;IAEO,UAAU,CAAC,MAAmB;QAClC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;IAC/E,CAAC;CACJ"}

package/dist/src/tools/quotas.d.ts

@@ -0,0 +1,11 @@
+import { type ToolDefinition } from "@opencode-ai/plugin";
+import { type QuotaService } from "../services/quota-service";
+import { type QuotaConfig } from "../interfaces";
+/**
+ * Creates a tool definition for fetching and displaying quota information.
+ * This tool always performs a fresh fetch of quotas from all providers.
+ *
+ * @returns A ToolDefinition that can be registered under Hooks.tool
+ */
+export declare function createQuotaTool(quotaService: QuotaService, getConfig: () => QuotaConfig): ToolDefinition;
+//# sourceMappingURL=quotas.d.ts.map

package/dist/src/tools/quotas.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quotas.d.ts","sourceRoot":"","sources":["../../../src/tools/quotas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAQ,KAAK,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAChE,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EAAE,KAAK,WAAW,EAAE,MAAM,eAAe,CAAC;AAIjD;;;;;GAKG;AACH,wBAAgB,eAAe,CAC3B,YAAY,EAAE,YAAY,EAC1B,SAAS,EAAE,MAAM,WAAW,GAC7B,cAAc,CA8DhB"}