sneks 0.1.3

simple macros for naming enum variants
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

// Copyright 2023, Offchain Labs, Inc.
// For licensing, see https://github.com/OffchainLabs/sneks/blob/stylus/licenses/COPYRIGHT.md

use convert_case::{Boundary, Case, Casing};
use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, Fields, ItemEnum};

#[proc_macro_derive(SimpleSnakeNames)]
pub fn derive_simple_snake_names(input: TokenStream) -> TokenStream {
    let ItemEnum {
        ident,
        generics,
        variants,
        ..
    } = parse_macro_input!(input as ItemEnum);

    let mut arms = quote!();
    for variant in variants {
        let name = variant.ident;
        let snek = name
            .to_string()
            .from_case(Case::Camel)
            .without_boundaries(&Boundary::letter_digit())
            .to_case(Case::Snake);

        arms.extend(match variant.fields {
            Fields::Unit => quote! { #name => #snek, },
            Fields::Unnamed(..) => quote! { #name(..) => #snek, },
            Fields::Named(..) => quote! { #name { .. } => #snek, },
        });
    }

    quote! {
        impl #generics #ident #generics {
            pub fn name(&self) -> &'static str {
                use #ident::*;
                match self {
                    #arms
                }
            }
        }
    }
    .into()
}