@apollo/client 3.9.0-alpha.4 → 3.9.0-alpha.5

package/testing/internal/profile/profile.d.ts

@@ -3,53 +3,107 @@ import type { Render, BaseRender } from "./Render.js";
 type ValidSnapshot = void | (object & {
     call?: never;
 });
+/** only used for passing around data internally */
 declare const _stackTrace: unique symbol;
+/** @internal */
 export interface NextRenderOptions {
     timeout?: number;
     [_stackTrace]?: string;
 }
-export interface ProfiledComponent<Props, Snapshot> extends React.FC<Props>, ProfiledComponentFields<Props, Snapshot>, ProfiledComponenOnlyFields<Props, Snapshot> {
+/** @internal */
+interface ProfilerProps {
+    children: React.ReactNode;
 }
-interface UpdateSnapshot<Snapshot> {
+/** @internal */
+export interface Profiler<Snapshot> extends React.FC<ProfilerProps>, ProfiledComponentFields<Snapshot>, ProfiledComponentOnlyFields<Snapshot> {
+}
+interface ReplaceSnapshot<Snapshot> {
     (newSnapshot: Snapshot): void;
     (updateSnapshot: (lastSnapshot: Readonly<Snapshot>) => Snapshot): void;
 }
-interface ProfiledComponenOnlyFields<Props, Snapshot> {
-    updateSnapshot: UpdateSnapshot<Snapshot>;
+interface MergeSnapshot<Snapshot> {
+    (partialSnapshot: Partial<Snapshot>): void;
+    (updatePartialSnapshot: (lastSnapshot: Readonly<Snapshot>) => Partial<Snapshot>): void;
+}
+interface ProfiledComponentOnlyFields<Snapshot> {
+    mergeSnapshot: MergeSnapshot<Snapshot>;
+    replaceSnapshot: ReplaceSnapshot<Snapshot>;
 }
-interface ProfiledComponentFields<Props, Snapshot> {
+interface ProfiledComponentFields<Snapshot> {
+    /**
+     * An array of all renders that have happened so far.
+     * Errors thrown during component render will be captured here, too.
+     */
     renders: Array<Render<Snapshot> | {
         phase: "snapshotError";
         count: number;
         error: unknown;
     }>;
+    /**
+     * Peeks the next render from the current iterator position, without advancing the iterator.
+     * If no render has happened yet, it will wait for the next render to happen.
+     * @throws {WaitForRenderTimeoutError} if no render happens within the timeout
+     */
     peekRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;
+    /**
+     * Iterates to the next render and returns it.
+     * If no render has happened yet, it will wait for the next render to happen.
+     * @throws {WaitForRenderTimeoutError} if no render happens within the timeout
+     */
     takeRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;
-    currentRenderCount(): number;
+    /**
+     * Returns the total number of renders.
+     */
+    totalRenderCount(): number;
+    /**
+     * Returns the current render.
+     * @throws {Error} if no render has happened yet
+     */
     getCurrentRender(): Render<Snapshot>;
-    takeUntilRenderCount(count: number, optionsPerRender?: NextRenderOptions): Promise<void>;
+    /**
+     * Waits for the next render to happen.
+     * Does not advance the render iterator.
+     */
     waitForNextRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;
 }
-export declare function profile<Snapshot extends ValidSnapshot = void, Props = Record<string, never>>({ Component, onRender, snapshotDOM, initialSnapshot, }: {
+export interface ProfiledComponent<Snapshot extends ValidSnapshot, Props = {}> extends React.FC<Props>, ProfiledComponentFields<Snapshot>, ProfiledComponentOnlyFields<Snapshot> {
+}
+/** @internal */
+export declare function profile<Snapshot extends ValidSnapshot = void, Props = {}>({ Component, ...options }: Parameters<typeof createProfiler<Snapshot>>[0] & {
     Component: React.ComponentType<Props>;
+}): ProfiledComponent<Snapshot, Props>;
+/** @internal */
+export declare function createProfiler<Snapshot extends ValidSnapshot = void>({ onRender, snapshotDOM, initialSnapshot, skipNonTrackingRenders, }?: {
     onRender?: (info: BaseRender & {
         snapshot: Snapshot;
-        updateSnapshot: UpdateSnapshot<Snapshot>;
+        replaceSnapshot: ReplaceSnapshot<Snapshot>;
+        mergeSnapshot: MergeSnapshot<Snapshot>;
     }) => void;
     snapshotDOM?: boolean;
     initialSnapshot?: Snapshot;
-}): ProfiledComponent<Props, Snapshot>;
+    /**
+     * This will skip renders during which no renders tracked by
+     * `useTrackRenders` occured.
+     */
+    skipNonTrackingRenders?: boolean;
+}): Profiler<Snapshot>;
+/** @internal */
 export declare class WaitForRenderTimeoutError extends Error {
     constructor();
 }
 type StringReplaceRenderWithSnapshot<T extends string> = T extends `${infer Pre}Render${infer Post}` ? `${Pre}Snapshot${Post}` : T;
 type ResultReplaceRenderWithSnapshot<T> = T extends (...args: infer Args) => Render<infer Snapshot> ? (...args: Args) => Snapshot : T extends (...args: infer Args) => Promise<Render<infer Snapshot>> ? (...args: Args) => Promise<Snapshot> : T;
-type ProfiledHookFields<Props, ReturnValue> = ProfiledComponentFields<Props, ReturnValue> extends infer PC ? {
+type ProfiledHookFields<ReturnValue> = ProfiledComponentFields<ReturnValue> extends infer PC ? {
     [K in keyof PC as StringReplaceRenderWithSnapshot<K & string>]: ResultReplaceRenderWithSnapshot<PC[K]>;
 } : never;
-export interface ProfiledHook<Props, ReturnValue> extends React.FC<Props>, ProfiledHookFields<Props, ReturnValue> {
-    ProfiledComponent: ProfiledComponent<Props, ReturnValue>;
+/** @internal */
+export interface ProfiledHook<Props, ReturnValue> extends React.FC<Props>, ProfiledHookFields<ReturnValue> {
+    Profiler: Profiler<ReturnValue>;
 }
+/** @internal */
 export declare function profileHook<ReturnValue extends ValidSnapshot, Props>(renderCallback: (props: Props) => ReturnValue): ProfiledHook<Props, ReturnValue>;
+export declare function useTrackRenders({ name }?: {
+    name?: string;
+}): void;
 export {};
 //# sourceMappingURL=profile.d.ts.map

package/testing/internal/profile/profile.js

@@ -1,32 +1,67 @@
 var _a, _b;
-import { __assign, __awaiter, __extends, __generator } from "tslib";
+import { __addDisposableResource, __assign, __awaiter, __disposeResources, __extends, __generator, __rest } from "tslib";
 import * as React from "react";
 import { TextEncoder, TextDecoder } from "util";
 (_a = global.TextEncoder) !== null && _a !== void 0 ? _a : (global.TextEncoder = TextEncoder);
+// @ts-ignore
 (_b = global.TextDecoder) !== null && _b !== void 0 ? _b : (global.TextDecoder = TextDecoder);
 import { RenderInstance } from "./Render.js";
 import { applyStackTrace, captureStackTrace } from "./traces.js";
+import { ProfilerContextProvider, useProfilerContext } from "./context.js";
+import { disableActWarnings } from "../disposables/index.js";
+/** only used for passing around data internally */
 var _stackTrace = Symbol();
+/** @internal */
 export function profile(_a) {
-    var Component = _a.Component, onRender = _a.onRender, _b = _a.snapshotDOM, snapshotDOM = _b === void 0 ? false : _b, initialSnapshot = _a.initialSnapshot;
-    var currentRender;
+    var Component = _a.Component, options = __rest(_a, ["Component"]);
+    var Profiler = createProfiler(options);
+    return Object.assign(function ProfiledComponent(props) {
+        return (React.createElement(Profiler, null,
+            React.createElement(Component, __assign({}, props))));
+    }, {
+        mergeSnapshot: Profiler.mergeSnapshot,
+        replaceSnapshot: Profiler.replaceSnapshot,
+        getCurrentRender: Profiler.getCurrentRender,
+        peekRender: Profiler.peekRender,
+        takeRender: Profiler.takeRender,
+        totalRenderCount: Profiler.totalRenderCount,
+        waitForNextRender: Profiler.waitForNextRender,
+        get renders() {
+            return Profiler.renders;
+        },
+    });
+}
+/** @internal */
+export function createProfiler(_a) {
+    var _b = _a === void 0 ? {} : _a, onRender = _b.onRender, _c = _b.snapshotDOM, snapshotDOM = _c === void 0 ? false : _c, initialSnapshot = _b.initialSnapshot, skipNonTrackingRenders = _b.skipNonTrackingRenders;
     var nextRender;
     var resolveNextRender;
     var rejectNextRender;
     var snapshotRef = { current: initialSnapshot };
-    var updateSnapshot = function (snap) {
+    var replaceSnapshot = function (snap) {
         if (typeof snap === "function") {
             if (!initialSnapshot) {
                 throw new Error("Cannot use a function to update the snapshot if no initial snapshot was provided.");
             }
-            snapshotRef.current = snap(typeof snapshotRef.current === "object"
-                ? __assign({}, snapshotRef.current) : snapshotRef.current);
+            snapshotRef.current = snap(typeof snapshotRef.current === "object" ? __assign({}, snapshotRef.current) : snapshotRef.current);
         }
         else {
             snapshotRef.current = snap;
         }
     };
+    var mergeSnapshot = function (partialSnapshot) {
+        replaceSnapshot(function (snapshot) { return (__assign(__assign({}, snapshot), (typeof partialSnapshot === "function" ?
+            partialSnapshot(snapshot)
+            : partialSnapshot))); });
+    };
+    var profilerContext = {
+        renderedComponents: [],
+    };
     var profilerOnRender = function (id, phase, actualDuration, baseDuration, startTime, commitTime) {
+        if (skipNonTrackingRenders &&
+            profilerContext.renderedComponents.length === 0) {
+            return;
+        }
         var baseRender = {
             id: id,
             phase: phase,
@@ -34,23 +69,28 @@ export function profile(_a) {
             baseDuration: baseDuration,
             startTime: startTime,
             commitTime: commitTime,
-            count: Profiled.renders.length + 1,
+            count: Profiler.renders.length + 1,
         };
         try {
-            onRender === null || onRender === void 0 ? void 0 : onRender(__assign(__assign({}, baseRender), { updateSnapshot: updateSnapshot, snapshot: snapshotRef.current }));
+            /*
+             * The `onRender` function could contain `expect` calls that throw
+             * `JestAssertionError`s - but we are still inside of React, where errors
+             * might be swallowed.
+             * So we record them and re-throw them in `takeRender`
+             * Additionally, we reject the `waitForNextRender` promise.
+             */
+            onRender === null || onRender === void 0 ? void 0 : onRender(__assign(__assign({}, baseRender), { replaceSnapshot: replaceSnapshot, mergeSnapshot: mergeSnapshot, snapshot: snapshotRef.current }));
             var snapshot = snapshotRef.current;
-            var domSnapshot = snapshotDOM
-                ? window.document.body.innerHTML
-                : undefined;
-            var render = new RenderInstance(baseRender, snapshot, domSnapshot);
-            currentRender = render;
-            Profiled.renders.push(render);
+            var domSnapshot = snapshotDOM ? window.document.body.innerHTML : undefined;
+            var render = new RenderInstance(baseRender, snapshot, domSnapshot, profilerContext.renderedComponents);
+            profilerContext.renderedComponents = [];
+            Profiler.renders.push(render);
             resolveNextRender === null || resolveNextRender === void 0 ? void 0 : resolveNextRender(render);
         }
         catch (error) {
-            Profiled.renders.push({
+            Profiler.renders.push({
                 phase: "snapshotError",
-                count: Profiled.renders.length,
+                count: Profiler.renders.length,
                 error: error,
             });
             rejectNextRender === null || rejectNextRender === void 0 ? void 0 : rejectNextRender(error);
@@ -60,83 +100,101 @@ export function profile(_a) {
         }
     };
     var iteratorPosition = 0;
-    var Profiled = Object.assign(function (props) { return (React.createElement(React.Profiler, { id: "test", onRender: profilerOnRender },
-        React.createElement(Component, __assign({}, props)))); }, {
-        updateSnapshot: updateSnapshot,
+    var Profiler = Object.assign(function (_a) {
+        var children = _a.children;
+        return (React.createElement(ProfilerContextProvider, { value: profilerContext },
+            React.createElement(React.Profiler, { id: "test", onRender: profilerOnRender }, children)));
+    }, {
+        replaceSnapshot: replaceSnapshot,
+        mergeSnapshot: mergeSnapshot,
     }, {
         renders: new Array(),
-        currentRenderCount: function () {
-            return Profiled.renders.length;
+        totalRenderCount: function () {
+            return Profiler.renders.length;
         },
         peekRender: function (options) {
             if (options === void 0) { options = {}; }
             return __awaiter(this, void 0, void 0, function () {
-                var render_1, render;
+                var render;
                 var _a;
                 return __generator(this, function (_b) {
-                    if (iteratorPosition < Profiled.renders.length) {
-                        render_1 = Profiled.renders[iteratorPosition];
-                        if (render_1.phase === "snapshotError") {
-                            throw render_1.error;
+                    if (iteratorPosition < Profiler.renders.length) {
+                        render = Profiler.renders[iteratorPosition];
+                        if (render.phase === "snapshotError") {
+                            throw render.error;
                         }
-                        return [2, render_1];
+                        return [2 /*return*/, render];
                     }
-                    render = Profiled.waitForNextRender(__assign((_a = {}, _a[_stackTrace] = captureStackTrace(Profiled.peekRender), _a), options));
-                    return [2, render];
+                    return [2 /*return*/, Profiler.waitForNextRender(__assign((_a = {}, _a[_stackTrace] = captureStackTrace(Profiler.peekRender), _a), options))];
                 });
             });
         },
         takeRender: function (options) {
             if (options === void 0) { options = {}; }
             return __awaiter(this, void 0, void 0, function () {
-                var error, e_1;
+                var env_1, _disabledActWarnings, error, e_1, e_2;
                 var _a;
                 return __generator(this, function (_b) {
                     switch (_b.label) {
                         case 0:
-                            error = undefined;
+                            env_1 = { stack: [], error: void 0, hasError: false };
                             _b.label = 1;
                         case 1:
-                            _b.trys.push([1, 3, 4, 5]);
-                            return [4, Profiled.peekRender(__assign((_a = {}, _a[_stackTrace] = captureStackTrace(Profiled.takeRender), _a), options))];
-                        case 2: return [2, _b.sent()];
-                        case 3:
+                            _b.trys.push([1, 7, 8, 9]);
+                            _disabledActWarnings = __addDisposableResource(env_1, disableActWarnings(), false);
+                            error = undefined;
+                            _b.label = 2;
+                        case 2:
+                            _b.trys.push([2, 4, 5, 6]);
+                            return [4 /*yield*/, Profiler.peekRender(__assign((_a = {}, _a[_stackTrace] = captureStackTrace(Profiler.takeRender), _a), options))];
+                        case 3: return [2 /*return*/, _b.sent()];
+                        case 4:
                             e_1 = _b.sent();
                             error = e_1;
                             throw e_1;
-                        case 4:
+                        case 5:
                             if (!(error && error instanceof WaitForRenderTimeoutError)) {
                                 iteratorPosition++;
                             }
-                            return [7];
-                        case 5: return [2];
+                            return [7 /*endfinally*/];
+                        case 6: return [3 /*break*/, 9];
+                        case 7:
+                            e_2 = _b.sent();
+                            env_1.error = e_2;
+                            env_1.hasError = true;
+                            return [3 /*break*/, 9];
+                        case 8:
+                            __disposeResources(env_1);
+                            return [7 /*endfinally*/];
+                        case 9: return [2 /*return*/];
                     }
                 });
             });
         },
         getCurrentRender: function () {
-            if (!currentRender) {
-                throw new Error("Has not been rendered yet!");
+            // The "current" render should point at the same render that the most
+            // recent `takeRender` call returned, so we need to get the "previous"
+            // iterator position, otherwise `takeRender` advances the iterator
+            // to the next render. This means we need to call `takeRender` at least
+            // once before we can get a current render.
+            var currentPosition = iteratorPosition - 1;
+            if (currentPosition < 0) {
+                throw new Error("No current render available. You need to call `takeRender` before you can get the current render.");
             }
-            return currentRender;
-        },
-        takeUntilRenderCount: function (count, optionsPerRender) {
-            return __awaiter(this, void 0, void 0, function () {
-                return __generator(this, function (_a) {
-                    switch (_a.label) {
-                        case 0:
-                            if (!(Profiled.renders.length < count)) return [3, 2];
-                            return [4, Profiled.takeRender(optionsPerRender)];
-                        case 1:
-                            _a.sent();
-                            return [3, 0];
-                        case 2: return [2];
-                    }
-                });
-            });
+            var render = Profiler.renders[currentPosition];
+            if (render.phase === "snapshotError") {
+                throw render.error;
+            }
+            return render;
         },
         waitForNextRender: function (_a) {
-            var _b = _a === void 0 ? {} : _a, _c = _b.timeout, timeout = _c === void 0 ? 1000 : _c, _d = _stackTrace, _e = _b[_d], stackTrace = _e === void 0 ? captureStackTrace(Profiled.waitForNextRender) : _e;
+            var _b = _a === void 0 ? {} : _a, _c = _b.timeout, timeout = _c === void 0 ? 1000 : _c, 
+            // capture the stack trace here so its stack trace is as close to the calling code as possible
+            _d = _stackTrace, 
+            // capture the stack trace here so its stack trace is as close to the calling code as possible
+            _e = _b[_d], 
+            // capture the stack trace here so its stack trace is as close to the calling code as possible
+            stackTrace = _e === void 0 ? captureStackTrace(Profiler.waitForNextRender) : _e;
             if (!nextRender) {
                 nextRender = Promise.race([
                     new Promise(function (resolve, reject) {
@@ -153,9 +211,10 @@ export function profile(_a) {
             return nextRender;
         },
     });
-    return Profiled;
+    return Profiler;
 }
-var WaitForRenderTimeoutError = (function (_super) {
+/** @internal */
+var WaitForRenderTimeoutError = /** @class */ (function (_super) {
     __extends(WaitForRenderTimeoutError, _super);
     function WaitForRenderTimeoutError() {
         var _newTarget = this.constructor;
@@ -166,29 +225,27 @@ var WaitForRenderTimeoutError = (function (_super) {
     return WaitForRenderTimeoutError;
 }(Error));
 export { WaitForRenderTimeoutError };
+/** @internal */
 export function profileHook(renderCallback) {
-    var returnValue;
-    var Component = function (props) {
-        ProfiledComponent.updateSnapshot(renderCallback(props));
+    var Profiler = createProfiler();
+    var ProfiledHook = function (props) {
+        Profiler.replaceSnapshot(renderCallback(props));
         return null;
     };
-    var ProfiledComponent = profile({
-        Component: Component,
-        onRender: function () { return returnValue; },
-    });
-    return Object.assign(function ProfiledHook(props) {
-        return React.createElement(ProfiledComponent, __assign({}, props));
+    return Object.assign(function App(props) {
+        return (React.createElement(Profiler, null,
+            React.createElement(ProfiledHook, __assign({}, props))));
     }, {
-        ProfiledComponent: ProfiledComponent,
+        Profiler: Profiler,
     }, {
-        renders: ProfiledComponent.renders,
-        currentSnapshotCount: ProfiledComponent.currentRenderCount,
+        renders: Profiler.renders,
+        totalSnapshotCount: Profiler.totalRenderCount,
         peekSnapshot: function (options) {
             return __awaiter(this, void 0, void 0, function () {
                 return __generator(this, function (_a) {
                     switch (_a.label) {
-                        case 0: return [4, ProfiledComponent.peekRender(options)];
-                        case 1: return [2, (_a.sent()).snapshot];
+                        case 0: return [4 /*yield*/, Profiler.peekRender(options)];
+                        case 1: return [2 /*return*/, (_a.sent()).snapshot];
                     }
                 });
             });
@@ -197,26 +254,43 @@ export function profileHook(renderCallback) {
             return __awaiter(this, void 0, void 0, function () {
                 return __generator(this, function (_a) {
                     switch (_a.label) {
-                        case 0: return [4, ProfiledComponent.takeRender(options)];
-                        case 1: return [2, (_a.sent()).snapshot];
+                        case 0: return [4 /*yield*/, Profiler.takeRender(options)];
+                        case 1: return [2 /*return*/, (_a.sent()).snapshot];
                     }
                 });
             });
         },
         getCurrentSnapshot: function () {
-            return ProfiledComponent.getCurrentRender().snapshot;
+            return Profiler.getCurrentRender().snapshot;
         },
-        takeUntilSnapshotCount: ProfiledComponent.takeUntilRenderCount,
         waitForNextSnapshot: function (options) {
             return __awaiter(this, void 0, void 0, function () {
                 return __generator(this, function (_a) {
                     switch (_a.label) {
-                        case 0: return [4, ProfiledComponent.waitForNextRender(options)];
-                        case 1: return [2, (_a.sent()).snapshot];
+                        case 0: return [4 /*yield*/, Profiler.waitForNextRender(options)];
+                        case 1: return [2 /*return*/, (_a.sent()).snapshot];
                     }
                 });
             });
         },
     });
 }
+function resolveHookOwner() {
+    var _a, _b, _c;
+    return (_c = (_b = (_a = React.__SECRET_INTERNALS_DO_NOT_USE_OR_YOU_WILL_BE_FIRED) === null || _a === void 0 ? void 0 : _a.ReactCurrentOwner) === null || _b === void 0 ? void 0 : _b.current) === null || _c === void 0 ? void 0 : _c.elementType;
+}
+export function useTrackRenders(_a) {
+    var _b = _a === void 0 ? {} : _a, name = _b.name;
+    var component = name || resolveHookOwner();
+    if (!component) {
+        throw new Error("useTrackRender: Unable to determine component. Please ensure the hook is called inside a rendered component or provide a `name` option.");
+    }
+    var ctx = useProfilerContext();
+    if (!ctx) {
+        throw new Error("useTrackComponentRender: A Profiler must be created and rendered to track component renders");
+    }
+    React.useLayoutEffect(function () {
+        ctx.renderedComponents.unshift(component);
+    });
+}
 //# sourceMappingURL=profile.js.map

package/testing/internal/profile/profile.js.map

@@ -1 +1 @@
-{"version":3,"file":"profile.js","sourceRoot":"","sources":["../../../../src/testing/internal/profile/profile.tsx"],"names":[],"mappings":";;AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAE/B,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,MAAM,CAAC;AAEhD,MAAA,MAAM,CAAC,WAAW,oCAAlB,MAAM,CAAC,WAAW,GAAK,WAAW,EAAC;AAEnC,MAAA,MAAM,CAAC,WAAW,oCAAlB,MAAM,CAAC,WAAW,GAAK,WAAW,EAAC;AAEnC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAKjE,IAAM,WAAW,GAAG,MAAM,EAAE,CAAC;AAiE7B,MAAM,UAAU,OAAO,CAGrB,EAeD;QAdC,SAAS,eAAA,EACT,QAAQ,cAAA,EACR,mBAAmB,EAAnB,WAAW,mBAAG,KAAK,KAAA,EACnB,eAAe,qBAAA;IAYf,IAAI,aAA2C,CAAC;IAChD,IAAI,UAAiD,CAAC;IACtD,IAAI,iBAAmE,CAAC;IACxE,IAAI,gBAAwD,CAAC;IAC7D,IAAM,WAAW,GAAG,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;IACjD,IAAM,cAAc,GAA6B,UAAC,IAAI;QACpD,IAAI,OAAO,IAAI,KAAK,UAAU,EAAE;YAC9B,IAAI,CAAC,eAAe,EAAE;gBACpB,MAAM,IAAI,KAAK,CACb,mFAAmF,CACpF,CAAC;aACH;YACD,WAAW,CAAC,OAAO,GAAG,IAAI,CACxB,OAAO,WAAW,CAAC,OAAO,KAAK,QAAQ;gBACrC,CAAC,cACM,WAAW,CAAC,OAAQ,EAC3B,CAAC,CAAC,WAAW,CAAC,OAAQ,CACzB,CAAC;SACH;aAAM;YACL,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;SAC5B;IACH,CAAC,CAAC;IACF,IAAM,gBAAgB,GAAmC,UACvD,EAAE,EACF,KAAK,EACL,cAAc,EACd,YAAY,EACZ,SAAS,EACT,UAAU;QAEV,IAAM,UAAU,GAAG;YACjB,EAAE,IAAA;YACF,KAAK,OAAA;YACL,cAAc,gBAAA;YACd,YAAY,cAAA;YACZ,SAAS,WAAA;YACT,UAAU,YAAA;YACV,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;SACnC,CAAC;QACF,IAAI;YAQF,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,uBACH,UAAU,KACb,cAAc,gBAAA,EACd,QAAQ,EAAE,WAAW,CAAC,OAAQ,IAC9B,CAAC;YAEH,IAAM,QAAQ,GAAG,WAAW,CAAC,OAAmB,CAAC;YACjD,IAAM,WAAW,GAAG,WAAW;gBAC7B,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS;gBAChC,CAAC,CAAC,SAAS,CAAC;YACd,IAAM,MAAM,GAAG,IAAI,cAAc,CAAC,UAAU,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;YAErE,aAAa,GAAG,MAAM,CAAC;YACvB,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC9B,iBAAiB,aAAjB,iBAAiB,uBAAjB,iBAAiB,CAAG,MAAM,CAAC,CAAC;SAC7B;QAAC,OAAO,KAAK,EAAE;YACd,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC;gBACpB,KAAK,EAAE,eAAe;gBACtB,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM;gBAC9B,KAAK,OAAA;aACN,CAAC,CAAC;YACH,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAG,KAAK,CAAC,CAAC;SAC3B;gBAAS;YACR,UAAU,GAAG,iBAAiB,GAAG,gBAAgB,GAAG,SAAS,CAAC;SAC/D;IACH,CAAC,CAAC;IAEF,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,IAAM,QAAQ,GAAuC,MAAM,CAAC,MAAM,CAChE,UAAC,KAAY,IAAK,OAAA,CAChB,oBAAC,KAAK,CAAC,QAAQ,IAAC,EAAE,EAAC,MAAM,EAAC,QAAQ,EAAE,gBAAgB;QAClD,oBAAC,SAAS,eAAM,KAAa,EAAI,CAClB,CAClB,EAJiB,CAIjB,EACD;QACE,cAAc,gBAAA;KACuC,EACvD;QACE,OAAO,EAAE,IAAI,KAAK,EAGf;QACH,kBAAkB;YAChB,OAAO,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC;QACjC,CAAC;QACK,UAAU,YAAC,OAA+B;YAA/B,wBAAA,EAAA,YAA+B;;;;;oBAC9C,IAAI,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE;wBACxC,WAAS,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;wBAClD,IAAI,QAAM,CAAC,KAAK,KAAK,eAAe,EAAE;4BACpC,MAAM,QAAM,CAAC,KAAK,CAAC;yBACpB;wBACD,WAAO,QAAM,EAAC;qBACf;oBACK,MAAM,GAAG,QAAQ,CAAC,iBAAiB,uBACtC,WAAW,IAAG,iBAAiB,CAAC,QAAQ,CAAC,UAAU,CAAC,OAClD,OAAO,EACV,CAAC;oBACH,WAAO,MAAM,EAAC;;;SACf;QACK,UAAU,YAAC,OAA+B;YAA/B,wBAAA,EAAA,YAA+B;;;;;;;4BAC1C,KAAK,GAAqC,SAAS,CAAC;;;;4BAE/C,WAAM,QAAQ,CAAC,UAAU,uBAC7B,WAAW,IAAG,iBAAiB,CAAC,QAAQ,CAAC,UAAU,CAAC,OAClD,OAAO,EACV,EAAA;gCAHF,WAAO,SAGL,EAAC;;;4BAEH,KAAK,GAAG,GAAC,CAAC;4BACV,MAAM,GAAC,CAAC;;4BAER,IAAI,CAAC,CAAC,KAAK,IAAI,KAAK,YAAY,yBAAyB,CAAC,EAAE;gCAC1D,gBAAgB,EAAE,CAAC;6BACpB;;;;;;SAEJ;QACD,gBAAgB;YACd,IAAI,CAAC,aAAa,EAAE;gBAClB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;aAC/C;YACD,OAAO,aAAa,CAAC;QACvB,CAAC;QACK,oBAAoB,YACxB,KAAa,EACb,gBAAoC;;;;;iCAE7B,CAAA,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,KAAK,CAAA;4BACpC,WAAM,QAAQ,CAAC,UAAU,CAAC,gBAAgB,CAAC,EAAA;;4BAA3C,SAA2C,CAAC;;;;;;SAE/C;QACD,iBAAiB,YAAC,EAMO;gBANP,qBAMK,EAAE,KAAA,EALvB,eAAc,EAAd,OAAO,mBAAG,IAAI,KAAA,EAEd,KAAC,WAAY,EAAb,WAEC,EAFc,UAAU,mBAAG,iBAAiB,CAC3C,QAAQ,CAAC,iBAAiB,CAC3B,KAAA;YAED,IAAI,CAAC,UAAU,EAAE;gBACf,UAAU,GAAG,OAAO,CAAC,IAAI,CAAmB;oBAC1C,IAAI,OAAO,CAAmB,UAAC,OAAO,EAAE,MAAM;wBAC5C,iBAAiB,GAAG,OAAO,CAAC;wBAC5B,gBAAgB,GAAG,MAAM,CAAC;oBAC5B,CAAC,CAAC;oBACF,IAAI,OAAO,CAAmB,UAAC,CAAC,EAAE,MAAM;wBACtC,OAAA,UAAU,CACR;4BACE,OAAA,MAAM,CACJ,eAAe,CAAC,IAAI,yBAAyB,EAAE,EAAE,UAAU,CAAC,CAC7D;wBAFD,CAEC,EACH,OAAO,CACR;oBAND,CAMC,CACF;iBACF,CAAC,CAAC;aACJ;YACD,OAAO,UAAU,CAAC;QACpB,CAAC;KACiD,CACrD,CAAC;IACF,OAAO,QAAQ,CAAC;AAClB,CAAC;AAGD;IAA+C,6CAAK;IAClD;;QAAA,YACE,kBAAM,2CAA2C,CAAC,SAEnD;QADC,MAAM,CAAC,cAAc,CAAC,KAAI,EAAE,WAAW,SAAS,CAAC,CAAC;;IACpD,CAAC;IACH,gCAAC;AAAD,CAAC,AALD,CAA+C,KAAK,GAKnD;;AAgCD,MAAM,UAAU,WAAW,CACzB,cAA6C;IAE7C,IAAI,WAAwB,CAAC;IAC7B,IAAM,SAAS,GAAG,UAAC,KAAY;QAC7B,iBAAiB,CAAC,cAAc,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;QACxD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IACF,IAAM,iBAAiB,GAAG,OAAO,CAAqB;QACpD,SAAS,WAAA;QACT,QAAQ,EAAE,cAAM,OAAA,WAAW,EAAX,CAAW;KAC5B,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,MAAM,CAClB,SAAS,YAAY,CAAC,KAAY;QAChC,OAAO,oBAAC,iBAAiB,eAAM,KAAa,EAAI,CAAC;IACnD,CAAC,EACD;QACE,iBAAiB,mBAAA;KAClB,EACD;QACE,OAAO,EAAE,iBAAiB,CAAC,OAAO;QAClC,oBAAoB,EAAE,iBAAiB,CAAC,kBAAkB;QACpD,YAAY,YAAC,OAAO;;;;gCAChB,WAAM,iBAAiB,CAAC,UAAU,CAAC,OAAO,CAAC,EAAA;gCAAnD,WAAO,CAAC,SAA2C,CAAC,CAAC,QAAQ,EAAC;;;;SAC/D;QACK,YAAY,YAAC,OAAO;;;;gCAChB,WAAM,iBAAiB,CAAC,UAAU,CAAC,OAAO,CAAC,EAAA;gCAAnD,WAAO,CAAC,SAA2C,CAAC,CAAC,QAAQ,EAAC;;;;SAC/D;QACD,kBAAkB;YAChB,OAAO,iBAAiB,CAAC,gBAAgB,EAAE,CAAC,QAAQ,CAAC;QACvD,CAAC;QACD,sBAAsB,EAAE,iBAAiB,CAAC,oBAAoB;QACxD,mBAAmB,YAAC,OAAO;;;;gCACvB,WAAM,iBAAiB,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAA;gCAA1D,WAAO,CAAC,SAAkD,CAAC,CAAC,QAAQ,EAAC;;;;SACtE;KAC+C,CACnD,CAAC;AACJ,CAAC","sourcesContent":["import * as React from \"react\";\n\nimport { TextEncoder, TextDecoder } from \"util\";\n\nglobal.TextEncoder ??= TextEncoder;\n// @ts-ignore\nglobal.TextDecoder ??= TextDecoder;\nimport type { Render, BaseRender } from \"./Render.js\";\nimport { RenderInstance } from \"./Render.js\";\nimport { applyStackTrace, captureStackTrace } from \"./traces.js\";\n\ntype ValidSnapshot = void | (object & { /* not a function */ call?: never });\n\n/** only used for passing around data internally */\nconst _stackTrace = Symbol();\n/** @internal */\nexport interface NextRenderOptions {\n  timeout?: number;\n  [_stackTrace]?: string;\n}\n\n/** @internal */\nexport interface ProfiledComponent<Props, Snapshot>\n  extends React.FC<Props>,\n    ProfiledComponentFields<Props, Snapshot>,\n    ProfiledComponenOnlyFields<Props, Snapshot> {}\n\ninterface UpdateSnapshot<Snapshot> {\n  (newSnapshot: Snapshot): void;\n  (updateSnapshot: (lastSnapshot: Readonly<Snapshot>) => Snapshot): void;\n}\n\ninterface ProfiledComponenOnlyFields<Props, Snapshot> {\n  updateSnapshot: UpdateSnapshot<Snapshot>;\n}\ninterface ProfiledComponentFields<Props, Snapshot> {\n  /**\n   * An array of all renders that have happened so far.\n   * Errors thrown during component render will be captured here, too.\n   */\n  renders: Array<\n    Render<Snapshot> | { phase: \"snapshotError\"; count: number; error: unknown }\n  >;\n  /**\n   * Peeks the next render from the current iterator position, without advancing the iterator.\n   * If no render has happened yet, it will wait for the next render to happen.\n   * @throws {WaitForRenderTimeoutError} if no render happens within the timeout\n   */\n  peekRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;\n  /**\n   * Iterates to the next render and returns it.\n   * If no render has happened yet, it will wait for the next render to happen.\n   * @throws {WaitForRenderTimeoutError} if no render happens within the timeout\n   */\n  takeRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;\n  /**\n   * Returns the current render count.\n   */\n  currentRenderCount(): number;\n  /**\n   * Returns the current render.\n   * @throws {Error} if no render has happened yet\n   */\n  getCurrentRender(): Render<Snapshot>;\n  /**\n   * Iterates the renders until the render count is reached.\n   */\n  takeUntilRenderCount(\n    count: number,\n    optionsPerRender?: NextRenderOptions\n  ): Promise<void>;\n  /**\n   * Waits for the next render to happen.\n   * Does not advance the render iterator.\n   */\n  waitForNextRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;\n}\n\n/** @internal */\nexport function profile<\n  Snapshot extends ValidSnapshot = void,\n  Props = Record<string, never>,\n>({\n  Component,\n  onRender,\n  snapshotDOM = false,\n  initialSnapshot,\n}: {\n  Component: React.ComponentType<Props>;\n  onRender?: (\n    info: BaseRender & {\n      snapshot: Snapshot;\n      updateSnapshot: UpdateSnapshot<Snapshot>;\n    }\n  ) => void;\n  snapshotDOM?: boolean;\n  initialSnapshot?: Snapshot;\n}) {\n  let currentRender: Render<Snapshot> | undefined;\n  let nextRender: Promise<Render<Snapshot>> | undefined;\n  let resolveNextRender: ((render: Render<Snapshot>) => void) | undefined;\n  let rejectNextRender: ((error: unknown) => void) | undefined;\n  const snapshotRef = { current: initialSnapshot };\n  const updateSnapshot: UpdateSnapshot<Snapshot> = (snap) => {\n    if (typeof snap === \"function\") {\n      if (!initialSnapshot) {\n        throw new Error(\n          \"Cannot use a function to update the snapshot if no initial snapshot was provided.\"\n        );\n      }\n      snapshotRef.current = snap(\n        typeof snapshotRef.current === \"object\"\n          ? // \"cheap best effort\" to prevent accidental mutation of the last snapshot\n            { ...snapshotRef.current! }\n          : snapshotRef.current!\n      );\n    } else {\n      snapshotRef.current = snap;\n    }\n  };\n  const profilerOnRender: React.ProfilerOnRenderCallback = (\n    id,\n    phase,\n    actualDuration,\n    baseDuration,\n    startTime,\n    commitTime\n  ) => {\n    const baseRender = {\n      id,\n      phase,\n      actualDuration,\n      baseDuration,\n      startTime,\n      commitTime,\n      count: Profiled.renders.length + 1,\n    };\n    try {\n      /*\n       * The `onRender` function could contain `expect` calls that throw\n       * `JestAssertionError`s - but we are still inside of React, where errors\n       * might be swallowed.\n       * So we record them and re-throw them in `takeRender`\n       * Additionally, we reject the `waitForNextRender` promise.\n       */\n      onRender?.({\n        ...baseRender,\n        updateSnapshot,\n        snapshot: snapshotRef.current!,\n      });\n\n      const snapshot = snapshotRef.current as Snapshot;\n      const domSnapshot = snapshotDOM\n        ? window.document.body.innerHTML\n        : undefined;\n      const render = new RenderInstance(baseRender, snapshot, domSnapshot);\n      // eslint-disable-next-line testing-library/render-result-naming-convention\n      currentRender = render;\n      Profiled.renders.push(render);\n      resolveNextRender?.(render);\n    } catch (error) {\n      Profiled.renders.push({\n        phase: \"snapshotError\",\n        count: Profiled.renders.length,\n        error,\n      });\n      rejectNextRender?.(error);\n    } finally {\n      nextRender = resolveNextRender = rejectNextRender = undefined;\n    }\n  };\n\n  let iteratorPosition = 0;\n  const Profiled: ProfiledComponent<Props, Snapshot> = Object.assign(\n    (props: Props) => (\n      <React.Profiler id=\"test\" onRender={profilerOnRender}>\n        <Component {...(props as any)} />\n      </React.Profiler>\n    ),\n    {\n      updateSnapshot,\n    } satisfies ProfiledComponenOnlyFields<Props, Snapshot>,\n    {\n      renders: new Array<\n        | Render<Snapshot>\n        | { phase: \"snapshotError\"; count: number; error: unknown }\n      >(),\n      currentRenderCount() {\n        return Profiled.renders.length;\n      },\n      async peekRender(options: NextRenderOptions = {}) {\n        if (iteratorPosition < Profiled.renders.length) {\n          const render = Profiled.renders[iteratorPosition];\n          if (render.phase === \"snapshotError\") {\n            throw render.error;\n          }\n          return render;\n        }\n        const render = Profiled.waitForNextRender({\n          [_stackTrace]: captureStackTrace(Profiled.peekRender),\n          ...options,\n        });\n        return render;\n      },\n      async takeRender(options: NextRenderOptions = {}) {\n        let error: { message?: string } | undefined = undefined;\n        try {\n          return await Profiled.peekRender({\n            [_stackTrace]: captureStackTrace(Profiled.takeRender),\n            ...options,\n          });\n        } catch (e) {\n          error = e;\n          throw e;\n        } finally {\n          if (!(error && error instanceof WaitForRenderTimeoutError)) {\n            iteratorPosition++;\n          }\n        }\n      },\n      getCurrentRender() {\n        if (!currentRender) {\n          throw new Error(\"Has not been rendered yet!\");\n        }\n        return currentRender;\n      },\n      async takeUntilRenderCount(\n        count: number,\n        optionsPerRender?: NextRenderOptions\n      ) {\n        while (Profiled.renders.length < count) {\n          await Profiled.takeRender(optionsPerRender);\n        }\n      },\n      waitForNextRender({\n        timeout = 1000,\n        // capture the stack trace here so its stack trace is as close to the calling code as possible\n        [_stackTrace]: stackTrace = captureStackTrace(\n          Profiled.waitForNextRender\n        ),\n      }: NextRenderOptions = {}) {\n        if (!nextRender) {\n          nextRender = Promise.race<Render<Snapshot>>([\n            new Promise<Render<Snapshot>>((resolve, reject) => {\n              resolveNextRender = resolve;\n              rejectNextRender = reject;\n            }),\n            new Promise<Render<Snapshot>>((_, reject) =>\n              setTimeout(\n                () =>\n                  reject(\n                    applyStackTrace(new WaitForRenderTimeoutError(), stackTrace)\n                  ),\n                timeout\n              )\n            ),\n          ]);\n        }\n        return nextRender;\n      },\n    } satisfies ProfiledComponentFields<Props, Snapshot>\n  );\n  return Profiled;\n}\n\n/** @internal */\nexport class WaitForRenderTimeoutError extends Error {\n  constructor() {\n    super(\"Exceeded timeout waiting for next render.\");\n    Object.setPrototypeOf(this, new.target.prototype);\n  }\n}\n\ntype StringReplaceRenderWithSnapshot<T extends string> =\n  T extends `${infer Pre}Render${infer Post}` ? `${Pre}Snapshot${Post}` : T;\n\ntype ResultReplaceRenderWithSnapshot<T> = T extends (\n  ...args: infer Args\n) => Render<infer Snapshot>\n  ? (...args: Args) => Snapshot\n  : T extends (...args: infer Args) => Promise<Render<infer Snapshot>>\n  ? (...args: Args) => Promise<Snapshot>\n  : T;\n\ntype ProfiledHookFields<Props, ReturnValue> = ProfiledComponentFields<\n  Props,\n  ReturnValue\n> extends infer PC\n  ? {\n      [K in keyof PC as StringReplaceRenderWithSnapshot<\n        K & string\n      >]: ResultReplaceRenderWithSnapshot<PC[K]>;\n    }\n  : never;\n\n/** @internal */\nexport interface ProfiledHook<Props, ReturnValue>\n  extends React.FC<Props>,\n    ProfiledHookFields<Props, ReturnValue> {\n  ProfiledComponent: ProfiledComponent<Props, ReturnValue>;\n}\n\n/** @internal */\nexport function profileHook<ReturnValue extends ValidSnapshot, Props>(\n  renderCallback: (props: Props) => ReturnValue\n): ProfiledHook<Props, ReturnValue> {\n  let returnValue: ReturnValue;\n  const Component = (props: Props) => {\n    ProfiledComponent.updateSnapshot(renderCallback(props));\n    return null;\n  };\n  const ProfiledComponent = profile<ReturnValue, Props>({\n    Component,\n    onRender: () => returnValue,\n  });\n  return Object.assign(\n    function ProfiledHook(props: Props) {\n      return <ProfiledComponent {...(props as any)} />;\n    },\n    {\n      ProfiledComponent,\n    },\n    {\n      renders: ProfiledComponent.renders,\n      currentSnapshotCount: ProfiledComponent.currentRenderCount,\n      async peekSnapshot(options) {\n        return (await ProfiledComponent.peekRender(options)).snapshot;\n      },\n      async takeSnapshot(options) {\n        return (await ProfiledComponent.takeRender(options)).snapshot;\n      },\n      getCurrentSnapshot() {\n        return ProfiledComponent.getCurrentRender().snapshot;\n      },\n      takeUntilSnapshotCount: ProfiledComponent.takeUntilRenderCount,\n      async waitForNextSnapshot(options) {\n        return (await ProfiledComponent.waitForNextRender(options)).snapshot;\n      },\n    } satisfies ProfiledHookFields<Props, ReturnValue>\n  );\n}\n"]}
+{"version":3,"file":"profile.js","sourceRoot":"","sources":["../../../../src/testing/internal/profile/profile.tsx"],"names":[],"mappings":";;AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAE/B,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,MAAM,CAAC;AAEhD,MAAA,MAAM,CAAC,WAAW,oCAAlB,MAAM,CAAC,WAAW,GAAK,WAAW,EAAC;AACnC,aAAa;AACb,MAAA,MAAM,CAAC,WAAW,oCAAlB,MAAM,CAAC,WAAW,GAAK,WAAW,EAAC;AAEnC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAEjE,OAAO,EAAE,uBAAuB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAC3E,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAI7D,mDAAmD;AACnD,IAAM,WAAW,GAAG,MAAM,EAAE,CAAC;AA+E7B,gBAAgB;AAChB,MAAM,UAAU,OAAO,CAAoD,EAK1E;IAJC,IAAA,SAAS,eAAA,EACN,OAAO,cAF+D,aAG1E,CADW;IAIV,IAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAEzC,OAAO,MAAM,CAAC,MAAM,CAClB,SAAS,iBAAiB,CAAC,KAAY;QACrC,OAAO,CACL,oBAAC,QAAQ;YACP,oBAAC,SAAS,eAAM,KAAa,EAAI,CACxB,CACZ,CAAC;IACJ,CAAC,EACD;QACE,aAAa,EAAE,QAAQ,CAAC,aAAa;QACrC,eAAe,EAAE,QAAQ,CAAC,eAAe;QACzC,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;QAC3C,UAAU,EAAE,QAAQ,CAAC,UAAU;QAC/B,UAAU,EAAE,QAAQ,CAAC,UAAU;QAC/B,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;QAC3C,iBAAiB,EAAE,QAAQ,CAAC,iBAAiB;QAC7C,IAAI,OAAO;YACT,OAAO,QAAQ,CAAC,OAAO,CAAC;QAC1B,CAAC;KACF,CACF,CAAC;AACJ,CAAC;AAED,gBAAgB;AAChB,MAAM,UAAU,cAAc,CAAwC,EAoBhE;QApBgE,qBAoBlE,EAAE,KAAA,EAnBJ,QAAQ,cAAA,EACR,mBAAmB,EAAnB,WAAW,mBAAG,KAAK,KAAA,EACnB,eAAe,qBAAA,EACf,sBAAsB,4BAAA;IAiBtB,IAAI,UAAiD,CAAC;IACtD,IAAI,iBAAmE,CAAC;IACxE,IAAI,gBAAwD,CAAC;IAC7D,IAAM,WAAW,GAAG,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;IACjD,IAAM,eAAe,GAA8B,UAAC,IAAI;QACtD,IAAI,OAAO,IAAI,KAAK,UAAU,EAAE,CAAC;YAC/B,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CACb,mFAAmF,CACpF,CAAC;YACJ,CAAC;YACD,WAAW,CAAC,OAAO,GAAG,IAAI,CACxB,OAAO,WAAW,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,cAElC,WAAW,CAAC,OAAQ,EAC3B,CAAC,CAAC,WAAW,CAAC,OAAQ,CACvB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC,CAAC;IAEF,IAAM,aAAa,GAA4B,UAAC,eAAe;QAC7D,eAAe,CAAC,UAAC,QAAQ,IAAK,OAAA,uBACzB,QAAQ,GACR,CAAC,OAAO,eAAe,KAAK,UAAU,CAAC,CAAC;YACzC,eAAe,CAAC,QAAQ,CAAC;YAC3B,CAAC,CAAC,eAAe,CAAC,EAClB,EAL4B,CAK5B,CAAC,CAAC;IACN,CAAC,CAAC;IAEF,IAAM,eAAe,GAAyB;QAC5C,kBAAkB,EAAE,EAAE;KACvB,CAAC;IAEF,IAAM,gBAAgB,GAAmC,UACvD,EAAE,EACF,KAAK,EACL,cAAc,EACd,YAAY,EACZ,SAAS,EACT,UAAU;QAEV,IACE,sBAAsB;YACtB,eAAe,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAC/C,CAAC;YACD,OAAO;QACT,CAAC;QACD,IAAM,UAAU,GAAG;YACjB,EAAE,IAAA;YACF,KAAK,OAAA;YACL,cAAc,gBAAA;YACd,YAAY,cAAA;YACZ,SAAS,WAAA;YACT,UAAU,YAAA;YACV,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;SACnC,CAAC;QACF,IAAI,CAAC;YACH;;;;;;eAMG;YACH,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,uBACH,UAAU,KACb,eAAe,iBAAA,EACf,aAAa,eAAA,EACb,QAAQ,EAAE,WAAW,CAAC,OAAQ,IAC9B,CAAC;YAEH,IAAM,QAAQ,GAAG,WAAW,CAAC,OAAmB,CAAC;YACjD,IAAM,WAAW,GACf,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;YAC3D,IAAM,MAAM,GAAG,IAAI,cAAc,CAC/B,UAAU,EACV,QAAQ,EACR,WAAW,EACX,eAAe,CAAC,kBAAkB,CACnC,CAAC;YACF,eAAe,CAAC,kBAAkB,GAAG,EAAE,CAAC;YACxC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC9B,iBAAiB,aAAjB,iBAAiB,uBAAjB,iBAAiB,CAAG,MAAM,CAAC,CAAC;QAC9B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC;gBACpB,KAAK,EAAE,eAAe;gBACtB,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM;gBAC9B,KAAK,OAAA;aACN,CAAC,CAAC;YACH,gBAAgB,aAAhB,gBAAgB,uBAAhB,gBAAgB,CAAG,KAAK,CAAC,CAAC;QAC5B,CAAC;gBAAS,CAAC;YACT,UAAU,GAAG,iBAAiB,GAAG,gBAAgB,GAAG,SAAS,CAAC;QAChE,CAAC;IACH,CAAC,CAAC;IAEF,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,IAAM,QAAQ,GAAuB,MAAM,CAAC,MAAM,CAChD,UAAC,EAA2B;YAAzB,QAAQ,cAAA;QACT,OAAO,CACL,oBAAC,uBAAuB,IAAC,KAAK,EAAE,eAAe;YAC7C,oBAAC,KAAK,CAAC,QAAQ,IAAC,EAAE,EAAC,MAAM,EAAC,QAAQ,EAAE,gBAAgB,IACjD,QAAQ,CACM,CACO,CAC3B,CAAC;IACJ,CAAC,EACD;QACE,eAAe,iBAAA;QACf,aAAa,eAAA;KACkC,EACjD;QACE,OAAO,EAAE,IAAI,KAAK,EAGf;QACH,gBAAgB;YACd,OAAO,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC;QACjC,CAAC;QACK,UAAU,YAAC,OAA+B;YAA/B,wBAAA,EAAA,YAA+B;;;;;oBAC9C,IAAI,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;wBACzC,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;wBAElD,IAAI,MAAM,CAAC,KAAK,KAAK,eAAe,EAAE,CAAC;4BACrC,MAAM,MAAM,CAAC,KAAK,CAAC;wBACrB,CAAC;wBAED,sBAAO,MAAM,EAAC;oBAChB,CAAC;oBACD,sBAAO,QAAQ,CAAC,iBAAiB,uBAC9B,WAAW,IAAG,iBAAiB,CAAC,QAAQ,CAAC,UAAU,CAAC,OAClD,OAAO,EACV,EAAC;;;SACJ;QACK,UAAU,YAAC,OAA+B;YAA/B,wBAAA,EAAA,YAA+B;;;;;;;;;;;4BAIxC,oBAAoB,kCAAG,kBAAkB,EAAE,QAAA,CAAC;4BAE9C,KAAK,GAAY,SAAS,CAAC;;;;4BAGtB,qBAAM,QAAQ,CAAC,UAAU,uBAC7B,WAAW,IAAG,iBAAiB,CAAC,QAAQ,CAAC,UAAU,CAAC,OAClD,OAAO,EACV,EAAA;gCAHF,sBAAO,SAGL,EAAC;;;4BAEH,KAAK,GAAG,GAAC,CAAC;4BACV,MAAM,GAAC,CAAC;;4BAER,IAAI,CAAC,CAAC,KAAK,IAAI,KAAK,YAAY,yBAAyB,CAAC,EAAE,CAAC;gCAC3D,gBAAgB,EAAE,CAAC;4BACrB,CAAC;;;;;;;;;;;;;;;SAEJ;QACD,gBAAgB;YACd,qEAAqE;YACrE,sEAAsE;YACtE,kEAAkE;YAClE,uEAAuE;YACvE,2CAA2C;YAC3C,IAAM,eAAe,GAAG,gBAAgB,GAAG,CAAC,CAAC;YAE7C,IAAI,eAAe,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CACb,mGAAmG,CACpG,CAAC;YACJ,CAAC;YAED,IAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;YAEjD,IAAI,MAAM,CAAC,KAAK,KAAK,eAAe,EAAE,CAAC;gBACrC,MAAM,MAAM,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,iBAAiB,YAAC,EAMO;gBANP,qBAMK,EAAE,KAAA,EALvB,eAAc,EAAd,OAAO,mBAAG,IAAI,KAAA;YACd,8FAA8F;YAC9F,KAAC,WAAY;YADb,8FAA8F;YAC9F,WAEC;YAHD,8FAA8F;YAC/E,UAAU,mBAAG,iBAAiB,CAC3C,QAAQ,CAAC,iBAAiB,CAC3B,KAAA;YAED,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,UAAU,GAAG,OAAO,CAAC,IAAI,CAAmB;oBAC1C,IAAI,OAAO,CAAmB,UAAC,OAAO,EAAE,MAAM;wBAC5C,iBAAiB,GAAG,OAAO,CAAC;wBAC5B,gBAAgB,GAAG,MAAM,CAAC;oBAC5B,CAAC,CAAC;oBACF,IAAI,OAAO,CAAmB,UAAC,CAAC,EAAE,MAAM;wBACtC,OAAA,UAAU,CACR;4BACE,OAAA,MAAM,CACJ,eAAe,CAAC,IAAI,yBAAyB,EAAE,EAAE,UAAU,CAAC,CAC7D;wBAFD,CAEC,EACH,OAAO,CACR;oBAND,CAMC,CACF;iBACF,CAAC,CAAC;YACL,CAAC;YACD,OAAO,UAAU,CAAC;QACpB,CAAC;KAC0C,CAC9C,CAAC;IACF,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,gBAAgB;AAChB;IAA+C,6CAAK;IAClD;;QACE,YAAA,MAAK,YAAC,2CAA2C,CAAC,SAAC;QACnD,MAAM,CAAC,cAAc,CAAC,KAAI,EAAE,WAAW,SAAS,CAAC,CAAC;;IACpD,CAAC;IACH,gCAAC;AAAD,CAAC,AALD,CAA+C,KAAK,GAKnD;;AA4BD,gBAAgB;AAChB,MAAM,UAAU,WAAW,CACzB,cAA6C;IAE7C,IAAM,QAAQ,GAAG,cAAc,EAAe,CAAC;IAE/C,IAAM,YAAY,GAAG,UAAC,KAAY;QAChC,QAAQ,CAAC,eAAe,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,OAAO,MAAM,CAAC,MAAM,CAClB,SAAS,GAAG,CAAC,KAAY;QACvB,OAAO,CACL,oBAAC,QAAQ;YACP,oBAAC,YAAY,eAAM,KAAa,EAAI,CAC3B,CACZ,CAAC;IACJ,CAAC,EACD;QACE,QAAQ,UAAA;KACT,EACD;QACE,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,kBAAkB,EAAE,QAAQ,CAAC,gBAAgB;QACvC,YAAY,YAAC,OAAO;;;;gCAChB,qBAAM,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAA;gCAA1C,sBAAO,CAAC,SAAkC,CAAC,CAAC,QAAQ,EAAC;;;;SACtD;QACK,YAAY,YAAC,OAAO;;;;gCAChB,qBAAM,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAA;gCAA1C,sBAAO,CAAC,SAAkC,CAAC,CAAC,QAAQ,EAAC;;;;SACtD;QACD,kBAAkB;YAChB,OAAO,QAAQ,CAAC,gBAAgB,EAAE,CAAC,QAAQ,CAAC;QAC9C,CAAC;QACK,mBAAmB,YAAC,OAAO;;;;gCACvB,qBAAM,QAAQ,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAA;gCAAjD,sBAAO,CAAC,SAAyC,CAAC,CAAC,QAAQ,EAAC;;;;SAC7D;KACwC,CAC5C,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB;;IACvB,OAAO,MAAA,MAAA,MAAC,KAAa,CAAC,kDAAkD,0CACpE,iBAAiB,0CAAE,OAAO,0CAAE,WAAW,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,EAAgC;QAAhC,qBAA8B,EAAE,KAAA,EAA9B,IAAI,UAAA;IACpC,IAAM,SAAS,GAAG,IAAI,IAAI,gBAAgB,EAAE,CAAC;IAE7C,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,yIAAyI,CAC1I,CAAC;IACJ,CAAC;IAED,IAAM,GAAG,GAAG,kBAAkB,EAAE,CAAC;IAEjC,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,IAAI,KAAK,CACb,6FAA6F,CAC9F,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CAAC;QACpB,GAAG,CAAC,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["import * as React from \"react\";\n\nimport { TextEncoder, TextDecoder } from \"util\";\n\nglobal.TextEncoder ??= TextEncoder;\n// @ts-ignore\nglobal.TextDecoder ??= TextDecoder;\nimport type { Render, BaseRender } from \"./Render.js\";\nimport { RenderInstance } from \"./Render.js\";\nimport { applyStackTrace, captureStackTrace } from \"./traces.js\";\nimport type { ProfilerContextValue } from \"./context.js\";\nimport { ProfilerContextProvider, useProfilerContext } from \"./context.js\";\nimport { disableActWarnings } from \"../disposables/index.js\";\n\ntype ValidSnapshot = void | (object & { /* not a function */ call?: never });\n\n/** only used for passing around data internally */\nconst _stackTrace = Symbol();\n/** @internal */\nexport interface NextRenderOptions {\n  timeout?: number;\n  [_stackTrace]?: string;\n}\n\n/** @internal */\ninterface ProfilerProps {\n  children: React.ReactNode;\n}\n\n/** @internal */\nexport interface Profiler<Snapshot>\n  extends React.FC<ProfilerProps>,\n    ProfiledComponentFields<Snapshot>,\n    ProfiledComponentOnlyFields<Snapshot> {}\n\ninterface ReplaceSnapshot<Snapshot> {\n  (newSnapshot: Snapshot): void;\n  (updateSnapshot: (lastSnapshot: Readonly<Snapshot>) => Snapshot): void;\n}\n\ninterface MergeSnapshot<Snapshot> {\n  (partialSnapshot: Partial<Snapshot>): void;\n  (\n    updatePartialSnapshot: (\n      lastSnapshot: Readonly<Snapshot>\n    ) => Partial<Snapshot>\n  ): void;\n}\n\ninterface ProfiledComponentOnlyFields<Snapshot> {\n  // Allows for partial updating of the snapshot by shallow merging the results\n  mergeSnapshot: MergeSnapshot<Snapshot>;\n  // Performs a full replacement of the snapshot\n  replaceSnapshot: ReplaceSnapshot<Snapshot>;\n}\ninterface ProfiledComponentFields<Snapshot> {\n  /**\n   * An array of all renders that have happened so far.\n   * Errors thrown during component render will be captured here, too.\n   */\n  renders: Array<\n    Render<Snapshot> | { phase: \"snapshotError\"; count: number; error: unknown }\n  >;\n  /**\n   * Peeks the next render from the current iterator position, without advancing the iterator.\n   * If no render has happened yet, it will wait for the next render to happen.\n   * @throws {WaitForRenderTimeoutError} if no render happens within the timeout\n   */\n  peekRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;\n  /**\n   * Iterates to the next render and returns it.\n   * If no render has happened yet, it will wait for the next render to happen.\n   * @throws {WaitForRenderTimeoutError} if no render happens within the timeout\n   */\n  takeRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;\n  /**\n   * Returns the total number of renders.\n   */\n  totalRenderCount(): number;\n  /**\n   * Returns the current render.\n   * @throws {Error} if no render has happened yet\n   */\n  getCurrentRender(): Render<Snapshot>;\n  /**\n   * Waits for the next render to happen.\n   * Does not advance the render iterator.\n   */\n  waitForNextRender(options?: NextRenderOptions): Promise<Render<Snapshot>>;\n}\n\nexport interface ProfiledComponent<Snapshot extends ValidSnapshot, Props = {}>\n  extends React.FC<Props>,\n    ProfiledComponentFields<Snapshot>,\n    ProfiledComponentOnlyFields<Snapshot> {}\n\n/** @internal */\nexport function profile<Snapshot extends ValidSnapshot = void, Props = {}>({\n  Component,\n  ...options\n}: Parameters<typeof createProfiler<Snapshot>>[0] & {\n  Component: React.ComponentType<Props>;\n}): ProfiledComponent<Snapshot, Props> {\n  const Profiler = createProfiler(options);\n\n  return Object.assign(\n    function ProfiledComponent(props: Props) {\n      return (\n        <Profiler>\n          <Component {...(props as any)} />\n        </Profiler>\n      );\n    },\n    {\n      mergeSnapshot: Profiler.mergeSnapshot,\n      replaceSnapshot: Profiler.replaceSnapshot,\n      getCurrentRender: Profiler.getCurrentRender,\n      peekRender: Profiler.peekRender,\n      takeRender: Profiler.takeRender,\n      totalRenderCount: Profiler.totalRenderCount,\n      waitForNextRender: Profiler.waitForNextRender,\n      get renders() {\n        return Profiler.renders;\n      },\n    }\n  );\n}\n\n/** @internal */\nexport function createProfiler<Snapshot extends ValidSnapshot = void>({\n  onRender,\n  snapshotDOM = false,\n  initialSnapshot,\n  skipNonTrackingRenders,\n}: {\n  onRender?: (\n    info: BaseRender & {\n      snapshot: Snapshot;\n      replaceSnapshot: ReplaceSnapshot<Snapshot>;\n      mergeSnapshot: MergeSnapshot<Snapshot>;\n    }\n  ) => void;\n  snapshotDOM?: boolean;\n  initialSnapshot?: Snapshot;\n  /**\n   * This will skip renders during which no renders tracked by\n   * `useTrackRenders` occured.\n   */\n  skipNonTrackingRenders?: boolean;\n} = {}) {\n  let nextRender: Promise<Render<Snapshot>> | undefined;\n  let resolveNextRender: ((render: Render<Snapshot>) => void) | undefined;\n  let rejectNextRender: ((error: unknown) => void) | undefined;\n  const snapshotRef = { current: initialSnapshot };\n  const replaceSnapshot: ReplaceSnapshot<Snapshot> = (snap) => {\n    if (typeof snap === \"function\") {\n      if (!initialSnapshot) {\n        throw new Error(\n          \"Cannot use a function to update the snapshot if no initial snapshot was provided.\"\n        );\n      }\n      snapshotRef.current = snap(\n        typeof snapshotRef.current === \"object\" ?\n          // \"cheap best effort\" to prevent accidental mutation of the last snapshot\n          { ...snapshotRef.current! }\n        : snapshotRef.current!\n      );\n    } else {\n      snapshotRef.current = snap;\n    }\n  };\n\n  const mergeSnapshot: MergeSnapshot<Snapshot> = (partialSnapshot) => {\n    replaceSnapshot((snapshot) => ({\n      ...snapshot,\n      ...(typeof partialSnapshot === \"function\" ?\n        partialSnapshot(snapshot)\n      : partialSnapshot),\n    }));\n  };\n\n  const profilerContext: ProfilerContextValue = {\n    renderedComponents: [],\n  };\n\n  const profilerOnRender: React.ProfilerOnRenderCallback = (\n    id,\n    phase,\n    actualDuration,\n    baseDuration,\n    startTime,\n    commitTime\n  ) => {\n    if (\n      skipNonTrackingRenders &&\n      profilerContext.renderedComponents.length === 0\n    ) {\n      return;\n    }\n    const baseRender = {\n      id,\n      phase,\n      actualDuration,\n      baseDuration,\n      startTime,\n      commitTime,\n      count: Profiler.renders.length + 1,\n    };\n    try {\n      /*\n       * The `onRender` function could contain `expect` calls that throw\n       * `JestAssertionError`s - but we are still inside of React, where errors\n       * might be swallowed.\n       * So we record them and re-throw them in `takeRender`\n       * Additionally, we reject the `waitForNextRender` promise.\n       */\n      onRender?.({\n        ...baseRender,\n        replaceSnapshot,\n        mergeSnapshot,\n        snapshot: snapshotRef.current!,\n      });\n\n      const snapshot = snapshotRef.current as Snapshot;\n      const domSnapshot =\n        snapshotDOM ? window.document.body.innerHTML : undefined;\n      const render = new RenderInstance(\n        baseRender,\n        snapshot,\n        domSnapshot,\n        profilerContext.renderedComponents\n      );\n      profilerContext.renderedComponents = [];\n      Profiler.renders.push(render);\n      resolveNextRender?.(render);\n    } catch (error) {\n      Profiler.renders.push({\n        phase: \"snapshotError\",\n        count: Profiler.renders.length,\n        error,\n      });\n      rejectNextRender?.(error);\n    } finally {\n      nextRender = resolveNextRender = rejectNextRender = undefined;\n    }\n  };\n\n  let iteratorPosition = 0;\n  const Profiler: Profiler<Snapshot> = Object.assign(\n    ({ children }: ProfilerProps) => {\n      return (\n        <ProfilerContextProvider value={profilerContext}>\n          <React.Profiler id=\"test\" onRender={profilerOnRender}>\n            {children}\n          </React.Profiler>\n        </ProfilerContextProvider>\n      );\n    },\n    {\n      replaceSnapshot,\n      mergeSnapshot,\n    } satisfies ProfiledComponentOnlyFields<Snapshot>,\n    {\n      renders: new Array<\n        | Render<Snapshot>\n        | { phase: \"snapshotError\"; count: number; error: unknown }\n      >(),\n      totalRenderCount() {\n        return Profiler.renders.length;\n      },\n      async peekRender(options: NextRenderOptions = {}) {\n        if (iteratorPosition < Profiler.renders.length) {\n          const render = Profiler.renders[iteratorPosition];\n\n          if (render.phase === \"snapshotError\") {\n            throw render.error;\n          }\n\n          return render;\n        }\n        return Profiler.waitForNextRender({\n          [_stackTrace]: captureStackTrace(Profiler.peekRender),\n          ...options,\n        });\n      },\n      async takeRender(options: NextRenderOptions = {}) {\n        // In many cases we do not control the resolution of the suspended\n        // promise which results in noisy tests when the profiler due to\n        // repeated act warnings.\n        using _disabledActWarnings = disableActWarnings();\n\n        let error: unknown = undefined;\n\n        try {\n          return await Profiler.peekRender({\n            [_stackTrace]: captureStackTrace(Profiler.takeRender),\n            ...options,\n          });\n        } catch (e) {\n          error = e;\n          throw e;\n        } finally {\n          if (!(error && error instanceof WaitForRenderTimeoutError)) {\n            iteratorPosition++;\n          }\n        }\n      },\n      getCurrentRender() {\n        // The \"current\" render should point at the same render that the most\n        // recent `takeRender` call returned, so we need to get the \"previous\"\n        // iterator position, otherwise `takeRender` advances the iterator\n        // to the next render. This means we need to call `takeRender` at least\n        // once before we can get a current render.\n        const currentPosition = iteratorPosition - 1;\n\n        if (currentPosition < 0) {\n          throw new Error(\n            \"No current render available. You need to call `takeRender` before you can get the current render.\"\n          );\n        }\n\n        const render = Profiler.renders[currentPosition];\n\n        if (render.phase === \"snapshotError\") {\n          throw render.error;\n        }\n        return render;\n      },\n      waitForNextRender({\n        timeout = 1000,\n        // capture the stack trace here so its stack trace is as close to the calling code as possible\n        [_stackTrace]: stackTrace = captureStackTrace(\n          Profiler.waitForNextRender\n        ),\n      }: NextRenderOptions = {}) {\n        if (!nextRender) {\n          nextRender = Promise.race<Render<Snapshot>>([\n            new Promise<Render<Snapshot>>((resolve, reject) => {\n              resolveNextRender = resolve;\n              rejectNextRender = reject;\n            }),\n            new Promise<Render<Snapshot>>((_, reject) =>\n              setTimeout(\n                () =>\n                  reject(\n                    applyStackTrace(new WaitForRenderTimeoutError(), stackTrace)\n                  ),\n                timeout\n              )\n            ),\n          ]);\n        }\n        return nextRender;\n      },\n    } satisfies ProfiledComponentFields<Snapshot>\n  );\n  return Profiler;\n}\n\n/** @internal */\nexport class WaitForRenderTimeoutError extends Error {\n  constructor() {\n    super(\"Exceeded timeout waiting for next render.\");\n    Object.setPrototypeOf(this, new.target.prototype);\n  }\n}\n\ntype StringReplaceRenderWithSnapshot<T extends string> =\n  T extends `${infer Pre}Render${infer Post}` ? `${Pre}Snapshot${Post}` : T;\n\ntype ResultReplaceRenderWithSnapshot<T> =\n  T extends (...args: infer Args) => Render<infer Snapshot> ?\n    (...args: Args) => Snapshot\n  : T extends (...args: infer Args) => Promise<Render<infer Snapshot>> ?\n    (...args: Args) => Promise<Snapshot>\n  : T;\n\ntype ProfiledHookFields<ReturnValue> =\n  ProfiledComponentFields<ReturnValue> extends infer PC ?\n    {\n      [K in keyof PC as StringReplaceRenderWithSnapshot<\n        K & string\n      >]: ResultReplaceRenderWithSnapshot<PC[K]>;\n    }\n  : never;\n\n/** @internal */\nexport interface ProfiledHook<Props, ReturnValue>\n  extends React.FC<Props>,\n    ProfiledHookFields<ReturnValue> {\n  Profiler: Profiler<ReturnValue>;\n}\n\n/** @internal */\nexport function profileHook<ReturnValue extends ValidSnapshot, Props>(\n  renderCallback: (props: Props) => ReturnValue\n): ProfiledHook<Props, ReturnValue> {\n  const Profiler = createProfiler<ReturnValue>();\n\n  const ProfiledHook = (props: Props) => {\n    Profiler.replaceSnapshot(renderCallback(props));\n    return null;\n  };\n\n  return Object.assign(\n    function App(props: Props) {\n      return (\n        <Profiler>\n          <ProfiledHook {...(props as any)} />\n        </Profiler>\n      );\n    },\n    {\n      Profiler,\n    },\n    {\n      renders: Profiler.renders,\n      totalSnapshotCount: Profiler.totalRenderCount,\n      async peekSnapshot(options) {\n        return (await Profiler.peekRender(options)).snapshot;\n      },\n      async takeSnapshot(options) {\n        return (await Profiler.takeRender(options)).snapshot;\n      },\n      getCurrentSnapshot() {\n        return Profiler.getCurrentRender().snapshot;\n      },\n      async waitForNextSnapshot(options) {\n        return (await Profiler.waitForNextRender(options)).snapshot;\n      },\n    } satisfies ProfiledHookFields<ReturnValue>\n  );\n}\n\nfunction resolveHookOwner(): React.ComponentType | undefined {\n  return (React as any).__SECRET_INTERNALS_DO_NOT_USE_OR_YOU_WILL_BE_FIRED\n    ?.ReactCurrentOwner?.current?.elementType;\n}\n\nexport function useTrackRenders({ name }: { name?: string } = {}) {\n  const component = name || resolveHookOwner();\n\n  if (!component) {\n    throw new Error(\n      \"useTrackRender: Unable to determine component. Please ensure the hook is called inside a rendered component or provide a `name` option.\"\n    );\n  }\n\n  const ctx = useProfilerContext();\n\n  if (!ctx) {\n    throw new Error(\n      \"useTrackComponentRender: A Profiler must be created and rendered to track component renders\"\n    );\n  }\n\n  React.useLayoutEffect(() => {\n    ctx.renderedComponents.unshift(component);\n  });\n}\n"]}

package/testing/internal/profile/traces.d.ts

@@ -1,3 +1,7 @@
+/**
+ * Captures a StackTrace and (if passed) cuts off
+ * the first lines including the calling function.
+ */
 export declare function captureStackTrace(callingFunction?: string | (() => {})): string;
 export declare function applyStackTrace(error: Error, stackTrace: string): Error;
 //# sourceMappingURL=traces.d.ts.map

package/testing/internal/profile/traces.js

@@ -1,3 +1,7 @@
+/**
+ * Captures a StackTrace and (if passed) cuts off
+ * the first lines including the calling function.
+ */
 export function captureStackTrace(callingFunction) {
     var stack = "";
     try {
@@ -6,15 +10,15 @@ export function captureStackTrace(callingFunction) {
     catch (e) {
         (stack = e.stack);
     }
-    var callerName = typeof callingFunction === "string"
-        ? callingFunction
-        : callingFunction
-            ? callingFunction.name
+    var callerName = typeof callingFunction === "string" ? callingFunction
+        : callingFunction ? callingFunction.name
             : undefined;
     if (callerName && stack.includes(callerName)) {
         var lines = stack.split("\n");
         stack = lines
-            .slice(lines.findLastIndex(function (line) { return line.includes(callerName); }) + 1)
+            .slice(
+        // @ts-expect-error this is too old of a TS target, but node has it
+        lines.findLastIndex(function (line) { return line.includes(callerName); }) + 1)
             .join("\n");
     }
     return stack;

package/testing/internal/profile/traces.js.map

@@ -1 +1 @@
-{"version":3,"file":"traces.js","sourceRoot":"","sources":["../../../../src/testing/internal/profile/traces.ts"],"names":[],"mappings":"AAIA,MAAM,UAAU,iBAAiB,CAAC,eAAqC;IACrE,IAAI,KAAK,GAAG,EAAE,CAAC;IACf,IAAI;QACF,MAAM,IAAI,KAAK,CAAC,EAAE,CAAC,CAAC;KACrB;IAAC,OAAO,CAAM,EAAE;QACf,CAAG,KAAK,GAAK,CAAC,MAAN,CAAO,CAAC;KACjB;IAED,IAAM,UAAU,GACd,OAAO,eAAe,KAAK,QAAQ;QACjC,CAAC,CAAC,eAAe;QACjB,CAAC,CAAC,eAAe;YACjB,CAAC,CAAC,eAAe,CAAC,IAAI;YACtB,CAAC,CAAC,SAAS,CAAC;IAEhB,IAAI,UAAU,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE;QAC5C,IAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhC,KAAK,GAAG,KAAK;aACV,KAAK,CAEJ,KAAK,CAAC,aAAa,CAAC,UAAC,IAAY,IAAK,OAAA,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAzB,CAAyB,CAAC,GAAG,CAAC,CACrE;aACA,IAAI,CAAC,IAAI,CAAC,CAAC;KACf;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,KAAY,EAAE,UAAkB;IAC9D,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,IAAI,GAAG,UAAU,CAAC;IAChD,OAAO,KAAK,CAAC;AACf,CAAC","sourcesContent":["/**\n * Captures a StackTrace and (if passed) cuts off\n * the first lines including the calling function.\n */\nexport function captureStackTrace(callingFunction?: string | (() => {})) {\n  let stack = \"\";\n  try {\n    throw new Error(\"\");\n  } catch (e: any) {\n    ({ stack } = e);\n  }\n\n  const callerName =\n    typeof callingFunction === \"string\"\n      ? callingFunction\n      : callingFunction\n      ? callingFunction.name\n      : undefined;\n\n  if (callerName && stack.includes(callerName)) {\n    const lines = stack.split(\"\\n\");\n\n    stack = lines\n      .slice(\n        // @ts-expect-error this is too old of a TS target, but node has it\n        lines.findLastIndex((line: string) => line.includes(callerName)) + 1\n      )\n      .join(\"\\n\");\n  }\n\n  return stack;\n}\n\nexport function applyStackTrace(error: Error, stackTrace: string) {\n  error.stack = error.message + \"\\n\" + stackTrace;\n  return error;\n}\n"]}
+{"version":3,"file":"traces.js","sourceRoot":"","sources":["../../../../src/testing/internal/profile/traces.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,eAAqC;IACrE,IAAI,KAAK,GAAG,EAAE,CAAC;IACf,IAAI,CAAC;QACH,MAAM,IAAI,KAAK,CAAC,EAAE,CAAC,CAAC;IACtB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,CAAG,KAAK,GAAK,CAAC,MAAN,CAAO,CAAC;IAClB,CAAC;IAED,IAAM,UAAU,GACd,OAAO,eAAe,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe;QACrD,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC,IAAI;YACxC,CAAC,CAAC,SAAS,CAAC;IAEd,IAAI,UAAU,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAC7C,IAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhC,KAAK,GAAG,KAAK;aACV,KAAK;QACJ,mEAAmE;QACnE,KAAK,CAAC,aAAa,CAAC,UAAC,IAAY,IAAK,OAAA,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAzB,CAAyB,CAAC,GAAG,CAAC,CACrE;aACA,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,KAAY,EAAE,UAAkB;IAC9D,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,GAAG,IAAI,GAAG,UAAU,CAAC;IAChD,OAAO,KAAK,CAAC;AACf,CAAC","sourcesContent":["/**\n * Captures a StackTrace and (if passed) cuts off\n * the first lines including the calling function.\n */\nexport function captureStackTrace(callingFunction?: string | (() => {})) {\n  let stack = \"\";\n  try {\n    throw new Error(\"\");\n  } catch (e: any) {\n    ({ stack } = e);\n  }\n\n  const callerName =\n    typeof callingFunction === \"string\" ? callingFunction\n    : callingFunction ? callingFunction.name\n    : undefined;\n\n  if (callerName && stack.includes(callerName)) {\n    const lines = stack.split(\"\\n\");\n\n    stack = lines\n      .slice(\n        // @ts-expect-error this is too old of a TS target, but node has it\n        lines.findLastIndex((line: string) => line.includes(callerName)) + 1\n      )\n      .join(\"\\n\");\n  }\n\n  return stack;\n}\n\nexport function applyStackTrace(error: Error, stackTrace: string) {\n  error.stack = error.message + \"\\n\" + stackTrace;\n  return error;\n}\n"]}