Simplify NodeGetAttributeTypeExtension

symplify · Oct 2, 2024 · afc2ca4 · afc2ca4
1 parent edf0856
commit afc2ca4
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/ReturnTypeExtension/NodeGetAttributeTypeExtension.php b/src/ReturnTypeExtension/NodeGetAttributeTypeExtension.php
@@ -51,18 +51,16 @@ public function getTypeFromMethodCall(
         MethodReflection $methodReflection,
         MethodCall $methodCall,
         Scope $scope
-    ): Type {
-        $returnType = ParametersAcceptorSelector::selectSingle($methodReflection->getVariants())->getReturnType();
-
+    ): ?Type {
         $firstArg = $methodCall->getArgs()[0];
 
         $argumentValue = $this->resolveArgumentValue($firstArg->value);
         if ($argumentValue === null) {
-            return $returnType;
+            return null;
         }
 
         if (! isset(self::ARGUMENT_KEY_TO_RETURN_TYPE[$argumentValue])) {
-            return $returnType;
+            return null;
         }
 
         $knownReturnType = self::ARGUMENT_KEY_TO_RETURN_TYPE[$argumentValue];