summaryrefslogtreecommitdiff
path: root/crates/bot-http/src/lib.rs
diff options
context:
space:
mode:
Diffstat (limited to 'crates/bot-http/src/lib.rs')
-rw-r--r--crates/bot-http/src/lib.rs63
1 files changed, 63 insertions, 0 deletions
diff --git a/crates/bot-http/src/lib.rs b/crates/bot-http/src/lib.rs
new file mode 100644
index 0000000..ab32cd4
--- /dev/null
+++ b/crates/bot-http/src/lib.rs
@@ -0,0 +1,63 @@
+use std::future::Future;
+
+use log::trace;
+use serde::de::DeserializeOwned;
+
+mod github;
+mod model;
+mod teawie;
+
+pub use github::ClientExt as GithubClientExt;
+pub use teawie::ClientExt as TeawieClientExt;
+
+pub type Client = reqwest::Client;
+pub type Response = reqwest::Response;
+pub type Error = reqwest::Error;
+
+/// Fun trait for functions we use with [Client]
+pub trait ClientExt {
+ fn default() -> Self;
+ fn get_request(&self, url: &str) -> impl Future<Output = Result<Response, Error>> + Send;
+ fn get_json<T: DeserializeOwned>(
+ &self,
+ url: &str,
+ ) -> impl Future<Output = Result<T, Error>> + Send;
+}
+
+impl ClientExt for Client {
+ /// Create the default [`Client`]
+ fn default() -> Self {
+ reqwest::Client::builder()
+ .user_agent(format!(
+ "nixpkgs-tracker-bot/{}",
+ option_env!("CARGO_PKG_VERSION").unwrap_or_else(|| "development")
+ ))
+ .build()
+ .unwrap()
+ }
+
+ /// Perform a GET request to [`url`]
+ ///
+ /// # Errors
+ ///
+ /// Will return [`Err`] if the request fails
+ async fn get_request(&self, url: &str) -> Result<Response, Error> {
+ trace!("Making GET request to {url}");
+
+ let resp = self.get(url).send().await?;
+ resp.error_for_status_ref()?;
+
+ Ok(resp)
+ }
+
+ /// Perform a GET request to [`url`] and decode the json response
+ ///
+ /// # Errors
+ ///
+ /// Will return [`Err`] if the request fails or cannot be deserialized
+ async fn get_json<T: DeserializeOwned>(&self, url: &str) -> Result<T, Error> {
+ let resp = self.get_request(url).await?;
+ let json = resp.json().await?;
+ Ok(json)
+ }
+}