@vercel/node 2.4.5 → 2.5.0

package/dist/dev-server.js

@@ -70,7 +70,7 @@ const static_config_1 = require("@vercel/static-config");
 const ts_morph_1 = require("ts-morph");
 const esbuild_1 = __importDefault(require("esbuild"));
 const node_fetch_1 = __importDefault(require("node-fetch"));
-const util_1 = require("util");
+const edge_wasm_plugin_1 = require("./edge-wasm-plugin");
 function logError(error) {
     console.error(error.message);
     if (error.stack) {
@@ -127,12 +127,14 @@ async function serializeRequest(message) {
     });
 }
 async function compileUserCode(entrypointPath, entrypointLabel) {
+    const { wasmAssets, plugin: edgeWasmPlugin } = edge_wasm_plugin_1.createEdgeWasmPlugin();
     try {
         const result = await esbuild_1.default.build({
             platform: 'node',
             target: 'node14',
             sourcemap: 'inline',
             bundle: true,
+            plugins: [edgeWasmPlugin],
             entryPoints: [entrypointPath],
             write: false,
             format: 'cjs',
@@ -141,9 +143,8 @@ async function compileUserCode(entrypointPath, entrypointLabel) {
         if (!compiledFile) {
             throw new Error(`Compilation of ${entrypointLabel} produced no output files.`);
         }
-        const userCode = new util_1.TextDecoder().decode(compiledFile.contents);
-        return `
-      ${userCode};
+        const userCode = `
+      ${compiledFile.text};
 
       addEventListener('fetch', async (event) => {
         try {
@@ -189,6 +190,7 @@ async function compileUserCode(entrypointPath, entrypointLabel) {
           }));
         }
       })`;
+        return { userCode, wasmAssets };
     }
     catch (error) {
         // We can't easily show a meaningful stack trace from ncc -> edge-runtime.
@@ -198,13 +200,14 @@ async function compileUserCode(entrypointPath, entrypointLabel) {
         return undefined;
     }
 }
-async function createEdgeRuntime(userCode) {
+async function createEdgeRuntime(params) {
     try {
-        if (!userCode) {
+        if (!params) {
             return undefined;
         }
+        const wasmBindings = await params.wasmAssets.getContext();
         const edgeRuntime = new edge_runtime_1.EdgeRuntime({
-            initialCode: userCode,
+            initialCode: params.userCode,
             extend: (context) => {
                 Object.assign(context, {
                     __dirname: '',
@@ -214,7 +217,7 @@ async function createEdgeRuntime(userCode) {
                     process: {
                         env: process.env,
                     },
-                });
+                }, wasmBindings);
                 return context;
             },
         });

package/dist/edge-wasm-plugin.d.ts

@@ -0,0 +1,21 @@
+import type { Plugin } from 'esbuild';
+export declare class WasmAssets {
+    private readonly assets;
+    /**
+     * Declare a WebAssembly binding
+     */
+    declare(filePath: string): Promise<string>;
+    /**
+     * Get an object with the context needed to execute the code
+     * built with the plugin
+     */
+    getContext(): Promise<Record<string, WebAssembly.Module>>;
+    /**
+     * Allow to iterate easily
+     */
+    [Symbol.iterator](): IterableIterator<[string, string]>;
+}
+export declare function createEdgeWasmPlugin(): {
+    plugin: Plugin;
+    wasmAssets: WasmAssets;
+};

package/dist/edge-wasm-plugin.js

@@ -0,0 +1,82 @@
+"use strict";
+// copied from `edge-functions-bridge`:
+// https://github.com/vercel/runtimes/blob/c076db9e3ce5635f7c2690396e3d9f791a0fd808/packages/edge-functions-bridge/src/get-edge-function-source.ts#L282-L317
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createEdgeWasmPlugin = exports.WasmAssets = void 0;
+const crypto_1 = require("crypto");
+const fs_1 = require("fs");
+class WasmAssets {
+    constructor() {
+        this.assets = new Map();
+    }
+    /**
+     * Declare a WebAssembly binding
+     */
+    async declare(filePath) {
+        const hash = sha1(await fs_1.promises.readFile(filePath));
+        const name = `wasm_${hash}`;
+        this.assets.set(name, filePath);
+        return name;
+    }
+    /**
+     * Get an object with the context needed to execute the code
+     * built with the plugin
+     */
+    async getContext() {
+        const promises = [];
+        const context = {};
+        for (const [name, filePath] of this.assets) {
+            promises.push((async () => {
+                const bytes = await fs_1.promises.readFile(filePath);
+                context[name] = await WebAssembly.compile(bytes);
+            })());
+        }
+        await Promise.all(promises);
+        return context;
+    }
+    /**
+     * Allow to iterate easily
+     */
+    [Symbol.iterator]() {
+        return this.assets[Symbol.iterator]();
+    }
+}
+exports.WasmAssets = WasmAssets;
+function createEdgeWasmPlugin() {
+    const wasmAssets = new WasmAssets();
+    const plugin = {
+        name: 'vercel-wasm',
+        setup(b) {
+            b.onResolve({ filter: /\.wasm\?module/i }, async (data) => {
+                const wasmFile = data.path.replace(/\?module$/, '');
+                const resolvedPath = await b.resolve(wasmFile, {
+                    importer: data.importer,
+                    resolveDir: data.resolveDir,
+                });
+                if (!resolvedPath.path) {
+                    return {
+                        errors: [
+                            { text: `WebAssembly file could not be located: ${wasmFile}` },
+                        ],
+                    };
+                }
+                const name = await wasmAssets.declare(resolvedPath.path);
+                return {
+                    path: name,
+                    namespace: 'vercel-wasm',
+                };
+            });
+            b.onLoad({ namespace: 'vercel-wasm', filter: /.+/ }, args => {
+                return {
+                    loader: 'js',
+                    contents: `export default globalThis[${JSON.stringify(args.path)}]`,
+                };
+            });
+        },
+    };
+    return { plugin, wasmAssets };
+}
+exports.createEdgeWasmPlugin = createEdgeWasmPlugin;
+function sha1(data) {
+    return crypto_1.createHash('sha1').update(data).digest('hex');
+}