fix!: Consider WITH a top level reserved word

shssoichiro · Jan 24, 2025 · 7142a12 · 7142a12
1 parent e205542
commit 7142a12
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -1991,7 +1991,8 @@ mod tests {
         };
         let expected = indoc! {
             "
-            WITH a AS (
+            WITH
+            a AS (
               SELECT a, b, c
               FROM t
               WHERE a > 100

diff --git a/src/tokenizer.rs b/src/tokenizer.rs
@@ -594,6 +594,7 @@ fn get_top_level_reserved_token_no_indent<'i>(input: &mut &'i str) -> PResult<To
         terminated("MINUS", end_of_word),
         terminated("UNION", end_of_word),
         terminated("UNION ALL", end_of_word),
+        terminated("WITH", end_of_word),
         terminated("$$", end_of_word),
     ))
     .parse_next(&mut uc_input);
@@ -987,7 +988,6 @@ fn get_plain_reserved_one_token<'i>(input: &mut &'i str) -> PResult<Token<'i>> {
 
         'W' => alt((
             terminated("WHEN", end_of_word),
-            terminated("WITH", end_of_word),
             terminated("WORK", end_of_word),
             terminated("WRITE", end_of_word),
         ))