codegen/src/gen.rs

use inflections::Inflect;
use proc_macro2::{Ident, Span, TokenStream};
use quote::quote;
use syn_codegen::{Data, Definitions, Features, 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 {
        let features = &s.features.any;
        let features = match features.len() {
            0 => quote!(),
            1 => quote!(#[cfg(feature = #(#features)*)]),
            _ => quote!(#[cfg(any(#(feature = #features),*))]),
        };
        traits.extend(features.clone());
        impls.extend(features);
        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)
}