@agent-scope/runtime 1.1.0 → 1.3.0

package/dist/index.cjs

@@ -697,6 +697,131 @@ function walkFiberRoot(fiberRoot, options = {}) {
     children
   };
 }
+var HookLayout2 = 4;
+function isLightweightEffectNode(node) {
+  const ms = node.memoizedState;
+  if (ms === null || typeof ms !== "object") return false;
+  const obj = ms;
+  return typeof obj.create === "function" && "deps" in obj && typeof obj.tag === "number";
+}
+function isLightweightRefNode(node) {
+  if (node.queue != null) return false;
+  const ms = node.memoizedState;
+  if (ms === null || typeof ms !== "object" || Array.isArray(ms)) return false;
+  const keys = Object.keys(ms);
+  return keys.length === 1 && keys[0] === "current";
+}
+function isLightweightMemoTuple(node) {
+  if (node.queue != null) return false;
+  const ms = node.memoizedState;
+  if (!Array.isArray(ms) || ms.length !== 2) return false;
+  return ms[1] === null || Array.isArray(ms[1]);
+}
+function isLightweightStateOrReducer(node) {
+  return node.queue != null && typeof node.queue === "object" && typeof node.queue.dispatch === "function";
+}
+function isLightweightReducer(node) {
+  if (!isLightweightStateOrReducer(node)) return false;
+  const q = node.queue;
+  if (typeof q.reducer === "function") return true;
+  const lrr = q.lastRenderedReducer;
+  if (typeof lrr !== "function") return false;
+  const name = lrr.name ?? "";
+  return name !== "basicStateReducer" && name !== "";
+}
+function classifyHookType(node) {
+  if (isLightweightEffectNode(node)) {
+    const ms = node.memoizedState;
+    return ms.tag & HookLayout2 ? "useLayoutEffect" : "useEffect";
+  }
+  if (isLightweightRefNode(node)) return "useRef";
+  if (isLightweightMemoTuple(node)) {
+    const [val] = node.memoizedState;
+    return typeof val === "function" ? "useCallback" : "useMemo";
+  }
+  if (isLightweightStateOrReducer(node)) {
+    return isLightweightReducer(node) ? "useReducer" : "useState";
+  }
+  return "custom";
+}
+function countAndClassifyHooks(fiber) {
+  const hookTypes = [];
+  let node = fiber.memoizedState ?? null;
+  if (node === null || typeof node !== "object" || !("next" in node)) {
+    return { hookCount: 0, hookTypes: [] };
+  }
+  while (node !== null) {
+    hookTypes.push(classifyHookType(node));
+    node = node.next ?? null;
+  }
+  return { hookCount: hookTypes.length, hookTypes };
+}
+function walkFiberLightweight(fiber, options = {}) {
+  if (fiber === null || fiber === void 0) return null;
+  const includeHost = options.includeHostElements ?? false;
+  return walkFiberLightweightInner(fiber, includeHost, /* @__PURE__ */ new Set(), 0);
+}
+function walkFiberLightweightInner(fiber, includeHost, visited, depth) {
+  if (fiber === null || fiber === void 0) return null;
+  if (visited.has(fiber)) return null;
+  if (shouldSkip(fiber, includeHost)) return null;
+  visited.add(fiber);
+  const id = typeof fiber._debugID === "number" ? fiber._debugID : nextId();
+  const { hookCount, hookTypes } = countAndClassifyHooks(fiber);
+  const children = collectLightweightChildren(fiber, includeHost, visited, depth + 1);
+  const node = {
+    id,
+    name: extractName(fiber),
+    type: classifyType(fiber),
+    hookCount,
+    hookTypes,
+    childCount: children.length,
+    depth,
+    children
+  };
+  return node;
+}
+function collectLightweightChildren(fiber, includeHost, visited, childDepth) {
+  const nodes = [];
+  let current = fiber.child ?? null;
+  while (current !== null) {
+    if (visited.has(current)) {
+      current = current.sibling ?? null;
+      continue;
+    }
+    if (shouldSkip(current, includeHost)) {
+      const promoted = collectLightweightChildren(current, includeHost, visited, childDepth);
+      nodes.push(...promoted);
+    } else {
+      const node = walkFiberLightweightInner(current, includeHost, visited, childDepth);
+      if (node !== null) {
+        nodes.push(node);
+      }
+    }
+    current = current.sibling ?? null;
+  }
+  return nodes;
+}
+function walkFiberRootLightweight(fiberRoot, options = {}) {
+  if (!fiberRoot) return null;
+  const hostRootFiber = fiberRoot.current ?? null;
+  if (!hostRootFiber) return null;
+  const includeHost = options.includeHostElements ?? false;
+  const visited = /* @__PURE__ */ new Set();
+  const children = collectLightweightChildren(hostRootFiber, includeHost, visited, 0);
+  if (children.length === 0) return null;
+  if (children.length === 1) return children[0] ?? null;
+  return {
+    id: nextId(),
+    name: "Root",
+    type: "function",
+    hookCount: 0,
+    hookTypes: [],
+    childCount: children.length,
+    depth: 0,
+    children
+  };
+}
 
 // src/suspense-detector.ts
 var SuspenseComponent2 = 13;
@@ -788,18 +913,35 @@ async function capture(options = {}) {
     const walkOptions = {
       includeHostElements: options.includeHostElements ?? false
     };
-    const tree = walkFiberRoot(fiberRoot, walkOptions);
-    if (!tree) {
-      throw new Error(
-        "capture(): Fiber tree is empty. Make sure React has rendered at least one component."
-      );
-    }
     const hostRootFiber = fiberRoot.current ?? null;
     const rootChild = hostRootFiber?.child ?? null;
     const errors = detectErrors(rootChild);
     const suspenseBoundaries = detectSuspenseBoundaries(rootChild);
     const capturedIn = Date.now() - startTime;
     const consoleEntries = getConsoleEntries();
+    if (options.lightweight) {
+      const tree2 = walkFiberRootLightweight(fiberRoot, walkOptions);
+      if (!tree2) {
+        throw new Error(
+          "capture(): Fiber tree is empty. Make sure React has rendered at least one component."
+        );
+      }
+      return {
+        url,
+        timestamp: startTime,
+        capturedIn,
+        tree: tree2,
+        consoleEntries,
+        errors,
+        suspenseBoundaries
+      };
+    }
+    const tree = walkFiberRoot(fiberRoot, walkOptions);
+    if (!tree) {
+      throw new Error(
+        "capture(): Fiber tree is empty. Make sure React has rendered at least one component."
+      );
+    }
     return {
       url,
       timestamp: startTime,
@@ -874,6 +1016,8 @@ exports.resetProfilingData = resetProfilingData;
 exports.uninstallConsoleInterceptor = uninstallConsoleInterceptor;
 exports.waitForReact = waitForReact;
 exports.walkFiber = walkFiber;
+exports.walkFiberLightweight = walkFiberLightweight;
 exports.walkFiberRoot = walkFiberRoot;
+exports.walkFiberRootLightweight = walkFiberRootLightweight;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map