diff options
| -rw-r--r-- | src/commands/mod.rs | 1 | ||||
| -rw-r--r-- | src/commands/random_shiggy.rs | 40 | ||||
| -rw-r--r-- | src/main.rs | 2 |
3 files changed, 43 insertions, 0 deletions
diff --git a/src/commands/mod.rs b/src/commands/mod.rs index fb7ee4b..1640707 100644 --- a/src/commands/mod.rs +++ b/src/commands/mod.rs @@ -3,4 +3,5 @@ pub mod bottom; pub mod convert; pub mod copypasta; pub mod random_lore; +pub mod random_shiggy; pub mod random_teawie; diff --git a/src/commands/random_shiggy.rs b/src/commands/random_shiggy.rs new file mode 100644 index 0000000..b1f1511 --- /dev/null +++ b/src/commands/random_shiggy.rs @@ -0,0 +1,40 @@ +use crate::api::guzzle::REQWEST_CLIENT; +use reqwest::StatusCode; +use serde::Deserialize; +use serenity::builder::CreateApplicationCommand; +use serenity::model::prelude::application_command::CommandDataOption; + +const URL: &str = "https://safebooru.donmai.us/posts/random.json?tags=kemomimi-chan_(naga_u)+naga_u&only=file_url"; +const ERROR_MSG: &str = "couldn't get a shiggy"; + +#[derive(Deserialize)] +struct SafebooruResponse { + file_url: String, +} + +pub async fn run(_: &[CommandDataOption]) -> String { + let resp = match REQWEST_CLIENT + .execute(REQWEST_CLIENT.get(URL).build().unwrap()) + .await + { + Ok(r) => r, + Err(e) => return format!("{} ({:?})", ERROR_MSG, e), + }; + + match resp.status() { + StatusCode::OK => match resp.json::<SafebooruResponse>().await { + Ok(sr) => sr.file_url, + Err(e) => format!("{} ({:?})", ERROR_MSG, e), + }, + other => { + println!("{}", resp.text().await.unwrap()); + format!("{} ({:?})", ERROR_MSG, other) + } + } +} + +pub fn register(command: &mut CreateApplicationCommand) -> &mut CreateApplicationCommand { + command + .name("random_shiggy") + .description("get a random shiggy!") +} diff --git a/src/main.rs b/src/main.rs index 8a3aa94..51f0f3c 100644 --- a/src/main.rs +++ b/src/main.rs @@ -111,6 +111,7 @@ impl EventHandler for Handler { .await } "random_lore" => commands::random_lore::run(&command.data.options), + "random_shiggy" => commands::random_shiggy::run(&command.data.options).await, "random_teawie" => commands::random_teawie::run(&command.data.options).await, _ => "not implemented :(".to_string(), }; @@ -145,6 +146,7 @@ impl EventHandler for Handler { .create_application_command(commands::bottom::register) .create_application_command(commands::convert::register) .create_application_command(commands::random_lore::register) + .create_application_command(commands::random_shiggy::register) .create_application_command(commands::random_teawie::register) }) .await; |