From 9a8816ce7ffc378ee172d177f1b4b50e54da5ed7 Mon Sep 17 00:00:00 2001
From: Robert Soriano <sorianorobertc@gmail.com>
Date: Wed, 18 May 2022 00:09:10 -0700
Subject: [PATCH] pass compatility event to createContext

---
 README.md          |  2 +-
 src/runtime/api.ts | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/README.md b/README.md
index 0b15d71..c394b38 100644
--- a/README.md
+++ b/README.md
@@ -43,7 +43,7 @@ export const router = trpc
   })
 
 // optional
-export const createContext = () => {
+export const createContext = (event: CompatibilityEvent) => {
   // ...
   return {
     /** context data */
diff --git a/src/runtime/api.ts b/src/runtime/api.ts
index 2b76d90..0fd1ca9 100644
--- a/src/runtime/api.ts
+++ b/src/runtime/api.ts
@@ -8,13 +8,13 @@ import type {
   inferRouterError,
 } from '@trpc/server'
 import { createURL } from 'ufo'
-import type { IncomingMessage } from 'h3'
+import type { CompatibilityEvent } from 'h3'
+import { defineEventHandler, isMethod, useBody } from 'h3'
 import type { TRPCResponse } from '@trpc/server/dist/declarations/src/rpc'
-import { isMethod, useBody } from 'h3'
 
 type MaybePromise<T> = T | Promise<T>
 
-type CreateContextFn<TRouter extends AnyRouter> = (req: IncomingMessage) => MaybePromise<inferRouterContext<TRouter>>
+type CreateContextFn<TRouter extends AnyRouter> = (event: CompatibilityEvent) => MaybePromise<inferRouterContext<TRouter>>
 
 type ResponseMetaFn<TRouter extends AnyRouter> = (opts: {
   data: TRPCResponse<unknown, inferRouterError<TRouter>>[]
@@ -35,7 +35,7 @@ export function createTRPCHandler<Router extends AnyRouter>({
 }) {
   const url = '/trpc'
 
-  return async (event) => {
+  return defineEventHandler(async (event) => {
     const {
       req,
       res,
@@ -52,7 +52,7 @@ export function createTRPCHandler<Router extends AnyRouter>({
         query: $url.searchParams,
       },
       path: $url.pathname.substring(url.length + 1),
-      createContext: async () => createContext?.(req),
+      createContext: async () => createContext?.(event),
       responseMeta,
     })
 
@@ -65,5 +65,5 @@ export function createTRPCHandler<Router extends AnyRouter>({
     })
 
     return body
-  }
+  })
 }