claude-brain 0.4.0 → 0.5.0

package/src/setup/wizard.ts

@@ -1,6 +1,7 @@
 import prompts from 'prompts'
 import fs from 'fs/promises'
 import { existsSync } from 'fs'
+import { execSync } from 'child_process'
 import path from 'path'
 import os from 'os'
 import { fileURLToPath } from 'url'
@@ -21,6 +22,7 @@ export interface SetupAnswers {
   vaultPath: string
   logLevel: string
   enableFileWatch: boolean
+  enableChromaDB: boolean
   createSampleProject: boolean
   installClaudeMd: boolean
 }
@@ -34,7 +36,7 @@ export class SetupWizard {
 
   async run(): Promise<SetupAnswers> {
     // Step 1: Detect Vaults
-    console.log(stepIndicator(1, 5, 'Detecting Obsidian Vaults'))
+    console.log(stepIndicator(1, 6, 'Detecting Obsidian Vaults'))
     await transition()
 
     const suggestedPaths = await withSpinner('Scanning for Obsidian vaults', () =>
@@ -49,7 +51,7 @@ export class SetupWizard {
     console.log()
 
     // Step 2: Vault Configuration
-    console.log(stepIndicator(2, 5, 'Vault Configuration'))
+    console.log(stepIndicator(2, 6, 'Vault Configuration'))
     await transition()
 
     const vaultAnswers = await prompts([
@@ -86,7 +88,7 @@ export class SetupWizard {
     const finalVaultPath = vaultAnswers.vaultPath || vaultAnswers.vaultPathChoice
 
     // Step 3: Logging
-    console.log(stepIndicator(3, 5, 'Logging Configuration'))
+    console.log(stepIndicator(3, 6, 'Logging Configuration'))
     await transition()
 
     const loggingAnswers = await prompts({
@@ -103,7 +105,7 @@ export class SetupWizard {
     })
 
     // Step 4: Features
-    console.log(stepIndicator(4, 5, 'Feature Selection'))
+    console.log(stepIndicator(4, 6, 'Feature Selection'))
     await transition()
 
     const featureAnswers = await prompts([
@@ -127,14 +129,40 @@ export class SetupWizard {
       }
     ])
 
-    // Step 5: Review
-    console.log(stepIndicator(5, 5, 'Review Configuration'))
+    // Step 5: ChromaDB
+    console.log(stepIndicator(5, 6, 'ChromaDB (Vector Database)'))
+    await transition()
+
+    console.log(dimText('ChromaDB enables semantic search, knowledge graph,'))
+    console.log(dimText('and advanced intelligence features. Requires Python 3.'))
+    console.log()
+
+    const chromaInstalled = this.isChromaInstalled()
+    if (chromaInstalled) {
+      console.log(successText('ChromaDB CLI detected'))
+    } else {
+      console.log(warningText('ChromaDB CLI not found — you can install it later'))
+    }
+    console.log()
+
+    const chromaAnswers = await prompts({
+      type: 'confirm',
+      name: 'enableChromaDB',
+      message: chromaInstalled
+        ? 'Enable ChromaDB for advanced features?'
+        : 'Enable ChromaDB? (install later: pip install chromadb)',
+      initial: chromaInstalled
+    })
+
+    // Step 6: Review
+    console.log(stepIndicator(6, 6, 'Review Configuration'))
     await transition()
 
     const answers: SetupAnswers = {
       vaultPath: finalVaultPath,
       logLevel: loggingAnswers.logLevel ?? 'warn',
       enableFileWatch: featureAnswers.enableFileWatch ?? true,
+      enableChromaDB: chromaAnswers.enableChromaDB ?? false,
       createSampleProject: featureAnswers.createSampleProject ?? true,
       installClaudeMd: featureAnswers.installClaudeMd ?? true,
     }
@@ -143,6 +171,7 @@ export class SetupWizard {
       { label: 'Vault Path', value: answers.vaultPath, status: 'success' },
       { label: 'Log Level', value: answers.logLevel, status: 'info' },
       { label: 'File Watching', value: answers.enableFileWatch ? 'Enabled' : 'Disabled', status: answers.enableFileWatch ? 'success' : 'warning' },
+      { label: 'ChromaDB', value: answers.enableChromaDB ? 'Enabled' : 'Disabled', status: answers.enableChromaDB ? 'success' : 'warning' },
       { label: 'Sample Project', value: answers.createSampleProject ? 'Yes' : 'No', status: 'info' },
       { label: 'Install CLAUDE.md', value: answers.installClaudeMd ? 'Yes' : 'No', status: 'info' },
     ]))
@@ -191,11 +220,31 @@ export class SetupWizard {
     return locations
   }
 
+  private isChromaInstalled(): boolean {
+    try {
+      execSync('chroma --version', { stdio: 'pipe', timeout: 5000 })
+      return true
+    } catch {
+      return false
+    }
+  }
+
   async applyConfiguration(answers: SetupAnswers): Promise<void> {
     console.log('\n' + heading('Applying configuration...') + '\n')
 
     const homePaths = getHomePaths()
 
+    const chromaEnvLines = answers.enableChromaDB
+      ? [
+          '',
+          '# ChromaDB Configuration',
+          'CHROMA_MODE=client-server',
+          'CHROMA_HOST=localhost',
+          'CHROMA_PORT=8000',
+          'CHROMA_EMBEDDING_PROVIDER=transformers',
+        ].join('\n')
+      : ''
+
     const envContent = `# Claude Brain Configuration
 VAULT_PATH=${answers.vaultPath}
 LOG_LEVEL=${answers.logLevel}
@@ -203,6 +252,7 @@ ENABLE_FILE_WATCH=${answers.enableFileWatch}
 DB_PATH=./data/memory.db
 LOG_FILE_PATH=./logs/claude-brain.log
 SERVER_NAME=claude-brain
+${chromaEnvLines}
 `
 
     await withSpinner('Writing .env configuration', async () => {
@@ -211,6 +261,7 @@ SERVER_NAME=claude-brain
 
     await withSpinner('Creating data and log directories', async () => {
       await fs.mkdir(homePaths.data, { recursive: true })
+      await fs.mkdir(homePaths.chroma, { recursive: true })
       await fs.mkdir(homePaths.logs, { recursive: true })
     })
 
@@ -229,16 +280,35 @@ SERVER_NAME=claude-brain
       }
     }
 
-    console.log()
-    console.log(box([
+    const nextSteps: string[] = [
       heading('Setup complete!'),
       '',
       dimText('Next steps:'),
       `  ${theme.primary('1.')} ${theme.bold('claude-brain install')}`,
       `     ${dimText('Register as MCP server')}`,
-      `  ${theme.primary('2.')} ${theme.bold('claude-brain health')}`,
-      `     ${dimText('Verify everything works')}`,
-    ].join('\n'), 'Done'))
+    ]
+
+    if (answers.enableChromaDB) {
+      nextSteps.push(
+        `  ${theme.primary('2.')} ${theme.bold('claude-brain chroma install')}`,
+        `     ${dimText('Install ChromaDB (if not already installed)')}`,
+        `  ${theme.primary('3.')} ${theme.bold('claude-brain chroma start')}`,
+        `     ${dimText('Start the ChromaDB server')}`,
+        `  ${theme.primary('4.')} ${theme.bold('claude-brain health')}`,
+        `     ${dimText('Verify everything works')}`,
+      )
+    } else {
+      nextSteps.push(
+        `  ${theme.primary('2.')} ${theme.bold('claude-brain health')}`,
+        `     ${dimText('Verify everything works')}`,
+        '',
+        dimText('Optional: Enable ChromaDB later for advanced features:'),
+        `  ${dimText('claude-brain chroma install && claude-brain chroma start')}`,
+      )
+    }
+
+    console.log()
+    console.log(box(nextSteps.join('\n'), 'Done'))
     console.log()
   }