@vibe-db/cli 1.0.0 → 1.1.0

package/bin/vibedb.js

@@ -4,6 +4,7 @@ const { Command } = require('commander');
 const chalk = require('chalk');
 const signupCommand = require('../src/commands/signup');
 const loginCommand = require('../src/commands/login');
+const logoutCommand = require('../src/commands/logout');
 const initCommand = require('../src/commands/init');
 const listCommand = require('../src/commands/list');
 const billingInfoCommand = require('../src/commands/billing-info');
@@ -16,7 +17,7 @@ const program = new Command();
 program
   .name('vibedb')
   .description('VibeDB CLI - Instant database provisioning for AI-assisted development')
-  .version('1.0.0');
+  .version('1.1.0');
 
 program
   .command('signup')
@@ -42,6 +43,18 @@ program
     }
   });
 
+program
+  .command('logout')
+  .description('Logout and remove saved credentials')
+  .action(async () => {
+    try {
+      await logoutCommand();
+    } catch (error) {
+      console.error(chalk.red('Unexpected error:'), error.message);
+      process.exit(1);
+    }
+  });
+
 program
   .command('init')
   .description('Download VIBEDB.md prompt file to your project')

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibe-db/cli",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "Command-line interface for VibeDB - instant database provisioning for AI-assisted development",
   "main": "bin/vibedb.js",
   "bin": {
@@ -27,7 +27,8 @@
     "chalk": "^4.1.2",
     "cli-table3": "^0.6.3",
     "commander": "^11.0.0",
-    "inquirer": "^8.2.5"
+    "inquirer": "^8.2.5",
+    "open": "^10.0.0"
   },
   "engines": {
     "node": ">=18.0.0"

package/src/api.js

@@ -183,6 +183,38 @@ async function getInvoices(apiKey) {
   }
 }
 
+/**
+ * Start device authorization flow
+ */
+async function startDeviceAuth() {
+  try {
+    const response = await axios.post(`${API_BASE_URL}/v1/auth/device`);
+    return response.data;
+  } catch (error) {
+    if (error.response) {
+      const { error: errorCode, message } = error.response.data;
+      throw new Error(message || errorCode || 'Failed to start device authorization');
+    }
+    throw new Error(`Network error: ${error.message}`);
+  }
+}
+
+/**
+ * Poll device authorization status
+ */
+async function pollDeviceAuth(deviceCode) {
+  try {
+    const response = await axios.get(`${API_BASE_URL}/v1/auth/device/${deviceCode}/poll`);
+    return response.data;
+  } catch (error) {
+    if (error.response) {
+      const { error: errorCode, message } = error.response.data;
+      throw new Error(message || errorCode || 'Failed to poll device authorization');
+    }
+    throw new Error(`Network error: ${error.message}`);
+  }
+}
+
 module.exports = {
   signup,
   login,
@@ -193,4 +225,6 @@ module.exports = {
   createSubscription,
   cancelSubscription,
   getInvoices,
+  startDeviceAuth,
+  pollDeviceAuth,
 };

package/src/commands/init.js

@@ -46,12 +46,21 @@ async function initCommand() {
 
     console.log(chalk.green.bold('\n✓ VIBEDB.md downloaded successfully!\n'));
     console.log(chalk.white('Location:'), chalk.cyan(promptFilePath));
-    console.log(chalk.white('API Key:'), chalk.cyan(userConfig.api_key));
     console.log(chalk.white('Email:'), chalk.cyan(userConfig.email));
-    console.log(chalk.white('\nNext steps:'));
-    console.log(chalk.gray('  1. Add VIBEDB.md to your project repository'));
-    console.log(chalk.gray('  2. Tell your AI assistant: "I need a database for my app"'));
-    console.log(chalk.gray('  3. The AI will read VIBEDB.md and provision a database for you!'));
+    
+    console.log(chalk.yellow('\n⚠️  Security reminder:'));
+    console.log(chalk.gray('  • Your API key is embedded in VIBEDB.md'));
+    console.log(chalk.gray('  • Keep it secure - don\'t share or commit to public repos'));
+    console.log(chalk.gray('  • Add VIBEDB.md to .gitignore if needed'));
+    
+    console.log(chalk.white('\nHow to use:'));
+    console.log(chalk.gray('  1. Tell your AI assistant: '), chalk.cyan('"Read VIBEDB.md and provision a database"'));
+    console.log(chalk.gray('  2. The AI will read the file and create a database for you'));
+    console.log(chalk.gray('  3. You\'ll get connection details instantly!'));
+    
+    console.log(chalk.white('\nNeed help?'));
+    console.log(chalk.gray('  • Visit: '), chalk.cyan('https://vibedb.dev/docs'));
+    console.log(chalk.gray('  • See examples in VIBEDB.md'));
     console.log();
   } catch (error) {
     console.error(chalk.red.bold('\n✗ Init failed:'), chalk.red(error.message));

package/src/commands/login.js

@@ -3,9 +3,82 @@ const chalk = require('chalk');
 const api = require('../api');
 const config = require('../config');
 
-async function loginCommand() {
-  console.log(chalk.blue.bold('\n🔐 VibeDB Login\n'));
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
 
+function formatUserCode(code) {
+  // Format as XXXX-XXXX for readability
+  return `${code.slice(0, 4)}-${code.slice(4)}`;
+}
+
+async function loginWithGitHub() {
+  console.log(chalk.blue('Starting GitHub authentication...\n'));
+  
+  try {
+    // 1. Start device auth
+    const data = await api.startDeviceAuth();
+    const { device_code, user_code, verification_uri, interval } = data;
+    
+    console.log(chalk.bold('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━'));
+    console.log(chalk.cyan.bold(`  Open:       ${verification_uri}`));
+    console.log(chalk.yellow.bold(`  Enter code: ${formatUserCode(user_code)}`));
+    console.log(chalk.bold('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n'));
+    
+    // Try to open browser automatically
+    try {
+      const open = (await import('open')).default;
+      await open(verification_uri);
+      console.log(chalk.gray('Opened browser automatically...\n'));
+    } catch (err) {
+      console.log(chalk.gray('Please open the URL manually\n'));
+    }
+    
+    console.log(chalk.gray('Waiting for authorization...'));
+    
+    // 2. Poll for completion
+    const pollInterval = (interval || 5) * 1000;
+    let attempts = 0;
+    const maxAttempts = 180 / (interval || 5); // 15 minutes / interval
+    
+    while (attempts < maxAttempts) {
+      await sleep(pollInterval);
+      attempts++;
+      
+      try {
+        const pollData = await api.pollDeviceAuth(device_code);
+        
+        if (pollData.status === 'complete') {
+          // Save API key
+          config.saveAuth(pollData.api_key, pollData.email);
+          console.log(chalk.green.bold('\n✓ Login successful!\n'));
+          console.log(chalk.white('Email:'), chalk.cyan(pollData.email));
+          console.log(chalk.gray(`API key saved to ${config.getConfigPath()}`));
+          console.log();
+          return;
+        }
+        
+        // Still pending, continue polling
+        process.stdout.write(chalk.gray('.'));
+      } catch (err) {
+        if (err.response?.data?.error === 'authorization_expired') {
+          console.error(chalk.red.bold('\n\n✗ Authorization expired. Please try again.'));
+          process.exit(1);
+        }
+        // Other errors, continue polling
+      }
+    }
+    
+    console.error(chalk.red.bold('\n\n✗ Authorization timed out. Please try again.'));
+    process.exit(1);
+    
+  } catch (error) {
+    console.error(chalk.red.bold('\n✗ GitHub login failed:'), chalk.red(error.message));
+    process.exit(1);
+  }
+}
+
+async function loginWithEmail() {
   try {
     const answers = await inquirer.prompt([
       {
@@ -50,4 +123,37 @@ async function loginCommand() {
   }
 }
 
+async function loginCommand() {
+  console.log(chalk.blue.bold('\n🔐 VibeDB Login\n'));
+
+  try {
+    const { method } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'method',
+        message: 'How would you like to log in?',
+        choices: [
+          { name: '🐙 GitHub (Recommended)', value: 'github' },
+          { name: '📧 Email & Password', value: 'email' },
+        ],
+      },
+    ]);
+
+    console.log(); // Empty line for spacing
+
+    if (method === 'github') {
+      await loginWithGitHub();
+    } else {
+      await loginWithEmail();
+    }
+  } catch (error) {
+    if (error.isTtyError) {
+      console.error(chalk.red('Prompt couldn\'t be rendered in the current environment'));
+    } else {
+      console.error(chalk.red('An unexpected error occurred:', error.message));
+    }
+    process.exit(1);
+  }
+}
+
 module.exports = loginCommand;

package/src/commands/logout.js

@@ -0,0 +1,47 @@
+const fs = require('fs');
+const chalk = require('chalk');
+const inquirer = require('inquirer');
+const config = require('../config');
+
+async function logoutCommand() {
+  console.log(chalk.blue.bold('\n👋 VibeDB Logout\n'));
+
+  try {
+    // Check if logged in
+    if (!config.isLoggedIn()) {
+      console.log(chalk.yellow('You are not currently logged in.'));
+      process.exit(0);
+    }
+
+    const userConfig = config.readConfig();
+    
+    // Confirm logout
+    const { confirm } = await inquirer.prompt([
+      {
+        type: 'confirm',
+        name: 'confirm',
+        message: `Log out of ${userConfig.email}?`,
+        default: true,
+      },
+    ]);
+
+    if (!confirm) {
+      console.log(chalk.yellow('\nLogout cancelled.'));
+      process.exit(0);
+    }
+
+    // Delete config file
+    const configPath = config.getConfigPath();
+    fs.unlinkSync(configPath);
+
+    console.log(chalk.green.bold('\n✓ Logged out successfully!\n'));
+    console.log(chalk.gray('Your API key has been removed from this device.'));
+    console.log(chalk.gray('To use VibeDB again, run:'), chalk.cyan('vibedb login'), chalk.gray('or'), chalk.cyan('vibedb signup'));
+    console.log();
+  } catch (error) {
+    console.error(chalk.red.bold('\n✗ Logout failed:'), chalk.red(error.message));
+    process.exit(1);
+  }
+}
+
+module.exports = logoutCommand;

package/src/commands/signup.js

@@ -3,9 +3,79 @@ const chalk = require('chalk');
 const api = require('../api');
 const config = require('../config');
 
-async function signupCommand() {
-  console.log(chalk.blue.bold('\n📝 VibeDB Signup\n'));
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+function formatUserCode(code) {
+  return `${code.slice(0, 4)}-${code.slice(4)}`;
+}
+
+async function signupWithGitHub() {
+  console.log(chalk.blue('Starting GitHub authentication...\n'));
+  
+  try {
+    const data = await api.startDeviceAuth();
+    const { device_code, user_code, verification_uri, interval } = data;
+    
+    console.log(chalk.bold('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━'));
+    console.log(chalk.cyan.bold(`  Open:       ${verification_uri}`));
+    console.log(chalk.yellow.bold(`  Enter code: ${formatUserCode(user_code)}`));
+    console.log(chalk.bold('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n'));
+    
+    try {
+      const open = (await import('open')).default;
+      await open(verification_uri);
+      console.log(chalk.gray('Opened browser automatically...\n'));
+    } catch (err) {
+      console.log(chalk.gray('Please open the URL manually\n'));
+    }
+    
+    console.log(chalk.gray('Waiting for authorization...'));
+    
+    const pollInterval = (interval || 5) * 1000;
+    let attempts = 0;
+    const maxAttempts = 180 / (interval || 5);
+    
+    while (attempts < maxAttempts) {
+      await sleep(pollInterval);
+      attempts++;
+      
+      try {
+        const pollData = await api.pollDeviceAuth(device_code);
+        
+        if (pollData.status === 'complete') {
+          config.saveAuth(pollData.api_key, pollData.email);
+          console.log(chalk.green.bold('\n✓ Account created successfully!\n'));
+          console.log(chalk.white('Email:'), chalk.cyan(pollData.email));
+          console.log(chalk.gray(`API key saved to ${config.getConfigPath()}`));
+          console.log(chalk.white('\nNext steps:'));
+          console.log(chalk.gray('  1. Run'), chalk.cyan('vibedb init'), chalk.gray('to download the prompt file'));
+          console.log(chalk.gray('  2. Add VIBEDB.md to your project'));
+          console.log(chalk.gray('  3. Tell your AI assistant to provision a database!'));
+          console.log();
+          return;
+        }
+        
+        process.stdout.write(chalk.gray('.'));
+      } catch (err) {
+        if (err.response?.data?.error === 'authorization_expired') {
+          console.error(chalk.red.bold('\n\n✗ Authorization expired. Please try again.'));
+          process.exit(1);
+        }
+      }
+    }
+    
+    console.error(chalk.red.bold('\n\n✗ Authorization timed out. Please try again.'));
+    process.exit(1);
+    
+  } catch (error) {
+    console.error(chalk.red.bold('\n✗ GitHub signup failed:'), chalk.red(error.message));
+    process.exit(1);
+  }
+}
 
+async function signupWithEmail() {
   try {
     const answers = await inquirer.prompt([
       {
@@ -52,13 +122,11 @@ async function signupCommand() {
 
     const result = await api.signup(answers.email.trim().toLowerCase(), answers.password);
 
-    // Save API key to config
     config.saveAuth(result.api_key, result.email);
 
     console.log(chalk.green.bold('\n✓ Account created successfully!\n'));
     console.log(chalk.white('Email:'), chalk.cyan(result.email));
-    console.log(chalk.white('API Key:'), chalk.cyan(result.api_key));
-    console.log(chalk.gray(`\nAPI key saved to ${config.getConfigPath()}`));
+    console.log(chalk.gray(`API key saved to ${config.getConfigPath()}`));
     console.log(chalk.white('\nNext steps:'));
     console.log(chalk.gray('  1. Run'), chalk.cyan('vibedb init'), chalk.gray('to download the prompt file'));
     console.log(chalk.gray('  2. Add VIBEDB.md to your project'));
@@ -70,4 +138,37 @@ async function signupCommand() {
   }
 }
 
+async function signupCommand() {
+  console.log(chalk.blue.bold('\n📝 VibeDB Signup\n'));
+
+  try {
+    const { method } = await inquirer.prompt([
+      {
+        type: 'list',
+        name: 'method',
+        message: 'How would you like to sign up?',
+        choices: [
+          { name: '🐙 GitHub (Recommended)', value: 'github' },
+          { name: '📧 Email & Password', value: 'email' },
+        ],
+      },
+    ]);
+
+    console.log();
+
+    if (method === 'github') {
+      await signupWithGitHub();
+    } else {
+      await signupWithEmail();
+    }
+  } catch (error) {
+    if (error.isTtyError) {
+      console.error(chalk.red('Prompt couldn\'t be rendered in the current environment'));
+    } else {
+      console.error(chalk.red('An unexpected error occurred:', error.message));
+    }
+    process.exit(1);
+  }
+}
+
 module.exports = signupCommand;

package/package.json.bak

@@ -1,35 +0,0 @@
-{
-  "name": "@vibedb/cli",
-  "version": "1.0.0",
-  "description": "Command-line interface for VibeDB - instant database provisioning for AI-assisted development",
-  "main": "bin/vibedb.js",
-  "bin": {
-    "vibedb": "./bin/vibedb.js"
-  },
-  "scripts": {
-    "test": "node bin/vibedb.js --help"
-  },
-  "keywords": [
-    "database",
-    "postgres",
-    "mysql",
-    "redis",
-    "provisioning",
-    "cli",
-    "vibedb",
-    "ai",
-    "development"
-  ],
-  "author": "VibeDB",
-  "license": "MIT",
-  "dependencies": {
-    "axios": "^1.6.0",
-    "chalk": "^4.1.2",
-    "cli-table3": "^0.6.3",
-    "commander": "^11.0.0",
-    "inquirer": "^8.2.5"
-  },
-  "engines": {
-    "node": ">=18.0.0"
-  }
-}