Merge branch 'main' of github.com:coratgerl/wobe

palixir · Jun 9, 2024 · fdbbbc0 · fdbbbc0
2 parents 7c8c6c0 + 22daccf
commit fdbbbc0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/packages/wobe-graphql-yoga/src/index.test.ts b/packages/wobe-graphql-yoga/src/index.test.ts
@@ -20,12 +20,16 @@ describe('Wobe GraphQL Yoga plugin', () => {
 					Query: {
 						hello: (_, __, context) => {
 							context.response.setCookie('tata', 'tata')
+							expect(context.test).toBeDefined()
 							expect(context.response).toBeDefined()
 							expect(context.request.headers).toBeDefined()
 							return 'Hello from Yoga!'
 						},
 					},
 				},
+				context: () => {
+					return { test: 'test' }
+				},
 			}),
 		)
 

diff --git a/packages/wobe-graphql-yoga/src/index.ts b/packages/wobe-graphql-yoga/src/index.ts
@@ -3,13 +3,15 @@ import {
 	createYoga,
 	type GraphQLSchemaWithContext,
 	type YogaServerOptions,
-	type YogaInitialContext,
 } from 'graphql-yoga'
 import type { MaybePromise, Wobe, WobePlugin, WobeResponse } from 'wobe'
 
 export type GraphqlYogaContext =
 	| MaybePromise<Record<string, unknown>>
-	| ((initialContext: YogaInitialContext) => MaybePromise<unknown>)
+	| ((context: {
+			request: Request
+			response: WobeResponse
+	  }) => MaybePromise<unknown>)
 
 export interface GraphqlYogaPluginOptions {
 	graphqlMiddleware?: (