Start9/rpc-toolkit
2
0
Fork 0
mirror of https://github.com/Start9Labs/rpc-toolkit.git synced 2026-03-26 10:21:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix cli

Browse Source
This commit is contained in:
Aiden McClelland
2022-08-01 10:25:53 -06:00
parent 9351e89240
commit ac4c8db6e1
2 changed files with 30 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
rpc-toolkit-macro-internals/src/command/build.rs
View File

@@ -796,10 +796,6 @@ fn cli_handler(
.unwrap_or_else(|| LitStr::new(&name.to_string(), name.span()));
let field_name = Ident::new(&format!("arg_{}", name), name.span());
let ty = arg.ty.clone();
let mut ty = quote! { #ty };
if arg.default.is_some() && !arg.optional {
ty = quote! { Option<#ty> };
}
arg_def.push(quote! {
#[serde(rename = #rename)]
#field_name: #ty,
@@ -849,7 +845,7 @@ fn cli_handler(
::rpc_toolkit::command_helpers::prelude::default_arg_parser(arg_val, matches)
}
};
if arg.optional || arg.default.is_some() {
if arg.optional {
quote! {
#field_name: if let Some(arg_val) = matches.value_of(#arg_name) {
Some(#parse_val?)
@@ -857,6 +853,28 @@ fn cli_handler(
None
},
}
} else if let Some(default) = &arg.default {
if let Some(default) = default {
let path: Path = match syn::parse_str(&default.value()) {
Ok(a) => a,
Err(e) => return e.into_compile_error(),
};
quote! {
#field_name: if let Some(arg_val) = matches.value_of(#arg_name) {
#parse_val?
} else {
#path()
},
}
} else {
quote! {
#field_name: if let Some(arg_val) = matches.value_of(#arg_name) {
#parse_val?
} else {
Default::default()
},
}
}
} else if arg.multiple.is_some() {
quote! {
#field_name: matches.values_of(#arg_name).iter().flatten().map(|arg_val| #parse_val).collect::<Result<_, _>>()?,