bluera-knowledge 0.11.9 → 0.11.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bluera-knowledge",
-  "version": "0.11.9",
+  "version": "0.11.10",
   "description": "CLI tool for managing knowledge stores with semantic search",
   "type": "module",
   "bin": {

package/src/plugin/commands.test.ts

@@ -9,9 +9,10 @@ import {
 } from './commands.js';
 import type { ServiceContainer } from '../services/index.js';
 
-// Mock the createServices function
+// Mock the createServices and destroyServices functions
 vi.mock('../services/index.js', () => ({
   createServices: vi.fn(),
+  destroyServices: vi.fn().mockResolvedValue(undefined),
 }));
 
 // Mock extractRepoName

package/src/plugin/commands.ts

@@ -2,7 +2,7 @@ import ora from 'ora';
 import { extractRepoName } from './git-clone.js';
 import { DependencyUsageAnalyzer } from '../analysis/dependency-usage-analyzer.js';
 import { RepoUrlResolver } from '../analysis/repo-url-resolver.js';
-import { createServices } from '../services/index.js';
+import { createServices, destroyServices } from '../services/index.js';
 
 /**
  * Options passed from CLI global options to plugin command handlers.
@@ -22,37 +22,42 @@ export async function handleSearch(args: {
 }): Promise<void> {
   // PWD is set by Claude Code to user's project directory
   const services = await createServices(undefined, undefined, process.env['PWD']);
-  const storeNames = args.stores?.split(',').map((s: string) => s.trim());
 
-  const allStores = await services.store.list();
-  const targetStores =
-    storeNames !== undefined ? allStores.filter((s) => storeNames.includes(s.name)) : allStores;
+  try {
+    const storeNames = args.stores?.split(',').map((s: string) => s.trim());
 
-  if (targetStores.length === 0) {
-    console.error('No stores found to search');
-    process.exit(1);
-  }
+    const allStores = await services.store.list();
+    const targetStores =
+      storeNames !== undefined ? allStores.filter((s) => storeNames.includes(s.name)) : allStores;
 
-  // Initialize stores
-  for (const store of targetStores) {
-    await services.lance.initialize(store.id);
-  }
+    if (targetStores.length === 0) {
+      console.error('No stores found to search');
+      process.exit(1);
+    }
 
-  const results = await services.search.search({
-    query: args.query,
-    stores: targetStores.map((s) => s.id),
-    mode: 'hybrid',
-    limit: parseInt(args.limit ?? '10', 10),
-    detail: 'contextual',
-  });
-
-  console.log(`Found ${String(results.totalResults)} results:\n`);
-  for (const r of results.results) {
-    if (r.summary !== undefined) {
-      console.log(`Score: ${r.score.toFixed(2)} - ${r.summary.location}`);
-      console.log(r.summary.purpose);
+    // Initialize stores
+    for (const store of targetStores) {
+      await services.lance.initialize(store.id);
     }
-    console.log('---');
+
+    const results = await services.search.search({
+      query: args.query,
+      stores: targetStores.map((s) => s.id),
+      mode: 'hybrid',
+      limit: parseInt(args.limit ?? '10', 10),
+      detail: 'contextual',
+    });
+
+    console.log(`Found ${String(results.totalResults)} results:\n`);
+    for (const r of results.results) {
+      if (r.summary !== undefined) {
+        console.log(`Score: ${r.score.toFixed(2)} - ${r.summary.location}`);
+        console.log(r.summary.purpose);
+      }
+      console.log('---');
+    }
+  } finally {
+    await destroyServices(services);
   }
 }
 
@@ -69,35 +74,40 @@ export async function handleAddRepo(
     options.dataDir,
     options.projectRoot ?? process.env['PWD']
   );
-  const storeName = args.name ?? extractRepoName(args.url);
 
-  console.log(`Cloning ${args.url}...`);
+  try {
+    const storeName = args.name ?? extractRepoName(args.url);
 
-  const result = await services.store.create({
-    name: storeName,
-    type: 'repo',
-    url: args.url,
-    ...(args.branch !== undefined ? { branch: args.branch } : {}),
-  });
+    console.log(`Cloning ${args.url}...`);
 
-  if (!result.success) {
-    console.error(`Error: ${result.error.message}`);
-    process.exit(1);
-  }
+    const result = await services.store.create({
+      name: storeName,
+      type: 'repo',
+      url: args.url,
+      ...(args.branch !== undefined ? { branch: args.branch } : {}),
+    });
 
-  console.log(`Created store: ${storeName} (${result.data.id})`);
-  if ('path' in result.data) {
-    console.log(`Location: ${result.data.path}`);
-  }
+    if (!result.success) {
+      console.error(`Error: ${result.error.message}`);
+      process.exit(1);
+    }
+
+    console.log(`Created store: ${storeName} (${result.data.id})`);
+    if ('path' in result.data) {
+      console.log(`Location: ${result.data.path}`);
+    }
 
-  // Auto-index
-  console.log('\nIndexing...');
-  const indexResult = await services.index.indexStore(result.data);
+    // Auto-index
+    console.log('\nIndexing...');
+    const indexResult = await services.index.indexStore(result.data);
 
-  if (indexResult.success) {
-    console.log(`Indexed ${String(indexResult.data.documentsIndexed)} files`);
-  } else {
-    console.error(`Indexing failed: ${indexResult.error.message}`);
+    if (indexResult.success) {
+      console.log(`Indexed ${String(indexResult.data.documentsIndexed)} files`);
+    } else {
+      console.error(`Indexing failed: ${indexResult.error.message}`);
+    }
+  } finally {
+    await destroyServices(services);
   }
 }
 
@@ -110,58 +120,68 @@ export async function handleAddFolder(
     options.dataDir,
     options.projectRoot ?? process.env['PWD']
   );
-  const { basename } = await import('node:path');
-  const storeName = args.name ?? basename(args.path);
 
-  console.log(`Adding folder: ${args.path}...`);
+  try {
+    const { basename } = await import('node:path');
+    const storeName = args.name ?? basename(args.path);
 
-  const result = await services.store.create({
-    name: storeName,
-    type: 'file',
-    path: args.path,
-  });
+    console.log(`Adding folder: ${args.path}...`);
 
-  if (!result.success) {
-    console.error(`Error: ${result.error.message}`);
-    process.exit(1);
-  }
+    const result = await services.store.create({
+      name: storeName,
+      type: 'file',
+      path: args.path,
+    });
 
-  console.log(`Created store: ${storeName} (${result.data.id})`);
-  if ('path' in result.data) {
-    console.log(`Location: ${result.data.path}`);
-  }
+    if (!result.success) {
+      console.error(`Error: ${result.error.message}`);
+      process.exit(1);
+    }
+
+    console.log(`Created store: ${storeName} (${result.data.id})`);
+    if ('path' in result.data) {
+      console.log(`Location: ${result.data.path}`);
+    }
 
-  // Auto-index
-  console.log('\nIndexing...');
-  const indexResult = await services.index.indexStore(result.data);
+    // Auto-index
+    console.log('\nIndexing...');
+    const indexResult = await services.index.indexStore(result.data);
 
-  if (indexResult.success) {
-    console.log(`Indexed ${String(indexResult.data.documentsIndexed)} files`);
-  } else {
-    console.error(`Indexing failed: ${indexResult.error.message}`);
+    if (indexResult.success) {
+      console.log(`Indexed ${String(indexResult.data.documentsIndexed)} files`);
+    } else {
+      console.error(`Indexing failed: ${indexResult.error.message}`);
+    }
+  } finally {
+    await destroyServices(services);
   }
 }
 
 export async function handleIndex(args: { store: string }): Promise<void> {
   // PWD is set by Claude Code to user's project directory
   const services = await createServices(undefined, undefined, process.env['PWD']);
-  const store = await services.store.getByIdOrName(args.store);
 
-  if (store === undefined) {
-    console.error(`Store not found: ${args.store}`);
-    process.exit(1);
-  }
+  try {
+    const store = await services.store.getByIdOrName(args.store);
 
-  console.log(`Indexing ${store.name}...`);
-  const result = await services.index.indexStore(store);
+    if (store === undefined) {
+      console.error(`Store not found: ${args.store}`);
+      process.exit(1);
+    }
 
-  if (result.success) {
-    console.log(
-      `Indexed ${String(result.data.documentsIndexed)} documents in ${String(result.data.timeMs)}ms`
-    );
-  } else {
-    console.error(`Error: ${result.error.message}`);
-    process.exit(1);
+    console.log(`Indexing ${store.name}...`);
+    const result = await services.index.indexStore(store);
+
+    if (result.success) {
+      console.log(
+        `Indexed ${String(result.data.documentsIndexed)} documents in ${String(result.data.timeMs)}ms`
+      );
+    } else {
+      console.error(`Error: ${result.error.message}`);
+      process.exit(1);
+    }
+  } finally {
+    await destroyServices(services);
   }
 }
 
@@ -171,34 +191,39 @@ export async function handleStores(options: CommandOptions = {}): Promise<void>
     options.dataDir,
     options.projectRoot ?? process.env['PWD']
   );
-  const stores = await services.store.list();
-
-  if (stores.length === 0) {
-    console.log('No stores found.');
-    console.log('\nCreate a store with:');
-    console.log('  /bluera-knowledge:add-repo <url> --name=<name>');
-    console.log('  /bluera-knowledge:add-folder <path> --name=<name>');
-    return;
-  }
 
-  // Table header
-  console.log('| Name | Type | ID | Source |');
-  console.log('|------|------|----|--------------------|');
-
-  // Table rows
-  for (const store of stores) {
-    const name = store.name;
-    const type = store.type;
-    const id = store.id;
-    let source = '';
-
-    if ('url' in store && store.url !== undefined) {
-      source = store.url;
-    } else if ('path' in store) {
-      source = store.path;
+  try {
+    const stores = await services.store.list();
+
+    if (stores.length === 0) {
+      console.log('No stores found.');
+      console.log('\nCreate a store with:');
+      console.log('  /bluera-knowledge:add-repo <url> --name=<name>');
+      console.log('  /bluera-knowledge:add-folder <path> --name=<name>');
+      return;
     }
 
-    console.log(`| ${name} | ${type} | ${id.substring(0, 8)}... | ${source} |`);
+    // Table header
+    console.log('| Name | Type | ID | Source |');
+    console.log('|------|------|----|--------------------|');
+
+    // Table rows
+    for (const store of stores) {
+      const name = store.name;
+      const type = store.type;
+      const id = store.id;
+      let source = '';
+
+      if ('url' in store && store.url !== undefined) {
+        source = store.url;
+      } else if ('path' in store) {
+        source = store.path;
+      }
+
+      console.log(`| ${name} | ${type} | ${id.substring(0, 8)}... | ${source} |`);
+    }
+  } finally {
+    await destroyServices(services);
   }
 }
 
@@ -209,73 +234,78 @@ export async function handleSuggest(options: CommandOptions = {}): Promise<void>
 
   // Create analyzer instance
   const services = await createServices(options.config, options.dataDir, projectRoot);
-  const analyzer = new DependencyUsageAnalyzer();
-  const resolver = new RepoUrlResolver();
-
-  // Analyze with progress indicator
-  const spinner = ora('Scanning source files...').start();
 
-  const result = await analyzer.analyze(projectRoot, (current, total, message) => {
-    spinner.text = `${message} (${String(current)}/${String(total)})`;
-  });
+  try {
+    const analyzer = new DependencyUsageAnalyzer();
+    const resolver = new RepoUrlResolver();
 
-  spinner.stop();
+    // Analyze with progress indicator
+    const spinner = ora('Scanning source files...').start();
 
-  if (!result.success) {
-    console.error(`Error: ${result.error.message}`);
-    process.exit(1);
-  }
-
-  const { usages, totalFilesScanned, skippedFiles } = result.data;
+    const result = await analyzer.analyze(projectRoot, (current, total, message) => {
+      spinner.text = `${message} (${String(current)}/${String(total)})`;
+    });
 
-  console.log(
-    `✔ Scanned ${String(totalFilesScanned)} files${skippedFiles > 0 ? ` (skipped ${String(skippedFiles)})` : ''}\n`
-  );
+    spinner.stop();
 
-  if (usages.length === 0) {
-    console.log('No external dependencies found in this project.');
-    console.log('\nMake sure you have a package.json or requirements.txt file.');
-    return;
-  }
+    if (!result.success) {
+      console.error(`Error: ${result.error.message}`);
+      process.exit(1);
+    }
 
-  // Filter out packages already in stores
-  const existingStores = await services.store.list();
-  const existingRepoNames = new Set(existingStores.map((s) => s.name));
+    const { usages, totalFilesScanned, skippedFiles } = result.data;
 
-  const newUsages = usages.filter((u) => !existingRepoNames.has(u.packageName));
+    console.log(
+      `✔ Scanned ${String(totalFilesScanned)} files${skippedFiles > 0 ? ` (skipped ${String(skippedFiles)})` : ''}\n`
+    );
 
-  if (newUsages.length === 0) {
-    console.log('✔ All dependencies are already in knowledge stores!');
-    return;
-  }
+    if (usages.length === 0) {
+      console.log('No external dependencies found in this project.');
+      console.log('\nMake sure you have a package.json or requirements.txt file.');
+      return;
+    }
 
-  // Show top 5 suggestions
-  const topSuggestions = newUsages.slice(0, 5);
+    // Filter out packages already in stores
+    const existingStores = await services.store.list();
+    const existingRepoNames = new Set(existingStores.map((s) => s.name));
 
-  console.log('Top dependencies by usage in this project:\n');
-  topSuggestions.forEach((usage, i) => {
-    console.log(`${String(i + 1)}. ${usage.packageName}`);
-    console.log(
-      `   ${String(usage.importCount)} imports across ${String(usage.fileCount)} files\n`
-    );
-  });
+    const newUsages = usages.filter((u) => !existingRepoNames.has(u.packageName));
 
-  console.log('Searching for repository URLs...\n');
+    if (newUsages.length === 0) {
+      console.log('✔ All dependencies are already in knowledge stores!');
+      return;
+    }
 
-  // For each package, find repo URL
-  for (const usage of topSuggestions) {
-    const repoResult = await resolver.findRepoUrl(usage.packageName, usage.language);
+    // Show top 5 suggestions
+    const topSuggestions = newUsages.slice(0, 5);
 
-    if (repoResult.url !== null) {
-      console.log(`✔ ${usage.packageName}: ${repoResult.url}`);
-      console.log(`  /bluera-knowledge:add-repo ${repoResult.url} --name=${usage.packageName}\n`);
-    } else {
-      console.log(`✗ ${usage.packageName}: Could not find repository URL`);
+    console.log('Top dependencies by usage in this project:\n');
+    topSuggestions.forEach((usage, i) => {
+      console.log(`${String(i + 1)}. ${usage.packageName}`);
       console.log(
-        `  You can manually add it: /bluera-knowledge:add-repo <url> --name=${usage.packageName}\n`
+        `   ${String(usage.importCount)} imports across ${String(usage.fileCount)} files\n`
       );
+    });
+
+    console.log('Searching for repository URLs...\n');
+
+    // For each package, find repo URL
+    for (const usage of topSuggestions) {
+      const repoResult = await resolver.findRepoUrl(usage.packageName, usage.language);
+
+      if (repoResult.url !== null) {
+        console.log(`✔ ${usage.packageName}: ${repoResult.url}`);
+        console.log(`  /bluera-knowledge:add-repo ${repoResult.url} --name=${usage.packageName}\n`);
+      } else {
+        console.log(`✗ ${usage.packageName}: Could not find repository URL`);
+        console.log(
+          `  You can manually add it: /bluera-knowledge:add-repo <url> --name=${usage.packageName}\n`
+        );
+      }
     }
-  }
 
-  console.log('Use the commands above to add repositories to your knowledge stores.');
+    console.log('Use the commands above to add repositories to your knowledge stores.');
+  } finally {
+    await destroyServices(services);
+  }
 }