@theguild/federation-composition
Version:
Open Source Composition library for Apollo Federation
80 lines (79 loc) • 4.22 kB
JavaScript
import { GraphQLError, Kind } from "graphql";
import { validateDirectiveAgainstOriginal } from "../../../helpers.js";
export function AuthenticatedRule(context) {
return {
DirectiveDefinition(node) {
validateDirectiveAgainstOriginal(node, "authenticated", context);
},
Directive(node, _key, _parent, paths, ancestors) {
if (!context.isAvailableFederationDirective("authenticated", node)) {
return;
}
context.stateBuilder.markSpecAsUsed("authenticated");
const directivesKeyAt = paths.findIndex((path) => path === "directives");
if (directivesKeyAt === -1) {
throw new Error('Could not find "directives" key in ancestors');
}
const parent = ancestors[directivesKeyAt];
if (!parent) {
throw new Error("Could not find the node annotated with @authenticated");
}
if (Array.isArray(parent)) {
throw new Error("Expected parent to be a single node");
}
if (!("kind" in parent)) {
throw new Error("Expected parent to be a node");
}
switch (parent.kind) {
case Kind.FIELD_DEFINITION: {
const typeDef = context.typeNodeInfo.getTypeDef();
if (!typeDef) {
throw new Error("Could not find the parent type of the field annotated with @authenticated");
}
if (typeDef.kind === Kind.INTERFACE_TYPE_DEFINITION ||
typeDef.kind === Kind.INTERFACE_TYPE_EXTENSION) {
context.reportError(new GraphQLError(`Invalid use of @authenticated on field "${typeDef.name.value}.${parent.name.value}": @authenticated cannot be applied on interfaces, interface fields and interface objects`, {
extensions: {
code: "AUTH_REQUIREMENTS_APPLIED_ON_INTERFACE",
},
}));
return;
}
if (typeDef.kind === Kind.OBJECT_TYPE_DEFINITION ||
typeDef.kind === Kind.OBJECT_TYPE_EXTENSION) {
context.stateBuilder.objectType.field.setAuthenticated(typeDef.name.value, parent.name.value);
}
break;
}
case Kind.OBJECT_TYPE_DEFINITION:
case Kind.OBJECT_TYPE_EXTENSION:
if (context.stateBuilder.isInterfaceObject(parent.name.value)) {
context.reportError(new GraphQLError(`Invalid use of @authenticated on interface object "${parent.name.value}": @authenticated cannot be applied on interfaces, interface fields and interface objects`, {
extensions: {
code: "AUTH_REQUIREMENTS_APPLIED_ON_INTERFACE",
},
}));
return;
}
context.stateBuilder.objectType.setAuthenticated(parent.name.value);
break;
case Kind.INTERFACE_TYPE_DEFINITION:
case Kind.INTERFACE_TYPE_DEFINITION:
context.reportError(new GraphQLError(`Invalid use of @authenticated on interface "${parent.name.value}": @authenticated cannot be applied on interfaces, interface fields and interface objects`, {
extensions: {
code: "AUTH_REQUIREMENTS_APPLIED_ON_INTERFACE",
},
}));
return;
case Kind.SCALAR_TYPE_DEFINITION:
case Kind.SCALAR_TYPE_EXTENSION:
context.stateBuilder.scalarType.setAuthenticated(parent.name.value);
break;
case Kind.ENUM_TYPE_DEFINITION:
case Kind.ENUM_TYPE_EXTENSION:
context.stateBuilder.enumType.setAuthenticated(parent.name.value);
break;
}
},
};
}