Function patchWebAssemblyWithForwarding

packages/wasm/src/index.ts:168–201

* Patches the WebAssembly object in the worker scope and forwards * registered modules to the parent thread.

(workerSelf: MinimalDedicatedWorkerGlobalScope)

 * registered modules to the parent thread.
 */
function patchWebAssemblyWithForwarding(workerSelf: MinimalDedicatedWorkerGlobalScope): void {
  if ('instantiateStreaming' in WebAssembly) {
    const origInstantiateStreaming = WebAssembly.instantiateStreaming;
    WebAssembly.instantiateStreaming = function instantiateStreaming(
      response: Response | PromiseLike<Response>,
      importObject: WebAssembly.Imports,
    ): Promise<WebAssembly.Module> {
      return Promise.resolve(response).then(response => {
        return origInstantiateStreaming(response, importObject).then(rv => {
          if (response.url) {
            registerModuleAndForward(rv.module, response.url, workerSelf);
          }
          return rv;
        });
      });
    } as typeof WebAssembly.instantiateStreaming;
  }

  if ('compileStreaming' in WebAssembly) {
    const origCompileStreaming = WebAssembly.compileStreaming;
    WebAssembly.compileStreaming = function compileStreaming(
      source: Response | Promise<Response>,
    ): Promise<WebAssembly.Module> {
      return Promise.resolve(source).then(response => {
        return origCompileStreaming(response).then(module => {
          if (response.url) {
            registerModuleAndForward(module, response.url, workerSelf);
          }
          return module;
        });
      });
    } as typeof WebAssembly.compileStreaming;
  }
}

/**
 * Registers a WASM module and forwards its debug image to the parent thread.