fixed search

Author: Omri Hopson

dashboard/src/pages/Repositories/Repositories.tsx

@@ -24,6 +24,9 @@ export default function Repositories() {
   const [addMode, setAddMode] = useState<'browse' | 'manual'>('browse');
   const [busyAction, setBusyAction] = useState<string | null>(null);
   const [workspaceRoot, setWorkspaceRoot] = useState('');
+  const [autoIndex, setAutoIndex] = useState(true);
+  const [autoIndexLocked, setAutoIndexLocked] = useState(false);
+  const [togglingAutoIndex, setTogglingAutoIndex] = useState(false);
   const { toasts, push: toast, dismiss: dismissToast } = useToast();
 
   useEffect(() => {
@@ -32,8 +35,20 @@ export default function Repositories() {
       .then((r) => r.json())
       .then((d) => setWorkspaceRoot(d.workspaceRoot || ''))
       .catch((err) => console.warn('Failed to load workspace info:', err));
+    loadConfig();
   }, []);
 
+  const loadConfig = async () => {
+    try {
+      const response = await fetch('/api/config');
+      const data = await response.json();
+      setAutoIndex(data.config.autoIndex || false);
+      setAutoIndexLocked(data.envLocked.autoIndex || false);
+    } catch (error) {
+      console.error('Failed to load config:', error);
+    }
+  };
+
   const loadRepositories = async () => {
     try {
       const response = await fetch('/api/repositories');
@@ -108,6 +123,63 @@ export default function Repositories() {
     }
   };
 
+  const handleToggleAutoIndex = async () => {
+    if (autoIndexLocked) {
+      toast('info', 'Auto-index is locked by environment variable. Cannot change via UI.');
+      return;
+    }
+
+    setTogglingAutoIndex(true);
+    try {
+      const response = await fetch('/api/config', {
+        method: 'PUT',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ autoIndex: !autoIndex }),
+      });
+
+      if (response.ok) {
+        const data = await response.json();
+        if (data.success) {
+          setAutoIndex(!autoIndex);
+          toast('success', !autoIndex 
+            ? 'Auto-indexing enabled. New repositories will index automatically.' 
+            : 'Auto-indexing disabled. Use manual indexing for new repositories.');
+        } else {
+          toast('error', 'Failed to update config.');
+        }
+      } else {
+        toast('error', 'Failed to update auto-index setting.');
+      }
+    } catch (error) {
+      toast('error', 'Network error — could not reach server.');
+    } finally {
+      setTogglingAutoIndex(false);
+    }
+  };
+
+  const handleQuickIndexWorkspace = async () => {
+    setBusyAction('quick-index');
+    try {
+      const response = await fetch('/api/repositories', {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ path: '/workspace', incremental: false }),
+      });
+
+      if (response.ok) {
+        toast('success', 'Workspace indexing started. This may take a few minutes for large projects.');
+        setTimeout(() => loadRepositories(), 2000);
+      } else {
+        const error = await response.text();
+        toast('error', `Failed to index workspace: ${error}`);
+      }
+    } catch (error) {
+      toast('error', 'Network error — could not reach server.');
+    } finally {
+      setBusyAction(null);
+    }
+  };
+
   const handleAddRepository = async () => {
     if (!newRepoPath.trim()) return;
 
@@ -181,6 +253,22 @@ export default function Repositories() {
           </p>
         </div>
         <div className="flex gap-3">
+          <button
+            onClick={handleToggleAutoIndex}
+            disabled={togglingAutoIndex || autoIndexLocked}
+            className={`px-6 py-3 font-semibold text-[0.875rem] rounded-xl transition-all flex items-center gap-2 ${
+              autoIndex 
+                ? 'bg-green-500/10 text-green-600 hover:bg-green-500/20 border border-green-500/30'
+                : 'bg-surface-container-highest text-on-surface hover:bg-surface-dim'
+            } disabled:opacity-50 disabled:cursor-not-allowed`}
+            title={autoIndexLocked ? 'Locked by environment variable' : 'Toggle auto-indexing'}
+          >
+            <span className={`material-symbols-outlined text-[18px] ${togglingAutoIndex ? 'animate-spin' : ''}`}>
+              {autoIndex ? 'toggle_on' : 'toggle_off'}
+            </span>
+            Auto-Index: {autoIndex ? 'ON' : 'OFF'}
+            {autoIndexLocked && <span className="material-symbols-outlined text-[14px]">lock</span>}
+          </button>
           {repos.length > 0 && (
             <button
               onClick={() => {
@@ -222,15 +310,27 @@ export default function Repositories() {
           </span>
           <h2 className="text-2xl font-bold mb-2">No Repositories Indexed</h2>
           <p className="text-on-surface-variant mb-6">
-            Get started by adding a repository to index for code intelligence.
+            Get started by indexing your workspace or adding a specific repository.
           </p>
-          <button
-            onClick={() => setShowAddDialog(true)}
-            className="px-6 py-3 primary-gradient text-white font-semibold text-[0.875rem] rounded-xl shadow-lg shadow-tertiary/10 active:scale-95 transition-all inline-flex items-center gap-2"
-          >
-            <span className="material-symbols-outlined text-[18px]">add</span>
-            Add Repository
-          </button>
+          <div className="flex gap-3 justify-center">
+            <button
+              onClick={handleQuickIndexWorkspace}
+              disabled={busyAction === 'quick-index'}
+              className="px-6 py-3 bg-tertiary/10 text-tertiary font-semibold text-[0.875rem] rounded-xl hover:bg-tertiary/20 active:scale-95 transition-all inline-flex items-center gap-2 disabled:opacity-50"
+            >
+              <span className={`material-symbols-outlined text-[18px] ${busyAction === 'quick-index' ? 'animate-spin' : ''}`}>
+                {busyAction === 'quick-index' ? 'refresh' : 'bolt'}
+              </span>
+              {busyAction === 'quick-index' ? 'Indexing...' : 'Quick Index Workspace'}
+            </button>
+            <button
+              onClick={() => setShowAddDialog(true)}
+              className="px-6 py-3 primary-gradient text-white font-semibold text-[0.875rem] rounded-xl shadow-lg shadow-tertiary/10 active:scale-95 transition-all inline-flex items-center gap-2"
+            >
+              <span className="material-symbols-outlined text-[18px]">add</span>
+              Add Specific Path
+            </button>
+          </div>
         </div>
       )}
 

src/api/routes/search.ts

@@ -68,6 +68,7 @@ export interface SearchRouteOptions {
   vectorSearch?: any;
   hybridSearch?: any;
   configManager?: any;
+  workspaceRoot?: string;
 }
 
 export async function registerSearchRoutes(
@@ -106,7 +107,7 @@ export async function registerSearchRoutes(
       let semanticTime = 0;
       let fusionTime = 0;
 
-      let results: unknown[] = [];
+      let results: any[] = [];
       let total = 0;
       let hasMore = false;
 
@@ -130,10 +131,23 @@ export async function registerSearchRoutes(
           });
         }
 
+        // Get repository ID - use provided one or first available
+        let repoId = input.repositoryId;
+        if (!repoId) {
+          const repos = options.storage.listRepositories();
+          if (repos.length === 0) {
+            return reply.status(400).send({
+              error: 'No repositories indexed',
+              message: 'Please index a repository first',
+            });
+          }
+          repoId = repos[0]!.id;
+        }
+
         const semanticStart = Date.now();
         const response = await vectorSearch.search(
           input.query,
-          input.repositoryId || 'default-repo',
+          repoId,
           input.limit,
           input.offset
         );
@@ -151,10 +165,23 @@ export async function registerSearchRoutes(
           });
         }
 
+        // Get repository ID - use provided one or first available
+        let repoId = input.repositoryId;
+        if (!repoId) {
+          const repos = options.storage.listRepositories();
+          if (repos.length === 0) {
+            return reply.status(400).send({
+              error: 'No repositories indexed',
+              message: 'Please index a repository first',
+            });
+          }
+          repoId = repos[0]!.id;
+        }
+
         const hybridStart = Date.now();
         const response = await hybridSearch.search(
           input.query,
-          input.repositoryId || 'default-repo',
+          repoId,
           input.limit,
           input.offset
         );
@@ -165,6 +192,32 @@ export async function registerSearchRoutes(
         hasMore = response.hasMore;
       }
 
+      // Add code snippets if workspace root is available
+      if (options.workspaceRoot && results.length > 0) {
+        try {
+          const { extractSnippetsBatch } = await import('../../search/snippet.js');
+          const snippets = await extractSnippetsBatch(
+            options.workspaceRoot,
+            results.map(r => ({
+              filePath: r.filePath,
+              lineStart: r.lineStart,
+              lineEnd: r.lineEnd,
+            })),
+            { maxLines: 10, maxChars: 500 }
+          );
+          
+          // Attach snippets to results
+          results = results.map(r => {
+            const key = `${r.filePath}:${r.lineStart}:${r.lineEnd}`;
+            const snippet = snippets.get(key);
+            return snippet ? { ...r, snippet } : r;
+          });
+        } catch (error) {
+          // Snippet extraction failed, continue without snippets
+          console.warn('Failed to extract snippets:', error);
+        }
+      }
+
       const totalTime = Date.now() - startTime;
 
       return {

src/api/server.ts

@@ -193,6 +193,7 @@ export async function createAPIServer(
     vectorSearch: options.vectorSearch,
     hybridSearch: options.hybridSearch,
     configManager: options.configManager,
+    workspaceRoot: options.workspaceRoot,
   });
 
   await registerGraphRoutes(fastify, {

src/core/config-manager.ts

@@ -36,6 +36,8 @@ import type { HybridSearch } from '../search/hybrid.js';
 export interface ConfigManagerOptions {
   storage: StorageProvider;
   vectorStore?: LanceDBVectorStore;
+  indexer?: any;
+  workspaceRoot?: string;
   onEmbeddingProviderChange?: (provider: EmbeddingProvider | undefined) => Promise<void>;
   onEmbeddingQueueChange?: (queue: EmbeddingQueue | undefined) => Promise<void>;
   onVectorSearchChange?: (vectorSearch: VectorSearch | undefined, hybridSearch: HybridSearch | undefined) => Promise<void>;
@@ -119,13 +121,19 @@ export class ConfigManager extends EventEmitter {
 
       const needsProviderReload = this.needsProviderReload(updates);
       const needsQueueUpdate = this.needsQueueUpdate(updates);
+      const needsAutoIndex = this.needsAutoIndexTrigger(updates);
 
       if (needsProviderReload) {
         await this.reloadEmbeddingProvider(newConfig, changes, warnings);
       } else if (needsQueueUpdate) {
         await this.updateQueueSettings(newConfig, changes);
       }
 
+      if (needsAutoIndex) {
+        changes.push('Auto-indexing enabled - will trigger on next config reload or manually via API');
+        warnings.push('Auto-indexing is now enabled. Use POST /api/repositories to trigger indexing.');
+      }
+
       this.emit('reloaded', { changes, warnings });
       return { success: true, changes, warnings };
     } catch (error) {
@@ -156,6 +164,10 @@ export class ConfigManager extends EventEmitter {
     return !!(updates.embeddingConcurrency || updates.embeddingBatchSize);
   }
 
+  private needsAutoIndexTrigger(updates: Record<string, unknown>): boolean {
+    return updates.autoIndex === true;
+  }
+
   private async reloadEmbeddingProvider(
     newConfig: AppConfig,
     changes: string[],

src/core/types.ts

@@ -145,6 +145,12 @@ export const SearchResultSchema = z.object({
   snippet: z.string().optional(),
   language: z.string(),
   repositoryId: z.string(),
+  // Enhanced metadata
+  docstring: z.string().optional(),
+  complexity: z.number().int().nonnegative().optional(),
+  visibility: VisibilitySchema.optional(),
+  isExported: z.boolean().optional(),
+  parentSymbol: z.string().optional(),
 });
 
 export type SearchResult = z.infer<typeof SearchResultSchema>;