feat: add useContext function

Author: wobsoriano

README.md

@@ -69,8 +69,16 @@ import { decodeAndVerifyJwtToken } from '~/somewhere/in/utils'
 export async function addTodo(todo: Todo) {
   const event = useEvent()
 
-  const authorization = getRequestHeader(event, 'authorization')
-  const user = await decodeAndVerifyJwtToken(authorization.split(' ')[1])
+  async function getUserFromHeader() {
+    const authorization = getRequestHeader(event, 'authorization')
+    if (authorization) {
+      const user = await decodeAndVerifyJwtToken(authorization.split(' ')[1])
+      return user
+    }
+    return null
+  }
+
+  const user = await getUserFromHeader()
 
   if (!user) {
     throw createError({ statusCode: 401 })
@@ -89,6 +97,37 @@ export async function addTodo(todo: Todo) {
 
 You can use all built-in [h3 utilities](https://github.com/unjs/h3#built-in) inside your exported functions.
 
+## createContext
+
+Each `.server.` file can also export a `createContext` function that is called for each incoming request:
+
+```ts
+export function createContext() {
+  const event = useEvent()
+
+  async function getUserFromHeader() {
+    const authorization = getRequestHeader(event, 'authorization')
+    if (authorization) {
+      const user = await decodeAndVerifyJwtToken(authorization.split(' ')[1])
+      return user
+    }
+    return null
+  }
+
+  event.context.user = await getUserFromHeader()
+}
+
+export async function addTodo(todo: Todo) {
+  const event = useEvent()
+
+  if (!event.context.user) {
+    throw createError({ statusCode: 401 })
+  }
+
+  // addTodo logic
+}
+```
+
 ## useAsyncData
 
 `nuxt-remote-fn` can work seamlessly with [`useAsyncData`](https://nuxt.com/docs/api/composables/use-async-data/):

playground/lib/todo.server.ts

@@ -15,9 +15,6 @@ export function getTodo (id: number) {
 }
 
 export async function toggleTodo (id: number) {
-  const event = useEvent()
-  console.log(event.context.params)
-
   const todo = await getTodo(id)
   return prisma.todo.update({
     where: { id },
@@ -38,3 +35,8 @@ export function addTodo ({ title, content }: { title: string; content: string })
     }
   })
 }
+
+export function createContext() {
+  const event = useEvent()
+  // console.log('event.context.params', event.context.params)
+}

src/runtime/server.ts

@@ -34,6 +34,10 @@ export function createRemoteFnHandler<
         statusMessage: `[nuxt-remote-fn]: ${functionName as string} is not a function.`
       })
     }
+
+    if ('createContext' in functions[moduleId]) {
+      await functions[moduleId]['createContext'].apply(event)
+    }
 
     const result = functions[moduleId][functionName].apply(event, body.args)
     return result