support array-size

Author: staabm

src/Analyser/TypeSpecifier.php

@@ -2263,14 +2263,23 @@ public function resolveIdentical(Expr\BinaryOp\Identical $expr, Scope $scope, Ty
 				$leftArrayType = $scope->getType($unwrappedLeftExpr->getArgs()[0]->value);
 				$rightArrayType = $scope->getType($unwrappedRightExpr->getArgs()[0]->value);
 
-				if (
-					$leftArrayType->isArray()->yes() && $rightArrayType->isArray()->yes()
-					&& ($leftArrayType->isIterableAtLeastOnce()->yes() || $rightArrayType->isIterableAtLeastOnce()->yes())
-				) {
-					$arrayTypes = $this->create($unwrappedLeftExpr->getArgs()[0]->value, new NonEmptyArrayType(), $context, $scope)->setRootExpr($expr);
-					return $arrayTypes->unionWith(
-						$this->create($unwrappedRightExpr->getArgs()[0]->value, new NonEmptyArrayType(), $context, $scope)->setRootExpr($expr),
-					);
+				if ($leftArrayType->isArray()->yes() && $rightArrayType->isArray()->yes()) {
+					$argType = $scope->getType($unwrappedRightExpr->getArgs()[0]->value);
+					$sizeType = $scope->getType($leftExpr);
+
+					if ($sizeType instanceof IntegerRangeType || $sizeType->isConstantScalarValue()->yes()) {
+						$specifiedTypes = $this->specifyTypesForCountFuncCall($unwrappedRightExpr, $argType, $sizeType, $context, $scope, $expr);
+						if ($specifiedTypes !== null) {
+							return $specifiedTypes;
+						}
+					}
+
+					if ($leftArrayType->isIterableAtLeastOnce()->yes() || $rightArrayType->isIterableAtLeastOnce()->yes()) {
+						$arrayTypes = $this->create($unwrappedLeftExpr->getArgs()[0]->value, new NonEmptyArrayType(), $context, $scope)->setRootExpr($expr);
+						return $arrayTypes->unionWith(
+							$this->create($unwrappedRightExpr->getArgs()[0]->value, new NonEmptyArrayType(), $context, $scope)->setRootExpr($expr),
+						);
+					}
 				}
 			}
 

tests/PHPStan/Analyser/nsrt/list-count2.php

@@ -78,4 +78,16 @@ public function sayUnknownArray($arrA, array $arrB): void
 		}
 		assertType('array<string>', $arrB);
 	}
+
+	/**
+	 * @param array{int, int, int} $arrA
+	 * @param list $arrB
+	 */
+	function sayEqualArrayShape($arrA, array $arrB): void
+	{
+		if (count($arrA) == count($arrB)) {
+			assertType('array{mixed, mixed, mixed}', $arrB);
+		}
+		assertType('list', $arrB);
+	}
 }