Create ComponentLookupMethodReflectionExtension.php

Author: adaamz

src/Type/Nette/ComponentLookupMethodReflectionExtension.php

@@ -0,0 +1,42 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Type\Nette;
+
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\Constant\ConstantBooleanType;
+use PHPStan\Type\DynamicMethodReturnTypeExtension;
+use PHPStan\Type\NullType;
+use PHPStan\Type\Type;
+use PHPStan\Type\UnionType;
+
+final class ComponentLookupMethodReflectionExtension implements DynamicMethodReturnTypeExtension
+{
+    public function getClass(): string
+    {
+        return \Nette\ComponentModel\Component::class;
+    }
+
+    public function isMethodSupported(MethodReflection $methodReflection): bool
+    {
+        return $methodReflection->getName() === 'lookup';
+    }
+
+    public function getTypeFromMethodCall(MethodReflection $methodReflection, MethodCall $methodCall, Scope $scope): Type
+    {
+        if (count($methodCall->args) > 1) {
+            $paramNeedExpr = $methodCall->args[1]->value;
+            $paramNeedType = $scope->getType($paramNeedExpr);
+
+            if ($paramNeedType instanceof ConstantBooleanType && $paramNeedType->getValue() === false) {
+                if ($methodReflection->getReturnType()->accepts(new NullType()) === false) {
+                    return new UnionType([$methodReflection->getReturnType(), new NullType()]);
+                }
+            }
+        }
+
+        return $methodReflection->getReturnType();
+    }
+
+}