Added AstManipulator::findArguments().

LastDragon-ru · Oct 3, 2023 · 543a3c3 · 543a3c3
1 parent c83c1ee
commit 543a3c3
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/packages/graphql/src/Utils/AstManipulator.php b/packages/graphql/src/Utils/AstManipulator.php
@@ -532,6 +532,29 @@ public function getField(
         return $field;
     }
 
+    /**
+     * @param callable(InputValueDefinitionNode|Argument|ArgumentNode): bool $closure
+     *
+     * @return array<string, ($node is FieldDefinitionNode ? InputValueDefinitionNode : ($node is FieldDefinition ? Argument : ArgumentNode))>
+     */
+    public function findArguments(
+        FieldDefinitionNode|FieldDefinition|DirectiveNode $node,
+        callable $closure,
+    ): array {
+        $arguments = [];
+        $args      = $node instanceof FieldDefinitionNode || $node instanceof DirectiveNode
+            ? $node->arguments
+            : $node->args;
+
+        foreach ($args as $arg) {
+            if ($closure($arg)) {
+                $arguments[$this->getName($arg)] = $arg;
+            }
+        }
+
+        return $arguments;
+    }
+
     /**
      * @param callable(InputValueDefinitionNode|Argument|ArgumentNode): bool $closure
      *