Fix phpstan/phpstan#14249: Propagate @phpstan-assert-* annotations to First-class callables (Closures) and string callables (#5201)

Author: phpstan-bot

src/Analyser/TypeSpecifier.php

@@ -26,6 +26,7 @@
 use PHPStan\Node\Printer\ExprPrinter;
 use PHPStan\Php\PhpVersion;
 use PHPStan\Reflection\Assertions;
+use PHPStan\Reflection\Callables\CallableParametersAcceptor;
 use PHPStan\Reflection\ExtendedParametersAcceptor;
 use PHPStan\Reflection\ParametersAcceptor;
 use PHPStan\Reflection\ParametersAcceptorSelector;
@@ -570,6 +571,13 @@ public function specifyTypesInCondition(
 				}
 			}
 
+			return $this->handleDefaultTruthyOrFalseyContext($context, $expr, $scope);
+		} elseif ($expr instanceof FuncCall && !($expr->name instanceof Name)) {
+			$specifiedTypes = $this->specifyTypesFromCallableCall($context, $expr, $scope);
+			if ($specifiedTypes !== null) {
+				return $specifiedTypes;
+			}
+
 			return $this->handleDefaultTruthyOrFalseyContext($context, $expr, $scope);
 		} elseif ($expr instanceof MethodCall && $expr->name instanceof Node\Identifier) {
 			$methodCalledOnType = $scope->getType($expr->var);
@@ -1764,6 +1772,38 @@ static function (Type $type, callable $traverse) use ($templateTypeMap, &$contai
 		return $types;
 	}
 
+	private function specifyTypesFromCallableCall(TypeSpecifierContext $context, FuncCall $call, Scope $scope): ?SpecifiedTypes
+	{
+		if (!$call->name instanceof Expr) {
+			return null;
+		}
+
+		$calleeType = $scope->getType($call->name);
+
+		$assertions = null;
+		$parametersAcceptor = null;
+		if ($calleeType->isCallable()->yes()) {
+			$variants = $calleeType->getCallableParametersAcceptors($scope);
+			$parametersAcceptor = ParametersAcceptorSelector::selectFromArgs($scope, $call->getArgs(), $variants);
+			if ($parametersAcceptor instanceof CallableParametersAcceptor) {
+				$assertions = $parametersAcceptor->getAsserts();
+			}
+		}
+
+		if ($assertions === null || $assertions->getAll() === [] || $parametersAcceptor === null) {
+			return null;
+		}
+
+		$asserts = $assertions->mapTypes(static fn (Type $type) => TemplateTypeHelper::resolveTemplateTypes(
+			$type,
+			$parametersAcceptor->getResolvedTemplateTypeMap(),
+			$parametersAcceptor instanceof ExtendedParametersAcceptor ? $parametersAcceptor->getCallSiteVarianceMap() : TemplateTypeVarianceMap::createEmpty(),
+			TemplateTypeVariance::createInvariant(),
+		));
+
+		return $this->specifyTypesFromAsserts($context, $call, $asserts, $parametersAcceptor, $scope);
+	}
+
 	/**
 	 * @return array<string, ConditionalExpressionHolder[]>
 	 */

src/Reflection/Callables/CallableParametersAcceptor.php

@@ -3,6 +3,7 @@
 namespace PHPStan\Reflection\Callables;
 
 use PHPStan\Node\InvalidateExprNode;
+use PHPStan\Reflection\Assertions;
 use PHPStan\Reflection\ParametersAcceptor;
 use PHPStan\TrinaryLogic;
 
@@ -57,4 +58,6 @@ public function getUsedVariables(): array;
 	 */
 	public function mustUseReturnValue(): TrinaryLogic;
 
+	public function getAsserts(): Assertions;
+
 }

src/Reflection/Callables/FunctionCallableVariant.php

@@ -2,6 +2,7 @@
 
 namespace PHPStan\Reflection\Callables;
 
+use PHPStan\Reflection\Assertions;
 use PHPStan\Reflection\ExtendedMethodReflection;
 use PHPStan\Reflection\ExtendedParameterReflection;
 use PHPStan\Reflection\ExtendedParametersAcceptor;
@@ -173,4 +174,9 @@ public function mustUseReturnValue(): TrinaryLogic
 		return $this->function->mustUseReturnValue();
 	}
 
+	public function getAsserts(): Assertions
+	{
+		return $this->function->getAsserts();
+	}
+
 }

src/Reflection/ExtendedCallableFunctionVariant.php

@@ -37,6 +37,7 @@ public function __construct(
 		private array $usedVariables,
 		private TrinaryLogic $acceptsNamedArguments,
 		private TrinaryLogic $mustUseReturnValue,
+		private ?Assertions $assertions = null,
 	)
 	{
 		parent::__construct(
@@ -86,4 +87,9 @@ public function mustUseReturnValue(): TrinaryLogic
 		return $this->mustUseReturnValue;
 	}
 
+	public function getAsserts(): Assertions
+	{
+		return $this->assertions ?? Assertions::createEmpty();
+	}
+
 }

src/Reflection/GenericParametersAcceptorResolver.php

@@ -130,6 +130,7 @@ public static function resolve(array $argTypes, ParametersAcceptor $parametersAc
 				$originalParametersAcceptor->getUsedVariables(),
 				$originalParametersAcceptor->acceptsNamedArguments(),
 				$originalParametersAcceptor->mustUseReturnValue(),
+				$originalParametersAcceptor->getAsserts(),
 			);
 		}
 

src/Reflection/InaccessibleMethod.php

@@ -93,4 +93,9 @@ public function mustUseReturnValue(): TrinaryLogic
 		return TrinaryLogic::createMaybe();
 	}
 
+	public function getAsserts(): Assertions
+	{
+		return Assertions::createEmpty();
+	}
+
 }

src/Reflection/InitializerExprTypeResolver.php

@@ -958,6 +958,7 @@ public function createFirstClassCallable(
 			}
 
 			$parameters = $variant->getParameters();
+			$assertions = $function !== null ? $function->getAsserts() : Assertions::createEmpty();
 			$closureTypes[] = new ClosureType(
 				$parameters,
 				$returnType,
@@ -970,6 +971,7 @@ public function createFirstClassCallable(
 				$impurePoints,
 				acceptsNamedArguments: $acceptsNamedArguments,
 				mustUseReturnValue: $mustUseReturnValue,
+				assertions: $assertions,
 			);
 		}
 

src/Reflection/ParametersAcceptorSelector.php

@@ -897,6 +897,7 @@ private static function wrapAcceptor(ParametersAcceptor $acceptor): ExtendedPara
 				$acceptor->getUsedVariables(),
 				$acceptor->acceptsNamedArguments(),
 				$acceptor->mustUseReturnValue(),
+				$acceptor->getAsserts(),
 			);
 		}
 

src/Reflection/ResolvedFunctionVariantWithCallable.php

@@ -29,6 +29,7 @@ public function __construct(
 		private array $usedVariables,
 		private TrinaryLogic $acceptsNamedArguments,
 		private TrinaryLogic $mustUseReturnValue,
+		private ?Assertions $assertions = null,
 	)
 	{
 	}
@@ -118,4 +119,9 @@ public function mustUseReturnValue(): TrinaryLogic
 		return $this->mustUseReturnValue;
 	}
 
+	public function getAsserts(): Assertions
+	{
+		return $this->assertions ?? Assertions::createEmpty();
+	}
+
 }

src/Reflection/TrivialParametersAcceptor.php

@@ -103,4 +103,9 @@ public function mustUseReturnValue(): TrinaryLogic
 		return TrinaryLogic::createMaybe();
 	}
 
+	public function getAsserts(): Assertions
+	{
+		return Assertions::createEmpty();
+	}
+
 }