From 252ea9e65eddf78e50fcd895db37bccfc891bc32 Mon Sep 17 00:00:00 2001 From: lifegpc Date: Sun, 8 Oct 2023 02:40:23 +0000 Subject: [PATCH] Add pushdeer client --- proc_macros/proc_macros.rs | 31 ++++++++++ src/ext/try_err.rs | 19 +++++++ src/push/mod.rs | 2 + src/push/pushdeer.rs | 113 +++++++++++++++++++++++++++++++++++++ 4 files changed, 165 insertions(+) create mode 100644 src/push/pushdeer.rs diff --git a/proc_macros/proc_macros.rs b/proc_macros/proc_macros.rs index 89163dbb..d2e6ca65 100644 --- a/proc_macros/proc_macros.rs +++ b/proc_macros/proc_macros.rs @@ -985,3 +985,34 @@ pub fn print_error(item: TokenStream) -> TokenStream { ); stream.into() } + +#[proc_macro] +pub fn pushdeer_api_quick_test(item: TokenStream) -> TokenStream { + let FanboxApiQuickTest { name, expr, errmsg } = parse_macro_input!(item as FanboxApiQuickTest); + let stream = quote! { + #[proc_macros::async_timeout_test(120s)] + #[tokio::test(flavor = "multi_thread")] + async fn #name() { + match std::env::var("PUSHDEER_SERVER") { + Ok(server) => match std::env::var("PUSHDEER_TOKEN") { + Ok(token) => { + let client = PushdeerClient::new(server); + match #expr.await { + Ok(_) => {}, + Err(e) => { + panic!("{}{}", #errmsg, e); + } + } + } + Err(_) => { + println!("No pushdeer token specified, skip test.") + } + }, + Err(_) => { + println!("No pushdeer server specified, skip test.") + } + } + } + }; + stream.into() +} diff --git a/src/ext/try_err.rs b/src/ext/try_err.rs index c65e8763..dc7382d9 100644 --- a/src/ext/try_err.rs +++ b/src/ext/try_err.rs @@ -22,6 +22,13 @@ pub trait TryErr3 { fn try_err3 + ?Sized>(self, code: i32, msg: &S) -> Result; } +/// A quick way to add detailed message before error +pub trait TryErr4 { + /// A quick way to add detailed message before error + /// * `message` - detailed message + fn try_err4 + ?Sized>(self, message: &S) -> Result; +} + impl, E> TryErr2 for Option { fn try_err2(&self, v: E) -> Result { match self { @@ -85,3 +92,15 @@ impl TryErr3 for Option { } } } + +impl TryErr4 for Result +where + E: std::fmt::Display, +{ + fn try_err4 + ?Sized>(self, message: &S) -> Result { + match self { + Ok(v) => Ok(v), + Err(e) => Err(format!("{} {}", message.as_ref(), e)), + } + } +} diff --git a/src/push/mod.rs b/src/push/mod.rs index 6ea4abd3..bba52da7 100644 --- a/src/push/mod.rs +++ b/src/push/mod.rs @@ -1,2 +1,4 @@ /// [EveryPush](https://github.com/PeanutMelonSeedBigAlmond/EveryPush) client pub mod every_push; +/// [Pushdeer](https://github.com/easychen/pushdeer) client +pub mod pushdeer; diff --git a/src/push/pushdeer.rs b/src/push/pushdeer.rs new file mode 100644 index 00000000..b4420ac6 --- /dev/null +++ b/src/push/pushdeer.rs @@ -0,0 +1,113 @@ +use crate::ext::try_err::TryErr4; +use crate::webclient::WebClient; +use proc_macros::pushdeer_api_quick_test; +use std::collections::HashMap; + +pub struct PushdeerClient { + client: WebClient, + server: String, +} + +impl PushdeerClient { + pub fn new(server: String) -> Self { + Self { + client: WebClient::default(), + server, + } + } + + async fn handle_result(re: reqwest::Response) -> Result<(), String> { + let obj = json::parse( + re.text() + .await + .try_err4("Failed to read text from response: ")? + .as_str(), + ) + .try_err4("Failed to parse JSON: ")?; + let code = obj["code"].as_i64().ok_or("Failed to get code.")?; + if code == 0 { + Ok(()) + } else { + let msg = obj["error"] + .as_str() + .ok_or("Failed to get error message.")?; + Err(msg.to_owned()) + } + } + + /// push text message to server + /// * `pushkey` - push key + /// * `text` - text + pub async fn push_text_message(&self, pushkey: String, text: String) -> Result<(), String> { + let mut params: HashMap = HashMap::new(); + params.insert(String::from("pushkey"), pushkey); + params.insert(String::from("text"), text); + let re = self + .client + .post(format!("{}/message/push", self.server), None, Some(params)) + .await + .ok_or("Failed to send text message.")?; + Self::handle_result(re).await + } + + /// push image message to server + /// * `pushkey` - push key + /// * `image` - image URL + pub async fn push_image(&self, pushkey: String, image: String) -> Result<(), String> { + let mut params: HashMap = HashMap::new(); + params.insert(String::from("pushkey"), pushkey); + params.insert(String::from("text"), image); + params.insert(String::from("type"), String::from("image")); + let re = self + .client + .post(format!("{}/message/push", self.server), None, Some(params)) + .await + .ok_or("Failed to send image message.")?; + Self::handle_result(re).await + } + + /// push markdown message to server + /// * `pushkey` - push key + /// * `title` - title + /// * `text` - markdown text + pub async fn push_markdown_message( + &self, + pushkey: String, + title: String, + text: String, + ) -> Result<(), String> { + let mut params: HashMap = HashMap::new(); + params.insert(String::from("pushkey"), pushkey); + params.insert(String::from("text"), title); + params.insert(String::from("desp"), text); + params.insert(String::from("type"), String::from("markdown")); + let re = self + .client + .post(format!("{}/message/push", self.server), None, Some(params)) + .await + .ok_or("Failed to send markdown message.")?; + Self::handle_result(re).await + } +} + +pushdeer_api_quick_test!( + test_push_text_message, + client.push_text_message(token, String::from("Test message")), + "Failed to send text message:" +); + +pushdeer_api_quick_test!( + test_push_image, + client.push_image(token, String::from("https://pd.lifegpc.com/proxy/pixiv/112199539_p0.jpg?url=https%3A%2F%2Fi.pximg.net%2Fimg-original%2Fimg%2F2023%2F10%2F02%2F00%2F00%2F38%2F112199539_p0.jpg&sign=4f429e69218669e226e1171d2499f49ac7f41f56c3a8275619641c0a3327a394dae63dcd145367d66a3735279dc59b39219eb4d4c20bfe5afb7e801a2e7c2496")), + "Failed to send image message:" +); + +pushdeer_api_quick_test!( + test_push_markdown_message, + client.push_markdown_message( + token, + String::from("Test title"), + String::from("# Test Header\n[link](https://github.com/lifegpc/pixiv_downloader)") + ), + "Failed to send markdown message:" +);