enum_all_values_const 0.1.1

a proc macro which provides access to all values of an enum even in const contexts
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

extern crate proc_macro;
extern crate syn;
#[macro_use]
extern crate quote;

use proc_macro::TokenStream;

#[proc_macro_derive(AllValues)]
pub fn derive_all_variants(input: TokenStream) -> TokenStream {
    let syn_item: syn::DeriveInput = syn::parse(input).unwrap();
    let variants = match syn_item.data {
        syn::Data::Enum(enum_item) => enum_item.variants.into_iter().map(|v| v.ident),
        _ => panic!("AllValues only works on enums"),
    };

    let variants_amount = variants.len();

    let enum_name = syn_item.ident;
    let expanded = quote! {
        impl #enum_name {
            pub const ALL_VALUES: [#enum_name; #variants_amount] = [ #(#enum_name::#variants),* ];
        }
    };

    expanded.into()
}