matching() returns the same type it was called on

Author: ondrejmirtes

composer.json

@@ -6,7 +6,7 @@
 	"prefer-stable": true,
 	"require": {
 		"php": "~7.0",
-		"phpstan/phpstan": "^0.6",
+		"phpstan/phpstan": "^0.6.3",
 		"doctrine/common": "^2.7",
 		"doctrine/orm": "^2.5"
 	},

extension.neon

@@ -3,6 +3,10 @@ services:
 		class: PHPStan\Reflection\Doctrine\DoctrineSelectableClassReflectionExtension
 		tags:
 			- phpstan.broker.methodsClassReflectionExtension
+	-
+		class: PHPStan\Type\Doctrine\DoctrineSelectableDynamicReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicMethodReturnTypeExtension
 	-
 		class: PHPStan\Type\Doctrine\EntityManagerFindDynamicReturnTypeExtension
 		tags:

src/Type/Doctrine/DoctrineSelectableDynamicReturnTypeExtension.php

@@ -0,0 +1,32 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Type\Doctrine;
+
+use PhpParser\Node\Expr\MethodCall;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Type\Type;
+
+class DoctrineSelectableDynamicReturnTypeExtension implements \PHPStan\Type\DynamicMethodReturnTypeExtension
+{
+
+	public static function getClass(): string
+	{
+		return \Doctrine\Common\Collections\Collection::class;
+	}
+
+	public function isMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return $methodReflection->getName() === 'matching';
+	}
+
+	public function getTypeFromMethodCall(
+		MethodReflection $methodReflection,
+		MethodCall $methodCall,
+		Scope $scope
+	): Type
+	{
+		return $scope->getType($methodCall->var);
+	}
+
+}