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

Add method builder

Browse Source
This commit is contained in:
Joel Wachsler 2022-07-12 13:06:07 +00:00
parent 95104b653b
commit 8ed9a98843
1 changed files with 47 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
qbittorrent-web-api-gen/src/generate/group/method/method_without_params.rs
View File

@ -3,6 +3,49 @@ use quote::quote;
use super::return_type::create_return_type;
struct MethodBuilder {
method_name: syn::Ident,
url: String,
return_type: Option<proc_macro2::TokenStream>,
}
impl MethodBuilder {
fn new(method_name: &syn::Ident, url: &str) -> Self {
Self {
method_name: method_name.clone(),
url: url.to_string(),
return_type: None,
}
}
fn return_type(mut self, value: &proc_macro2::TokenStream) -> Self {
self.return_type = Some(value.clone());
self
}
fn build(&self) -> proc_macro2::TokenStream {
let method_name = &self.method_name;
let (return_type, parse_type) = match &self.return_type {
Some(t) => (t.clone(), quote! { .json::<#t>() }),
None => (quote! { String }, quote! { .text() }),
};
let url = &self.url;
quote! {
pub async fn #method_name(&self) -> Result<#return_type> {
let res = self.auth
.authenticated_client(#url)
.send()
.await?
#parse_type
.await?;
Ok(res)
}
}
}
}
pub fn create_method_without_params(
group: &parser::ApiGroup,
method: &parser::ApiMethod,
@ -11,33 +54,13 @@ pub fn create_method_without_params(
) -> (proc_macro2::TokenStream, Option<proc_macro2::TokenStream>) {
let res = match create_return_type(group, method) {
Some((return_type_name, return_type)) => (
quote! {
pub async fn #method_name(&self) -> Result<#return_type_name> {
let res = self.auth
.authenticated_client(#url)
.send()
.await?
.json::<#return_type_name>()
.await?;
Ok(res)
}
},
MethodBuilder::new(&method_name, url)
.return_type(&return_type_name)
.build(),
Some(return_type),
),
None => (
quote! {
pub async fn #method_name(&self) -> Result<String> {
let res = self.auth
.authenticated_client(#url)
.send()
.await?
.text()
.await?;
Ok(res)
}
},
MethodBuilder::new(&method_name, url).build(),
// assume that all methods without a return type returns a string
None,
),