claude-brain 0.30.2 → 0.30.3

package/src/cli/commands/update.ts

@@ -1,124 +1,130 @@
-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', async () => 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}`, async () => {
-        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()
-}
+import { spawnSync } 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 = spawnSync('npm', ['view', 'claude-brain', 'version'], {
+      encoding: 'utf-8',
+      stdio: ['pipe', 'pipe', 'pipe'],
+      timeout: 15_000,
+    })
+    if (result.error || result.status !== 0) return null
+    return result.stdout?.trim() || null
+  } 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', async () => 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}`, async () => {
+        const result = spawnSync('bun', ['update', '-g', 'claude-brain'], {
+          encoding: 'utf-8',
+          stdio: ['pipe', 'pipe', 'pipe'],
+          timeout: 120_000,
+        })
+        if (result.error || result.status !== 0) {
+          throw new Error(result.stderr || 'bun update failed')
+        }
+      })
+      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/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/ui/animations.ts

@@ -1,80 +1,80 @@
-import ora from 'ora'
-import { theme } from './theme.js'
-
-const isTTY = process.stdout.isTTY === true
-
-// Async delay utility
-export function delay(ms: number): Promise<void> {
-  return new Promise(resolve => setTimeout(resolve, ms))
-}
-
-// Character-by-character text output with configurable speed
-export async function typewrite(text: string, speed = 30): Promise<void> {
-  if (!isTTY) {
-    process.stdout.write(text + '\n')
-    return
-  }
-  for (const char of text) {
-    process.stdout.write(char)
-    await delay(speed)
-  }
-  process.stdout.write('\n')
-}
-
-// Wraps async operations with ora spinner, auto succeed/fail
-export async function withSpinner<T>(label: string, fn: () => Promise<T>): Promise<T> {
-  if (!isTTY) {
-    process.stdout.write(`${label}...`)
-    try {
-      const result = await fn()
-      process.stdout.write(' done\n')
-      return result
-    } catch (error) {
-      process.stdout.write(' failed\n')
-      throw error
-    }
-  }
-
-  const spinner = ora({
-    text: label,
-    color: 'magenta',
-  }).start()
-
-  try {
-    const result = await fn()
-    spinner.succeed(theme.success(label))
-    return result
-  } catch (error) {
-    spinner.fail(theme.error(label))
-    throw error
-  }
-}
-
-// Pause between sections for visual breathing room
-export async function transition(ms = 300): Promise<void> {
-  if (!isTTY) return
-  await delay(ms)
-}
-
-// Progress bar that fills over time
-export async function animateProgress(label: string, durationMs = 1000, width = 30): Promise<void> {
-  if (!isTTY) {
-    console.log(`${label}... done`)
-    return
-  }
-
-  const steps = 20
-  const stepDelay = durationMs / steps
-
-  for (let i = 0; i <= steps; i++) {
-    const percent = Math.round((i / steps) * 100)
-    const filled = Math.round((percent / 100) * width)
-    const empty = width - filled
-    const bar = theme.primary('='.repeat(Math.max(0, filled - 1)) + (filled > 0 ? '>' : '')) +
-      ' '.repeat(empty)
-    const pct = theme.dim(`${percent}%`)
-    process.stdout.write(`\r  ${label} [${bar}] ${pct}`)
-    await delay(stepDelay)
-  }
-  process.stdout.write('\n')
-}
+import ora from 'ora'
+import { theme } from './theme.js'
+
+const isTTY = process.stdout.isTTY === true
+
+// Async delay utility
+export function delay(ms: number): Promise<void> {
+  return new Promise(resolve => setTimeout(resolve, ms))
+}
+
+// Character-by-character text output with configurable speed
+export async function typewrite(text: string, speed = 30): Promise<void> {
+  if (!isTTY) {
+    process.stdout.write(text + '\n')
+    return
+  }
+  for (const char of text) {
+    process.stdout.write(char)
+    await delay(speed)
+  }
+  process.stdout.write('\n')
+}
+
+// Wraps async operations with ora spinner, auto succeed/fail
+export async function withSpinner<T>(label: string, fn: () => Promise<T>): Promise<T> {
+  if (!isTTY) {
+    process.stdout.write(`${label}...`)
+    try {
+      const result = await fn()
+      process.stdout.write(' done\n')
+      return result
+    } catch (error) {
+      process.stdout.write(' failed\n')
+      throw error
+    }
+  }
+
+  const spinner = ora({
+    text: label,
+    color: 'magenta',
+  }).start()
+
+  try {
+    const result = await fn()
+    spinner.succeed(theme.success(label))
+    return result
+  } catch (error) {
+    spinner.fail(theme.error(label))
+    throw error
+  }
+}
+
+// Pause between sections for visual breathing room
+export async function transition(ms = 300): Promise<void> {
+  if (!isTTY) return
+  await delay(ms)
+}
+
+// Progress bar that fills over time
+export async function animateProgress(label: string, durationMs = 1000, width = 30): Promise<void> {
+  if (!isTTY) {
+    console.log(`${label}... done`)
+    return
+  }
+
+  const steps = 20
+  const stepDelay = durationMs / steps
+
+  for (let i = 0; i <= steps; i++) {
+    const percent = Math.round((i / steps) * 100)
+    const filled = Math.round((percent / 100) * width)
+    const empty = width - filled
+    const bar = theme.primary('='.repeat(Math.max(0, filled - 1)) + (filled > 0 ? '>' : '')) +
+      ' '.repeat(empty)
+    const pct = theme.dim(`${percent}%`)
+    process.stdout.write(`\r  ${label} [${bar}] ${pct}`)
+    await delay(stepDelay)
+  }
+  process.stdout.write('\n')
+}