Rely on RuleLevelHelper in InvalidKeyInArrayItemRule

Author: VincentLanglet

src/Rules/Arrays/InvalidKeyInArrayItemRule.php

@@ -4,11 +4,12 @@
 
 use PhpParser\Node;
 use PHPStan\Analyser\Scope;
-use PHPStan\DependencyInjection\AutowiredParameter;
 use PHPStan\DependencyInjection\RegisteredRule;
 use PHPStan\Rules\Rule;
 use PHPStan\Rules\RuleErrorBuilder;
-use PHPStan\Type\MixedType;
+use PHPStan\Rules\RuleLevelHelper;
+use PHPStan\Type\ErrorType;
+use PHPStan\Type\Type;
 use PHPStan\Type\VerbosityLevel;
 use function sprintf;
 
@@ -20,8 +21,7 @@ final class InvalidKeyInArrayItemRule implements Rule
 {
 
 	public function __construct(
-		#[AutowiredParameter]
-		private bool $reportMaybes,
+		private RuleLevelHelper $ruleLevelHelper,
 	)
 	{
 	}
@@ -37,23 +37,28 @@ public function processNode(Node $node, Scope $scope): array
 			return [];
 		}
 
-		$dimensionType = $scope->getType($node->key);
+		$dimensionType = $this->ruleLevelHelper->findTypeToCheck(
+			$scope,
+			$node->key,
+			'',
+			static fn (Type $dimType): bool => AllowedArrayKeysTypes::getType()->isSuperTypeOf($dimType)->yes(),
+		)->getType();
+		if ($dimensionType instanceof ErrorType) {
+			return [];
+		}
+
 		$isSuperType = AllowedArrayKeysTypes::getType()->isSuperTypeOf($dimensionType);
-		if ($isSuperType->no()) {
-			return [
-				RuleErrorBuilder::message(
-					sprintf('Invalid array key type %s.', $dimensionType->describe(VerbosityLevel::typeOnly())),
-				)->identifier('array.invalidKey')->build(),
-			];
-		} elseif ($this->reportMaybes && $isSuperType->maybe() && !$dimensionType instanceof MixedType) {
-			return [
-				RuleErrorBuilder::message(
-					sprintf('Possibly invalid array key type %s.', $dimensionType->describe(VerbosityLevel::typeOnly())),
-				)->identifier('array.invalidKey')->build(),
-			];
+		if ($isSuperType->yes()) {
+			return [];
 		}
 
-		return [];
+		return [
+			RuleErrorBuilder::message(sprintf(
+				'%s array key type %s.',
+				$isSuperType->no() ? 'Invalid' : 'Possibly invalid',
+				$dimensionType->describe(VerbosityLevel::typeOnly())
+			))->identifier('array.invalidKey')->build(),
+		];
 	}
 
 }

tests/PHPStan/Rules/Arrays/InvalidKeyInArrayItemRuleTest.php

@@ -3,6 +3,7 @@
 namespace PHPStan\Rules\Arrays;
 
 use PHPStan\Rules\Rule;
+use PHPStan\Rules\RuleLevelHelper;
 use PHPStan\Testing\RuleTestCase;
 use PHPUnit\Framework\Attributes\RequiresPhp;
 
@@ -14,7 +15,9 @@ class InvalidKeyInArrayItemRuleTest extends RuleTestCase
 
 	protected function getRule(): Rule
 	{
-		return new InvalidKeyInArrayItemRule(true);
+		$ruleLevelHelper = new RuleLevelHelper(self::createReflectionProvider(), true, false, true, false, false, false, true);
+
+		return new InvalidKeyInArrayItemRule($ruleLevelHelper);
 	}
 
 	public function testInvalidKey(): void