@memlab/cli 1.0.28 → 1.0.30

package/bin/memlab.js

@@ -1,4 +1,4 @@
-#!/usr/bin/env node --expose-gc --max-old-space-size=4096
+#!/usr/bin/env -S node --expose-gc --max-old-space-size=4096
 
 /**
  * Copyright (c) Meta Platforms, Inc. and affiliates.

package/dist/Dispatcher.d.ts

@@ -27,6 +27,8 @@ export declare class CommandDispatcher {
     protected resetData(): void;
     dispatch(args: ParsedArgs): Promise<void>;
     parseOptions(command: BaseCommand, config: MemLabConfig, args: ParsedArgs): Promise<AnyRecord>;
+    private mergeConfigFromOptions;
+    private mergeArrays;
     private runCommand;
     runSubCommandIfAny(command: BaseCommand, args: ParsedArgs, runCmdOpt: RunCommandOptions): Promise<void>;
     private helper;

package/dist/Dispatcher.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Dispatcher.d.ts","sourceRoot":"","sources":["../src/Dispatcher.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,KAAK,EAAC,SAAS,EAAE,YAAY,EAAC,MAAM,cAAc,CAAC;AAG1D,OAAO,WAAW,MAAM,eAAe,CAAC;AAGxC,OAAO,aAAa,MAAM,iBAAiB,CAAC;AAE5C,KAAK,iBAAiB,GAAG;IACvB,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,SAAS,CAAC;CAC9B,CAAC;AAEF,KAAK,wBAAwB,GAAG;IAC9B,aAAa,CAAC,EAAE,aAAa,CAAC;CAC/B,CAAC;AAIF,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,OAAO,CAA2B;IAC1C,OAAO,CAAC,gBAAgB,CAAc;IACtC,OAAO,CAAC,qBAAqB,CAAgB;gBAEjC,OAAO,GAAE,wBAA6B;IAOlD,SAAS,CAAC,SAAS;IAKb,QAAQ,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IA+BzC,YAAY,CAChB,OAAO,EAAE,WAAW,EACpB,MAAM,EAAE,YAAY,EACpB,IAAI,EAAE,UAAU,GACf,OAAO,CAAC,SAAS,CAAC;YAYP,UAAU;IAiDlB,kBAAkB,CACtB,OAAO,EAAE,WAAW,EACpB,IAAI,EAAE,UAAU,EAChB,SAAS,EAAE,iBAAiB,GAC3B,OAAO,CAAC,IAAI,CAAC;YA4BF,MAAM;CAWrB;;AAED,wBAAuC"}
+{"version":3,"file":"Dispatcher.d.ts","sourceRoot":"","sources":["../src/Dispatcher.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,KAAK,EAAC,SAAS,EAAY,YAAY,EAAW,MAAM,cAAc,CAAC;AAG9E,OAAO,WAAW,MAAM,eAAe,CAAC;AAGxC,OAAO,aAAa,MAAM,iBAAiB,CAAC;AAE5C,KAAK,iBAAiB,GAAG;IACvB,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,SAAS,CAAC;CAC9B,CAAC;AAEF,KAAK,wBAAwB,GAAG;IAC9B,aAAa,CAAC,EAAE,aAAa,CAAC;CAC/B,CAAC;AAIF,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,OAAO,CAA2B;IAC1C,OAAO,CAAC,gBAAgB,CAAc;IACtC,OAAO,CAAC,qBAAqB,CAAgB;gBAEjC,OAAO,GAAE,wBAA6B;IAOlD,SAAS,CAAC,SAAS;IAKb,QAAQ,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IA+BzC,YAAY,CAChB,OAAO,EAAE,WAAW,EACpB,MAAM,EAAE,YAAY,EACpB,IAAI,EAAE,UAAU,GACf,OAAO,CAAC,SAAS,CAAC;IAYrB,OAAO,CAAC,sBAAsB;IAgB9B,OAAO,CAAC,WAAW;YAaL,UAAU;IAiDlB,kBAAkB,CACtB,OAAO,EAAE,WAAW,EACpB,IAAI,EAAE,UAAU,EAChB,SAAS,EAAE,iBAAiB,GAC3B,OAAO,CAAC,IAAI,CAAC;YA4BF,MAAM;CAWrB;;AAED,wBAAuC"}

package/dist/Dispatcher.js

@@ -77,12 +77,41 @@ class CommandDispatcher {
             for (const option of options) {
                 const ret = yield option.run(config, args);
                 if (ret) {
-                    Object.assign(configFromOptions, ret);
+                    this.mergeConfigFromOptions(configFromOptions, ret);
                 }
             }
             return configFromOptions;
         });
     }
+    mergeConfigFromOptions(to, from) {
+        for (const key in from) {
+            if (Array.isArray(to[key]) && Array.isArray(from[key])) {
+                // both are arrays, merge them
+                this.mergeArrays(to[key], from[key]);
+            }
+            else if (from[key] == null || to[key] == null) {
+                // one of them is null, use the other one
+                to[key] = to[key] || from[key];
+            }
+            else {
+                // both have existing values, first one wins
+                core_1.info.warning(`Merge conflict CLI options key: ${key}`);
+            }
+        }
+        return to;
+    }
+    mergeArrays(arr1, arr2) {
+        if (arr1 == null) {
+            return arr2;
+        }
+        if (arr2 == null) {
+            return arr1;
+        }
+        for (const v of arr2) {
+            arr1.push(v);
+        }
+        return arr1;
+    }
     runCommand(command, args, runCmdOpt = { configFromOptions: {} }) {
         var _a;
         return __awaiter(this, void 0, void 0, function* () {

package/dist/commands/RunMeasureCommand.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"RunMeasureCommand.d.ts","sourceRoot":"","sources":["../../src/commands/RunMeasureCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AAEnE,OAAO,WAAW,MAAM,gBAAgB,CAAC;AAEzC,OAAO,EAAgB,UAAU,EAAC,MAAM,cAAc,CAAC;AAwBvD,MAAM,CAAC,OAAO,OAAO,iBAAkB,SAAQ,WAAW;IACxD,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,gBAAgB,IAAI,WAAW,EAAE;IAIjC,WAAW,IAAI,oBAAoB,EAAE;IAQrC,UAAU,IAAI,UAAU,EAAE;IAyBpB,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAS9C"}
+{"version":3,"file":"RunMeasureCommand.d.ts","sourceRoot":"","sources":["../../src/commands/RunMeasureCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AAEnE,OAAO,WAAW,MAAM,gBAAgB,CAAC;AAEzC,OAAO,EAAgB,UAAU,EAAC,MAAM,cAAc,CAAC;AAyBvD,MAAM,CAAC,OAAO,OAAO,iBAAkB,SAAQ,WAAW;IACxD,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,gBAAgB,IAAI,WAAW,EAAE;IAIjC,WAAW,IAAI,oBAAoB,EAAE;IAQrC,UAAU,IAAI,UAAU,EAAE;IA0BpB,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAS9C"}

package/dist/commands/RunMeasureCommand.js

@@ -46,6 +46,7 @@ const DisableWebSecurityOption_1 = __importDefault(require("../options/e2e/Disab
 const EnableJSRewriteOption_1 = __importDefault(require("../options/e2e/EnableJSRewriteOption"));
 const EnableJSInterceptOption_1 = __importDefault(require("../options/e2e/EnableJSInterceptOption"));
 const SetChromiumBinaryOption_1 = __importDefault(require("../options/e2e/SetChromiumBinaryOption"));
+const SetChromiumProtocolTimeoutOption_1 = __importDefault(require("../options/e2e/SetChromiumProtocolTimeoutOption"));
 class RunMeasureCommand extends BaseCommand_1.default {
     getCommandName() {
         return 'measure';
@@ -79,6 +80,7 @@ class RunMeasureCommand extends BaseCommand_1.default {
             new RemoteBrowserDebugOption_1.default(),
             new ScenarioFileOption_1.default(),
             new SetChromiumBinaryOption_1.default(),
+            new SetChromiumProtocolTimeoutOption_1.default(),
             new SetDeviceOption_1.default(),
             new SetUserAgentOption_1.default(),
             new DisableXvfbOption_1.default(),

package/dist/commands/WarmupAppCommand.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"WarmupAppCommand.d.ts","sourceRoot":"","sources":["../../src/commands/WarmupAppCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AAEnE,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AACxC,OAAO,WAAW,MAAM,gBAAgB,CAAC;AAmBzC,MAAM,CAAC,OAAO,OAAO,kBAAmB,SAAQ,WAAW;IACzD,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,gBAAgB,IAAI,WAAW,EAAE;IAIjC,WAAW,IAAI,oBAAoB,EAAE;IAOrC,UAAU,IAAI,UAAU,EAAE;IAoBpB,GAAG,CAAC,QAAQ,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAG/C"}
+{"version":3,"file":"WarmupAppCommand.d.ts","sourceRoot":"","sources":["../../src/commands/WarmupAppCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AAEnE,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AACxC,OAAO,WAAW,MAAM,gBAAgB,CAAC;AAoBzC,MAAM,CAAC,OAAO,OAAO,kBAAmB,SAAQ,WAAW;IACzD,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,gBAAgB,IAAI,WAAW,EAAE;IAIjC,WAAW,IAAI,oBAAoB,EAAE;IAOrC,UAAU,IAAI,UAAU,EAAE;IAqBpB,GAAG,CAAC,QAAQ,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAG/C"}

package/dist/commands/WarmupAppCommand.js

@@ -39,6 +39,7 @@ const DisableWebSecurityOption_1 = __importDefault(require("../options/e2e/Disab
 const EnableJSRewriteOption_1 = __importDefault(require("../options/e2e/EnableJSRewriteOption"));
 const EnableJSInterceptOption_1 = __importDefault(require("../options/e2e/EnableJSInterceptOption"));
 const SetChromiumBinaryOption_1 = __importDefault(require("../options/e2e/SetChromiumBinaryOption"));
+const SetChromiumProtocolTimeoutOption_1 = __importDefault(require("../options/e2e/SetChromiumProtocolTimeoutOption"));
 class FBWarmupAppCommand extends BaseCommand_1.default {
     getCommandName() {
         return 'warmup';
@@ -64,6 +65,7 @@ class FBWarmupAppCommand extends BaseCommand_1.default {
             new RemoteBrowserDebugOption_1.default(),
             new ScenarioFileOption_1.default(),
             new SetChromiumBinaryOption_1.default(),
+            new SetChromiumProtocolTimeoutOption_1.default(),
             new SetDeviceOption_1.default(),
             new SetUserAgentOption_1.default(),
             new DisableXvfbOption_1.default(),

package/dist/commands/heap/DiffLeakCommand.d.ts

@@ -26,7 +26,9 @@ export default class CheckLeakCommand extends BaseCommand {
     getCategory(): CommandCategory;
     getPrerequisites(): BaseCommand[];
     getOptions(): BaseOption[];
+    protected showWorkingDirErrorMessageAndHalt(): never;
     protected getWorkDirs(options: CLIOptions): WorkDirSettings;
+    private dumpVerboseInfo;
     run(options: CLIOptions): Promise<void>;
 }
 //# sourceMappingURL=DiffLeakCommand.d.ts.map

package/dist/commands/heap/DiffLeakCommand.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"DiffLeakCommand.d.ts","sourceRoot":"","sources":["../../../src/commands/heap/DiffLeakCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,EAAC,UAAU,EAAE,UAAU,EAAC,MAAM,cAAc,CAAC;AACpD,OAAO,KAAK,EAAC,uBAAuB,EAAC,MAAM,oBAAoB,CAAC;AAGhE,OAAO,WAAW,EAAE,EAAC,eAAe,EAAC,MAAM,mBAAmB,CAAC;AAiB/D,MAAM,MAAM,eAAe,GAAG;IAC5B,eAAe,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/B,iBAAiB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;CAClC,CAAC;AAEF,MAAM,CAAC,OAAO,OAAO,gBAAiB,SAAQ,WAAW;IACvD,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,0BAA0B,CAAS;IAE3C,SAAS,CAAC,6BAA6B,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAOlE,SAAS,CAAC,iCAAiC,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;gBAM1D,OAAO,GAAE,uBAA4B;IAKjD,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,WAAW,IAAI,eAAe;IAI9B,gBAAgB,IAAI,WAAW,EAAE;IAIjC,UAAU,IAAI,UAAU,EAAE;IAkB1B,SAAS,CAAC,WAAW,CAAC,OAAO,EAAE,UAAU,GAAG,eAAe;IAsBrD,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAa9C"}
+{"version":3,"file":"DiffLeakCommand.d.ts","sourceRoot":"","sources":["../../../src/commands/heap/DiffLeakCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,EAAC,UAAU,EAAE,UAAU,EAAC,MAAM,cAAc,CAAC;AACpD,OAAO,KAAK,EAAC,uBAAuB,EAAC,MAAM,oBAAoB,CAAC;AAGhE,OAAO,WAAW,EAAE,EAAC,eAAe,EAAC,MAAM,mBAAmB,CAAC;AAmB/D,MAAM,MAAM,eAAe,GAAG;IAC5B,eAAe,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/B,iBAAiB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;CAClC,CAAC;AAEF,MAAM,CAAC,OAAO,OAAO,gBAAiB,SAAQ,WAAW;IACvD,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,0BAA0B,CAAS;IAE3C,SAAS,CAAC,6BAA6B,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAOlE,SAAS,CAAC,iCAAiC,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;gBAM1D,OAAO,GAAE,uBAA4B;IAKjD,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,WAAW,IAAI,eAAe;IAI9B,gBAAgB,IAAI,WAAW,EAAE;IAIjC,UAAU,IAAI,UAAU,EAAE;IAoB1B,SAAS,CAAC,iCAAiC,IAAI,KAAK;IAYpD,SAAS,CAAC,WAAW,CAAC,OAAO,EAAE,UAAU,GAAG,eAAe;IAqB3D,OAAO,CAAC,eAAe;IAejB,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAc9C"}

package/dist/commands/heap/DiffLeakCommand.js

@@ -52,6 +52,8 @@ const SetControlWorkDirOption_1 = __importDefault(require("../../options/experim
 const SetTreatmentWorkDirOption_1 = __importDefault(require("../../options/experiment/SetTreatmentWorkDirOption"));
 const SetMaxClusterSampleSizeOption_1 = __importDefault(require("../../options/SetMaxClusterSampleSizeOption"));
 const SetTraceContainsFilterOption_1 = __importDefault(require("../../options/heap/SetTraceContainsFilterOption"));
+const SetControlSnapshotOption_1 = __importDefault(require("../../options/experiment/SetControlSnapshotOption"));
+const SetTreatmentSnapshotOption_1 = __importDefault(require("../../options/experiment/SetTreatmentSnapshotOption"));
 class CheckLeakCommand extends BaseCommand_1.default {
     useDefaultMLClusteringSetting(cliArgs) {
         if (!MLClusteringOption_1.default.hasOptionSet(cliArgs)) {
@@ -84,7 +86,9 @@ class CheckLeakCommand extends BaseCommand_1.default {
     }
     getOptions() {
         return [
+            new SetControlSnapshotOption_1.default(),
             new SetControlWorkDirOption_1.default(),
+            new SetTreatmentSnapshotOption_1.default(),
             new SetTreatmentWorkDirOption_1.default(),
             new JSEngineOption_1.default(),
             new LeakFilterFileOption_1.default(),
@@ -99,13 +103,21 @@ class CheckLeakCommand extends BaseCommand_1.default {
             new SetTraceContainsFilterOption_1.default(),
         ];
     }
+    showWorkingDirErrorMessageAndHalt() {
+        throw core_1.utils.haltOrThrow('No control or test working directory or snapshot location specified, ' +
+            'please specify them via: \n' +
+            ` --${new SetControlSnapshotOption_1.default().getOptionName()} with ` +
+            ` --${new SetTreatmentSnapshotOption_1.default().getOptionName()}\n` +
+            'alternatively, you can also specify them via: \n' +
+            ` --${new SetControlWorkDirOption_1.default().getOptionName()} with ` +
+            ` --${new SetTreatmentWorkDirOption_1.default().getOptionName()}`);
+    }
     getWorkDirs(options) {
         var _a, _b;
         // double check parameters
         if (!((_a = options.configFromOptions) === null || _a === void 0 ? void 0 : _a.controlWorkDirs) ||
             !((_b = options.configFromOptions) === null || _b === void 0 ? void 0 : _b.treatmentWorkDirs)) {
-            core_1.info.error('Please specify control and test working directory');
-            throw core_1.utils.haltOrThrow('No control or test working directory specified');
+            this.showWorkingDirErrorMessageAndHalt();
         }
         // get parameters
         const controlWorkDirs = options.configFromOptions['controlWorkDirs'];
@@ -115,6 +127,13 @@ class CheckLeakCommand extends BaseCommand_1.default {
             treatmentWorkDirs,
         };
     }
+    dumpVerboseInfo(controlWorkDirs, treatmentWorkDirs) {
+        if (core_1.config.verbose) {
+            core_1.info.lowLevel(`control working directories: ${controlWorkDirs.join(', ')}`);
+            core_1.info.lowLevel(`treatment working directories: ${treatmentWorkDirs.join(', ')}`);
+            core_1.info.lowLevel(`diffing working directory: ${treatmentWorkDirs[0]}`);
+        }
+    }
     run(options) {
         return __awaiter(this, void 0, void 0, function* () {
             core_1.config.chaseWeakMapEdge = false;
@@ -124,6 +143,7 @@ class CheckLeakCommand extends BaseCommand_1.default {
                 workDir: treatmentWorkDirs[0],
                 silentFail: true,
             });
+            this.dumpVerboseInfo(controlWorkDirs, treatmentWorkDirs);
             // diff memory leaks
             this.useDefaultMLClusteringSetting(options.cliArgs);
             yield core_1.analysis.diffLeakByWorkDir({ controlWorkDirs, treatmentWorkDirs });

package/dist/commands/heap/HeapAnalysisCommand.d.ts

@@ -7,15 +7,17 @@
  * @format
  * @oncall web_perf_infra
  */
-import type { CLIOptions, CommandOptionExample } from '@memlab/core';
+import type { BaseOption, CLIOptions, CommandOptionExample } from '@memlab/core';
 import BaseCommand, { CommandCategory } from '../../BaseCommand';
 export default class RunHeapAnalysisCommand extends BaseCommand {
     getCommandName(): string;
     getDescription(): string;
     getCategory(): CommandCategory;
     getPrerequisites(): BaseCommand[];
+    getOptions(): BaseOption[];
     getSubCommands(): BaseCommand[];
     getExamples(): CommandOptionExample[];
+    private errorIfNoSubCommand;
     run(options: CLIOptions): Promise<void>;
 }
 //# sourceMappingURL=HeapAnalysisCommand.d.ts.map

package/dist/commands/heap/HeapAnalysisCommand.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"HeapAnalysisCommand.d.ts","sourceRoot":"","sources":["../../../src/commands/heap/HeapAnalysisCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AAGnE,OAAO,WAAW,EAAE,EAAC,eAAe,EAAC,MAAM,mBAAmB,CAAC;AAO/D,MAAM,CAAC,OAAO,OAAO,sBAAuB,SAAQ,WAAW;IAC7D,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,WAAW,IAAI,eAAe;IAI9B,gBAAgB,IAAI,WAAW,EAAE;IAIjC,cAAc,IAAI,WAAW,EAAE;IAU/B,WAAW,IAAI,oBAAoB,EAAE;IAI/B,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAqB9C"}
+{"version":3,"file":"HeapAnalysisCommand.d.ts","sourceRoot":"","sources":["../../../src/commands/heap/HeapAnalysisCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAE,UAAU,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AAG/E,OAAO,WAAW,EAAE,EAAC,eAAe,EAAC,MAAM,mBAAmB,CAAC;AAS/D,MAAM,CAAC,OAAO,OAAO,sBAAuB,SAAQ,WAAW;IAC7D,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,WAAW,IAAI,eAAe;IAI9B,gBAAgB,IAAI,WAAW,EAAE;IAIjC,UAAU,IAAI,UAAU,EAAE;IAI1B,cAAc,IAAI,WAAW,EAAE;IAU/B,WAAW,IAAI,oBAAoB,EAAE;YAIvB,mBAAmB;IAwB3B,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAc9C"}

package/dist/commands/heap/HeapAnalysisCommand.js

@@ -50,12 +50,13 @@ const heap_analysis_1 = require("@memlab/heap-analysis");
 const HeapAnalysisSubCommandWrapper_1 = __importDefault(require("./HeapAnalysisSubCommandWrapper"));
 const HelperCommand_1 = __importDefault(require("../helper/HelperCommand"));
 const InitDirectoryCommand_1 = __importDefault(require("../InitDirectoryCommand"));
+const HeapAnalysisPluginOption_1 = __importDefault(require("../../options/heap/HeapAnalysisPluginOption"));
 class RunHeapAnalysisCommand extends BaseCommand_1.default {
     getCommandName() {
         return 'analyze';
     }
     getDescription() {
-        return 'Run heap analysis plugins';
+        return 'Run heap analysis on heap snapshots.\n';
     }
     getCategory() {
         return BaseCommand_1.CommandCategory.COMMON;
@@ -63,6 +64,9 @@ class RunHeapAnalysisCommand extends BaseCommand_1.default {
     getPrerequisites() {
         return [new InitDirectoryCommand_1.default()];
     }
+    getOptions() {
+        return [new HeapAnalysisPluginOption_1.default()];
+    }
     getSubCommands() {
         const analyses = [...heap_analysis_1.heapAnalysisLoader.loadAllAnalysis().values()];
         return analyses.map((analysis) => {
@@ -75,23 +79,38 @@ class RunHeapAnalysisCommand extends BaseCommand_1.default {
     getExamples() {
         return ['<PLUGIN_NAME> [PLUGIN_OPTIONS]'];
     }
+    errorIfNoSubCommand(args, analysisMap) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (args && args._.length >= 2 && analysisMap.has(args._[1])) {
+                return;
+            }
+            const helper = new HelperCommand_1.default();
+            const modules = new Map();
+            for (const subCommand of this.getSubCommands()) {
+                modules.set(subCommand.getCommandName(), subCommand);
+            }
+            const errMsg = args && args._.length < 2
+                ? `\n  Heap analysis plugin name missing\n`
+                : `\n  Invalid command \`memlab ${this.getCommandName()} ${args === null || args === void 0 ? void 0 : args._[1]}\`\n`;
+            core_1.info.error(errMsg);
+            yield helper.run({ cliArgs: args, modules, command: this });
+            core_1.utils.haltOrThrow(errMsg, { printErrorBeforeHalting: false });
+        });
+    }
     run(options) {
+        var _a;
         return __awaiter(this, void 0, void 0, function* () {
+            // process command line arguments and load analysis modules
             const args = options.cliArgs;
-            const analysisMap = heap_analysis_1.heapAnalysisLoader.loadAllAnalysis();
-            if (!args || args._.length < 2 || !analysisMap.has(args._[1])) {
-                const helper = new HelperCommand_1.default();
-                const modules = new Map();
-                for (const subCommand of this.getSubCommands()) {
-                    modules.set(subCommand.getCommandName(), subCommand);
-                }
-                const errMsg = args && args._.length < 2
-                    ? `\n  Heap analysis plugin name missing\n`
-                    : `\n  Invalid command \`memlab ${this.getCommandName()} ${args === null || args === void 0 ? void 0 : args._[1]}\`\n`;
-                core_1.info.error(errMsg);
-                yield helper.run({ cliArgs: args, modules, command: this });
-                core_1.utils.haltOrThrow(errMsg, { printErrorBeforeHalting: false });
+            let plugin = (_a = options.configFromOptions) === null || _a === void 0 ? void 0 : _a.heapAnalysisPlugin;
+            if (plugin != null) {
+                plugin = `${plugin}`;
             }
+            const analysisMap = heap_analysis_1.heapAnalysisLoader.loadAllAnalysis({
+                heapAnalysisPlugin: plugin,
+                errorWhenPluginFailed: true,
+            });
+            yield this.errorIfNoSubCommand(args, analysisMap);
         });
     }
 }

package/dist/commands/snapshot/TakeSnapshotCommand.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TakeSnapshotCommand.d.ts","sourceRoot":"","sources":["../../../src/commands/snapshot/TakeSnapshotCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AAEnE,OAAO,WAAW,MAAM,mBAAmB,CAAC;AAG5C,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AAwBxC,MAAM,CAAC,OAAO,OAAO,mBAAoB,SAAQ,WAAW;IAC1D,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,gBAAgB,IAAI,WAAW,EAAE;IAQjC,WAAW,IAAI,oBAAoB,EAAE;IAQrC,UAAU,IAAI,UAAU,EAAE;IA0BpB,GAAG,CAAC,QAAQ,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAG/C"}
+{"version":3,"file":"TakeSnapshotCommand.d.ts","sourceRoot":"","sources":["../../../src/commands/snapshot/TakeSnapshotCommand.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAE,oBAAoB,EAAC,MAAM,cAAc,CAAC;AAEnE,OAAO,WAAW,MAAM,mBAAmB,CAAC;AAG5C,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AAyBxC,MAAM,CAAC,OAAO,OAAO,mBAAoB,SAAQ,WAAW;IAC1D,cAAc,IAAI,MAAM;IAIxB,cAAc,IAAI,MAAM;IAIxB,gBAAgB,IAAI,WAAW,EAAE;IAQjC,WAAW,IAAI,oBAAoB,EAAE;IAQrC,UAAU,IAAI,UAAU,EAAE;IA2BpB,GAAG,CAAC,QAAQ,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAG/C"}

package/dist/commands/snapshot/TakeSnapshotCommand.js

@@ -46,6 +46,7 @@ const EnableJSRewriteOption_1 = __importDefault(require("../../options/e2e/Enabl
 const EnableJSInterceptOption_1 = __importDefault(require("../../options/e2e/EnableJSInterceptOption"));
 const TargetWorkerOption_1 = __importDefault(require("../../options/e2e/TargetWorkerOption"));
 const SetChromiumBinaryOption_1 = __importDefault(require("../../options/e2e/SetChromiumBinaryOption"));
+const SetChromiumProtocolTimeoutOption_1 = __importDefault(require("../../options/e2e/SetChromiumProtocolTimeoutOption"));
 class TakeSnapshotCommand extends BaseCommand_1.default {
     getCommandName() {
         return 'snapshot';
@@ -82,6 +83,7 @@ class TakeSnapshotCommand extends BaseCommand_1.default {
             new RemoteBrowserDebugOption_1.default(),
             new ScenarioFileOption_1.default(),
             new SetChromiumBinaryOption_1.default(),
+            new SetChromiumProtocolTimeoutOption_1.default(),
             new SetDeviceOption_1.default(),
             new SetUserAgentOption_1.default(),
             new DisableXvfbOption_1.default(),

package/dist/options/e2e/SetChromiumBinaryOption.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SetChromiumBinaryOption.d.ts","sourceRoot":"","sources":["../../../src/options/e2e/SetChromiumBinaryOption.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,EAAC,YAAY,EAAC,MAAM,cAAc,CAAC;AAI1C,OAAO,EAAc,UAAU,EAAC,MAAM,cAAc,CAAC;AAGrD,MAAM,CAAC,OAAO,OAAO,uBAAwB,SAAQ,UAAU;IAC7D,aAAa,IAAI,MAAM;IAIvB,cAAc,IAAI,MAAM;IAIlB,KAAK,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAgBnE"}
+{"version":3,"file":"SetChromiumBinaryOption.d.ts","sourceRoot":"","sources":["../../../src/options/e2e/SetChromiumBinaryOption.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,EAAC,YAAY,EAAC,MAAM,cAAc,CAAC;AAE1C,OAAO,EAAQ,UAAU,EAAC,MAAM,cAAc,CAAC;AAG/C,MAAM,CAAC,OAAO,OAAO,uBAAwB,SAAQ,UAAU;IAC7D,aAAa,IAAI,MAAM;IAIvB,cAAc,IAAI,MAAM;IAIlB,KAAK,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAOnE"}

package/dist/options/e2e/SetChromiumBinaryOption.js

@@ -12,8 +12,6 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-const path_1 = __importDefault(require("path"));
-const fs_1 = __importDefault(require("fs"));
 const core_1 = require("@memlab/core");
 const OptionConstant_1 = __importDefault(require("../lib/OptionConstant"));
 class SetChromiumBinaryOption extends core_1.BaseOption {
@@ -28,14 +26,7 @@ class SetChromiumBinaryOption extends core_1.BaseOption {
             const name = this.getOptionName();
             const arg = args[name];
             if (arg) {
-                const binaryPath = path_1.default.resolve(process.cwd(), arg);
-                if (!fs_1.default.existsSync(binaryPath)) {
-                    throw core_1.utils.haltOrThrow(`Chromium binary does not exist: ${binaryPath}`);
-                }
-                if (config.verbose) {
-                    core_1.info.lowLevel(`Using ${binaryPath} as Chromium binary for E2E run`);
-                }
-                config.puppeteerConfig.executablePath = binaryPath;
+                core_1.utils.setChromiumBinary(config, arg);
             }
         });
     }

package/dist/options/e2e/SetChromiumProtocolTimeoutOption.d.ts

@@ -0,0 +1,18 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+import type { ParsedArgs } from 'minimist';
+import { MemLabConfig } from '@memlab/core';
+import { BaseOption } from '@memlab/core';
+export default class SetChromiumProtocolTimeoutOption extends BaseOption {
+    getOptionName(): string;
+    getDescription(): string;
+    parse(config: MemLabConfig, args: ParsedArgs): Promise<void>;
+}
+//# sourceMappingURL=SetChromiumProtocolTimeoutOption.d.ts.map

package/dist/options/e2e/SetChromiumProtocolTimeoutOption.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SetChromiumProtocolTimeoutOption.d.ts","sourceRoot":"","sources":["../../../src/options/e2e/SetChromiumProtocolTimeoutOption.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,EAAC,YAAY,EAAC,MAAM,cAAc,CAAC;AAE1C,OAAO,EAAc,UAAU,EAAC,MAAM,cAAc,CAAC;AAGrD,MAAM,CAAC,OAAO,OAAO,gCAAiC,SAAQ,UAAU;IACtE,aAAa,IAAI,MAAM;IAIvB,cAAc,IAAI,MAAM;IAUlB,KAAK,CAAC,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;CAiBnE"}

package/dist/options/e2e/SetChromiumProtocolTimeoutOption.js

@@ -0,0 +1,46 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@memlab/core");
+const OptionConstant_1 = __importDefault(require("../lib/OptionConstant"));
+class SetChromiumProtocolTimeoutOption extends core_1.BaseOption {
+    getOptionName() {
+        return OptionConstant_1.default.optionNames.CHROMIUM_PROTOCOL_TIMEOUT;
+    }
+    getDescription() {
+        return ('set the protocol timeout for chromium connection (in ms). \n' +
+            'The current default value is 180000, you may want to increase the ' +
+            'timeout via this flag when the heap snapshot is ' +
+            'too big (e.g., over 1GB) and the Page crashed with error: ' +
+            "'ProtocolError: HeapProfiler.takeHeapSnapshot timed out'.");
+    }
+    parse(config, args) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const name = this.getOptionName();
+            const arg = args[name];
+            if (arg) {
+                const timeout = parseInt(arg, 10);
+                if (Number.isNaN(timeout)) {
+                    core_1.utils.haltOrThrow(`Invalid Chromium protocol timeout value: ${arg}. ` +
+                        'It must be a number.');
+                }
+                if (config.verbose) {
+                    core_1.info.lowLevel(`Set Chromium protocol timeout to ${timeout}.`);
+                }
+                config.puppeteerConfig.protocolTimeout = timeout;
+            }
+        });
+    }
+}
+exports.default = SetChromiumProtocolTimeoutOption;

package/dist/options/experiment/SetControlSnapshotOption.d.ts

@@ -0,0 +1,20 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+import type { ParsedArgs } from 'minimist';
+import { MemLabConfig, Nullable } from '@memlab/core';
+import { BaseOption } from '@memlab/core';
+export default class SetControlWorkDirOption extends BaseOption {
+    getOptionName(): string;
+    getDescription(): string;
+    parse(config: MemLabConfig, args: ParsedArgs): Promise<{
+        controlWorkDirs?: Nullable<string[]>;
+    }>;
+}
+//# sourceMappingURL=SetControlSnapshotOption.d.ts.map

package/dist/options/experiment/SetControlSnapshotOption.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SetControlSnapshotOption.d.ts","sourceRoot":"","sources":["../../../src/options/experiment/SetControlSnapshotOption.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,EAAC,YAAY,EAAE,QAAQ,EAAC,MAAM,cAAc,CAAC;AAEpD,OAAO,EAAC,UAAU,EAAO,MAAM,cAAc,CAAC;AAO9C,MAAM,CAAC,OAAO,OAAO,uBAAwB,SAAQ,UAAU;IAC7D,aAAa,IAAI,MAAM;IAIvB,cAAc,IAAI,MAAM;IAIlB,KAAK,CACT,MAAM,EAAE,YAAY,EACpB,IAAI,EAAE,UAAU,GACf,OAAO,CAAC;QAAC,eAAe,CAAC,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAA;KAAC,CAAC;CAYnD"}

package/dist/options/experiment/SetControlSnapshotOption.js

@@ -0,0 +1,49 @@
+"use strict";
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@memlab/core");
+const OptionConstant_1 = __importDefault(require("../lib/OptionConstant"));
+const ExperimentOptionsUtils_1 = require("./utils/ExperimentOptionsUtils");
+class SetControlWorkDirOption extends core_1.BaseOption {
+    getOptionName() {
+        return OptionConstant_1.default.optionNames.CONTROL_SNAPSHOT;
+    }
+    getDescription() {
+        return 'set the single (target) snapshot of control run';
+    }
+    parse(config, args) {
+        return __awaiter(this, void 0, void 0, function* () {
+            let dirs = null;
+            const optionName = this.getOptionName();
+            if (optionName in args) {
+                const snapshotFile = (0, ExperimentOptionsUtils_1.validateHeapSnapshotFileOrThrow)(args[optionName]);
+                dirs = [(0, ExperimentOptionsUtils_1.createTransientWorkDirFromSingleHeapSnapshot)(snapshotFile)];
+            }
+            if (config.verbose && dirs && dirs[0]) {
+                core_1.info.lowLevel(`creating control working directory: ${dirs[0]}`);
+            }
+            return { controlWorkDirs: dirs };
+        });
+    }
+}
+exports.default = SetControlWorkDirOption;

package/dist/options/experiment/SetTreatmentSnapshotOption.d.ts

@@ -0,0 +1,20 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+import type { ParsedArgs } from 'minimist';
+import { MemLabConfig, Nullable } from '@memlab/core';
+import { BaseOption } from '@memlab/core';
+export default class SetTreatmentWorkDirOption extends BaseOption {
+    getOptionName(): string;
+    getDescription(): string;
+    parse(config: MemLabConfig, args: ParsedArgs): Promise<{
+        treatmentWorkDirs?: Nullable<string[]>;
+    }>;
+}
+//# sourceMappingURL=SetTreatmentSnapshotOption.d.ts.map

package/dist/options/experiment/SetTreatmentSnapshotOption.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SetTreatmentSnapshotOption.d.ts","sourceRoot":"","sources":["../../../src/options/experiment/SetTreatmentSnapshotOption.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,EAAC,YAAY,EAAE,QAAQ,EAAC,MAAM,cAAc,CAAC;AAEpD,OAAO,EAAC,UAAU,EAAO,MAAM,cAAc,CAAC;AAO9C,MAAM,CAAC,OAAO,OAAO,yBAA0B,SAAQ,UAAU;IAC/D,aAAa,IAAI,MAAM;IAIvB,cAAc,IAAI,MAAM;IAIlB,KAAK,CACT,MAAM,EAAE,YAAY,EACpB,IAAI,EAAE,UAAU,GACf,OAAO,CAAC;QAAC,iBAAiB,CAAC,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAA;KAAC,CAAC;CAYrD"}

package/dist/options/experiment/SetTreatmentSnapshotOption.js

@@ -0,0 +1,49 @@
+"use strict";
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@memlab/core");
+const OptionConstant_1 = __importDefault(require("../lib/OptionConstant"));
+const ExperimentOptionsUtils_1 = require("./utils/ExperimentOptionsUtils");
+class SetTreatmentWorkDirOption extends core_1.BaseOption {
+    getOptionName() {
+        return OptionConstant_1.default.optionNames.TREATMENT_SNAPSHOT;
+    }
+    getDescription() {
+        return 'set the single (target) snapshot of treatment run';
+    }
+    parse(config, args) {
+        return __awaiter(this, void 0, void 0, function* () {
+            let dirs = null;
+            const optionName = this.getOptionName();
+            if (optionName in args) {
+                const snapshotFile = (0, ExperimentOptionsUtils_1.validateHeapSnapshotFileOrThrow)(args[optionName]);
+                dirs = [(0, ExperimentOptionsUtils_1.createTransientWorkDirFromSingleHeapSnapshot)(snapshotFile)];
+            }
+            if (config.verbose && dirs && dirs[0]) {
+                core_1.info.lowLevel(`creating treatment working directory: ${dirs[0]}`);
+            }
+            return { treatmentWorkDirs: dirs };
+        });
+    }
+}
+exports.default = SetTreatmentWorkDirOption;

package/dist/options/experiment/utils/ExperimentOptionsUtils.d.ts

@@ -0,0 +1,13 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+import { AnyValue } from '@memlab/core';
+export declare function validateHeapSnapshotFileOrThrow(file: AnyValue): string;
+export declare function createTransientWorkDirFromSingleHeapSnapshot(file: string): string;
+//# sourceMappingURL=ExperimentOptionsUtils.d.ts.map

package/dist/options/experiment/utils/ExperimentOptionsUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ExperimentOptionsUtils.d.ts","sourceRoot":"","sources":["../../../../src/options/experiment/utils/ExperimentOptionsUtils.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAC,QAAQ,EAAmC,MAAM,cAAc,CAAC;AAGxE,wBAAgB,+BAA+B,CAAC,IAAI,EAAE,QAAQ,GAAG,MAAM,CAetE;AAED,wBAAgB,4CAA4C,CAC1D,IAAI,EAAE,MAAM,GACX,MAAM,CAIR"}

package/dist/options/experiment/utils/ExperimentOptionsUtils.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTransientWorkDirFromSingleHeapSnapshot = exports.validateHeapSnapshotFileOrThrow = void 0;
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+const core_1 = require("@memlab/core");
+const fs_extra_1 = require("fs-extra");
+function validateHeapSnapshotFileOrThrow(file) {
+    if (typeof file !== 'string') {
+        throw core_1.utils.haltOrThrow(`Heap snapshot file must be a string, but got ${typeof file}`);
+    }
+    if (!file.endsWith('.heapsnapshot')) {
+        throw core_1.utils.haltOrThrow(`Heap snapshot file must end with .heapsnapshot, but got ${file}`);
+    }
+    if (!(0, fs_extra_1.existsSync)(file)) {
+        throw core_1.utils.haltOrThrow(`Heap snapshot file ${file} does not exist`);
+    }
+    return file;
+}
+exports.validateHeapSnapshotFileOrThrow = validateHeapSnapshotFileOrThrow;
+function createTransientWorkDirFromSingleHeapSnapshot(file) {
+    const config = core_1.MemLabConfig.resetConfigWithTransientDir();
+    core_1.fileManager.createOrOverrideVisitOrderMetaFileForExternalSnapshot(file);
+    return config.workDir;
+}
+exports.createTransientWorkDirFromSingleHeapSnapshot = createTransientWorkDirFromSingleHeapSnapshot;

package/dist/options/heap/HeapAnalysisPluginOption.d.ts

@@ -0,0 +1,20 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+import type { ParsedArgs } from 'minimist';
+import { MemLabConfig } from '@memlab/core';
+import { BaseOption } from '@memlab/core';
+export default class HeapAnalysisPluginOption extends BaseOption {
+    getOptionName(): string;
+    getDescription(): string;
+    parse(config: MemLabConfig, args: ParsedArgs): Promise<{
+        workDir?: string;
+    }>;
+}
+//# sourceMappingURL=HeapAnalysisPluginOption.d.ts.map

package/dist/options/heap/HeapAnalysisPluginOption.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"HeapAnalysisPluginOption.d.ts","sourceRoot":"","sources":["../../../src/options/heap/HeapAnalysisPluginOption.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,UAAU,CAAC;AACzC,OAAO,EAAY,YAAY,EAAC,MAAM,cAAc,CAAC;AACrD,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AAGxC,MAAM,CAAC,OAAO,OAAO,wBAAyB,SAAQ,UAAU;IAC9D,aAAa,IAAI,MAAM;IAIvB,cAAc,IAAI,MAAM;IAOlB,KAAK,CACT,MAAM,EAAE,YAAY,EACpB,IAAI,EAAE,UAAU,GACf,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAC,CAAC;CAS/B"}

package/dist/options/heap/HeapAnalysisPluginOption.js

@@ -0,0 +1,46 @@
+"use strict";
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @oncall web_perf_infra
+ */
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@memlab/core");
+const OptionConstant_1 = __importDefault(require("../lib/OptionConstant"));
+class HeapAnalysisPluginOption extends core_1.BaseOption {
+    getOptionName() {
+        return OptionConstant_1.default.optionNames.HEAP_ANALYSIS_PLUGIN_FILE;
+    }
+    getDescription() {
+        return ('specify the external heap analysis plugin file ' +
+            '(must be a vanilla JS file ended with `Analysis.js` suffix)');
+    }
+    parse(config, args) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const name = this.getOptionName();
+            const ret = {};
+            const heapAnalysisPlugin = args[name];
+            if (heapAnalysisPlugin) {
+                ret.heapAnalysisPlugin = heapAnalysisPlugin;
+            }
+            return ret;
+        });
+    }
+}
+exports.default = HeapAnalysisPluginOption;

package/dist/options/lib/OptionConstant.d.ts

@@ -13,8 +13,10 @@ declare const optionConstants: {
         BASELINE: string;
         CLEAN_UP_SNAPSHOT: string;
         CONTINUS_TEST: string;
+        CONTROL_SNAPSHOT: string;
         CONTROL_WORK_DIR: string;
         CHROMIUM_BINARY: string;
+        CHROMIUM_PROTOCOL_TIMEOUT: string;
         DEVICE: string;
         DISABLE_WEB_SECURITY: string;
         DISABLE_XVFB: string;
@@ -22,6 +24,7 @@ declare const optionConstants: {
         FINAL: string;
         FULL: string;
         HEADFUL: string;
+        HEAP_ANALYSIS_PLUGIN_FILE: string;
         HELP: string;
         IGNORE_LEAK_CLUSTER_SIZE_BELOW: string;
         INTERACTION: string;
@@ -53,6 +56,7 @@ declare const optionConstants: {
         TRACE_ALL_OBJECTS: string;
         TRACE_CONTAINS: string;
         TRACE_OBJECT_SIZE_ABOVE: string;
+        TREATMENT_SNAPSHOT: string;
         TREATMENT_WORK_DIR: string;
         USER_AGENT: string;
         VERBOSE: string;
@@ -73,8 +77,10 @@ declare const _default: {
         BASELINE: string;
         CLEAN_UP_SNAPSHOT: string;
         CONTINUS_TEST: string;
+        CONTROL_SNAPSHOT: string;
         CONTROL_WORK_DIR: string;
         CHROMIUM_BINARY: string;
+        CHROMIUM_PROTOCOL_TIMEOUT: string;
         DEVICE: string;
         DISABLE_WEB_SECURITY: string;
         DISABLE_XVFB: string;
@@ -82,6 +88,7 @@ declare const _default: {
         FINAL: string;
         FULL: string;
         HEADFUL: string;
+        HEAP_ANALYSIS_PLUGIN_FILE: string;
         HELP: string;
         IGNORE_LEAK_CLUSTER_SIZE_BELOW: string;
         INTERACTION: string;
@@ -113,6 +120,7 @@ declare const _default: {
         TRACE_ALL_OBJECTS: string;
         TRACE_CONTAINS: string;
         TRACE_OBJECT_SIZE_ABOVE: string;
+        TREATMENT_SNAPSHOT: string;
         TREATMENT_WORK_DIR: string;
         USER_AGENT: string;
         VERBOSE: string;

package/dist/options/lib/OptionConstant.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"OptionConstant.d.ts","sourceRoot":"","sources":["../../../src/options/lib/OptionConstant.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AA+DH,QAAA,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAGpB,CAAC;AACF,MAAM,MAAM,eAAe,GAAG,OAAO,eAAe,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACrD,wBAIE"}
+{"version":3,"file":"OptionConstant.d.ts","sourceRoot":"","sources":["../../../src/options/lib/OptionConstant.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAmEH,QAAA,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAGpB,CAAC;AACF,MAAM,MAAM,eAAe,GAAG,OAAO,eAAe,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACrD,wBAIE"}

package/dist/options/lib/OptionConstant.js

@@ -15,8 +15,10 @@ const optionNames = {
     BASELINE: 'baseline',
     CLEAN_UP_SNAPSHOT: 'clean-up-snapshot',
     CONTINUS_TEST: 'ContinuousTest',
+    CONTROL_SNAPSHOT: 'control-snapshot',
     CONTROL_WORK_DIR: 'control-work-dir',
     CHROMIUM_BINARY: 'chromium-binary',
+    CHROMIUM_PROTOCOL_TIMEOUT: 'protocol-timeout',
     DEVICE: 'device',
     DISABLE_WEB_SECURITY: 'disable-web-security',
     DISABLE_XVFB: 'disable-xvfb',
@@ -24,6 +26,7 @@ const optionNames = {
     FINAL: 'final',
     FULL: 'full',
     HEADFUL: 'headful',
+    HEAP_ANALYSIS_PLUGIN_FILE: 'analysis-plugin',
     HELP: 'help',
     IGNORE_LEAK_CLUSTER_SIZE_BELOW: 'ignore-leak-cluster-size-below',
     INTERACTION: 'interaction',
@@ -55,6 +58,7 @@ const optionNames = {
     TRACE_ALL_OBJECTS: 'trace-all-objects',
     TRACE_CONTAINS: 'trace-contains',
     TRACE_OBJECT_SIZE_ABOVE: 'trace-object-size-above',
+    TREATMENT_SNAPSHOT: 'treatment-snapshot',
     TREATMENT_WORK_DIR: 'treatment-work-dir',
     USER_AGENT: 'user-agent',
     VERBOSE: 'verbose',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memlab/cli",
-  "version": "1.0.28",
+  "version": "1.0.30",
   "license": "MIT",
   "description": "command line interface for memlab",
   "author": "Liang Gong <lgong@fb.com>",
@@ -26,10 +26,10 @@
     "access": "public"
   },
   "dependencies": {
-    "@memlab/api": "^1.0.11",
-    "@memlab/core": "^1.1.13",
-    "@memlab/e2e": "^1.0.11",
-    "@memlab/heap-analysis": "^1.0.7",
+    "@memlab/api": "^1.0.27",
+    "@memlab/core": "^1.1.28",
+    "@memlab/e2e": "^1.0.28",
+    "@memlab/heap-analysis": "^1.0.25",
     "ansi": "^0.3.1",
     "babar": "^0.2.0",
     "blessed": "^0.1.81",