3 files changed, 91 insertions, 0 deletions
diff --git a/src/http/mod.rs b/src/http/mod.rs
new file mode 100644
index 0000000..0f16852
--- /dev/null
+++ b/src/http/mod.rs
@@ -0,0 +1,43 @@
+use eyre::Result;
+use log::trace;
+use serde::de::DeserializeOwned;
+
+pub mod shiggy;
+pub mod teawie;
+
+pub type Client = reqwest::Client;
+pub type Response = reqwest::Response;
+
+/// Primary extensions for HTTP Client
+pub trait Ext {
+	async fn get_request(&self, url: &str) -> Result<Response>;
+	async fn get_json<T: DeserializeOwned>(&self, url: &str) -> Result<T>;
+	fn default() -> Self;
+}
+
+impl Ext for Client {
+	fn default() -> Self {
+		reqwest::ClientBuilder::new()
+			.user_agent(format!(
+				"teawie-bot/{}",
+				option_env!("CARGO_PKG_VERSION").unwrap_or("development")
+			))
+			.build()
+			.unwrap()
+	}
+
+	async fn get_request(&self, url: &str) -> Result<Response> {
+		trace!("Making request to {url}");
+		let resp = self.get(url).send().await?;
+		resp.error_for_status_ref()?;
+
+		Ok(resp)
+	}
+
+	async fn get_json<T: DeserializeOwned>(&self, url: &str) -> Result<T> {
+		let resp = self.get_request(url).await?;
+		let json = resp.json().await?;
+
+		Ok(json)
+	}
+}
diff --git a/src/http/shiggy.rs b/src/http/shiggy.rs
new file mode 100644
index 0000000..397d397
--- /dev/null
+++ b/src/http/shiggy.rs
@@ -0,0 +1,20 @@
+use eyre::Result;
+use serde::Deserialize;
+
+const SHIGGY: &str = "https://safebooru.donmai.us";
+const RANDOM: &str = "/posts/random.json?tags=kemomimi-chan_(naga_u)+naga_u&only=file_url";
+
+#[derive(Deserialize)]
+struct SafebooruResponse {
+	file_url: String,
+}
+
+pub async fn random<T>(http: &T) -> Result<String>
+where
+	T: super::Ext,
+{
+	let url = format!("{SHIGGY}{RANDOM}");
+	let resp: SafebooruResponse = http.get_json(&url).await?;
+
+	Ok(resp.file_url)
+}
diff --git a/src/http/teawie.rs b/src/http/teawie.rs
new file mode 100644
index 0000000..368fad5
--- /dev/null
+++ b/src/http/teawie.rs
@@ -0,0 +1,28 @@
+use eyre::{bail, OptionExt, Result};
+use serde::{Deserialize, Serialize};
+
+// https://github.com/getchoo/teawieAPI
+#[derive(Deserialize, Serialize)]
+struct RandomTeawieResponse {
+	url: Option<String>,
+	error: Option<String>,
+}
+
+// TODO: read this from an env var
+const TEAWIE: &str = "https://api.getchoo.com";
+const RANDOM: &str = "/random_teawie";
+
+pub async fn random<T>(http: &T) -> Result<String>
+where
+	T: super::Ext,
+{
+	let url = format!("{TEAWIE}{RANDOM}");
+	let json: RandomTeawieResponse = http.get_json(&url).await?;
+
+	if let Some(error) = json.error {
+		bail!("TeawieAPI reported error: {error}");
+	};
+
+	json.url
+		.ok_or_eyre("TeawieAPI didn't return an error or URL???")
+}