From cd379012774431226c9fdd56c438ad65ebfb41fb Mon Sep 17 00:00:00 2001 From: cloudgazing Date: Wed, 25 Sep 2024 07:10:44 +0300 Subject: [PATCH] Added some delay and cli arguments --- rust/Cargo.lock | 122 ++++++++++++++++++++++++++++++++++++++++++++++- rust/Cargo.toml | 3 +- rust/src/main.rs | 29 ++++++++++- 3 files changed, 151 insertions(+), 3 deletions(-) diff --git a/rust/Cargo.lock b/rust/Cargo.lock index 4edd75a..bce0ba0 100644 --- a/rust/Cargo.lock +++ b/rust/Cargo.lock @@ -17,6 +17,55 @@ version = "2.0.0" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "512761e0bb2578dd7380c6baaa0f4ce03e84f95e960231d1dec8bf4d7d6e2627" +[[package]] +name = "anstream" +version = "0.6.15" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "64e15c1ab1f89faffbf04a634d5e1962e9074f2741eef6d97f3c4e322426d526" +dependencies = [ + "anstyle", + "anstyle-parse", + "anstyle-query", + "anstyle-wincon", + "colorchoice", + "is_terminal_polyfill", + "utf8parse", +] + +[[package]] +name = "anstyle" +version = "1.0.8" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1bec1de6f59aedf83baf9ff929c98f2ad654b97c9510f4e70cf6f661d49fd5b1" + +[[package]] +name = "anstyle-parse" +version = "0.2.5" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "eb47de1e80c2b463c735db5b217a0ddc39d612e7ac9e2e96a5aed1f57616c1cb" +dependencies = [ + "utf8parse", +] + +[[package]] +name = "anstyle-query" +version = "1.1.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "6d36fc52c7f6c869915e99412912f22093507da8d9e942ceaf66fe4b7c14422a" +dependencies = [ + "windows-sys 0.52.0", +] + +[[package]] +name = "anstyle-wincon" +version = "3.0.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "5bf74e1b6e971609db8ca7a9ce79fd5768ab6ae46441c572e46cf596f59e57f8" +dependencies = [ + "anstyle", + "windows-sys 0.52.0", +] + [[package]] name = "atomic-waker" version = "1.1.2" @@ -83,6 +132,52 @@ version = "1.0.0" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd" +[[package]] +name = "clap" +version = "4.5.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b0956a43b323ac1afaffc053ed5c4b7c1f1800bacd1683c353aabbb752515dd3" +dependencies = [ + "clap_builder", + "clap_derive", +] + +[[package]] +name = "clap_builder" +version = "4.5.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "4d72166dd41634086d5803a47eb71ae740e61d84709c36f3c34110173db3961b" +dependencies = [ + "anstream", + "anstyle", + "clap_lex", + "strsim", +] + +[[package]] +name = "clap_derive" +version = "4.5.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "4ac6a0c7b1a9e9a5186361f67dfa1b88213572f427fb9ab038efb2bd8c582dab" +dependencies = [ + "heck", + "proc-macro2", + "quote", + "syn", +] + +[[package]] +name = "clap_lex" +version = "0.7.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1462739cb27611015575c0c11df5df7601141071f07518d56fcc1be504cbec97" + +[[package]] +name = "colorchoice" +version = "1.0.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "d3fd119d74b830634cea2a0f58bbd0d54540518a14397557951e79340abc28c0" + [[package]] name = "core-foundation" version = "0.9.4" @@ -252,6 +347,12 @@ version = "0.14.5" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "e5274423e17b7c9fc20b6e7e208532f9b19825d82dfd615708b70edd83df41f1" +[[package]] +name = "heck" +version = "0.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea" + [[package]] name = "hermit-abi" version = "0.3.9" @@ -396,6 +497,12 @@ version = "2.10.0" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "187674a687eed5fe42285b40c6291f9a01517d415fad1c3cbc6a9f778af7fcd4" +[[package]] +name = "is_terminal_polyfill" +version = "1.70.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf" + [[package]] name = "itoa" version = "1.0.11" @@ -464,8 +571,9 @@ dependencies = [ [[package]] name = "mkbsd" -version = "0.1.0" +version = "0.1.1" dependencies = [ + "clap", "reqwest", "serde", "serde_json", @@ -826,6 +934,12 @@ version = "0.9.8" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "6980e8d7511241f8acf4aebddbb1ff938df5eebe98691418c4468d0b72a96a67" +[[package]] +name = "strsim" +version = "0.11.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f" + [[package]] name = "subtle" version = "2.6.1" @@ -1019,6 +1133,12 @@ dependencies = [ "percent-encoding", ] +[[package]] +name = "utf8parse" +version = "0.2.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821" + [[package]] name = "vcpkg" version = "0.2.15" diff --git a/rust/Cargo.toml b/rust/Cargo.toml index 1fa3117..164d148 100644 --- a/rust/Cargo.toml +++ b/rust/Cargo.toml @@ -1,9 +1,10 @@ [package] name = "mkbsd" -version = "0.1.0" +version = "0.1.1" edition = "2021" [dependencies] +clap = { version = "4.5.18", features = ["derive"] } reqwest = { version = "0.12.7", features = ["blocking"] } serde = { version = "1.0.210", features = ["derive"] } serde_json = "1.0.128" diff --git a/rust/src/main.rs b/rust/src/main.rs index 3024621..380207c 100644 --- a/rust/src/main.rs +++ b/rust/src/main.rs @@ -1,7 +1,20 @@ use std::collections::HashMap; +use clap::Parser; use serde::{Deserialize, Serialize}; +#[derive(Parser, Debug)] +#[command(version, about, long_about = None)] +struct Args { + /// Skip the delay and download files faster (only enable this if you absolutely cannot wait a bit longer) + #[arg(short, long, default_value_t = false)] + fast: bool, + + /// Download all files, not just the "high res" ones + #[arg(short, long, default_value_t = false)] + all: bool, +} + #[derive(Debug, Deserialize, Serialize)] struct JsonData { version: i8, @@ -28,6 +41,8 @@ fn download_file(url: &str, filename: &str) -> Result<(), Box Result<(), Box> { + let args = Args::parse(); + let file_url = "https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s"; let Ok(json_text) = reqwest::blocking::get(file_url)?.text() else { @@ -55,9 +70,17 @@ fn main() -> Result<(), Box> { ); println!("\n 🤑 Starting downloads from your favorite sellout grifter's wallpaper app... \n"); + if !args.fast { + std::thread::sleep(std::time::Duration::from_millis(5000)); + } + for (key_1, inner_map) in json_data.data { for (key_2, url) in inner_map { - if key_2 != "dhd" { + if !args.fast { + std::thread::sleep(std::time::Duration::from_millis(100)); + } + + if !args.all && key_2 != "dhd" { continue; } @@ -81,6 +104,10 @@ fn main() -> Result<(), Box> { println!("Error downloading file: {}", filename); } } + + if !args.fast { + std::thread::sleep(std::time::Duration::from_millis(200)); + } } }