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

Open Source Composition library for Apollo Federation

import { GraphQLError, Kind } from "graphql";
import { validateDirectiveAgainstOriginal } from "../../../helpers.js";
export function ShareableRules(context) {
    return {
        DirectiveDefinition(node) {
            validateDirectiveAgainstOriginal(node, "shareable", context);
        },
        Directive(node) {
            if (!context.isAvailableFederationDirective("shareable", node)) {
                return;
            }
            const typeDef = context.typeNodeInfo.getTypeDef();
            const fieldDef = context.typeNodeInfo.getFieldDef();
            if (!typeDef) {
                return;
            }
            if ((typeDef.kind === Kind.OBJECT_TYPE_DEFINITION ||
                typeDef.kind === Kind.OBJECT_TYPE_EXTENSION) &&
                !context.stateBuilder.isInterfaceObject(typeDef.name.value)) {
                if (fieldDef) {
                    context.stateBuilder.objectType.field.setShareable(typeDef.name.value, fieldDef.name.value);
                }
                else {
                    context.stateBuilder.objectType.setShareable(typeDef.name.value);
                    if (typeDef.fields) {
                        for (const fieldDef of typeDef.fields) {
                            context.stateBuilder.objectType.field.setShareable(typeDef.name.value, fieldDef.name.value);
                        }
                    }
                }
            }
            if (!fieldDef) {
                return;
            }
            if (typeDef.kind === Kind.INTERFACE_TYPE_DEFINITION ||
                typeDef.kind === Kind.INTERFACE_TYPE_EXTENSION) {
                context.reportError(new GraphQLError(`Invalid use of @shareable on field "${typeDef.name.value}.${fieldDef.name.value}": only object type fields can be marked with @shareable`, {
                    nodes: node,
                    extensions: { code: "INVALID_SHAREABLE_USAGE" },
                }));
                return;
            }
        },
    };
}