@wrongstack/tools 0.148.0 → 0.155.0

package/dist/pack.js

@@ -2748,6 +2748,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_IGNORE = [
   "node_modules",
   ".git",
@@ -2759,7 +2769,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 = [
@@ -2774,13 +2784,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 fs13.readdir(dir, { withFileTypes: true });
     } catch {
       return;
     }
+    dirCount++;
     for (const e of entries) {
       if (ignoreSet.has(e.name)) continue;
       const full = path2.join(dir, e.name);
@@ -2825,7 +2842,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 = [];
@@ -2837,7 +2854,7 @@ async function runIndexer(_ctx, opts) {
   if (opts.files && opts.files.length > 0) {
     files = opts.files.map((f) => path2.resolve(projectRoot, f)).filter((f) => !isGitIgnored(path2.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);
@@ -2856,11 +2873,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 fs13.stat(file);
-    } catch {
+      const statOpts = signal ? { signal } : {};
+      stat10 = await fs13.stat(file, statOpts);
+    } catch (e) {
+      if (isAbortError(e)) throw e;
       store.deleteFile(file);
       continue;
     }
@@ -2878,8 +2898,9 @@ async function runIndexer(_ctx, opts) {
     store.deleteSymbolsForFile(file);
     let content;
     try {
-      content = await fs13.readFile(file, "utf8");
+      content = await fs13.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;
     }
@@ -2969,12 +2990,13 @@ var codebaseIndexTool = {
       }
     }
   },
-  async execute(input, ctx) {
+  async execute(input, ctx, execOpts) {
     const result = await runIndexer(ctx, {
       projectRoot: ctx.projectRoot,
       force: input.force ?? false,
       langs: input.langs,
-      indexDir: codebaseIndexDirOverride(ctx)
+      indexDir: codebaseIndexDirOverride(ctx),
+      signal: execOpts?.signal
     });
     setIndexReady();
     return result;