refactor: using async storage in the local runtime context and folder structure

Author: jcbsfilho

lib/build/bundlers/polyfills/polyfills-manager.js

@@ -146,7 +146,10 @@ class PolyfillsManager {
     this.setAlias('util', require.resolve('util/'));
     this.setAlias('process', `${nodePolyfillsPath}/globals/process.js`);
 
-    this.setExternal('async_hooks', `${externalPolyfillsPath}/async_hooks.js`);
+    this.setExternal(
+      'async_hooks',
+      `${externalPolyfillsPath}/async_hooks/index.js`,
+    );
 
     return {
       libs: this.libs,

lib/env/polyfills/async_hooks.js

@@ -0,0 +1,10 @@
+/* eslint-disable */
+import * as async_hooks from 'async_hooks';
+
+export class AsyncLocalStorage extends async_hooks.AsyncLocalStorage {}
+export class AsyncResource extends async_hooks.AsyncResource {}
+
+export default {
+  AsyncLocalStorage,
+  AsyncResource,
+};

lib/env/polyfills/async_hooks/context/index.js

@@ -0,0 +1,12 @@
+/* eslint-disable */
+/**
+ * ASYNC_LOCAL_STORAGE is defined in runtime.env.js for use on the local server
+ */
+
+export class AsyncLocalStorage extends ASYNC_LOCAL_STORAGE.AsyncLocalStorage {}
+export class AsyncResource extends ASYNC_LOCAL_STORAGE.AsyncResource {}
+
+export default {
+  AsyncLocalStorage,
+  AsyncResource,
+};

lib/env/polyfills/async_hooks/index.js

@@ -1,4 +1,5 @@
-import fetchPolyfill from './fetch.polyfills.js';
-import FetchEventPolyfill from './FetchEvent.polyfills.js';
+import fetchPolyfill from './fetch/fetch.polyfills.js';
+import FetchEventPolyfill from './fetch/FetchEvent.polyfills.js';
+import AsyncHooks from './async_hooks/context/index.js';
 
-export { fetchPolyfill, FetchEventPolyfill };
+export { fetchPolyfill, FetchEventPolyfill, AsyncHooks };

lib/env/polyfills/FetchEvent.polyfills.js lib/env/polyfills/fetch/FetchEvent.polyfills.js

@@ -1,6 +1,10 @@
 import { EdgeRuntime } from 'edge-runtime';
 
-import { fetchPolyfill, FetchEventPolyfill } from './polyfills/index.js';
+import {
+  fetchPolyfill,
+  FetchEventPolyfill,
+  AsyncHooks,
+} from './polyfills/index.js';
 
 /**
  * Executes the specified JavaScript code within a sandbox environment,
@@ -56,6 +60,9 @@ function runtime(code) {
     context.WebSocket = undefined;
     /* ========================================================== */
 
+    // Async Hooks
+    context.ASYNC_LOCAL_STORAGE = AsyncHooks;
+
     return context;
   };