fix: recognise CTE (WITH ...) queries in getQueryType

Etienne V. Labelle · Etienne V. Labelle · commit ff77a7c56c23 · 2026-05-17T18:32:01.000-04:00
diff --git a/src/QueryReflection/QueryReflection.php b/src/QueryReflection/QueryReflection.php
@@ -407,6 +407,12 @@ public static function getQueryType(string $query): ?string
             return strtoupper($matches[1]);
         }
 
+        // WITH [RECURSIVE] cte [(col_list)] AS (subquery) [, ...] <SELECT|INSERT|UPDATE|DELETE|REPLACE>
+        // (?1) recurses group 1 to balance parentheses inside subqueries.
+        if (1 === preg_match('/^\s*WITH\s+(?:RECURSIVE\s+)?[`"\w]+\s*(?:\([^)]*\))?\s*AS\s*(\((?:[^()]|(?1))*\))(?:\s*,\s*[`"\w]+\s*(?:\([^)]*\))?\s*AS\s*(?1))*\s*(SELECT|INSERT|UPDATE|DELETE|REPLACE)\b/i', $query, $matches)) {
+            return strtoupper($matches[2]);
+        }
+
         return null;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -407,6 +407,12 @@ public static function getQueryType(string $query): ?string`
`407`	`407`	`return strtoupper($matches[1]);`
`408`	`408`	`}`
`409`	`409`
	`410`	`+ // WITH [RECURSIVE] cte [(col_list)] AS (subquery) [, ...] <SELECT\|INSERT\|UPDATE\|DELETE\|REPLACE>`
	`411`	`+ // (?1) recurses group 1 to balance parentheses inside subqueries.`
	`412`	+ if (1 === preg_match('/^\sWITH\s+(?:RECURSIVE\s+)?[`"\w]+\s(?:\([^)]\))?\sAS\s(\((?:[^()]\|(?1))\))(?:\s,\s[`"\w]+\s(?:\([^)]\))?\sAS\s(?1))\s(SELECT\|INSERT\|UPDATE\|DELETE\|REPLACE)\b/i', $query, $matches)) {
	`413`	`+ return strtoupper($matches[2]);`
	`414`	`+ }`
	`415`	`+`
`410`	`416`	`return null;`
`411`	`417`	`}`
`412`	`418`