From 139d52a9a111a1d5fb789dde99f23c675dea0f81 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Julien=20Valverd=C3=A9?= <julien.valverde@mailo.com>
Date: Tue, 9 Jul 2024 02:36:37 +0200
Subject: [PATCH] TRPC context creator

---
 packages/server/src/trpc/TRPCContext.ts        | 2 +-
 packages/server/src/trpc/TRPCContextCreator.ts | 8 ++++++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/packages/server/src/trpc/TRPCContext.ts b/packages/server/src/trpc/TRPCContext.ts
index 7c57bb4..51d7534 100644
--- a/packages/server/src/trpc/TRPCContext.ts
+++ b/packages/server/src/trpc/TRPCContext.ts
@@ -10,5 +10,5 @@ export interface TRPCContext {
         options?:  { readonly signal?: AbortSignal }
     ) => Promise<A>
 
-    req: Request
+    // req: Request
 }
diff --git a/packages/server/src/trpc/TRPCContextCreator.ts b/packages/server/src/trpc/TRPCContextCreator.ts
index bb8f3bd..5b2bd55 100644
--- a/packages/server/src/trpc/TRPCContextCreator.ts
+++ b/packages/server/src/trpc/TRPCContextCreator.ts
@@ -1,5 +1,6 @@
 import { TRPCError } from "@trpc/server"
 import type { CreateExpressContextOptions } from "@trpc/server/adapters/express"
+import type { CreateWSSContextFnOptions } from "@trpc/server/adapters/ws"
 import { Context, Effect, Layer, Runtime } from "effect"
 import type { Services } from "../Services"
 import type { TRPCContext } from "./TRPCContext"
@@ -9,7 +10,10 @@ import type { TRPCContext } from "./TRPCContext"
  * Provides a function that instantiates a fresh context for each tRPC procedure call
  */
 export class TRPCContextCreator extends Context.Tag("TRPCContextCreator")<TRPCContextCreator,
-    (opts: CreateExpressContextOptions) => TRPCContext
+    (opts:
+        | CreateExpressContextOptions
+        | CreateWSSContextFnOptions
+    ) => TRPCContext
 >() {}
 
 export module TRPCContextCreator {
@@ -20,7 +24,7 @@ export module TRPCContextCreator {
         return ({ req }) => ({
             runtime,
             run,
-            req,
+            // req,
         })
     }))
 }