use proc_macro2::TokenStream; | |
use syn; | |
use helpers::{extract_attrs, extract_meta, is_disabled, unique_attr}; | |
fn get_arms(ast: &syn::DeriveInput) -> Vec<TokenStream> { | |
let name = &ast.ident; | |
let mut arms = Vec::new(); | |
let variants = match ast.data { | |
syn::Data::Enum(ref v) => &v.variants, | |
_ => panic!("This macro only works on Enums"), | |
}; | |
for variant in variants { | |
use syn::Fields::*; | |
let ident = &variant.ident; | |
let meta = extract_meta(&variant.attrs); | |
if is_disabled(&meta) { | |
continue; | |
} | |
// Look at all the serialize attributes. | |
// Use `to_string` attribute (not `as_ref_str` or something) to keep things consistent | |
// (i.e. always `enum.as_ref().to_string() == enum.to_string()`). | |
let output = if let Some(n) = unique_attr(&meta, "strum", "to_string") { | |
n | |
} else { | |
let mut attrs = extract_attrs(&meta, "strum", "serialize"); | |
// We always take the longest one. This is arbitary, but is *mostly* deterministic | |
attrs.sort_by_key(|s| s.len()); | |
if let Some(n) = attrs.pop() { | |
n | |
} else { | |
ident.to_string() | |
} | |
}; | |
let params = match variant.fields { | |
Unit => quote!{}, | |
Unnamed(..) => quote!{ (..) }, | |
Named(..) => quote!{ {..} }, | |
}; | |
arms.push(quote!{ #name::#ident #params => #output }); | |
} | |
if arms.len() < variants.len() { | |
arms.push(quote!{ | |
_ => panic!("AsRef::<str>::as_ref() or AsStaticRef::<str>::as_static() \ | |
called on disabled variant.") | |
}) | |
} | |
arms | |
} | |
pub fn as_ref_str_inner(ast: &syn::DeriveInput) -> TokenStream { | |
let name = &ast.ident; | |
let (impl_generics, ty_generics, where_clause) = ast.generics.split_for_impl(); | |
let arms = get_arms(ast); | |
quote!{ | |
impl #impl_generics ::std::convert::AsRef<str> for #name #ty_generics #where_clause { | |
fn as_ref(&self) -> &str { | |
match *self { | |
#(#arms),* | |
} | |
} | |
} | |
} | |
} | |
pub fn as_static_str_inner(ast: &syn::DeriveInput) -> TokenStream { | |
let name = &ast.ident; | |
let (impl_generics, ty_generics, where_clause) = ast.generics.split_for_impl(); | |
let arms = get_arms(ast); | |
quote!{ | |
impl #impl_generics ::strum::AsStaticRef<str> for #name #ty_generics #where_clause { | |
fn as_static(&self) -> &'static str { | |
match *self { | |
#(#arms),* | |
} | |
} | |
} | |
} | |
} |