Specify types on match() condition correctly

Author: rvanvelzen

src/Analyser/TypeSpecifier.php

@@ -1991,11 +1991,16 @@ public function resolveIdentical(Expr\BinaryOp\Identical $expr, Scope $scope, Ty
 			$exprNode = $expressions[0];
 			$constantType = $expressions[1];
 
-			$specifiedType = $this->specifyTypesForConstantBinaryExpression($exprNode, $constantType, $context, $scope, $rootExpr);
+			$unwrappedExprNode = $exprNode;
+			if ($exprNode instanceof AlwaysRememberedExpr) {
+				$unwrappedExprNode = $exprNode->getExpr();
+			}
+
+			$specifiedType = $this->specifyTypesForConstantBinaryExpression($unwrappedExprNode, $constantType, $context, $scope, $rootExpr);
 			if ($specifiedType !== null) {
-				if ($exprNode instanceof AlwaysRememberedExpr) {
-					$specifiedType->unionWith(
-						$this->create($exprNode->getExpr(), $constantType, $context, false, $scope, $rootExpr),
+				if ($exprNode !== $unwrappedExprNode) {
+					$specifiedType = $specifiedType->unionWith(
+						$this->create($exprNode, $constantType, $context, false, $scope, $rootExpr),
 					);
 				}
 				return $specifiedType;

tests/PHPStan/Rules/Functions/CallToFunctionParametersRuleTest.php

@@ -1669,4 +1669,13 @@ public function testBug10297(): void
 		$this->analyse([__DIR__ . '/data/bug-10297.php'], []);
 	}
 
+	public function testBug10974(): void
+	{
+		if (PHP_VERSION_ID < 80000) {
+			$this->markTestSkipped('Test requires PHP 8.0.');
+		}
+
+		$this->analyse([__DIR__ . '/data/bug-10974.php'], []);
+	}
+
 }

tests/PHPStan/Rules/Functions/data/bug-10974.php

@@ -0,0 +1,27 @@
+<?php declare(strict_types = 1); // lint >= 8.0
+
+namespace Bug10974;
+
+function non(): void {}
+function single(string $str): void {}
+/** @param non-empty-array<string> $strs */
+function multiple(array $strs): void {}
+
+/** @param array<string> $arr */
+function test(array $arr): void
+{
+	match (count($arr))
+	{
+		0 => non(),
+		1 => single(reset($arr)),
+		default => multiple($arr)
+	};
+
+	if (empty($arr)) {
+		non();
+	} elseif (count($arr) === 1) {
+		single(reset($arr));
+	} else {
+		multiple($arr);
+	}
+}

tests/PHPStan/Rules/Properties/AccessPropertiesRuleTest.php

@@ -937,4 +937,16 @@ public function testBug8629(): void
 		$this->analyse([__DIR__ . '/data/bug-8629.php'], []);
 	}
 
+	public function testBug9694(): void
+	{
+		if (PHP_VERSION_ID < 80000) {
+			$this->markTestSkipped('Test requires PHP 8.0.');
+		}
+
+		$this->checkThisOnly = false;
+		$this->checkUnionTypes = true;
+		$this->checkDynamicProperties = true;
+		$this->analyse([__DIR__ . '/data/bug-9694.php'], []);
+	}
+
 }

tests/PHPStan/Rules/Properties/data/bug-9694.php

@@ -0,0 +1,20 @@
+<?php declare(strict_types = 1); // lint >= 8.0
+
+class TotpEnrollment
+{
+	public bool $confirmed;
+}
+
+class User
+{
+	public ?TotpEnrollment $totpEnrollment;
+}
+
+function () {
+	$user = new User();
+
+	return match ($user->totpEnrollment === null) {
+		true => false,
+		false => $user->totpEnrollment->confirmed,
+	};
+};