@memlab/core 1.1.16 → 1.1.18

package/dist/lib/HeapAnalyzer.js

@@ -23,7 +23,6 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const fs_1 = __importDefault(require("fs"));
 const babar_1 = __importDefault(require("babar"));
-const chalk_1 = __importDefault(require("chalk"));
 const LeakClusterLogger_1 = __importDefault(require("../logger/LeakClusterLogger"));
 const LeakTraceDetailsLogger_1 = __importDefault(require("../logger/LeakTraceDetailsLogger"));
 const TraceFinder_1 = __importDefault(require("../paths/TraceFinder"));
@@ -42,156 +41,6 @@ class MemoryAnalyst {
             return yield this.detectMemoryLeaks();
         });
     }
-    checkUnbound(options = {}) {
-        return __awaiter(this, void 0, void 0, function* () {
-            this.visualizeMemoryUsage(options);
-            Utils_1.default.checkSnapshots(options);
-            yield this.detectUnboundGrowth(options);
-        });
-    }
-    breakDownMemoryByShapes(options = {}) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const opt = { buildNodeIdIndex: true, verbose: true };
-            const file = options.file ||
-                Utils_1.default.getSnapshotFilePathWithTabType(/.*/) ||
-                '<EMPTY_FILE_PATH>';
-            const snapshot = yield Utils_1.default.getSnapshotFromFile(file, opt);
-            this.preparePathFinder(snapshot);
-            const heapInfo = this.getOverallHeapInfo(snapshot, { force: true });
-            if (heapInfo) {
-                this.printHeapInfo(heapInfo);
-            }
-            this.breakDownSnapshotByShapes(snapshot);
-        });
-    }
-    // find any objects that keeps growing
-    detectUnboundGrowth(options = {}) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const nodeInfo = Object.create(null);
-            let hasCheckedFirstSnapshot = false;
-            let snapshot = null;
-            const isValidNode = (node) => node.type === 'object' ||
-                node.type === 'closure' ||
-                node.type === 'regexp';
-            const initNodeInfo = (node) => {
-                if (!isValidNode(node)) {
-                    return;
-                }
-                const n = node.retainedSize;
-                nodeInfo[node.id] = {
-                    type: node.type,
-                    name: node.name,
-                    min: n,
-                    max: n,
-                    history: [n],
-                    node,
-                };
-            };
-            const updateNodeInfo = (node) => {
-                const item = nodeInfo[node.id];
-                if (!item) {
-                    return;
-                }
-                if (node.name !== item.name || node.type !== item.type) {
-                    nodeInfo[node.id] = null;
-                    return;
-                }
-                const n = node.retainedSize;
-                // only monotonic increase?
-                if (Config_1.default.monotonicUnboundGrowthOnly && n < item.max) {
-                    nodeInfo[node.id] = null;
-                    return;
-                }
-                item.history.push(n);
-                item.max = Math.max(item.max, n);
-                item.min = Math.min(item.min, n);
-            };
-            // summarize the heap objects info in current heap snapshot
-            // this is mainly used for better understanding of the % of
-            // objects released and allocated over time
-            const maybeSummarizeNodeInfo = () => {
-                if (!Config_1.default.verbose) {
-                    return;
-                }
-                let n = 0;
-                for (const k in nodeInfo) {
-                    if (nodeInfo[k]) {
-                        ++n;
-                    }
-                }
-                Console_1.default.lowLevel(`Objects tracked: ${n}`);
-            };
-            Console_1.default.overwrite('Checking unbounded objects...');
-            const snapshotFiles = options.snapshotDir
-                ? // load snapshots from a directory
-                    Utils_1.default.getSnapshotFilesInDir(options.snapshotDir)
-                : // load snapshots based on the visit sequence meta data
-                    Utils_1.default.getSnapshotFilesFromTabsOrder();
-            for (const file of snapshotFiles) {
-                // force GC before loading each snapshot
-                if (global.gc) {
-                    global.gc();
-                }
-                // load and preprocess heap snapshot
-                const opt = { buildNodeIdIndex: true, verbose: true };
-                snapshot = yield Utils_1.default.getSnapshotFromFile(file, opt);
-                this.calculateRetainedSizes(snapshot);
-                // keep track of heap objects
-                if (!hasCheckedFirstSnapshot) {
-                    // record Ids in the snapshot
-                    snapshot.nodes.forEach(initNodeInfo);
-                    hasCheckedFirstSnapshot = true;
-                }
-                else {
-                    snapshot.nodes.forEach(updateNodeInfo);
-                    maybeSummarizeNodeInfo();
-                }
-            }
-            // exit if no heap snapshot found
-            if (!hasCheckedFirstSnapshot) {
-                return;
-            }
-            // post process and print the unbounded objects
-            const idsInLastSnapshot = new Set();
-            snapshot === null || snapshot === void 0 ? void 0 : snapshot.nodes.forEach(node => {
-                idsInLastSnapshot.add(node.id);
-            });
-            let ids = [];
-            for (const key in nodeInfo) {
-                const id = parseInt(key, 10);
-                const item = nodeInfo[id];
-                if (!item) {
-                    continue;
-                }
-                if (!idsInLastSnapshot.has(id)) {
-                    continue;
-                }
-                if (item.min === item.max) {
-                    continue;
-                }
-                // filter out non-significant leaks
-                if (item.history[item.history.length - 1] < Config_1.default.unboundSizeThreshold) {
-                    continue;
-                }
-                ids.push(Object.assign({ id }, item));
-            }
-            if (ids.length === 0) {
-                Console_1.default.midLevel('No increasing objects found.');
-                return;
-            }
-            ids = ids
-                .sort((o1, o2) => o2.history[o2.history.length - 1] - o1.history[o1.history.length - 1])
-                .slice(0, 20);
-            // print on terminal
-            const str = Serializer_1.default.summarizeUnboundedObjects(ids, { color: true });
-            Console_1.default.topLevel('Top growing objects in sizes:');
-            Console_1.default.lowLevel(' (Use `memlab trace --node-id=@ID` to get trace)');
-            Console_1.default.topLevel('\n' + str);
-            // save results to file
-            const csv = Serializer_1.default.summarizeUnboundedObjectsToCSV(ids);
-            fs_1.default.writeFileSync(Config_1.default.unboundObjectCSV, csv, 'UTF-8');
-        });
-    }
     // find all unique pattern of leaks
     detectMemoryLeaks() {
         return __awaiter(this, void 0, void 0, function* () {
@@ -368,11 +217,6 @@ class MemoryAnalyst {
             };
         });
     }
-    calculateRetainedSizes(snapshot) {
-        const finder = new TraceFinder_1.default();
-        // dominator and retained size
-        finder.calculateAllNodesRetainedSizes(snapshot);
-    }
     // initialize the path finder
     preparePathFinder(snapshot) {
         const finder = new TraceFinder_1.default();
@@ -442,14 +286,6 @@ class MemoryAnalyst {
             Console_1.default.midLevel(`${leakedNodeIds.size} Fiber nodes and Detached elements`);
         }
     }
-    aggregateDominatorMetrics(ids, snapshot, checkNodeCb, nodeMetricsCb) {
-        let ret = 0;
-        const dominators = Utils_1.default.getConditionalDominatorIds(ids, snapshot, checkNodeCb);
-        Utils_1.default.applyToNodes(dominators, snapshot, node => {
-            ret += nodeMetricsCb(node);
-        });
-        return ret;
-    }
     getOverallHeapInfo(snapshot, options = {}) {
         if (!Config_1.default.verbose && !options.force) {
             return;
@@ -457,22 +293,19 @@ class MemoryAnalyst {
         Console_1.default.overwrite('summarizing heap info...');
         const allIds = Utils_1.default.getNodesIdSet(snapshot);
         const heapInfo = {
-            fiberNodeSize: this.aggregateDominatorMetrics(allIds, snapshot, Utils_1.default.isFiberNode, this.getRetainedSize),
-            regularFiberNodeSize: this.aggregateDominatorMetrics(allIds, snapshot, Utils_1.default.isRegularFiberNode, this.getRetainedSize),
-            detachedFiberNodeSize: this.aggregateDominatorMetrics(allIds, snapshot, Utils_1.default.isDetachedFiberNode, this.getRetainedSize),
-            alternateFiberNodeSize: this.aggregateDominatorMetrics(allIds, snapshot, Utils_1.default.isAlternateNode, this.getRetainedSize),
-            error: this.aggregateDominatorMetrics(allIds, snapshot, node => node.name === 'Error', this.getRetainedSize),
+            fiberNodeSize: Utils_1.default.aggregateDominatorMetrics(allIds, snapshot, Utils_1.default.isFiberNode, Utils_1.default.getRetainedSize),
+            regularFiberNodeSize: Utils_1.default.aggregateDominatorMetrics(allIds, snapshot, Utils_1.default.isRegularFiberNode, Utils_1.default.getRetainedSize),
+            detachedFiberNodeSize: Utils_1.default.aggregateDominatorMetrics(allIds, snapshot, Utils_1.default.isDetachedFiberNode, Utils_1.default.getRetainedSize),
+            alternateFiberNodeSize: Utils_1.default.aggregateDominatorMetrics(allIds, snapshot, Utils_1.default.isAlternateNode, Utils_1.default.getRetainedSize),
+            error: Utils_1.default.aggregateDominatorMetrics(allIds, snapshot, node => node.name === 'Error', Utils_1.default.getRetainedSize),
         };
         return heapInfo;
     }
-    getRetainedSize(node) {
-        return node.retainedSize;
-    }
     getOverallLeakInfo(leakedNodeIds, snapshot) {
         if (!Config_1.default.verbose) {
             return;
         }
-        const leakInfo = Object.assign(Object.assign({}, this.getOverallHeapInfo(snapshot)), { leakedSize: this.aggregateDominatorMetrics(leakedNodeIds, snapshot, () => true, this.getRetainedSize), leakedFiberNodeSize: this.aggregateDominatorMetrics(leakedNodeIds, snapshot, Utils_1.default.isFiberNode, this.getRetainedSize), leakedAlternateFiberNodeSize: this.aggregateDominatorMetrics(leakedNodeIds, snapshot, Utils_1.default.isAlternateNode, this.getRetainedSize) });
+        const leakInfo = Object.assign(Object.assign({}, this.getOverallHeapInfo(snapshot)), { leakedSize: Utils_1.default.aggregateDominatorMetrics(leakedNodeIds, snapshot, () => true, Utils_1.default.getRetainedSize), leakedFiberNodeSize: Utils_1.default.aggregateDominatorMetrics(leakedNodeIds, snapshot, Utils_1.default.isFiberNode, Utils_1.default.getRetainedSize), leakedAlternateFiberNodeSize: Utils_1.default.aggregateDominatorMetrics(leakedNodeIds, snapshot, Utils_1.default.isAlternateNode, Utils_1.default.getRetainedSize) });
         return leakInfo;
     }
     printHeapInfo(leakInfo) {
@@ -485,111 +318,6 @@ class MemoryAnalyst {
             Console_1.default.topLevel(`· ${name}: ${value}`);
         });
     }
-    breakDownSnapshotByShapes(snapshot) {
-        Console_1.default.overwrite('Breaking down memory by shapes...');
-        const breakdown = Object.create(null);
-        const population = Object.create(null);
-        // group objects based on their shapes
-        snapshot.nodes.forEach(node => {
-            if ((node.type !== 'object' && !Utils_1.default.isStringNode(node)) ||
-                Config_1.default.nodeIgnoreSetInShape.has(node.name)) {
-                return;
-            }
-            const key = Serializer_1.default.summarizeNodeShape(node);
-            breakdown[key] = breakdown[key] || new Set();
-            breakdown[key].add(node.id);
-            if (population[key] === undefined) {
-                population[key] = { examples: [], n: 0 };
-            }
-            ++population[key].n;
-            // retain the top 5 examples
-            const examples = population[key].examples;
-            examples.push(node);
-            examples.sort((n1, n2) => n2.retainedSize - n1.retainedSize);
-            if (examples.length > 5) {
-                examples.pop();
-            }
-        });
-        // calculate and sort based on retained sizes
-        const ret = [];
-        for (const key in breakdown) {
-            const size = this.aggregateDominatorMetrics(breakdown[key], snapshot, () => true, this.getRetainedSize);
-            ret.push({ key, retainedSize: size });
-        }
-        ret.sort((o1, o2) => o2.retainedSize - o1.retainedSize);
-        Console_1.default.topLevel('Object shapes with top retained sizes:');
-        Console_1.default.lowLevel(' (Use `memlab trace --node-id=@ID` to get trace)\n');
-        const topList = ret.slice(0, 40);
-        // print settings
-        const opt = { color: true, compact: true };
-        const dot = chalk_1.default.grey('· ');
-        const colon = chalk_1.default.grey(': ');
-        // print the shapes with the biggest retained size
-        for (const o of topList) {
-            const referrerInfo = this.breakDownByReferrers(breakdown[o.key], snapshot);
-            const { examples, n } = population[o.key];
-            const shapeStr = Serializer_1.default.summarizeNodeShape(examples[0], opt);
-            const bytes = Utils_1.default.getReadableBytes(o.retainedSize);
-            const examplesStr = examples
-                .map(e => `@${e.id} [${Utils_1.default.getReadableBytes(e.retainedSize)}]`)
-                .join(' | ');
-            const meta = chalk_1.default.grey(` (N: ${n}, Examples: ${examplesStr})`);
-            Console_1.default.topLevel(`${dot}${shapeStr}${colon}${bytes}${meta}`);
-            Console_1.default.lowLevel(referrerInfo + '\n');
-        }
-    }
-    isTrivialEdgeForBreakDown(edge) {
-        const source = edge.fromNode;
-        return (source.type === 'array' ||
-            source.name === '(object elements)' ||
-            source.name === 'system' ||
-            edge.name_or_index === '__proto__' ||
-            edge.name_or_index === 'prototype');
-    }
-    breakDownByReferrers(ids, snapshot) {
-        const edgeNames = Object.create(null);
-        for (const id of ids) {
-            const node = snapshot.getNodeById(id);
-            for (const edge of (node === null || node === void 0 ? void 0 : node.referrers) || []) {
-                const source = edge.fromNode;
-                if (!Utils_1.default.isMeaningfulEdge(edge) ||
-                    this.isTrivialEdgeForBreakDown(edge)) {
-                    continue;
-                }
-                const sourceName = Serializer_1.default.summarizeNodeName(source, {
-                    color: false,
-                });
-                const edgeName = Serializer_1.default.summarizeEdgeName(edge, {
-                    color: false,
-                    abstract: true,
-                });
-                const edgeKey = `[${sourceName}] --${edgeName}--> `;
-                edgeNames[edgeKey] = edgeNames[edgeKey] || {
-                    numberOfEdgesToNode: 0,
-                    source,
-                    edge,
-                };
-                ++edgeNames[edgeKey].numberOfEdgesToNode;
-            }
-        }
-        const referrerInfo = Object.entries(edgeNames)
-            .sort((i1, i2) => i2[1].numberOfEdgesToNode - i1[1].numberOfEdgesToNode)
-            .slice(0, 4)
-            .map(i => {
-            const meta = i[1];
-            const source = Serializer_1.default.summarizeNodeName(meta.source, {
-                color: true,
-            });
-            const edgeName = Serializer_1.default.summarizeEdgeName(meta.edge, {
-                color: true,
-                abstract: true,
-            });
-            const edgeSummary = `${source} --${edgeName}-->`;
-            return `  · ${edgeSummary}: ${meta.numberOfEdgesToNode}`;
-        })
-            .join('\n');
-        return referrerInfo;
-    }
     printHeapAndLeakInfo(leakedNodeIds, snapshot) {
         // write page interaction summary to the leaks text file
         this.dumpPageInteractionSummary();
@@ -616,10 +344,6 @@ class MemoryAnalyst {
             this.printHeapAndLeakInfo(leakedNodeIds, snapshot);
             // get all leaked objects
             this.filterLeakedObjects(leakedNodeIds, snapshot);
-            if (Config_1.default.verbose) {
-                // show a breakdown of different object structures
-                this.breakDownSnapshotByShapes(snapshot);
-            }
             const nodeIdInPaths = new Set();
             const paths = [];
             let numOfLeakedObjects = 0;
@@ -642,7 +366,7 @@ class MemoryAnalyst {
                 Console_1.default.midLevel(`${numOfLeakedObjects} leaked objects`);
             }
             // cluster traces from the current run
-            const clusters = TraceBucket_1.default.clusterPaths(paths, snapshot, this.aggregateDominatorMetrics, {
+            const clusters = TraceBucket_1.default.clusterPaths(paths, snapshot, Utils_1.default.aggregateDominatorMetrics, {
                 strategy: Config_1.default.isMLClustering
                     ? new MLTraceSimilarityStrategy_1.default()
                     : undefined,
@@ -651,7 +375,7 @@ class MemoryAnalyst {
             yield this.serializeClusterUpdate(clusters);
             if (Config_1.default.logUnclassifiedClusters) {
                 // cluster traces from the current run
-                const clustersUnclassified = TraceBucket_1.default.generateUnClassifiedClusters(paths, snapshot, this.aggregateDominatorMetrics);
+                const clustersUnclassified = TraceBucket_1.default.generateUnClassifiedClusters(paths, snapshot, Utils_1.default.aggregateDominatorMetrics);
                 LeakClusterLogger_1.default.logUnclassifiedClusters(clustersUnclassified);
             }
             return {
@@ -682,7 +406,7 @@ class MemoryAnalyst {
             }
         }, { reverse: true });
         // cluster traces from the current run
-        const clusters = TraceBucket_1.default.clusterPaths(paths, snapshot, this.aggregateDominatorMetrics, {
+        const clusters = TraceBucket_1.default.clusterPaths(paths, snapshot, Utils_1.default.aggregateDominatorMetrics, {
             strategy: Config_1.default.isMLClustering
                 ? new MLTraceSimilarityStrategy_1.default()
                 : undefined,

package/dist/lib/Types.d.ts

@@ -8,7 +8,7 @@
  * @oncall web_perf_infra
  */
 import { ParsedArgs } from 'minimist';
-import type { LaunchOptions, Page } from 'puppeteer';
+import type { LaunchOptions, Page as PuppeteerPage } from 'puppeteer';
 import type { ErrorHandling, MemLabConfig } from './Config';
 /** @internal */
 export declare type AnyValue = any;
@@ -43,6 +43,12 @@ export declare type HaltOrThrowOptions = {
     printCallback?: () => void;
 };
 /** @internal */
+export declare type FileOption = {
+    workDir?: Optional<string>;
+    clear?: boolean;
+    transient?: boolean;
+};
+/** @internal */
 export declare type CLIOptions = {
     cliArgs: ParsedArgs;
     configFromOptions?: AnyRecord;
@@ -56,6 +62,12 @@ export declare type XvfbType = {
     display: () => string;
 };
 /** @internal */
+export declare type ShellOptions = {
+    dir?: Optional<string>;
+    ignoreError?: Optional<boolean>;
+    disconnectStdio?: Optional<boolean>;
+};
+/** @internal */
 export declare type CLIArgs = {
     verbose: boolean;
     app: string;
@@ -85,6 +97,57 @@ export declare type CLIArgs = {
     'local-puppeteer': boolean;
     'snapshot-dir': string;
 };
+/**
+ * This is the puppeteer [`Page`](https://pptr.dev/api/puppeteer.page)
+ * class used by MemLab. The puppeteer `Page` class instance provides
+ * APIs to interact with the web browser.
+ *
+ * The puppeteer `Page` type can be incompatible across different versions.
+ * Your local npm-installed puppeteer version may be different from
+ * the puppeteer used by MemLab. This may cause some type errors, for example:
+ *
+ * ```typescript
+ * import type {Page} from 'puppeteer';
+ * import type {RunOptions} from '@memlab/api';
+ *
+ * const runOptions: RunOptions = {
+ *   scenario: {
+ *     // initial page load url: Google Maps
+ *     url: () => {
+ *       return "https://www.google.com/maps/@37.386427,-122.0428214,11z";
+ *     },
+ *     // type error here if your local puppeeter version is different
+ *     // from the puppeteer used by MemLab
+ *     action: async function (page: Page) {
+ *       await page.click('button[aria-label="Hotels"]');
+ *     },
+ *   },
+ * };
+ * ```
+ *
+ * To avoid the type error in the code example above, MemLab exports the
+ * puppeteer `Page` type used by MemLab so that your code can import it
+ * when necessary:
+ *
+ * ```typescript
+ * import type {Page} from '@memlab/core' // import Page type from memlab
+ * import type {RunOptions} from 'memlab';
+ *
+ * const runOptions: RunOptions = {
+ *   scenario: {
+ *     // initial page load url: Google Maps
+ *     url: () => {
+ *       return "https://www.google.com/maps/@37.386427,-122.0428214,11z";
+ *     },
+ *     // no type error here
+ *     action: async function (page: Page) {
+ *       await page.click('button[aria-label="Hotels"]');
+ *     },
+ *   },
+ * };
+ * ```
+ */
+export declare type Page = PuppeteerPage;
 /**
  * the predicate callback is used to decide if a
  * entity of type `T`.
@@ -338,7 +401,8 @@ export declare type LeakFilterCallback = (node: IHeapNode, snapshot: IHeapSnapsh
  * For concrete examples, check out {@link action} or {@link back}.
  *
  * @param page the puppeteer [`Page`](https://pptr.dev/api/puppeteer.page)
- * object, which provides APIs to interact with the web browser
+ * object, which provides APIs to interact with the web browser.
+ * To import this type, check out {@link Page}.
  * @returns no return value
  */
 export declare type InteractionsCallback = (page: Page, args?: OperationArgs) => Promise<void>;
@@ -420,7 +484,8 @@ export interface IScenario {
      *
      * * **Parameters**:
      *   * page: `Page` | the puppeteer [`Page`](https://pptr.dev/api/puppeteer.page)
-     *     object, which provides APIs to interact with the web browser
+     *     object, which provides APIs to interact with the web browser. To import
+     *     this type, check out {@link Page}.
      *
      * * **Examples**:
      * ```typescript
@@ -467,7 +532,8 @@ export interface IScenario {
      *
      * * **Parameters**:
      *   * page: `Page` | the puppeteer [`Page`](https://pptr.dev/api/puppeteer.page)
-     *     object, which provides APIs to interact with the web browser
+     *     object, which provides APIs to interact with the web browser. To import
+     *     this type, check out {@link Page}.
      *
      * * **Examples**:
      * ```typescript
@@ -496,7 +562,8 @@ export interface IScenario {
      *
      * * **Parameters**:
      *   * page: `Page` | the puppeteer [`Page`](https://pptr.dev/api/puppeteer.page)
-     *     object, which provides APIs to interact with the web browser
+     *     object, which provides APIs to interact with the web browser. To import
+     *     this type, check out {@link Page}.
      *
      * * **Examples**:
      * ```typescript
@@ -539,7 +606,8 @@ export interface IScenario {
      *
      * * **Parameters**:
      *   * page: `Page` | the puppeteer [`Page`](https://pptr.dev/api/puppeteer.page)
-     *     object, which provides APIs to interact with the web browser
+     *     object, which provides APIs to interact with the web browser. To import
+     *     this type, check out {@link Page}.
      *
      * * **Examples**:
      * ```typescript
@@ -582,7 +650,8 @@ export interface IScenario {
      *
      * * **Parameters**:
      *   * page: `Page` | the puppeteer [`Page`](https://pptr.dev/api/puppeteer.page)
-     *     object, which provides APIs to interact with the web browser
+     *     object, which provides APIs to interact with the web browser. To import
+     *     this type, check out {@link Page}.
      * * **Returns**: a boolean value, if it returns `true`, memlab will consider
      *   the navigation completes, if it returns `false`, memlab will keep calling
      *   this callback until it returns `true`. This is an async callback, you can
@@ -766,6 +835,7 @@ export interface IDataBuilder {
  * Callback function to provide if the page is loaded.
  * For concrete example, check out {@link isPageLoaded}.
  * @param page - puppeteer's [Page](https://pptr.dev/api/puppeteer.page/) object.
+ * To import this type, check out {@link Page}.
  * @returns a boolean value, if it returns `true`, memlab will consider
  * the navigation completes, if it returns `false`, memlab will keep calling
  * this callback until it returns `true`. This is an async callback, you can
@@ -773,6 +843,8 @@ export interface IDataBuilder {
  */
 export declare type CheckPageLoadCallback = (page: Page) => Promise<boolean>;
 /** @internal */
+export declare type PageSetupCallback = (page: Page) => Promise<void>;
+/** @internal */
 export interface IE2EScenarioVisitPlan {
     name: string;
     appName: string;
@@ -786,6 +858,7 @@ export interface IE2EScenarioVisitPlan {
     dataBuilder: Optional<IDataBuilder>;
     isPageLoaded?: CheckPageLoadCallback;
     scenario?: IScenario;
+    pageSetup?: PageSetupCallback;
 }
 /** @internal */
 export declare type OperationArgs = {

package/dist/lib/Utils.d.ts

@@ -7,7 +7,7 @@
  * @format
  * @oncall web_perf_infra
  */
-import type { HaltOrThrowOptions } from './Types';
+import type { HaltOrThrowOptions, HeapNodeIdSet, ShellOptions } from './Types';
 import type { Browser, Page } from 'puppeteer';
 import type { AnyAyncFunction, AnyOptions, E2EStepInfo, IHeapSnapshot, IHeapNode, IHeapEdge, IScenario, ILeakFilter, LeakTracePathItem, RunMetaInfo, RawHeapSnapshot, Nullable, Optional } from './Types';
 declare function isHermesInternalObject(node: IHeapNode): boolean;
@@ -122,7 +122,11 @@ declare function getError(maybeError: unknown): Error;
 declare function isNodeDominatedByDeletionsArray(node: IHeapNode): boolean;
 declare function getUniqueID(): string;
 declare function getClosureSourceUrl(node: IHeapNode): Nullable<string>;
+export declare function runShell(command: string, options?: ShellOptions): Nullable<string>;
+export declare function getRetainedSize(node: IHeapNode): number;
+export declare function aggregateDominatorMetrics(ids: HeapNodeIdSet, snapshot: IHeapSnapshot, checkNodeCb: (node: IHeapNode) => boolean, nodeMetricsCb: (node: IHeapNode) => number): number;
 declare const _default: {
+    aggregateDominatorMetrics: typeof aggregateDominatorMetrics;
     applyToNodes: typeof applyToNodes;
     callAsync: typeof callAsync;
     camelCaseToReadableString: typeof camelCaseToReadableString;
@@ -148,6 +152,7 @@ declare const _default: {
     getReadableBytes: typeof getReadableBytes;
     getReadablePercent: typeof getReadablePercent;
     getReadableTime: typeof getReadableTime;
+    getRetainedSize: typeof getRetainedSize;
     getRunMetaFilePath: typeof getRunMetaFilePath;
     getScenarioName: typeof getScenarioName;
     getSingleSnapshotFileForAnalysis: typeof getSingleSnapshotFileForAnalysis;
@@ -215,6 +220,7 @@ declare const _default: {
     repeat: typeof repeat;
     resolveFilePath: typeof resolveFilePath;
     resolveSnapshotFilePath: typeof resolveSnapshotFilePath;
+    runShell: typeof runShell;
     setIsAlternateNode: typeof setIsAlternateNode;
     setIsRegularFiberNode: typeof setIsRegularFiberNode;
     shouldShowMoreInfo: typeof shouldShowMoreInfo;

package/dist/lib/Utils.js

@@ -44,9 +44,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.resolveSnapshotFilePath = void 0;
+exports.aggregateDominatorMetrics = exports.getRetainedSize = exports.runShell = exports.resolveSnapshotFilePath = void 0;
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
+const child_process_1 = __importDefault(require("child_process"));
 const process_1 = __importDefault(require("process"));
 const Config_1 = __importStar(require("./Config"));
 const Console_1 = __importDefault(require("./Console"));
@@ -1751,7 +1752,52 @@ function getClosureSourceUrl(node) {
     const url = (_b = (_a = urlNode === null || urlNode === void 0 ? void 0 : urlNode.toStringNode()) === null || _a === void 0 ? void 0 : _a.stringValue) !== null && _b !== void 0 ? _b : null;
     return url;
 }
+function runShell(command, options = {}) {
+    var _a, _b, _c;
+    const runningDir = (_b = (_a = options.dir) !== null && _a !== void 0 ? _a : Config_1.default.workDir) !== null && _b !== void 0 ? _b : FileManager_1.default.getTmpDir();
+    const execOptions = {
+        cwd: runningDir,
+        stdio: options.disconnectStdio
+            ? []
+            : [process_1.default.stdin, process_1.default.stdout, process_1.default.stderr],
+    };
+    if (process_1.default.platform !== 'win32') {
+        execOptions.shell = '/bin/bash';
+    }
+    let ret = null;
+    try {
+        ret = child_process_1.default.execSync(command, execOptions);
+    }
+    catch (ex) {
+        if (Config_1.default.verbose) {
+            if (ex instanceof Error) {
+                Console_1.default.lowLevel(ex.message);
+                Console_1.default.lowLevel((_c = ex.stack) !== null && _c !== void 0 ? _c : '');
+            }
+        }
+        if (options.ignoreError === true) {
+            return '';
+        }
+        __1.utils.haltOrThrow(`Error when executing command: ${command}`);
+    }
+    return ret && ret.toString('UTF-8');
+}
+exports.runShell = runShell;
+function getRetainedSize(node) {
+    return node.retainedSize;
+}
+exports.getRetainedSize = getRetainedSize;
+function aggregateDominatorMetrics(ids, snapshot, checkNodeCb, nodeMetricsCb) {
+    let ret = 0;
+    const dominators = __1.utils.getConditionalDominatorIds(ids, snapshot, checkNodeCb);
+    __1.utils.applyToNodes(dominators, snapshot, node => {
+        ret += nodeMetricsCb(node);
+    });
+    return ret;
+}
+exports.aggregateDominatorMetrics = aggregateDominatorMetrics;
 exports.default = {
+    aggregateDominatorMetrics,
     applyToNodes,
     callAsync,
     camelCaseToReadableString,
@@ -1777,6 +1823,7 @@ exports.default = {
     getReadableBytes,
     getReadablePercent,
     getReadableTime,
+    getRetainedSize,
     getRunMetaFilePath,
     getScenarioName,
     getSingleSnapshotFileForAnalysis,
@@ -1844,6 +1891,7 @@ exports.default = {
     repeat,
     resolveFilePath,
     resolveSnapshotFilePath,
+    runShell,
     setIsAlternateNode,
     setIsRegularFiberNode,
     shouldShowMoreInfo,

package/dist/lib/heap-data/HeapNode.js

@@ -257,10 +257,9 @@ class HeapNode {
     get location() {
         const heapSnapshot = this.heapSnapshot;
         const locationIdx = heapSnapshot._nodeIdx2LocationIdx[this.idx];
-        if (locationIdx == null) {
-            return null;
-        }
-        return new HeapLocation_1.default(heapSnapshot, locationIdx);
+        return locationIdx == null
+            ? null
+            : new HeapLocation_1.default(heapSnapshot, locationIdx);
     }
     // search reference by edge name and edge type
     getReference(edgeName, edgeType) {