From 43922cda4a8ca955c133e390a4745996a49dec93 Mon Sep 17 00:00:00 2001 From: Hatter Jiang Date: Sun, 17 Dec 2023 15:33:21 +0800 Subject: [PATCH] feat: olt ask --- post-rs/src/main.rs | 29 ++++++++++++++++------------- 1 file changed, 16 insertions(+), 13 deletions(-) diff --git a/post-rs/src/main.rs b/post-rs/src/main.rs index d9dff75..db44098 100644 --- a/post-rs/src/main.rs +++ b/post-rs/src/main.rs @@ -17,6 +17,7 @@ use std::path::PathBuf; use base64::Engine; use clap::{arg, Parser}; use rust_util::{debugging, failure_and_exit, information, success, util_size, warning}; +use rust_util::util_term::read_yes_no; use serde_json::Value; #[derive(Parser, Debug)] @@ -112,19 +113,21 @@ fn main() { Err(e) => failure_and_exit!("Post file: {} response failed: {}", file_name, e), Ok(text) => { success!("Post file: {} succeed: {}", file_name, &text); - let post_response: Value = serde_json::from_str(&text).unwrap(); - if let Some(post_response_data) = post_response.get("data") { - if let Some(Value::String(full_link)) = post_response_data.get("fullLink") { - let mut params = HashMap::new(); - params.insert("pretty", "true".to_string()); - params.insert("targetUrl", full_link.to_string()); - params.insert("aliveTimeMin", "10".to_string()); - let otl_result = client.post("https://hatter.ink/onetimelink/add.json") - .form(¶ms) - .send(); - if let Ok(response) = otl_result { - if let Ok(text) = response.text() { - success!("File one time link: {}", text); + if read_yes_no("Generate One Time Link") { + let post_response: Value = serde_json::from_str(&text).unwrap(); + if let Some(post_response_data) = post_response.get("data") { + if let Some(Value::String(full_link)) = post_response_data.get("fullLink") { + let mut params = HashMap::new(); + params.insert("pretty", "true".to_string()); + params.insert("targetUrl", full_link.to_string()); + params.insert("aliveTimeMin", "10".to_string()); + let otl_result = client.post("https://hatter.ink/onetimelink/add.json") + .form(¶ms) + .send(); + if let Ok(response) = otl_result { + if let Ok(text) = response.text() { + success!("File one time link: {}", text); + } } } }