Type::getFiniteTypes()

phpstan · Jun 19, 2023 · 7912caf · 7912caf
1 parent 175307e
commit 7912caf
Show file tree

Hide file tree

Showing 38 changed files with 423 additions and 1 deletion.
diff --git a/src/Type/Accessory/AccessoryArrayListType.php b/src/Type/Accessory/AccessoryArrayListType.php
@@ -474,6 +474,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		return new IdentifierTypeNode('list');

diff --git a/src/Type/Accessory/AccessoryLiteralStringType.php b/src/Type/Accessory/AccessoryLiteralStringType.php
@@ -335,6 +335,11 @@ public function exponentiate(Type $exponent): Type
 		]);
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		return new IdentifierTypeNode('literal-string');

diff --git a/src/Type/Accessory/AccessoryNonEmptyStringType.php b/src/Type/Accessory/AccessoryNonEmptyStringType.php
@@ -340,6 +340,11 @@ public function exponentiate(Type $exponent): Type
 		]);
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		return new IdentifierTypeNode('non-empty-string');

diff --git a/src/Type/Accessory/AccessoryNonFalsyStringType.php b/src/Type/Accessory/AccessoryNonFalsyStringType.php
@@ -331,6 +331,11 @@ public function exponentiate(Type $exponent): Type
 		]);
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		return new IdentifierTypeNode('non-falsy-string');

diff --git a/src/Type/Accessory/AccessoryNumericStringType.php b/src/Type/Accessory/AccessoryNumericStringType.php
@@ -342,6 +342,11 @@ public function exponentiate(Type $exponent): Type
 		]);
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		return new IdentifierTypeNode('numeric-string');

diff --git a/src/Type/Accessory/HasMethodType.php b/src/Type/Accessory/HasMethodType.php
@@ -190,6 +190,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public static function __set_state(array $properties): Type
 	{
 		return new self($properties['methodName']);

diff --git a/src/Type/Accessory/HasOffsetType.php b/src/Type/Accessory/HasOffsetType.php
@@ -363,6 +363,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public static function __set_state(array $properties): Type
 	{
 		return new self($properties['offsetType']);

diff --git a/src/Type/Accessory/HasOffsetValueType.php b/src/Type/Accessory/HasOffsetValueType.php
@@ -419,6 +419,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public static function __set_state(array $properties): Type
 	{
 		return new self($properties['offsetType'], $properties['valueType']);

diff --git a/src/Type/Accessory/HasPropertyType.php b/src/Type/Accessory/HasPropertyType.php
@@ -152,6 +152,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public static function __set_state(array $properties): Type
 	{
 		return new self($properties['propertyName']);

diff --git a/src/Type/Accessory/NonEmptyArrayType.php b/src/Type/Accessory/NonEmptyArrayType.php
@@ -429,6 +429,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public static function __set_state(array $properties): Type
 	{
 		return new self();

diff --git a/src/Type/Accessory/OversizedArrayType.php b/src/Type/Accessory/OversizedArrayType.php
@@ -415,6 +415,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public static function __set_state(array $properties): Type
 	{
 		return new self();

diff --git a/src/Type/ArrayType.php b/src/Type/ArrayType.php
@@ -728,6 +728,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	/**
 	 * @param mixed[] $properties
 	 */

diff --git a/src/Type/BooleanType.php b/src/Type/BooleanType.php
@@ -135,6 +135,14 @@ public function tryRemove(Type $typeToRemove): ?Type
 		return null;
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [
+			new ConstantBooleanType(true),
+			new ConstantBooleanType(false),
+		];
+	}
+
 	public function exponentiate(Type $exponent): Type
 	{
 		return ExponentiateHelper::exponentiate($this, $exponent);

diff --git a/src/Type/CallableType.php b/src/Type/CallableType.php
@@ -524,6 +524,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		if ($this->isCommonCallable) {

diff --git a/src/Type/ClosureType.php b/src/Type/ClosureType.php
@@ -588,6 +588,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		$parameters = [];

diff --git a/src/Type/Constant/ConstantArrayType.php b/src/Type/Constant/ConstantArrayType.php
@@ -3,6 +3,7 @@
 namespace PHPStan\Type\Constant;
 
 use Nette\Utils\Strings;
+use PHPStan\Internal\CombinationsHelper;
 use PHPStan\Php\PhpVersion;
 use PHPStan\PhpDocParser\Ast\ConstExpr\ConstExprIntegerNode;
 use PHPStan\PhpDocParser\Ast\ConstExpr\ConstExprStringNode;
@@ -13,6 +14,7 @@
 use PHPStan\PhpDocParser\Ast\Type\TypeNode;
 use PHPStan\Reflection\ClassMemberAccessAnswerer;
 use PHPStan\Reflection\InaccessibleMethod;
+use PHPStan\Reflection\InitializerExprTypeResolver;
 use PHPStan\Reflection\ParametersAcceptor;
 use PHPStan\Reflection\TrivialParametersAcceptor;
 use PHPStan\ShouldNotHappenException;
@@ -1606,6 +1608,44 @@ public static function isValidIdentifier(string $value): bool
 		return $result !== null;
 	}
 
+	public function getFiniteTypes(): array
+	{
+		$arraysArraysForCombinations = [];
+		$count = 0;
+		foreach ($this->getAllArrays() as $array) {
+			$values = $array->getValueTypes();
+			$arraysForCombinations = [];
+			$combinationCount = 1;
+			foreach ($values as $valueType) {
+				if ($valueType->getFiniteTypes() === []) {
+					return [];
+				}
+				$arraysForCombinations[] = $valueType->getFiniteTypes();
+				$combinationCount *= count($valueType->getFiniteTypes());
+			}
+			$arraysArraysForCombinations[] = $arraysForCombinations;
+			$count += $combinationCount;
+		}
+
+		if ($count > InitializerExprTypeResolver::CALCULATE_SCALARS_LIMIT) {
+			return [];
+		}
+
+		$finiteTypes = [];
+		foreach ($arraysArraysForCombinations as $arraysForCombinations) {
+			$combinations = CombinationsHelper::combinations($arraysForCombinations);
+			foreach ($combinations as $combination) {
+				$builder = ConstantArrayTypeBuilder::createEmpty();
+				foreach ($combination as $i => $v) {
+					$builder->setOffsetValueType($this->keyTypes[$i], $v);
+				}
+				$finiteTypes[] = $builder->getArray();
+			}
+		}
+
+		return $finiteTypes;
+	}
+
 	/**
 	 * @param mixed[] $properties
 	 */

diff --git a/src/Type/FloatType.php b/src/Type/FloatType.php
@@ -269,6 +269,11 @@ public function toPhpDocNode(): TypeNode
 		return new IdentifierTypeNode('float');
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	/**
 	 * @param mixed[] $properties
 	 */

diff --git a/src/Type/IntegerRangeType.php b/src/Type/IntegerRangeType.php
@@ -7,6 +7,7 @@
 use PHPStan\PhpDocParser\Ast\Type\GenericTypeNode;
 use PHPStan\PhpDocParser\Ast\Type\IdentifierTypeNode;
 use PHPStan\PhpDocParser\Ast\Type\TypeNode;
+use PHPStan\Reflection\InitializerExprTypeResolver;
 use PHPStan\TrinaryLogic;
 use PHPStan\Type\Constant\ConstantBooleanType;
 use PHPStan\Type\Constant\ConstantIntegerType;
@@ -615,6 +616,25 @@ public function exponentiate(Type $exponent): Type
 		return parent::exponentiate($exponent);
 	}
 
+	public function getFiniteTypes(): array
+	{
+		if ($this->min === null || $this->max === null) {
+			return [];
+		}
+
+		$size = $this->max - $this->min;
+		if ($size > InitializerExprTypeResolver::CALCULATE_SCALARS_LIMIT) {
+			return [];
+		}
+
+		$types = [];
+		for ($i = $this->min; $i <= $this->max; $i++) {
+			$types[] = new ConstantIntegerType($i);
+		}
+
+		return $types;
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		if ($this->min === null) {

diff --git a/src/Type/IntegerType.php b/src/Type/IntegerType.php
@@ -157,6 +157,11 @@ public function tryRemove(Type $typeToRemove): ?Type
 		return null;
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function exponentiate(Type $exponent): Type
 	{
 		return ExponentiateHelper::exponentiate($this, $exponent);

diff --git a/src/Type/IntersectionType.php b/src/Type/IntersectionType.php
@@ -10,6 +10,7 @@
 use PHPStan\Reflection\ClassMemberAccessAnswerer;
 use PHPStan\Reflection\ConstantReflection;
 use PHPStan\Reflection\ExtendedMethodReflection;
+use PHPStan\Reflection\InitializerExprTypeResolver;
 use PHPStan\Reflection\ParametersAcceptor;
 use PHPStan\Reflection\PropertyReflection;
 use PHPStan\Reflection\TrivialParametersAcceptor;
@@ -40,6 +41,7 @@
 use function implode;
 use function in_array;
 use function ksort;
+use function md5;
 use function sprintf;
 use function strlen;
 use function substr;
@@ -719,7 +721,7 @@ public function getEnumCases(): array
 		foreach ($this->types as $type) {
 			$oneType = [];
 			foreach ($type->getEnumCases() as $enumCase) {
-				$oneType[$enumCase->describe(VerbosityLevel::typeOnly())] = $enumCase;
+				$oneType[md5($enumCase->describe(VerbosityLevel::typeOnly()))] = $enumCase;
 			}
 			$compare[] = $oneType;
 		}
@@ -995,6 +997,26 @@ public function exponentiate(Type $exponent): Type
 		return $this->intersectTypes(static fn (Type $type): Type => $type->exponentiate($exponent));
 	}
 
+	public function getFiniteTypes(): array
+	{
+		$compare = [];
+		foreach ($this->types as $type) {
+			$oneType = [];
+			foreach ($type->getFiniteTypes() as $finiteType) {
+				$oneType[md5($finiteType->describe(VerbosityLevel::typeOnly()))] = $finiteType;
+			}
+			$compare[] = $oneType;
+		}
+
+		$result = array_values(array_intersect_key(...$compare));
+
+		if (count($result) > InitializerExprTypeResolver::CALCULATE_SCALARS_LIMIT) {
+			return [];
+		}
+
+		return $result;
+	}
+
 	/**
 	 * @param mixed[] $properties
 	 */

diff --git a/src/Type/IterableType.php b/src/Type/IterableType.php
@@ -470,6 +470,11 @@ public function exponentiate(Type $exponent): Type
 		return new ErrorType();
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		$isMixedKeyType = $this->keyType instanceof MixedType && $this->keyType->describe(VerbosityLevel::precise()) === 'mixed';

diff --git a/src/Type/MixedType.php b/src/Type/MixedType.php
@@ -909,6 +909,11 @@ public function exponentiate(Type $exponent): Type
 		]);
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		return new IdentifierTypeNode('mixed');

diff --git a/src/Type/NeverType.php b/src/Type/NeverType.php
@@ -496,6 +496,11 @@ public function exponentiate(Type $exponent): Type
 		return $this;
 	}
 
+	public function getFiniteTypes(): array
+	{
+		return [];
+	}
+
 	public function toPhpDocNode(): TypeNode
 	{
 		return new IdentifierTypeNode('never');