@mstuercke/api-utils 5.0.1 → 5.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mstuercke/api-utils",
-  "version": "5.0.1",
+  "version": "5.1.0",
   "type": "module",
   "main": "src/index.ts",
   "files": [

package/src/ws/server/trpc/awsLambdaTrpcSubscriptionRequestHandler.ts

@@ -1,5 +1,5 @@
 import {PostToConnectionCommand} from '@aws-sdk/client-apigatewaymanagementapi';
-import {AnyTRPCRouter, callTRPCProcedure} from '@trpc/server';
+import {AnyTRPCRouter, callTRPCProcedure, inferRouterContext} from '@trpc/server';
 import type {APIGatewayResult} from '@trpc/server/adapters/aws-lambda';
 import type {Observable} from '@trpc/server/observable';
 import {APIGatewayProxyEvent} from 'aws-lambda';
@@ -10,8 +10,9 @@ import {TrpcSubscriptionResponse} from './TrpcSubscriptionResponse';
 
 export const awsLambdaTrpcSubscriptionRequestHandler = <Router extends AnyTRPCRouter>(opts: {
   router: Router
+  createContext?: (event: APIGatewayProxyEvent) => Promise<inferRouterContext<Router>>
 }) => {
-  const {router} = opts;
+  const {router, createContext} = opts;
 
   return async (event: APIGatewayProxyEvent): Promise<APIGatewayResult> => {
     const connection = getConnection(event);
@@ -51,7 +52,7 @@ export const awsLambdaTrpcSubscriptionRequestHandler = <Router extends AnyTRPCRo
           procedures: router._def.procedures,
           path: request.params.path,
           type: 'subscription',
-          ctx: {},
+          ctx: createContext?.(event) || {},
           getRawInput: async () => request.params.input,
         })) as Observable<unknown, unknown>;
 
@@ -94,7 +95,7 @@ export const awsLambdaTrpcSubscriptionRequestHandler = <Router extends AnyTRPCRo
       }
     } catch (err) {
       console.error(err);
-      await stopConnection()
+      await stopConnection();
     }
 
     return {