vibeteam 0.2.3 → 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
 
@@ -1008,6 +1076,78 @@ async function checkRequiredDependencies() {
 // 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()
 
@@ -1024,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()