Explicitly check for enum type when enum literal is met

graphpql · Aug 30, 2024 · 6f73b19 · 6f73b19
1 parent 94ce372
commit 6f73b19
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/Value/ConvertParserValueVisitor.php b/src/Value/ConvertParserValueVisitor.php
@@ -19,6 +19,7 @@
 use Graphpinator\Parser\Value\ValueVisitor;
 use Graphpinator\Parser\Value\VariableRef;
 use Graphpinator\Typesystem\Contract\Inputable;
+use Graphpinator\Typesystem\EnumType;
 use Graphpinator\Typesystem\InputType;
 use Graphpinator\Typesystem\ListType;
 use Graphpinator\Typesystem\NotNullType;
@@ -40,6 +41,16 @@ public function visitLiteral(Literal $literal) : InputedValue
 
     public function visitEnumLiteral(EnumLiteral $enumLiteral) : InputedValue
     {
+        if ($this->type instanceof NotNullType) {
+            $this->type = $this->type->getInnerType();
+
+            return $enumLiteral->accept($this);
+        }
+
+        if (!$this->type instanceof EnumType) {
+            throw new InvalidValue($this->type->printName(), $enumLiteral->getRawValue(), true);
+        }
+
         return $this->type->accept(new ConvertRawValueVisitor($enumLiteral->getRawValue(), $this->path));
     }