feat: shim additional native modules

orinokai · orinokai · commit 409354dc9a76 · 2023-05-02T19:03:54.000+01:00
diff --git a/packages/runtime/src/templates/edge/shims.js b/packages/runtime/src/templates/edge/shims.js
@@ -1,8 +1,11 @@
 // @ts-check
-// deno-lint-ignore-file no-var prefer-const no-unused-vars no-explicit-any
+// deno-lint-ignore-file prefer-const no-unused-vars
 import { decode as _base64Decode } from 'https://deno.land/std@0.175.0/encoding/base64.ts'
-import { AsyncLocalStorage as ALSCompat } from 'https://deno.land/std@0.175.0/node/async_hooks.ts'
-import { Buffer as BufferCompat } from 'https://deno.land/std@0.175.0/io/buffer.ts'
+import BufferCompat from 'https://deno.land/std@0.175.0/node/buffer.ts'
+import EventsCompat from 'https://deno.land/std@0.175.0/node/events.ts'
+import AsyncHooksCompat from 'https://deno.land/std@0.175.0/node/async_hooks.ts'
+import AssertCompat from 'https://deno.land/std@0.175.0/node/assert.ts'
+import UtilCompat from 'https://deno.land/std@0.175.0/node/util.ts'
 
 /**
  * These are the shims, polyfills and other kludges to make Next.js work in standards-compliant runtime.
@@ -19,7 +22,7 @@ globalThis.EdgeRuntime = 'netlify-edge'
 let _ENTRIES = {}
 
 // Next.js expects this as a global
-globalThis.AsyncLocalStorage = ALSCompat
+globalThis.AsyncLocalStorage = AsyncHooksCompat.AsyncLocalStorage
 
 // Next.js uses this extension to the Headers API implemented by Cloudflare workerd
 if (!('getAll' in Headers.prototype)) {
@@ -49,13 +52,23 @@ const fetch /* type {typeof globalThis.fetch} */ = async (url, init) => {
   }
 }
 
-// Turbopack aliases "Buffer" to a module import, so we need to provide a shim for that
+// Shim native modules that Vercel makes available
 if (typeof require === 'undefined') {
   globalThis.require = (name) => {
-    if (name === 'buffer' || name === 'node:buffer') {
-      return { Buffer: BufferCompat }
+    switch (name.replace(/^node:/, '')) {
+      case 'buffer':
+        return BufferCompat
+      case 'events':
+        return EventsCompat
+      case 'async_hooks':
+        return AsyncHooksCompat
+      case 'assert':
+        return AssertCompat
+      case 'util':
+        return UtilCompat
+      default:
+        throw new ReferenceError(`Native module not found: ${name}`)
     }
-    throw new ReferenceError('require is not defined')
   }
 }
 
