feat: add bearer token authentication to GraphQL requests

Author: hrideshmg

.env.sample

@@ -3,3 +3,4 @@ ROOT_URL=https://root.amfoss.in/
 OWNER_ID=
 DEBUG=true
 ENABLE_DEBUG_LIBRARIES=false
+AMD_API_KEY=

src/config.rs

@@ -34,6 +34,7 @@ pub struct Config {
     pub owner_id: Option<UserId>,
     pub prefix_string: String,
     pub root_url: String,
+    pub api_key: String,
 }
 
 impl Default for Config {
@@ -46,6 +47,8 @@ impl Default for Config {
             owner_id: parse_owner_id_env("OWNER_ID"),
             prefix_string: String::from("$"),
             root_url: std::env::var("ROOT_URL").expect("ROOT_URL was not found in env"),
+            api_key: std::env::var("AMD_API_KEY")
+                .expect("AMD_API_KEY was not found in env"),
         }
     }
 }

src/graphql/mod.rs

@@ -26,20 +26,26 @@ use reqwest::Client;
 pub struct GraphQLClient {
     http: Client,
     root_url: Arc<String>,
+    api_key: Arc<String>,
 }
 
 impl GraphQLClient {
-    pub fn new(root_url: String) -> Self {
+    pub fn new(root_url: String, api_key: String) -> Self {
         Self {
             http: Client::new(),
             root_url: Arc::new(root_url),
+            api_key: Arc::new(api_key),
         }
     }
 
     pub fn root_url(&self) -> &str {
         &self.root_url
     }
 
+    pub fn api_key(&self) -> &str {
+        &self.api_key
+    }
+
     pub fn http(&self) -> Client {
         self.http.clone()
     }

src/graphql/queries.rs

@@ -60,6 +60,7 @@ impl GraphQLClient {
         let response = self
             .http()
             .post(self.root_url())
+            .bearer_auth(self.api_key())
             .json(&serde_json::json!({"query": query, "variables":variables}))
             .send()
             .await
@@ -115,6 +116,7 @@ impl GraphQLClient {
         let response = self
             .http()
             .post(self.root_url())
+            .bearer_auth(self.api_key())
             .json(&serde_json::json!({ "query": query }))
             .send()
             .await

src/main.rs

@@ -55,11 +55,11 @@ struct Data {
 
 impl Data {
     /// Returns a new [`Data`] with an empty `reaction_roles` field and the passed-in `reload_handle`.
-    fn new(reload_handle: ReloadHandle, root_url: String) -> Self {
+    fn new(reload_handle: ReloadHandle, root_url: String, api_key: String) -> Self {
         Data {
             reaction_roles: HashMap::new(),
             log_reload_handle: reload_handle,
-            graphql_client: GraphQLClient::new(root_url),
+            graphql_client: GraphQLClient::new(root_url, api_key),
         }
     }
 }
@@ -95,7 +95,11 @@ fn build_framework(
 }
 
 fn prepare_data(config: &Config, reload_handle: ReloadHandle) -> Data {
-    let mut data = Data::new(reload_handle, config.root_url.clone());
+    let mut data = Data::new(
+        reload_handle,
+        config.root_url.clone(),
+        config.api_key.clone(),
+    );
     data.populate_with_reaction_roles();
     data
 }