@wrongstack/tools 0.141.0 → 0.155.0

package/dist/codebase-index/index.d.ts

@@ -1,5 +1,5 @@
-import { I as IndexResult, S as Symbol, F as FileMeta, a as SymbolKind, b as SymbolLang, c as SearchResult, d as IndexStats, R as Ref } from '../background-indexer-C2014mH0.js';
-export { e as FileSymbols, f as SCHEMA_VERSION, g as cancelPendingReindexes, h as codebaseIndexTool, i as codebaseSearchTool, j as codebaseStatsTool, k as enqueueReindex, l as getIndexState, m as isIndexReady, n as isIndexableFile, o as isIndexing, p as onIndexStateChange, r as runStartupIndex } from '../background-indexer-C2014mH0.js';
+import { I as IndexResult, S as Symbol, F as FileMeta, a as SymbolKind, b as SymbolLang, c as SearchResult, d as IndexStats, R as Ref } from '../background-indexer-CtbgPExj.js';
+export { e as FileSymbols, f as SCHEMA_VERSION, g as cancelPendingReindexes, h as codebaseIndexTool, i as codebaseSearchTool, j as codebaseStatsTool, k as enqueueReindex, l as getIndexState, m as isIndexReady, n as isIndexableFile, o as isIndexing, p as onIndexStateChange, r as runStartupIndex } from '../background-indexer-CtbgPExj.js';
 import { Context } from '@wrongstack/core';
 
 interface IndexerOptions {
@@ -10,6 +10,13 @@ interface IndexerOptions {
     ignore?: string[] | undefined;
     /** Override the index directory (default: the global per-project dir). */
     indexDir?: string | undefined;
+    /**
+     * Signal that cancels indexing cooperatively. Polled at yield points
+     * (file walk, per-file loop) so a hung filesystem won't lock up the
+     * process. When the tool executor's timeout fires, this signal aborts
+     * and `runIndexer` throws, releasing the mutex and resetting flags.
+     */
+    signal?: AbortSignal | undefined;
 }
 /** Run a full or incremental index and return statistics. */
 declare function runIndexer(_ctx: Context, opts: IndexerOptions): Promise<IndexResult>;

package/dist/codebase-index/index.js

@@ -1742,7 +1742,8 @@ async function runStartupIndex(opts) {
       () => runIndexer(stubCtx(opts.projectRoot), {
         projectRoot: opts.projectRoot,
         indexDir: opts.indexDir,
-        force: opts.force
+        force: opts.force,
+        signal: opts.signal
       })
     );
     _ready = true;
@@ -1788,6 +1789,16 @@ var YIELD_EVERY_N = 50;
 function yieldEventLoop() {
   return new Promise((resolve2) => setImmediate(resolve2));
 }
+function throwIfAborted(signal) {
+  if (!signal?.aborted) return;
+  if (signal.reason instanceof Error) throw signal.reason;
+  throw new Error(
+    typeof signal.reason === "string" ? signal.reason : "Indexing cancelled"
+  );
+}
+function isAbortError(err) {
+  return err instanceof DOMException && err.name === "AbortError";
+}
 var DEFAULT_IGNORE = [
   "node_modules",
   ".git",
@@ -1799,7 +1810,7 @@ var DEFAULT_IGNORE = [
   "__snapshots__",
   ".nyc_output"
 ];
-async function findSourceFiles(projectRoot, ignore, isGitIgnored) {
+async function findSourceFiles(projectRoot, ignore, isGitIgnored, signal) {
   const results = [];
   const ignoreSet = /* @__PURE__ */ new Set([...DEFAULT_IGNORE, ...ignore]);
   const globs = [
@@ -1814,13 +1825,20 @@ async function findSourceFiles(projectRoot, ignore, isGitIgnored) {
     { ext: ".yaml", pat: compileGlob("**/*.yaml") },
     { ext: ".yml", pat: compileGlob("**/*.yml") }
   ];
+  let dirCount = 0;
   const walk = async (dir) => {
+    throwIfAborted(signal);
+    if (dirCount > 0 && dirCount % YIELD_EVERY_N === 0) {
+      await yieldEventLoop();
+      throwIfAborted(signal);
+    }
     let entries;
     try {
       entries = await fs3.readdir(dir, { withFileTypes: true });
     } catch {
       return;
     }
+    dirCount++;
     for (const e of entries) {
       if (ignoreSet.has(e.name)) continue;
       const full = path4.join(dir, e.name);
@@ -1865,7 +1883,7 @@ async function parseFile(file, content, lang) {
   }
 }
 async function runIndexer(_ctx, opts) {
-  const { projectRoot, force = false, langs, ignore = [], indexDir } = opts;
+  const { projectRoot, force = false, langs, ignore = [], indexDir, signal } = opts;
   const store = new IndexStore(projectRoot, { indexDir });
   const startMs = Date.now();
   const errors = [];
@@ -1877,7 +1895,7 @@ async function runIndexer(_ctx, opts) {
   if (opts.files && opts.files.length > 0) {
     files = opts.files.map((f) => path4.resolve(projectRoot, f)).filter((f) => !isGitIgnored(path4.relative(projectRoot, f).replace(/\\/g, "/"), false));
   } else {
-    files = await findSourceFiles(projectRoot, ignore, isGitIgnored);
+    files = await findSourceFiles(projectRoot, ignore, isGitIgnored, signal);
   }
   if (langs && langs.length > 0) {
     const langSet = new Set(langs);
@@ -1896,11 +1914,14 @@ async function runIndexer(_ctx, opts) {
     _setIndexProgress(fi + 1, files.length);
     if (fi > 0 && fi % YIELD_EVERY_N === 0) {
       await yieldEventLoop();
+      throwIfAborted(signal);
     }
     let stat2;
     try {
-      stat2 = await fs3.stat(file);
-    } catch {
+      const statOpts = signal ? { signal } : {};
+      stat2 = await fs3.stat(file, statOpts);
+    } catch (e) {
+      if (isAbortError(e)) throw e;
       store.deleteFile(file);
       continue;
     }
@@ -1918,8 +1939,9 @@ async function runIndexer(_ctx, opts) {
     store.deleteSymbolsForFile(file);
     let content;
     try {
-      content = await fs3.readFile(file, "utf8");
+      content = await fs3.readFile(file, { encoding: "utf8", signal });
     } catch (e) {
+      if (isAbortError(e)) throw e;
       errors.push(`read error: ${file}: ${e instanceof Error ? e.message : String(e)}`);
       continue;
     }
@@ -2009,7 +2031,7 @@ var codebaseIndexTool = {
       }
     }
   },
-  async execute(input, ctx) {
+  async execute(input, ctx, execOpts) {
     if (isIndexing()) {
       return {
         filesIndexed: 0,
@@ -2024,7 +2046,8 @@ var codebaseIndexTool = {
       projectRoot: ctx.projectRoot,
       force: input.force ?? false,
       langs: input.langs,
-      indexDir: codebaseIndexDirOverride(ctx)
+      indexDir: codebaseIndexDirOverride(ctx),
+      signal: execOpts?.signal
     });
     setIndexReady();
     return result;
@@ -2039,7 +2062,7 @@ function tokenise(text) {
   return sanitised.toLowerCase().split(" ").filter(Boolean);
 }
 function splitName(name) {
-  return name.replace(/([a-z])([A-Z])/g, "$1 $2").replace(/[_\-]+/g, " ").trim();
+  return name.replace(/([a-z])([A-Z])/g, "$1 $2").replace(/[_-]+/g, " ").trim();
 }
 function buildIndexableText(name, signature, docComment) {
   return [splitName(name), name, signature, docComment].filter(Boolean).join(" ");