codegen/src/gen.rs

use inflections::Inflect;
use proc_macro2::{Ident, Span, TokenStream};
use syn_codegen::{Data, Features, Definitions, Node};

pub const TERMINAL_TYPES: &[&str] = &["Span", "Ident"];

pub fn under_name(name: &str) -> Ident {
    Ident::new(&name.to_snake_case(), Span::call_site())
}

pub fn traverse(
    defs: &Definitions,
    node: fn(&mut TokenStream, &mut TokenStream, &Node, &Definitions),
) -> (TokenStream, TokenStream) {
    let mut traits = TokenStream::new();
    let mut impls = TokenStream::new();
    for s in &defs.types {
        node(&mut traits, &mut impls, s, defs);
    }
    for tt in TERMINAL_TYPES {
        let s = Node {
            ident: tt.to_string(),
            features: Features::default(),
            data: Data::Private,
        };
        node(&mut traits, &mut impls, &s, defs);
    }
    (traits, impls)
}