Add Args::getFirstCallArg and Args::getFirstCallArgType

fp4php · Sep 11, 2022 · 252c2eb · 252c2eb
1 parent f76505b
commit 252c2eb
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/Toolkit/Args.php b/src/Toolkit/Args.php
@@ -67,4 +67,28 @@ public function getCallArgs(
                 fn(Union $type) => new CallArg($arg, new CodeLocation($source, $arg), $type)
             ));
     }
+
+    /**
+     * @return Option<CallArg>
+     */
+    public function getFirstCallArg(
+        MethodReturnTypeProviderEvent |
+        FunctionReturnTypeProviderEvent |
+        AfterFunctionCallAnalysisEvent |
+        AfterMethodCallAnalysisEvent $from,
+    ): Option {
+        return $this->getCallArgs($from)->flatMap(fn(ArrayList $args) => $args->head());
+    }
+
+    /**
+     * @return Option<Union>
+     */
+    public function getFirstCallArgType(
+        MethodReturnTypeProviderEvent |
+        FunctionReturnTypeProviderEvent |
+        AfterFunctionCallAnalysisEvent |
+        AfterMethodCallAnalysisEvent $from,
+    ): Option {
+        return $this->getFirstCallArg($from)->map(fn(CallArg $arg) => $arg->type);
+    }
 }