vibecodingmachine-cli 2025.12.1-534 → 2025.12.6-1702

package/src/utils/auth.js

@@ -31,7 +31,30 @@ class CLIAuth {
    * Check if authenticated (uses shared storage)
    */
   async isAuthenticated() {
-    return await sharedAuth.isAuthenticated();
+    // First check if current token is valid
+    const isValid = await sharedAuth.isAuthenticated();
+    if (isValid) return true;
+
+    // If not valid, try to refresh
+    try {
+      const refreshToken = await sharedAuth.getRefreshToken();
+      if (refreshToken) {
+        // console.log(chalk.gray('Refreshing session...'));
+        const newTokens = await this._refreshTokens(refreshToken);
+
+        // Validate new ID token
+        await this._validateToken(newTokens.id_token);
+
+        // Save new tokens
+        await sharedAuth.saveToken(newTokens);
+        return true;
+      }
+    } catch (error) {
+      // Refresh failed (token expired or revoked), user needs to login again
+      // console.log(chalk.gray('Session refresh failed, please login again.'));
+    }
+
+    return false;
   }
 
   /**
@@ -82,7 +105,7 @@ class CLIAuth {
         res.on('end', () => {
           if (res.statusCode === 200) {
             const tokens = JSON.parse(data);
-            resolve(tokens.id_token); // Return the ID token
+            resolve(tokens); // Return full tokens object
           } else {
             reject(new Error(`Token exchange failed: ${res.statusCode} - ${data}`));
           }
@@ -97,7 +120,53 @@ class CLIAuth {
       req.end();
     });
   }
+  /**
+   * Refresh tokens using refresh_token
+   */
+  async _refreshTokens(refreshToken) {
+    const https = require('https');
+
+    const tokenEndpoint = `https://${COGNITO_DOMAIN}/oauth2/token`;
+    const postData = new URLSearchParams({
+      grant_type: 'refresh_token',
+      client_id: CLIENT_ID,
+      refresh_token: refreshToken
+    }).toString();
+
+    return new Promise((resolve, reject) => {
+      const options = {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/x-www-form-urlencoded',
+          'Content-Length': postData.length
+        }
+      };
 
+      const req = https.request(tokenEndpoint, options, (res) => {
+        let data = '';
+        res.on('data', (chunk) => { data += chunk; });
+        res.on('end', () => {
+          if (res.statusCode === 200) {
+            const tokens = JSON.parse(data);
+            // Cognito might not return a new refresh token, so we keep the old one if not provided
+            if (!tokens.refresh_token) {
+              tokens.refresh_token = refreshToken;
+            }
+            resolve(tokens);
+          } else {
+            reject(new Error(`Token refresh failed: ${res.statusCode} - ${data}`));
+          }
+        });
+      });
+
+      req.on('error', (error) => {
+        reject(new Error(`Token refresh request failed: ${error.message}`));
+      });
+
+      req.write(postData);
+      req.end();
+    });
+  }
   /**
    * Validate JWT token with signature verification
    * @param {string} idToken - JWT token to validate
@@ -312,18 +381,22 @@ class CLIAuth {
             try {
               // Exchange code for tokens
               console.log(chalk.gray('Exchanging authorization code for tokens...'));
-              const idToken = await this._exchangeCodeForTokens(
+              // Exchange code for tokens
+              console.log(chalk.gray('Exchanging authorization code for tokens...'));
+              const tokens = await this._exchangeCodeForTokens(
                 code,
                 codeVerifier,
                 `http://localhost:${PORT}/callback`
               );
 
+              const idToken = tokens.id_token;
+
               // Validate token (signature + claims)
               console.log(chalk.gray('Validating token...'));
               await this._validateToken(idToken);
 
               // Save token
-              await sharedAuth.saveToken(idToken);
+              await sharedAuth.saveToken(tokens);
 
               // Show success page
               res.writeHead(200, {
@@ -428,7 +501,7 @@ class CLIAuth {
                 serverClosed = true;
                 server.close();
                 console.log(chalk.green('✓ Authentication successful!\n'));
-                resolve(idToken);
+                resolve(tokens.id_token);
               }
             } catch (error) {
               // Token exchange or validation failed
@@ -560,12 +633,16 @@ class CLIAuth {
 
       // Exchange code for tokens
       console.log(chalk.gray('Exchanging authorization code for tokens...'));
-      const idToken = await this._exchangeCodeForTokens(
+      // Exchange code for tokens
+      console.log(chalk.gray('Exchanging authorization code for tokens...'));
+      const tokens = await this._exchangeCodeForTokens(
         code,
         codeVerifier,
         'http://localhost:3000/callback'
       );
 
+      const idToken = tokens.id_token;
+
       // Validate JWT token
       console.log(chalk.gray('Validating token...'));
       try {
@@ -577,7 +654,7 @@ class CLIAuth {
       }
 
       // Save token using shared storage (only if validation passed)
-      await sharedAuth.saveToken(idToken);
+      await sharedAuth.saveToken(tokens);
 
       console.log(chalk.green('\n✓ Authentication successful!'));
       return idToken;

package/src/utils/auto-mode-simple-ui.js

@@ -13,9 +13,6 @@ class AutoModeSimpleUI {
     this.chatCount = 0;
     this.maxChats = null;
     this.progress = 0;
-    this.outputLineCount = 0; // Track output lines for periodic re-render
-    this.lastCardPrintTime = Date.now(); // Track when we last printed the card
-    this.lastCardPrintLine = 0; // Track line count when we last printed the card
 
     // Print header once
     console.log('\n' + chalk.bold.cyan('═══════════════════════════════════════════════════════════'));
@@ -109,30 +106,13 @@ class AutoModeSimpleUI {
       borderColor: 'magenta',
       backgroundColor: 'black'
     });
-    
-    // Print the card with a separator line before it for visibility
-    if (this.outputLineCount > 0) {
-      // Add separator when re-printing after output
-      console.log(chalk.gray('─'.repeat(80)));
-    }
+
     console.log(card);
-    this.lastCardPrintTime = Date.now();
-    this.lastCardPrintLine = this.outputLineCount;
   }
 
   appendOutput(line) {
-    // Periodically re-print the workflow card to keep it visible
-    // Re-print every 20 lines of output, or if more than 5 seconds have passed
-    const linesSinceLastCard = this.outputLineCount - this.lastCardPrintLine;
-    const timeSinceLastCard = Date.now() - this.lastCardPrintTime;
-    
-    if (linesSinceLastCard >= 20 || timeSinceLastCard > 5000) {
-      this.printWorkflowCard();
-    }
-    
-    // Increment output line counter
+    // Just print the line - no automatic card re-printing
     if (line && line.trim()) {
-      this.outputLineCount++;
       console.log(line);
     }
   }

package/src/utils/first-run.js

@@ -0,0 +1,293 @@
+const inquirer = require('inquirer');
+const chalk = require('chalk');
+const boxen = require('boxen');
+const fs = require('fs-extra');
+const path = require('path');
+const os = require('os');
+const { getProviderDefinitions, saveProviderPreferences, getDefaultProviderOrder } = require('./provider-registry');
+const { isKiroInstalled, installKiro } = require('./kiro-installer');
+
+async function checkFirstRun() {
+    const configDir = path.join(os.homedir(), '.vibecodingmachine');
+
+    // If directory exists, we assume it's not the first run
+    if (await fs.pathExists(configDir)) {
+        return;
+    }
+
+    // FIRST RUN EXPERIENCE
+    console.clear();
+
+    // 1. Big Welcome Banner
+    console.log('\n' + boxen(
+        chalk.bold.cyan('Welcome to Vibe Coding Machine') + '\n\n' +
+        chalk.white('The future of coding is here.') + '\n' +
+        chalk.gray('Just describe what you want, and we code it for you.'),
+        {
+            padding: 1,
+            margin: 1,
+            borderStyle: 'double',
+            borderColor: 'cyan',
+            align: 'center'
+        }
+    ));
+
+    console.log(chalk.cyan('Let\'s get you set up with the best AI coding environment.\n'));
+    await new Promise(resolve => setTimeout(resolve, 1500));
+
+    // --- NEW: Vibe Coding Introduction ---
+    const { showIntro } = await inquirer.prompt([
+        {
+            type: 'confirm',
+            name: 'showIntro',
+            message: 'Are you new to Vibe Coding? Would you like a quick 30-second tour?',
+            default: true
+        }
+    ]);
+
+    if (showIntro) {
+        const introContent = [
+            chalk.bold.cyan('╔════════════════════════ Vibe Coding 101 ════════════════════════╗'),
+            chalk.cyan('║                                                                 ║'),
+            chalk.cyan('║  1. What is an IDE?                                             ║'),
+            chalk.cyan('║     Think of it like MS Word, but for writing code.             ║'),
+            chalk.cyan('║     You need one to build apps, but don\'t worry—we\'ll install   ║'),
+            chalk.cyan('║     the best ones for you automatically.                        ║'),
+            chalk.cyan('║                                                                 ║'),
+            chalk.cyan('║  2. How it works                                                ║'),
+            chalk.cyan('║     You simply describe the app or web page you want.           ║'),
+            chalk.cyan('║     VibeCodingMachine then takes over your keyboard to write    ║'),
+            chalk.cyan('║     the code inside the IDE (GUI mode) or runs in the           ║'),
+            chalk.cyan('║     background (CLI mode) which is even faster.                 ║'),
+            chalk.cyan('║                                                                 ║'),
+            chalk.bold.cyan('╚═════════════════════════════════════════════════════════════════╝')
+        ].join('\n');
+
+        console.log('\n' + introContent + '\n');
+
+        await inquirer.prompt([{
+            type: 'input',
+            name: 'continue',
+            message: 'Press Enter to continue...'
+        }]);
+    }
+
+
+
+
+    // 2. Detect IDEs
+    const definitions = getProviderDefinitions();
+    const ideDefinitions = definitions.filter(d => d.type === 'ide');
+
+    const detectedIDEs = [];
+    const otherIDEs = [];
+
+    // Simple simulation of detection for standard apps on macOS
+    // In a real scenario, we might check other paths or registry on Windows
+    const platform = os.platform();
+    const getAppPath = (name) => platform === 'darwin' ? `/Applications/${name}.app` : null;
+
+    for (const ide of ideDefinitions) {
+        let installed = false;
+
+        if (ide.id === 'kiro') {
+            installed = isKiroInstalled();
+        } else if (ide.id === 'cursor') {
+            if (platform === 'darwin') installed = await fs.pathExists(getAppPath('Cursor'));
+        } else if (ide.id === 'windsurf') {
+            if (platform === 'darwin') installed = await fs.pathExists(getAppPath('Windsurf'));
+        } else if (ide.id === 'vscode') {
+            if (platform === 'darwin') installed = await fs.pathExists(getAppPath('Visual Studio Code'));
+        } else if (ide.id === 'antigravity') {
+            // Antigravity is just a provider here, maybe assume installed if running? 
+            // Or check for some component. For now default to true or false based on something simple.
+            installed = true; // Assume available
+        }
+
+        if (installed) {
+            detectedIDEs.push(ide);
+        } else {
+            otherIDEs.push(ide);
+        }
+    }
+
+    // 3. Status Report & Selection
+    if (detectedIDEs.length > 0) {
+        console.log(chalk.green('\n✓ We found these IDEs and enabled them for you:'));
+        detectedIDEs.forEach(ide => {
+            console.log(chalk.gray(`  • ${ide.name}`));
+        });
+    }
+
+    let selectedIDEs = [];
+
+    // Only prompt for uninstalled IDEs if there are any
+    if (otherIDEs.length > 0) {
+        console.log(); // Spacing
+        const choices = otherIDEs.map(ide => ({
+            name: ide.name,
+            value: ide.id,
+            checked: true // Select by default as requested
+        }));
+
+        const response = await inquirer.prompt([{
+            type: 'checkbox',
+            name: 'selectedIDEs',
+            message: 'Select additional IDEs to install & enable:',
+            choices: choices,
+            pageSize: 10
+        }]);
+        selectedIDEs = response.selectedIDEs;
+    }
+
+    // 4. Handle Installations
+    if (selectedIDEs.includes('kiro')) {
+        const kiroInstalled = isKiroInstalled();
+        if (!kiroInstalled) {
+            // installKiro logs its own inputs, but we want to handle failure gracefully
+            const success = await installKiro();
+            if (!success) {
+                console.log(chalk.yellow('\n⚠️  Kiro installation failed or was skipped.'));
+                console.log(chalk.gray('You can continue setting up other IDEs.'));
+                await new Promise(resolve => setTimeout(resolve, 2000));
+            } else {
+                // Post-install notification with auto-launch
+                console.log('\n' + boxen(
+                    chalk.bold.yellow('⚠️  ACTION REQUIRED: SIGN IN ') + '\n\n' +
+                    chalk.white('1. I will launch AWS Kiro IDE for you.') + '\n' +
+                    chalk.white('2. Sign in with Google, GitHub, or AWS Builder ID.') + '\n' +
+                    chalk.white('3. Click "Skip all" to skip survey questions.') + '\n' +
+                    chalk.white('4. Once signed in, you can minimize or close the IDE.') + '\n\n' +
+                    chalk.cyan('You must sign in for the CLI to control the IDE.'),
+                    {
+                        padding: 1,
+                        margin: 1,
+                        borderStyle: 'round',
+                        borderColor: 'yellow'
+                    }
+                ));
+
+                await inquirer.prompt([{
+                    type: 'input',
+                    name: 'launch',
+                    message: 'Press Enter and I will launch Kiro for you to sign in to...'
+                }]);
+
+                const { exec } = require('child_process');
+                console.log(chalk.gray('Launching AWS Kiro...'));
+
+                try {
+                    // Try to open AWS Kiro
+                    exec('open -a "AWS Kiro"', (error) => {
+                        if (error) {
+                            // Fallback to just Kiro
+                            exec('open -a "Kiro"', (err) => {
+                                if (err) {
+                                    console.log(chalk.red('Could not auto-launch Kiro. Please open it manually.'));
+                                }
+                            });
+                        }
+                    });
+                } catch (e) {
+                    // Ignore errors, user can launch manually
+                }
+
+                await inquirer.prompt([{
+                    type: 'input',
+                    name: 'continue',
+                    message: 'Press Enter once you have signed in...'
+                }]);
+            }
+        }
+    }
+
+    // 5. Configure Preferences
+    // Enable detected IDEs AND selected IDEs, disable others
+    const defaultOrder = getDefaultProviderOrder();
+    const enabledMap = {};
+
+    defaultOrder.forEach(id => {
+        // Enable if it was detected OR selected
+        // For LLMs (not 'ide' type), keep enabled by default
+        const def = definitions.find(d => d.id === id);
+        if (def && def.type === 'ide') {
+            const isDetected = detectedIDEs.some(d => d.id === id);
+            const isSelected = selectedIDEs.includes(id);
+            enabledMap[id] = isDetected || isSelected;
+        } else {
+            enabledMap[id] = true; // Keep LLMs enabled by default
+        }
+    });
+
+    // Save initial preferences. provider-registry.js usually saves to ~/.config/vibecodingmachine/config.json
+    // But here we are checking ~/.vibecodingmachine.
+    // We should create the ~/.vibecodingmachine directory to mark first run as complete.
+    await fs.ensureDir(configDir);
+    await saveProviderPreferences(defaultOrder, enabledMap);
+
+    // --- NEW: CLI Usage Onboarding ---
+    // Moved here so IDEs are set up first
+    const { isFamiliar } = await inquirer.prompt([
+        {
+            type: 'confirm',
+            name: 'isFamiliar',
+            message: 'Are you familiar with how to use VibeCodingMachine CLI?',
+            default: false
+        }
+    ]);
+
+    if (!isFamiliar) {
+        const cliUsageContent = [
+            chalk.bold.green('╔════════════════════════ CLI Crash Course ═══════════════════════╗'),
+            chalk.green('║                                                                 ║'),
+            chalk.green('║  1. Navigation                                                  ║'),
+            chalk.green('║     [↑/↓] Arrow keys to move selection                          ║'),
+            chalk.green('║     [Space] to toggle checkboxes [x]                            ║'),
+            chalk.green('║     [Enter] to confirm your choice                              ║'),
+            chalk.green('║                                                                 ║'),
+            chalk.green('║  2. Key Features                                                ║'),
+            chalk.green('║     • Auto Mode: The autonomous coding agent                    ║'),
+            chalk.green('║     • Direct Mode: Chat directly with LLMs                      ║'),
+            chalk.green('║     • IDE Setup: Install and configure tools                    ║'),
+            chalk.green('║                                                                 ║'),
+            chalk.green('║  3. Pro Tip                                                     ║'),
+            chalk.green('║     If you ever get stuck, just restart the CLI with \'vcm\'.     ║'),
+            chalk.green('║                                                                 ║'),
+            chalk.bold.green('╚═════════════════════════════════════════════════════════════════╝')
+        ].join('\n');
+
+        console.log('\n' + cliUsageContent + '\n');
+
+        await inquirer.prompt([{
+            type: 'input',
+            name: 'continue',
+            message: 'Press Enter to continue...'
+        }]);
+    }
+
+    // 6. Explanation
+    console.log('\n' + boxen(
+        chalk.bold.green(' Setup Complete! ') + '\n\n' +
+        chalk.white('Vibe Coding is about flow. You describe, we code.') + '\n' +
+        chalk.white('Using your local IDEs + Cloud LLMs for maximum power.'),
+        {
+            padding: 1,
+            margin: 1,
+            borderStyle: 'round',
+            borderColor: 'green'
+        }
+    ));
+
+    console.log(chalk.cyan('Press any key to continue to the main menu...'));
+
+    await new Promise((resolve) => {
+        process.stdin.setRawMode(true);
+        process.stdin.resume();
+        process.stdin.once('data', () => {
+            process.stdin.setRawMode(false);
+            resolve();
+        });
+    });
+}
+
+module.exports = { checkFirstRun };