allow non-nullable field argument (#6)

efabrica-team · Feb 28, 2024 · 03d298f · 03d298f
1 parent 0df1d1a
commit 03d298f
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,8 @@
 # Changelog
 
 ## [Unreleased]
+### Added
+- Allow non-nullable field argument
 
 ## [0.2.2] - 2024-01-26
 ### Added

diff --git a/src/Schema/Definition/Arguments/FieldArgument.php b/src/Schema/Definition/Arguments/FieldArgument.php
@@ -22,6 +22,8 @@ class FieldArgument
     protected bool $isDefaultValueSet = false;
 
     protected bool $multi = false;
+
+    protected bool $nullable = true;
 
     public function __construct(string $name, Type $type)
     {
@@ -92,4 +94,21 @@ public function setMulti(bool $multi = true): self
         $this->multi = $multi;
         return $this;
     }
+
+    public function isNullable(): bool
+    {
+        return $this->nullable;
+    }
+
+    public function setNullable(bool $nullable = true): FieldArgument
+    {
+        $this->nullable = $nullable;
+        return $this;
+    }
+
+    public function setNonNullable(bool $nonNullable = true): FieldArgument
+    {
+        $this->nullable = !$nonNullable;
+        return $this;
+    }
 }
diff --git a/src/Schema/Transformers/WebonyxSchemaTransformer.php b/src/Schema/Transformers/WebonyxSchemaTransformer.php
@@ -190,6 +190,10 @@ private function transformFieldArgument(FieldArgument $fieldArgument): array
             $argumentType = WebonyxType::listOf($argumentType);
         }
 
+        if (!$fieldArgument->isNullable()) {
+            $argumentType = WebonyxType::nonNull($argumentType);
+        }
+
         $argumentResult = [
             'name' => $fieldArgument->getName(),
             'type' => $argumentType,