add some spice to cli

Author: mrlynn

src/cli/celebration.js

@@ -0,0 +1,48 @@
+// src/cli/celebration.js
+import chalk from 'chalk';
+
+const frames = [
+  `
+    🎉 
+   \\|/
+    |  
+   / \\
+  `,
+  `
+    🎊
+    /|\\
+     |
+    / \\
+  `,
+  `
+    ✨
+    \\|/
+     |
+    / \\
+  `
+];
+
+export function celebrate(message) {
+  let frameIndex = 0;
+  const animation = setInterval(() => {
+    process.stdout.write('\x1B[2J\x1B[3J\x1B[H');
+    console.log(chalk.rainbow(`
+    ===============================
+         ${message}
+    ===============================
+    `));
+    console.log(chalk.cyan(frames[frameIndex]));
+    frameIndex = (frameIndex + 1) % frames.length;
+  }, 200);
+
+  // Stop after 2 seconds
+  setTimeout(() => {
+    clearInterval(animation);
+    process.stdout.write('\x1B[2J\x1B[3J\x1B[H');
+    console.log(chalk.green(`
+    ===============================
+         ${message}
+    ===============================
+    `));
+  }, 2000);
+}

src/cli/createRagApp.js

@@ -4,19 +4,36 @@ import path from 'path';
 import { execSync } from 'child_process';
 import chalk from 'chalk';
 import { displayLogo } from './asciiLogo.js';
+import MongoSpinner from './spinner.js';
+import { celebrate } from './celebration.js';
+import FunProgressBar from './progressBar.js';
 
-export function createRagApp(projectName) {
-  const projectPath = path.resolve(process.cwd(), projectName);
+export async function createRagApp(projectName) {
+  const spinner = new MongoSpinner();
+  const progressBar = new FunProgressBar();
+  
+  // Display the ASCII logo
   displayLogo();
-
+  
+  const projectPath = path.resolve(process.cwd(), projectName);
 
   if (fs.existsSync(projectPath)) {
     console.error(chalk.red(`Error: Directory "${projectName}" already exists.`));
     process.exit(1);
   }
 
   console.log(chalk.green(`\n🚀 Creating a new RAG app in ${projectPath}\n`));
+  
+  // Start the creation process with spinner
+  spinner.start('Initializing your RAG app');
   fs.mkdirSync(projectPath, { recursive: true });
+  await new Promise(resolve => setTimeout(resolve, 1000));
+  spinner.stop(true);
+
+  // Show progress while creating files
+  console.log(chalk.cyan('\nPreparing your MongoDB RAG environment...'));
+  let currentProgress = 0;
+  progressBar.update(currentProgress);
 
   // Step 1: Create package.json
   fs.writeFileSync(path.join(projectPath, 'package.json'), JSON.stringify({
@@ -35,8 +52,10 @@ export function createRagApp(projectName) {
       "mongodb-rag": "^0.15.0"
     }
   }, null, 2));
+  currentProgress += 0.25;
+  progressBar.update(currentProgress);
 
-  // Step 2: Create .env file with corrected variables
+  // Step 2: Create .env file
   fs.writeFileSync(path.join(projectPath, '.env'), `
 MONGODB_URI=mongodb+srv://your_user:your_password@your-cluster.mongodb.net/mongorag
 PORT=5000
@@ -49,8 +68,10 @@ EMBEDDING_MODEL=text-embedding-3-small  # Adjust for different providers
 # MongoDB Vector Search Index
 VECTOR_INDEX=default
   `);
+  currentProgress += 0.25;
+  progressBar.update(currentProgress);
 
-  // Step 3: Create server.js with environment-aware embedding configuration
+  // Step 3: Create server.js
   fs.writeFileSync(path.join(projectPath, 'server.js'), `
 import express from 'express';
 import cors from 'cors';
@@ -101,13 +122,26 @@ app.listen(process.env.PORT || 5000, () => {
   console.log(\`🚀 Server running on port \${process.env.PORT || 5000}\`);
 });
   `);
+  currentProgress += 0.25;
+  progressBar.update(currentProgress);
 
-  // Step 4: Install dependencies
+  // Step 4: Install dependencies with spinner
   console.log(chalk.blue(`\n📦 Installing dependencies...\n`));
+  spinner.start('Installing packages');
   execSync(`cd ${projectPath} && npm install`, { stdio: 'inherit' });
-
-  console.log(chalk.green(`\n✅ Project created successfully!`));
-  console.log(chalk.yellow(`\nNext steps:`));
-  console.log(chalk.cyan(`  cd ${projectName}`));
-  console.log(chalk.cyan(`  npm run dev`));
-}
+  spinner.stop(true);
+  currentProgress = 1;
+  progressBar.update(currentProgress);
+
+  // Show celebration
+  await new Promise(resolve => setTimeout(resolve, 500));
+  celebrate('RAG App Created Successfully! 🎉');
+
+  // Show next steps after celebration (keeping original format)
+  setTimeout(() => {
+    console.log(chalk.green(`\n✅ Project created successfully!`));
+    console.log(chalk.yellow(`\nNext steps:`));
+    console.log(chalk.cyan(`  cd ${projectName}`));
+    console.log(chalk.cyan(`  npm run dev`));
+  }, 2500);
+}

src/cli/progressBar.js

@@ -0,0 +1,39 @@
+// src/cli/progressBar.js
+import chalk from 'chalk';
+
+const funFacts = [
+  "Did you know? Vector search helps find similar items even if they use different words!",
+  "MongoDB Atlas Vector Search uses cosine similarity by default 📐",
+  "RAG helps combine the power of vector search with your own data 🔋",
+  "Vector embeddings can capture semantic meaning beyond keywords 🎯",
+  "MongoDB can handle billions of vectors efficiently! 🚀",
+  "Vector search is like giving your database a human-like understanding 🧠"
+];
+
+class FunProgressBar {
+  constructor() {
+    this.width = 40;
+    this.currentFact = 0;
+  }
+
+  update(progress) {
+    const filled = Math.round(this.width * progress);
+    const empty = this.width - filled;
+    
+    const filledBar = '█'.repeat(filled);
+    const emptyBar = '░'.repeat(empty);
+    
+    process.stdout.clearLine();
+    process.stdout.cursorTo(0);
+    
+    const percentage = Math.round(progress * 100);
+    process.stdout.write(
+      chalk.blue(`[${filledBar}${emptyBar}] ${percentage}%\n`) +
+      chalk.yellow(`Fun Fact: ${funFacts[this.currentFact]}\n`)
+    );
+    
+    this.currentFact = (this.currentFact + 1) % funFacts.length;
+  }
+}
+
+export default FunProgressBar;

src/cli/spinner.js

@@ -0,0 +1,56 @@
+// src/cli/spinner.js
+import chalk from 'chalk';
+
+class MongoSpinner {
+  constructor() {
+    this.frames = [
+      '🔍 ○ ○ ○ ',  // Search frame
+      '🔍 ● ○ ○ ',  // Loading embeddings
+      '🔍 ● ● ○ ',  // Processing vectors
+      '🔍 ● ● ● ',  // Complete
+    ];
+    this.interval = null;
+    this.frameIndex = 0;
+    this.messages = [
+      'Preparing vector magic...',
+      'Calculating embeddings...',
+      'Optimizing search space...',
+      'Almost there...'
+    ];
+  }
+
+  start(text = '') {
+    process.stdout.write('\n');
+    this.interval = setInterval(() => {
+      const frame = this.frames[this.frameIndex];
+      const message = this.messages[this.frameIndex];
+      process.stdout.clearLine();
+      process.stdout.cursorTo(0);
+      process.stdout.write(
+        chalk.blue(frame) + ' ' + 
+        chalk.cyan(message) + ' ' + 
+        chalk.yellow(text)
+      );
+      this.frameIndex = (this.frameIndex + 1) % this.frames.length;
+    }, 800);
+  }
+
+  stop(success = true) {
+    clearInterval(this.interval);
+    process.stdout.clearLine();
+    process.stdout.cursorTo(0);
+    if (success) {
+      process.stdout.write(
+        chalk.green('✨ Vector magic complete! ') + 
+        chalk.blue('🎉\n')
+      );
+    } else {
+      process.stdout.write(
+        chalk.red('✖ Operation failed ') + 
+        chalk.yellow('😢\n')
+      );
+    }
+  }
+}
+
+export default MongoSpinner;