use syn::{Meta, Token, punctuated::Punctuated};
use crate::{
common::{
boolean::meta_2_bool, range::Range, rocket_options::RocketOptions,
serde_options::SerdeOptions,
},
panic,
};
pub(crate) struct ByteAttribute {
pub(crate) range: Range<u128>,
pub(crate) ignore_case: bool,
#[cfg_attr(not(feature = "serde"), allow(dead_code))]
pub(crate) serde_options: SerdeOptions,
#[cfg_attr(not(feature = "rocket"), allow(dead_code))]
pub(crate) rocket_options: RocketOptions,
}
impl ByteAttribute {
pub(crate) fn build_from_meta(meta: &Meta) -> syn::Result<Self> {
let correct_parameters = ["range", "ignore_case", "serde", "rocket"];
let mut range = Range::new();
let mut ignore_case = true;
let mut serde_options = SerdeOptions::default();
let mut rocket_options = RocketOptions::default();
match meta {
Meta::Path(_) => (),
Meta::NameValue(_) => {
return Err(panic::attribute_incorrect_format(meta.path()));
},
Meta::List(list) => {
let result =
list.parse_args_with(Punctuated::<Meta, Token![,]>::parse_terminated)?;
let mut range_is_set = false;
let mut ignore_case_is_set = false;
let mut serde_options_is_set = false;
let mut rocket_options_is_set = false;
let mut handler = |meta: &Meta| -> syn::Result<bool> {
if let Some(ident) = meta.path().get_ident() {
match ident.to_string().as_str() {
"range" => {
let v = Range::from_meta(meta)?;
if range_is_set {
return Err(panic::parameter_reset(ident));
}
range_is_set = true;
range = v;
return Ok(true);
},
"ignore_case" => {
let v = meta_2_bool(meta)?;
if ignore_case_is_set {
return Err(panic::parameter_reset(ident));
}
ignore_case_is_set = true;
ignore_case = v;
return Ok(true);
},
"serde" => {
let v = SerdeOptions::from_meta(meta)?;
if serde_options_is_set {
return Err(panic::parameter_reset(ident));
}
serde_options_is_set = true;
serde_options = v;
return Ok(true);
},
"rocket" => {
let v = RocketOptions::from_meta(meta)?;
if rocket_options_is_set {
return Err(panic::parameter_reset(ident));
}
rocket_options_is_set = true;
rocket_options = v;
return Ok(true);
},
_ => (),
}
}
Ok(false)
};
for p in result {
if !handler(&p)? {
return Err(panic::parameter_incorrect_format(
p.path(),
&correct_parameters,
));
}
}
},
}
Ok(Self {
range,
ignore_case,
serde_options,
rocket_options,
})
}
}