Implement BinaryPowHandler

Author: staabm

src/Analyser/Generator/ExprHandler/BinaryPowHandler.php

@@ -0,0 +1,53 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Analyser\Generator\ExprHandler;
+
+use Generator;
+use PhpParser\Node\Expr;
+use PhpParser\Node\Stmt;
+use PHPStan\Analyser\ExpressionContext;
+use PHPStan\Analyser\Generator\ExprAnalysisRequest;
+use PHPStan\Analyser\Generator\ExprAnalysisResult;
+use PHPStan\Analyser\Generator\ExprHandler;
+use PHPStan\Analyser\Generator\GeneratorScope;
+use PHPStan\Analyser\SpecifiedTypes;
+use PHPStan\DependencyInjection\AutowiredService;
+use PHPStan\Reflection\InitializerExprTypeResolver;
+use function array_merge;
+
+/**
+ * @implements ExprHandler<Expr\BinaryOp\Pow>
+ */
+#[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(),
+		);
+	}
+
+}

src/Reflection/InitializerExprTypeResolver.php

@@ -1653,6 +1653,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) {

tests/PHPStan/Analyser/Generator/data/gnsr.php

@@ -117,6 +117,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