teleportation-cli 1.1.5 → 1.2.1

package/lib/remote/providers/provider-factory.js

@@ -1,15 +1,21 @@
 /**
  * Provider Factory for Intelligent Provider Selection
  *
- * Automatically selects the best cloud provider (Fly.io or Daytona) based on:
+ * Automatically selects the best cloud provider (Fly.io, Daytona, or Sprites.dev) based on:
  * - Task description analysis
  * - Estimated duration
  * - Keyword detection
  * - User preferences
+ *
+ * Provider characteristics:
+ * - **Daytona**: Quick, ephemeral tasks (PRs, bug fixes, prototypes)
+ * - **Fly.io**: Long-running persistent VMs (overnight, migrations)
+ * - **Sprites.dev**: Tasks requiring checkpoints/rollback (large refactors, experiments)
  */
 
 import { FlyProvider } from './fly-provider.js';
 import { DaytonaProvider } from './daytona-provider.js';
+import { SpritesProvider } from './sprites-provider.js';
 
 export class ProviderFactory {
   /**
@@ -19,7 +25,8 @@ export class ProviderFactory {
    * @param {LivePortClient} config.livePortClient - LivePort client (required)
    * @param {string} [config.flyApiToken] - Fly.io API token
    * @param {string} [config.daytonaApiKey] - Daytona API key
-   * @param {string} [config.defaultProvider] - Default provider ('fly' or 'daytona')
+   * @param {string} [config.spritesToken] - Sprites.dev API token
+   * @param {string} [config.defaultProvider] - Default provider ('fly', 'daytona', or 'sprites')
    */
   constructor(config = {}) {
     if (!config.vaultClient) {
@@ -47,6 +54,14 @@ export class ProviderFactory {
       'feature', 'branch', 'bug fix', 'hotfix',
       'test', 'experiment', 'prototype', 'spike',
     ];
+
+    // Keywords that suggest checkpoint/rollback needs (prefer Sprites)
+    this.spritesKeywords = [
+      'checkpoint', 'rollback', 'restore', 'snapshot',
+      'risky', 'experimental', 'major refactor', 'large refactor',
+      'database migration', 'data migration', 'breaking change',
+      'incremental', 'save point', 'recovery',
+    ];
   }
 
   /**
@@ -54,14 +69,15 @@ export class ProviderFactory {
    *
    * Decision logic:
    * 1. If provider explicitly specified in options, use it
-   * 2. Analyze task description for duration indicators
-   * 3. Match keywords to provider preferences
-   * 4. Default to Daytona for quick, ephemeral tasks
+   * 2. Check for Sprites keywords (checkpoint/rollback needs)
+   * 3. Analyze task description for duration indicators
+   * 4. Match keywords to provider preferences
+   * 5. Default to Daytona for quick, ephemeral tasks
    *
    * @param {string} taskDescription - Description of the task to perform
    * @param {Object} [options] - Selection options
-   * @param {string} [options.provider] - Force specific provider ('fly' or 'daytona')
-   * @returns {string} Selected provider type ('fly' or 'daytona')
+   * @param {string} [options.provider] - Force specific provider ('fly', 'daytona', or 'sprites')
+   * @returns {string} Selected provider type ('fly', 'daytona', or 'sprites')
    */
   selectProvider(taskDescription, options = {}) {
     // 1. Explicit override takes precedence
@@ -69,20 +85,28 @@ export class ProviderFactory {
       return options.provider;
     }
 
-    // 2. Estimate task duration
-    const duration = this._estimateDuration(taskDescription);
+    const lowerDesc = taskDescription.toLowerCase();
 
-    // 3. Analyze keywords
-    const keywords = this._analyzeKeywords(taskDescription);
+    // 2. Check for Sprites-specific keywords (checkpoint/rollback needs)
+    const hasSpritesKeyword = this.spritesKeywords.some(keyword =>
+      lowerDesc.includes(keyword)
+    );
+
+    if (hasSpritesKeyword) {
+      return 'sprites';
+    }
 
-    // 4. Decision logic
+    // 3. Estimate task duration
+    const duration = this._estimateDuration(taskDescription);
+
+    // 4. Decision logic based on duration
     if (duration === 'long') {
       return 'fly';
     }
 
-    // Check for Fly-specific keywords
+    // 5. Check for Fly-specific keywords
     const hasFlyKeyword = this.flyKeywords.some(keyword =>
-      taskDescription.toLowerCase().includes(keyword)
+      lowerDesc.includes(keyword)
     );
 
     if (hasFlyKeyword) {
@@ -96,9 +120,9 @@ export class ProviderFactory {
   /**
    * Create a provider instance of the specified type
    *
-   * @param {string} type - Provider type ('fly' or 'daytona')
+   * @param {string} type - Provider type ('fly', 'daytona', or 'sprites')
    * @param {Object} [providerConfig] - Provider-specific configuration
-   * @returns {FlyProvider|DaytonaProvider} Provider instance
+   * @returns {FlyProvider|DaytonaProvider|SpritesProvider} Provider instance
    * @throws {Error} If provider type is unknown
    */
   createProvider(type, providerConfig = {}) {
@@ -122,6 +146,13 @@ export class ProviderFactory {
           ...providerConfig,
         });
 
+      case 'sprites':
+        return new SpritesProvider({
+          ...baseConfig,
+          spritesToken: this.config.spritesToken,
+          ...providerConfig,
+        });
+
       default:
         throw new Error(`Unknown provider type: ${type}`);
     }
@@ -134,8 +165,8 @@ export class ProviderFactory {
    *
    * @param {string} taskDescription - Description of the task to perform
    * @param {Object} [options] - Selection and provider options
-   * @param {string} [options.provider] - Force specific provider ('fly' or 'daytona')
-   * @returns {FlyProvider|DaytonaProvider} Configured provider instance
+   * @param {string} [options.provider] - Force specific provider ('fly', 'daytona', or 'sprites')
+   * @returns {FlyProvider|DaytonaProvider|SpritesProvider} Configured provider instance
    */
   selectAndCreate(taskDescription, options = {}) {
     const providerType = this.selectProvider(taskDescription, options);
@@ -204,7 +235,7 @@ export class ProviderFactory {
   /**
    * Get default configuration for a provider type
    *
-   * @param {string} type - Provider type ('fly' or 'daytona')
+   * @param {string} type - Provider type ('fly', 'daytona', or 'sprites')
    * @returns {Object} Default configuration for the provider
    */
   getDefaultProviderConfig(type) {
@@ -221,8 +252,48 @@ export class ProviderFactory {
           autoStopMinutes: 60,
         };
 
+      case 'sprites':
+        return {
+          // Sprites uses auto-hibernate, no explicit stop timer needed
+          cpus: 1,
+          memory: 2048,
+        };
+
       default:
         return {};
     }
   }
+
+  /**
+   * Get list of available providers
+   *
+   * @returns {string[]} Array of available provider types
+   */
+  getAvailableProviders() {
+    const available = [];
+
+    if (this.config.flyApiToken || process.env.FLY_API_TOKEN) {
+      available.push('fly');
+    }
+
+    if (this.config.daytonaApiKey || process.env.DAYTONA_API_KEY) {
+      available.push('daytona');
+    }
+
+    if (this.config.spritesToken || process.env.SPRITES_TOKEN) {
+      available.push('sprites');
+    }
+
+    return available;
+  }
+
+  /**
+   * Check if a specific provider is available
+   *
+   * @param {string} type - Provider type
+   * @returns {boolean} True if provider credentials are configured
+   */
+  isProviderAvailable(type) {
+    return this.getAvailableProviders().includes(type);
+  }
 }