fix: add "or replace"

Author: cvng

crates/codegen/src/get_node_properties.rs

@@ -529,6 +529,10 @@ fn custom_handlers(node: &Node) -> TokenStream {
  },
  "DefineStmt" => quote! {
  tokens.push(TokenProperty::from(Token::Create));
+ if n.replace {
+ tokens.push(TokenProperty::from(Token::Or));
+ tokens.push(TokenProperty::from(Token::Replace));
+ }
  match n.kind() {
  protobuf::ObjectType::ObjectAggregate => tokens.push(TokenProperty::from(Token::Aggregate)),
  _ => todo!("Unknown DefineStmt {:#?}", n.kind())

crates/parser/src/codegen.rs

@@ -186,6 +186,8 @@ mod tests {
  SyntaxKind::DefineStmt,
  vec![
  TokenProperty::from(SyntaxKind::Create),
+ TokenProperty::from(SyntaxKind::Or),
+ TokenProperty::from(SyntaxKind::Replace),
  TokenProperty::from(SyntaxKind::Aggregate),
  TokenProperty::from(SyntaxKind::Order),
  TokenProperty::from(SyntaxKind::By),