From 59c3d50bd31c821fa84e817b75c3aadbf8bd9446 Mon Sep 17 00:00:00 2001
From: adrew <klimichkartorgnusov@gmail.com>
Date: Sun, 11 Sep 2022 21:16:31 +0300
Subject: [PATCH] Add AfterFunctionCallAnalysisEvent support for Types::setType

---
 src/Toolkit/Types.php | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/Toolkit/Types.php b/src/Toolkit/Types.php
index 09aafc7..a50821d 100644
--- a/src/Toolkit/Types.php
+++ b/src/Toolkit/Types.php
@@ -167,6 +167,7 @@ public function setType(
         StatementsSource |
         NodeTypeProvider |
         AfterMethodCallAnalysisEvent |
+        AfterFunctionCallAnalysisEvent |
         MethodReturnTypeProviderEvent |
         AfterStatementAnalysisEvent |
         FunctionReturnTypeProviderEvent |
@@ -183,6 +184,7 @@ public function setType(
             $to instanceof FunctionReturnTypeProviderEvent => $to->getStatementsSource()->getNodeTypeProvider(),
             $to instanceof AfterExpressionAnalysisEvent => $to->getStatementsSource()->getNodeTypeProvider(),
             $to instanceof AfterMethodCallAnalysisEvent => $to->getStatementsSource()->getNodeTypeProvider(),
+            $to instanceof AfterFunctionCallAnalysisEvent => $to->getStatementsSource()->getNodeTypeProvider(),
         };
 
         $provider->setType($for, $type);