Added Nette\Caching\Cache dynamic return type extensions

Author: adaamz

extension.neon

@@ -43,6 +43,16 @@ services:
 			- phpstan.broker.propertiesClassReflectionExtension
 			- phpstan.broker.methodsClassReflectionExtension
 
+	-
+		class: PHPStan\Type\Nette\CachingFallbacksDynamicReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicMethodReturnTypeExtension
+
+	-
+		class: PHPStan\Type\Nette\CachingSaveDynamicReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicMethodReturnTypeExtension
+
 	-
 		class: PHPStan\Type\Nette\ComponentModelArrayAccessDynamicReturnTypeExtension
 		tags:

src/Type/Nette/CachingFallbacksDynamicReturnTypeExtension.php

@@ -0,0 +1,55 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Type\Nette;
+
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Reflection\ParametersAcceptor;
+use PHPStan\Reflection\ParametersAcceptorSelector;
+use PHPStan\Type\DynamicMethodReturnTypeExtension;
+use PHPStan\Type\Type;
+
+final class CachingFallbacksDynamicReturnTypeExtension implements DynamicMethodReturnTypeExtension
+{
+
+	/** @var array<string, int> */
+	private $fallbackMethods = [
+		'load' => 1,
+		'call' => 0,
+		'wrap' => 0,
+	];
+
+	public function getClass(): string
+	{
+		return 'Nette\Caching\Cache';
+	}
+
+	public function isMethodSupported(MethodReflection $methodReflection): bool
+	{
+		$methodName = $methodReflection->getName();
+
+		return array_key_exists($methodName, $this->fallbackMethods);
+	}
+
+	public function getTypeFromMethodCall(
+		MethodReflection $methodReflection,
+		MethodCall $methodCall,
+		Scope $scope
+	): Type
+	{
+		$fallbackParameterIndex = $this->fallbackMethods[$methodReflection->getName()];
+
+		if ($fallbackParameterIndex >= count($methodCall->args)) {
+			return ParametersAcceptorSelector::selectSingle($methodReflection->getVariants())->getReturnType();
+		}
+
+		$fallbackParameterType = $scope->getType($methodCall->args[$fallbackParameterIndex]->value);
+		if (!$fallbackParameterType->isCallable()->yes()) {
+			return ParametersAcceptorSelector::selectSingle($methodReflection->getVariants())->getReturnType();
+		}
+
+		return ParametersAcceptorSelector::selectFromArgs($scope, $methodCall->args, $fallbackParameterType->getCallableParametersAcceptors($scope))->getReturnType();
+	}
+
+}

src/Type/Nette/CachingSaveDynamicReturnTypeExtension.php

@@ -0,0 +1,38 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Type\Nette;
+
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Reflection\ParametersAcceptorSelector;
+use PHPStan\Type\DynamicMethodReturnTypeExtension;
+use PHPStan\Type\Type;
+
+final class CachingSaveDynamicReturnTypeExtension implements DynamicMethodReturnTypeExtension
+{
+
+	public function getClass(): string
+	{
+		return 'Nette\Caching\Cache';
+	}
+
+	public function isMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return $methodReflection->getName() === 'save';
+	}
+
+	public function getTypeFromMethodCall(
+		MethodReflection $methodReflection,
+		MethodCall $methodCall,
+		Scope $scope
+	): Type
+	{
+		if (count($methodCall->args) < 2) {
+			return ParametersAcceptorSelector::selectSingle($methodReflection->getVariants())->getReturnType();
+		}
+
+		return $scope->getType($methodCall->args[1]->value);
+	}
+
+}