programmer/qbittorrent_web_api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify parameter handling

Browse Source
This commit is contained in:
Joel Wachsler 2022-07-14 23:04:55 +00:00
parent 72074f60c1
commit 4a862d6e36
1 changed files with 20 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
qbittorrent-web-api-gen/src/generate/group/method/method_with_params.rs
View File

@ -147,27 +147,26 @@ impl<'a> Parameters<'a> {
#[derive(Debug)]
struct MandatoryParams<'a> {
params: &'a ApiParameters,
params: Vec<Parameter<'a>>,
}
impl<'a> MandatoryParams<'a> {
fn new(params: &'a ApiParameters) -> Self {
Self { params }
Self {
params: Parameter::from(&params.mandatory),
}
}
fn generate_params(&self) -> Vec<TokenStream> {
self.params
.mandatory
.iter()
.map(|p| p.to_parameter().generate_param_with_name())
.map(|p| p.generate_param_with_name())
.collect()
}
fn form_builder(&self) -> Vec<TokenStream> {
self.params
.mandatory
.iter()
.map(|p| p.to_parameter())
.map(|param| {
let name_ident = param.name_ident();
let name = param.name();
@ -178,9 +177,8 @@ impl<'a> MandatoryParams<'a> {
fn names(&self) -> Vec<TokenStream> {
self.params
.mandatory
.iter()
.map(|p| p.to_parameter().name_ident())
.map(|p| p.name_ident())
.map(|name_ident| quote! { #name_ident })
.collect()
}
@ -188,37 +186,37 @@ impl<'a> MandatoryParams<'a> {
#[derive(Debug)]
struct OptionalParams<'a> {
params: &'a ApiParameters,
params: Vec<Parameter<'a>>,
}
impl<'a> OptionalParams<'a> {
fn new(params: &'a ApiParameters) -> Self {
Self { params }
Self {
params: Parameter::from(&params.optional),
}
}
fn generate_builder_methods(&self) -> Vec<TokenStream> {
self.params
.optional
.iter()
.map(Self::generate_builder_method)
.collect()
}
fn generate_builder_method(param: &types::Type) -> TokenStream {
let parameter = param.to_parameter();
let name = parameter.name();
let name_ident = parameter.name_ident();
fn generate_builder_method(param: &Parameter) -> TokenStream {
let name = param.name();
let name_ident = param.name_ident();
let param_type = util::to_ident(&param.to_borrowed_type());
let param_type = util::to_ident(&param.p_type.to_borrowed_type());
let builder_param = if param.should_borrow() {
let builder_param = if param.p_type.should_borrow() {
quote! { &#param_type }
} else {
quote! { #param_type }
};
util::add_docs(
&param.get_type_info().description,
&param.p_type.get_type_info().description,
quote! {
pub fn #name_ident(mut self, value: #builder_param) -> Self {
self.form = self.form.text(#name, value.to_string());
@ -239,6 +237,10 @@ impl<'a> Parameter<'a> {
Self { p_type }
}
fn from(parameters: &[types::Type]) -> Vec<Parameter<'_>> {
parameters.iter().map(Parameter::new).collect()
}
fn name(&self) -> String {
self.p_type.get_type_info().name.to_snake()
}
@ -260,9 +262,3 @@ impl<'a> Parameter<'a> {
quote! { #name_ident: #t }
}
}
impl types::Type {
fn to_parameter(&self) -> Parameter<'_> {
Parameter::new(self)
}
}