Function transform

tfjs-backend-wasm/scripts/patch_wechat_webassembly.ts:27–61

(code, file)

  return {
    name: 'patchWechatWebAssembly',
    transform(code, file) {
      // remove node imports
      if (
        file.endsWith('tfjs-backend-wasm-threaded-simd.worker.js') ||
        file.endsWith('tfjs-backend-wasm-threaded-simd.js')
      ) {
        code = code.replace(`require("worker_threads")`, 'null');
        code = code.replace(`require("perf_hooks")`, 'null');
      }

      // it is not a nice way, but WebAssembly.validate not working and SIMD is not support in WXWebAssembly
      // tf.env().set('WASM_HAS_SIMD_SUPPORT', false) will be done in tfjs-wechat or application code
      // so does the WASM_HAS_MULTITHREAD_SUPPORT
      if (file.endsWith('backend_wasm.ts')) {
        const ast = (this as {parse: (code: string) => BaseNode}).parse(code);
        walk(ast, {
          enter(node: NodeType) {
            if (
              node.type === 'FunctionDeclaration' &&
              'id' in node &&
              node.id.name === 'createInstantiateWasmFunc'
            ) {
              code = code.replace(
                code.slice(node.start, node.end),
                createInstantiateWasmFuncString,
              );
            }
          },
        });
      }

      code = code.replace(/WebAssembly\./g, `WXWebAssembly.`);
      code = code.replace(/typeof WebAssembly/g, `typeof WXWebAssembly`);
      return { code, map: null };
    },
  };
}