ChildProcessSnapExecutor.ts

import ObjectMultiplex from '@metamask/object-multiplex';
import { ProcessMessageStream } from '@metamask/post-message-stream/node';
import { logError, SNAP_STREAM_NAMES } from '@metamask/snaps-utils';
import { pipeline } from 'readable-stream';

import { BaseSnapExecutor } from '../common/BaseSnapExecutor';
import { log } from '../logging';

export class ChildProcessSnapExecutor extends BaseSnapExecutor {
  static initialize() {
    log('Worker: Connecting to parent.');

    const parentStream = new ProcessMessageStream();
    const mux = new ObjectMultiplex();
    pipeline(parentStream, mux as any, parentStream, (error) => {
      if (error) {
        logError(`Parent stream failure, closing worker.`, error);
      }
      self.close();
    });

    const commandStream = mux.createStream(SNAP_STREAM_NAMES.COMMAND);
    const rpcStream = mux.createStream(SNAP_STREAM_NAMES.JSON_RPC) as any;
    return new ChildProcessSnapExecutor(commandStream, rpcStream);
  }
}