fix: restore watcher lifecycle for routing hints

Author: Helweg

src/index.ts

@@ -25,6 +25,14 @@ import {
 import { loadCommandsFromDirectory } from "./commands/loader.js";
 import { RoutingHintController } from "./routing-hints.js";
 import { hasProjectMarker } from "./utils/files.js";
+import type { CombinedWatcher } from "./watcher/index.js";
+
+let activeWatcher: CombinedWatcher | null = null;
+
+function replaceActiveWatcher(nextWatcher: CombinedWatcher | null): void {
+  activeWatcher?.stop();
+  activeWatcher = nextWatcher;
+}
 
 function getCommandsDir(): string {
   let currentDir = process.cwd();
@@ -65,7 +73,9 @@ const plugin: Plugin = async ({ directory }) => {
     }
 
     if (config.indexing.watchFiles && isValidProject) {
-      createWatcherWithIndexer(indexer, projectRoot, config);
+      replaceActiveWatcher(createWatcherWithIndexer(getSharedIndexer, projectRoot, config));
+    } else {
+      replaceActiveWatcher(null);
     }
 
     return {

src/watcher/index.ts

@@ -1,9 +1,9 @@
 import chokidar, { FSWatcher } from "chokidar";
 import * as path from "path";
 
-import { CodebaseIndexConfig } from "../config/schema.js";
+import type { CodebaseIndexConfig } from "../config/schema.js";
 import { createIgnoreFilter, shouldIncludeFile } from "../utils/files.js";
-import { Indexer } from "../indexer/index.js";
+import type { Indexer } from "../indexer/index.js";
 import { isGitRepo, getHeadPath, getCurrentBranch } from "../git/index.js";
 
 export type FileChangeType = "add" | "change" | "unlink";
@@ -243,7 +243,7 @@ export interface CombinedWatcher {
 }
 
 export function createWatcherWithIndexer(
-  indexer: Indexer,
+  getIndexer: () => Indexer,
   projectRoot: string,
   config: CodebaseIndexConfig
 ): CombinedWatcher {
@@ -256,7 +256,7 @@ export function createWatcherWithIndexer(
     const hasDelete = changes.some((c) => c.type === "unlink");
 
     if (hasAddOrChange || hasDelete) {
-      await indexer.index();
+      await getIndexer().index();
     }
   });
 
@@ -266,7 +266,7 @@ export function createWatcherWithIndexer(
     gitWatcher = new GitHeadWatcher(projectRoot);
     gitWatcher.start(async (oldBranch, newBranch) => {
       console.log(`Branch changed: ${oldBranch ?? "(none)"} -> ${newBranch}`);
-      await indexer.index();
+      await getIndexer().index();
     });
   }
 

tests/watcher.test.ts

@@ -2,7 +2,12 @@ import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
 import * as fs from "fs";
 import * as path from "path";
 import * as os from "os";
-import { FileWatcher, GitHeadWatcher, FileChange } from "../src/watcher/index.js";
+import {
+  FileWatcher,
+  GitHeadWatcher,
+  FileChange,
+  createWatcherWithIndexer,
+} from "../src/watcher/index.js";
 import { ParsedCodebaseIndexConfig } from "../src/config/schema.js";
 
 const createTestConfig = (overrides: Partial<ParsedCodebaseIndexConfig> = {}): ParsedCodebaseIndexConfig => ({
@@ -119,6 +124,73 @@ describe("FileWatcher", () => {
       expect(tsChanges.length).toBeGreaterThanOrEqual(0);
       expect(mdChanges.length).toBe(0);
     });
+
+    it("should include matching root-level files", async () => {
+      const changes: FileChange[] = [];
+      watcher = new FileWatcher(tempDir, createTestConfig({ include: ["**/*.ts"] }));
+
+      watcher.start(async (c) => {
+        changes.push(...c);
+      });
+
+      await new Promise((r) => setTimeout(r, 100));
+
+      fs.writeFileSync(path.join(tempDir, "root.ts"), "export const root = 1;");
+
+      await new Promise((r) => setTimeout(r, 1500));
+
+      expect(changes.some((c) => c.path.endsWith("root.ts"))).toBe(true);
+    });
+  });
+
+  describe("createWatcherWithIndexer", () => {
+    it("uses the latest indexer instance for file-triggered reindexing", async () => {
+      const staleIndexer = {
+        index: vi.fn().mockResolvedValue(undefined),
+      };
+      const refreshedIndexer = {
+        index: vi.fn().mockResolvedValue(undefined),
+      };
+
+      let currentIndexer = staleIndexer;
+      const combinedWatcher = createWatcherWithIndexer(
+        () => currentIndexer,
+        tempDir,
+        createTestConfig()
+      );
+
+      await new Promise((r) => setTimeout(r, 100));
+      currentIndexer = refreshedIndexer;
+
+      fs.writeFileSync(path.join(tempDir, "src", "reindex-me.ts"), "export const value = 1;");
+
+      await new Promise((r) => setTimeout(r, 1500));
+
+      expect(refreshedIndexer.index).toHaveBeenCalledTimes(1);
+      expect(staleIndexer.index).not.toHaveBeenCalled();
+
+      combinedWatcher.stop();
+    });
+
+    it("stops the watcher cleanly after start", () => {
+      const indexer = {
+        index: vi.fn().mockResolvedValue(undefined),
+      };
+
+      const combinedWatcher = createWatcherWithIndexer(
+        () => indexer,
+        tempDir,
+        createTestConfig()
+      );
+
+      expect(combinedWatcher.fileWatcher.isRunning()).toBe(true);
+      expect(combinedWatcher.gitWatcher?.isRunning() ?? false).toBe(false);
+
+      combinedWatcher.stop();
+
+      expect(combinedWatcher.fileWatcher.isRunning()).toBe(false);
+      expect(combinedWatcher.gitWatcher?.isRunning() ?? false).toBe(false);
+    });
   });
 });