Fix phpstan/phpstan#12601: Array keys cannot be narrowed to non-empty-string when used in templates

src/Type/Generic/TemplateTypeHelper.php

@@ -142,9 +142,7 @@
 	{
 		if (!$templateType->getVariance()->covariant()) {
 			$isArrayKey = $templateType->getBound()->describe(VerbosityLevel::precise()) === '(int|string)';
-			if ($type->isScalar()->yes() && $isArrayKey) {
-				$type = $type->generalize(GeneralizePrecision::templateArgument());
-			} elseif ($type->isConstantValue()->yes() && (!$templateType->getBound()->isScalar()->yes() || $isArrayKey)) {
+			if ($type->isConstantValue()->yes() && ($isArrayKey || !$templateType->getBound()->isScalar()->yes())) {
 				$type = $type->generalize(GeneralizePrecision::templateArgument());
 			}
 		}

tests/PHPStan/Analyser/nsrt/bug-12601.php

@@ -0,0 +1,54 @@
+<?php // lint >= 8.0
+
+declare(strict_types = 1);
+
+namespace Bug12601;
+
+use ArrayIterator;
+use IteratorAggregate;
+use Traversable;
+use function PHPStan\Testing\assertType;
+
+/** @implements IteratorAggregate<non-empty-string, non-empty-string> */
+class HelloWorld implements IteratorAggregate
+{
+	/** @param array<non-empty-string, non-empty-string> $map */
+	public function __construct(private array $map) {}
+
+	/** @return Traversable<non-empty-string, non-empty-string> */
+	public function getIterator(): Traversable
+	{
+		$iterator = new ArrayIterator($this->map);
+		assertType('ArrayIterator<non-empty-string, non-empty-string>', $iterator);
+		return $iterator;
+	}
+}
+
+class HelloWorld3
+{
+	/** @var ArrayIterator<int, string> */
+	private ArrayIterator $a;
+
+	/** @param list<string> $map */
+	public function __construct(private array $map) {
+		$a = new ArrayIterator($this->map);
+		assertType('ArrayIterator<int<0, max>, string>', $a);
+
+		$this->a = $a;
+	}
+}
+
+/** @implements IteratorAggregate<int<0, max>, non-empty-string> */
+class HelloWorld2 implements IteratorAggregate
+{
+	/** @param array<int<0, max>, non-empty-string> $map */
+	public function __construct(private array $map) {}
+
+	/** @return ArrayIterator<int<0, max>, non-empty-string> */
+	public function getIterator(): Traversable
+	{
+		$iterator = new ArrayIterator($this->map);
+		assertType('ArrayIterator<int<0, max>, non-empty-string>', $iterator);
+		return $iterator;
+	}
+}

tests/PHPStan/Analyser/nsrt/generics-do-not-generalize.php

@@ -97,7 +97,7 @@ function (): void {
 	/** @var list<string> $a */
 	$a = doFoo();
 
-	assertType('ArrayIterator<int, string>', new ArrayIterator($a));
+	assertType('ArrayIterator<int<0, max>, string>', new ArrayIterator($a));
 };
 
 /**
@@ -115,7 +115,7 @@ function (): void {
 	/** @var list<string> $a */
 	$a = doFoo();
 
-	assertType('ArrayIterator<int, string>', createArrayIterator($a));
+	assertType('ArrayIterator<int<0, max>, string>', createArrayIterator($a));
 };
 
 /** @template T */