testchimp-runner-core 0.0.34 → 0.0.36

package/src/credit-usage-service.ts

@@ -1,188 +0,0 @@
-import axios from 'axios';
-import { AuthConfig, getAuthHeaders } from './auth-config';
-import { loadEnvConfig } from './env-loader';
-
-export enum CreditUsageReason {
-  UNKNOWN_CREDIT_USAGE_REASON = 0,
-  EXPLORER_STEP = 1,
-  TEST_REPAIR = 2,
-  SCRIPT_GENERATE = 3
-}
-
-export interface InsertCreditUsageRequest {
-  credits?: number;
-  usage_reason?: CreditUsageReason;
-  job_id?: string;
-}
-
-export interface InsertCreditUsageResponse {
-  // Empty response as per the protobuf definition
-}
-
-/**
- * Credit usage information for callback
- */
-export interface CreditUsage {
-  credits: number;
-  usageReason: CreditUsageReason;
-  jobId?: string;
-  timestamp: number;
-}
-
-/**
- * Callback for credit usage reporting
- * Allows consumers to track credit usage in their own systems
- */
-export type CreditUsageCallback = (usage: CreditUsage) => void | Promise<void>;
-
-export class CreditUsageService {
-  private backendUrl: string;
-  private authConfig: AuthConfig | null;
-  private logger?: (message: string, level?: 'log' | 'error' | 'warn') => void;
-  private creditUsageCallback?: CreditUsageCallback;
-
-  constructor(authConfig?: AuthConfig, backendUrl?: string, creditUsageCallback?: CreditUsageCallback) {
-    // Use provided backend URL or fall back to environment configuration
-    if (backendUrl) {
-      this.backendUrl = backendUrl;
-    } else {
-      // Fall back to environment configuration for backward compatibility
-      const envConfig = loadEnvConfig();
-      this.backendUrl = envConfig.TESTCHIMP_BACKEND_URL;
-    }
-    
-    this.authConfig = authConfig || null;
-    this.creditUsageCallback = creditUsageCallback;
-  }
-
-  /**
-   * Set a logger callback for capturing execution logs
-   */
-  setLogger(logger: (message: string, level?: 'log' | 'error' | 'warn') => void): void {
-    this.logger = logger;
-  }
-
-  /**
-   * Set credit usage callback
-   * Allows consumers to track credit usage in their own systems
-   */
-  setCreditUsageCallback(callback: CreditUsageCallback): void {
-    this.creditUsageCallback = callback;
-  }
-
-  /**
-   * Log a message using the configured logger
-   */
-  private log(message: string, level: 'log' | 'error' | 'warn' = 'log'): void {
-    if (this.logger) {
-      this.logger(message, level);
-    }
-    // No console fallback - logs are routed to consumer
-  }
-
-  /**
-   * Update authentication configuration
-   */
-  setAuthConfig(authConfig: AuthConfig): void {
-    this.authConfig = authConfig;
-  }
-
-  /**
-   * Get current authentication configuration
-   */
-  getAuthConfig(): AuthConfig | null {
-    return this.authConfig;
-  }
-
-  /**
-   * Report credit usage via callback and/or backend
-   * 
-   * Behavior:
-   * 1. If callback provided: Call callback (server-side uses this to update DB directly)
-   * 2. If NO callback but auth configured: Make axios call to backend (client-side: vs-ext, github action)
-   * 3. Server-side: Provides callback, no axios calls made
-   * 4. Client-side: No callback, uses auth for axios calls to backend
-   */
-  async reportCreditUsage(
-    credits: number = 1,
-    usageReason: CreditUsageReason,
-    jobId?: string
-  ): Promise<InsertCreditUsageResponse> {
-    const creditUsage: CreditUsage = {
-      credits,
-      usageReason,
-      jobId,
-      timestamp: Date.now()
-    };
-    
-    // 1. If callback provided: Use callback (server-side path)
-    if (this.creditUsageCallback) {
-      try {
-        await this.creditUsageCallback(creditUsage);
-        this.log(`Credit usage reported via callback: ${credits} credits for ${CreditUsageReason[usageReason]}`);
-        return {}; // Callback succeeded, no need for backend call
-      } catch (error: any) {
-        this.log(`Credit usage callback failed: ${error.message}`, 'error');
-        throw error; // Callback failure is critical for server-side
-      }
-    }
-    
-    // 2. No callback: Make axios call if auth configured (client-side path: vs-ext, github action)
-    if (this.authConfig) {
-      try {
-        const authHeaders = getAuthHeaders(this.authConfig);
-        const url = `${this.backendUrl}/localagent/insert_credit_usage`;
-        
-        const request: InsertCreditUsageRequest = {
-          credits,
-          usage_reason: usageReason,
-          job_id: jobId
-        };
-
-        const response = await axios.post(url, request, {
-          headers: {
-            ...authHeaders,
-            'Content-Type': 'application/json'
-          },
-          timeout: 10000 // 10 second timeout for credit usage calls
-        });
-
-        this.log(`Credit usage reported to backend: ${credits} credits for ${CreditUsageReason[usageReason]}`);
-        return response.data;
-      } catch (error: any) {
-        this.log(`Credit usage backend report failed: ${error.message}`, 'error');
-        throw error; // Backend failure is critical for client-side
-      }
-    }
-    
-    // 3. No callback and no auth - no credit tracking (development mode)
-    this.log(`Credit usage not tracked (no callback or auth configured): ${credits} credits for ${CreditUsageReason[usageReason]}`, 'warn');
-    return {};
-  }
-
-  /**
-   * Report script generation credit usage
-   */
-  async reportScriptGenerationCredit(jobId?: string): Promise<void> {
-    try {
-      await this.reportCreditUsage(1, CreditUsageReason.SCRIPT_GENERATE, jobId);
-      this.log(`Credit usage reported for script generation${jobId ? ` (job: ${jobId})` : ''}`);
-    } catch (error) {
-      this.log(`Failed to report script generation credit usage: ${error}`, 'error');
-      // Don't throw - credit reporting should not break the main flow
-    }
-  }
-
-  /**
-   * Report AI repair credit usage
-   */
-  async reportAIRepairCredit(jobId?: string): Promise<void> {
-    try {
-      await this.reportCreditUsage(1, CreditUsageReason.TEST_REPAIR, jobId);
-      this.log(`Credit usage reported for AI repair${jobId ? ` (job: ${jobId})` : ''}`);
-    } catch (error) {
-      this.log(`Failed to report AI repair credit usage: ${error}`, 'error');
-      // Don't throw - credit reporting should not break the main flow
-    }
-  }
-}

package/src/env-loader.ts

@@ -1,103 +0,0 @@
-/**
- * Environment Configuration Loader
- * 
- * This module loads environment configuration from packaged env files
- * instead of relying on process.env which may not be available in VS Code extensions
- */
-
-interface EnvConfig {
-  TESTCHIMP_BACKEND_URL: string;
-}
-
-let cachedConfig: EnvConfig | null = null;
-
-/**
- * Load environment configuration from the packaged env file
- * Falls back to process.env if env file is not available
- */
-export function loadEnvConfig(): EnvConfig {
-  if (cachedConfig) {
-    return cachedConfig;
-  }
-
-  try {
-    // Try multiple possible paths for the env file
-    const possiblePaths = [
-      require('path').join(String(__dirname), 'env'),
-      require('path').join(String(__dirname), './env'),
-      require('path').join(String(__dirname), '../env'),
-      require('path').join(String(__dirname), '../../runner-core/env'),
-      require('path').join(String(__dirname), '../../runner-core/dist/env'),
-      require('path').join(String(process.cwd()), 'env'),
-      require('path').join(String(process.cwd()), 'runner-core/env'),
-      require('path').join(String(process.cwd()), 'runner-core/dist/env'),
-      require('path').join(String(process.cwd()), 'local/runner-core/env'),
-      require('path').join(String(process.cwd()), 'local/runner-core/dist/env')
-    ];
-    
-    let envContent = '';
-    let envPath = '';
-    
-    for (const path of possiblePaths) {
-      try {
-        envContent = require('fs').readFileSync(path, 'utf8');
-        envPath = path;
-        break;
-      } catch (error) {
-        // Silently continue to next path
-      }
-    }
-    
-    if (!envContent) {
-      throw new Error('Could not find env file in any of the expected locations');
-    }
-    
-    const config: Partial<EnvConfig> = {};
-    
-    envContent.split('\n').forEach((line: string) => {
-      const trimmedLine = line.trim();
-      if (trimmedLine && !trimmedLine.startsWith('#')) {
-        const [key, value] = trimmedLine.split('=');
-        if (key && value) {
-          const trimmedKey = key.trim() as keyof EnvConfig;
-          config[trimmedKey] = value.trim();
-        }
-      }
-    });
-    
-    // Ensure required properties are present
-    const finalConfig: EnvConfig = {
-      TESTCHIMP_BACKEND_URL: config.TESTCHIMP_BACKEND_URL || 'https://featureservice.testchimp.io'
-    };
-    
-    cachedConfig = finalConfig;
-    return finalConfig;
-  } catch (error) {
-    // Fallback to process.env
-    let backendUrl = process.env.TESTCHIMP_BACKEND_URL;
-    
-    // If no explicit backend URL, determine from TESTCHIMP_ENV
-    if (!backendUrl) {
-      const testchimpEnv = process.env.TESTCHIMP_ENV;
-      if (testchimpEnv === 'staging') {
-        backendUrl = 'https://featureservice-staging.testchimp.io';
-      } else {
-        backendUrl = 'https://featureservice.testchimp.io';
-      }
-    }
-    
-    const config: EnvConfig = {
-      TESTCHIMP_BACKEND_URL: backendUrl
-    };
-    cachedConfig = config;
-    return config;
-  }
-}
-
-/**
- * Get a specific environment variable
- */
-export function getEnvVar(key: keyof EnvConfig): string | undefined {
-  const config = loadEnvConfig();
-  return config[key];
-}