Merge pull request #61 from cvng/try-create-domain

Author: psteinroe

crates/codegen/src/get_node_properties.rs

@@ -529,6 +529,13 @@ fn custom_handlers(node: &Node) -> TokenStream {
         "TypeCast" => quote! {
             tokens.push(TokenProperty::from(Token::Typecast));
         },
+        "CreateDomainStmt" => quote! {
+            tokens.push(TokenProperty::from(Token::Create));
+            tokens.push(TokenProperty::from(Token::DomainP));
+            if n.type_name.is_some() {
+                tokens.push(TokenProperty::from(Token::As));
+            }
+        },
         _ => quote! {},
     }
 }

crates/parser/src/codegen.rs

@@ -91,4 +91,18 @@ mod tests {
             ],
         )
     }
+
+    #[test]
+    fn test_create_domain() {
+        test_get_node_properties(
+            "create domain us_postal_code as text check (value is not null);",
+            SyntaxKind::CreateDomainStmt,
+            vec![
+                TokenProperty::from(SyntaxKind::Create),
+                TokenProperty::from(SyntaxKind::DomainP),
+                TokenProperty::from(SyntaxKind::As),
+                TokenProperty::from("us_postal_code".to_string()),
+            ],
+        )
+    }
 }