Add extension calls for shift left/right operators

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: 2 people

src/Reflection/InitializerExprTypeResolver.php

@@ -1774,6 +1774,12 @@ public function getShiftLeftType(Expr $left, Expr $right, callable $getTypeCallb
 		$leftType = $getTypeCallback($left);
 		$rightType = $getTypeCallback($right);
 
+		$specifiedTypes = $this->operatorTypeSpecifyingExtensionRegistryProvider->getRegistry()
+			->callOperatorTypeSpecifyingExtensions(new BinaryOp\ShiftLeft($left, $right), $leftType, $rightType);
+		if ($specifiedTypes !== null) {
+			return $specifiedTypes;
+		}
+
 		return $this->getShiftLeftTypeFromTypes($left, $right, $leftType, $rightType);
 	}
 
@@ -1838,6 +1844,12 @@ public function getShiftRightType(Expr $left, Expr $right, callable $getTypeCall
 		$leftType = $getTypeCallback($left);
 		$rightType = $getTypeCallback($right);
 
+		$specifiedTypes = $this->operatorTypeSpecifyingExtensionRegistryProvider->getRegistry()
+			->callOperatorTypeSpecifyingExtensions(new BinaryOp\ShiftRight($left, $right), $leftType, $rightType);
+		if ($specifiedTypes !== null) {
+			return $specifiedTypes;
+		}
+
 		return $this->getShiftRightTypeFromTypes($left, $right, $leftType, $rightType);
 	}
 

tests/PHPStan/Analyser/data/operator-type-specifying-extension.php

@@ -25,6 +25,16 @@ function testBitwiseXor(TestBitwiseOperand $a, TestBitwiseOperand $b): void
 	assertType('PHPStan\Fixture\TestBitwiseOperand', $a ^ $b);
 }
 
+function testShiftLeft(TestBitwiseOperand $a, TestBitwiseOperand $b): void
+{
+	assertType('PHPStan\Fixture\TestBitwiseOperand', $a << $b);
+}
+
+function testShiftRight(TestBitwiseOperand $a, TestBitwiseOperand $b): void
+{
+	assertType('PHPStan\Fixture\TestBitwiseOperand', $a >> $b);
+}
+
 // =============================================================================
 // Arithmetic operator extension tests (via TestDecimal)
 // =============================================================================

tests/PHPStan/Type/TestBitwiseOperatorTypeSpecifyingExtension.php

@@ -15,7 +15,7 @@ public function isOperatorSupported(string $operatorSigil, Type $leftSide, Type
 	{
 		$testType = new ObjectType(TestBitwiseOperand::class);
 
-		return in_array($operatorSigil, ['&', '|', '^'], true)
+		return in_array($operatorSigil, ['&', '|', '^', '<<', '>>'], true)
 			&& $testType->isSuperTypeOf($leftSide)->yes()
 			&& $testType->isSuperTypeOf($rightSide)->yes();
 	}