max/macroconf
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove unwraps and panics

Browse Source
This commit is contained in:
Max Känner 2024-04-15 21:40:32 +02:00
parent 3ad76d1c00
commit 65b47996e3
1 changed files with 33 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
src/lib.rs
View File

@ -3,7 +3,7 @@ use std::iter::once;
use convert_case::{Case, Casing}; use convert_case::{Case, Casing};
use proc_macro::TokenStream; use proc_macro::TokenStream;
use proc_macro2::{Span as Span2, TokenStream as TokenStream2}; use proc_macro2::{Span as Span2, TokenStream as TokenStream2};
use quote::{format_ident, quote}; use quote::{format_ident, quote, ToTokens};
use syn::{ use syn::{
parse_macro_input, parse_quote, Attribute, DataStruct, DeriveInput, Expr, ExprLit, Field, parse_macro_input, parse_quote, Attribute, DataStruct, DeriveInput, Expr, ExprLit, Field,
Ident, Lit, Type, Visibility, Ident, Lit, Type, Visibility,
@ -12,10 +12,6 @@ use syn::{
/// Creates structs for the values to extend them with extra metadata. /// Creates structs for the values to extend them with extra metadata.
/// ///
/// supported metadata is `min`, `max` and `default`. Doc comments are parsed as `description` /// supported metadata is `min`, `max` and `default`. Doc comments are parsed as `description`
///
/// # Panics
///
/// Panics if it is unable to generate a config struct from the input
#[proc_macro_attribute] #[proc_macro_attribute]
pub fn config(_attr: TokenStream, item: TokenStream) -> TokenStream { pub fn config(_attr: TokenStream, item: TokenStream) -> TokenStream {
let mut input = parse_macro_input!(item as DeriveInput); let mut input = parse_macro_input!(item as DeriveInput);
@ -33,8 +29,12 @@ pub fn config(_attr: TokenStream, item: TokenStream) -> TokenStream {
} }
} }
} }
syn::Data::Enum(_) => panic!("Enums are not supported"), syn::Data::Enum(_) => {
syn::Data::Union(_) => panic!("Unions are not supported"), return quote! {compile_error!("Enums are not supported")}.into();
}
syn::Data::Union(_) => {
return quote! {compile_error!("Unions are not supported")}.into();
}
} }
quote! { quote! {
@ -145,9 +145,12 @@ fn parse_min(
let const_ident = Ident::new(KEY.to_case(Case::Upper).as_str(), Span2::mixed_site()); let const_ident = Ident::new(KEY.to_case(Case::Upper).as_str(), Span2::mixed_site());
let (value, new_check) = attr.meta.require_path_only().map_or_else( let (value, new_check) = attr.meta.require_path_only().map_or_else(
|_| { |_| {
let value = &attr.meta.require_name_value().unwrap().value; let value = match &attr.meta.require_name_value() {
Ok(meta) => &meta.value,
Err(e) => return (e.to_owned().into_compile_error(), None),
};
( (
quote!(#value), value.to_token_stream(),
Some(quote! { Some(quote! {
if (value < Self::#const_ident.0) { if (value < Self::#const_ident.0) {
return None; return None;
@ -177,9 +180,12 @@ fn parse_max(
let const_ident = Ident::new(KEY.to_case(Case::Upper).as_str(), Span2::mixed_site()); let const_ident = Ident::new(KEY.to_case(Case::Upper).as_str(), Span2::mixed_site());
let (value, new_check) = attr.meta.require_path_only().map_or_else( let (value, new_check) = attr.meta.require_path_only().map_or_else(
|_| { |_| {
let value = &attr.meta.require_name_value().unwrap().value; let value = match &attr.meta.require_name_value() {
Ok(meta) => &meta.value,
Err(e) => return (e.to_owned().into_compile_error(), None),
};
( (
quote!(#value), value.to_token_stream(),
Some(quote! { Some(quote! {
if (value > Self::#const_ident.0) { if (value > Self::#const_ident.0) {
return None; return None;
@ -208,9 +214,9 @@ fn parse_default(
} }
let const_ident = Ident::new(KEY.to_case(Case::Upper).as_str(), Span2::mixed_site()); let const_ident = Ident::new(KEY.to_case(Case::Upper).as_str(), Span2::mixed_site());
let value = attr.meta.require_path_only().map_or_else( let value = attr.meta.require_path_only().map_or_else(
|_| { |_| match &attr.meta.require_name_value() {
let value = &attr.meta.require_name_value().unwrap().value; Ok(meta) => meta.value.to_token_stream(),
quote!(#value) Err(e) => e.to_owned().into_compile_error(),
}, },
|_| quote!(#ty::#const_ident), |_| quote!(#ty::#const_ident),
); );
@ -237,13 +243,24 @@ fn parse_description(
return None; return None;
} }
let const_ident = Ident::new(KEY.to_case(Case::Upper).as_str(), Span2::mixed_site()); let const_ident = Ident::new(KEY.to_case(Case::Upper).as_str(), Span2::mixed_site());
let value = &attr.meta.require_name_value().ok()?.value; let value = match attr.meta.require_name_value() {
Ok(meta) => &meta.value,
Err(e) => {
return Some((e.into_compile_error(), const_ident, KEY));
}
};
let value = match value { let value = match value {
Expr::Lit(ExprLit { Expr::Lit(ExprLit {
attrs: _, attrs: _,
lit: Lit::Str(ref string), lit: Lit::Str(ref string),
}) => string.value(), }) => string.value(),
_ => panic!("doc comment must be a string"), _ => {
return Some((
quote!(compile_error!("doc comment must be a string")),
const_ident,
KEY,
))
}
}; };
let trimmed_value = value.trim(); let trimmed_value = value.trim();
let impl_quote = quote! { let impl_quote = quote! {