github
diff --git a/‎rust/ast-generator/src/main.rs
Copy file name to clipboardExpand all lines: rust/ast-generator/src/main.rs
+9-3Lines changed: 9 additions & 3 deletions b/‎rust/ast-generator/src/main.rs
Copy file name to clipboardExpand all lines: rust/ast-generator/src/main.rs
+9-3Lines changed: 9 additions & 3 deletions
diff --git a/‎rust/ast-generator/src/templates/extractor.mustache
Copy file name to clipboardExpand all lines: rust/ast-generator/src/templates/extractor.mustache
+1-1Lines changed: 1 addition & 1 deletion b/‎rust/ast-generator/src/templates/extractor.mustache
Copy file name to clipboardExpand all lines: rust/ast-generator/src/templates/extractor.mustache
+1-1Lines changed: 1 addition & 1 deletion
@@ -432,6 +432,7 @@ fn get_fields(node: &AstNodeSrc) -> Vec<FieldInfo> {
 struct EnumVariantInfo {
     name: String,
     snake_case_name: String,
+    variant_ast_name: String,
 }
 
 #[derive(Serialize)]
@@ -480,9 +481,14 @@ fn enum_to_extractor_info(node: &AstEnumSrc) -> Option<ExtractorEnumInfo> {
         variants: node
             .variants
             .iter()
-            .map(|v| EnumVariantInfo {
-                name: v.clone(),
-                snake_case_name: to_lower_snake_case(v),
+            .map(|v| {
+                let name = class_name(v);
+                let snake_case_name = to_lower_snake_case(v);
+                EnumVariantInfo {
+                    name,
+                    snake_case_name,
+                    variant_ast_name: v.clone(),
+                }
             })
             .collect(),
     })
 
@@ -25,7 +25,7 @@ impl Translator<'_> {
     pub(crate) fn emit_{{snake_case_name}}(&mut self, node: ast::{{ast_name}}) -> Option<Label<generated::{{name}}>> {
         match node {
             {{#variants}}
-            ast::{{ast_name}}::{{name}}(inner) => self.emit_{{snake_case_name}}(inner).map(Into::into),
+            ast::{{ast_name}}::{{variant_ast_name}}(inner) => self.emit_{{snake_case_name}}(inner).map(Into::into),
             {{/variants}}
         }
     }
Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@ impl Translator<'_> {`
`25`	`25`	`pub(crate) fn emit_{{snake_case_name}}(&mut self, node: ast::{{ast_name}}) -> Option<Label<generated::{{name}}>> {`
`26`	`26`	`match node {`
`27`	`27`	`{{#variants}}`
`28`		`- ast::{{ast_name}}::{{name}}(inner) => self.emit_{{snake_case_name}}(inner).map(Into::into),`
	`28`	`+ ast::{{ast_name}}::{{variant_ast_name}}(inner) => self.emit_{{snake_case_name}}(inner).map(Into::into),`
`29`	`29`	`{{/variants}}`
`30`	`30`	`}`
`31`	`31`	`}`