Fix false positive non-existing-offset after count() - 1

Author: staabm

src/Analyser/TypeSpecifier.php

@@ -667,7 +667,7 @@ public function specifyTypesInCondition(
 				if (
 					$expr->expr instanceof FuncCall
 					&& $expr->expr->name instanceof Name
-					&& $expr->expr->name->toLowerString() === 'array_key_last'
+					&& in_array($expr->expr->name->toLowerString(), ['array_key_first', 'array_key_last'], true)
 					&& count($expr->expr->getArgs()) >= 1
 				) {
 					$arrayArg = $expr->expr->getArgs()[0]->value;
@@ -677,6 +677,32 @@ public function specifyTypesInCondition(
 						&& $arrayType->isIterableAtLeastOnce()->yes()
 					) {
 						$dimFetch = new ArrayDimFetch($arrayArg, $expr->var);
+						$iterableValueType = $expr->expr->name->toLowerString() === 'array_key_first'
+							? $arrayType->getFirstIterableValueType()
+							: $arrayType->getLastIterableValueType();
+
+						return $specifiedTypes->unionWith(
+							$this->create($dimFetch, $iterableValueType, TypeSpecifierContext::createTrue(), $scope),
+						);
+					}
+				}
+
+				if (
+					$expr->expr instanceof Expr\BinaryOp\Minus
+					&& $expr->expr->left instanceof FuncCall
+					&& $expr->expr->left->name instanceof Name
+					&& $expr->expr->left->name->toLowerString() === 'count'
+					&& count($expr->expr->left->getArgs()) >= 1
+					&& $expr->expr->right instanceof Node\Scalar\Int_
+					&& $expr->expr->right->value === 1
+				) {
+					$arrayArg = $expr->expr->left->getArgs()[0]->value;
+					$arrayType = $scope->getType($arrayArg);
+					if (
+						$arrayType->isList()->yes()
+						&& $arrayType->isIterableAtLeastOnce()->yes()
+					) {
+						$dimFetch = new ArrayDimFetch($arrayArg, $expr->var);
 
 						return $specifiedTypes->unionWith(
 							$this->create($dimFetch, $arrayType->getLastIterableValueType(), TypeSpecifierContext::createTrue(), $scope),

tests/PHPStan/Rules/Arrays/NonexistentOffsetInArrayDimFetchRuleTest.php

@@ -792,4 +792,24 @@ public function testArrayDimFetchAfterArrayKeyFirstOrLast(): void
 		]);
 	}
 
+	public function testArrayDimFetchAfterCount(): void
+	{
+		$this->reportPossiblyNonexistentGeneralArrayOffset = true;
+
+		$this->analyse([__DIR__ . '/data/array-dim-after-count.php'], [
+			[
+				'Offset int<0, max> might not exist on list<string>.',
+				26,
+			],
+			[
+				'Offset int<-1, max> might not exist on array<string>.',
+				35,
+			],
+			[
+				'Offset int<0, max> might not exist on non-empty-array<string>.',
+				42,
+			],
+		]);
+	}
+
 }

tests/PHPStan/Rules/Arrays/data/array-dim-after-array-key-first-or-last.php

@@ -10,22 +10,48 @@ class HelloWorld
 	public function last(array $hellos): string
 	{
 		if ($hellos !== []) {
-			$lastHelloKey = array_key_last($hellos);
-			return $hellos[$lastHelloKey];
+			$last = array_key_last($hellos);
+			return $hellos[$last];
 		} else {
-			$lastHelloKey = array_key_last($hellos);
-			return $hellos[$lastHelloKey];
+			$last = array_key_last($hellos);
+			return $hellos[$last];
 		}
 	}
 
+	/**
+	 * @param array<string> $hellos
+	 */
+	public function lastOnArray(array $hellos): string
+	{
+		if ($hellos !== []) {
+			$last = array_key_last($hellos);
+			return $hellos[$last];
+		}
+
+		return 'nothing';
+	}
+
 	/**
 	 * @param list<string> $hellos
 	 */
 	public function first(array $hellos): string
 	{
 		if ($hellos !== []) {
-			$firstHelloKey = array_key_first($hellos);
-			return $hellos[$firstHelloKey];
+			$first = array_key_first($hellos);
+			return $hellos[$first];
+		}
+
+		return 'nothing';
+	}
+
+	/**
+	 * @param array<string> $hellos
+	 */
+	public function firstOnArray(array $hellos): string
+	{
+		if ($hellos !== []) {
+			$first = array_key_first($hellos);
+			return $hellos[$first];
 		}
 
 		return 'nothing';
@@ -36,12 +62,12 @@ public function first(array $hellos): string
 	 */
 	public function shape(array $hellos): int|bool
 	{
-		$firstHelloKey = array_key_first($hellos);
-		$lastHelloKey = array_key_last($hellos);
+		$first = array_key_first($hellos);
+		$last = array_key_last($hellos);
 
 		if (rand(0,1)) {
-			return $hellos[$firstHelloKey];
+			return $hellos[$first];
 		}
-		return $hellos[$lastHelloKey];
+		return $hellos[$last];
 	}
 }

tests/PHPStan/Rules/Arrays/data/array-dim-after-count.php

@@ -0,0 +1,45 @@
+<?php declare(strict_types = 1);
+
+namespace ArrayDimFetchOnCount;
+
+class HelloWorld
+{
+	/**
+	 * @param list<string> $hellos
+	 */
+	public function works(array $hellos): string
+	{
+		if ($hellos === []) {
+			return 'nothing';
+		}
+
+		$count = count($hellos) - 1;
+		return $hellos[$count];
+	}
+
+	/**
+	 * @param list<string> $hellos
+	 */
+	public function offByOne(array $hellos): string
+	{
+		$count = count($hellos);
+		return $hellos[$count];
+	}
+
+	/**
+	 * @param array<string> $hellos
+	 */
+	public function maybeInvalid(array $hellos): string
+	{
+		$count = count($hellos) - 1;
+		echo $hellos[$count];
+
+		if ($hellos === []) {
+			return 'nothing';
+		}
+
+		$count = count($hellos) - 1;
+		return $hellos[$count];
+	}
+
+}