rerender-lens 0.3.0 → 0.4.0

package/dist/vitest-setup.cjs

@@ -30,10 +30,49 @@ function remember(seen, a, b) {
   set.add(b);
   return false;
 }
+var scope = null;
+var DEFAULT_DIFF_BUDGET = 5e4;
+function beginDiffScope(budget = 2e5) {
+  const previous = scope;
+  scope = { cache: /* @__PURE__ */ new WeakMap(), remaining: budget, exhausted: false };
+  return () => {
+    scope = previous;
+  };
+}
+function diffBudgetExhausted() {
+  return scope?.exhausted === true;
+}
 function deepEqual(a, b, seen = /* @__PURE__ */ new Map()) {
   if (Object.is(a, b)) return true;
   if (typeof a !== typeof b) return false;
   if (typeof a !== "object" || a === null || b === null) return false;
+  if (!scope) {
+    const end = beginDiffScope(DEFAULT_DIFF_BUDGET);
+    try {
+      return deepEqual(a, b, seen);
+    } finally {
+      end();
+    }
+  }
+  const sc = scope;
+  const objA = a;
+  const objB = b;
+  const hit = sc.cache.get(objA)?.get(objB);
+  if (hit !== void 0) return hit;
+  if (--sc.remaining < 0) {
+    sc.exhausted = true;
+    return false;
+  }
+  const result = deepEqualObjects(objA, objB, seen);
+  let m = sc.cache.get(objA);
+  if (!m) {
+    m = /* @__PURE__ */ new WeakMap();
+    sc.cache.set(objA, m);
+  }
+  m.set(objB, result);
+  return result;
+}
+function deepEqualObjects(a, b, seen) {
   const objA = a;
   const objB = b;
   if (remember(seen, objA, objB)) return true;
@@ -62,6 +101,10 @@ function deepEqual(a, b, seen = /* @__PURE__ */ new Map()) {
   }
   if (a instanceof Set) {
     if (!(b instanceof Set) || a.size !== b.size) return false;
+    if (a.size > 50) {
+      for (const v of a) if (!b.has(v)) return false;
+      return true;
+    }
     outer: for (const v of a) {
       if (b.has(v)) continue;
       for (const w of b) if (deepEqual(v, w, seen)) continue outer;
@@ -77,9 +120,10 @@ function deepEqual(a, b, seen = /* @__PURE__ */ new Map()) {
     const ka = Object.keys(a);
     const kb = Object.keys(b);
     if (ka.length !== kb.length) return false;
+    const rb = b;
     for (const k of ka) {
       if (!Object.prototype.hasOwnProperty.call(b, k)) return false;
-      if (!deepEqual(a[k], b[k], seen)) return false;
+      if (!deepEqual(a[k], rb[k], seen)) return false;
     }
     return true;
   }
@@ -121,7 +165,7 @@ function diffRecords(prev, next, basePath = "") {
     if (Object.is(a, b)) continue;
     const kind = classify(a, b);
     if (kind === "different") {
-      changes.push({ path: firstDifferentPath(a, b, path), kind, prev: a, next: b });
+      changes.push({ path: diffBudgetExhausted() ? path : firstDifferentPath(a, b, path), kind, prev: a, next: b });
     } else {
       changes.push({ path, kind, prev: a, next: b });
     }
@@ -332,7 +376,7 @@ function getState() {
   const g = globalThis;
   let s = g[KEY];
   if (!s) {
-    s = { options: {}, enabled: false, printed: /* @__PURE__ */ new Map(), detach: null, nextInstanceId: 1, nextCommitId: 1, warnedOnce: /* @__PURE__ */ new Set(), scheduled: 0, commits: 0, overheadMs: 0, maxCommitMs: 0 };
+    s = { options: {}, enabled: false, printed: /* @__PURE__ */ new Map(), detach: null, nextInstanceId: 1, nextCommitId: 1, warnedOnce: /* @__PURE__ */ new Set(), scheduled: 0, commits: 0, overheadMs: 0, maxCommitMs: 0, truncated: 0 };
     g[KEY] = s;
   }
   return s;
@@ -362,7 +406,7 @@ function dispatch(report, override) {
     try {
       options.notifier(report);
     } catch (err) {
-      (options.console ?? console).warn("[rerender-lens] notifier threw", err);
+      warnOnce("notifier", `notifier threw: ${String(err)}`);
     }
   }
 }
@@ -750,11 +794,14 @@ var counts = /* @__PURE__ */ new WeakMap();
 var ids = /* @__PURE__ */ new WeakMap();
 var fibersById = /* @__PURE__ */ new Map();
 var hasWeakRef = typeof WeakRef === "function";
+var pruneAt = 5e3;
 function remember2(id, fiber) {
   if (!hasWeakRef) return;
+  if (fibersById.get(id)?.deref() === fiber) return;
   fibersById.set(id, new WeakRef(fiber));
-  if (fibersById.size > 5e3) {
+  if (fibersById.size > pruneAt) {
     for (const [k, ref] of fibersById) if (!ref.deref()) fibersById.delete(k);
+    pruneAt = Math.max(5e3, fibersById.size * 2);
   }
 }
 function instanceIdOf(fiber) {
@@ -783,7 +830,7 @@ function nearestComponent(fiber) {
 }
 var INTERNAL_FRAME = /react-dom|react_jsx|jsx-(dev-)?runtime|\/react\/|node_modules\/react|react-stack-bottom-frame|react_stack_bottom_frame|scheduler/;
 function parseStackLocation(stack) {
-  for (const line of stack.split("\n")) {
+  for (const line of stack.split("\n", 24)) {
     const m = /(?:at\s+(?:.*?\s+)?\(?|@)?((?:https?|file|webpack|vite|blob):[^\s()]+?):(\d+):(\d+)\)?\s*$/.exec(line.trim());
     if (!m || !m[1] || INTERNAL_FRAME.test(m[1])) continue;
     return { fileName: m[1], lineNumber: Number(m[2]), columnNumber: Number(m[3]) };
@@ -799,9 +846,16 @@ function sourceOf(fiber) {
     return out;
   }
   const st = fiber._debugStack;
-  const text = typeof st === "string" ? st : st && typeof st.stack === "string" ? st.stack : null;
-  return text ? parseStackLocation(text) : void 0;
+  if (st && typeof st === "object") {
+    if (sourceCache.has(st)) return sourceCache.get(st);
+    const text = typeof st.stack === "string" ? st.stack : null;
+    const out = text ? parseStackLocation(text) : void 0;
+    sourceCache.set(st, out);
+    return out;
+  }
+  return typeof st === "string" ? parseStackLocation(st) : void 0;
 }
+var sourceCache = /* @__PURE__ */ new WeakMap();
 function bumpCount(fiber) {
   const prev = counts.get(fiber) ?? (fiber.alternate ? counts.get(fiber.alternate) : void 0) ?? 0;
   const n = prev + 1;
@@ -989,6 +1043,8 @@ function updatersOf(root) {
 }
 var lastCommit = null;
 var EFFECT_LOOP_WINDOW_MS = 50;
+var MAX_REPORTS_PER_COMMIT = 200;
+var COMMIT_TIME_BUDGET_MS = 25;
 function onCommit(root, commitPriority) {
   const s = getState();
   if (!s.enabled) return;
@@ -1019,7 +1075,8 @@ function onCommit(root, commitPriority) {
   const updaters = updatersOf(root);
   let commitCause;
   let afterCommit;
-  if (lastCommit && at - lastCommit.at < EFFECT_LOOP_WINDOW_MS && updaters.some((u) => lastCommit.rendered.has(u))) {
+  const inputDriven = commitPriority === "immediate" || commitPriority === "user-blocking";
+  if (!inputDriven && lastCommit && at - lastCommit.at < EFFECT_LOOP_WINDOW_MS && updaters.some((u) => lastCommit.rendered.has(u))) {
     commitCause = "effect-after-commit";
     afterCommit = lastCommit.id;
   } else if (suspenseResolved) commitCause = "suspense-resolved";
@@ -1031,47 +1088,59 @@ function onCommit(root, commitPriority) {
   lastCommit = { id: commitId, at, rendered: renderedNames };
   const dispatcherRef = o.resolveHookNames ? getDispatcherRef() : null;
   const parentCache = /* @__PURE__ */ new Map();
-  for (const fiber of rendered) {
-    const alt = fiber.alternate;
-    const a = analyze(fiber, alt, trackHooks);
-    const durations = durationsOf(fiber);
-    let hookState = o.includeState !== false && trackHooks ? snapshotHooks(fiber) : void 0;
-    if (dispatcherRef && fiber.tag !== ClassComponent && (a.hookChanges.length || hookState && hookState.length)) {
-      const names = resolveHookNames(fiber, dispatcherRef);
-      if (names.size) {
-        for (const c of a.hookChanges) if (c.hook !== "useContext" && names.has(c.index)) c.custom = names.get(c.index);
-        if (hookState) hookState = hookState.map((h) => names.has(h.index) ? { ...h, custom: names.get(h.index) } : h);
+  const endDiffScope = beginDiffScope();
+  const deadline = at + COMMIT_TIME_BUDGET_MS;
+  try {
+    for (let i = 0; i < rendered.length; i++) {
+      if (i >= MAX_REPORTS_PER_COMMIT || (i & 15) === 15 && nowMs() > deadline) {
+        s.truncated += rendered.length - i;
+        warnOnce("truncated", `a commit rendered ${rendered.length} tracked components; only ${i} were reported (see info().truncated). Narrow \`include\` or turn off includeState.`);
+        break;
+      }
+      const fiber = rendered[i];
+      const alt = fiber.alternate;
+      const a = analyze(fiber, alt, trackHooks);
+      const durations = durationsOf(fiber);
+      let hookState = o.includeState !== false && trackHooks ? snapshotHooks(fiber) : void 0;
+      if (dispatcherRef && fiber.tag !== ClassComponent && (a.hookChanges.length || hookState && hookState.length)) {
+        const names = resolveHookNames(fiber, dispatcherRef);
+        if (names.size) {
+          for (const c of a.hookChanges) if (c.hook !== "useContext" && names.has(c.index)) c.custom = names.get(c.index);
+          if (hookState) hookState = hookState.map((h) => names.has(h.index) ? { ...h, custom: names.get(h.index) } : h);
+        }
       }
+      const report = buildReport({
+        component: fiberName(fiber),
+        instanceId: instanceId(fiber),
+        renderCount: bumpCount(fiber),
+        prevProps: alt.memoizedProps ?? {},
+        nextProps: fiber.memoizedProps ?? {},
+        propChanges: a.propChanges,
+        stateChanges: a.stateChanges,
+        hookChanges: a.hookChanges,
+        ...o.includeState !== false ? {
+          hookState,
+          contexts: trackHooks ? snapshotContexts(fiber) : void 0,
+          state: fiber.tag === ClassComponent && fiber.memoizedState && typeof fiber.memoizedState === "object" ? fiber.memoizedState : void 0
+        } : {},
+        updaters,
+        commitCause,
+        afterCommit,
+        key: fiber.key === null || fiber.key === void 0 ? null : String(fiber.key),
+        parent: a.trigger === "parent" ? nearestRenderedAncestor(fiber, parentCache, trackHooks) : null,
+        owner: ownerName(fiber),
+        path: componentPath(fiber),
+        memoized: isMemoizedFiber(fiber),
+        selfDuration: durations ? durations.self : void 0,
+        treeDuration: durations ? durations.tree : void 0,
+        commitId,
+        commitPriority,
+        source: sourceOf(fiber)
+      });
+      dispatch(report);
     }
-    const report = buildReport({
-      component: fiberName(fiber),
-      instanceId: instanceId(fiber),
-      renderCount: bumpCount(fiber),
-      prevProps: alt.memoizedProps ?? {},
-      nextProps: fiber.memoizedProps ?? {},
-      propChanges: a.propChanges,
-      stateChanges: a.stateChanges,
-      hookChanges: a.hookChanges,
-      ...o.includeState !== false ? {
-        hookState,
-        contexts: trackHooks ? snapshotContexts(fiber) : void 0,
-        state: fiber.tag === ClassComponent && fiber.memoizedState && typeof fiber.memoizedState === "object" ? fiber.memoizedState : void 0
-      } : {},
-      updaters,
-      commitCause,
-      afterCommit,
-      key: fiber.key === null || fiber.key === void 0 ? null : String(fiber.key),
-      parent: a.trigger === "parent" ? nearestRenderedAncestor(fiber, parentCache, trackHooks) : null,
-      owner: ownerName(fiber),
-      path: componentPath(fiber),
-      memoized: isMemoizedFiber(fiber),
-      selfDuration: durations ? durations.self : void 0,
-      treeDuration: durations ? durations.tree : void 0,
-      commitId,
-      commitPriority,
-      source: sourceOf(fiber)
-    });
-    dispatch(report);
+  } finally {
+    endDiffScope();
   }
 }