@graphql-yoga/nestjs 3.0.2 → 3.1.0

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # @graphql-yoga/nestjs
 
+## 3.1.0
+
+### Minor Changes
+
+- [#2992](https://github.com/dotansimha/graphql-yoga/pull/2992)
+  [`b1f0e3a2`](https://github.com/dotansimha/graphql-yoga/commit/b1f0e3a2986956c6791a251df908e3f8b50ec966)
+  Thanks [@magrinj](https://github.com/magrinj)! - Adding support of conditionalSchema option.
+  (Currently without support for WebSocket subscriptions)
+
+### Patch Changes
+
+- Updated dependencies
+  [[`b1f0e3a2`](https://github.com/dotansimha/graphql-yoga/commit/b1f0e3a2986956c6791a251df908e3f8b50ec966)]:
+  - graphql-yoga@5.1.0
+
 ## 3.0.2
 
 ### Patch Changes

package/__tests__/graphql.spec.ts

@@ -1,3 +1,11 @@
+import {
+  GraphQLID,
+  GraphQLList,
+  GraphQLNonNull,
+  GraphQLObjectType,
+  GraphQLSchema,
+  GraphQLString,
+} from 'graphql';
 import { INestApplication } from '@nestjs/common';
 import { Test } from '@nestjs/testing';
 import { fetch } from '@whatwg-node/fetch';
@@ -7,7 +15,44 @@ let app: INestApplication, url: string;
 
 beforeAll(async () => {
   const module = await Test.createTestingModule({
-    imports: [AppModule.forRoot()],
+    imports: [
+      AppModule.forRoot({
+        conditionalSchema: async () => {
+          const DogObjectType = new GraphQLObjectType({
+            name: 'Dog',
+            fields: {
+              id: {
+                type: new GraphQLNonNull(GraphQLID),
+              },
+              name: {
+                type: new GraphQLNonNull(GraphQLString),
+              },
+              breed: {
+                type: new GraphQLNonNull(GraphQLString),
+              },
+            },
+          });
+
+          return new GraphQLSchema({
+            query: new GraphQLObjectType({
+              name: 'Query',
+              fields: {
+                getDogs: {
+                  type: new GraphQLList(new GraphQLNonNull(DogObjectType)),
+                  resolve: () => [
+                    {
+                      id: '1',
+                      name: 'Bingo',
+                      breed: 'Dalmatian',
+                    },
+                  ],
+                },
+              },
+            }),
+          });
+        },
+      }),
+    ],
   }).compile();
   app = module.createNestApplication();
   await app.listen(0);
@@ -48,3 +93,36 @@ it('should return query result', async () => {
     }
   `);
 });
+
+it('should return query result for conditional schema', async () => {
+  const res = await fetch(url, {
+    method: 'POST',
+    headers: {
+      'content-type': 'application/json',
+    },
+    body: JSON.stringify({
+      query: /* GraphQL */ `
+        {
+          getDogs {
+            id
+            name
+            breed
+          }
+        }
+      `,
+    }),
+  });
+  await expect(res.json()).resolves.toMatchInlineSnapshot(`
+    {
+      "data": {
+        "getDogs": [
+          {
+            "breed": "Dalmatian",
+            "id": "1",
+            "name": "Bingo",
+          },
+        ],
+      },
+    }
+  `);
+});

package/dist/cjs/index.js

@@ -28,7 +28,7 @@ class AbstractYogaDriver extends graphql_2.AbstractGraphQLDriver {
     async stop() {
         // noop
     }
-    registerExpress(options, { preStartHook } = {}) {
+    registerExpress({ conditionalSchema, ...options }, { preStartHook } = {}) {
         const app = this.httpAdapterHost.httpAdapter.getInstance();
         preStartHook?.(app);
         // nest's logger doesnt have the info method
@@ -41,8 +41,10 @@ class AbstractYogaDriver extends graphql_2.AbstractGraphQLDriver {
                 this.log(message, ...args);
             }
         }
+        const schema = this.mergeConditionalSchema(conditionalSchema, options.schema);
         const yoga = (0, graphql_yoga_1.createYoga)({
             ...options,
+            schema,
             graphqlEndpoint: options.path,
             // disable logging by default
             // however, if `true` use nest logger
@@ -55,11 +57,13 @@ class AbstractYogaDriver extends graphql_2.AbstractGraphQLDriver {
         this.yoga = yoga;
         app.use(yoga.graphqlEndpoint, (req, res) => yoga(req, res, { req, res }));
     }
-    registerFastify(options, { preStartHook } = {}) {
+    registerFastify({ conditionalSchema, ...options }, { preStartHook } = {}) {
         const app = this.httpAdapterHost.httpAdapter.getInstance();
         preStartHook?.(app);
+        const schema = this.mergeConditionalSchema(conditionalSchema, options.schema);
         const yoga = (0, graphql_yoga_1.createYoga)({
             ...options,
+            schema,
             graphqlEndpoint: options.path,
             // disable logging by default
             // however, if `true` use fastify logger
@@ -78,6 +82,27 @@ class AbstractYogaDriver extends graphql_2.AbstractGraphQLDriver {
             return reply;
         });
     }
+    mergeConditionalSchema(conditionalSchema, schema) {
+        let mergedSchema = schema;
+        if (conditionalSchema) {
+            mergedSchema = async (request) => {
+                const schemas = [];
+                if (schema) {
+                    schemas.push(schema);
+                }
+                const conditionalSchemaResult = typeof conditionalSchema === 'function'
+                    ? await conditionalSchema(request)
+                    : await conditionalSchema;
+                if (conditionalSchemaResult) {
+                    schemas.push(conditionalSchemaResult);
+                }
+                return (0, graphql_yoga_1.mergeSchemas)({
+                    schemas,
+                });
+            };
+        }
+        return mergedSchema;
+    }
     subscriptionWithFilter(instanceRef, filterFn, 
     // disable next error, the original function in @nestjs/graphql is also untyped
     // eslint-disable-next-line @typescript-eslint/ban-types
@@ -111,6 +136,11 @@ let YogaDriver = exports.YogaDriver = class YogaDriver extends AbstractYogaDrive
                 : options.subscriptions;
             if (config['graphql-ws']) {
                 config['graphql-ws'] = typeof config['graphql-ws'] === 'object' ? config['graphql-ws'] : {};
+                if (options.conditionalSchema) {
+                    throw new Error(`
+            Conditional schema is not supported with graphql-ws.
+          `);
+                }
                 config['graphql-ws'].onSubscribe = async (ctx, msg) => {
                     const { schema, execute, subscribe, contextFactory, parse, validate } = this.yoga.getEnveloped({
                         ...ctx,
@@ -138,6 +168,11 @@ let YogaDriver = exports.YogaDriver = class YogaDriver extends AbstractYogaDrive
                     typeof config['subscriptions-transport-ws'] === 'object'
                         ? config['subscriptions-transport-ws']
                         : {};
+                if (options.conditionalSchema) {
+                    throw new Error(`
+            Conditional schema is not supported with subscriptions-transport-ws.
+          `);
+                }
                 config['subscriptions-transport-ws'].onOperation = async (_msg, params, ws) => {
                     const { schema, execute, subscribe, contextFactory, parse, validate } = this.yoga.getEnveloped({
                         ...params.context,

package/dist/esm/index.js

@@ -1,6 +1,6 @@
 import { __decorate } from "tslib";
 import { printSchema } from 'graphql';
-import { createYoga, filter, pipe } from 'graphql-yoga';
+import { createYoga, filter, mergeSchemas, pipe, } from 'graphql-yoga';
 import { Injectable, Logger } from '@nestjs/common';
 import { AbstractGraphQLDriver, GqlSubscriptionService, } from '@nestjs/graphql';
 export class AbstractYogaDriver extends AbstractGraphQLDriver {
@@ -25,7 +25,7 @@ export class AbstractYogaDriver extends AbstractGraphQLDriver {
     async stop() {
         // noop
     }
-    registerExpress(options, { preStartHook } = {}) {
+    registerExpress({ conditionalSchema, ...options }, { preStartHook } = {}) {
         const app = this.httpAdapterHost.httpAdapter.getInstance();
         preStartHook?.(app);
         // nest's logger doesnt have the info method
@@ -38,8 +38,10 @@ export class AbstractYogaDriver extends AbstractGraphQLDriver {
                 this.log(message, ...args);
             }
         }
+        const schema = this.mergeConditionalSchema(conditionalSchema, options.schema);
         const yoga = createYoga({
             ...options,
+            schema,
             graphqlEndpoint: options.path,
             // disable logging by default
             // however, if `true` use nest logger
@@ -52,11 +54,13 @@ export class AbstractYogaDriver extends AbstractGraphQLDriver {
         this.yoga = yoga;
         app.use(yoga.graphqlEndpoint, (req, res) => yoga(req, res, { req, res }));
     }
-    registerFastify(options, { preStartHook } = {}) {
+    registerFastify({ conditionalSchema, ...options }, { preStartHook } = {}) {
         const app = this.httpAdapterHost.httpAdapter.getInstance();
         preStartHook?.(app);
+        const schema = this.mergeConditionalSchema(conditionalSchema, options.schema);
         const yoga = createYoga({
             ...options,
+            schema,
             graphqlEndpoint: options.path,
             // disable logging by default
             // however, if `true` use fastify logger
@@ -75,6 +79,27 @@ export class AbstractYogaDriver extends AbstractGraphQLDriver {
             return reply;
         });
     }
+    mergeConditionalSchema(conditionalSchema, schema) {
+        let mergedSchema = schema;
+        if (conditionalSchema) {
+            mergedSchema = async (request) => {
+                const schemas = [];
+                if (schema) {
+                    schemas.push(schema);
+                }
+                const conditionalSchemaResult = typeof conditionalSchema === 'function'
+                    ? await conditionalSchema(request)
+                    : await conditionalSchema;
+                if (conditionalSchemaResult) {
+                    schemas.push(conditionalSchemaResult);
+                }
+                return mergeSchemas({
+                    schemas,
+                });
+            };
+        }
+        return mergedSchema;
+    }
     subscriptionWithFilter(instanceRef, filterFn, 
     // disable next error, the original function in @nestjs/graphql is also untyped
     // eslint-disable-next-line @typescript-eslint/ban-types
@@ -107,6 +132,11 @@ export let YogaDriver = class YogaDriver extends AbstractYogaDriver {
                 : options.subscriptions;
             if (config['graphql-ws']) {
                 config['graphql-ws'] = typeof config['graphql-ws'] === 'object' ? config['graphql-ws'] : {};
+                if (options.conditionalSchema) {
+                    throw new Error(`
+            Conditional schema is not supported with graphql-ws.
+          `);
+                }
                 config['graphql-ws'].onSubscribe = async (ctx, msg) => {
                     const { schema, execute, subscribe, contextFactory, parse, validate } = this.yoga.getEnveloped({
                         ...ctx,
@@ -134,6 +164,11 @@ export let YogaDriver = class YogaDriver extends AbstractYogaDriver {
                     typeof config['subscriptions-transport-ws'] === 'object'
                         ? config['subscriptions-transport-ws']
                         : {};
+                if (options.conditionalSchema) {
+                    throw new Error(`
+            Conditional schema is not supported with subscriptions-transport-ws.
+          `);
+                }
                 config['subscriptions-transport-ws'].onOperation = async (_msg, params, ws) => {
                     const { schema, execute, subscribe, contextFactory, parse, validate } = this.yoga.getEnveloped({
                         ...params.context,

package/dist/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@graphql-yoga/nestjs",
-  "version": "3.0.2",
+  "version": "3.1.0",
   "description": "GraphQL Yoga driver for NestJS GraphQL.",
   "sideEffects": false,
   "peerDependencies": {
@@ -8,7 +8,7 @@
     "@nestjs/core": "^10.0.0",
     "@nestjs/graphql": "^12.0.0",
     "graphql": "^15.0.0 || ^16.0.0",
-    "graphql-yoga": "^5.0.2"
+    "graphql-yoga": "^5.1.0"
   },
   "repository": {
     "type": "git",

package/dist/typings/index.d.cts

@@ -1,6 +1,6 @@
 import type { Express, Request as ExpressRequest, Response as ExpressResponse } from 'express';
 import type { FastifyInstance, FastifyReply, FastifyRequest } from 'fastify';
-import { YogaServerInstance, YogaServerOptions } from 'graphql-yoga';
+import { YogaSchemaDefinition, YogaServerInstance, YogaServerOptions } from 'graphql-yoga';
 import { AbstractGraphQLDriver, GqlModuleOptions, SubscriptionConfig } from '@nestjs/graphql';
 export type YogaDriverPlatform = 'express' | 'fastify';
 export type YogaDriverServerContext<Platform extends YogaDriverPlatform> = Platform extends 'fastify' ? {
@@ -10,14 +10,23 @@ export type YogaDriverServerContext<Platform extends YogaDriverPlatform> = Platf
     req: ExpressRequest;
     res: ExpressResponse;
 };
-export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<YogaServerOptions<YogaDriverServerContext<Platform>, never>, 'context' | 'schema'>;
+export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<YogaServerOptions<YogaDriverServerContext<Platform>, never>, 'context' | 'schema'> & {
+    conditionalSchema?: YogaSchemaDefinition<YogaDriverServerContext<Platform>, never> | undefined;
+};
 export type YogaDriverServerInstance<Platform extends YogaDriverPlatform> = YogaServerInstance<YogaDriverServerContext<Platform>, never>;
-export type YogaDriverConfig<Platform extends YogaDriverPlatform = 'express'> = GqlModuleOptions & YogaDriverServerOptions<Platform> & {
+export type YogaDriverConfig<Platform extends YogaDriverPlatform = 'express'> = GqlModuleOptions & YogaDriverServerOptions<Platform> & ({
     /**
      * Subscriptions configuration. Passing `true` will install only `graphql-ws`.
      */
     subscriptions?: boolean | YogaDriverSubscriptionConfig;
-};
+    conditionalSchema?: never;
+} | {
+    /**
+     * TODO: Support conditional schema with subscriptions
+     */
+    subscriptions?: never;
+    conditionalSchema?: YogaSchemaDefinition<YogaDriverServerContext<Platform>, never> | undefined;
+});
 export type YogaDriverSubscriptionConfig = {
     'graphql-ws'?: Omit<SubscriptionConfig['graphql-ws'], 'onSubscribe'>;
     'subscriptions-transport-ws'?: Omit<SubscriptionConfig['subscriptions-transport-ws'], 'onOperation'>;
@@ -26,12 +35,13 @@ export declare abstract class AbstractYogaDriver<Platform extends YogaDriverPlat
     protected yoga: YogaDriverServerInstance<Platform>;
     start(options: YogaDriverConfig<Platform>): Promise<void>;
     stop(): Promise<void>;
-    protected registerExpress(options: YogaDriverConfig<'express'>, { preStartHook }?: {
+    protected registerExpress({ conditionalSchema, ...options }: YogaDriverConfig<'express'>, { preStartHook }?: {
         preStartHook?: (app: Express) => void;
     }): void;
-    protected registerFastify(options: YogaDriverConfig<'fastify'>, { preStartHook }?: {
+    protected registerFastify({ conditionalSchema, ...options }: YogaDriverConfig<'fastify'>, { preStartHook }?: {
         preStartHook?: (app: FastifyInstance) => void;
     }): void;
+    private mergeConditionalSchema;
     subscriptionWithFilter<TPayload, TVariables, TContext>(instanceRef: unknown, filterFn: (payload: TPayload, variables: TVariables, context: TContext) => boolean | Promise<boolean>, createSubscribeContext: Function): (args_0: TPayload, args_1: TVariables, args_2: TContext) => Promise<import("graphql-yoga").Repeater<TPayload, void, unknown>>;
 }
 export declare class YogaDriver<Platform extends YogaDriverPlatform = 'express'> extends AbstractYogaDriver<Platform> {

package/dist/typings/index.d.ts

@@ -1,6 +1,6 @@
 import type { Express, Request as ExpressRequest, Response as ExpressResponse } from 'express';
 import type { FastifyInstance, FastifyReply, FastifyRequest } from 'fastify';
-import { YogaServerInstance, YogaServerOptions } from 'graphql-yoga';
+import { YogaSchemaDefinition, YogaServerInstance, YogaServerOptions } from 'graphql-yoga';
 import { AbstractGraphQLDriver, GqlModuleOptions, SubscriptionConfig } from '@nestjs/graphql';
 export type YogaDriverPlatform = 'express' | 'fastify';
 export type YogaDriverServerContext<Platform extends YogaDriverPlatform> = Platform extends 'fastify' ? {
@@ -10,14 +10,23 @@ export type YogaDriverServerContext<Platform extends YogaDriverPlatform> = Platf
     req: ExpressRequest;
     res: ExpressResponse;
 };
-export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<YogaServerOptions<YogaDriverServerContext<Platform>, never>, 'context' | 'schema'>;
+export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<YogaServerOptions<YogaDriverServerContext<Platform>, never>, 'context' | 'schema'> & {
+    conditionalSchema?: YogaSchemaDefinition<YogaDriverServerContext<Platform>, never> | undefined;
+};
 export type YogaDriverServerInstance<Platform extends YogaDriverPlatform> = YogaServerInstance<YogaDriverServerContext<Platform>, never>;
-export type YogaDriverConfig<Platform extends YogaDriverPlatform = 'express'> = GqlModuleOptions & YogaDriverServerOptions<Platform> & {
+export type YogaDriverConfig<Platform extends YogaDriverPlatform = 'express'> = GqlModuleOptions & YogaDriverServerOptions<Platform> & ({
     /**
      * Subscriptions configuration. Passing `true` will install only `graphql-ws`.
      */
     subscriptions?: boolean | YogaDriverSubscriptionConfig;
-};
+    conditionalSchema?: never;
+} | {
+    /**
+     * TODO: Support conditional schema with subscriptions
+     */
+    subscriptions?: never;
+    conditionalSchema?: YogaSchemaDefinition<YogaDriverServerContext<Platform>, never> | undefined;
+});
 export type YogaDriverSubscriptionConfig = {
     'graphql-ws'?: Omit<SubscriptionConfig['graphql-ws'], 'onSubscribe'>;
     'subscriptions-transport-ws'?: Omit<SubscriptionConfig['subscriptions-transport-ws'], 'onOperation'>;
@@ -26,12 +35,13 @@ export declare abstract class AbstractYogaDriver<Platform extends YogaDriverPlat
     protected yoga: YogaDriverServerInstance<Platform>;
     start(options: YogaDriverConfig<Platform>): Promise<void>;
     stop(): Promise<void>;
-    protected registerExpress(options: YogaDriverConfig<'express'>, { preStartHook }?: {
+    protected registerExpress({ conditionalSchema, ...options }: YogaDriverConfig<'express'>, { preStartHook }?: {
         preStartHook?: (app: Express) => void;
     }): void;
-    protected registerFastify(options: YogaDriverConfig<'fastify'>, { preStartHook }?: {
+    protected registerFastify({ conditionalSchema, ...options }: YogaDriverConfig<'fastify'>, { preStartHook }?: {
         preStartHook?: (app: FastifyInstance) => void;
     }): void;
+    private mergeConditionalSchema;
     subscriptionWithFilter<TPayload, TVariables, TContext>(instanceRef: unknown, filterFn: (payload: TPayload, variables: TVariables, context: TContext) => boolean | Promise<boolean>, createSubscribeContext: Function): (args_0: TPayload, args_1: TVariables, args_2: TContext) => Promise<import("graphql-yoga").Repeater<TPayload, void, unknown>>;
 }
 export declare class YogaDriver<Platform extends YogaDriverPlatform = 'express'> extends AbstractYogaDriver<Platform> {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@graphql-yoga/nestjs",
-  "version": "3.0.2",
+  "version": "3.1.0",
   "type": "module",
   "description": "GraphQL Yoga driver for NestJS GraphQL.",
   "repository": {
@@ -46,7 +46,7 @@
     "@nestjs/core": "^10.0.0",
     "@nestjs/graphql": "^12.0.0",
     "graphql": "^15.0.0 || ^16.0.0",
-    "graphql-yoga": "^5.0.2"
+    "graphql-yoga": "^5.1.0"
   },
   "devDependencies": {
     "@nestjs/common": "^10.0.0",
@@ -66,7 +66,7 @@
     "graphql": "^16.6.0",
     "graphql-http": "^1.18.0",
     "graphql-ws": "^5.11.3",
-    "graphql-yoga": "^5.0.2",
+    "graphql-yoga": "^5.1.0",
     "prettier": "^3.0.0",
     "reflect-metadata": "^0.1.13",
     "rxjs": "^7.8.0",

package/src/index.ts

@@ -1,7 +1,15 @@
 import type { Express, Request as ExpressRequest, Response as ExpressResponse } from 'express';
 import type { FastifyInstance, FastifyReply, FastifyRequest } from 'fastify';
-import { printSchema } from 'graphql';
-import { createYoga, filter, pipe, YogaServerInstance, YogaServerOptions } from 'graphql-yoga';
+import { GraphQLSchema, printSchema } from 'graphql';
+import {
+  createYoga,
+  filter,
+  mergeSchemas,
+  pipe,
+  YogaSchemaDefinition,
+  YogaServerInstance,
+  YogaServerOptions,
+} from 'graphql-yoga';
 import type { ExecutionParams } from 'subscriptions-transport-ws';
 import { Injectable, Logger } from '@nestjs/common';
 import {
@@ -27,7 +35,9 @@ export type YogaDriverServerContext<Platform extends YogaDriverPlatform> =
 export type YogaDriverServerOptions<Platform extends YogaDriverPlatform> = Omit<
   YogaServerOptions<YogaDriverServerContext<Platform>, never>,
   'context' | 'schema'
->;
+> & {
+  conditionalSchema?: YogaSchemaDefinition<YogaDriverServerContext<Platform>, never> | undefined;
+};
 
 export type YogaDriverServerInstance<Platform extends YogaDriverPlatform> = YogaServerInstance<
   YogaDriverServerContext<Platform>,
@@ -35,12 +45,25 @@ export type YogaDriverServerInstance<Platform extends YogaDriverPlatform> = Yoga
 >;
 
 export type YogaDriverConfig<Platform extends YogaDriverPlatform = 'express'> = GqlModuleOptions &
-  YogaDriverServerOptions<Platform> & {
-    /**
-     * Subscriptions configuration. Passing `true` will install only `graphql-ws`.
-     */
-    subscriptions?: boolean | YogaDriverSubscriptionConfig;
-  };
+  YogaDriverServerOptions<Platform> &
+  (
+    | {
+        /**
+         * Subscriptions configuration. Passing `true` will install only `graphql-ws`.
+         */
+        subscriptions?: boolean | YogaDriverSubscriptionConfig;
+        conditionalSchema?: never;
+      }
+    | {
+        /**
+         * TODO: Support conditional schema with subscriptions
+         */
+        subscriptions?: never;
+        conditionalSchema?:
+          | YogaSchemaDefinition<YogaDriverServerContext<Platform>, never>
+          | undefined;
+      }
+  );
 
 export type YogaDriverSubscriptionConfig = {
   'graphql-ws'?: Omit<SubscriptionConfig['graphql-ws'], 'onSubscribe'>;
@@ -78,7 +101,7 @@ export abstract class AbstractYogaDriver<
   }
 
   protected registerExpress(
-    options: YogaDriverConfig<'express'>,
+    { conditionalSchema, ...options }: YogaDriverConfig<'express'>,
     { preStartHook }: { preStartHook?: (app: Express) => void } = {},
   ) {
     const app: Express = this.httpAdapterHost.httpAdapter.getInstance();
@@ -96,8 +119,11 @@ export abstract class AbstractYogaDriver<
       }
     }
 
+    const schema = this.mergeConditionalSchema<'express'>(conditionalSchema, options.schema);
+
     const yoga = createYoga<YogaDriverServerContext<'express'>>({
       ...options,
+      schema,
       graphqlEndpoint: options.path,
       // disable logging by default
       // however, if `true` use nest logger
@@ -115,15 +141,18 @@ export abstract class AbstractYogaDriver<
   }
 
   protected registerFastify(
-    options: YogaDriverConfig<'fastify'>,
+    { conditionalSchema, ...options }: YogaDriverConfig<'fastify'>,
     { preStartHook }: { preStartHook?: (app: FastifyInstance) => void } = {},
   ) {
     const app: FastifyInstance = this.httpAdapterHost.httpAdapter.getInstance();
 
     preStartHook?.(app);
 
+    const schema = this.mergeConditionalSchema<'fastify'>(conditionalSchema, options.schema);
+
     const yoga = createYoga<YogaDriverServerContext<'fastify'>>({
       ...options,
+      schema,
       graphqlEndpoint: options.path,
       // disable logging by default
       // however, if `true` use fastify logger
@@ -144,6 +173,38 @@ export abstract class AbstractYogaDriver<
     });
   }
 
+  private mergeConditionalSchema<T extends YogaDriverPlatform>(
+    conditionalSchema: YogaSchemaDefinition<YogaDriverServerContext<T>, never> | undefined,
+    schema?: GraphQLSchema,
+  ) {
+    let mergedSchema: YogaSchemaDefinition<YogaDriverServerContext<T>, never> | undefined = schema;
+
+    if (conditionalSchema) {
+      mergedSchema = async request => {
+        const schemas: GraphQLSchema[] = [];
+
+        if (schema) {
+          schemas.push(schema);
+        }
+
+        const conditionalSchemaResult =
+          typeof conditionalSchema === 'function'
+            ? await conditionalSchema(request)
+            : await conditionalSchema;
+
+        if (conditionalSchemaResult) {
+          schemas.push(conditionalSchemaResult);
+        }
+
+        return mergeSchemas({
+          schemas,
+        });
+      };
+    }
+
+    return mergedSchema;
+  }
+
   public subscriptionWithFilter<TPayload, TVariables, TContext>(
     instanceRef: unknown,
     filterFn: (
@@ -198,6 +259,12 @@ export class YogaDriver<
       if (config['graphql-ws']) {
         config['graphql-ws'] = typeof config['graphql-ws'] === 'object' ? config['graphql-ws'] : {};
 
+        if (options.conditionalSchema) {
+          throw new Error(`
+            Conditional schema is not supported with graphql-ws.
+          `);
+        }
+
         config['graphql-ws'].onSubscribe = async (ctx, msg) => {
           const { schema, execute, subscribe, contextFactory, parse, validate } =
             this.yoga.getEnveloped({
@@ -229,6 +296,12 @@ export class YogaDriver<
             ? config['subscriptions-transport-ws']
             : {};
 
+        if (options.conditionalSchema) {
+          throw new Error(`
+            Conditional schema is not supported with subscriptions-transport-ws.
+          `);
+        }
+
         config['subscriptions-transport-ws'].onOperation = async (
           _msg: unknown,
           params: ExecutionParams,