pi-lens 4.4.3 → 4.4.4

package/dist/{chunk-QHJ6DQ2R.js → chunk-US7ECLD7.js}

@@ -6,26 +6,26 @@ import {
   EXTENSION_TO_GRAMMAR,
   getUserRuleRoot,
   ruleCorpusFingerprintForCycle
-} from "./chunk-724UL2TJ.js";
+} from "./chunk-GXFV6MYU.js";
 import {
   getPackageRoot,
   js_yaml_default,
   resolvePackagePath
-} from "./chunk-QOZXJ4NA.js";
+} from "./chunk-FM4RDMKT.js";
 import {
   transientRetryDelayMs
-} from "./chunk-VRZP76YM.js";
+} from "./chunk-P63GXC37.js";
 import {
   getProjectIgnoreMatcher,
   isExcludedDirName
-} from "./chunk-C34ZXZ7O.js";
+} from "./chunk-KUA4OJ5Y.js";
 import {
   notifyUserDegradation
-} from "./chunk-2BBLFX75.js";
+} from "./chunk-J6MPGQTS.js";
 import {
   assertInstallAllowed,
   getProjectTrustGeneration
-} from "./chunk-ZANLILYD.js";
+} from "./chunk-M2IFVFWU.js";
 import {
   writeFileAtomic
 } from "./chunk-2EECBNC5.js";
@@ -36,20 +36,20 @@ import {
   incrementDegradationCount,
   recordDegradation,
   recordDegradationOnce
-} from "./chunk-JAU72XNY.js";
+} from "./chunk-J64T55EI.js";
 import {
   createNdjsonLogger,
   getGlobalPiLensLogDir,
   getMaxLogSizeMB,
   isTestMode
-} from "./chunk-AOHZU2BM.js";
+} from "./chunk-YNE3RP52.js";
 import {
   BoundedFifoMap
 } from "./chunk-ABBOA7UD.js";
 import {
   normalizeFilePath,
   normalizeLoggedPath
-} from "./chunk-N2H3ZPZY.js";
+} from "./chunk-3RKP3FMS.js";
 
 // dist/clients/tree-sitter-logger.js
 import * as path from "node:path";
@@ -319,17 +319,18 @@ function shortHash(hash) {
   const hex = hash.startsWith("sha256:") ? hash.slice("sha256:".length) : hash;
   return `sha256:${hex.slice(0, 12)}\u2026`;
 }
-async function downloadGrammarDetailed(destDir, filename) {
+async function downloadGrammarDetailed(destDir, filename, options = {}) {
   if (isVendoredGrammar(filename))
     return vendoredGrammarRefusal(filename);
+  const stall = options.stallMs === void 0 ? void 0 : armStallBound(options.stallMs);
   try {
     fs.mkdirSync(destDir, { recursive: true });
-    const res = await fetch(grammarSourceUrl(filename));
+    const res = await fetch(grammarSourceUrl(filename), stall ? { signal: stall.signal } : void 0);
     if (!res.ok) {
       const retryable = res.status !== 404 && res.status !== 410;
       return { ok: false, retryable };
     }
-    const data = Buffer.from(await res.arrayBuffer());
+    const data = await readResponseBody(res, stall);
     if (!hasWasmMagic(data)) {
       return { ok: false, retryable: true, reason: describeNonWasmBody(data) };
     }
@@ -358,7 +359,42 @@ async function downloadGrammarDetailed(destDir, filename) {
     writeFileAtomic(path2.join(destDir, filename), data, { bestEffort: false });
     return { ok: true, retryable: true };
   } catch {
+    if (stall?.signal.aborted) {
+      return {
+        ok: false,
+        retryable: true,
+        reason: `No bytes arrived for ${options.stallMs} ms \u2014 a stalled transfer.`
+      };
+    }
     return { ok: false, retryable: true };
+  } finally {
+    stall?.dispose();
+  }
+}
+var GRAMMAR_STALL_MS = 15e3;
+function armStallBound(ms) {
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), ms);
+  timer.unref();
+  return {
+    signal: controller.signal,
+    touch: () => {
+      timer.refresh();
+    },
+    dispose: () => clearTimeout(timer)
+  };
+}
+async function readResponseBody(res, stall) {
+  if (!stall || !res.body)
+    return Buffer.from(await res.arrayBuffer());
+  const reader = res.body.getReader();
+  const chunks = [];
+  for (; ; ) {
+    const { done, value } = await reader.read();
+    if (done)
+      return Buffer.concat(chunks);
+    stall.touch();
+    chunks.push(value);
   }
 }
 
@@ -2110,6 +2146,8 @@ var TreeSitterClient = class _TreeSitterClient {
    * (the `transientRetryDelayMs` shape from availability-policy.ts) instead
    * of latching for the process lifetime. */
   grammarRetryAtMs = /* @__PURE__ */ new Map();
+  /** Instance field so a test can shorten it without waiting out the real bound. */
+  grammarStallMs = GRAMMAR_STALL_MS;
   /**
    * Grammar path → the `size:mtimeMs` stamp at which its wasm preamble was
    * verified (#1548). A POSITIVE-only memo, so the steady-state resolve costs
@@ -2634,6 +2672,10 @@ var TreeSitterClient = class _TreeSitterClient {
    * `vendor/grammars` comes first because it is the ONLY source for a grammar
    * we build ourselves (`VENDORED_GRAMMARS`) — nothing downloads into the
    * later dirs for it, so a miss here is a miss everywhere.
+   *
+   * Public so the Vitest grammar prefetch (`tests/support/prewarm-grammars.ts`)
+   * counts a grammar as present exactly where this resolver would find it
+   * (#4342), instead of keeping its own copy of the list.
    */
   grammarSourceDirs() {
     const dirs = [];
@@ -2944,7 +2986,7 @@ var TreeSitterClient = class _TreeSitterClient {
       );
       return false;
     }
-    const { ok, retryable, reason } = await downloadGrammarDetailed(dir, grammarFile);
+    const { ok, retryable, reason } = await downloadGrammarDetailed(dir, grammarFile, { stallMs: this.grammarStallMs });
     if (ok) {
       if (!this.grammarsDir)
         this.grammarsDir = dir;

package/dist/{chunk-AB3YYG6N.js → chunk-UZ2XCLYS.js}

@@ -4,32 +4,32 @@ import {
   discoverManagedTool,
   getManagedToolEnvironment,
   resolveAvailableOrInstall
-} from "./chunk-YSZNHOBY.js";
+} from "./chunk-ZHH37KZL.js";
 import {
   mapWithConcurrency
 } from "./chunk-XIMYZ3JA.js";
 import {
   findNodeToolBinary
-} from "./chunk-AW7R2GWG.js";
+} from "./chunk-5VSF4UUQ.js";
 import {
   getIsolatedNpxSpawnOptions
-} from "./chunk-RWWYJ623.js";
+} from "./chunk-3KAZU3VR.js";
 import {
   createAvailabilityLatch
-} from "./chunk-VRZP76YM.js";
+} from "./chunk-P63GXC37.js";
 import {
   safeSpawnAsync
-} from "./chunk-C34ZXZ7O.js";
+} from "./chunk-KUA4OJ5Y.js";
 import {
   compareOrdinal,
   recordDegradationOnce
-} from "./chunk-JAU72XNY.js";
+} from "./chunk-J64T55EI.js";
 import {
   createSubsystemLogger
-} from "./chunk-XTMJ7XDT.js";
+} from "./chunk-POMLXPMY.js";
 import {
   isFullyQualified
-} from "./chunk-N2H3ZPZY.js";
+} from "./chunk-3RKP3FMS.js";
 
 // dist/clients/dependency-checker.js
 import * as fs from "node:fs";