All files / omnium-gatherum/src offscreen.ts

0% Statements 0/18
0% Branches 0/2
0% Functions 0/4
0% Lines 0/17

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88                                                                                                                                                                               
import {
  makeIframeVatWorker,
  PlatformServicesServer,
  createRelayQueryString,
} from '@metamask/kernel-browser-runtime';
import { delay, isJsonRpcMessage } from '@metamask/kernel-utils';
import type { JsonRpcMessage } from '@metamask/kernel-utils';
import { Logger } from '@metamask/logger';
import type { DuplexStream } from '@metamask/streams';
import {
  initializeMessageChannel,
  ChromeRuntimeDuplexStream,
  MessagePortDuplexStream,
} from '@metamask/streams/browser';
import type { PostMessageTarget } from '@metamask/streams/browser';
 
const logger = new Logger('offscreen');
 
main().catch(logger.error);
 
/**
 * Main function to initialize the offscreen document.
 */
async function main(): Promise<void> {
  // Without this delay, sending messages via the chrome.runtime API can fail.
  await delay(50);
 
  // Create stream for CapTP messages from the background script
  const backgroundStream = await ChromeRuntimeDuplexStream.make<
    JsonRpcMessage,
    JsonRpcMessage
  >(chrome.runtime, 'offscreen', 'background', isJsonRpcMessage);
 
  const kernelStream = await makeKernelWorker();
 
  // Handle messages from the background script / kernel
  await Promise.all([
    kernelStream.pipe(backgroundStream),
    backgroundStream.pipe(kernelStream),
  ]);
}
 
/**
 * Creates and initializes the kernel worker.
 *
 * @returns The message port stream for worker communication
 */
async function makeKernelWorker(): Promise<
  DuplexStream<JsonRpcMessage, JsonRpcMessage>
> {
  // Assign local relay address generated from `yarn ocap relay`
  const relayQueryString = createRelayQueryString([
    '/ip4/127.0.0.1/tcp/9001/ws/p2p/12D3KooWJBDqsyHQF2MWiCdU4kdqx4zTsSTLRdShg7Ui6CRWB4uc',
  ]);
 
  const workerUrlParams = new URLSearchParams(relayQueryString);
  workerUrlParams.set('reset-storage', process.env.RESET_STORAGE ?? 'false');
 
  const workerUrl = new URL('kernel-worker.js', import.meta.url);
  workerUrl.search = workerUrlParams.toString();
 
  const worker = new Worker(workerUrl, {
    type: 'module',
  });
 
  const port = await initializeMessageChannel((message, transfer) =>
    worker.postMessage(message, transfer),
  );
 
  const kernelStream = await MessagePortDuplexStream.make<
    JsonRpcMessage,
    JsonRpcMessage
  >(port, isJsonRpcMessage);
 
  await PlatformServicesServer.make(worker as PostMessageTarget, (vatId) =>
    makeIframeVatWorker({
      id: vatId,
      iframeUri: 'iframe.html',
      getPort: initializeMessageChannel,
      logger: logger.subLogger({
        tags: ['iframe-vat-worker', vatId],
      }),
    }),
  );
 
  return kernelStream;
}