@metamask/snaps-execution-environments 0.34.0-flask.1 → 0.35.0-flask.1

package/dist/openrpc.json

@@ -1,210 +0,0 @@
-{
-    "openrpc": "1.2.4",
-    "info": {
-        "title": "MetaMask Snaps Execution Environment API",
-        "version": "0.0.0-development"
-    },
-    "methods": [
-        {
-            "name": "ping",
-            "description": "confirms that a connection has been established with the execution environment",
-            "params": [],
-            "result": {
-                "name": "PingResult",
-                "schema": {
-                    "$ref": "#/components/schemas/OK"
-                }
-            }
-        },
-        {
-            "name": "terminate",
-            "description": "tells the execution environment that it is being shut down",
-            "params": [],
-            "result": {
-                "name": "TerminateResult",
-                "schema": {
-                    "$ref": "#/components/schemas/OK"
-                }
-            }
-        },
-        {
-            "name": "executeSnap",
-            "description": "executes a snap in the environment. The snap can then be interacted with via `snapRpc`",
-            "params": [
-                {
-                    "name": "snapId",
-                    "required": true,
-                    "description": "the name of the snap",
-                    "schema": {
-                        "title": "SnapName",
-                        "type": "string"
-                    }
-                },
-                {
-                    "name": "sourceCode",
-                    "description": "a snaps source code that gets executed in the environment",
-                    "required": true,
-                    "schema": {
-                        "title": "SourceCode",
-                        "type": "string"
-                    }
-                },
-                {
-                    "name": "endowments",
-                    "description": "A list of endowments to provide to SES",
-                    "required": false,
-                    "schema": {
-                        "title": "Endowments",
-                        "description": "An array of the names of the endowments",
-                        "type": "array",
-                        "items": {
-                            "title": "Endowment",
-                            "type": "string"
-                        }
-                    }
-                }
-            ],
-            "result": {
-                "name": "executeSnapResult",
-                "schema": {
-                    "$ref": "#/components/schemas/OK"
-                }
-            },
-            "examples": [
-                {
-                    "name": "BasicTestSnapExecuteExample",
-                    "params": [
-                        {
-                            "name": "snapId",
-                            "value": "TestSnap"
-                        },
-                        {
-                            "name": "sourceCode",
-                            "value": "module.exports.onRpcRequest = async ({ request }) => { return request.method + request.id }"
-                        }
-                    ],
-                    "result": {
-                        "name": "BasicTestSnapExampleResult",
-                        "value": "OK"
-                    }
-                }
-            ]
-        },
-        {
-            "name": "snapRpc",
-            "params": [
-                {
-                    "name": "target",
-                    "required": true,
-                    "description": "the name of the snap",
-                    "schema": {
-                        "title": "Target",
-                        "type": "string"
-                    }
-                },
-                {
-                    "name": "handler",
-                    "required": true,
-                    "description": "the handler to trigger on the snap",
-                    "schema": {
-                        "title": "Handler",
-                        "type": "string",
-                        "enum": ["onRpcRequest", "onTransaction", "keyring"]
-                    }
-                },
-                {
-                    "name": "origin",
-                    "required": true,
-                    "description": "Origin of the snap JSON-RPC request",
-                    "schema": {
-                        "title": "Origin",
-                        "type": "string"
-                    }
-                },
-                {
-                    "name": "request",
-                    "required": true,
-                    "description": "JSON-RPC request",
-                    "schema": {
-                        "$ref": "#/components/schemas/JsonRpcRequest"
-                    }
-                }
-            ],
-            "result": {
-                "name": "HandleSnapRpcResult",
-                "schema": {
-                    "title": "SnapRpcResult"
-                }
-            },
-            "examples": [
-                {
-                    "name": "TestSnapExample",
-                    "params": [
-                        {
-                            "name": "target",
-                            "value": "TestSnap"
-                        },
-                        {
-                            "name": "origin",
-                            "value": "foo.com"
-                        },
-                        {
-                            "name": "request",
-                            "value": {
-                                "jsonrpc": "2.0",
-                                "method": "hello",
-                                "params": [],
-                                "id": 1
-                            }
-                        }
-                    ],
-                    "result": {
-                        "name": "TestSnapResultExample",
-                        "value": "hello1"
-                    }
-                }
-            ]
-        }
-    ],
-    "components": {
-        "schemas": {
-            "OK": {
-                "title": "OK",
-                "type": "string",
-                "const": "OK"
-            },
-            "JsonRpcRequest": {
-                "title": "JsonRpcRequest",
-                "type": "object",
-                "required": ["jsonrpc", "method"],
-                "properties": {
-                    "jsonrpc": {
-                        "title": "JSONRPCString",
-                        "type": "string",
-                        "const": "2.0"
-                    },
-                    "id": {
-                        "title": "JSONRPCID",
-                        "oneOf": [
-                            {
-                                "type": "string"
-                            },
-                            {
-                                "type": "number"
-                            }
-                        ]
-                    },
-                    "method": {
-                        "title": "JSONRPCMethod",
-                        "description": "the name of the method",
-                        "type": "string"
-                    },
-                    "params": {
-                        "title": "JSONRPCParams",
-                        "type": ["array", "object"]
-                    }
-                }
-            }
-        }
-    }
-}

package/dist/webworker/executor/WebWorkerSnapExecutor.js

@@ -1,37 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.WebWorkerSnapExecutor = void 0;
-const object_multiplex_1 = __importDefault(require("@metamask/object-multiplex"));
-const post_message_stream_1 = require("@metamask/post-message-stream");
-const snaps_utils_1 = require("@metamask/snaps-utils");
-const pump_1 = __importDefault(require("pump"));
-const BaseSnapExecutor_1 = require("../../common/BaseSnapExecutor");
-const logging_1 = require("../../logging");
-class WebWorkerSnapExecutor extends BaseSnapExecutor_1.BaseSnapExecutor {
-    /**
-     * Initialize the WebWorkerSnapExecutor. This creates a post message stream
-     * from and to the parent window, for two-way communication with the iframe.
-     *
-     * @param stream - The stream to use for communication.
-     * @returns An instance of `WebWorkerSnapExecutor`, with the initialized post
-     * message streams.
-     */
-    static initialize(stream = new post_message_stream_1.WebWorkerPostMessageStream()) {
-        (0, logging_1.log)('Worker: Connecting to parent.');
-        const mux = new object_multiplex_1.default();
-        (0, pump_1.default)(stream, mux, stream, (error) => {
-            if (error) {
-                (0, snaps_utils_1.logError)(`Parent stream failure, closing worker.`, error);
-            }
-            self.close();
-        });
-        const commandStream = mux.createStream(snaps_utils_1.SNAP_STREAM_NAMES.COMMAND);
-        const rpcStream = mux.createStream(snaps_utils_1.SNAP_STREAM_NAMES.JSON_RPC);
-        return new WebWorkerSnapExecutor(commandStream, rpcStream);
-    }
-}
-exports.WebWorkerSnapExecutor = WebWorkerSnapExecutor;
-//# sourceMappingURL=WebWorkerSnapExecutor.js.map

package/dist/webworker/executor/WebWorkerSnapExecutor.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"WebWorkerSnapExecutor.js","sourceRoot":"","sources":["../../../src/webworker/executor/WebWorkerSnapExecutor.ts"],"names":[],"mappings":";;;;;;AAAA,kFAAyD;AACzD,uEAGuC;AACvC,uDAAoE;AACpE,gDAAwB;AAExB,oEAAiE;AACjE,2CAAoC;AAEpC,MAAa,qBAAsB,SAAQ,mCAAgB;IACzD;;;;;;;OAOG;IACH,MAAM,CAAC,UAAU,CACf,SAAgC,IAAI,gDAA0B,EAAE;QAEhE,IAAA,aAAG,EAAC,+BAA+B,CAAC,CAAC;QAErC,MAAM,GAAG,GAAG,IAAI,0BAAe,EAAE,CAAC;QAClC,IAAA,cAAI,EAAC,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAClC,IAAI,KAAK,EAAE;gBACT,IAAA,sBAAQ,EAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;aAC3D;YACD,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,GAAG,CAAC,YAAY,CAAC,+BAAiB,CAAC,OAAO,CAAC,CAAC;QAClE,MAAM,SAAS,GAAG,GAAG,CAAC,YAAY,CAAC,+BAAiB,CAAC,QAAQ,CAAC,CAAC;QAE/D,OAAO,IAAI,qBAAqB,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IAC7D,CAAC;CACF;AA3BD,sDA2BC","sourcesContent":["import ObjectMultiplex from '@metamask/object-multiplex';\nimport {\n  BasePostMessageStream,\n  WebWorkerPostMessageStream,\n} from '@metamask/post-message-stream';\nimport { logError, SNAP_STREAM_NAMES } from '@metamask/snaps-utils';\nimport pump from 'pump';\n\nimport { BaseSnapExecutor } from '../../common/BaseSnapExecutor';\nimport { log } from '../../logging';\n\nexport class WebWorkerSnapExecutor extends BaseSnapExecutor {\n  /**\n   * Initialize the WebWorkerSnapExecutor. This creates a post message stream\n   * from and to the parent window, for two-way communication with the iframe.\n   *\n   * @param stream - The stream to use for communication.\n   * @returns An instance of `WebWorkerSnapExecutor`, with the initialized post\n   * message streams.\n   */\n  static initialize(\n    stream: BasePostMessageStream = new WebWorkerPostMessageStream(),\n  ) {\n    log('Worker: Connecting to parent.');\n\n    const mux = new ObjectMultiplex();\n    pump(stream, mux, stream, (error) => {\n      if (error) {\n        logError(`Parent stream failure, closing worker.`, error);\n      }\n      self.close();\n    });\n\n    const commandStream = mux.createStream(SNAP_STREAM_NAMES.COMMAND);\n    const rpcStream = mux.createStream(SNAP_STREAM_NAMES.JSON_RPC);\n\n    return new WebWorkerSnapExecutor(commandStream, rpcStream);\n  }\n}\n"]}

package/dist/webworker/executor/index.js

@@ -1,10 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const lockdown_events_1 = require("../../common/lockdown/lockdown-events");
-const lockdown_more_1 = require("../../common/lockdown/lockdown-more");
-const WebWorkerSnapExecutor_1 = require("./WebWorkerSnapExecutor");
-// Lockdown is already applied in LavaMoat
-(0, lockdown_more_1.executeLockdownMore)();
-(0, lockdown_events_1.executeLockdownEvents)();
-WebWorkerSnapExecutor_1.WebWorkerSnapExecutor.initialize();
-//# sourceMappingURL=index.js.map

package/dist/webworker/executor/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/webworker/executor/index.ts"],"names":[],"mappings":";;AAAA,2EAA8E;AAC9E,uEAA0E;AAC1E,mEAAgE;AAEhE,0CAA0C;AAC1C,IAAA,mCAAmB,GAAE,CAAC;AACtB,IAAA,uCAAqB,GAAE,CAAC;AAExB,6CAAqB,CAAC,UAAU,EAAE,CAAC","sourcesContent":["import { executeLockdownEvents } from '../../common/lockdown/lockdown-events';\nimport { executeLockdownMore } from '../../common/lockdown/lockdown-more';\nimport { WebWorkerSnapExecutor } from './WebWorkerSnapExecutor';\n\n// Lockdown is already applied in LavaMoat\nexecuteLockdownMore();\nexecuteLockdownEvents();\n\nWebWorkerSnapExecutor.initialize();\n"]}

package/dist/webworker/pool/WebWorkerPool.js

@@ -1,168 +0,0 @@
-"use strict";
-var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
-    if (kind === "m") throw new TypeError("Private method is not writable");
-    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
-    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
-    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
-};
-var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
-    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
-    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
-    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
-};
-var _WebWorkerPool_instances, _WebWorkerPool_poolSize, _WebWorkerPool_stream, _WebWorkerPool_url, _WebWorkerPool_workerSourceURL, _WebWorkerPool_onData, _WebWorkerPool_initializeJob, _WebWorkerPool_terminateJob, _WebWorkerPool_getWorker, _WebWorkerPool_updatePool, _WebWorkerPool_createWorker, _WebWorkerPool_getWorkerURL;
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.WebWorkerPool = void 0;
-const post_message_stream_1 = require("@metamask/post-message-stream");
-const snaps_utils_1 = require("@metamask/snaps-utils");
-const utils_1 = require("@metamask/utils");
-const non_secure_1 = require("nanoid/non-secure");
-/**
- * A snap executor using the WebWorker API.
- *
- * This is not a traditional snap executor, as it does not execute snaps itself.
- * Instead, it creates a pool of webworkers for each snap execution, and sends
- * the snap execution request to the webworker. The webworker is responsible for
- * executing the snap.
- */
-class WebWorkerPool {
-    constructor(stream, url, poolSize = 3) {
-        _WebWorkerPool_instances.add(this);
-        _WebWorkerPool_poolSize.set(this, void 0);
-        _WebWorkerPool_stream.set(this, void 0);
-        _WebWorkerPool_url.set(this, void 0);
-        this.pool = [];
-        this.jobs = new Map();
-        _WebWorkerPool_workerSourceURL.set(this, void 0);
-        __classPrivateFieldSet(this, _WebWorkerPool_stream, stream, "f");
-        __classPrivateFieldSet(this, _WebWorkerPool_url, url, "f");
-        __classPrivateFieldSet(this, _WebWorkerPool_poolSize, poolSize, "f");
-        __classPrivateFieldGet(this, _WebWorkerPool_stream, "f").on('data', __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_onData).bind(this));
-    }
-    /* istanbul ignore next - Constructor arguments. */
-    static initialize(stream = new post_message_stream_1.WindowPostMessageStream({
-        name: 'child',
-        target: 'parent',
-        targetWindow: self.parent,
-        targetOrigin: '*',
-    }), url = '../executor/bundle.js', poolSize) {
-        return new WebWorkerPool(stream, url, poolSize);
-    }
-}
-exports.WebWorkerPool = WebWorkerPool;
-_WebWorkerPool_poolSize = new WeakMap(), _WebWorkerPool_stream = new WeakMap(), _WebWorkerPool_url = new WeakMap(), _WebWorkerPool_workerSourceURL = new WeakMap(), _WebWorkerPool_instances = new WeakSet(), _WebWorkerPool_onData = function _WebWorkerPool_onData(data) {
-    const { jobId, data: request } = data;
-    const job = this.jobs.get(jobId);
-    if (!job) {
-        // This ensures that a job is initialized before it is used. To avoid
-        // code duplication, we call the `#onData` method again, which will
-        // run the rest of the logic after initialization.
-        __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_initializeJob).call(this, jobId)
-            .then(() => {
-            __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_onData).call(this, data);
-        })
-            .catch((error) => {
-            (0, snaps_utils_1.logError)('[Worker] Error initializing job:', error.toString());
-            __classPrivateFieldGet(this, _WebWorkerPool_stream, "f").write({
-                jobId,
-                data: {
-                    name: 'command',
-                    data: {
-                        jsonrpc: '2.0',
-                        id: request.id ?? null,
-                        error: {
-                            code: -32000,
-                            message: 'Internal error',
-                        },
-                    },
-                },
-            });
-        });
-        return;
-    }
-    // This is a method specific to the `WebWorkerPool`, as the service itself
-    // does not have access to the workers directly.
-    if (request.method === 'terminateJob') {
-        __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_terminateJob).call(this, jobId);
-        return;
-    }
-    job.stream.write(request);
-}, _WebWorkerPool_initializeJob = 
-/**
- * Create a new worker and set up a stream to communicate with it.
- *
- * @param jobId - The job ID.
- * @returns The job.
- */
-async function _WebWorkerPool_initializeJob(jobId) {
-    const worker = await __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_getWorker).call(this);
-    const jobStream = new post_message_stream_1.WebWorkerParentPostMessageStream({
-        worker,
-    });
-    // Write messages from the worker to the parent, wrapped with the job ID.
-    jobStream.on('data', (data) => {
-        __classPrivateFieldGet(this, _WebWorkerPool_stream, "f").write({ data, jobId });
-    });
-    const job = { id: jobId, worker, stream: jobStream };
-    this.jobs.set(jobId, job);
-    return job;
-}, _WebWorkerPool_terminateJob = function _WebWorkerPool_terminateJob(jobId) {
-    const job = this.jobs.get(jobId);
-    (0, utils_1.assert)(job, `Job "${jobId}" not found.`);
-    job.stream.destroy();
-    job.worker.terminate();
-    this.jobs.delete(jobId);
-}, _WebWorkerPool_getWorker = 
-/**
- * Get a worker from the pool. A new worker will be created automatically.
- *
- * @returns The worker.
- */
-async function _WebWorkerPool_getWorker() {
-    // Lazily create the pool of workers.
-    if (this.pool.length === 0) {
-        await __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_updatePool).call(this);
-    }
-    const worker = this.pool.shift();
-    (0, utils_1.assert)(worker, 'Worker not found.');
-    await __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_updatePool).call(this);
-    return worker;
-}, _WebWorkerPool_updatePool = 
-/**
- * Update the pool of workers. This will create new workers if the pool is
- * below the minimum size.
- */
-async function _WebWorkerPool_updatePool() {
-    while (this.pool.length < __classPrivateFieldGet(this, _WebWorkerPool_poolSize, "f")) {
-        const worker = await __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_createWorker).call(this);
-        this.pool.push(worker);
-    }
-}, _WebWorkerPool_createWorker = 
-/**
- * Create a new worker. This will fetch the worker source if it has not
- * already been fetched.
- *
- * @returns The worker.
- */
-async function _WebWorkerPool_createWorker() {
-    return new Worker(await __classPrivateFieldGet(this, _WebWorkerPool_instances, "m", _WebWorkerPool_getWorkerURL).call(this), {
-        name: `worker-${(0, non_secure_1.nanoid)()}`,
-    });
-}, _WebWorkerPool_getWorkerURL = 
-/**
- * Get the URL of the worker source. This will fetch the worker source if it
- * has not already been fetched.
- *
- * @returns The worker source URL, as a `blob:` URL.
- */
-async function _WebWorkerPool_getWorkerURL() {
-    if (__classPrivateFieldGet(this, _WebWorkerPool_workerSourceURL, "f")) {
-        return __classPrivateFieldGet(this, _WebWorkerPool_workerSourceURL, "f");
-    }
-    const blob = await fetch(__classPrivateFieldGet(this, _WebWorkerPool_url, "f"))
-        .then(async (response) => response.blob())
-        .then(URL.createObjectURL.bind(URL));
-    __classPrivateFieldSet(this, _WebWorkerPool_workerSourceURL, blob, "f");
-    return blob;
-};
-//# sourceMappingURL=WebWorkerPool.js.map

package/dist/webworker/pool/WebWorkerPool.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"WebWorkerPool.js","sourceRoot":"","sources":["../../../src/webworker/pool/WebWorkerPool.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,uEAIuC;AACvC,uDAAiD;AACjD,2CAAyD;AACzD,kDAA2C;AAQ3C;;;;;;;GAOG;AACH,MAAa,aAAa;IA2BxB,YAAY,MAA6B,EAAE,GAAW,EAAE,QAAQ,GAAG,CAAC;;QA1BpE,0CAAmB;QAEnB,wCAAwC;QAExC,qCAAsB;QAEb,SAAI,GAAa,EAAE,CAAC;QAEpB,SAAI,GAA6B,IAAI,GAAG,EAAE,CAAC;QAEpD,iDAA0B;QAiBxB,uBAAA,IAAI,yBAAW,MAAM,MAAA,CAAC;QACtB,uBAAA,IAAI,sBAAQ,GAAG,MAAA,CAAC;QAChB,uBAAA,IAAI,2BAAa,QAAQ,MAAA,CAAC;QAE1B,uBAAA,IAAI,6BAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,uBAAA,IAAI,uDAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACnD,CAAC;IApBD,mDAAmD;IACnD,MAAM,CAAC,UAAU,CACf,SAAgC,IAAI,6CAAuB,CAAC;QAC1D,IAAI,EAAE,OAAO;QACb,MAAM,EAAE,QAAQ;QAChB,YAAY,EAAE,IAAI,CAAC,MAAM;QACzB,YAAY,EAAE,GAAG;KAClB,CAAC,EACF,GAAG,GAAG,uBAAuB,EAC7B,QAAiB;QAEjB,OAAO,IAAI,aAAa,CAAC,MAAM,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;IAClD,CAAC;CAiKF;AA1LD,sCA0LC;qQA9IS,IAA6C;IACnD,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAEtC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,GAAG,EAAE;QACR,qEAAqE;QACrE,mEAAmE;QACnE,kDAAkD;QAClD,uBAAA,IAAI,8DAAe,MAAnB,IAAI,EAAgB,KAAK,CAAC;aACvB,IAAI,CAAC,GAAG,EAAE;YACT,uBAAA,IAAI,uDAAQ,MAAZ,IAAI,EAAS,IAAI,CAAC,CAAC;QACrB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,IAAA,sBAAQ,EAAC,kCAAkC,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;YAE/D,uBAAA,IAAI,6BAAQ,CAAC,KAAK,CAAC;gBACjB,KAAK;gBACL,IAAI,EAAE;oBACJ,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE;wBACJ,OAAO,EAAE,KAAK;wBACd,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,IAAI;wBACtB,KAAK,EAAE;4BACL,IAAI,EAAE,CAAC,KAAK;4BACZ,OAAO,EAAE,gBAAgB;yBAC1B;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEL,OAAO;KACR;IAED,0EAA0E;IAC1E,gDAAgD;IAChD,IAAI,OAAO,CAAC,MAAM,KAAK,cAAc,EAAE;QACrC,uBAAA,IAAI,6DAAc,MAAlB,IAAI,EAAe,KAAK,CAAC,CAAC;QAC1B,OAAO;KACR;IAED,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AAC5B,CAAC;AAED;;;;;GAKG;AACH,KAAK,uCAAgB,KAAa;IAChC,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,0DAAW,MAAf,IAAI,CAAa,CAAC;IACvC,MAAM,SAAS,GAAG,IAAI,sDAAgC,CAAC;QACrD,MAAM;KACP,CAAC,CAAC;IAEH,yEAAyE;IACzE,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;QAC5B,uBAAA,IAAI,6BAAQ,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IACrD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC1B,OAAO,GAAG,CAAC;AACb,CAAC,qEAQa,KAAa;IACzB,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAA,cAAM,EAAC,GAAG,EAAE,QAAQ,KAAK,cAAc,CAAC,CAAC;IAEzC,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;IACrB,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;IAEvB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC1B,CAAC;AAED;;;;GAIG;AACH,KAAK;IACH,qCAAqC;IACrC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE;QAC1B,MAAM,uBAAA,IAAI,2DAAY,MAAhB,IAAI,CAAc,CAAC;KAC1B;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;IACjC,IAAA,cAAM,EAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;IAEpC,MAAM,uBAAA,IAAI,2DAAY,MAAhB,IAAI,CAAc,CAAC;IAEzB,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,KAAK;IACH,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,uBAAA,IAAI,+BAAU,EAAE;QACxC,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,6DAAc,MAAlB,IAAI,CAAgB,CAAC;QAC1C,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;KACxB;AACH,CAAC;AAED;;;;;GAKG;AACH,KAAK;IACH,OAAO,IAAI,MAAM,CAAC,MAAM,uBAAA,IAAI,6DAAc,MAAlB,IAAI,CAAgB,EAAE;QAC5C,IAAI,EAAE,UAAU,IAAA,mBAAM,GAAE,EAAE;KAC3B,CAAC,CAAC;AACL,CAAC;AAED;;;;;GAKG;AACH,KAAK;IACH,IAAI,uBAAA,IAAI,sCAAiB,EAAE;QACzB,OAAO,uBAAA,IAAI,sCAAiB,CAAC;KAC9B;IAED,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,uBAAA,IAAI,0BAAK,CAAC;SAChC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;SACzC,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAEvC,uBAAA,IAAI,kCAAoB,IAAI,MAAA,CAAC;IAC7B,OAAO,IAAI,CAAC;AACd,CAAC","sourcesContent":["import {\n  BasePostMessageStream,\n  WebWorkerParentPostMessageStream,\n  WindowPostMessageStream,\n} from '@metamask/post-message-stream';\nimport { logError } from '@metamask/snaps-utils';\nimport { JsonRpcRequest, assert } from '@metamask/utils';\nimport { nanoid } from 'nanoid/non-secure';\n\ntype ExecutorJob = {\n  id: string;\n  worker: Worker;\n  stream: WebWorkerParentPostMessageStream;\n};\n\n/**\n * A snap executor using the WebWorker API.\n *\n * This is not a traditional snap executor, as it does not execute snaps itself.\n * Instead, it creates a pool of webworkers for each snap execution, and sends\n * the snap execution request to the webworker. The webworker is responsible for\n * executing the snap.\n */\nexport class WebWorkerPool {\n  readonly #poolSize;\n\n  readonly #stream: BasePostMessageStream;\n\n  readonly #url: string;\n\n  readonly pool: Worker[] = [];\n\n  readonly jobs: Map<string, ExecutorJob> = new Map();\n\n  #workerSourceURL?: string;\n\n  /* istanbul ignore next - Constructor arguments. */\n  static initialize(\n    stream: BasePostMessageStream = new WindowPostMessageStream({\n      name: 'child',\n      target: 'parent',\n      targetWindow: self.parent,\n      targetOrigin: '*',\n    }),\n    url = '../executor/bundle.js',\n    poolSize?: number,\n  ) {\n    return new WebWorkerPool(stream, url, poolSize);\n  }\n\n  constructor(stream: BasePostMessageStream, url: string, poolSize = 3) {\n    this.#stream = stream;\n    this.#url = url;\n    this.#poolSize = poolSize;\n\n    this.#stream.on('data', this.#onData.bind(this));\n  }\n\n  /**\n   * Handle an incoming message from the `WebWorkerExecutionService`. This\n   * assumes that the message contains a `jobId` property, and a JSON-RPC\n   * request in the `data` property.\n   *\n   * @param data - The message data.\n   * @param data.data - The JSON-RPC request.\n   * @param data.jobId - The job ID.\n   */\n  #onData(data: { data: JsonRpcRequest; jobId: string }) {\n    const { jobId, data: request } = data;\n\n    const job = this.jobs.get(jobId);\n    if (!job) {\n      // This ensures that a job is initialized before it is used. To avoid\n      // code duplication, we call the `#onData` method again, which will\n      // run the rest of the logic after initialization.\n      this.#initializeJob(jobId)\n        .then(() => {\n          this.#onData(data);\n        })\n        .catch((error) => {\n          logError('[Worker] Error initializing job:', error.toString());\n\n          this.#stream.write({\n            jobId,\n            data: {\n              name: 'command',\n              data: {\n                jsonrpc: '2.0',\n                id: request.id ?? null,\n                error: {\n                  code: -32000,\n                  message: 'Internal error',\n                },\n              },\n            },\n          });\n        });\n\n      return;\n    }\n\n    // This is a method specific to the `WebWorkerPool`, as the service itself\n    // does not have access to the workers directly.\n    if (request.method === 'terminateJob') {\n      this.#terminateJob(jobId);\n      return;\n    }\n\n    job.stream.write(request);\n  }\n\n  /**\n   * Create a new worker and set up a stream to communicate with it.\n   *\n   * @param jobId - The job ID.\n   * @returns The job.\n   */\n  async #initializeJob(jobId: string): Promise<ExecutorJob> {\n    const worker = await this.#getWorker();\n    const jobStream = new WebWorkerParentPostMessageStream({\n      worker,\n    });\n\n    // Write messages from the worker to the parent, wrapped with the job ID.\n    jobStream.on('data', (data) => {\n      this.#stream.write({ data, jobId });\n    });\n\n    const job = { id: jobId, worker, stream: jobStream };\n    this.jobs.set(jobId, job);\n    return job;\n  }\n\n  /**\n   * Terminate the job with the given ID. This will close the worker and delete\n   * the job from the internal job map.\n   *\n   * @param jobId - The job ID.\n   */\n  #terminateJob(jobId: string) {\n    const job = this.jobs.get(jobId);\n    assert(job, `Job \"${jobId}\" not found.`);\n\n    job.stream.destroy();\n    job.worker.terminate();\n\n    this.jobs.delete(jobId);\n  }\n\n  /**\n   * Get a worker from the pool. A new worker will be created automatically.\n   *\n   * @returns The worker.\n   */\n  async #getWorker() {\n    // Lazily create the pool of workers.\n    if (this.pool.length === 0) {\n      await this.#updatePool();\n    }\n\n    const worker = this.pool.shift();\n    assert(worker, 'Worker not found.');\n\n    await this.#updatePool();\n\n    return worker;\n  }\n\n  /**\n   * Update the pool of workers. This will create new workers if the pool is\n   * below the minimum size.\n   */\n  async #updatePool() {\n    while (this.pool.length < this.#poolSize) {\n      const worker = await this.#createWorker();\n      this.pool.push(worker);\n    }\n  }\n\n  /**\n   * Create a new worker. This will fetch the worker source if it has not\n   * already been fetched.\n   *\n   * @returns The worker.\n   */\n  async #createWorker() {\n    return new Worker(await this.#getWorkerURL(), {\n      name: `worker-${nanoid()}`,\n    });\n  }\n\n  /**\n   * Get the URL of the worker source. This will fetch the worker source if it\n   * has not already been fetched.\n   *\n   * @returns The worker source URL, as a `blob:` URL.\n   */\n  async #getWorkerURL() {\n    if (this.#workerSourceURL) {\n      return this.#workerSourceURL;\n    }\n\n    const blob = await fetch(this.#url)\n      .then(async (response) => response.blob())\n      .then(URL.createObjectURL.bind(URL));\n\n    this.#workerSourceURL = blob;\n    return blob;\n  }\n}\n"]}

package/dist/webworker/pool/index.js

@@ -1,10 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const lockdown_events_1 = require("../../common/lockdown/lockdown-events");
-const lockdown_more_1 = require("../../common/lockdown/lockdown-more");
-const WebWorkerPool_1 = require("./WebWorkerPool");
-// Lockdown is already applied in LavaMoat
-(0, lockdown_more_1.executeLockdownMore)();
-(0, lockdown_events_1.executeLockdownEvents)();
-WebWorkerPool_1.WebWorkerPool.initialize();
-//# sourceMappingURL=index.js.map

package/dist/webworker/pool/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/webworker/pool/index.ts"],"names":[],"mappings":";;AAAA,2EAA8E;AAC9E,uEAA0E;AAC1E,mDAAgD;AAEhD,0CAA0C;AAC1C,IAAA,mCAAmB,GAAE,CAAC;AACtB,IAAA,uCAAqB,GAAE,CAAC;AAExB,6BAAa,CAAC,UAAU,EAAE,CAAC","sourcesContent":["import { executeLockdownEvents } from '../../common/lockdown/lockdown-events';\nimport { executeLockdownMore } from '../../common/lockdown/lockdown-more';\nimport { WebWorkerPool } from './WebWorkerPool';\n\n// Lockdown is already applied in LavaMoat\nexecuteLockdownMore();\nexecuteLockdownEvents();\n\nWebWorkerPool.initialize();\n"]}