feat: support CreatedbStmt

Author: cvng

Diff for: crates/codegen/src/get_node_properties.rs

@@ -683,6 +683,21 @@ fn custom_handlers(node: &Node) -> TokenStream {
             tokens.push(TokenProperty::from(Token::Create));
             tokens.push(TokenProperty::from(Token::TypeP));
         },
+        "CreatedbStmt" => quote! {
+            tokens.push(TokenProperty::from(Token::Create));
+            tokens.push(TokenProperty::from(Token::Database));
+            for option in &n.options {
+                if let Some(NodeEnum::DefElem(node)) = &option.node {
+                    if node.defname == "location" {
+                        tokens.push(TokenProperty::from(Token::Default));
+                    }
+                    if node.defname == "connection_limit" {
+                        tokens.push(TokenProperty::from(Token::Limit));
+                        tokens.push(TokenProperty::from(Token::Iconst));
+                    }
+                }
+            }
+        },
         _ => quote! {},
     }
 }

Diff for: crates/parser/src/codegen.rs

@@ -268,4 +268,19 @@ mod tests {
             ],
         )
     }
+
+    #[test]
+    fn test_create_database() {
+        test_get_node_properties(
+            "create database x owner abc connection limit 5;",
+            SyntaxKind::CreatedbStmt,
+            vec![
+                TokenProperty::from(SyntaxKind::Create),
+                TokenProperty::from(SyntaxKind::Database),
+                TokenProperty::from(SyntaxKind::Limit),
+                TokenProperty::from(SyntaxKind::Iconst),
+                TokenProperty::from("x".to_string()),
+            ],
+        )
+    }
 }

Diff for: crates/parser/tests/data/statements/valid/0047.sql

@@ -0,0 +1,6 @@
+CREATE DATABASE x OWNER abc CONNECTION LIMIT 5;
+CREATE DATABASE x ENCODING \"SQL_ASCII\";
+CREATE DATABASE x LC_COLLATE \"en_US.UTF-8\";
+CREATE DATABASE x LOCATION DEFAULT;
+CREATE DATABASE x TABLESPACE abc;
+CREATE DATABASE x TEMPLATE TRUE;