Almost final StaticCallHandler

Author: ondrejmirtes

src/Analyser/Generator/ExprHandler/MethodCallHelper.php

@@ -0,0 +1,94 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Analyser\Generator\ExprHandler;
+
+use Generator;
+use PhpParser\Node\Expr;
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Generator\GeneratorNodeScopeResolver;
+use PHPStan\Analyser\Generator\GeneratorScope;
+use PHPStan\Analyser\Generator\RunInFiberRequest;
+use PHPStan\DependencyInjection\AutowiredService;
+use PHPStan\DependencyInjection\Type\DynamicReturnTypeExtensionRegistryProvider;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Reflection\ParametersAcceptor;
+use PHPStan\Type\Type;
+use PHPStan\Type\TypeCombinator;
+use function count;
+
+/**
+ * @phpstan-import-type GeneratorTValueType from GeneratorNodeScopeResolver
+ * @phpstan-import-type GeneratorTSendType from GeneratorNodeScopeResolver
+ */
+#[AutowiredService]
+final class MethodCallHelper
+{
+
+	public function __construct(
+		private DynamicReturnTypeExtensionRegistryProvider $dynamicReturnTypeExtensionRegistryProvider,
+	)
+	{
+	}
+
+	/**
+	 * @param list<non-empty-string> $objectClassNamesFromType
+	 * @return Generator<int, GeneratorTValueType, GeneratorTSendType, ?Type>
+	 */
+	public function methodCallReturnType(
+		GeneratorScope $scope,
+		MethodReflection $methodReflection,
+		ParametersAcceptor $parametersAcceptor,
+		MethodCall|Expr\StaticCall|null $normalizedMethodCall,
+		array $objectClassNamesFromType,
+	): Generator
+	{
+		if ($normalizedMethodCall === null) {
+			//return $this->transformVoidToNull($parametersAcceptor->getReturnType(), $methodCall);
+			return $parametersAcceptor->getReturnType();
+		}
+
+		$resolvedTypes = [];
+		foreach ($objectClassNamesFromType as $className) {
+			if ($normalizedMethodCall instanceof MethodCall) {
+				foreach ($this->dynamicReturnTypeExtensionRegistryProvider->getRegistry()->getDynamicMethodReturnTypeExtensionsForClass($className) as $dynamicMethodReturnTypeExtension) {
+					if (!$dynamicMethodReturnTypeExtension->isMethodSupported($methodReflection)) {
+						continue;
+					}
+
+					$resolvedType = (yield new RunInFiberRequest(static fn () => $dynamicMethodReturnTypeExtension->getTypeFromMethodCall($methodReflection, $normalizedMethodCall, $scope)))->value;
+					if ($resolvedType === null) {
+						continue;
+					}
+
+					$resolvedTypes[] = $resolvedType;
+				}
+			} else {
+				foreach ($this->dynamicReturnTypeExtensionRegistryProvider->getRegistry()->getDynamicStaticMethodReturnTypeExtensionsForClass($className) as $dynamicStaticMethodReturnTypeExtension) {
+					if (!$dynamicStaticMethodReturnTypeExtension->isStaticMethodSupported($methodReflection)) {
+						continue;
+					}
+
+					$resolvedType = (yield new RunInFiberRequest(static fn () => $dynamicStaticMethodReturnTypeExtension->getTypeFromStaticMethodCall(
+						$methodReflection,
+						$normalizedMethodCall,
+						$scope,
+					)))->value;
+					if ($resolvedType === null) {
+						continue;
+					}
+
+					$resolvedTypes[] = $resolvedType;
+				}
+			}
+		}
+
+		if (count($resolvedTypes) > 0) {
+			//return $this->transformVoidToNull(TypeCombinator::union(...$resolvedTypes), $methodCall);
+			return TypeCombinator::union(...$resolvedTypes);
+		}
+
+		//return $this->transformVoidToNull($parametersAcceptor->getReturnType(), $methodCall);
+		return $parametersAcceptor->getReturnType();
+	}
+
+}

src/Analyser/Generator/ExprHandler/NullsafeShortCircuitingHelper.php

@@ -0,0 +1,70 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Analyser\Generator\ExprHandler;
+
+use Generator;
+use PhpParser\Node\Expr;
+use PhpParser\Node\Expr\MethodCall;
+use PhpParser\Node\Expr\PropertyFetch;
+use PHPStan\Analyser\Generator\GeneratorNodeScopeResolver;
+use PHPStan\Analyser\Generator\TypeExprRequest;
+use PHPStan\DependencyInjection\AutowiredService;
+use PHPStan\Type\Type;
+use PHPStan\Type\TypeCombinator;
+
+/**
+ * @phpstan-import-type GeneratorTValueType from GeneratorNodeScopeResolver
+ * @phpstan-import-type GeneratorTSendType from GeneratorNodeScopeResolver
+ */
+#[AutowiredService]
+final class NullsafeShortCircuitingHelper
+{
+
+	/**
+	 * @return Generator<int, GeneratorTValueType, GeneratorTSendType, Type>
+	 */
+	public function getNullsafeShortCircuitingType(Expr $expr, Type $type): Generator
+	{
+		if ($expr instanceof Expr\NullsafePropertyFetch || $expr instanceof Expr\NullsafeMethodCall) {
+			$varType = (yield new TypeExprRequest($expr->var))->type;
+			if (TypeCombinator::containsNull($varType)) {
+				return TypeCombinator::addNull($type);
+			}
+
+			return $type;
+		}
+
+		if ($expr instanceof Expr\ArrayDimFetch) {
+			$gen = $this->getNullsafeShortCircuitingType($expr->var, $type);
+			yield from $gen;
+			return $gen->getReturn();
+		}
+
+		if ($expr instanceof PropertyFetch) {
+			$gen = $this->getNullsafeShortCircuitingType($expr->var, $type);
+			yield from $gen;
+			return $gen->getReturn();
+		}
+
+		if ($expr instanceof Expr\StaticPropertyFetch && $expr->class instanceof Expr) {
+			$gen = $this->getNullsafeShortCircuitingType($expr->class, $type);
+			yield from $gen;
+			return $gen->getReturn();
+		}
+
+		if ($expr instanceof MethodCall) {
+			$gen = $this->getNullsafeShortCircuitingType($expr->var, $type);
+			yield from $gen;
+			return $gen->getReturn();
+		}
+
+		if ($expr instanceof Expr\StaticCall && $expr->class instanceof Expr) {
+			$gen = $this->getNullsafeShortCircuitingType($expr->class, $type);
+			yield from $gen;
+			return $gen->getReturn();
+		}
+
+		return $type;
+	}
+
+}