vibeteam 0.2.2 → 0.2.4

package/bin/cli.js

@@ -384,7 +384,71 @@ if (args[0] === 'setup') {
   }
 
   // ==========================================================================
-  // Step 5: Verify and report
+  // Step 5: MCP Server Installation
+  // ==========================================================================
+
+  let mcpInstalled = false
+
+  // Check if MCP is already configured
+  const mcpAlreadyConfigured = (() => {
+    try {
+      const s = JSON.parse(readFileSync(settingsPath, 'utf-8'))
+      return !!s.mcpServers?.vibeteam
+    } catch { return false }
+  })()
+
+  // Also check settings.local.json
+  const mcpInLocal = (() => {
+    try {
+      const localPath = settingsPath.replace('settings.json', 'settings.local.json')
+      if (!existsSync(localPath)) return false
+      const s = JSON.parse(readFileSync(localPath, 'utf-8'))
+      return !!s.mcpServers?.vibeteam
+    } catch { return false }
+  })()
+
+  if (mcpAlreadyConfigured || mcpInLocal) {
+    console.log('\nMCP server: already configured')
+    mcpInstalled = true
+  } else {
+    // Check if claude CLI is available
+    let hasClaudeCli = false
+    try {
+      execSync('which claude', { stdio: 'ignore' })
+      hasClaudeCli = true
+    } catch {}
+
+    if (hasClaudeCli) {
+      console.log('\n  VibeTeam MCP server lets Claude Code control your agents remotely.')
+      const mcpAnswer = await askQuestion('  Install MCP integration for Claude Code? (y/n): ')
+
+      if (mcpAnswer.toLowerCase() === 'y' || mcpAnswer.toLowerCase() === 'yes') {
+        // Determine install method: global npm binary vs local dev path
+        let mcpCommand = 'vibeteam-mcp'
+        try {
+          execSync('which vibeteam-mcp', { stdio: 'ignore' })
+        } catch {
+          // vibeteam-mcp not in PATH — use absolute path to mcp/server.js
+          mcpCommand = resolve(ROOT, 'mcp/server.js')
+        }
+
+        try {
+          execSync(
+            `CLAUDECODE= claude mcp add vibeteam --scope user -- ${mcpCommand}`,
+            { stdio: 'pipe', timeout: 10000 }
+          )
+          console.log('  ✓ MCP server installed for Claude Code')
+          mcpInstalled = true
+        } catch (e) {
+          console.log(`  ✗ Failed to install MCP server: ${e.message}`)
+          console.log(`    You can install manually: claude mcp add vibeteam -- ${mcpCommand}`)
+        }
+      }
+    }
+  }
+
+  // ==========================================================================
+  // Step 6: Verify and report
   // ==========================================================================
 
   console.log('\n' + '='.repeat(50))
@@ -401,6 +465,10 @@ if (args[0] === 'setup') {
   console.log('  - UserPromptSubmit')
   console.log('  - Notification')
 
+  if (mcpInstalled) {
+    console.log('\nMCP server: ✓ configured')
+  }
+
   // Check dependencies
   let hasWarnings = false
 
@@ -866,7 +934,221 @@ console.log(`
   ╰─────────────────────────────────────╯
 `)
 
-// Run health checks
+// ============================================================================
+// Critical Dependency Check (blocks startup if missing)
+// ============================================================================
+
+function detectPackageManager() {
+  const platform = process.platform
+
+  if (platform === 'darwin') {
+    // macOS - check for brew
+    try {
+      execSync('which brew', { stdio: 'ignore' })
+      return { name: 'brew', install: (pkg) => `brew install ${pkg}` }
+    } catch {
+      return null
+    }
+  } else if (platform === 'linux') {
+    // Linux - check for apt, then yum/dnf
+    try {
+      execSync('which apt', { stdio: 'ignore' })
+      return { name: 'apt', install: (pkg) => `sudo apt install -y ${pkg}` }
+    } catch {
+      try {
+        execSync('which dnf', { stdio: 'ignore' })
+        return { name: 'dnf', install: (pkg) => `sudo dnf install -y ${pkg}` }
+      } catch {
+        try {
+          execSync('which yum', { stdio: 'ignore' })
+          return { name: 'yum', install: (pkg) => `sudo yum install -y ${pkg}` }
+        } catch {
+          return null
+        }
+      }
+    }
+  }
+  return null
+}
+
+async function installDependency(pkgManager, depName) {
+  const cmd = pkgManager.install(depName)
+  console.log(`  Running: ${cmd}`)
+  console.log()
+
+  try {
+    execSync(cmd, { stdio: 'inherit' })
+    return true
+  } catch (e) {
+    console.log(`  ✗ Failed to install ${depName}`)
+    return false
+  }
+}
+
+async function checkRequiredDependencies() {
+  const missing = []
+
+  // Check tmux (REQUIRED - agents won't spawn without it)
+  if (!checkTmux()) {
+    missing.push({
+      name: 'tmux',
+      reason: 'Required for running Claude agents in background sessions',
+    })
+  }
+
+  // Check jq (REQUIRED - hooks won't work without it)
+  if (!checkJq()) {
+    missing.push({
+      name: 'jq',
+      reason: 'Required for Claude Code hooks to capture events',
+    })
+  }
+
+  if (missing.length === 0) {
+    return true // All good
+  }
+
+  // Show what's missing
+  console.log('  ╭─────────────────────────────────────╮')
+  console.log('  │  Missing Required Dependencies     │')
+  console.log('  ╰─────────────────────────────────────╯')
+  console.log()
+
+  for (const dep of missing) {
+    console.log(`  ✗ ${dep.name} - ${dep.reason}`)
+  }
+  console.log()
+
+  // Detect package manager
+  const pkgManager = detectPackageManager()
+
+  if (pkgManager) {
+    const answer = await askQuestion(`  Install missing dependencies using ${pkgManager.name}? (y/n): `)
+
+    if (answer.toLowerCase() === 'y' || answer.toLowerCase() === 'yes') {
+      console.log()
+
+      let allInstalled = true
+      for (const dep of missing) {
+        const success = await installDependency(pkgManager, dep.name)
+        if (!success) {
+          allInstalled = false
+        }
+      }
+
+      if (allInstalled) {
+        console.log()
+        console.log('  ✓ All dependencies installed successfully!')
+        console.log()
+        return true
+      } else {
+        console.log()
+        console.log('  Some dependencies failed to install. Please install them manually.')
+        process.exit(1)
+      }
+    }
+  } else {
+    // No package manager detected - show manual instructions
+    console.log('  Could not detect package manager. Please install manually:')
+    console.log()
+    console.log('    macOS:  brew install tmux jq')
+    console.log('    Ubuntu: sudo apt install tmux jq')
+    console.log('    Fedora: sudo dnf install tmux jq')
+    console.log()
+  }
+
+  // Ask if they want to continue anyway
+  const continueAnswer = await askQuestion('  Continue without these dependencies? (y/n): ')
+
+  if (continueAnswer.toLowerCase() === 'y' || continueAnswer.toLowerCase() === 'yes') {
+    console.log()
+    console.log('  ⚠ Continuing without all dependencies - some features may not work!')
+    console.log()
+    return true
+  }
+
+  console.log()
+  console.log('  Install the missing dependencies and run vibeteam again.')
+  console.log()
+  process.exit(1)
+}
+
+// Check dependencies before starting
+await checkRequiredDependencies()
+
+// ============================================================================
+// First-run MCP install offer
+// ============================================================================
+
+async function offerMcpInstall() {
+  // Check if we already offered (don't nag on every start)
+  const flagFile = join(homedir(), '.vibeteam', '.mcp-offered')
+  if (existsSync(flagFile)) return
+
+  // Check if MCP is already configured in settings or settings.local
+  const settingsPaths = [
+    join(homedir(), '.claude', 'settings.json'),
+    join(homedir(), '.claude', 'settings.local.json'),
+    join(homedir(), '.config', 'claude', 'settings.json'),
+    join(homedir(), '.config', 'claude', 'settings.local.json'),
+  ]
+
+  for (const sp of settingsPaths) {
+    try {
+      if (!existsSync(sp)) continue
+      const s = JSON.parse(readFileSync(sp, 'utf-8'))
+      if (s.mcpServers?.vibeteam) return // Already configured
+    } catch {}
+  }
+
+  // Check if claude CLI is available
+  try {
+    execSync('which claude', { stdio: 'ignore' })
+  } catch {
+    return // No claude CLI, skip
+  }
+
+  console.log('  VibeTeam MCP server lets Claude Code control your agents remotely.')
+  const answer = await askQuestion('  Install MCP integration for Claude Code? (y/n): ')
+
+  // Ensure .vibeteam dir exists for the flag file
+  const vibeteamDir = join(homedir(), '.vibeteam')
+  if (!existsSync(vibeteamDir)) {
+    mkdirSync(vibeteamDir, { recursive: true })
+  }
+
+  if (answer.toLowerCase() === 'y' || answer.toLowerCase() === 'yes') {
+    let mcpCommand = 'vibeteam-mcp'
+    try {
+      execSync('which vibeteam-mcp', { stdio: 'ignore' })
+    } catch {
+      mcpCommand = resolve(ROOT, 'mcp/server.js')
+    }
+
+    try {
+      execSync(
+        `CLAUDECODE= claude mcp add vibeteam --scope user -- ${mcpCommand}`,
+        { stdio: 'pipe', timeout: 10000 }
+      )
+      console.log('  ✓ MCP server installed for Claude Code\n')
+    } catch (e) {
+      console.log(`  ✗ Failed to install MCP: ${e.message}`)
+      console.log(`    Install manually: claude mcp add vibeteam -- ${mcpCommand}\n`)
+    }
+  } else {
+    console.log()
+  }
+
+  // Write flag so we don't ask again
+  const { writeFileSync } = await import('fs')
+  try {
+    writeFileSync(flagFile, new Date().toISOString())
+  } catch {}
+}
+
+await offerMcpInstall()
+
+// Run health checks (warnings only)
 printHealthCheck()
 
 console.log(`Starting server on port ${port}...`)
@@ -882,44 +1164,58 @@ console.log()
 const compiledPath = resolve(ROOT, 'dist/server/server/index.js')
 const sourcePath = resolve(ROOT, 'server/index.ts')
 
-let server
-if (existsSync(compiledPath)) {
-  // Use compiled JS (production/npm install)
-  server = spawn('node', [compiledPath], {
-    cwd: ROOT,
-    env: {
-      ...process.env,
-      VIBETEAM_PORT: port,
-    },
-    stdio: 'inherit',
-  })
-} else {
-  // Fall back to tsx (development)
-  console.log('(dev mode - using tsx)')
-  server = spawn('npx', ['tsx', sourcePath], {
-    cwd: ROOT,
-    env: {
-      ...process.env,
-      VIBETEAM_PORT: port,
-    },
-    stdio: 'inherit',
+/** Exit code 75 = restart requested (e.g. via POST /api/restart) */
+const RESTART_EXIT_CODE = 75
+
+function spawnServer() {
+  let server
+  if (existsSync(compiledPath)) {
+    // Use compiled JS (production/npm install)
+    server = spawn('node', [compiledPath], {
+      cwd: ROOT,
+      env: {
+        ...process.env,
+        VIBETEAM_PORT: port,
+      },
+      stdio: 'inherit',
+    })
+  } else {
+    // Fall back to tsx (development)
+    console.log('(dev mode - using tsx)')
+    server = spawn('npx', ['tsx', sourcePath], {
+      cwd: ROOT,
+      env: {
+        ...process.env,
+        VIBETEAM_PORT: port,
+      },
+      stdio: 'inherit',
+    })
+  }
+
+  server.on('error', (err) => {
+    console.error('Failed to start server:', err.message)
+    process.exit(1)
   })
-}
 
-server.on('error', (err) => {
-  console.error('Failed to start server:', err.message)
-  process.exit(1)
-})
+  server.on('close', (code) => {
+    if (code === RESTART_EXIT_CODE) {
+      console.log('\n  Restarting server...\n')
+      // Small delay to let port release
+      setTimeout(() => spawnServer(), 1000)
+      return
+    }
+    process.exit(code || 0)
+  })
 
-server.on('close', (code) => {
-  process.exit(code || 0)
-})
+  // Handle signals - forward to child
+  const onSigInt = () => server.kill('SIGINT')
+  const onSigTerm = () => server.kill('SIGTERM')
 
-// Handle signals
-process.on('SIGINT', () => {
-  server.kill('SIGINT')
-})
+  // Remove old listeners to avoid stacking on restart
+  process.removeAllListeners('SIGINT')
+  process.removeAllListeners('SIGTERM')
+  process.on('SIGINT', onSigInt)
+  process.on('SIGTERM', onSigTerm)
+}
 
-process.on('SIGTERM', () => {
-  server.kill('SIGTERM')
-})
+spawnServer()