Implemented Doctrine\DBAL\Types\Type::getType handling

README.md

@@ -17,6 +17,7 @@ This extension provides following features:
 * Provides correct return type for `Doctrine\ORM\EntityManager::find`, `getReference` and `getPartialReference` when `Foo::class` entity class name is provided as the first argument
 * Adds missing `matching` method on `Doctrine\Common\Collections\Collection`. This can be turned off by setting `parameters.doctrine.allCollectionsSelectable` to `false`.
 * Also supports Doctrine ODM.
+* Adds `Doctrine\DBAL\Types\Type::getType` return type support.
 
 ## Installation
 

composer.json

@@ -34,7 +34,8 @@
 		"doctrine/common": "^2.7",
 		"doctrine/orm": "^2.5",
 		"doctrine/collections": "^1.0",
-		"doctrine/mongodb-odm": "^1.2"
+		"doctrine/mongodb-odm": "^1.2",
+		"doctrine/dbal": "^2.9"
 	},
 	"conflict": {
 		"doctrine/common": "<2.7",

extension.neon

@@ -119,6 +119,11 @@ services:
 		arguments:
 			managerClass: Doctrine\Common\Persistence\ObjectManager
 
+	dbalTypeGetType:
+		class: PHPStan\Type\Doctrine\GetDBALTypeDynamicReturnTypeExtension
+		tags:
+			- phpstan.broker.dynamicStaticMethodReturnTypeExtension
+
 	## NewExprDynamicReturnTypeExtensions
 
 	-

src/Type/Doctrine/GetDBALTypeDynamicReturnTypeExtension.php

@@ -0,0 +1,60 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Type\Doctrine;
+
+use PhpParser\Node\Expr;
+use PhpParser\Node\Expr\StaticCall;
+use PhpParser\Node\Name;
+use PhpParser\Node\Scalar;
+use PHPStan\Analyser\Scope;
+use PHPStan\Reflection\MethodReflection;
+use PHPStan\Reflection\ParametersAcceptorSelector;
+use PHPStan\Type\DynamicStaticMethodReturnTypeExtension;
+use PHPStan\Type\ObjectType;
+use PHPStan\Type\Type;
+
+final class GetDBALTypeDynamicReturnTypeExtension implements DynamicStaticMethodReturnTypeExtension
+{
+
+	public function getClass(): string
+	{
+		return 'Doctrine\DBAL\Types\Type';
+	}
+
+	public function isStaticMethodSupported(MethodReflection $methodReflection): bool
+	{
+		return $methodReflection->getName() === 'getType';
+	}
+
+	public function getTypeFromStaticMethodCall(
+		MethodReflection $methodReflection,
+		StaticCall $methodCall,
+		Scope $scope
+	): Type
+	{
+		if (count($methodCall->args) === 0) {
+			return ParametersAcceptorSelector::selectSingle(
+				$methodReflection->getVariants()
+			)->getReturnType();
+		}
+
+		$dbalTypeArg = $methodCall->args[0]->value;
+		return $this->detectReturnType($methodReflection, $dbalTypeArg);
+	}
+
+	private function detectReturnType(MethodReflection $methodReflection, Expr $dbalTypeArg): Type
+	{
+		if ($dbalTypeArg instanceof Expr\ClassConstFetch && $dbalTypeArg->class instanceof Name) {
+			return new ObjectType($dbalTypeArg->class->toString());
+		}
+
+		if ($dbalTypeArg instanceof Scalar\String_) {
+			return new ObjectType($dbalTypeArg->value);
+		}
+
+		return ParametersAcceptorSelector::selectSingle(
+			$methodReflection->getVariants()
+		)->getReturnType();
+	}
+
+}

tests/DoctrineIntegration/DBAL/data/dbalTypesTypeStaticGetTypeUsage.php

@@ -0,0 +1,20 @@
+<?php
+declare(strict_types=1);
+
+namespace PHPStan\DoctrineIntegration\DBAL\DBALTypesTypeStaticGetTypeUsage;
+
+use Doctrine\DBAL\Types\JsonType;
+use Doctrine\DBAL\Types\Type;
+
+final class Example
+{
+    public function typeWithClassConstant(): JsonType
+    {
+        return Type::getType(JsonType::class);
+    }
+
+    public function typeWithString(): JsonType
+    {
+        return Type::getType('Doctrine\DBAL\Types\JsonType');
+    }
+}

tests/DoctrineIntegration/DBAL/phpstan.neon

@@ -0,0 +1,3 @@
+includes:
+	- ../../../extension.neon
+	- ../../../vendor/phpstan/phpstan/conf/bleedingEdge.neon