@vibe-db/cli 1.2.1 → 1.3.0

package/README.md

@@ -99,6 +99,42 @@ List all your databases.
 
 **Output:**
 - Table showing all databases with ID, Name, Type, and Status
+- Note: Shows databases for your current team only
+
+### `vibedb teams list` (alias: `teams ls`)
+List all teams you belong to.
+
+**Requirements:**
+- Must be logged in
+
+**Output:**
+- Table showing team ID, name, role, type, and current indicator
+- Shows which team is currently selected
+
+### `vibedb teams switch <team_id>`
+Switch to a different team.
+
+**Requirements:**
+- Must be logged in
+- Team ID from `vibedb teams list`
+
+**Example:**
+```bash
+vibedb teams switch team_abc123
+```
+
+**Output:**
+- Confirmation of team switch
+- All subsequent commands use this team context
+
+### `vibedb teams current`
+Show currently selected team.
+
+**Requirements:**
+- Must be logged in
+
+**Output:**
+- Current team details (ID, name, role, type)
 
 ## Configuration
 
@@ -108,10 +144,13 @@ The CLI stores your credentials in `~/.vibedb`:
 {
   "api_key": "vdb_xxx...",
   "email": "you@example.com",
-  "api_url": "https://api.vibedb.dev"
+  "api_url": "https://api.vibedb.dev",
+  "current_team_id": "team_abc123"
 }
 ```
 
+**Team Context:** The CLI remembers your current team selection. Use `vibedb teams switch` to change teams.
+
 **Security Note:** Keep this file secure. Anyone with your API key can manage your databases.
 
 ## Examples
@@ -150,6 +189,50 @@ $ npx @vibedb/cli list
 Total: 1 database
 ```
 
+### Working with Teams
+```bash
+# List all your teams
+$ vibedb teams list
+
+┌──────────────────┬──────────────┬────────┬──────────┬─────────┐
+│ ID               │ Name         │ Role   │ Type     │ Current │
+├──────────────────┼──────────────┼────────┼──────────┼─────────┤
+│ team_abc123      │ John's Team  │ owner  │ Personal │ ✓       │
+│ team_xyz789      │ Acme Corp    │ member │ Team     │         │
+└──────────────────┴──────────────┴────────┴──────────┴─────────┘
+
+Total: 2 teams
+
+# Switch to a different team
+$ vibedb teams switch team_xyz789
+
+✓ Successfully switched to team
+Team: Acme Corp
+Role: member
+
+# List databases in the new team context
+$ vibedb list
+
+┌─────────────────┬────────────────┬──────────┬────────┐
+│ ID              │ Name           │ Type     │ Status │
+├─────────────────┼────────────────┼──────────┼────────┤
+│ db_def456       │ production-db  │ postgres │ ready  │
+│ db_ghi789       │ staging-cache  │ redis    │ ready  │
+└─────────────────┴────────────────┴──────────┴────────┘
+
+Total: 2 databases
+
+# Check current team
+$ vibedb teams current
+
+📍 Current Team
+
+ID: team_xyz789
+Name: Acme Corp
+Role: member
+Type: Team
+```
+
 ## Supported Databases
 
 - **PostgreSQL** - General purpose relational database

package/bin/vibedb.js

@@ -12,6 +12,9 @@ const billingInfoCommand = require('../src/commands/billing-info');
 const billingSubscribeCommand = require('../src/commands/billing-subscribe');
 const billingCancelCommand = require('../src/commands/billing-cancel');
 const billingInvoicesCommand = require('../src/commands/billing-invoices');
+const teamsListCommand = require('../src/commands/teams-list');
+const teamsSwitchCommand = require('../src/commands/teams-switch');
+const teamsCurrentCommand = require('../src/commands/teams-current');
 
 const program = new Command();
 
@@ -134,6 +137,48 @@ billing
     }
   });
 
+// Teams commands
+const teams = program
+  .command('teams')
+  .description('Manage teams and team membership');
+
+teams
+  .command('list')
+  .alias('ls')
+  .description('List all teams you belong to')
+  .action(async () => {
+    try {
+      await teamsListCommand();
+    } catch (error) {
+      console.error(chalk.red('Unexpected error:'), error.message);
+      process.exit(1);
+    }
+  });
+
+teams
+  .command('switch <team_id>')
+  .description('Switch to a different team')
+  .action(async (teamId) => {
+    try {
+      await teamsSwitchCommand(teamId);
+    } catch (error) {
+      console.error(chalk.red('Unexpected error:'), error.message);
+      process.exit(1);
+    }
+  });
+
+teams
+  .command('current')
+  .description('Show current team')
+  .action(async () => {
+    try {
+      await teamsCurrentCommand();
+    } catch (error) {
+      console.error(chalk.red('Unexpected error:'), error.message);
+      process.exit(1);
+    }
+  });
+
 // Global error handler
 process.on('unhandledRejection', (error) => {
   console.error(chalk.red.bold('\n✗ Unhandled error:'), error.message);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibe-db/cli",
-  "version": "1.2.1",
+  "version": "1.3.0",
   "description": "Command-line interface for VibeDB - instant database provisioning for AI-assisted development",
   "main": "bin/vibedb.js",
   "bin": {

package/src/api.js

@@ -43,9 +43,13 @@ async function login(email, password) {
 /**
  * List all databases for authenticated user
  */
-async function listDatabases(apiKey) {
+async function listDatabases(apiKey, teamId = null) {
   try {
-    const response = await axios.get(`${API_BASE_URL}/v1/databases`, {
+    const url = teamId 
+      ? `${API_BASE_URL}/v1/databases?team_id=${teamId}`
+      : `${API_BASE_URL}/v1/databases`;
+    
+    const response = await axios.get(url, {
       headers: {
         Authorization: `Bearer ${apiKey}`,
       },
@@ -78,9 +82,13 @@ async function getPromptFile() {
 /**
  * Get account information
  */
-async function getAccount(apiKey) {
+async function getAccount(apiKey, teamId = null) {
   try {
-    const response = await axios.get(`${API_BASE_URL}/v1/account`, {
+    const url = teamId 
+      ? `${API_BASE_URL}/v1/account?team_id=${teamId}`
+      : `${API_BASE_URL}/v1/account`;
+    
+    const response = await axios.get(url, {
       headers: {
         Authorization: `Bearer ${apiKey}`,
       },
@@ -215,6 +223,46 @@ async function pollDeviceAuth(deviceCode) {
   }
 }
 
+/**
+ * List all teams for authenticated user
+ */
+async function listTeams(apiKey) {
+  try {
+    const response = await axios.get(`${API_BASE_URL}/v1/teams`, {
+      headers: {
+        Authorization: `Bearer ${apiKey}`,
+      },
+    });
+    return response.data.teams || [];
+  } catch (error) {
+    if (error.response) {
+      const { error: errorCode, message } = error.response.data;
+      throw new Error(message || errorCode || 'Failed to list teams');
+    }
+    throw new Error(`Network error: ${error.message}`);
+  }
+}
+
+/**
+ * Get team details
+ */
+async function getTeam(apiKey, teamId) {
+  try {
+    const response = await axios.get(`${API_BASE_URL}/v1/teams/${teamId}`, {
+      headers: {
+        Authorization: `Bearer ${apiKey}`,
+      },
+    });
+    return response.data;
+  } catch (error) {
+    if (error.response) {
+      const { error: errorCode, message } = error.response.data;
+      throw new Error(message || errorCode || 'Failed to get team');
+    }
+    throw new Error(`Network error: ${error.message}`);
+  }
+}
+
 module.exports = {
   signup,
   login,
@@ -227,4 +275,6 @@ module.exports = {
   getInvoices,
   startDeviceAuth,
   pollDeviceAuth,
+  listTeams,
+  getTeam,
 };

package/src/commands/list.js

@@ -15,10 +15,11 @@ async function listCommand() {
     }
 
     const apiKey = config.getAPIKey();
+    const currentTeamId = config.getCurrentTeamId();
 
     console.log(chalk.gray('Fetching databases...'));
 
-    const databases = await api.listDatabases(apiKey);
+    const databases = await api.listDatabases(apiKey, currentTeamId);
 
     if (databases.length === 0) {
       console.log(chalk.yellow('\nNo databases found.'));

package/src/commands/login.js

@@ -54,6 +54,21 @@ async function loginWithGitHub() {
         if (pollData.status === 'complete') {
           // Save API key
           config.saveAuth(pollData.api_key, pollData.email);
+          
+          // Auto-select first team (usually personal team)
+          try {
+            const teams = await api.listTeams(pollData.api_key);
+            if (teams.length > 0) {
+              // Prefer personal team, otherwise first team
+              const personalTeam = teams.find(t => t.is_personal);
+              const defaultTeam = personalTeam || teams[0];
+              config.setCurrentTeamId(defaultTeam.id);
+            }
+          } catch (err) {
+            // Non-fatal, just log
+            console.log(chalk.gray('Note: Could not auto-select team'));
+          }
+          
           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()}`));
@@ -116,6 +131,20 @@ async function loginWithEmail() {
     // Save API key to config
     config.saveAuth(result.api_key, result.email);
 
+    // Auto-select first team (usually personal team)
+    try {
+      const teams = await api.listTeams(result.api_key);
+      if (teams.length > 0) {
+        // Prefer personal team, otherwise first team
+        const personalTeam = teams.find(t => t.is_personal);
+        const defaultTeam = personalTeam || teams[0];
+        config.setCurrentTeamId(defaultTeam.id);
+      }
+    } catch (err) {
+      // Non-fatal, just log
+      console.log(chalk.gray('Note: Could not auto-select team'));
+    }
+
     console.log(chalk.green.bold('\n✓ Login successful!\n'));
     console.log(chalk.white('Email:'), chalk.cyan(result.email));
     console.log(chalk.gray(`API key saved to ${config.getConfigPath()}`));

package/src/commands/signup.js

@@ -18,15 +18,19 @@ async function signupWithGitHub() {
     const data = await api.startDeviceAuth();
     const { device_code, user_code, verification_uri, interval } = data;
     
+    // Build URL with pre-populated code for better UX
+    const urlWithCode = `${verification_uri}?code=${formatUserCode(user_code)}`;
+    
     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.cyan.bold(`  Open:       ${urlWithCode}`));
+    console.log(chalk.yellow.bold(`  Your code:  ${formatUserCode(user_code)}`));
     console.log(chalk.bold('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n'));
     
+    // Try to open browser automatically with pre-populated code
     try {
       const open = (await import('open')).default;
-      await open(verification_uri);
-      console.log(chalk.gray('Opened browser automatically...\n'));
+      await open(urlWithCode);
+      console.log(chalk.gray('Opened browser with code pre-filled...\n'));
     } catch (err) {
       console.log(chalk.gray('Please open the URL manually\n'));
     }
@@ -46,6 +50,19 @@ async function signupWithGitHub() {
         
         if (pollData.status === 'complete') {
           config.saveAuth(pollData.api_key, pollData.email);
+          
+          // Auto-select first team (usually personal team)
+          try {
+            const teams = await api.listTeams(pollData.api_key);
+            if (teams.length > 0) {
+              const personalTeam = teams.find(t => t.is_personal);
+              const defaultTeam = personalTeam || teams[0];
+              config.setCurrentTeamId(defaultTeam.id);
+            }
+          } catch (err) {
+            // Non-fatal
+          }
+          
           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()}`));
@@ -124,6 +141,18 @@ async function signupWithEmail() {
 
     config.saveAuth(result.api_key, result.email);
 
+    // Auto-select first team (usually personal team)
+    try {
+      const teams = await api.listTeams(result.api_key);
+      if (teams.length > 0) {
+        const personalTeam = teams.find(t => t.is_personal);
+        const defaultTeam = personalTeam || teams[0];
+        config.setCurrentTeamId(defaultTeam.id);
+      }
+    } catch (err) {
+      // Non-fatal
+    }
+
     console.log(chalk.green.bold('\n✓ Account created successfully!\n'));
     console.log(chalk.white('Email:'), chalk.cyan(result.email));
     console.log(chalk.gray(`API key saved to ${config.getConfigPath()}`));

package/src/commands/teams-current.js

@@ -0,0 +1,42 @@
+const chalk = require('chalk');
+const api = require('../api');
+const config = require('../config');
+
+async function teamsCurrentCommand() {
+  try {
+    // Check if logged in
+    if (!config.isLoggedIn()) {
+      console.error(chalk.red.bold('✗ Not logged in'));
+      console.log(chalk.gray('\nPlease run'), chalk.cyan('vibedb login'), chalk.gray('or'), chalk.cyan('vibedb signup'), chalk.gray('first.'));
+      process.exit(1);
+    }
+
+    const apiKey = config.getAPIKey();
+    const currentTeamId = config.getCurrentTeamId();
+
+    if (!currentTeamId) {
+      console.log(chalk.yellow('\nNo team currently selected.'));
+      console.log(chalk.gray('\nSwitch to a team with:'), chalk.cyan('vibedb teams switch <team_id>'));
+      console.log(chalk.gray('List your teams with:'), chalk.cyan('vibedb teams list'));
+      console.log();
+      return;
+    }
+
+    console.log(chalk.gray('Fetching current team...'));
+
+    const team = await api.getTeam(apiKey, currentTeamId);
+
+    console.log(chalk.blue.bold('\n📍 Current Team\n'));
+    console.log(chalk.gray('ID:'), chalk.cyan(team.id));
+    console.log(chalk.gray('Name:'), chalk.white(team.name));
+    console.log(chalk.gray('Role:'), chalk.blue(team.role));
+    console.log(chalk.gray('Type:'), team.is_personal ? chalk.gray('Personal') : chalk.white('Team'));
+    console.log();
+  } catch (error) {
+    console.error(chalk.red.bold('\n✗ Failed to get current team:'), chalk.red(error.message));
+    console.log(chalk.gray('\nThe team may have been deleted. Switch to a different team with:'), chalk.cyan('vibedb teams switch <team_id>'));
+    process.exit(1);
+  }
+}
+
+module.exports = teamsCurrentCommand;

package/src/commands/teams-list.js

@@ -0,0 +1,84 @@
+const chalk = require('chalk');
+const Table = require('cli-table3');
+const api = require('../api');
+const config = require('../config');
+
+async function teamsListCommand() {
+  console.log(chalk.blue.bold('\n👥 Your Teams\n'));
+
+  try {
+    // Check if logged in
+    if (!config.isLoggedIn()) {
+      console.error(chalk.red.bold('✗ Not logged in'));
+      console.log(chalk.gray('\nPlease run'), chalk.cyan('vibedb login'), chalk.gray('or'), chalk.cyan('vibedb signup'), chalk.gray('first.'));
+      process.exit(1);
+    }
+
+    const apiKey = config.getAPIKey();
+    const currentTeamId = config.getCurrentTeamId();
+
+    console.log(chalk.gray('Fetching teams...'));
+
+    const teams = await api.listTeams(apiKey);
+
+    if (teams.length === 0) {
+      console.log(chalk.yellow('\nNo teams found.'));
+      console.log();
+      return;
+    }
+
+    // Create table
+    const table = new Table({
+      head: [
+        chalk.white.bold('ID'),
+        chalk.white.bold('Name'),
+        chalk.white.bold('Role'),
+        chalk.white.bold('Type'),
+        chalk.white.bold('Current'),
+      ],
+      colWidths: [30, 25, 12, 12, 10],
+      style: {
+        head: [],
+        border: ['gray'],
+      },
+    });
+
+    teams.forEach((team) => {
+      const isCurrent = team.id === currentTeamId;
+      const currentMark = isCurrent ? chalk.green('✓') : '';
+      const teamName = team.is_personal 
+        ? chalk.gray(`${team.name} (Personal)`)
+        : team.name;
+
+      table.push([
+        chalk.cyan(team.id),
+        teamName,
+        chalk.blue(team.role),
+        team.is_personal ? chalk.gray('Personal') : chalk.white('Team'),
+        currentMark,
+      ]);
+    });
+
+    console.log();
+    console.log(table.toString());
+    console.log();
+    
+    if (currentTeamId) {
+      const currentTeam = teams.find(t => t.id === currentTeamId);
+      if (currentTeam) {
+        console.log(chalk.gray(`Current team: ${currentTeam.name}`));
+      }
+    } else {
+      console.log(chalk.yellow('No team selected. Use'), chalk.cyan('vibedb teams switch <team_id>'), chalk.yellow('to select a team.'));
+    }
+    
+    console.log();
+    console.log(chalk.gray(`Total: ${teams.length} team${teams.length === 1 ? '' : 's'}`));
+    console.log();
+  } catch (error) {
+    console.error(chalk.red.bold('\n✗ Failed to list teams:'), chalk.red(error.message));
+    process.exit(1);
+  }
+}
+
+module.exports = teamsListCommand;

package/src/commands/teams-switch.js

@@ -0,0 +1,41 @@
+const chalk = require('chalk');
+const api = require('../api');
+const config = require('../config');
+
+async function teamsSwitchCommand(teamId) {
+  try {
+    // Check if logged in
+    if (!config.isLoggedIn()) {
+      console.error(chalk.red.bold('✗ Not logged in'));
+      console.log(chalk.gray('\nPlease run'), chalk.cyan('vibedb login'), chalk.gray('or'), chalk.cyan('vibedb signup'), chalk.gray('first.'));
+      process.exit(1);
+    }
+
+    if (!teamId) {
+      console.error(chalk.red.bold('✗ Team ID required'));
+      console.log(chalk.gray('\nUsage:'), chalk.cyan('vibedb teams switch <team_id>'));
+      console.log(chalk.gray('\nList your teams with:'), chalk.cyan('vibedb teams list'));
+      process.exit(1);
+    }
+
+    const apiKey = config.getAPIKey();
+
+    console.log(chalk.gray(`Switching to team ${teamId}...`));
+
+    // Verify team exists and user has access
+    const team = await api.getTeam(apiKey, teamId);
+
+    // Save as current team
+    config.setCurrentTeamId(teamId);
+
+    console.log(chalk.green.bold('\n✓ Successfully switched to team'));
+    console.log(chalk.gray('Team:'), chalk.white(team.name));
+    console.log(chalk.gray('Role:'), chalk.blue(team.role));
+    console.log();
+  } catch (error) {
+    console.error(chalk.red.bold('\n✗ Failed to switch team:'), chalk.red(error.message));
+    process.exit(1);
+  }
+}
+
+module.exports = teamsSwitchCommand;

package/src/config.js

@@ -68,6 +68,23 @@ function saveAuth(apiKey, email) {
   writeConfig(config);
 }
 
+/**
+ * Get current team ID from config
+ */
+function getCurrentTeamId() {
+  const config = readConfig();
+  return config?.current_team_id || null;
+}
+
+/**
+ * Set current team ID in config
+ */
+function setCurrentTeamId(teamId) {
+  const config = readConfig() || {};
+  config.current_team_id = teamId;
+  writeConfig(config);
+}
+
 module.exports = {
   getConfigPath,
   readConfig,
@@ -75,4 +92,6 @@ module.exports = {
   isLoggedIn,
   getAPIKey,
   saveAuth,
+  getCurrentTeamId,
+  setCurrentTeamId,
 };