@wrongstack/tools 0.148.0 → 0.155.0

package/dist/index.d.ts

@@ -33,7 +33,7 @@ export { forgetTool, relatedMemoryTool, rememberTool, searchMemoryTool } from '.
 export { createModeTool } from './mode.js';
 export { KillOpts, RegistryStats, TrackedProcess, _resetProcessRegistry, getProcessRegistry } from './process-registry.js';
 export { CircuitBreaker, CircuitBreakerConfig, CircuitBreakerSnapshot } from './circuit-breaker.js';
-export { 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';
+export { 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';
 export { builtinTools } from './builtin.js';
 export { builtinToolsPack } from './pack.js';
 import 'node:child_process';

package/dist/index.js

@@ -7202,7 +7202,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;
@@ -7248,6 +7249,16 @@ var YIELD_EVERY_N = 50;
 function yieldEventLoop() {
   return new Promise((resolve7) => setImmediate(resolve7));
 }
+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_IGNORE5 = [
   "node_modules",
   ".git",
@@ -7259,7 +7270,7 @@ var DEFAULT_IGNORE5 = [
   "__snapshots__",
   ".nyc_output"
 ];
-async function findSourceFiles(projectRoot, ignore, isGitIgnored) {
+async function findSourceFiles(projectRoot, ignore, isGitIgnored, signal) {
   const results = [];
   const ignoreSet = /* @__PURE__ */ new Set([...DEFAULT_IGNORE5, ...ignore]);
   const globs = [
@@ -7274,13 +7285,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 fs4.readdir(dir, { withFileTypes: true });
     } catch {
       return;
     }
+    dirCount++;
     for (const e of entries) {
       if (ignoreSet.has(e.name)) continue;
       const full = path.join(dir, e.name);
@@ -7325,7 +7343,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 = [];
@@ -7337,7 +7355,7 @@ async function runIndexer(_ctx, opts) {
   if (opts.files && opts.files.length > 0) {
     files = opts.files.map((f) => path.resolve(projectRoot, f)).filter((f) => !isGitIgnored(path.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);
@@ -7356,11 +7374,14 @@ async function runIndexer(_ctx, opts) {
     _setIndexProgress(fi + 1, files.length);
     if (fi > 0 && fi % YIELD_EVERY_N === 0) {
       await yieldEventLoop();
+      throwIfAborted(signal);
     }
     let stat10;
     try {
-      stat10 = await fs4.stat(file);
-    } catch {
+      const statOpts = signal ? { signal } : {};
+      stat10 = await fs4.stat(file, statOpts);
+    } catch (e) {
+      if (isAbortError(e)) throw e;
       store.deleteFile(file);
       continue;
     }
@@ -7378,8 +7399,9 @@ async function runIndexer(_ctx, opts) {
     store.deleteSymbolsForFile(file);
     let content;
     try {
-      content = await fs4.readFile(file, "utf8");
+      content = await fs4.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;
     }
@@ -7469,7 +7491,7 @@ var codebaseIndexTool = {
       }
     }
   },
-  async execute(input, ctx) {
+  async execute(input, ctx, execOpts) {
     if (isIndexing()) {
       return {
         filesIndexed: 0,
@@ -7484,7 +7506,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;