feat: add support for truncate statement

Author: pert5432

crates/codegen/src/get_node_properties.rs

@@ -952,6 +952,23 @@ fn custom_handlers(node: &Node) -> TokenStream {
                 }
             }
         },
+        "TruncateStmt" => quote! {
+            tokens.push(TokenProperty::from(Token::Truncate));
+            if n.restart_seqs {
+                tokens.push(TokenProperty::from(Token::Restart));
+                tokens.push(TokenProperty::from(Token::IdentityP));
+            } else {
+                tokens.push(TokenProperty::from(Token::ContinueP));
+                tokens.push(TokenProperty::from(Token::IdentityP));
+            }
+            match n.behavior {
+                // DropRestrict
+                1 => tokens.push(TokenProperty::from(Token::Restrict)),
+                // DropCascade
+                2 => tokens.push(TokenProperty::from(Token::Cascade)),
+                _ => {}
+            }
+        },
         _ => quote! {},
     }
 }

crates/parser/src/codegen.rs

@@ -358,4 +358,18 @@ mod tests {
             ],
         )
     }
+
+    #[test]
+    fn test_truncate() {
+        test_get_node_properties(
+            "TRUNCATE TABLE users CONTINUE IDENTITY RESTRICT",
+            SyntaxKind::TruncateStmt,
+            vec![
+                TokenProperty::from(SyntaxKind::Truncate),
+                TokenProperty::from(SyntaxKind::ContinueP),
+                TokenProperty::from(SyntaxKind::IdentityP),
+                TokenProperty::from(SyntaxKind::Restrict),
+            ],
+        )
+    }
 }