Consolidate under_name code
diff --git a/codegen/src/fold.rs b/codegen/src/fold.rs
index bb6e37d..e0f49d1 100644
--- a/codegen/src/fold.rs
+++ b/codegen/src/fold.rs
@@ -1,5 +1,4 @@
use crate::{file, full, gen};
-use inflections::Inflect;
use proc_macro2::{Span, TokenStream};
use quote::{quote, TokenStreamExt};
use syn::*;
@@ -13,12 +12,8 @@
fold_impl: TokenStream,
}
-fn under_name(name: &str) -> Ident {
- Ident::new(&name.to_snake_case(), Span::call_site())
-}
-
fn simple_visit(item: &str, name: &TokenStream) -> TokenStream {
- let ident = under_name(item);
+ let ident = gen::under_name(item);
let method = Ident::new(&format!("fold_{}", ident), Span::call_site());
quote! {
_visitor.#method(#name)
@@ -118,7 +113,7 @@
fn node(state: &mut State, s: &types::Node, defs: &types::Definitions) {
let features = visit_features(&s.features);
- let under_name = under_name(&s.ident);
+ let under_name = gen::under_name(&s.ident);
let ty = Ident::new(&s.ident, Span::call_site());
let fold_fn = Ident::new(&format!("fold_{}", under_name), Span::call_site());