move client & logic into api module

1 files changed, 29 insertions, 0 deletions

diff --git a/src/api/shiggy.rs b/src/api/shiggy.rs
new file mode 100644
index 0000000..0e9fd19
--- /dev/null
+++ b/src/api/shiggy.rs
@@ -0,0 +1,29 @@
+use crate::api::REQWEST_CLIENT;
+use reqwest::StatusCode;
+use serde::Deserialize;
+
+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 get_random_shiggy() -> 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 => format!("{} ({:?})", ERROR_MSG, other),
+	}
+}