Narrow explode return to constant arrays for small positive limits

Author: herndlm

Diff for: src/Type/Php/ExplodeFunctionDynamicReturnTypeExtension.php

@@ -11,7 +11,9 @@
 use PHPStan\Type\Accessory\AccessoryUppercaseStringType;
 use PHPStan\Type\Accessory\NonEmptyArrayType;
 use PHPStan\Type\ArrayType;
+use PHPStan\Type\Constant\ConstantArrayTypeBuilder;
 use PHPStan\Type\Constant\ConstantBooleanType;
+use PHPStan\Type\Constant\ConstantIntegerType;
 use PHPStan\Type\Constant\ConstantStringType;
 use PHPStan\Type\DynamicFunctionReturnTypeExtension;
 use PHPStan\Type\IntegerRangeType;
@@ -24,10 +26,13 @@
 use PHPStan\Type\TypeCombinator;
 use PHPStan\Type\TypeUtils;
 use function count;
+use const PHP_INT_MAX;
 
 final class ExplodeFunctionDynamicReturnTypeExtension implements DynamicFunctionReturnTypeExtension
 {
 
+	private const CONST_ARRAY_LIMIT = 8;
+
 	public function __construct(private PhpVersion $phpVersion)
 	{
 	}
@@ -73,11 +78,21 @@ public function getTypeFromFunctionCall(
 		}
 
 		$returnType = TypeCombinator::intersect(new ArrayType(new IntegerType(), $returnValueType), new AccessoryArrayListType());
-		if (
-			!isset($args[2])
-			|| IntegerRangeType::fromInterval(0, null)->isSuperTypeOf($scope->getType($args[2]->value))->yes()
-		) {
-			$returnType = TypeCombinator::intersect($returnType, new NonEmptyArrayType());
+		$limitType = isset($args[2]) ? $scope->getType($args[2]->value) : new ConstantIntegerType(PHP_INT_MAX);
+		if (IntegerRangeType::fromInterval(0, null)->isSuperTypeOf($limitType)->yes()) {
+			$constantScalarTypes = $limitType->getConstantScalarTypes();
+			if (count($constantScalarTypes) === 1 && IntegerRangeType::fromInterval(0, self::CONST_ARRAY_LIMIT)->isSuperTypeOf($limitType)->yes()) {
+				$limit = (int) $constantScalarTypes[0]->getValue() ?: 1; // 0 is treated as 1
+
+				$builder = ConstantArrayTypeBuilder::createEmpty();
+				for ($i = 0; $i < $limit; $i++) {
+					$builder->setOffsetValueType(null, $returnValueType, $i !== 0);
+				}
+
+				$returnType = $builder->getArray();
+			} else {
+				$returnType = TypeCombinator::intersect($returnType, new NonEmptyArrayType());
+			}
 		}
 
 		if (!$this->phpVersion->throwsValueErrorForInternalFunctions() && $isEmptyString->maybe()) {

Diff for: tests/PHPStan/Analyser/nsrt/bug-3961-php8.php

@@ -12,8 +12,8 @@ public function doFoo(string $v, string $d, $m): void
 		assertType('non-empty-list<string>', explode('.', $v));
 		assertType('*NEVER*', explode('', $v));
 		assertType('list<string>', explode('.', $v, -2));
-		assertType('non-empty-list<string>', explode('.', $v, 0));
-		assertType('non-empty-list<string>', explode('.', $v, 1));
+		assertType('array{string}', explode('.', $v, 0));
+		assertType('array{string}', explode('.', $v, 1));
 		assertType('non-empty-list<string>', explode($d, $v));
 		assertType('non-empty-list<string>', explode($m, $v));
 	}

Diff for: tests/PHPStan/Analyser/nsrt/bug-3961.php

@@ -12,8 +12,8 @@ public function doFoo(string $v, string $d, $m): void
 		assertType('non-empty-list<string>', explode('.', $v));
 		assertType('false', explode('', $v));
 		assertType('list<string>', explode('.', $v, -2));
-		assertType('non-empty-list<string>', explode('.', $v, 0));
-		assertType('non-empty-list<string>', explode('.', $v, 1));
+		assertType('array{string}', explode('.', $v, 0));
+		assertType('array{string}', explode('.', $v, 1));
 		assertType('non-empty-list<string>|false', explode($d, $v));
 		assertType('(non-empty-list<string>|false)', explode($m, $v));
 	}

Diff for: tests/PHPStan/Analyser/nsrt/explode-php7.php

@@ -0,0 +1,34 @@
+<?php // lint < 8.0
+
+namespace ExplodePhp7;
+
+use function PHPStan\Testing\assertType;
+
+class Foo
+{
+
+	/**
+	 * @param non-empty-string $nonEmptyString
+	 */
+	public function constantArrays(string $string, string $nonEmptyString): void
+	{
+		$strings = explode(',', $string, 0);
+		assertType('array{string}', $strings);
+
+		$strings = explode(',', $string, 2);
+		assertType('array{0: string, 1?: string}', $strings);
+
+		$strings = explode(rand(0, 1) ? '' : ',', $string, 2);
+		assertType('array{0: string, 1?: string}|false', $strings);
+
+		$strings = explode(',', $string, 16);
+		assertType('non-empty-list<string>', $strings);
+
+		$strings = explode(',', $nonEmptyString, 2);
+		assertType('array{0: string, 1?: string}', $strings);
+
+		$strings = explode(',', $nonEmptyString, 16);
+		assertType('non-empty-list<string>', $strings);
+	}
+
+}

Diff for: tests/PHPStan/Analyser/nsrt/explode-php8.php

@@ -0,0 +1,34 @@
+<?php // lint >= 8.0
+
+namespace ExplodePhp8;
+
+use function PHPStan\Testing\assertType;
+
+class Foo
+{
+
+	/**
+	 * @param non-empty-string $nonEmptyString
+	 */
+	public function constantArrays(string $string, string $nonEmptyString): void
+	{
+		$strings = explode(',', $string, 0);
+		assertType('array{string}', $strings);
+
+		$strings = explode(',', $string, 2);
+		assertType('array{0: string, 1?: string}', $strings);
+
+		$strings = explode(rand(0, 1) ? '' : ',', $string, 2);
+		assertType('array{0: string, 1?: string}', $strings);
+
+		$strings = explode(',', $string, 16);
+		assertType('non-empty-list<string>', $strings);
+
+		$strings = explode(',', $nonEmptyString, 2);
+		assertType('array{0: string, 1?: string}', $strings);
+
+		$strings = explode(',', $nonEmptyString, 16);
+		assertType('non-empty-list<string>', $strings);
+	}
+
+}