centaurus-cli 3.0.1 → 3.1.0

package/dist/services/session-quota-manager.js

@@ -1,242 +1,226 @@
-/**
- * Session Quota Manager
- *
- * Tracks and enforces rate limits based on backend configuration.
- * Each agent loop turn counts as one message.
- * Configuration is fetched from backend API on startup with fallback defaults.
- */
-import * as fs from 'fs';
-import * as path from 'path';
-import * as os from 'os';
-import { logError, logWarning, logInfo } from '../utils/logger.js';
-import { apiClient } from './api-client.js';
-// Default constants (fallback if backend is unreachable)
+import * as fs from "fs";
+import * as path from "path";
+import * as os from "os";
+import { logError, logWarning, logInfo } from "../utils/logger.js";
+import { apiClient } from "./api-client.js";
 let MAX_MESSAGES_PER_SESSION = 50;
-let SESSION_DURATION_MS = 4 * 60 * 60 * 1000; // 4 hours in milliseconds
+let SESSION_DURATION_MS = 4 * 60 * 60 * 1e3;
 let WARNING_THRESHOLD = 10;
-/**
- * SessionQuotaManager class for managing rate limits
- */
 class SessionQuotaManager {
-    quotaFilePath;
-    configFilePath;
-    state;
-    configLoaded = false;
-    constructor() {
-        const homeDir = os.homedir();
-        const centaurusDir = path.join(homeDir, '.centaurus');
-        this.quotaFilePath = path.join(centaurusDir, 'session-quota.json');
-        this.configFilePath = path.join(centaurusDir, 'rate-limits-config.json');
-        // Ensure directory exists
-        if (!fs.existsSync(centaurusDir)) {
-            fs.mkdirSync(centaurusDir, { recursive: true });
-        }
-        // Load cached config first (sync, for immediate use)
-        this.loadCachedConfig();
-        // Load or initialize state
-        this.state = this.loadState();
-    }
-    /**
-     * Load cached config from disk (synced on previous fetch)
-     */
-    loadCachedConfig() {
-        try {
-            if (fs.existsSync(this.configFilePath)) {
-                const data = fs.readFileSync(this.configFilePath, 'utf-8');
-                const config = JSON.parse(data);
-                this.applyConfig(config);
-                this.configLoaded = true;
-            }
-        }
-        catch (error) {
-            logWarning(`Failed to load cached rate limits config: ${error instanceof Error ? error.message : 'Unknown error'}`);
-        }
-    }
-    /**
-     * Apply configuration values
-     */
-    applyConfig(config) {
-        MAX_MESSAGES_PER_SESSION = config.maxMessagesPerSession;
-        SESSION_DURATION_MS = config.sessionDurationHours * 60 * 60 * 1000;
-        WARNING_THRESHOLD = config.warningThreshold || 10;
-    }
-    /**
-     * Save config to disk for caching
-     */
-    saveConfigCache(config) {
-        try {
-            fs.writeFileSync(this.configFilePath, JSON.stringify(config, null, 2), 'utf-8');
-        }
-        catch (error) {
-            logWarning(`Failed to cache rate limits config: ${error instanceof Error ? error.message : 'Unknown error'}`);
-        }
-    }
-    /**
-     * Fetch configuration from backend API
-     * Call this on CLI startup to get latest config
-     */
-    async fetchConfigFromBackend() {
-        try {
-            const config = await apiClient.getRateLimitsConfig();
-            if (config) {
-                this.applyConfig(config);
-                this.saveConfigCache(config);
-                this.configLoaded = true;
-                logInfo(`Rate limits config fetched from backend: ${MAX_MESSAGES_PER_SESSION} messages per ${SESSION_DURATION_MS / (60 * 60 * 1000)}h`);
-            }
-        }
-        catch (error) {
-            // Backend unreachable - use cached or default config
-            if (!this.configLoaded) {
-                logWarning(`Backend unreachable for rate limits config, using defaults: ${MAX_MESSAGES_PER_SESSION} messages per ${SESSION_DURATION_MS / (60 * 60 * 1000)}h`);
-            }
-        }
-    }
-    /**
-     * Load quota state from disk, or create fresh state if expired or missing
-     */
-    loadState() {
-        try {
-            if (fs.existsSync(this.quotaFilePath)) {
-                const data = fs.readFileSync(this.quotaFilePath, 'utf-8');
-                const state = JSON.parse(data);
-                // Check if session has expired
-                const now = Date.now();
-                const sessionAge = now - state.sessionStartTime;
-                if (sessionAge >= SESSION_DURATION_MS) {
-                    // Session expired, return fresh state (but don't save yet)
-                    return this.createFreshState();
-                }
-                return state;
-            }
-        }
-        catch (error) {
-            logWarning(`Failed to load session quota state: ${error instanceof Error ? error.message : 'Unknown error'}`);
-        }
-        return this.createFreshState();
-    }
-    /**
-     * Create a fresh session state
-     */
-    createFreshState() {
-        return {
-            sessionStartTime: 0, // Will be set on first message
-            messagesUsed: 0
-        };
-    }
-    /**
-     * Save current state to disk
-     */
-    saveState() {
-        try {
-            fs.writeFileSync(this.quotaFilePath, JSON.stringify(this.state, null, 2), 'utf-8');
-        }
-        catch (error) {
-            logError('Failed to save session quota state', error instanceof Error ? error : new Error(String(error)));
-        }
-    }
-    /**
-     * Check if session has expired and reset if needed
-     */
-    checkAndResetIfExpired() {
-        if (this.state.sessionStartTime === 0) {
-            // No session started yet
-            return;
-        }
+  quotaFilePath;
+  configFilePath;
+  state;
+  configLoaded = false;
+  currentPlan = "free";
+  constructor() {
+    const homeDir = os.homedir();
+    const centaurusDir = path.join(homeDir, ".centaurus");
+    this.quotaFilePath = path.join(centaurusDir, "session-quota.json");
+    this.configFilePath = path.join(centaurusDir, "rate-limits-config.json");
+    if (!fs.existsSync(centaurusDir)) {
+      fs.mkdirSync(centaurusDir, { recursive: true });
+    }
+    this.loadCachedConfig();
+    this.state = this.loadState();
+  }
+  /**
+   * Load cached config from disk (synced on previous fetch)
+   */
+  loadCachedConfig() {
+    try {
+      if (fs.existsSync(this.configFilePath)) {
+        const data = fs.readFileSync(this.configFilePath, "utf-8");
+        const config = JSON.parse(data);
+        this.applyConfig(config);
+        this.configLoaded = true;
+      }
+    } catch (error) {
+      logWarning(`Failed to load cached rate limits config: ${error instanceof Error ? error.message : "Unknown error"}`);
+    }
+  }
+  /**
+   * Apply configuration values
+   */
+  applyConfig(config) {
+    MAX_MESSAGES_PER_SESSION = config.maxMessagesPerSession;
+    SESSION_DURATION_MS = config.sessionDurationHours * 60 * 60 * 1e3;
+    WARNING_THRESHOLD = config.warningThreshold || 10;
+    if (config.defaultPlan) {
+      this.currentPlan = config.defaultPlan;
+    }
+  }
+  /**
+   * Save config to disk for caching
+   */
+  saveConfigCache(config) {
+    try {
+      fs.writeFileSync(this.configFilePath, JSON.stringify(config, null, 2), "utf-8");
+    } catch (error) {
+      logWarning(`Failed to cache rate limits config: ${error instanceof Error ? error.message : "Unknown error"}`);
+    }
+  }
+  /**
+   * Fetch configuration from backend API
+   * Call this on CLI startup to get latest config
+   */
+  async fetchConfigFromBackend() {
+    try {
+      const config = await apiClient.getRateLimitsConfig();
+      if (config) {
+        this.applyConfig(config);
+        this.saveConfigCache(config);
+        this.configLoaded = true;
+        logInfo(`Rate limits config fetched from backend: ${MAX_MESSAGES_PER_SESSION} messages per ${SESSION_DURATION_MS / (60 * 60 * 1e3)}h`);
+      }
+    } catch (error) {
+      if (!this.configLoaded) {
+        logWarning(`Backend unreachable for rate limits config, using defaults: ${MAX_MESSAGES_PER_SESSION} messages per ${SESSION_DURATION_MS / (60 * 60 * 1e3)}h`);
+      }
+    }
+  }
+  /**
+   * Load quota state from disk, or create fresh state if expired or missing
+   */
+  loadState() {
+    try {
+      if (fs.existsSync(this.quotaFilePath)) {
+        const data = fs.readFileSync(this.quotaFilePath, "utf-8");
+        const state = JSON.parse(data);
         const now = Date.now();
-        const sessionAge = now - this.state.sessionStartTime;
+        const sessionAge = now - state.sessionStartTime;
         if (sessionAge >= SESSION_DURATION_MS) {
-            // Session expired, reset state
-            this.state = this.createFreshState();
-            // Don't save yet - will be saved when first message is sent
-        }
-    }
-    /**
-     * Check if user can send a message (has remaining quota)
-     */
-    canSendMessage() {
-        this.checkAndResetIfExpired();
-        return this.state.messagesUsed < MAX_MESSAGES_PER_SESSION;
-    }
-    /**
-     * Increment the message count (call this on each agent loop turn)
-     */
-    incrementMessageCount() {
-        this.checkAndResetIfExpired();
-        // Start session on first message
-        if (this.state.sessionStartTime === 0) {
-            this.state.sessionStartTime = Date.now();
-        }
-        this.state.messagesUsed++;
-        this.saveState();
-    }
-    /**
-     * Get the number of remaining messages in the current session
-     */
-    getRemainingMessages() {
-        this.checkAndResetIfExpired();
-        return Math.max(0, MAX_MESSAGES_PER_SESSION - this.state.messagesUsed);
-    }
-    /**
-     * Get the total messages used in the current session
-     */
-    getMessagesUsed() {
-        this.checkAndResetIfExpired();
-        return this.state.messagesUsed;
-    }
-    /**
-     * Get time remaining until session resets (in milliseconds)
-     * Returns 0 if no session has started
-     */
-    getTimeUntilReset() {
-        if (this.state.sessionStartTime === 0) {
-            return 0;
-        }
-        const now = Date.now();
-        const sessionAge = now - this.state.sessionStartTime;
-        const timeRemaining = SESSION_DURATION_MS - sessionAge;
-        return Math.max(0, timeRemaining);
-    }
-    /**
-     * Get formatted time remaining as "Xh Ym" string
-     */
-    getFormattedTimeRemaining() {
-        const timeMs = this.getTimeUntilReset();
-        if (timeMs <= 0) {
-            return '0h 0m';
-        }
-        const totalMinutes = Math.ceil(timeMs / (60 * 1000));
-        const hours = Math.floor(totalMinutes / 60);
-        const minutes = totalMinutes % 60;
-        if (hours > 0) {
-            return `${hours}h ${minutes}m`;
-        }
-        return `${minutes}m`;
-    }
-    /**
-     * Get session start time (for debugging)
-     */
-    getSessionStartTime() {
-        return this.state.sessionStartTime;
-    }
-    /**
-     * Force reset the session (for testing purposes)
-     */
-    resetSession() {
-        this.state = this.createFreshState();
-        this.saveState();
-    }
-    /**
-     * Get current config values (for debugging)
-     */
-    getCurrentConfig() {
-        return {
-            maxMessagesPerSession: MAX_MESSAGES_PER_SESSION,
-            sessionDurationMs: SESSION_DURATION_MS,
-            warningThreshold: WARNING_THRESHOLD
-        };
-    }
+          return this.createFreshState();
+        }
+        return state;
+      }
+    } catch (error) {
+      logWarning(`Failed to load session quota state: ${error instanceof Error ? error.message : "Unknown error"}`);
+    }
+    return this.createFreshState();
+  }
+  /**
+   * Create a fresh session state
+   */
+  createFreshState() {
+    return {
+      sessionStartTime: 0,
+      // Will be set on first message
+      messagesUsed: 0
+    };
+  }
+  /**
+   * Save current state to disk
+   */
+  saveState() {
+    try {
+      fs.writeFileSync(this.quotaFilePath, JSON.stringify(this.state, null, 2), "utf-8");
+    } catch (error) {
+      logError("Failed to save session quota state", error instanceof Error ? error : new Error(String(error)));
+    }
+  }
+  /**
+   * Check if session has expired and reset if needed
+   */
+  checkAndResetIfExpired() {
+    if (this.state.sessionStartTime === 0) {
+      return;
+    }
+    const now = Date.now();
+    const sessionAge = now - this.state.sessionStartTime;
+    if (sessionAge >= SESSION_DURATION_MS) {
+      this.state = this.createFreshState();
+    }
+  }
+  /**
+   * Check if user can send a message (has remaining quota)
+   */
+  canSendMessage() {
+    this.checkAndResetIfExpired();
+    return this.state.messagesUsed < MAX_MESSAGES_PER_SESSION;
+  }
+  /**
+   * Increment the message count (call this on each agent loop turn)
+   */
+  incrementMessageCount() {
+    this.checkAndResetIfExpired();
+    if (this.state.sessionStartTime === 0) {
+      this.state.sessionStartTime = Date.now();
+    }
+    this.state.messagesUsed++;
+    this.saveState();
+  }
+  /**
+   * Get the number of remaining messages in the current session
+   */
+  getRemainingMessages() {
+    this.checkAndResetIfExpired();
+    return Math.max(0, MAX_MESSAGES_PER_SESSION - this.state.messagesUsed);
+  }
+  /**
+   * Get the total messages used in the current session
+   */
+  getMessagesUsed() {
+    this.checkAndResetIfExpired();
+    return this.state.messagesUsed;
+  }
+  /**
+   * Get time remaining until session resets (in milliseconds)
+   * Returns 0 if no session has started
+   */
+  getTimeUntilReset() {
+    if (this.state.sessionStartTime === 0) {
+      return 0;
+    }
+    const now = Date.now();
+    const sessionAge = now - this.state.sessionStartTime;
+    const timeRemaining = SESSION_DURATION_MS - sessionAge;
+    return Math.max(0, timeRemaining);
+  }
+  /**
+   * Get formatted time remaining as "Xh Ym" string
+   */
+  getFormattedTimeRemaining() {
+    const timeMs = this.getTimeUntilReset();
+    if (timeMs <= 0) {
+      return "0h 0m";
+    }
+    const totalMinutes = Math.ceil(timeMs / (60 * 1e3));
+    const hours = Math.floor(totalMinutes / 60);
+    const minutes = totalMinutes % 60;
+    if (hours > 0) {
+      return `${hours}h ${minutes}m`;
+    }
+    return `${minutes}m`;
+  }
+  /**
+   * Get session start time (for debugging)
+   */
+  getSessionStartTime() {
+    return this.state.sessionStartTime;
+  }
+  /**
+   * Force reset the session (for testing purposes)
+   */
+  resetSession() {
+    this.state = this.createFreshState();
+    this.saveState();
+  }
+  /**
+   * Get current config values (for debugging)
+   */
+  getCurrentConfig() {
+    return {
+      maxMessagesPerSession: MAX_MESSAGES_PER_SESSION,
+      sessionDurationMs: SESSION_DURATION_MS,
+      warningThreshold: WARNING_THRESHOLD
+    };
+  }
+  getPlan() {
+    return this.currentPlan;
+  }
 }
-// Export singleton instance
-export const sessionQuotaManager = new SessionQuotaManager();
+const sessionQuotaManager = new SessionQuotaManager();
+export {
+  sessionQuotaManager
+};
 //# sourceMappingURL=session-quota-manager.js.map

package/dist/services/session-quota-manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"session-quota-manager.js","sourceRoot":"","sources":["../../src/services/session-quota-manager.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAE5C,yDAAyD;AACzD,IAAI,wBAAwB,GAAG,EAAE,CAAC;AAClC,IAAI,mBAAmB,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,0BAA0B;AACxE,IAAI,iBAAiB,GAAG,EAAE,CAAC;AAoB3B;;GAEG;AACH,MAAM,mBAAmB;IACb,aAAa,CAAS;IACtB,cAAc,CAAS;IACvB,KAAK,CAAoB;IACzB,YAAY,GAAY,KAAK,CAAC;IAEtC;QACI,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAC7B,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,oBAAoB,CAAC,CAAC;QACnE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,yBAAyB,CAAC,CAAC;QAEzE,0BAA0B;QAC1B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;YAC/B,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACpD,CAAC;QAED,qDAAqD;QACrD,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAExB,2BAA2B;QAC3B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAED;;OAEG;IACK,gBAAgB;QACpB,IAAI,CAAC;YACD,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;gBACrC,MAAM,IAAI,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;gBAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAuB,CAAC;gBACtD,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBACzB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAC7B,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,UAAU,CAAC,6CAA6C,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QACxH,CAAC;IACL,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,MAA0B;QAC1C,wBAAwB,GAAG,MAAM,CAAC,qBAAqB,CAAC;QACxD,mBAAmB,GAAG,MAAM,CAAC,oBAAoB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QACnE,iBAAiB,GAAG,MAAM,CAAC,gBAAgB,IAAI,EAAE,CAAC;IACtD,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,MAA0B;QAC9C,IAAI,CAAC;YACD,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACpF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,UAAU,CAAC,uCAAuC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QAClH,CAAC;IACL,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,sBAAsB;QACxB,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,mBAAmB,EAAE,CAAC;YAErD,IAAI,MAAM,EAAE,CAAC;gBACT,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBACzB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;gBAC7B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;gBACzB,OAAO,CAAC,4CAA4C,wBAAwB,iBAAiB,mBAAmB,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5I,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,qDAAqD;YACrD,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;gBACrB,UAAU,CAAC,+DAA+D,wBAAwB,iBAAiB,mBAAmB,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;YAClK,CAAC;QACL,CAAC;IACL,CAAC;IAED;;OAEG;IACK,SAAS;QACb,IAAI,CAAC;YACD,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;gBACpC,MAAM,IAAI,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;gBAC1D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAsB,CAAC;gBAEpD,+BAA+B;gBAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACvB,MAAM,UAAU,GAAG,GAAG,GAAG,KAAK,CAAC,gBAAgB,CAAC;gBAEhD,IAAI,UAAU,IAAI,mBAAmB,EAAE,CAAC;oBACpC,2DAA2D;oBAC3D,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACnC,CAAC;gBAED,OAAO,KAAK,CAAC;YACjB,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,UAAU,CAAC,uCAAuC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QAClH,CAAC;QAED,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACK,gBAAgB;QACpB,OAAO;YACH,gBAAgB,EAAE,CAAC,EAAE,+BAA+B;YACpD,YAAY,EAAE,CAAC;SAClB,CAAC;IACN,CAAC;IAED;;OAEG;IACK,SAAS;QACb,IAAI,CAAC;YACD,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACvF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,QAAQ,CAAC,oCAAoC,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9G,CAAC;IACL,CAAC;IAED;;OAEG;IACK,sBAAsB;QAC1B,IAAI,IAAI,CAAC,KAAK,CAAC,gBAAgB,KAAK,CAAC,EAAE,CAAC;YACpC,yBAAyB;YACzB,OAAO;QACX,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,UAAU,GAAG,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC;QAErD,IAAI,UAAU,IAAI,mBAAmB,EAAE,CAAC;YACpC,+BAA+B;YAC/B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACrC,4DAA4D;QAChE,CAAC;IACL,CAAC;IAED;;OAEG;IACH,cAAc;QACV,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,wBAAwB,CAAC;IAC9D,CAAC;IAED;;OAEG;IACH,qBAAqB;QACjB,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAE9B,iCAAiC;QACjC,IAAI,IAAI,CAAC,KAAK,CAAC,gBAAgB,KAAK,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;QAC1B,IAAI,CAAC,SAAS,EAAE,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,oBAAoB;QAChB,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,wBAAwB,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC3E,CAAC;IAED;;OAEG;IACH,eAAe;QACX,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;IACnC,CAAC;IAED;;;OAGG;IACH,iBAAiB;QACb,IAAI,IAAI,CAAC,KAAK,CAAC,gBAAgB,KAAK,CAAC,EAAE,CAAC;YACpC,OAAO,CAAC,CAAC;QACb,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,UAAU,GAAG,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC;QACrD,MAAM,aAAa,GAAG,mBAAmB,GAAG,UAAU,CAAC;QAEvD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,yBAAyB;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAExC,IAAI,MAAM,IAAI,CAAC,EAAE,CAAC;YACd,OAAO,OAAO,CAAC;QACnB,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;QACrD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,EAAE,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,YAAY,GAAG,EAAE,CAAC;QAElC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACZ,OAAO,GAAG,KAAK,KAAK,OAAO,GAAG,CAAC;QACnC,CAAC;QACD,OAAO,GAAG,OAAO,GAAG,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,mBAAmB;QACf,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,YAAY;QACR,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACrC,IAAI,CAAC,SAAS,EAAE,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACZ,OAAO;YACH,qBAAqB,EAAE,wBAAwB;YAC/C,iBAAiB,EAAE,mBAAmB;YACtC,gBAAgB,EAAE,iBAAiB;SACtC,CAAC;IACN,CAAC;CACJ;AAED,4BAA4B;AAC5B,MAAM,CAAC,MAAM,mBAAmB,GAAG,IAAI,mBAAmB,EAAE,CAAC"}
+{"version":3,"sources":["../../src/services/session-quota-manager.ts"],"sourcesContent":["/**\r\n * Session Quota Manager\r\n * \r\n * Tracks and enforces rate limits based on backend configuration.\r\n * Each agent loop turn counts as one message.\r\n * Configuration is fetched from backend API on startup with fallback defaults.\r\n */\r\n\r\nimport * as fs from 'fs';\r\nimport * as path from 'path';\r\nimport * as os from 'os';\r\nimport { logError, logWarning, logInfo } from '../utils/logger.js';\r\nimport { apiClient } from './api-client.js';\r\n\r\n// Default constants (fallback if backend is unreachable)\r\nlet MAX_MESSAGES_PER_SESSION = 50;\r\nlet SESSION_DURATION_MS = 4 * 60 * 60 * 1000; // 4 hours in milliseconds\r\nlet WARNING_THRESHOLD = 10;\r\n\r\n/**\r\n * Session quota state stored on disk\r\n */\r\ninterface SessionQuotaState {\r\n    sessionStartTime: number;  // Unix timestamp (ms)\r\n    messagesUsed: number;      // Count of agent loop turns\r\n}\r\n\r\n/**\r\n * Backend config response\r\n */\r\ninterface SessionQuotaConfig {\r\n    maxMessagesPerSession: number;\r\n    sessionDurationHours: number;\r\n    warningThreshold: number;\r\n    defaultPlan: string;\r\n}\r\n\r\n/**\r\n * SessionQuotaManager class for managing rate limits\r\n */\r\nclass SessionQuotaManager {\r\n    private quotaFilePath: string;\r\n    private configFilePath: string;\r\n    private state: SessionQuotaState;\r\n    private configLoaded: boolean = false;\r\n    private currentPlan: string = 'free';\r\n\r\n    constructor() {\r\n        const homeDir = os.homedir();\r\n        const centaurusDir = path.join(homeDir, '.centaurus');\r\n        this.quotaFilePath = path.join(centaurusDir, 'session-quota.json');\r\n        this.configFilePath = path.join(centaurusDir, 'rate-limits-config.json');\r\n\r\n        // Ensure directory exists\r\n        if (!fs.existsSync(centaurusDir)) {\r\n            fs.mkdirSync(centaurusDir, { recursive: true });\r\n        }\r\n\r\n        // Load cached config first (sync, for immediate use)\r\n        this.loadCachedConfig();\r\n\r\n        // Load or initialize state\r\n        this.state = this.loadState();\r\n    }\r\n\r\n    /**\r\n     * Load cached config from disk (synced on previous fetch)\r\n     */\r\n    private loadCachedConfig(): void {\r\n        try {\r\n            if (fs.existsSync(this.configFilePath)) {\r\n                const data = fs.readFileSync(this.configFilePath, 'utf-8');\r\n                const config = JSON.parse(data) as SessionQuotaConfig;\r\n                this.applyConfig(config);\r\n                this.configLoaded = true;\r\n            }\r\n        } catch (error) {\r\n            logWarning(`Failed to load cached rate limits config: ${error instanceof Error ? error.message : 'Unknown error'}`);\r\n        }\r\n    }\r\n\r\n    /**\r\n     * Apply configuration values\r\n     */\r\n    private applyConfig(config: SessionQuotaConfig): void {\r\n        MAX_MESSAGES_PER_SESSION = config.maxMessagesPerSession;\r\n        SESSION_DURATION_MS = config.sessionDurationHours * 60 * 60 * 1000;\r\n        WARNING_THRESHOLD = config.warningThreshold || 10;\r\n        if (config.defaultPlan) {\r\n            this.currentPlan = config.defaultPlan;\r\n        }\r\n    }\r\n\r\n    /**\r\n     * Save config to disk for caching\r\n     */\r\n    private saveConfigCache(config: SessionQuotaConfig): void {\r\n        try {\r\n            fs.writeFileSync(this.configFilePath, JSON.stringify(config, null, 2), 'utf-8');\r\n        } catch (error) {\r\n            logWarning(`Failed to cache rate limits config: ${error instanceof Error ? error.message : 'Unknown error'}`);\r\n        }\r\n    }\r\n\r\n    /**\r\n     * Fetch configuration from backend API\r\n     * Call this on CLI startup to get latest config\r\n     */\r\n    async fetchConfigFromBackend(): Promise<void> {\r\n        try {\r\n            const config = await apiClient.getRateLimitsConfig();\r\n\r\n            if (config) {\r\n                this.applyConfig(config);\r\n                this.saveConfigCache(config);\r\n                this.configLoaded = true;\r\n                logInfo(`Rate limits config fetched from backend: ${MAX_MESSAGES_PER_SESSION} messages per ${SESSION_DURATION_MS / (60 * 60 * 1000)}h`);\r\n            }\r\n        } catch (error) {\r\n            // Backend unreachable - use cached or default config\r\n            if (!this.configLoaded) {\r\n                logWarning(`Backend unreachable for rate limits config, using defaults: ${MAX_MESSAGES_PER_SESSION} messages per ${SESSION_DURATION_MS / (60 * 60 * 1000)}h`);\r\n            }\r\n        }\r\n    }\r\n\r\n    /**\r\n     * Load quota state from disk, or create fresh state if expired or missing\r\n     */\r\n    private loadState(): SessionQuotaState {\r\n        try {\r\n            if (fs.existsSync(this.quotaFilePath)) {\r\n                const data = fs.readFileSync(this.quotaFilePath, 'utf-8');\r\n                const state = JSON.parse(data) as SessionQuotaState;\r\n\r\n                // Check if session has expired\r\n                const now = Date.now();\r\n                const sessionAge = now - state.sessionStartTime;\r\n\r\n                if (sessionAge >= SESSION_DURATION_MS) {\r\n                    // Session expired, return fresh state (but don't save yet)\r\n                    return this.createFreshState();\r\n                }\r\n\r\n                return state;\r\n            }\r\n        } catch (error) {\r\n            logWarning(`Failed to load session quota state: ${error instanceof Error ? error.message : 'Unknown error'}`);\r\n        }\r\n\r\n        return this.createFreshState();\r\n    }\r\n\r\n    /**\r\n     * Create a fresh session state\r\n     */\r\n    private createFreshState(): SessionQuotaState {\r\n        return {\r\n            sessionStartTime: 0, // Will be set on first message\r\n            messagesUsed: 0\r\n        };\r\n    }\r\n\r\n    /**\r\n     * Save current state to disk\r\n     */\r\n    private saveState(): void {\r\n        try {\r\n            fs.writeFileSync(this.quotaFilePath, JSON.stringify(this.state, null, 2), 'utf-8');\r\n        } catch (error) {\r\n            logError('Failed to save session quota state', error instanceof Error ? error : new Error(String(error)));\r\n        }\r\n    }\r\n\r\n    /**\r\n     * Check if session has expired and reset if needed\r\n     */\r\n    private checkAndResetIfExpired(): void {\r\n        if (this.state.sessionStartTime === 0) {\r\n            // No session started yet\r\n            return;\r\n        }\r\n\r\n        const now = Date.now();\r\n        const sessionAge = now - this.state.sessionStartTime;\r\n\r\n        if (sessionAge >= SESSION_DURATION_MS) {\r\n            // Session expired, reset state\r\n            this.state = this.createFreshState();\r\n            // Don't save yet - will be saved when first message is sent\r\n        }\r\n    }\r\n\r\n    /**\r\n     * Check if user can send a message (has remaining quota)\r\n     */\r\n    canSendMessage(): boolean {\r\n        this.checkAndResetIfExpired();\r\n        return this.state.messagesUsed < MAX_MESSAGES_PER_SESSION;\r\n    }\r\n\r\n    /**\r\n     * Increment the message count (call this on each agent loop turn)\r\n     */\r\n    incrementMessageCount(): void {\r\n        this.checkAndResetIfExpired();\r\n\r\n        // Start session on first message\r\n        if (this.state.sessionStartTime === 0) {\r\n            this.state.sessionStartTime = Date.now();\r\n        }\r\n\r\n        this.state.messagesUsed++;\r\n        this.saveState();\r\n    }\r\n\r\n    /**\r\n     * Get the number of remaining messages in the current session\r\n     */\r\n    getRemainingMessages(): number {\r\n        this.checkAndResetIfExpired();\r\n        return Math.max(0, MAX_MESSAGES_PER_SESSION - this.state.messagesUsed);\r\n    }\r\n\r\n    /**\r\n     * Get the total messages used in the current session\r\n     */\r\n    getMessagesUsed(): number {\r\n        this.checkAndResetIfExpired();\r\n        return this.state.messagesUsed;\r\n    }\r\n\r\n    /**\r\n     * Get time remaining until session resets (in milliseconds)\r\n     * Returns 0 if no session has started\r\n     */\r\n    getTimeUntilReset(): number {\r\n        if (this.state.sessionStartTime === 0) {\r\n            return 0;\r\n        }\r\n\r\n        const now = Date.now();\r\n        const sessionAge = now - this.state.sessionStartTime;\r\n        const timeRemaining = SESSION_DURATION_MS - sessionAge;\r\n\r\n        return Math.max(0, timeRemaining);\r\n    }\r\n\r\n    /**\r\n     * Get formatted time remaining as \"Xh Ym\" string\r\n     */\r\n    getFormattedTimeRemaining(): string {\r\n        const timeMs = this.getTimeUntilReset();\r\n\r\n        if (timeMs <= 0) {\r\n            return '0h 0m';\r\n        }\r\n\r\n        const totalMinutes = Math.ceil(timeMs / (60 * 1000));\r\n        const hours = Math.floor(totalMinutes / 60);\r\n        const minutes = totalMinutes % 60;\r\n\r\n        if (hours > 0) {\r\n            return `${hours}h ${minutes}m`;\r\n        }\r\n        return `${minutes}m`;\r\n    }\r\n\r\n    /**\r\n     * Get session start time (for debugging)\r\n     */\r\n    getSessionStartTime(): number {\r\n        return this.state.sessionStartTime;\r\n    }\r\n\r\n    /**\r\n     * Force reset the session (for testing purposes)\r\n     */\r\n    resetSession(): void {\r\n        this.state = this.createFreshState();\r\n        this.saveState();\r\n    }\r\n\r\n    /**\r\n     * Get current config values (for debugging)\r\n     */\r\n    getCurrentConfig() {\r\n        return {\r\n            maxMessagesPerSession: MAX_MESSAGES_PER_SESSION,\r\n            sessionDurationMs: SESSION_DURATION_MS,\r\n            warningThreshold: WARNING_THRESHOLD\r\n        };\r\n    }\r\n\r\n    getPlan(): string {\r\n        return this.currentPlan;\r\n    }\r\n}\r\n\r\n// Export singleton instance\r\nexport const sessionQuotaManager = new SessionQuotaManager();\r\n\r\n// Export types for use elsewhere\r\nexport { SessionQuotaState };\r\n"],"mappings":"AAQA,YAAY,QAAQ;AACpB,YAAY,UAAU;AACtB,YAAY,QAAQ;AACpB,SAAS,UAAU,YAAY,eAAe;AAC9C,SAAS,iBAAiB;AAG1B,IAAI,2BAA2B;AAC/B,IAAI,sBAAsB,IAAI,KAAK,KAAK;AACxC,IAAI,oBAAoB;AAuBxB,MAAM,oBAAoB;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EACA,eAAwB;AAAA,EACxB,cAAsB;AAAA,EAE9B,cAAc;AACV,UAAM,UAAU,GAAG,QAAQ;AAC3B,UAAM,eAAe,KAAK,KAAK,SAAS,YAAY;AACpD,SAAK,gBAAgB,KAAK,KAAK,cAAc,oBAAoB;AACjE,SAAK,iBAAiB,KAAK,KAAK,cAAc,yBAAyB;AAGvE,QAAI,CAAC,GAAG,WAAW,YAAY,GAAG;AAC9B,SAAG,UAAU,cAAc,EAAE,WAAW,KAAK,CAAC;AAAA,IAClD;AAGA,SAAK,iBAAiB;AAGtB,SAAK,QAAQ,KAAK,UAAU;AAAA,EAChC;AAAA;AAAA;AAAA;AAAA,EAKQ,mBAAyB;AAC7B,QAAI;AACA,UAAI,GAAG,WAAW,KAAK,cAAc,GAAG;AACpC,cAAM,OAAO,GAAG,aAAa,KAAK,gBAAgB,OAAO;AACzD,cAAM,SAAS,KAAK,MAAM,IAAI;AAC9B,aAAK,YAAY,MAAM;AACvB,aAAK,eAAe;AAAA,MACxB;AAAA,IACJ,SAAS,OAAO;AACZ,iBAAW,6CAA6C,iBAAiB,QAAQ,MAAM,UAAU,eAAe,EAAE;AAAA,IACtH;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKQ,YAAY,QAAkC;AAClD,+BAA2B,OAAO;AAClC,0BAAsB,OAAO,uBAAuB,KAAK,KAAK;AAC9D,wBAAoB,OAAO,oBAAoB;AAC/C,QAAI,OAAO,aAAa;AACpB,WAAK,cAAc,OAAO;AAAA,IAC9B;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKQ,gBAAgB,QAAkC;AACtD,QAAI;AACA,SAAG,cAAc,KAAK,gBAAgB,KAAK,UAAU,QAAQ,MAAM,CAAC,GAAG,OAAO;AAAA,IAClF,SAAS,OAAO;AACZ,iBAAW,uCAAuC,iBAAiB,QAAQ,MAAM,UAAU,eAAe,EAAE;AAAA,IAChH;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,yBAAwC;AAC1C,QAAI;AACA,YAAM,SAAS,MAAM,UAAU,oBAAoB;AAEnD,UAAI,QAAQ;AACR,aAAK,YAAY,MAAM;AACvB,aAAK,gBAAgB,MAAM;AAC3B,aAAK,eAAe;AACpB,gBAAQ,4CAA4C,wBAAwB,iBAAiB,uBAAuB,KAAK,KAAK,IAAK,GAAG;AAAA,MAC1I;AAAA,IACJ,SAAS,OAAO;AAEZ,UAAI,CAAC,KAAK,cAAc;AACpB,mBAAW,+DAA+D,wBAAwB,iBAAiB,uBAAuB,KAAK,KAAK,IAAK,GAAG;AAAA,MAChK;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKQ,YAA+B;AACnC,QAAI;AACA,UAAI,GAAG,WAAW,KAAK,aAAa,GAAG;AACnC,cAAM,OAAO,GAAG,aAAa,KAAK,eAAe,OAAO;AACxD,cAAM,QAAQ,KAAK,MAAM,IAAI;AAG7B,cAAM,MAAM,KAAK,IAAI;AACrB,cAAM,aAAa,MAAM,MAAM;AAE/B,YAAI,cAAc,qBAAqB;AAEnC,iBAAO,KAAK,iBAAiB;AAAA,QACjC;AAEA,eAAO;AAAA,MACX;AAAA,IACJ,SAAS,OAAO;AACZ,iBAAW,uCAAuC,iBAAiB,QAAQ,MAAM,UAAU,eAAe,EAAE;AAAA,IAChH;AAEA,WAAO,KAAK,iBAAiB;AAAA,EACjC;AAAA;AAAA;AAAA;AAAA,EAKQ,mBAAsC;AAC1C,WAAO;AAAA,MACH,kBAAkB;AAAA;AAAA,MAClB,cAAc;AAAA,IAClB;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKQ,YAAkB;AACtB,QAAI;AACA,SAAG,cAAc,KAAK,eAAe,KAAK,UAAU,KAAK,OAAO,MAAM,CAAC,GAAG,OAAO;AAAA,IACrF,SAAS,OAAO;AACZ,eAAS,sCAAsC,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAAC;AAAA,IAC5G;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKQ,yBAA+B;AACnC,QAAI,KAAK,MAAM,qBAAqB,GAAG;AAEnC;AAAA,IACJ;AAEA,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,aAAa,MAAM,KAAK,MAAM;AAEpC,QAAI,cAAc,qBAAqB;AAEnC,WAAK,QAAQ,KAAK,iBAAiB;AAAA,IAEvC;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,iBAA0B;AACtB,SAAK,uBAAuB;AAC5B,WAAO,KAAK,MAAM,eAAe;AAAA,EACrC;AAAA;AAAA;AAAA;AAAA,EAKA,wBAA8B;AAC1B,SAAK,uBAAuB;AAG5B,QAAI,KAAK,MAAM,qBAAqB,GAAG;AACnC,WAAK,MAAM,mBAAmB,KAAK,IAAI;AAAA,IAC3C;AAEA,SAAK,MAAM;AACX,SAAK,UAAU;AAAA,EACnB;AAAA;AAAA;AAAA;AAAA,EAKA,uBAA+B;AAC3B,SAAK,uBAAuB;AAC5B,WAAO,KAAK,IAAI,GAAG,2BAA2B,KAAK,MAAM,YAAY;AAAA,EACzE;AAAA;AAAA;AAAA;AAAA,EAKA,kBAA0B;AACtB,SAAK,uBAAuB;AAC5B,WAAO,KAAK,MAAM;AAAA,EACtB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,oBAA4B;AACxB,QAAI,KAAK,MAAM,qBAAqB,GAAG;AACnC,aAAO;AAAA,IACX;AAEA,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,aAAa,MAAM,KAAK,MAAM;AACpC,UAAM,gBAAgB,sBAAsB;AAE5C,WAAO,KAAK,IAAI,GAAG,aAAa;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA,EAKA,4BAAoC;AAChC,UAAM,SAAS,KAAK,kBAAkB;AAEtC,QAAI,UAAU,GAAG;AACb,aAAO;AAAA,IACX;AAEA,UAAM,eAAe,KAAK,KAAK,UAAU,KAAK,IAAK;AACnD,UAAM,QAAQ,KAAK,MAAM,eAAe,EAAE;AAC1C,UAAM,UAAU,eAAe;AAE/B,QAAI,QAAQ,GAAG;AACX,aAAO,GAAG,KAAK,KAAK,OAAO;AAAA,IAC/B;AACA,WAAO,GAAG,OAAO;AAAA,EACrB;AAAA;AAAA;AAAA;AAAA,EAKA,sBAA8B;AAC1B,WAAO,KAAK,MAAM;AAAA,EACtB;AAAA;AAAA;AAAA;AAAA,EAKA,eAAqB;AACjB,SAAK,QAAQ,KAAK,iBAAiB;AACnC,SAAK,UAAU;AAAA,EACnB;AAAA;AAAA;AAAA;AAAA,EAKA,mBAAmB;AACf,WAAO;AAAA,MACH,uBAAuB;AAAA,MACvB,mBAAmB;AAAA,MACnB,kBAAkB;AAAA,IACtB;AAAA,EACJ;AAAA,EAEA,UAAkB;AACd,WAAO,KAAK;AAAA,EAChB;AACJ;AAGO,MAAM,sBAAsB,IAAI,oBAAoB;","names":[]}