diff --git a/src/Analyser/Generator/ExprHandler/BinaryPowHandler.php b/src/Analyser/Generator/ExprHandler/BinaryPowHandler.php new file mode 100644 index 0000000000..b68517d398 --- /dev/null +++ b/src/Analyser/Generator/ExprHandler/BinaryPowHandler.php @@ -0,0 +1,53 @@ + + */ +#[AutowiredService] +final class BinaryPowHandler implements ExprHandler +{ + + public function __construct(private InitializerExprTypeResolver $initializerExprTypeResolver) + { + } + + public function supports(Expr $expr): bool + { + return $expr instanceof Expr\BinaryOp\Pow; + } + + public function analyseExpr(Stmt $stmt, Expr $expr, GeneratorScope $scope, ExpressionContext $context, ?callable $alternativeNodeCallback): Generator + { + $leftResult = yield new ExprAnalysisRequest($stmt, $expr->left, $scope, $context, $alternativeNodeCallback); + $rightResult = yield new ExprAnalysisRequest($stmt, $expr->right, $leftResult->scope, $context, $alternativeNodeCallback); + + return new ExprAnalysisResult( + $this->initializerExprTypeResolver->getPowTypeFromTypes($expr->left, $expr->right, $leftResult->type, $rightResult->type), + $this->initializerExprTypeResolver->getPowTypeFromTypes($expr->left, $expr->right, $leftResult->nativeType, $rightResult->nativeType), + $rightResult->scope, + hasYield: $leftResult->hasYield || $rightResult->hasYield, + isAlwaysTerminating: $leftResult->isAlwaysTerminating || $rightResult->isAlwaysTerminating, + throwPoints: array_merge($leftResult->throwPoints, $rightResult->throwPoints), + impurePoints: array_merge($leftResult->impurePoints, $rightResult->impurePoints), + specifiedTruthyTypes: new SpecifiedTypes(), + specifiedFalseyTypes: new SpecifiedTypes(), + specifiedNullTypes: new SpecifiedTypes(), + ); + } + +} diff --git a/src/Reflection/InitializerExprTypeResolver.php b/src/Reflection/InitializerExprTypeResolver.php index 27ceac0a4a..b75042e418 100644 --- a/src/Reflection/InitializerExprTypeResolver.php +++ b/src/Reflection/InitializerExprTypeResolver.php @@ -1648,6 +1648,11 @@ public function getPowType(Expr $left, Expr $right, callable $getTypeCallback): $leftType = $getTypeCallback($left); $rightType = $getTypeCallback($right); + return $this->getPowTypeFromTypes($left, $right, $leftType, $rightType); + } + + public function getPowTypeFromTypes(Expr $left, Expr $right, Type $leftType, Type $rightType): Type + { $extensionSpecified = $this->operatorTypeSpecifyingExtensionRegistryProvider->getRegistry() ->callOperatorTypeSpecifyingExtensions(new BinaryOp\Pow($left, $right), $leftType, $rightType); if ($extensionSpecified !== null) { diff --git a/tests/PHPStan/Analyser/Generator/data/gnsr.php b/tests/PHPStan/Analyser/Generator/data/gnsr.php index b915536172..7329db8812 100644 --- a/tests/PHPStan/Analyser/Generator/data/gnsr.php +++ b/tests/PHPStan/Analyser/Generator/data/gnsr.php @@ -97,6 +97,21 @@ public function doMul($a, $b, int $c, int $d): void assertNativeType('int', $c * $d); } + /** + * @param int $a + * @param int $b + * @return void + */ + public function doPow($a, $b, int $c, int $d): void + { + assertType('(float|int)', $a ** $b); + assertNativeType('(float|int)', $a ** $b); + assertType('1', 1 ** 1); + assertNativeType('1', 1 ** 1); + assertType('(float|int)', $c ** $d); + assertNativeType('(float|int)', $c ** $d); + } + /** * @param string $a * @param string $b