You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.6 KiB
85 lines
2.6 KiB
import { GraphQLError } from '../../error/GraphQLError.mjs'; |
|
import { Kind } from '../../language/kinds.mjs'; |
|
import { |
|
isTypeDefinitionNode, |
|
isTypeExtensionNode, |
|
} from '../../language/predicates.mjs'; |
|
import { specifiedDirectives } from '../../type/directives.mjs'; |
|
|
|
/** |
|
* Unique directive names per location |
|
* |
|
* A GraphQL document is only valid if all non-repeatable directives at |
|
* a given location are uniquely named. |
|
* |
|
* See https://spec.graphql.org/draft/#sec-Directives-Are-Unique-Per-Location |
|
*/ |
|
export function UniqueDirectivesPerLocationRule(context) { |
|
const uniqueDirectiveMap = Object.create(null); |
|
const schema = context.getSchema(); |
|
const definedDirectives = schema |
|
? schema.getDirectives() |
|
: specifiedDirectives; |
|
|
|
for (const directive of definedDirectives) { |
|
uniqueDirectiveMap[directive.name] = !directive.isRepeatable; |
|
} |
|
|
|
const astDefinitions = context.getDocument().definitions; |
|
|
|
for (const def of astDefinitions) { |
|
if (def.kind === Kind.DIRECTIVE_DEFINITION) { |
|
uniqueDirectiveMap[def.name.value] = !def.repeatable; |
|
} |
|
} |
|
|
|
const schemaDirectives = Object.create(null); |
|
const typeDirectivesMap = Object.create(null); |
|
return { |
|
// Many different AST nodes may contain directives. Rather than listing |
|
// them all, just listen for entering any node, and check to see if it |
|
// defines any directives. |
|
enter(node) { |
|
if (!('directives' in node) || !node.directives) { |
|
return; |
|
} |
|
|
|
let seenDirectives; |
|
|
|
if ( |
|
node.kind === Kind.SCHEMA_DEFINITION || |
|
node.kind === Kind.SCHEMA_EXTENSION |
|
) { |
|
seenDirectives = schemaDirectives; |
|
} else if (isTypeDefinitionNode(node) || isTypeExtensionNode(node)) { |
|
const typeName = node.name.value; |
|
seenDirectives = typeDirectivesMap[typeName]; |
|
|
|
if (seenDirectives === undefined) { |
|
typeDirectivesMap[typeName] = seenDirectives = Object.create(null); |
|
} |
|
} else { |
|
seenDirectives = Object.create(null); |
|
} |
|
|
|
for (const directive of node.directives) { |
|
const directiveName = directive.name.value; |
|
|
|
if (uniqueDirectiveMap[directiveName]) { |
|
if (seenDirectives[directiveName]) { |
|
context.reportError( |
|
new GraphQLError( |
|
`The directive "@${directiveName}" can only be used once at this location.`, |
|
{ |
|
nodes: [seenDirectives[directiveName], directive], |
|
}, |
|
), |
|
); |
|
} else { |
|
seenDirectives[directiveName] = directive; |
|
} |
|
} |
|
} |
|
}, |
|
}; |
|
}
|
|
|