claude-brain 0.15.2 → 0.17.0

package/src/cli/commands/start.ts

@@ -1,42 +1,42 @@
-/**
- * Start Command
- * Starts ChromaDB + MCP server together, or just ChromaDB with --chroma-only
- *
- * Usage:
- *   claude-brain start              Start ChromaDB + MCP server
- *   claude-brain start --chroma-only  Start only ChromaDB server
- */
-
-import {
-  heading, successText, warningText, dimText,
-} from '@/cli/ui/index.js'
-import { ensureChromaRunning } from '@/cli/commands/chroma'
-
-export async function runStart(): Promise<void> {
-  const chromaOnly = process.argv.includes('--chroma-only')
-
-  if (chromaOnly) {
-    console.log()
-    console.log(heading('Starting ChromaDB'))
-    console.log()
-
-    const chromaReady = await ensureChromaRunning()
-    console.log()
-
-    if (chromaReady) {
-      console.log(successText('ChromaDB is running and ready.'))
-    } else {
-      console.log(warningText('ChromaDB could not be started. Check installation with: claude-brain chroma status'))
-    }
-    console.log()
-    return
-  }
-
-  // Full start: ChromaDB + MCP server
-  // serve.ts already calls ensureChromaRunning(), so just delegate
-  console.error(dimText('Starting ChromaDB + MCP server...'))
-  console.error()
-
-  const { runServe } = await import('./serve')
-  await runServe()
-}
+/**
+ * Start Command
+ * Starts ChromaDB + MCP server together, or just ChromaDB with --chroma-only
+ *
+ * Usage:
+ *   claude-brain start              Start ChromaDB + MCP server
+ *   claude-brain start --chroma-only  Start only ChromaDB server
+ */
+
+import {
+  heading, successText, warningText, dimText,
+} from '@/cli/ui/index.js'
+import { ensureChromaRunning } from '@/cli/commands/chroma'
+
+export async function runStart(): Promise<void> {
+  const chromaOnly = process.argv.includes('--chroma-only')
+
+  if (chromaOnly) {
+    console.log()
+    console.log(heading('Starting ChromaDB'))
+    console.log()
+
+    const chromaReady = await ensureChromaRunning()
+    console.log()
+
+    if (chromaReady) {
+      console.log(successText('ChromaDB is running and ready.'))
+    } else {
+      console.log(warningText('ChromaDB could not be started. Check installation with: claude-brain chroma status'))
+    }
+    console.log()
+    return
+  }
+
+  // Full start: ChromaDB + MCP server
+  // serve.ts already calls ensureChromaRunning(), so just delegate
+  console.error(dimText('Starting ChromaDB + MCP server...'))
+  console.error()
+
+  const { runServe } = await import('./serve')
+  await runServe()
+}

package/src/cli/commands/uninstall-mcp.ts

@@ -1,41 +1,41 @@
-import { execSync } from 'node:child_process'
-import {
-  renderLogo, theme, heading, successText, errorText, dimText,
-  box, withSpinner,
-} from '@/cli/ui/index.js'
-
-export async function runUninstall() {
-  console.log()
-  console.log(renderLogo())
-  console.log()
-  console.log(heading('MCP Uninstall'))
-  console.log()
-
-  try {
-    await withSpinner('Removing Claude Brain from Claude Code', async () => {
-      try {
-        execSync('claude mcp remove claude-brain', {
-          encoding: 'utf-8',
-          stdio: ['pipe', 'pipe', 'pipe'],
-        })
-      } catch {
-        // Falls back to scoped removal when registered in multiple scopes
-        execSync('claude mcp remove claude-brain -s local', {
-          encoding: 'utf-8',
-          stdio: ['pipe', 'pipe', 'pipe'],
-        })
-      }
-    })
-    console.log()
-    console.log(box(successText('Claude Brain MCP server removed from Claude Code.'), 'Success'))
-  } catch {
-    console.log()
-    console.log(box([
-      errorText('Failed to remove automatically.'),
-      '',
-      dimText('Run manually:'),
-      `  ${theme.bold('claude mcp remove claude-brain')}`,
-    ].join('\n'), 'Error'))
-  }
-  console.log()
-}
+import { execSync } from 'node:child_process'
+import {
+  renderLogo, theme, heading, successText, errorText, dimText,
+  box, withSpinner,
+} from '@/cli/ui/index.js'
+
+export async function runUninstall() {
+  console.log()
+  console.log(renderLogo())
+  console.log()
+  console.log(heading('MCP Uninstall'))
+  console.log()
+
+  try {
+    await withSpinner('Removing Claude Brain from Claude Code', async () => {
+      try {
+        execSync('claude mcp remove claude-brain', {
+          encoding: 'utf-8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+        })
+      } catch {
+        // Falls back to scoped removal when registered in multiple scopes
+        execSync('claude mcp remove claude-brain -s local', {
+          encoding: 'utf-8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+        })
+      }
+    })
+    console.log()
+    console.log(box(successText('Claude Brain MCP server removed from Claude Code.'), 'Success'))
+  } catch {
+    console.log()
+    console.log(box([
+      errorText('Failed to remove automatically.'),
+      '',
+      dimText('Run manually:'),
+      `  ${theme.bold('claude mcp remove claude-brain')}`,
+    ].join('\n'), 'Error'))
+  }
+  console.log()
+}

package/src/cli/commands/update.ts

@@ -1,121 +1,124 @@
-import { execSync } from 'node:child_process'
-import { existsSync, readFileSync } from 'node:fs'
-import fs from 'fs/promises'
-import path from 'path'
-import os from 'os'
-import { fileURLToPath } from 'url'
-import {
-  renderLogo, theme, heading, successText, warningText, dimText,
-  box, withSpinner,
-} from '@/cli/ui/index.js'
-
-const __filename = fileURLToPath(import.meta.url)
-const __dirname = path.dirname(__filename)
-const PACKAGE_ROOT = path.resolve(__dirname, '..', '..', '..')
-
-function getInstalledVersion(): string {
-  try {
-    const pkg = JSON.parse(readFileSync(path.join(PACKAGE_ROOT, 'package.json'), 'utf-8'))
-    return pkg.version || 'unknown'
-  } catch {
-    return 'unknown'
-  }
-}
-
-function getLatestVersion(): string | null {
-  try {
-    const result = execSync('npm view claude-brain version', {
-      encoding: 'utf-8',
-      stdio: ['pipe', 'pipe', 'pipe'],
-    })
-    return result.trim()
-  } catch {
-    return null
-  }
-}
-
-export async function runUpdate() {
-  console.log()
-  console.log(renderLogo())
-  console.log()
-  console.log(heading('Update Claude Brain'))
-  console.log()
-
-  const currentVersion = getInstalledVersion()
-  console.log(`  ${theme.bold('Installed:')} ${dimText('v' + currentVersion)}`)
-
-  const latestVersion = await withSpinner('Checking for updates', () => getLatestVersion())
-
-  if (latestVersion) {
-    console.log(`  ${theme.bold('Latest:')}    ${dimText('v' + latestVersion)}`)
-  }
-  console.log()
-
-  // Step 1: Update package if newer version available
-  if (latestVersion && latestVersion !== currentVersion) {
-    try {
-      await withSpinner(`Updating claude-brain to v${latestVersion}`, () => {
-        execSync('bun update -g claude-brain', {
-          encoding: 'utf-8',
-          stdio: ['pipe', 'pipe', 'pipe'],
-        })
-      })
-      console.log(successText(`  Updated to v${latestVersion}`))
-    } catch {
-      console.log(warningText('  Package update failed. Try manually: bun update -g claude-brain'))
-    }
-  } else {
-    console.log(successText('  Package is up to date'))
-  }
-  console.log()
-
-  // Step 2: Update CLAUDE.md to ~/.claude/CLAUDE.md (where Claude Code reads it)
-  const sourcePath = path.join(PACKAGE_ROOT, 'assets', 'CLAUDE.md')
-  const claudeDir = path.join(os.homedir(), '.claude')
-  const destPath = path.join(claudeDir, 'CLAUDE.md')
-
-  if (!existsSync(sourcePath)) {
-    console.log(warningText('  CLAUDE.md asset not found in package, skipping'))
-  } else if (!existsSync(destPath)) {
-    await withSpinner('Installing CLAUDE.md (brain works automatically)', async () => {
-      await fs.mkdir(claudeDir, { recursive: true })
-      await fs.copyFile(sourcePath, destPath)
-    })
-    console.log(successText('  CLAUDE.md installed to ~/.claude/CLAUDE.md (brain is now automatic)'))
-  } else {
-    const sourceContent = readFileSync(sourcePath, 'utf-8')
-    const destContent = readFileSync(destPath, 'utf-8')
-
-    if (sourceContent === destContent) {
-      console.log(successText('  CLAUDE.md is already up to date'))
-    } else {
-      await withSpinner('Updating CLAUDE.md', async () => {
-        await fs.copyFile(sourcePath, destPath)
-      })
-      console.log(successText('  CLAUDE.md updated to ~/.claude/CLAUDE.md (brain is now automatic)'))
-    }
-  }
-
-  // Clean up old ~/CLAUDE.md if it exists (from previous versions)
-  const oldDestPath = path.join(os.homedir(), 'CLAUDE.md')
-  if (existsSync(oldDestPath)) {
-    try {
-      const oldContent = readFileSync(oldDestPath, 'utf-8')
-      if (oldContent.includes('Claude Brain')) {
-        await fs.unlink(oldDestPath)
-        console.log(dimText('  Removed old ~/CLAUDE.md (migrated to ~/.claude/)'))
-      }
-    } catch {
-      // ignore cleanup errors
-    }
-  }
-
-  console.log()
-  console.log(box([
-    heading('Update complete!'),
-    '',
-    dimText('Run health check to verify:'),
-    `  ${theme.bold('claude-brain health')}`,
-  ].join('\n'), 'Done'))
-  console.log()
-}
+import { execSync } from 'node:child_process'
+import { existsSync, readFileSync } from 'node:fs'
+import fs from 'fs/promises'
+import path from 'path'
+import os from 'os'
+import { fileURLToPath } from 'url'
+import {
+  renderLogo, theme, heading, successText, warningText, dimText,
+  box, withSpinner,
+} from '@/cli/ui/index.js'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = path.dirname(__filename)
+const PACKAGE_ROOT = path.resolve(__dirname, '..', '..', '..')
+
+function getInstalledVersion(): string {
+  try {
+    const pkg = JSON.parse(readFileSync(path.join(PACKAGE_ROOT, 'package.json'), 'utf-8'))
+    return pkg.version || 'unknown'
+  } catch {
+    return 'unknown'
+  }
+}
+
+function getLatestVersion(): string | null {
+  try {
+    const result = execSync('npm view claude-brain version', {
+      encoding: 'utf-8',
+      stdio: ['pipe', 'pipe', 'pipe'],
+    })
+    return result.trim()
+  } catch {
+    return null
+  }
+}
+
+export async function runUpdate() {
+  console.log()
+  console.log(renderLogo())
+  console.log()
+  console.log(heading('Update Claude Brain'))
+  console.log()
+
+  const currentVersion = getInstalledVersion()
+  console.log(`  ${theme.bold('Installed:')} ${dimText('v' + currentVersion)}`)
+
+  const latestVersion = await withSpinner('Checking for updates', () => getLatestVersion())
+
+  if (latestVersion) {
+    console.log(`  ${theme.bold('Latest:')}    ${dimText('v' + latestVersion)}`)
+  }
+  console.log()
+
+  // Step 1: Update package if newer version available
+  if (latestVersion && latestVersion !== currentVersion) {
+    try {
+      await withSpinner(`Updating claude-brain to v${latestVersion}`, () => {
+        execSync('bun update -g claude-brain', {
+          encoding: 'utf-8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+        })
+      })
+      console.log(successText(`  Updated to v${latestVersion}`))
+    } catch {
+      console.log(warningText('  Package update failed. Try manually: bun update -g claude-brain'))
+    }
+  } else {
+    console.log(successText('  Package is up to date'))
+  }
+  console.log()
+
+  // Step 2: Update CLAUDE.md to ~/.claude/CLAUDE.md (where Claude Code reads it)
+  const sourcePath = path.join(PACKAGE_ROOT, 'assets', 'CLAUDE.md')
+  const claudeDir = path.join(os.homedir(), '.claude')
+  const destPath = path.join(claudeDir, 'CLAUDE.md')
+
+  if (!existsSync(sourcePath)) {
+    console.log(warningText('  CLAUDE.md asset not found in package, skipping'))
+  } else if (!existsSync(destPath)) {
+    await withSpinner('Installing CLAUDE.md (brain works automatically)', async () => {
+      await fs.mkdir(claudeDir, { recursive: true })
+      await fs.copyFile(sourcePath, destPath)
+    })
+    console.log(successText('  CLAUDE.md installed to ~/.claude/CLAUDE.md (brain is now automatic)'))
+  } else {
+    const sourceContent = readFileSync(sourcePath, 'utf-8')
+    const destContent = readFileSync(destPath, 'utf-8')
+
+    if (sourceContent === destContent) {
+      console.log(successText('  CLAUDE.md is already up to date'))
+    } else {
+      await withSpinner('Updating CLAUDE.md', async () => {
+        await fs.copyFile(sourcePath, destPath)
+      })
+      console.log(successText('  CLAUDE.md updated to ~/.claude/CLAUDE.md (brain is now automatic)'))
+    }
+  }
+
+  // Clean up old ~/CLAUDE.md if it exists (from previous versions)
+  const oldDestPath = path.join(os.homedir(), 'CLAUDE.md')
+  if (existsSync(oldDestPath)) {
+    try {
+      const oldContent = readFileSync(oldDestPath, 'utf-8')
+      if (oldContent.includes('Claude Brain')) {
+        await fs.unlink(oldDestPath)
+        console.log(dimText('  Removed old ~/CLAUDE.md (migrated to ~/.claude/)'))
+      }
+    } catch {
+      // ignore cleanup errors
+    }
+  }
+
+  console.log()
+  console.log(box([
+    heading('Update complete!'),
+    '',
+    dimText('Run health check to verify:'),
+    `  ${theme.bold('claude-brain health')}`,
+    '',
+    dimText('For a full kill + update + restart:'),
+    `  ${theme.bold('claude-brain refresh')}`,
+  ].join('\n'), 'Done'))
+  console.log()
+}

package/src/cli/diagnose.ts

@@ -1,4 +1,4 @@
-#!/usr/bin/env bun
-import { runDiagnostics } from '@/diagnostics'
-
-runDiagnostics()
+#!/usr/bin/env bun
+import { runDiagnostics } from '@/diagnostics'
+
+runDiagnostics()

package/src/cli/health-check.ts

@@ -1,4 +1,4 @@
-#!/usr/bin/env bun
-import { runHealthCheck } from '@/health'
-
-runHealthCheck()
+#!/usr/bin/env bun
+import { runHealthCheck } from '@/health'
+
+runHealthCheck()

package/src/cli/migrate-chroma.ts

@@ -1,106 +1,106 @@
-#!/usr/bin/env bun
-
-import { MemoryManager } from '@/memory'
-import { createLogger } from '@/utils'
-import fs from 'node:fs/promises'
-
-// Parse CLI arguments (simple argv parsing)
-const args = process.argv.slice(2)
-const options = {
-  batchSize: 100,
-  validateAfter: false,
-  verbose: false
-}
-
-for (const arg of args) {
-  if (arg === '--validate-after') options.validateAfter = true
-  if (arg === '--verbose') options.verbose = true
-  const batchMatch = arg.match(/--batch-size=(\d+)/)
-  if (batchMatch) options.batchSize = parseInt(batchMatch[1])
-}
-
-// Create dedicated migration logger
-const logger = createLogger('info', './logs/migration.log')
-
-async function main() {
-  const startTime = Date.now()
-  
-  console.log('📦 SQLite → ChromaDB Migration\n')
-  
-  // Ensure logs directory exists
-  await fs.mkdir('./logs', { recursive: true })
-  
-  // Initialize MemoryManager with ChromaDB enabled
-  if (options.verbose) {
-    console.log('⏳ Initializing MemoryManager...')
-  }
-  
-  const memoryManager = new MemoryManager(
-    './data/memory.db',
-    logger,
-    true // useChromaDB = true
-  )
-  
-  await memoryManager.initialize()
-
-  // Clear existing collections to avoid embedding function conflicts
-  if (options.verbose) {
-    console.log('⏳ Clearing existing ChromaDB collections...')
-  }
-  await memoryManager.chroma.collections.clearAll()
-
-  // Check ChromaDB connection
-  if (options.verbose) {
-    console.log('⏳ Verifying ChromaDB connection...')
-  }
-  
-  const healthy = await memoryManager.chroma.healthCheck()
-  if (!healthy) {
-    console.error('❌ ChromaDB health check failed')
-    logger.error('ChromaDB health check failed')
-    process.exit(1)
-  }
-  
-  if (options.verbose) {
-    console.log('✅ ChromaDB connected')
-  }
-  
-  // Get SQLite stats before migration
-  const sqliteStats = memoryManager.database.getStats()
-  console.log(`📊 SQLite Database: ${sqliteStats.decisions} decisions, ${sqliteStats.memories} memories\n`)
-  
-  // Run migration
-  console.log('🚀 Starting migration...\n')
-  
-  const stats = await memoryManager.migrateToChromaDB({
-    batchSize: options.batchSize,
-    validateAfter: options.validateAfter,
-    skipExisting: false, // Always migrate, don't skip
-    dryRun: false
-  })
-  
-  const duration = Date.now() - startTime
-  
-  // Display results
-  console.log('\n✅ Migration completed!')
-  console.log(`   Duration: ${(duration / 1000).toFixed(2)}s`)
-  console.log(`   Decisions: ${stats.migratedDecisions}/${stats.totalDecisions} (${stats.failedDecisions} failed)`)
-  console.log(`   Memories: ${stats.migratedMemories}/${stats.totalMemories} (${stats.failedMemories} failed)`)
-  
-  if (options.validateAfter) {
-    console.log('\n🔍 Validation passed ✓')
-  }
-  
-  await memoryManager.close()
-  
-  logger.info({
-    stats,
-    duration
-  }, 'Migration completed successfully')
-}
-
-main().catch(error => {
-  console.error('\n❌ Migration failed:', error.message)
-  logger.error({ error }, 'Migration failed')
-  process.exit(1)
-})
+#!/usr/bin/env bun
+
+import { MemoryManager } from '@/memory'
+import { createLogger } from '@/utils'
+import fs from 'node:fs/promises'
+
+// Parse CLI arguments (simple argv parsing)
+const args = process.argv.slice(2)
+const options = {
+  batchSize: 100,
+  validateAfter: false,
+  verbose: false
+}
+
+for (const arg of args) {
+  if (arg === '--validate-after') options.validateAfter = true
+  if (arg === '--verbose') options.verbose = true
+  const batchMatch = arg.match(/--batch-size=(\d+)/)
+  if (batchMatch) options.batchSize = parseInt(batchMatch[1])
+}
+
+// Create dedicated migration logger
+const logger = createLogger('info', './logs/migration.log')
+
+async function main() {
+  const startTime = Date.now()
+  
+  console.log('📦 SQLite → ChromaDB Migration\n')
+  
+  // Ensure logs directory exists
+  await fs.mkdir('./logs', { recursive: true })
+  
+  // Initialize MemoryManager with ChromaDB enabled
+  if (options.verbose) {
+    console.log('⏳ Initializing MemoryManager...')
+  }
+  
+  const memoryManager = new MemoryManager(
+    './data/memory.db',
+    logger,
+    true // useChromaDB = true
+  )
+  
+  await memoryManager.initialize()
+
+  // Clear existing collections to avoid embedding function conflicts
+  if (options.verbose) {
+    console.log('⏳ Clearing existing ChromaDB collections...')
+  }
+  await memoryManager.chroma.collections.clearAll()
+
+  // Check ChromaDB connection
+  if (options.verbose) {
+    console.log('⏳ Verifying ChromaDB connection...')
+  }
+  
+  const healthy = await memoryManager.chroma.healthCheck()
+  if (!healthy) {
+    console.error('❌ ChromaDB health check failed')
+    logger.error('ChromaDB health check failed')
+    process.exit(1)
+  }
+  
+  if (options.verbose) {
+    console.log('✅ ChromaDB connected')
+  }
+  
+  // Get SQLite stats before migration
+  const sqliteStats = memoryManager.database.getStats()
+  console.log(`📊 SQLite Database: ${sqliteStats.decisions} decisions, ${sqliteStats.memories} memories\n`)
+  
+  // Run migration
+  console.log('🚀 Starting migration...\n')
+  
+  const stats = await memoryManager.migrateToChromaDB({
+    batchSize: options.batchSize,
+    validateAfter: options.validateAfter,
+    skipExisting: false, // Always migrate, don't skip
+    dryRun: false
+  })
+  
+  const duration = Date.now() - startTime
+  
+  // Display results
+  console.log('\n✅ Migration completed!')
+  console.log(`   Duration: ${(duration / 1000).toFixed(2)}s`)
+  console.log(`   Decisions: ${stats.migratedDecisions}/${stats.totalDecisions} (${stats.failedDecisions} failed)`)
+  console.log(`   Memories: ${stats.migratedMemories}/${stats.totalMemories} (${stats.failedMemories} failed)`)
+  
+  if (options.validateAfter) {
+    console.log('\n🔍 Validation passed ✓')
+  }
+  
+  await memoryManager.close()
+  
+  logger.info({
+    stats,
+    duration
+  }, 'Migration completed successfully')
+}
+
+main().catch(error => {
+  console.error('\n❌ Migration failed:', error.message)
+  logger.error({ error }, 'Migration failed')
+  process.exit(1)
+})

package/src/cli/setup.ts

@@ -1,4 +1,4 @@
-#!/usr/bin/env bun
-import { runSetup } from '@/setup'
-
-runSetup()
+#!/usr/bin/env bun
+import { runSetup } from '@/setup'
+
+runSetup()