Merge #564 - Fix query flags for lower-case functions

Pull-request: #564
Signed-off-by: William Desportes <[email protected]>

Author: williamdes

phpstan-baseline.neon

@@ -950,6 +950,11 @@ parameters:
 			count: 1
 			path: src/Utils/Query.php
 
+		-
+			message: "#^Parameter \\#1 \\$str of function strtoupper expects string, mixed given\\.$#"
+			count: 1
+			path: src/Utils/Query.php
+
 		-
 			message: "#^Cannot access offset 'value' on mixed\\.$#"
 			count: 3

psalm-baseline.xml

@@ -1336,8 +1336,9 @@
     <InvalidNullableReturnType occurrences="1">
       <code>int</code>
     </InvalidNullableReturnType>
-    <MixedArgument occurrences="1">
+    <MixedArgument occurrences="2">
       <code>$expr</code>
+      <code>$expr-&gt;function</code>
     </MixedArgument>
     <MixedArrayOffset occurrences="2">
       <code>$clauses[$token-&gt;keyword]</code>

src/Utils/Query.php

@@ -37,6 +37,7 @@
 use function count;
 use function in_array;
 use function is_string;
+use function strtoupper;
 use function trim;
 
 /**
@@ -318,9 +319,10 @@ private static function getFlagsSelect($statement, $flags)
 
         foreach ($expressions as $expr) {
             if (! empty($expr->function)) {
-                if ($expr->function === 'COUNT') {
+                $function = strtoupper($expr->function);
+                if ($function === 'COUNT') {
                     $flags['is_count'] = true;
-                } elseif (in_array($expr->function, static::$FUNCTIONS)) {
+                } elseif (in_array($function, static::$FUNCTIONS, true)) {
                     $flags['is_func'] = true;
                 }
             }

tests/Utils/QueryTest.php

@@ -181,6 +181,24 @@ public static function getFlagsProvider(): array
                     'querytype' => 'SELECT',
                 ],
             ],
+            [
+                'SELECT count(*) FROM tbl',
+                [
+                    'is_count' => true,
+                    'is_select' => true,
+                    'select_from' => true,
+                    'querytype' => 'SELECT',
+                ],
+            ],
+            [
+                'SELECT sum(*) FROM tbl',
+                [
+                    'is_func' => true,
+                    'is_select' => true,
+                    'select_from' => true,
+                    'querytype' => 'SELECT',
+                ],
+            ],
             [
                 'SELECT (SELECT "foo")',
                 [