54 lines
1.7 KiB
Rust
54 lines
1.7 KiB
Rust
use proc_macro2::{Ident, Span, TokenStream};
|
|
use proc_macro_crate::{crate_name, FoundCrate};
|
|
use quote::quote;
|
|
use syn::{ItemEnum, LitStr, Variant};
|
|
|
|
use crate::{
|
|
attr::{RenameAllAttr, RenameAttr},
|
|
case::RenameRule,
|
|
};
|
|
|
|
pub fn import_ruma_serde() -> TokenStream {
|
|
if let Ok(FoundCrate::Name(possibly_renamed)) = crate_name("ruma-serde") {
|
|
let import = Ident::new(&possibly_renamed, Span::call_site());
|
|
quote! { ::#import }
|
|
} else if let Ok(FoundCrate::Name(possibly_renamed)) = crate_name("ruma") {
|
|
let import = Ident::new(&possibly_renamed, Span::call_site());
|
|
quote! { ::#import::serde }
|
|
} else {
|
|
quote! { ::ruma_serde }
|
|
}
|
|
}
|
|
|
|
pub fn get_rename_rule(input: &ItemEnum) -> syn::Result<RenameRule> {
|
|
let rules: Vec<_> = input
|
|
.attrs
|
|
.iter()
|
|
.filter(|attr| attr.path.is_ident("ruma_enum"))
|
|
.map(|attr| attr.parse_args::<RenameAllAttr>().map(RenameAllAttr::into_inner))
|
|
.collect::<syn::Result<_>>()?;
|
|
|
|
match rules.len() {
|
|
0 => Ok(RenameRule::None),
|
|
1 => Ok(rules[0]),
|
|
_ => Err(syn::Error::new(
|
|
Span::call_site(),
|
|
"found multiple ruma_enum(rename_all) attributes",
|
|
)),
|
|
}
|
|
}
|
|
|
|
pub fn get_rename(input: &Variant) -> syn::Result<Option<LitStr>> {
|
|
let renames: Vec<_> = input
|
|
.attrs
|
|
.iter()
|
|
.filter(|attr| attr.path.is_ident("ruma_enum"))
|
|
.map(|attr| attr.parse_args::<RenameAttr>().map(RenameAttr::into_inner))
|
|
.collect::<syn::Result<_>>()?;
|
|
|
|
match renames.len() {
|
|
0 | 1 => Ok(renames.into_iter().next()),
|
|
_ => Err(syn::Error::new(Span::call_site(), "found multiple ruma_enum(rename) attributes")),
|
|
}
|
|
}
|