round

Author: staabm

src/ParserExtension/QueryScope.php

@@ -56,6 +56,7 @@ public function __construct(Table $fromTable, array $joinedTables)
             new AvgReturnTypeExtension(),
             new IsNullReturnTypeExtension(),
             new AbsReturnTypeExtension(),
+            new RoundReturnTypeExtension(),
         ];
     }
 

src/ParserExtension/RoundReturnTypeExtension.php

@@ -0,0 +1,63 @@
+<?php
+
+declare(strict_types=1);
+
+namespace staabm\PHPStanDba\ParserExtension;
+
+use PHPStan\Type\FloatType;
+use PHPStan\Type\IntegerRangeType;
+use PHPStan\Type\IntegerType;
+use PHPStan\Type\NullType;
+use PHPStan\Type\Type;
+use PHPStan\Type\TypeCombinator;
+use PHPStan\Type\UnionType;
+use SqlFtw\Sql\Expression\BuiltInFunction;
+use SqlFtw\Sql\Expression\ExpressionNode;
+use SqlFtw\Sql\Expression\FunctionCall;
+
+/**
+ * @implements QueryExpressionReturnTypeExtension<FunctionCall>
+ */
+final class RoundReturnTypeExtension implements QueryExpressionReturnTypeExtension
+{
+    public function isExpressionSupported(ExpressionNode $expression): bool
+    {
+        return
+            $expression instanceof FunctionCall
+            && \in_array($expression->getFunction()->getName(), [BuiltInFunction::ROUND], true);
+    }
+
+    public function getTypeFromExpression(ExpressionNode $expression, QueryScope $scope): ?Type
+    {
+        $args = $expression->getArguments();
+
+        if (\count($args) < 1) {
+            return null;
+        }
+
+        $argType = $scope->getType($args[0]);
+        if ($argType->isNull()->yes()) {
+            return new NullType();
+        }
+
+        $containedNull = TypeCombinator::containsNull($argType);
+        $argType = TypeCombinator::removeNull($argType);
+
+        if (1 === \count($args)) {
+            if (!$argType instanceof IntegerRangeType) {
+                $argType = new IntegerType();
+            }
+        } else {
+            $argType = new UnionType([
+                new IntegerType(),
+                new FloatType(),
+            ]);
+        }
+
+        if ($containedNull) {
+            $argType = TypeCombinator::addNull($argType);
+        }
+
+        return $argType;
+    }
+}

tests/default/config/.phpunit-phpstan-dba-mysqli.cache

@@ -207,4 +207,16 @@ public function abs(PDO $pdo): void
         $stmt = $pdo->query('SELECT abs(eladaid) as abs from ak');
         assertType('PDOStatement<array{abs: int<0, 2147483647>|null, 0: int<0, 2147483647>|null}>', $stmt);
     }
+
+    public function round(PDO $pdo): void
+    {
+        $stmt = $pdo->query('SELECT round(null) as abs from ada');
+        assertType('PDOStatement<array{abs: null, 0: null}>', $stmt);
+
+        $stmt = $pdo->query('SELECT round(freigabe1u1) as abs from ada');
+        assertType('PDOStatement<array{abs: int<-128, 127>, 0: int<-128, 127>}>', $stmt);
+
+        $stmt = $pdo->query('SELECT round(1.12, 1) as abs from ak');
+        assertType('PDOStatement<array{abs: float|int, 0: float|int}>', $stmt);
+    }
 }