Version 4.5.0

system/3rd_party/graphql-php/Validator/Rules/KnownDirectives.php

@@ -0,0 +1,77 @@
+<?php
+namespace GraphQL\Validator\Rules;
+
+
+use GraphQL\Error\Error;
+use GraphQL\Language\AST\DirectiveNode;
+use GraphQL\Language\AST\Node;
+use GraphQL\Language\AST\NodeKind;
+use GraphQL\Validator\ValidationContext;
+use GraphQL\Type\Definition\DirectiveLocation;
+
+class KnownDirectives extends AbstractValidationRule
+{
+    static function unknownDirectiveMessage($directiveName)
+    {
+        return "Unknown directive \"$directiveName\".";
+    }
+
+    static function misplacedDirectiveMessage($directiveName, $location)
+    {
+        return "Directive \"$directiveName\" may not be used on \"$location\".";
+    }
+
+    public function getVisitor(ValidationContext $context)
+    {
+        return [
+            NodeKind::DIRECTIVE => function (DirectiveNode $node, $key, $parent, $path, $ancestors) use ($context) {
+                $directiveDef = null;
+                foreach ($context->getSchema()->getDirectives() as $def) {
+                    if ($def->name === $node->name->value) {
+                        $directiveDef = $def;
+                        break;
+                    }
+                }
+
+                if (!$directiveDef) {
+                    $context->reportError(new Error(
+                        self::unknownDirectiveMessage($node->name->value),
+                        [$node]
+                    ));
+                    return ;
+                }
+                $appliedTo = $ancestors[count($ancestors) - 1];
+                $candidateLocation = $this->getLocationForAppliedNode($appliedTo);
+
+                if (!$candidateLocation) {
+                    $context->reportError(new Error(
+                        self::misplacedDirectiveMessage($node->name->value, $node->type),
+                        [$node]
+                    ));
+                } else if (!in_array($candidateLocation, $directiveDef->locations)) {
+                    $context->reportError(new Error(
+                        self::misplacedDirectiveMessage($node->name->value, $candidateLocation),
+                        [ $node ]
+                    ));
+                }
+            }
+        ];
+    }
+
+    private function getLocationForAppliedNode(Node $appliedTo)
+    {
+        switch ($appliedTo->kind) {
+            case NodeKind::OPERATION_DEFINITION:
+                switch ($appliedTo->operation) {
+                    case 'query': return DirectiveLocation::QUERY;
+                    case 'mutation': return DirectiveLocation::MUTATION;
+                    case 'subscription': return DirectiveLocation::SUBSCRIPTION;
+                }
+                break;
+            case NodeKind::FIELD: return DirectiveLocation::FIELD;
+            case NodeKind::FRAGMENT_SPREAD: return DirectiveLocation::FRAGMENT_SPREAD;
+            case NodeKind::INLINE_FRAGMENT: return DirectiveLocation::INLINE_FRAGMENT;
+            case NodeKind::FRAGMENT_DEFINITION: return DirectiveLocation::FRAGMENT_DEFINITION;
+        }
+    }
+}