claude-brain 0.30.2 → 0.31.1

package/src/code-intelligence/query.ts

@@ -1,291 +1,291 @@
-import { Database } from 'bun:sqlite'
-import type { Logger } from 'pino'
-import type { SymbolResult, FileResult, DependencyResult, FileMapEntry, IndexStats } from './types'
-
-export class CodeQuery {
-  private db: Database
-  private logger: Logger
-  private memoryDb?: Database
-
-  constructor(db: Database, logger: Logger) {
-    this.db = db
-    this.logger = logger.child({ component: 'code-query' })
-  }
-
-  /** Phase 29: Set memory DB for file-linked memory annotations */
-  setMemoryDb(db: Database): void {
-    this.memoryDb = db
-  }
-
-  findSymbols(query: string, project: string, limit: number = 20): SymbolResult[] {
-    if (!query.trim()) return []
-
-    // First try exact match
-    const exact = this.db.query(
-      `SELECT symbol_name, symbol_type, file_path, line_start, line_end, signature
-       FROM code_symbols
-       WHERE project = ? AND symbol_name = ?
-       LIMIT ?`
-    ).all(project, query, limit) as RawSymbolRow[]
-
-    if (exact.length > 0) {
-      return exact.map(row => ({
-        symbol: row.symbol_name,
-        type: row.symbol_type as any,
-        filePath: row.file_path,
-        lineStart: row.line_start,
-        lineEnd: row.line_end ?? undefined,
-        signature: row.signature ?? undefined,
-        confidence: 1.0,
-      }))
-    }
-
-    // Try prefix match
-    const prefix = this.db.query(
-      `SELECT symbol_name, symbol_type, file_path, line_start, line_end, signature
-       FROM code_symbols
-       WHERE project = ? AND symbol_name LIKE ? || '%'
-       LIMIT ?`
-    ).all(project, query, limit) as RawSymbolRow[]
-
-    if (prefix.length > 0) {
-      return prefix.map(row => ({
-        symbol: row.symbol_name,
-        type: row.symbol_type as any,
-        filePath: row.file_path,
-        lineStart: row.line_start,
-        lineEnd: row.line_end ?? undefined,
-        signature: row.signature ?? undefined,
-        confidence: 0.9,
-      }))
-    }
-
-    // Fall back to FTS5 search
-    try {
-      const ftsResults = this.db.query(
-        `SELECT s.symbol_name, s.symbol_type, s.file_path, s.line_start, s.line_end, s.signature,
-                rank
-         FROM code_symbols_fts fts
-         JOIN code_symbols s ON s.id = fts.rowid
-         WHERE code_symbols_fts MATCH ? AND s.project = ?
-         ORDER BY rank
-         LIMIT ?`
-      ).all(query, project, limit) as (RawSymbolRow & { rank: number })[]
-
-      if (ftsResults.length === 0) return []
-
-      // Normalize FTS5 rank to 0.5-0.85 confidence
-      const maxRank = Math.abs(ftsResults[0].rank) || 1
-      return ftsResults.map(row => ({
-        symbol: row.symbol_name,
-        type: row.symbol_type as any,
-        filePath: row.file_path,
-        lineStart: row.line_start,
-        lineEnd: row.line_end ?? undefined,
-        signature: row.signature ?? undefined,
-        confidence: 0.5 + 0.35 * (Math.abs(row.rank) / maxRank),
-      }))
-    } catch (error) {
-      // FTS5 can throw on invalid query syntax
-      this.logger.debug({ error, query }, 'FTS5 query failed, falling back to LIKE')
-
-      const likeResults = this.db.query(
-        `SELECT symbol_name, symbol_type, file_path, line_start, line_end, signature
-         FROM code_symbols
-         WHERE project = ? AND symbol_name LIKE '%' || ? || '%'
-         LIMIT ?`
-      ).all(project, query, limit) as RawSymbolRow[]
-
-      return likeResults.map(row => ({
-        symbol: row.symbol_name,
-        type: row.symbol_type as any,
-        filePath: row.file_path,
-        lineStart: row.line_start,
-        lineEnd: row.line_end ?? undefined,
-        signature: row.signature ?? undefined,
-        confidence: 0.5,
-      }))
-    }
-  }
-
-  findFiles(query: string, project: string): FileResult[] {
-    if (!query.trim()) return []
-
-    const rows = this.db.query(
-      `SELECT file_path, language, summary, symbol_count, line_count
-       FROM code_files
-       WHERE project = ? AND (file_path LIKE '%' || ? || '%' OR summary LIKE '%' || ? || '%')
-       ORDER BY file_path
-       LIMIT 50`
-    ).all(project, query, query) as RawFileRow[]
-
-    return rows.map(row => ({
-      filePath: row.file_path,
-      language: row.language ?? undefined,
-      summary: row.summary ?? undefined,
-      symbolCount: row.symbol_count,
-      lineCount: row.line_count,
-    }))
-  }
-
-  getDependencies(filePath: string, project: string): DependencyResult {
-    const imports = this.db.query(
-      `SELECT target_file, import_names
-       FROM code_dependencies
-       WHERE project = ? AND source_file = ?`
-    ).all(project, filePath) as { target_file: string; import_names: string | null }[]
-
-    const importedBy = this.db.query(
-      `SELECT source_file, import_names
-       FROM code_dependencies
-       WHERE project = ? AND target_file = ?`
-    ).all(project, filePath) as { source_file: string; import_names: string | null }[]
-
-    return {
-      file: filePath,
-      imports: imports.map(row => ({
-        file: row.target_file,
-        names: row.import_names ? row.import_names.split(', ').filter(Boolean) : [],
-      })),
-      importedBy: importedBy.map(row => ({
-        file: row.source_file,
-        names: row.import_names ? row.import_names.split(', ').filter(Boolean) : [],
-      })),
-    }
-  }
-
-  getDependents(filePath: string, project: string): DependencyResult {
-    // getDependents is the inverse view — same data, different perspective
-    return this.getDependencies(filePath, project)
-  }
-
-  getFileMap(project: string, maxEntries: number = 100): FileMapEntry[] {
-    const rows = this.db.query(
-      `SELECT f.file_path, f.language, f.summary,
-              GROUP_CONCAT(CASE WHEN s.exported = 1 THEN s.symbol_name END) as symbols
-       FROM code_files f
-       LEFT JOIN code_symbols s ON s.project = f.project AND s.file_path = f.file_path
-       WHERE f.project = ?
-       GROUP BY f.file_path
-       ORDER BY f.file_path
-       LIMIT ?`
-    ).all(project, maxEntries) as {
-      file_path: string
-      language: string | null
-      summary: string | null
-      symbols: string | null
-    }[]
-
-    // Phase 29: Count linked memories per file from observations table
-    const memoryCounts = this.getMemoryCounts(project)
-
-    return rows.map(row => ({
-      path: row.file_path,
-      language: row.language ?? undefined,
-      summary: row.summary ?? undefined,
-      symbols: row.symbols ? [...new Set(row.symbols.split(',').filter(Boolean))] : [],
-      linkedMemories: memoryCounts.get(row.file_path) ?? undefined,
-    }))
-  }
-
-  /** Phase 29: Query observations table for file path reference counts */
-  private getMemoryCounts(project: string): Map<string, number> {
-    const counts = new Map<string, number>()
-    if (!this.memoryDb) return counts
-
-    try {
-      const rows = this.memoryDb.prepare(
-        `SELECT file_paths FROM observations
-         WHERE project = ? AND file_paths IS NOT NULL AND archived = 0`
-      ).all(project) as { file_paths: string }[]
-
-      for (const row of rows) {
-        try {
-          const paths: string[] = JSON.parse(row.file_paths)
-          if (Array.isArray(paths)) {
-            for (const fp of paths) {
-              counts.set(fp, (counts.get(fp) || 0) + 1)
-            }
-          }
-        } catch {
-          // Skip malformed JSON
-        }
-      }
-    } catch (error) {
-      this.logger.debug({ error }, 'Failed to query memory counts from observations')
-    }
-
-    return counts
-  }
-
-  formatFileMap(entries: FileMapEntry[]): string {
-    const lines: string[] = []
-    let totalSize = 0
-    const MAX_SIZE = 4096 // 4KB budget
-
-    for (const entry of entries) {
-      let line = entry.path
-      if (entry.symbols.length > 0) {
-        const symbolStr = entry.symbols.length <= 5
-          ? entry.symbols.join(', ')
-          : `${entry.symbols.slice(0, 5).join(', ')} +${entry.symbols.length - 5}`
-        line += ` — ${symbolStr}`
-      }
-      if (entry.summary) {
-        line += ` — ${entry.summary}`
-      }
-      // Phase 29: Append linked memory count annotation
-      if (entry.linkedMemories && entry.linkedMemories > 0) {
-        line += ` [${entry.linkedMemories} ${entry.linkedMemories === 1 ? 'memory' : 'memories'}]`
-      }
-
-      // Check if adding this line would exceed budget
-      if (totalSize + line.length + 1 > MAX_SIZE) break
-      totalSize += line.length + 1
-      lines.push(line)
-    }
-
-    return lines.join('\n')
-  }
-
-  getStats(project: string): IndexStats {
-    const fileRow = this.db.query(
-      'SELECT COUNT(*) as cnt FROM code_files WHERE project = ?'
-    ).get(project) as { cnt: number } | null
-
-    const symbolRow = this.db.query(
-      'SELECT COUNT(*) as cnt FROM code_symbols WHERE project = ?'
-    ).get(project) as { cnt: number } | null
-
-    const lastRow = this.db.query(
-      'SELECT MAX(last_indexed) as last_indexed FROM code_files WHERE project = ?'
-    ).get(project) as { last_indexed: string | null } | null
-
-    return {
-      project,
-      filesIndexed: fileRow?.cnt ?? 0,
-      totalSymbols: symbolRow?.cnt ?? 0,
-      lastIndexed: lastRow?.last_indexed ?? null,
-      staleFiles: 0,
-    }
-  }
-}
-
-// ─── Internal Row Types ─────────────────────────────────────
-
-interface RawSymbolRow {
-  symbol_name: string
-  symbol_type: string
-  file_path: string
-  line_start: number
-  line_end: number | null
-  signature: string | null
-}
-
-interface RawFileRow {
-  file_path: string
-  language: string | null
-  summary: string | null
-  symbol_count: number
-  line_count: number
-}
+import { Database } from 'bun:sqlite'
+import type { Logger } from 'pino'
+import type { SymbolResult, FileResult, DependencyResult, FileMapEntry, IndexStats, SymbolType } from './types'
+
+export class CodeQuery {
+  private db: Database
+  private logger: Logger
+  private memoryDb?: Database
+
+  constructor(db: Database, logger: Logger) {
+    this.db = db
+    this.logger = logger.child({ component: 'code-query' })
+  }
+
+  /** Phase 29: Set memory DB for file-linked memory annotations */
+  setMemoryDb(db: Database): void {
+    this.memoryDb = db
+  }
+
+  findSymbols(query: string, project: string, limit: number = 20): SymbolResult[] {
+    if (!query.trim()) return []
+
+    // First try exact match
+    const exact = this.db.query(
+      `SELECT symbol_name, symbol_type, file_path, line_start, line_end, signature
+       FROM code_symbols
+       WHERE project = ? AND symbol_name = ?
+       LIMIT ?`
+    ).all(project, query, limit) as RawSymbolRow[]
+
+    if (exact.length > 0) {
+      return exact.map(row => ({
+        symbol: row.symbol_name,
+        type: row.symbol_type as SymbolType,
+        filePath: row.file_path,
+        lineStart: row.line_start,
+        lineEnd: row.line_end ?? undefined,
+        signature: row.signature ?? undefined,
+        confidence: 1.0,
+      }))
+    }
+
+    // Try prefix match
+    const prefix = this.db.query(
+      `SELECT symbol_name, symbol_type, file_path, line_start, line_end, signature
+       FROM code_symbols
+       WHERE project = ? AND symbol_name LIKE ? || '%'
+       LIMIT ?`
+    ).all(project, query, limit) as RawSymbolRow[]
+
+    if (prefix.length > 0) {
+      return prefix.map(row => ({
+        symbol: row.symbol_name,
+        type: row.symbol_type as SymbolType,
+        filePath: row.file_path,
+        lineStart: row.line_start,
+        lineEnd: row.line_end ?? undefined,
+        signature: row.signature ?? undefined,
+        confidence: 0.9,
+      }))
+    }
+
+    // Fall back to FTS5 search
+    try {
+      const ftsResults = this.db.query(
+        `SELECT s.symbol_name, s.symbol_type, s.file_path, s.line_start, s.line_end, s.signature,
+                rank
+         FROM code_symbols_fts fts
+         JOIN code_symbols s ON s.id = fts.rowid
+         WHERE code_symbols_fts MATCH ? AND s.project = ?
+         ORDER BY rank
+         LIMIT ?`
+      ).all(query, project, limit) as (RawSymbolRow & { rank: number })[]
+
+      if (ftsResults.length === 0) return []
+
+      // Normalize FTS5 rank to 0.5-0.85 confidence
+      const maxRank = Math.abs(ftsResults[0].rank) || 1
+      return ftsResults.map(row => ({
+        symbol: row.symbol_name,
+        type: row.symbol_type as SymbolType,
+        filePath: row.file_path,
+        lineStart: row.line_start,
+        lineEnd: row.line_end ?? undefined,
+        signature: row.signature ?? undefined,
+        confidence: 0.5 + 0.35 * (Math.abs(row.rank) / maxRank),
+      }))
+    } catch (error) {
+      // FTS5 can throw on invalid query syntax
+      this.logger.debug({ error, query }, 'FTS5 query failed, falling back to LIKE')
+
+      const likeResults = this.db.query(
+        `SELECT symbol_name, symbol_type, file_path, line_start, line_end, signature
+         FROM code_symbols
+         WHERE project = ? AND symbol_name LIKE '%' || ? || '%'
+         LIMIT ?`
+      ).all(project, query, limit) as RawSymbolRow[]
+
+      return likeResults.map(row => ({
+        symbol: row.symbol_name,
+        type: row.symbol_type as SymbolType,
+        filePath: row.file_path,
+        lineStart: row.line_start,
+        lineEnd: row.line_end ?? undefined,
+        signature: row.signature ?? undefined,
+        confidence: 0.5,
+      }))
+    }
+  }
+
+  findFiles(query: string, project: string): FileResult[] {
+    if (!query.trim()) return []
+
+    const rows = this.db.query(
+      `SELECT file_path, language, summary, symbol_count, line_count
+       FROM code_files
+       WHERE project = ? AND (file_path LIKE '%' || ? || '%' OR summary LIKE '%' || ? || '%')
+       ORDER BY file_path
+       LIMIT 50`
+    ).all(project, query, query) as RawFileRow[]
+
+    return rows.map(row => ({
+      filePath: row.file_path,
+      language: row.language ?? undefined,
+      summary: row.summary ?? undefined,
+      symbolCount: row.symbol_count,
+      lineCount: row.line_count,
+    }))
+  }
+
+  getDependencies(filePath: string, project: string): DependencyResult {
+    const imports = this.db.query(
+      `SELECT target_file, import_names
+       FROM code_dependencies
+       WHERE project = ? AND source_file = ?`
+    ).all(project, filePath) as { target_file: string; import_names: string | null }[]
+
+    const importedBy = this.db.query(
+      `SELECT source_file, import_names
+       FROM code_dependencies
+       WHERE project = ? AND target_file = ?`
+    ).all(project, filePath) as { source_file: string; import_names: string | null }[]
+
+    return {
+      file: filePath,
+      imports: imports.map(row => ({
+        file: row.target_file,
+        names: row.import_names ? row.import_names.split(', ').filter(Boolean) : [],
+      })),
+      importedBy: importedBy.map(row => ({
+        file: row.source_file,
+        names: row.import_names ? row.import_names.split(', ').filter(Boolean) : [],
+      })),
+    }
+  }
+
+  getDependents(filePath: string, project: string): DependencyResult {
+    // getDependents is the inverse view — same data, different perspective
+    return this.getDependencies(filePath, project)
+  }
+
+  getFileMap(project: string, maxEntries: number = 100): FileMapEntry[] {
+    const rows = this.db.query(
+      `SELECT f.file_path, f.language, f.summary,
+              GROUP_CONCAT(CASE WHEN s.exported = 1 THEN s.symbol_name END) as symbols
+       FROM code_files f
+       LEFT JOIN code_symbols s ON s.project = f.project AND s.file_path = f.file_path
+       WHERE f.project = ?
+       GROUP BY f.file_path
+       ORDER BY f.file_path
+       LIMIT ?`
+    ).all(project, maxEntries) as {
+      file_path: string
+      language: string | null
+      summary: string | null
+      symbols: string | null
+    }[]
+
+    // Phase 29: Count linked memories per file from observations table
+    const memoryCounts = this.getMemoryCounts(project)
+
+    return rows.map(row => ({
+      path: row.file_path,
+      language: row.language ?? undefined,
+      summary: row.summary ?? undefined,
+      symbols: row.symbols ? [...new Set(row.symbols.split(',').filter(Boolean))] : [],
+      linkedMemories: memoryCounts.get(row.file_path) ?? undefined,
+    }))
+  }
+
+  /** Phase 29: Query observations table for file path reference counts */
+  private getMemoryCounts(project: string): Map<string, number> {
+    const counts = new Map<string, number>()
+    if (!this.memoryDb) return counts
+
+    try {
+      const rows = this.memoryDb.prepare(
+        `SELECT file_paths FROM observations
+         WHERE project = ? AND file_paths IS NOT NULL AND archived = 0`
+      ).all(project) as { file_paths: string }[]
+
+      for (const row of rows) {
+        try {
+          const paths: string[] = JSON.parse(row.file_paths)
+          if (Array.isArray(paths)) {
+            for (const fp of paths) {
+              counts.set(fp, (counts.get(fp) || 0) + 1)
+            }
+          }
+        } catch {
+          // Skip malformed JSON
+        }
+      }
+    } catch (error) {
+      this.logger.debug({ error }, 'Failed to query memory counts from observations')
+    }
+
+    return counts
+  }
+
+  formatFileMap(entries: FileMapEntry[]): string {
+    const lines: string[] = []
+    let totalSize = 0
+    const MAX_SIZE = 4096 // 4KB budget
+
+    for (const entry of entries) {
+      let line = entry.path
+      if (entry.symbols.length > 0) {
+        const symbolStr = entry.symbols.length <= 5
+          ? entry.symbols.join(', ')
+          : `${entry.symbols.slice(0, 5).join(', ')} +${entry.symbols.length - 5}`
+        line += ` — ${symbolStr}`
+      }
+      if (entry.summary) {
+        line += ` — ${entry.summary}`
+      }
+      // Phase 29: Append linked memory count annotation
+      if (entry.linkedMemories && entry.linkedMemories > 0) {
+        line += ` [${entry.linkedMemories} ${entry.linkedMemories === 1 ? 'memory' : 'memories'}]`
+      }
+
+      // Check if adding this line would exceed budget
+      if (totalSize + line.length + 1 > MAX_SIZE) break
+      totalSize += line.length + 1
+      lines.push(line)
+    }
+
+    return lines.join('\n')
+  }
+
+  getStats(project: string): IndexStats {
+    const fileRow = this.db.query(
+      'SELECT COUNT(*) as cnt FROM code_files WHERE project = ?'
+    ).get(project) as { cnt: number } | null
+
+    const symbolRow = this.db.query(
+      'SELECT COUNT(*) as cnt FROM code_symbols WHERE project = ?'
+    ).get(project) as { cnt: number } | null
+
+    const lastRow = this.db.query(
+      'SELECT MAX(last_indexed) as last_indexed FROM code_files WHERE project = ?'
+    ).get(project) as { last_indexed: string | null } | null
+
+    return {
+      project,
+      filesIndexed: fileRow?.cnt ?? 0,
+      totalSymbols: symbolRow?.cnt ?? 0,
+      lastIndexed: lastRow?.last_indexed ?? null,
+      staleFiles: 0,
+    }
+  }
+}
+
+// ─── Internal Row Types ─────────────────────────────────────
+
+interface RawSymbolRow {
+  symbol_name: string
+  symbol_type: string
+  file_path: string
+  line_start: number
+  line_end: number | null
+  signature: string | null
+}
+
+interface RawFileRow {
+  file_path: string
+  language: string | null
+  summary: string | null
+  symbol_count: number
+  line_count: number
+}