Check count_args in create_parser

This commit is contained in:
Robin Hundt 2020-09-02 14:29:52 +02:00
parent 1f295f6f36
commit fc7126d727

View file

@ -71,8 +71,9 @@ fn create_parser<'a>(
count_args: usize,
) -> quote::__private::TokenStream {
let function_to_parse = match parser_type {
ParserType::Default => match types.next() {
Some(ty) => {
ParserType::Default => match count_args {
1 => {
let ty = types.next().expect("count_args != types.len()");
quote! { (|s: String| {
let res = <#ty>::from_str(&s)
.map_err(|e|ParseError::IncorrectFormat({ let e: Box<dyn std::error::Error + Send + Sync + 'static> = e.into(); e }))?;
@ -80,7 +81,7 @@ fn create_parser<'a>(
})
}
}
None => quote! { compile_error!("Expected 1 argument") },
_ => quote! { compile_error!("Expected exactly 1 argument") },
},
ParserType::Split { separator } => parser_with_separator(
&separator.clone().unwrap_or_else(|| " ".to_owned()),