@graphql-yoga/plugin-defer-stream/esm/index.js

import { GraphQLSchema } from 'graphql';
import { GraphQLDeferDirective, GraphQLStreamDirective, } from '@graphql-tools/utils';
import { DeferStreamDirectiveLabelRule } from './validations/defer-stream-directive-label.js';
import { DeferStreamDirectiveOnRootFieldRule } from './validations/defer-stream-directive-on-root-field.js';
import { OverlappingFieldsCanBeMergedRule } from './validations/overlapping-fields-can-be-merged.js';
import { StreamDirectiveOnListFieldRule } from './validations/stream-directive-on-list-field.js';
export function useDeferStream() {
    return {
        onSchemaChange: ({ schema, replaceSchema, }) => {
            const directives = [];
            const deferInSchema = schema.getDirective('defer');
            if (deferInSchema == null) {
                directives.push(GraphQLDeferDirective);
            }
            const streamInSchema = schema.getDirective('stream');
            if (streamInSchema == null) {
                directives.push(GraphQLStreamDirective);
            }
            if (directives.length) {
                replaceSchema(new GraphQLSchema({
                    ...schema.toConfig(),
                    directives: [...schema.getDirectives(), ...directives],
                }));
            }
        },
        onValidate: ({ params, addValidationRule, }) => {
            // Just to make TS happy because rules are always defined by useEngine.
            params.rules = params.rules || [];
            params.rules = params.rules.filter((rule) => rule.name !== 'OverlappingFieldsCanBeMergedRule');
            addValidationRule(OverlappingFieldsCanBeMergedRule);
            addValidationRule(DeferStreamDirectiveLabelRule);
            addValidationRule(DeferStreamDirectiveOnRootFieldRule);
            addValidationRule(StreamDirectiveOnListFieldRule);
        },
    };
}