@wundr.io/cli 1.0.11 → 1.0.12

package/src/commands/rag.ts

@@ -67,8 +67,8 @@ function getTimestamp(): string {
 
 function formatBytes(bytes: number): string {
   if (bytes === 0) {
-return '0 Bytes';
-}
+    return '0 Bytes';
+  }
   const k = 1024;
   const sizes = ['Bytes', 'KB', 'MB', 'GB'];
   const i = Math.floor(Math.log(bytes) / Math.log(k));
@@ -77,8 +77,8 @@ return '0 Bytes';
 
 function getDirSize(dirPath: string): number {
   if (!existsSync(dirPath)) {
-return 0;
-}
+    return 0;
+  }
 
   let totalSize = 0;
   try {
@@ -208,7 +208,9 @@ async function getStoreMetadata(storePath: string): Promise<StoreMetadata> {
   let lastPrune: string | null = null;
   if (existsSync(pruneMetadataPath)) {
     try {
-      const pruneData = JSON.parse(await fs.readFile(pruneMetadataPath, 'utf-8'));
+      const pruneData = JSON.parse(
+        await fs.readFile(pruneMetadataPath, 'utf-8')
+      );
       lastPrune = pruneData.lastPrune || null;
     } catch {
       // Ignore parse errors
@@ -218,7 +220,9 @@ async function getStoreMetadata(storePath: string): Promise<StoreMetadata> {
   let lastIndex: string | null = null;
   if (existsSync(indexMetadataPath)) {
     try {
-      const indexData = JSON.parse(await fs.readFile(indexMetadataPath, 'utf-8'));
+      const indexData = JSON.parse(
+        await fs.readFile(indexMetadataPath, 'utf-8')
+      );
       lastIndex = indexData.lastIndex || null;
     } catch {
       // Ignore parse errors
@@ -240,7 +244,9 @@ async function getStoreMetadata(storePath: string): Promise<StoreMetadata> {
 // Create RAG command
 export function createRAGCommand(): Command {
   const command = new Command('rag')
-    .description('Manage RAG (Retrieval-Augmented Generation) stores for AI-powered code understanding')
+    .description(
+      'Manage RAG (Retrieval-Augmented Generation) stores for AI-powered code understanding'
+    )
     .addHelpText(
       'after',
       chalk.gray(`
@@ -251,7 +257,7 @@ Examples:
   ${chalk.green('wundr rag reindex')}          Re-index stores
   ${chalk.green('wundr rag create myproject')} Create a new project-specific store
   ${chalk.green('wundr rag setup')}            Run initial RAG infrastructure setup
-      `),
+      `)
     );
 
   // Status command (default)
@@ -259,7 +265,7 @@ Examples:
     .command('status', { isDefault: true })
     .description('Show RAG store status and statistics')
     .option('--json', 'Output as JSON')
-    .action(async (options) => {
+    .action(async options => {
       await showStatus(options);
     });
 
@@ -269,7 +275,7 @@ Examples:
     .description('Sync all RAG stores with their source directories')
     .option('--store <name>', 'Only sync specific store')
     .option('--dry-run', 'Show what would be synced without making changes')
-    .action(async (options) => {
+    .action(async options => {
       await syncStores(options);
     });
 
@@ -279,7 +285,7 @@ Examples:
     .description('Remove deleted files from RAG stores')
     .option('--store <name>', 'Only prune specific store')
     .option('--dry-run', 'Show what would be pruned without making changes')
-    .action(async (options) => {
+    .action(async options => {
       await pruneStores(options);
     });
 
@@ -289,7 +295,7 @@ Examples:
     .description('Re-index RAG stores with updated configurations')
     .option('--store <name>', 'Only reindex specific store')
     .option('--force', 'Force complete reindex')
-    .action(async (options) => {
+    .action(async options => {
       await reindexStores(options);
     });
 
@@ -309,7 +315,7 @@ Examples:
     .command('setup')
     .description('Run initial RAG infrastructure setup')
     .option('--skip-api-key', 'Skip GEMINI_API_KEY configuration')
-    .action(async (options) => {
+    .action(async options => {
       await runSetup(options);
     });
 
@@ -320,7 +326,7 @@ Examples:
     .option('--get <key>', 'Get configuration value')
     .option('--set <key=value>', 'Set configuration value')
     .option('--reset', 'Reset to default configuration')
-    .action(async (options) => {
+    .action(async options => {
       await manageConfig(options);
     });
 
@@ -357,13 +363,19 @@ async function showStatus(options: { json?: boolean }): Promise<void> {
     spinner.stop();
 
     if (options.json) {
-      console.log(JSON.stringify({
-        timestamp: getTimestamp(),
-        stores: storeMetadata,
-        totalEmbeddings,
-        totalDiskUsage: formatBytes(totalSize),
-        configPath: CONFIG_FILE,
-      }, null, 2));
+      console.log(
+        JSON.stringify(
+          {
+            timestamp: getTimestamp(),
+            stores: storeMetadata,
+            totalEmbeddings,
+            totalDiskUsage: formatBytes(totalSize),
+            configPath: CONFIG_FILE,
+          },
+          null,
+          2
+        )
+      );
       return;
     }
 
@@ -371,21 +383,28 @@ async function showStatus(options: { json?: boolean }): Promise<void> {
     console.log(chalk.gray('='.repeat(70)));
     console.log(chalk.white('Config:'), chalk.gray(CONFIG_FILE));
     console.log(chalk.white('Total Embeddings:'), chalk.green(totalEmbeddings));
-    console.log(chalk.white('Disk Usage:'), chalk.green(formatBytes(totalSize)));
+    console.log(
+      chalk.white('Disk Usage:'),
+      chalk.green(formatBytes(totalSize))
+    );
     console.log(chalk.gray('-'.repeat(70)));
 
     if (storeMetadata.length === 0) {
-      console.log(chalk.yellow('\nNo RAG stores found. Run "wundr rag setup" to get started.\n'));
+      console.log(
+        chalk.yellow(
+          '\nNo RAG stores found. Run "wundr rag setup" to get started.\n'
+        )
+      );
       return;
     }
 
     console.log(
       chalk.cyan(
         padRight('Store', 25) +
-        padRight('Embeddings', 12) +
-        padRight('Last Sync', 18) +
-        padRight('Status', 15),
-      ),
+          padRight('Embeddings', 12) +
+          padRight('Last Sync', 18) +
+          padRight('Status', 15)
+      )
     );
     console.log(chalk.gray('-'.repeat(70)));
 
@@ -396,17 +415,17 @@ async function showStatus(options: { json?: boolean }): Promise<void> {
 
       let statusColor = chalk.yellow;
       if (store.status === 'ready') {
-statusColor = chalk.green;
-}
+        statusColor = chalk.green;
+      }
       if (store.status === 'error') {
-statusColor = chalk.red;
-}
+        statusColor = chalk.red;
+      }
 
       console.log(
         padRight(store.name, 25) +
-        padRight(String(store.embeddingCount), 12) +
-        padRight(lastSync, 18) +
-        statusColor(padRight(store.status, 15)),
+          padRight(String(store.embeddingCount), 12) +
+          padRight(lastSync, 18) +
+          statusColor(padRight(store.status, 15))
       );
     }
 
@@ -414,11 +433,16 @@ statusColor = chalk.red;
     console.log('');
   } catch (error) {
     spinner.fail('Failed to load RAG status');
-    console.error(chalk.red(error instanceof Error ? error.message : String(error)));
+    console.error(
+      chalk.red(error instanceof Error ? error.message : String(error))
+    );
   }
 }
 
-async function syncStores(options: { store?: string; dryRun?: boolean }): Promise<void> {
+async function syncStores(options: {
+  store?: string;
+  dryRun?: boolean;
+}): Promise<void> {
   const spinner = ora('Syncing RAG stores...').start();
 
   try {
@@ -436,7 +460,11 @@ async function syncStores(options: { store?: string; dryRun?: boolean }): Promis
         continue;
       }
 
-      const sourceMetadataPath = path.join(storePath, 'metadata', 'source.json');
+      const sourceMetadataPath = path.join(
+        storePath,
+        'metadata',
+        'source.json'
+      );
 
       if (!existsSync(sourceMetadataPath)) {
         console.log(chalk.yellow(`  [SKIP] ${storeName}: No source metadata`));
@@ -445,11 +473,15 @@ async function syncStores(options: { store?: string; dryRun?: boolean }): Promis
       }
 
       try {
-        const sourceMetadata = JSON.parse(await fs.readFile(sourceMetadataPath, 'utf-8'));
+        const sourceMetadata = JSON.parse(
+          await fs.readFile(sourceMetadataPath, 'utf-8')
+        );
         const sourceDir = sourceMetadata.sourceDir;
 
         if (!sourceDir || !existsSync(sourceDir)) {
-          console.log(chalk.yellow(`  [SKIP] ${storeName}: Source directory not found`));
+          console.log(
+            chalk.yellow(`  [SKIP] ${storeName}: Source directory not found`)
+          );
           skipped++;
           continue;
         }
@@ -458,16 +490,23 @@ async function syncStores(options: { store?: string; dryRun?: boolean }): Promis
         let fileCount = 0;
         const countFiles = (dir: string): void => {
           if (!existsSync(dir)) {
-return;
-}
+            return;
+          }
           try {
             const entries = readdirSync(dir);
             for (const entry of entries) {
               const fullPath = path.join(dir, entry);
               const stat = statSync(fullPath);
-              if (stat.isDirectory() && !entry.startsWith('.') && entry !== 'node_modules') {
+              if (
+                stat.isDirectory() &&
+                !entry.startsWith('.') &&
+                entry !== 'node_modules'
+              ) {
                 countFiles(fullPath);
-              } else if (stat.isFile() && /\.(ts|js|tsx|jsx|md|json)$/.test(entry)) {
+              } else if (
+                stat.isFile() &&
+                /\.(ts|js|tsx|jsx|md|json)$/.test(entry)
+              ) {
                 fileCount++;
               }
             }
@@ -478,7 +517,11 @@ return;
         countFiles(sourceDir);
 
         if (options.dryRun) {
-          console.log(chalk.blue(`  [DRY-RUN] ${storeName}: Would sync ${fileCount} files`));
+          console.log(
+            chalk.blue(
+              `  [DRY-RUN] ${storeName}: Would sync ${fileCount} files`
+            )
+          );
         } else {
           // Update sync metadata
           const syncMetadata = {
@@ -491,26 +534,39 @@ return;
           await fs.mkdir(path.join(storePath, 'metadata'), { recursive: true });
           await fs.writeFile(
             path.join(storePath, 'metadata', 'sync.json'),
-            JSON.stringify(syncMetadata, null, 2),
+            JSON.stringify(syncMetadata, null, 2)
           );
 
-          console.log(chalk.green(`  [OK] ${storeName}: Synced ${fileCount} files`));
+          console.log(
+            chalk.green(`  [OK] ${storeName}: Synced ${fileCount} files`)
+          );
         }
         synced++;
       } catch (error) {
-        console.log(chalk.red(`  [ERROR] ${storeName}: ${error instanceof Error ? error.message : String(error)}`));
+        console.log(
+          chalk.red(
+            `  [ERROR] ${storeName}: ${error instanceof Error ? error.message : String(error)}`
+          )
+        );
       }
     }
 
     console.log(chalk.gray('\n' + '-'.repeat(50)));
-    console.log(chalk.green(`Sync complete: ${synced} synced, ${skipped} skipped\n`));
+    console.log(
+      chalk.green(`Sync complete: ${synced} synced, ${skipped} skipped\n`)
+    );
   } catch (error) {
     spinner.fail('Failed to sync stores');
-    console.error(chalk.red(error instanceof Error ? error.message : String(error)));
+    console.error(
+      chalk.red(error instanceof Error ? error.message : String(error))
+    );
   }
 }
 
-async function pruneStores(options: { store?: string; dryRun?: boolean }): Promise<void> {
+async function pruneStores(options: {
+  store?: string;
+  dryRun?: boolean;
+}): Promise<void> {
   const spinner = ora('Pruning RAG stores...').start();
 
   try {
@@ -532,13 +588,17 @@ async function pruneStores(options: { store?: string; dryRun?: boolean }): Promi
 
       if (existsSync(embeddingsDir)) {
         try {
-          const embeddingFiles = readdirSync(embeddingsDir).filter(f => f.endsWith('.json'));
+          const embeddingFiles = readdirSync(embeddingsDir).filter(f =>
+            f.endsWith('.json')
+          );
 
           for (const file of embeddingFiles) {
             const embeddingPath = path.join(embeddingsDir, file);
 
             try {
-              const embeddingData = JSON.parse(await fs.readFile(embeddingPath, 'utf-8'));
+              const embeddingData = JSON.parse(
+                await fs.readFile(embeddingPath, 'utf-8')
+              );
               const originalPath = embeddingData.originalPath;
 
               if (originalPath && !existsSync(originalPath)) {
@@ -569,7 +629,7 @@ async function pruneStores(options: { store?: string; dryRun?: boolean }): Promi
         await fs.mkdir(path.join(storePath, 'metadata'), { recursive: true });
         await fs.writeFile(
           path.join(storePath, 'metadata', 'prune.json'),
-          JSON.stringify(pruneMetadata, null, 2),
+          JSON.stringify(pruneMetadata, null, 2)
         );
       }
 
@@ -578,14 +638,21 @@ async function pruneStores(options: { store?: string; dryRun?: boolean }): Promi
     }
 
     console.log(chalk.gray('\n' + '-'.repeat(50)));
-    console.log(chalk.green(`Prune complete: ${totalPruned} entries removed\n`));
+    console.log(
+      chalk.green(`Prune complete: ${totalPruned} entries removed\n`)
+    );
   } catch (error) {
     spinner.fail('Failed to prune stores');
-    console.error(chalk.red(error instanceof Error ? error.message : String(error)));
+    console.error(
+      chalk.red(error instanceof Error ? error.message : String(error))
+    );
   }
 }
 
-async function reindexStores(options: { store?: string; force?: boolean }): Promise<void> {
+async function reindexStores(options: {
+  store?: string;
+  force?: boolean;
+}): Promise<void> {
   const spinner = ora('Re-indexing RAG stores...').start();
 
   try {
@@ -609,7 +676,9 @@ async function reindexStores(options: { store?: string; force?: boolean }): Prom
       let embeddingCount = 0;
       if (existsSync(embeddingsDir)) {
         try {
-          const files = readdirSync(embeddingsDir).filter(f => f.endsWith('.json'));
+          const files = readdirSync(embeddingsDir).filter(f =>
+            f.endsWith('.json')
+          );
           embeddingCount = files.length;
         } catch {
           // Ignore permission errors
@@ -640,21 +709,28 @@ async function reindexStores(options: { store?: string; force?: boolean }): Prom
       await fs.mkdir(path.join(storePath, 'metadata'), { recursive: true });
       await fs.writeFile(
         path.join(storePath, 'metadata', 'index.json'),
-        JSON.stringify(indexMetadata, null, 2),
+        JSON.stringify(indexMetadata, null, 2)
       );
 
-      console.log(chalk.green(`  [OK] ${storeName}: Indexed ${embeddingCount} entries`));
+      console.log(
+        chalk.green(`  [OK] ${storeName}: Indexed ${embeddingCount} entries`)
+      );
     }
 
     console.log(chalk.gray('\n' + '-'.repeat(50)));
     console.log(chalk.green('Re-index complete\n'));
   } catch (error) {
     spinner.fail('Failed to reindex stores');
-    console.error(chalk.red(error instanceof Error ? error.message : String(error)));
+    console.error(
+      chalk.red(error instanceof Error ? error.message : String(error))
+    );
   }
 }
 
-async function createStore(name: string, options: { source?: string; global?: boolean }): Promise<void> {
+async function createStore(
+  name: string,
+  options: { source?: string; global?: boolean }
+): Promise<void> {
   const spinner = ora(`Creating RAG store: ${name}...`).start();
 
   try {
@@ -706,7 +782,7 @@ async function createStore(name: string, options: { source?: string; global?: bo
 
     await fs.writeFile(
       path.join(baseDir, 'metadata', 'source.json'),
-      JSON.stringify(sourceMetadata, null, 2),
+      JSON.stringify(sourceMetadata, null, 2)
     );
 
     spinner.succeed(`RAG store created: ${name}`);
@@ -715,13 +791,17 @@ async function createStore(name: string, options: { source?: string; global?: bo
     console.log(chalk.green('\nRun "wundr rag sync" to sync the store.\n'));
   } catch (error) {
     spinner.fail('Failed to create store');
-    console.error(chalk.red(error instanceof Error ? error.message : String(error)));
+    console.error(
+      chalk.red(error instanceof Error ? error.message : String(error))
+    );
   }
 }
 
 async function runSetup(options: { skipApiKey?: boolean }): Promise<void> {
   console.log(chalk.cyan('\nRAG Infrastructure Setup\n'));
-  console.log(chalk.gray('This will set up the RAG infrastructure for Wundr.\n'));
+  console.log(
+    chalk.gray('This will set up the RAG infrastructure for Wundr.\n')
+  );
 
   const spinner = ora('Checking prerequisites...').start();
 
@@ -751,14 +831,22 @@ async function runSetup(options: { skipApiKey?: boolean }): Promise<void> {
     if (!options.skipApiKey) {
       spinner.stop();
       console.log(chalk.yellow('\nGEMINI_API_KEY Configuration'));
-      console.log(chalk.gray('Get your API key from: https://makersuite.google.com/app/apikey\n'));
+      console.log(
+        chalk.gray(
+          'Get your API key from: https://makersuite.google.com/app/apikey\n'
+        )
+      );
 
       if (process.env.GEMINI_API_KEY) {
-        console.log(chalk.green('GEMINI_API_KEY is already set in environment.\n'));
+        console.log(
+          chalk.green('GEMINI_API_KEY is already set in environment.\n')
+        );
       } else {
         console.log(chalk.yellow('GEMINI_API_KEY is not set.'));
         console.log(chalk.gray('Add the following to your shell profile:'));
-        console.log(chalk.white('  export GEMINI_API_KEY="your-api-key-here"\n'));
+        console.log(
+          chalk.white('  export GEMINI_API_KEY="your-api-key-here"\n')
+        );
       }
     }
 
@@ -766,7 +854,9 @@ async function runSetup(options: { skipApiKey?: boolean }): Promise<void> {
     spinner.start('Creating RAG store directories...');
 
     await fs.mkdir(RAG_GLOBAL_DIR, { recursive: true });
-    await fs.mkdir(path.join(RAG_GLOBAL_DIR, 'embeddings'), { recursive: true });
+    await fs.mkdir(path.join(RAG_GLOBAL_DIR, 'embeddings'), {
+      recursive: true,
+    });
     await fs.mkdir(path.join(RAG_GLOBAL_DIR, 'indexes'), { recursive: true });
     await fs.mkdir(path.join(RAG_GLOBAL_DIR, 'metadata'), { recursive: true });
     await fs.mkdir(path.join(RAG_GLOBAL_DIR, 'cache'), { recursive: true });
@@ -788,16 +878,24 @@ async function runSetup(options: { skipApiKey?: boolean }): Promise<void> {
     console.log(chalk.green('\nRAG infrastructure setup complete!\n'));
     console.log(chalk.gray('Next steps:'));
     console.log(chalk.white('  1. Set GEMINI_API_KEY if not already set'));
-    console.log(chalk.white('  2. Run "wundr rag create <name>" to create a store'));
+    console.log(
+      chalk.white('  2. Run "wundr rag create <name>" to create a store')
+    );
     console.log(chalk.white('  3. Run "wundr rag sync" to sync your stores'));
     console.log(chalk.white('  4. Run "wundr rag status" to check status\n'));
   } catch (error) {
     spinner.fail('Setup failed');
-    console.error(chalk.red(error instanceof Error ? error.message : String(error)));
+    console.error(
+      chalk.red(error instanceof Error ? error.message : String(error))
+    );
   }
 }
 
-async function manageConfig(options: { get?: string; set?: string; reset?: boolean }): Promise<void> {
+async function manageConfig(options: {
+  get?: string;
+  set?: string;
+  reset?: boolean;
+}): Promise<void> {
   try {
     if (options.reset) {
       const config = getDefaultConfig();
@@ -828,7 +926,10 @@ async function manageConfig(options: { get?: string; set?: string; reset?: boole
       }
 
       const keys = keyPath.split('.');
-      let obj: Record<string, unknown> = config as unknown as Record<string, unknown>;
+      let obj: Record<string, unknown> = config as unknown as Record<
+        string,
+        unknown
+      >;
       for (let i = 0; i < keys.length - 1; i++) {
         const key = keys[i];
         if (key) {
@@ -858,14 +959,18 @@ async function manageConfig(options: { get?: string; set?: string; reset?: boole
     console.log(chalk.cyan('\nRAG Configuration\n'));
     console.log(JSON.stringify(config, null, 2));
   } catch (error) {
-    console.error(chalk.red(error instanceof Error ? error.message : String(error)));
+    console.error(
+      chalk.red(error instanceof Error ? error.message : String(error))
+    );
   }
 }
 
-async function deleteStore(name: string, options: { force?: boolean }): Promise<void> {
-  const storePath = name === 'global'
-    ? RAG_GLOBAL_DIR
-    : path.join(RAG_PROJECT_DIR, name);
+async function deleteStore(
+  name: string,
+  options: { force?: boolean }
+): Promise<void> {
+  const storePath =
+    name === 'global' ? RAG_GLOBAL_DIR : path.join(RAG_PROJECT_DIR, name);
 
   if (!existsSync(storePath)) {
     console.log(chalk.red(`Store not found: ${name}`));
@@ -895,7 +1000,9 @@ async function deleteStore(name: string, options: { force?: boolean }): Promise<
     spinner.succeed(`Store deleted: ${name}`);
   } catch (error) {
     spinner.fail('Failed to delete store');
-    console.error(chalk.red(error instanceof Error ? error.message : String(error)));
+    console.error(
+      chalk.red(error instanceof Error ? error.message : String(error))
+    );
   }
 }