feat: use async_hooks for retrieving event context

Author: wobsoriano

README.md

@@ -59,14 +59,15 @@ Checkout [the playground example](/playground).
 
 ## H3 Event
 
-The h3 event is available as `this` for functions:
+The access the H3 event, you can use the `getEvent` function:
 
 ```ts
+import { getEvent } from 'nuxt-remote-fn/server'
 import { readBody } from 'h3'
-import type { H3Event } from 'h3'
 
-export async function getTodos(this: H3Event, otherArg: any) {
-  const body = await readBody(this)
+export async function getTodos() {
+  const event = getEvent()
+  const body = await readBody(event)
   // ...
 }
 ```
@@ -92,7 +93,6 @@ Wouldn't it be nice if all of that was automatically handled and all you'd need
 ## Development
 
 - Run `cp playground/.env.example playground/.env`
-- Run `pnpm dev:prepare` to generate type stubs.
 - Use `pnpm dev` to start [playground](./playground) in development mode.
 
 ## Credits

client.d.ts

@@ -0,0 +1 @@
+export * from './dist/runtime/client'

package.json

@@ -29,11 +29,12 @@
   "main": "./dist/module.cjs",
   "types": "./dist/types.d.ts",
   "files": [
-    "dist"
+    "dist",
+    "*.d.ts"
   ],
   "scripts": {
     "prepack": "nuxt-module-build",
-    "dev": "nuxi dev playground",
+    "dev": "pnpm prepack && nuxi dev playground",
     "dev:build": "nuxi build playground",
     "dev:prepare": "nuxt-module-build --stub && nuxi prepare playground",
     "lint": "eslint .",

playground/nuxt.config.ts

@@ -1,8 +1,7 @@
 import { defineNuxtConfig } from 'nuxt/config'
-import MyModule from '..'
 
 export default defineNuxtConfig({
   modules: [
-    MyModule
+    'nuxt-remote-fn'
   ]
 })

playground/package.json

@@ -12,6 +12,7 @@
     "schema": "./prisma/schema.prisma"
   },
   "dependencies": {
-    "@prisma/client": "^4.6.1"
+    "@prisma/client": "^4.6.1",
+    "nuxt-remote-fn": "workspace:*"
   }
 }

pnpm-lock.yaml

@@ -0,0 +1 @@
+export * from './dist/runtime/server'

server.d.ts

@@ -1,8 +1,28 @@
-import type { EventHandler } from 'h3'
+import { AsyncLocalStorage } from 'async_hooks'
+import type { EventHandler, H3Event } from 'h3'
 import { eventHandler, isMethod, readBody, createError } from 'h3'
 
-export function createRemoteFnHandler<T> (functions: T): EventHandler<T> {
-  return eventHandler(async (event) => {
+const DEFAULT_CONTEXT = {}
+
+const asyncLocalStorage = new AsyncLocalStorage<H3Event>()
+
+export function getEvent (): H3Event {
+  return asyncLocalStorage.getStore() || DEFAULT_CONTEXT as H3Event
+}
+
+function wrapEventHandler (handler: EventHandler): EventHandler {
+  return eventHandler((event) => {
+    const context = {
+      node: event.node,
+      context: event.context,
+      path: event.path
+    }
+    return asyncLocalStorage.run(context as H3Event, () => handler(event))
+  })
+}
+
+export function createRemoteFnHandler<T> (functions: T): any {
+  return wrapEventHandler(eventHandler(async (event) => {
     if (!isMethod(event, 'POST')) {
       throw createError({
         statusCode: 405,
@@ -22,7 +42,7 @@ export function createRemoteFnHandler<T> (functions: T): EventHandler<T> {
     }
 
     // @ts-ignore
-    const result = await functions[moduleId][functionName].apply(event, input)
+    const result = await functions[moduleId][functionName](...input)
     return result
-  })
+  }))
 }