with context extension

Author: enisdenjo

packages/core/src/orchestrator.ts

@@ -564,7 +564,7 @@ export function createEnvelopOrchestrator<PluginsContext extends DefaultContext>
     const validate = customValidate(initialContext);
     const contextFactory = customContextFactory(initialContext);
 
-    return async params => {
+    return async (params, contextExtension) => {
       let document;
       try {
         document = parse(params.query);
@@ -577,8 +577,7 @@ export function createEnvelopOrchestrator<PluginsContext extends DefaultContext>
         return { errors: validationErrors };
       }
 
-      // TODO: add context to perform
-      const context = await contextFactory();
+      const context = await contextFactory(contextExtension);
 
       if (isSubscriptionOperation(document, params.operationName)) {
         return await customSubscribe({

packages/types/src/get-enveloped.ts

@@ -1,7 +1,7 @@
 import { Plugin } from './plugin.js';
-import { ExecuteFunction, ParseFunction, SubscribeFunction, ValidateFunction } from './graphql.js';
-import { ArbitraryObject, Spread, PromiseOrValue } from './utils.js';
-import { PerformFunction } from './hooks.js';
+import { ExecuteFunction, ExecutionResult, ParseFunction, SubscribeFunction, ValidateFunction } from './graphql.js';
+import { ArbitraryObject, Spread, PromiseOrValue, AsyncIterableIteratorOrValue } from './utils.js';
+import { PerformParams } from './hooks.js';
 export { ArbitraryObject } from './utils.js';
 
 export type EnvelopContextFnWrapper<TFunction extends Function, ContextType = unknown> = (
@@ -18,7 +18,10 @@ export type GetEnvelopedFn<PluginsContext> = {
       contextExtension?: ContextExtension
     ) => PromiseOrValue<Spread<[InitialContext, PluginsContext, ContextExtension]>>;
     schema: any;
-    perform: PerformFunction;
+    perform: <ContextExtension = unknown>(
+      params: PerformParams,
+      contextExtension?: ContextExtension
+    ) => Promise<AsyncIterableIteratorOrValue<ExecutionResult>>;
   };
   _plugins: Plugin[];
 };

packages/types/src/hooks.ts

@@ -514,7 +514,10 @@ export interface PerformParams {
  *
  * Will never throw GraphQL errors, they will be constructed accordingly and placed in the result.
  */
-export type PerformFunction = (params: PerformParams) => Promise<AsyncIterableIteratorOrValue<ExecutionResult>>;
+export type PerformFunction<ContextExtension = unknown> = (
+  params: PerformParams,
+  contextExtension?: ContextExtension
+) => Promise<AsyncIterableIteratorOrValue<ExecutionResult>>;
 
 export type OnPerformEventPayload = {
   params: PerformParams;