UniqueDirectivesPerLocation.php 1.34 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
<?php
namespace GraphQL\Validator\Rules;

use GraphQL\Error\Error;
use GraphQL\Language\AST\DirectiveNode;
use GraphQL\Language\AST\Node;
use GraphQL\Validator\ValidationContext;

class UniqueDirectivesPerLocation extends AbstractValidationRule
{
    static function duplicateDirectiveMessage($directiveName)
    {
        return 'The directive "'.$directiveName.'" can only be used once at this location.';
    }

    public function getVisitor(ValidationContext $context)
    {
        return [
            'enter' => function(Node $node) use ($context) {
                if (isset($node->directives)) {
                    $knownDirectives = [];
                    foreach ($node->directives as $directive) {
                        /** @var DirectiveNode $directive */
                        $directiveName = $directive->name->value;
                        if (isset($knownDirectives[$directiveName])) {
                            $context->reportError(new Error(
                                self::duplicateDirectiveMessage($directiveName),
                                [$knownDirectives[$directiveName], $directive]
                            ));
                        } else {
                            $knownDirectives[$directiveName] = $directive;
                        }
                    }
                }
            }
        ];
    }
}