fix: support define stmt

Author: cvng

crates/codegen/src/get_node_properties.rs

@@ -630,6 +630,9 @@ fn custom_handlers(node: &Node) -> TokenStream {
                         }                     }
                         // if its a list, we handle it in the handler for `List`
                 },
+                protobuf::ObjectType::ObjectType => {
+                    tokens.push(TokenProperty::from(Token::TypeP));
+                },
                 _ => panic!("Unknown DefineStmt {:#?}", n.kind()),
             }
         },

crates/parser/src/codegen.rs

@@ -272,6 +272,19 @@ mod tests {
 
     #[test]
     fn test_create_type() {
+        test_get_node_properties(
+            "create type type1",
+            SyntaxKind::DefineStmt,
+            vec![
+                TokenProperty::from(SyntaxKind::Create),
+                TokenProperty::from(SyntaxKind::TypeP),
+                TokenProperty::from("type1".to_string()),
+            ],
+        )
+    }
+
+    #[test]
+    fn test_create_composite_type() {
         test_get_node_properties(
             "create type type1 as (attr1 int4, attr2 bool);",
             SyntaxKind::CompositeTypeStmt,