@theguild/federation-composition/esm/subgraph/validation/rules/elements/compose-directive.js

Open Source Composition library for Apollo Federation

import { GraphQLError, Kind } from 'graphql';
import { isDirectiveDefinitionNode, validateDirectiveAgainstOriginal } from '../../../helpers.js';
export function ComposeDirectiveRules(context) {
    return {
        DirectiveDefinition(node) {
            validateDirectiveAgainstOriginal(node, 'composeDirective', context);
        },
        Directive(node) {
            if (!context.isAvailableFederationDirective('composeDirective', node)) {
                return;
            }
            if (!context.satisfiesVersionRange('>= v2.1')) {
                return;
            }
            const nameArg = node.arguments?.find(arg => arg.name.value === 'name');
            if (!nameArg || nameArg.value.kind !== Kind.STRING) {
                return;
            }
            const name = nameArg.value.value.replace(/^@/, '');
            const definedDirectives = context.getDocument().definitions.filter(isDirectiveDefinitionNode);
            const matchingDirective = definedDirectives.find(directive => directive.name.value === name);
            if (matchingDirective) {
                context.stateBuilder.directive.setComposed(matchingDirective.name.value);
                context.stateBuilder.composedDirectives.add(matchingDirective.name.value);
            }
            else {
                context.reportError(new GraphQLError(`Could not find matching directive definition for argument to @composeDirective "@${name}" in subgraph "${context.getSubgraphName()}".`, {
                    nodes: node,
                    extensions: {
                        code: 'DIRECTIVE_COMPOSITION_ERROR',
                        subgraphName: context.getSubgraphName(),
                    },
                }));
            }
        },
    };
}