vibecodingmachine-core 1.0.0 → 1.0.1

package/src/ide-integration/quota-detector.js

@@ -1,349 +1,349 @@
-// @vibecodingmachine/core - Quota Detector
-// Handles quota detection for different IDEs using CDP and AppleScript
-
-import CDP from 'chrome-remote-interface';
-import { AppleScriptManager } from './applescript-manager.js';
-
-/**
- * Quota Detector for IDE interactions
- * Detects quota warnings in different IDEs using CDP and AppleScript
- */
-export class QuotaDetector {
-  constructor() {
-    this.logger = console;
-    this.appleScriptManager = new AppleScriptManager();
-  }
-
-  /**
-   * Timeout utility function
-   * @param {number} ms - Timeout in milliseconds
-   * @param {Promise} promise - Promise to timeout
-   * @returns {Promise} Promise with timeout
-   */
-  timeout(ms, promise) {
-    return Promise.race([
-      promise,
-      new Promise((_, reject) =>
-        setTimeout(() => reject(new Error(`Operation timed out after ${ms}ms`)), ms)
-      )
-    ]);
-  }
-
-  /**
-   * Detect quota warnings in an IDE
-   * @param {string} ide - The IDE name ('vscode', 'cursor', 'windsurf')
-   * @returns {Promise<Object>} Quota detection result
-   */
-  async detectQuotaWarning(ide = 'vscode') {
-    this.logger.log(`🔍 Starting quota detection for ${ide}...`);
-
-    // Determine the port based on the IDE
-    const port = ide === 'windsurf' ? 9224 : ide === 'cursor' ? 9225 : 9222;
-    const ideName = ide === 'windsurf' ? 'Windsurf' : ide === 'cursor' ? 'Cursor' : 'VS Code';
-
-    this.logger.log(`🔌 Checking ${ideName} on port ${port}...`);
-
-    // For Windsurf, use AppleScript detection since CDP is not supported
-    if (ide === 'windsurf') {
-      this.logger.log('🔍 Windsurf detected - using AppleScript quota detection');
-      return await this.appleScriptManager.detectQuotaWarning(ide);
-    }
-
-    // For VS Code and Cursor, use CDP detection
-    try {
-      this.logger.log(`🔍 Attempting CDP connection to ${ideName} on port ${port}...`);
-
-      // Try to connect to CDP
-      const targets = await this.timeout(5000, CDP.List({ port }));
-
-      if (!targets || targets.length === 0) {
-        this.logger.log(`❌ No CDP targets found on port ${port}`);
-
-        // For Windsurf, fall back to AppleScript
-        if (ide === 'windsurf') {
-          this.logger.log('🔍 Falling back to enhanced AppleScript quota detection...');
-          return await this.appleScriptManager.detectQuotaWarning(ide);
-        }
-
-        return {
-          hasQuotaWarning: false,
-          error: `Could not find ${ideName}. Make sure ${ideName} is running with --remote-debugging-port=${port}`,
-          note: 'CDP connection failed'
-        };
-      }
-
-      this.logger.log(`✅ Found ${targets.length} CDP targets`);
-
-      // For Cursor, prefer the workspace target over settings
-      let workbench;
-      if (ide === 'cursor') {
-        workbench = targets.find(t => t.title !== 'Cursor Settings' && t.type === 'page') ||
-                    targets.find(t => t.url && t.url.includes('workbench')) ||
-                    targets[0];
-      } else {
-        workbench = targets.find(t => t.url && t.url.includes('workbench')) || targets[0];
-      }
-
-      if (!workbench) {
-        this.logger.log(`❌ No suitable workbench target found`);
-        return {
-          hasQuotaWarning: false,
-          error: `No ${ideName} workbench target found.`,
-          note: 'No workbench target available'
-        };
-      }
-
-      this.logger.log(`✅ Selected workbench: ${workbench.title}`);
-
-      // Connect to CDP
-      const client = await this.timeout(10000, CDP({ port, target: workbench }));
-      const { Runtime, Page } = client;
-
-      await this.timeout(5000, Runtime.enable());
-      if (Page && Page.bringToFront) {
-        try {
-          await this.timeout(3000, Page.bringToFront());
-        } catch (error) {
-          this.logger.log(`⚠️ Bring to front failed: ${error.message}`);
-        }
-      }
-
-      // Execute JavaScript to detect quota warnings
-      const detectionScript = `
-        (() => {
-          console.log('🔍 Starting quota detection in ${ideName}...');
-
-          // Common quota warning patterns
-          const quotaPatterns = [
-            'not enough credits',
-            'upgrade to a paid plan',
-            'switch to swe-1-lite',
-            'quota exceeded',
-            'credits exhausted',
-            'payment required',
-            'out of credits',
-            'insufficient credits',
-            'billing required',
-            'subscription needed'
-          ];
-
-          // Function to check if text contains quota warnings
-          function containsQuotaWarning(text) {
-            if (!text) return false;
-            const lowerText = text.toLowerCase();
-            return quotaPatterns.some(pattern => lowerText.includes(pattern));
-          }
-
-          // Method 1: Check all text content in the document
-          const allText = document.body.innerText || document.body.textContent || '';
-          if (containsQuotaWarning(allText)) {
-            console.log('❌ Quota warning detected in document text');
-            return {
-              hasQuotaWarning: true,
-              method: 'document-text',
-              matchedText: allText.substring(0, 200) + '...'
-            };
-          }
-
-          // Method 2: Check specific elements that might contain quota warnings
-          const quotaElements = [
-            // VS Code specific selectors
-            '.monaco-workbench .part.auxiliarybar',
-            '.monaco-workbench .part.sidebar.right',
-            '.monaco-workbench .chat-view',
-            '.monaco-workbench .chat-input',
-
-            // Cursor specific selectors
-            '.aislash-editor-input',
-            '.aislash-chat-container',
-            '.aislash-panel',
-
-            // Generic selectors
-            '[data-testid*="chat"]',
-            '[aria-label*="chat"]',
-            '[class*="chat"]',
-            '[class*="quota"]',
-            '[class*="credits"]',
-            '[class*="billing"]'
-          ];
-
-          for (const selector of quotaElements) {
-            const elements = document.querySelectorAll(selector);
-            for (const element of elements) {
-              const elementText = element.innerText || element.textContent || '';
-              if (containsQuotaWarning(elementText)) {
-                console.log('❌ Quota warning detected in element:', selector);
-                return {
-                  hasQuotaWarning: true,
-                  method: 'element-specific',
-                  selector: selector,
-                  matchedText: elementText.substring(0, 200) + '...'
-                };
-              }
-            }
-          }
-
-          // Method 3: Check for disabled input fields (common when quota is exceeded)
-          const disabledInputs = document.querySelectorAll('input[disabled], textarea[disabled], [contenteditable="true"][aria-disabled="true"]');
-          if (disabledInputs.length > 0) {
-            console.log('⚠️ Found disabled input fields, checking for quota context...');
-
-            // Check if there are quota-related messages near disabled inputs
-            for (const input of disabledInputs) {
-              const parentText = input.parentElement?.innerText || '';
-              if (containsQuotaWarning(parentText)) {
-                console.log('❌ Quota warning detected near disabled input');
-                return {
-                  hasQuotaWarning: true,
-                  method: 'disabled-input-context',
-                  matchedText: parentText.substring(0, 200) + '...'
-                };
-              }
-            }
-          }
-
-          // Method 4: Check for specific UI elements that indicate quota issues
-          const quotaButtons = document.querySelectorAll('button, a, [role="button"]');
-          for (const button of quotaButtons) {
-            const buttonText = button.innerText || button.textContent || '';
-            if (containsQuotaWarning(buttonText)) {
-              console.log('❌ Quota warning detected in button:', buttonText);
-              return {
-                hasQuotaWarning: true,
-                method: 'button-text',
-                matchedText: buttonText
-              };
-            }
-          }
-
-          // Method 5: Check for error messages or notifications
-          const errorElements = document.querySelectorAll('[class*="error"], [class*="warning"], [class*="notification"], [role="alert"]');
-          for (const error of errorElements) {
-            const errorText = error.innerText || error.textContent || '';
-            if (containsQuotaWarning(errorText)) {
-              console.log('❌ Quota warning detected in error element');
-              return {
-                hasQuotaWarning: true,
-                method: 'error-element',
-                matchedText: errorText.substring(0, 200) + '...'
-              };
-            }
-          }
-
-          console.log('✅ No quota warnings detected');
-          return {
-            hasQuotaWarning: false,
-            method: 'cdp-comprehensive',
-            note: 'No quota warnings found in ${ideName}'
-          };
-        })()
-      `;
-
-      const { result } = await this.timeout(10000, Runtime.evaluate({
-        expression: detectionScript,
-        returnByValue: true
-      }));
-
-      const detectionResult = result.value || result.unserializableValue;
-
-      this.logger.log('🔍 CDP quota detection result:', detectionResult);
-
-      if (detectionResult && detectionResult.hasQuotaWarning) {
-        this.logger.log('❌ Quota warning detected via CDP');
-        return {
-          hasQuotaWarning: true,
-          method: detectionResult.method,
-          note: detectionResult.matchedText || 'Quota warning detected via CDP',
-          debug: detectionResult
-        };
-      }
-
-      this.logger.log('✅ No quota warnings detected via CDP');
-      return {
-        hasQuotaWarning: false,
-        method: 'cdp-comprehensive',
-        note: 'No quota warnings found in ' + ideName,
-        debug: detectionResult
-      };
-
-    } catch (error) {
-      this.logger.log(`❌ CDP quota detection failed: ${error.message}`);
-
-      // For Windsurf, fall back to AppleScript
-      if (ide === 'windsurf') {
-        this.logger.log('🔍 Falling back to AppleScript quota detection...');
-        return await this.appleScriptManager.detectQuotaWarning(ide);
-      }
-
-      return {
-        hasQuotaWarning: false,
-        error: error.message,
-        note: `CDP quota detection failed for ${ideName}`
-      };
-    }
-  }
-
-  /**
-   * Get available IDEs without quota warnings
-   * @param {Array<string>} ides - Array of IDE names to check
-   * @returns {Promise<Array<string>>} Array of available IDE names
-   */
-  async getAvailableIdes(ides = ['vscode', 'cursor', 'windsurf']) {
-    this.logger.log('🔍 Checking available IDEs without quota warnings...');
-
-    const availableIdes = [];
-
-    for (const ide of ides) {
-      try {
-        const quotaResult = await this.detectQuotaWarning(ide);
-
-        if (!quotaResult.hasQuotaWarning) {
-          availableIdes.push(ide);
-          this.logger.log(`✅ ${ide} is available (no quota warnings)`);
-        } else {
-          this.logger.log(`❌ ${ide} has quota warnings: ${quotaResult.note || 'Unknown quota issue'}`);
-        }
-      } catch (error) {
-        this.logger.log(`⚠️ Error checking ${ide}: ${error.message}`);
-        // If we can't check quota, assume the IDE is available
-        availableIdes.push(ide);
-      }
-    }
-
-    this.logger.log(`📋 Available IDEs: ${availableIdes.join(', ')}`);
-    return availableIdes;
-  }
-
-  /**
-   * Test quota detection with a test message
-   * @param {string} ide - The IDE name to test
-   * @returns {Promise<Object>} Test result
-   */
-  async testQuotaDetection(ide) {
-    this.logger.log(`🧪 Testing quota detection for ${ide}...`);
-
-    try {
-      const quotaResult = await this.detectQuotaWarning(ide);
-
-      this.logger.log(`🧪 Quota detection test result for ${ide}:`, quotaResult);
-
-      return {
-        ide,
-        success: true,
-        hasQuotaWarning: quotaResult.hasQuotaWarning,
-        method: quotaResult.method,
-        note: quotaResult.note,
-        debug: quotaResult
-      };
-    } catch (error) {
-      this.logger.log(`❌ Quota detection test failed for ${ide}: ${error.message}`);
-
-      return {
-        ide,
-        success: false,
-        error: error.message,
-        note: `Quota detection test failed for ${ide}`
-      };
-    }
-  }
-}
+// @vibecodingmachine/core - Quota Detector
+// Handles quota detection for different IDEs using CDP and AppleScript
+
+import CDP from 'chrome-remote-interface';
+import { AppleScriptManager } from './applescript-manager.js';
+
+/**
+ * Quota Detector for IDE interactions
+ * Detects quota warnings in different IDEs using CDP and AppleScript
+ */
+export class QuotaDetector {
+  constructor() {
+    this.logger = console;
+    this.appleScriptManager = new AppleScriptManager();
+  }
+
+  /**
+   * Timeout utility function
+   * @param {number} ms - Timeout in milliseconds
+   * @param {Promise} promise - Promise to timeout
+   * @returns {Promise} Promise with timeout
+   */
+  timeout(ms, promise) {
+    return Promise.race([
+      promise,
+      new Promise((_, reject) =>
+        setTimeout(() => reject(new Error(`Operation timed out after ${ms}ms`)), ms)
+      )
+    ]);
+  }
+
+  /**
+   * Detect quota warnings in an IDE
+   * @param {string} ide - The IDE name ('vscode', 'cursor', 'windsurf')
+   * @returns {Promise<Object>} Quota detection result
+   */
+  async detectQuotaWarning(ide = 'vscode') {
+    this.logger.log(`🔍 Starting quota detection for ${ide}...`);
+
+    // Determine the port based on the IDE
+    const port = ide === 'windsurf' ? 9224 : ide === 'cursor' ? 9225 : 9222;
+    const ideName = ide === 'windsurf' ? 'Windsurf' : ide === 'cursor' ? 'Cursor' : 'VS Code';
+
+    this.logger.log(`🔌 Checking ${ideName} on port ${port}...`);
+
+    // For Windsurf, use AppleScript detection since CDP is not supported
+    if (ide === 'windsurf') {
+      this.logger.log('🔍 Windsurf detected - using AppleScript quota detection');
+      return await this.appleScriptManager.detectQuotaWarning(ide);
+    }
+
+    // For VS Code and Cursor, use CDP detection
+    try {
+      this.logger.log(`🔍 Attempting CDP connection to ${ideName} on port ${port}...`);
+
+      // Try to connect to CDP
+      const targets = await this.timeout(5000, CDP.List({ port }));
+
+      if (!targets || targets.length === 0) {
+        this.logger.log(`❌ No CDP targets found on port ${port}`);
+
+        // For Windsurf, fall back to AppleScript
+        if (ide === 'windsurf') {
+          this.logger.log('🔍 Falling back to enhanced AppleScript quota detection...');
+          return await this.appleScriptManager.detectQuotaWarning(ide);
+        }
+
+        return {
+          hasQuotaWarning: false,
+          error: `Could not find ${ideName}. Make sure ${ideName} is running with --remote-debugging-port=${port}`,
+          note: 'CDP connection failed'
+        };
+      }
+
+      this.logger.log(`✅ Found ${targets.length} CDP targets`);
+
+      // For Cursor, prefer the workspace target over settings
+      let workbench;
+      if (ide === 'cursor') {
+        workbench = targets.find(t => t.title !== 'Cursor Settings' && t.type === 'page') ||
+                    targets.find(t => t.url && t.url.includes('workbench')) ||
+                    targets[0];
+      } else {
+        workbench = targets.find(t => t.url && t.url.includes('workbench')) || targets[0];
+      }
+
+      if (!workbench) {
+        this.logger.log(`❌ No suitable workbench target found`);
+        return {
+          hasQuotaWarning: false,
+          error: `No ${ideName} workbench target found.`,
+          note: 'No workbench target available'
+        };
+      }
+
+      this.logger.log(`✅ Selected workbench: ${workbench.title}`);
+
+      // Connect to CDP
+      const client = await this.timeout(10000, CDP({ port, target: workbench }));
+      const { Runtime, Page } = client;
+
+      await this.timeout(5000, Runtime.enable());
+      if (Page && Page.bringToFront) {
+        try {
+          await this.timeout(3000, Page.bringToFront());
+        } catch (error) {
+          this.logger.log(`⚠️ Bring to front failed: ${error.message}`);
+        }
+      }
+
+      // Execute JavaScript to detect quota warnings
+      const detectionScript = `
+        (() => {
+          console.log('🔍 Starting quota detection in ${ideName}...');
+
+          // Common quota warning patterns
+          const quotaPatterns = [
+            'not enough credits',
+            'upgrade to a paid plan',
+            'switch to swe-1-lite',
+            'quota exceeded',
+            'credits exhausted',
+            'payment required',
+            'out of credits',
+            'insufficient credits',
+            'billing required',
+            'subscription needed'
+          ];
+
+          // Function to check if text contains quota warnings
+          function containsQuotaWarning(text) {
+            if (!text) return false;
+            const lowerText = text.toLowerCase();
+            return quotaPatterns.some(pattern => lowerText.includes(pattern));
+          }
+
+          // Method 1: Check all text content in the document
+          const allText = document.body.innerText || document.body.textContent || '';
+          if (containsQuotaWarning(allText)) {
+            console.log('❌ Quota warning detected in document text');
+            return {
+              hasQuotaWarning: true,
+              method: 'document-text',
+              matchedText: allText.substring(0, 200) + '...'
+            };
+          }
+
+          // Method 2: Check specific elements that might contain quota warnings
+          const quotaElements = [
+            // VS Code specific selectors
+            '.monaco-workbench .part.auxiliarybar',
+            '.monaco-workbench .part.sidebar.right',
+            '.monaco-workbench .chat-view',
+            '.monaco-workbench .chat-input',
+
+            // Cursor specific selectors
+            '.aislash-editor-input',
+            '.aislash-chat-container',
+            '.aislash-panel',
+
+            // Generic selectors
+            '[data-testid*="chat"]',
+            '[aria-label*="chat"]',
+            '[class*="chat"]',
+            '[class*="quota"]',
+            '[class*="credits"]',
+            '[class*="billing"]'
+          ];
+
+          for (const selector of quotaElements) {
+            const elements = document.querySelectorAll(selector);
+            for (const element of elements) {
+              const elementText = element.innerText || element.textContent || '';
+              if (containsQuotaWarning(elementText)) {
+                console.log('❌ Quota warning detected in element:', selector);
+                return {
+                  hasQuotaWarning: true,
+                  method: 'element-specific',
+                  selector: selector,
+                  matchedText: elementText.substring(0, 200) + '...'
+                };
+              }
+            }
+          }
+
+          // Method 3: Check for disabled input fields (common when quota is exceeded)
+          const disabledInputs = document.querySelectorAll('input[disabled], textarea[disabled], [contenteditable="true"][aria-disabled="true"]');
+          if (disabledInputs.length > 0) {
+            console.log('⚠️ Found disabled input fields, checking for quota context...');
+
+            // Check if there are quota-related messages near disabled inputs
+            for (const input of disabledInputs) {
+              const parentText = input.parentElement?.innerText || '';
+              if (containsQuotaWarning(parentText)) {
+                console.log('❌ Quota warning detected near disabled input');
+                return {
+                  hasQuotaWarning: true,
+                  method: 'disabled-input-context',
+                  matchedText: parentText.substring(0, 200) + '...'
+                };
+              }
+            }
+          }
+
+          // Method 4: Check for specific UI elements that indicate quota issues
+          const quotaButtons = document.querySelectorAll('button, a, [role="button"]');
+          for (const button of quotaButtons) {
+            const buttonText = button.innerText || button.textContent || '';
+            if (containsQuotaWarning(buttonText)) {
+              console.log('❌ Quota warning detected in button:', buttonText);
+              return {
+                hasQuotaWarning: true,
+                method: 'button-text',
+                matchedText: buttonText
+              };
+            }
+          }
+
+          // Method 5: Check for error messages or notifications
+          const errorElements = document.querySelectorAll('[class*="error"], [class*="warning"], [class*="notification"], [role="alert"]');
+          for (const error of errorElements) {
+            const errorText = error.innerText || error.textContent || '';
+            if (containsQuotaWarning(errorText)) {
+              console.log('❌ Quota warning detected in error element');
+              return {
+                hasQuotaWarning: true,
+                method: 'error-element',
+                matchedText: errorText.substring(0, 200) + '...'
+              };
+            }
+          }
+
+          console.log('✅ No quota warnings detected');
+          return {
+            hasQuotaWarning: false,
+            method: 'cdp-comprehensive',
+            note: 'No quota warnings found in ${ideName}'
+          };
+        })()
+      `;
+
+      const { result } = await this.timeout(10000, Runtime.evaluate({
+        expression: detectionScript,
+        returnByValue: true
+      }));
+
+      const detectionResult = result.value || result.unserializableValue;
+
+      this.logger.log('🔍 CDP quota detection result:', detectionResult);
+
+      if (detectionResult && detectionResult.hasQuotaWarning) {
+        this.logger.log('❌ Quota warning detected via CDP');
+        return {
+          hasQuotaWarning: true,
+          method: detectionResult.method,
+          note: detectionResult.matchedText || 'Quota warning detected via CDP',
+          debug: detectionResult
+        };
+      }
+
+      this.logger.log('✅ No quota warnings detected via CDP');
+      return {
+        hasQuotaWarning: false,
+        method: 'cdp-comprehensive',
+        note: 'No quota warnings found in ' + ideName,
+        debug: detectionResult
+      };
+
+    } catch (error) {
+      this.logger.log(`❌ CDP quota detection failed: ${error.message}`);
+
+      // For Windsurf, fall back to AppleScript
+      if (ide === 'windsurf') {
+        this.logger.log('🔍 Falling back to AppleScript quota detection...');
+        return await this.appleScriptManager.detectQuotaWarning(ide);
+      }
+
+      return {
+        hasQuotaWarning: false,
+        error: error.message,
+        note: `CDP quota detection failed for ${ideName}`
+      };
+    }
+  }
+
+  /**
+   * Get available IDEs without quota warnings
+   * @param {Array<string>} ides - Array of IDE names to check
+   * @returns {Promise<Array<string>>} Array of available IDE names
+   */
+  async getAvailableIdes(ides = ['vscode', 'cursor', 'windsurf']) {
+    this.logger.log('🔍 Checking available IDEs without quota warnings...');
+
+    const availableIdes = [];
+
+    for (const ide of ides) {
+      try {
+        const quotaResult = await this.detectQuotaWarning(ide);
+
+        if (!quotaResult.hasQuotaWarning) {
+          availableIdes.push(ide);
+          this.logger.log(`✅ ${ide} is available (no quota warnings)`);
+        } else {
+          this.logger.log(`❌ ${ide} has quota warnings: ${quotaResult.note || 'Unknown quota issue'}`);
+        }
+      } catch (error) {
+        this.logger.log(`⚠️ Error checking ${ide}: ${error.message}`);
+        // If we can't check quota, assume the IDE is available
+        availableIdes.push(ide);
+      }
+    }
+
+    this.logger.log(`📋 Available IDEs: ${availableIdes.join(', ')}`);
+    return availableIdes;
+  }
+
+  /**
+   * Test quota detection with a test message
+   * @param {string} ide - The IDE name to test
+   * @returns {Promise<Object>} Test result
+   */
+  async testQuotaDetection(ide) {
+    this.logger.log(`🧪 Testing quota detection for ${ide}...`);
+
+    try {
+      const quotaResult = await this.detectQuotaWarning(ide);
+
+      this.logger.log(`🧪 Quota detection test result for ${ide}:`, quotaResult);
+
+      return {
+        ide,
+        success: true,
+        hasQuotaWarning: quotaResult.hasQuotaWarning,
+        method: quotaResult.method,
+        note: quotaResult.note,
+        debug: quotaResult
+      };
+    } catch (error) {
+      this.logger.log(`❌ Quota detection test failed for ${ide}: ${error.message}`);
+
+      return {
+        ide,
+        success: false,
+        error: error.message,
+        note: `Quota detection test failed for ${ide}`
+      };
+    }
+  }
+}