implement case-expression inference (#522)

staabm · Feb 17, 2023 · a5322f9 · a5322f9
1 parent 5ae4471
commit a5322f9
Show file tree

Hide file tree

Showing 5 changed files with 455 additions and 0 deletions.
diff --git a/.phpstan-dba-mysqli.cache b/.phpstan-dba-mysqli.cache
diff --git a/src/SqlAst/QueryScope.php b/src/SqlAst/QueryScope.php
@@ -12,7 +12,9 @@
 use PHPStan\Type\MixedType;
 use PHPStan\Type\NullType;
 use PHPStan\Type\Type;
+use PHPStan\Type\TypeCombinator;
 use SqlFtw\Sql\Expression\BoolValue;
+use SqlFtw\Sql\Expression\CaseExpression;
 use SqlFtw\Sql\Expression\ExpressionNode;
 use SqlFtw\Sql\Expression\FunctionCall;
 use SqlFtw\Sql\Expression\Identifier;
@@ -112,6 +114,15 @@ public function getType($expression): Type
             throw new ShouldNotHappenException('Unable to resolve column '.$expression->getName());
         }
 
+        if ($expression instanceof CaseExpression) {
+            $resultTypes = [];
+            foreach ($expression->getResults() as $result) {
+                $resultTypes[] = $this->getType($result);
+            }
+
+            return TypeCombinator::union(...$resultTypes);
+        }
+
         if ($expression instanceof FunctionCall) {
             foreach ($this->extensions as $extension) {
                 if (!$extension->isFunctionSupported($expression)) {

diff --git a/tests/default/config/.phpunit-phpstan-dba-mysqli.cache b/tests/default/config/.phpunit-phpstan-dba-mysqli.cache