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

Open Source Composition library for Apollo Federation

import { Kind } from 'graphql';
import { validateDirectiveAgainstOriginal } from '../../../helpers.js';
export function ExtendsRules(context) {
    return {
        DirectiveDefinition(node) {
            validateDirectiveAgainstOriginal(node, 'extends', context);
        },
        Directive(node) {
            if (!context.isAvailableFederationDirective('extends', node)) {
                return;
            }
            const typeDef = context.typeNodeInfo.getTypeDef();
            if (!typeDef ||
                !(typeDef.kind === Kind.OBJECT_TYPE_DEFINITION ||
                    typeDef.kind === Kind.INTERFACE_TYPE_DEFINITION)) {
                return;
            }
            if (typeDef.kind === Kind.OBJECT_TYPE_DEFINITION) {
                context.stateBuilder.objectType.setExtension(typeDef.name.value, '@extends');
            }
            else {
                context.stateBuilder.interfaceType.setExtension(typeDef.name.value);
            }
            const fields = typeDef.fields;
            for (const field of fields ?? []) {
                context.markAsUsed('@extends', typeDef.kind, typeDef.name.value, field.name.value);
            }
        },
    };
}