fix: REFERENCES xyz ON UPDATE .. causes formatter to treat the remaining as an UPDATE statement

src/lib.rs

@@ -1,4 +1,4 @@
 //! This crate is a port of https://github.com/kufii/sql-formatter-plus
 //! written in Rust. It is intended to be usable as a pure-Rust library
 //! for formatting SQL queries.
 
@@ -1616,4 +1616,20 @@
 
         assert_eq!(format(input, &QueryParams::None, options), expected);
     }
+
+    #[test]
+    fn it_recognizes_on_update_clause() {
+        let input = indoc!(
+            "CREATE TABLE a (b integer REFERENCES c (id) ON UPDATE RESTRICT, other integer);"
+        );
+        let options = FormatOptions::default();
+        let expected = indoc!(
+            "
+          CREATE TABLE a (
+            b integer REFERENCES c (id) ON UPDATE RESTRICT,
+            other integer
+          );"
+        );
+        assert_eq!(format(input, &QueryParams::None, options), expected);
+    }
 }

src/tokenizer.rs

@@ -569,8 +569,10 @@ fn get_top_level_reserved_token_no_indent(input: &str) -> IResult<&str, Token<'_
         Err(Err::Error(Error::new(input, ErrorKind::Alt)))
     }
 }
-
 fn get_plain_reserved_token(input: &str) -> IResult<&str, Token<'_>> {
+    alt((get_plain_reserved_two_token, get_plain_reserved_one_token))(input)
+}
+fn get_plain_reserved_one_token(input: &str) -> IResult<&str, Token<'_>> {
     let uc_input = get_uc_words(input, 1);
     let result: IResult<&str, &str> = alt((
         terminated(tag("ACCESSIBLE"), end_of_word),
@@ -595,7 +597,6 @@ fn get_plain_reserved_token(input: &str) -> IResult<&str, Token<'_>> {
         alt((
             terminated(tag("CHANGE"), end_of_word),
             terminated(tag("CHANGED"), end_of_word),
-            terminated(tag("CHARACTER SET"), end_of_word),
             terminated(tag("CHARSET"), end_of_word),
             terminated(tag("CHECK"), end_of_word),
             terminated(tag("CHECKSUM"), end_of_word),
@@ -736,8 +737,6 @@ fn get_plain_reserved_token(input: &str) -> IResult<&str, Token<'_>> {
                                     terminated(tag("NOW()"), end_of_word),
                                     terminated(tag("NULL"), end_of_word),
                                     terminated(tag("OFFSET"), end_of_word),
-                                    terminated(tag("ON DELETE"), end_of_word),
-                                    terminated(tag("ON UPDATE"), end_of_word),
                                     alt((
                                         terminated(tag("ON"), end_of_word),
                                         terminated(tag("ONLY"), end_of_word),
@@ -929,6 +928,29 @@ fn get_plain_reserved_token(input: &str) -> IResult<&str, Token<'_>> {
     }
 }
 
+fn get_plain_reserved_two_token(input: &str) -> IResult<&str, Token<'_>> {
+    let uc_input = get_uc_words(input, 2);
+    let result: IResult<&str, &str> = alt((
+        terminated(tag("CHARACTER SET"), end_of_word),
+        terminated(tag("ON DELETE"), end_of_word),
+        terminated(tag("ON UPDATE"), end_of_word),
+    ))(&uc_input);
+    if let Ok((_, token)) = result {
+        let input_end_pos = token.len();
+        let (token, input) = input.split_at(input_end_pos);
+        Ok((
+            input,
+            Token {
+                kind: TokenKind::Reserved,
+                value: token,
+                key: None,
+            },
+        ))
+    } else {
+        Err(Err::Error(Error::new(input, ErrorKind::Alt)))
+    }
+}
+
 fn get_word_token(input: &str) -> IResult<&str, Token<'_>> {
     take_while1(is_word_character)(input).map(|(input, token)| {
         (