added UserAgent to referral and points

Author: erhant

Cargo.lock

@@ -1,7 +1,7 @@
 [package]
 name = "dkn-compute-launcher"
 edition = "2021"
-version = "0.1.12"
+version = "0.1.13"
 license = "Apache-2.0"
 readme = "README.md"
 description = "Dria Compute Node Launcher"

Cargo.toml

@@ -1,6 +1,6 @@
 use eyre::Context;
 
-use crate::utils::DriaEnv;
+use crate::utils::{DriaEnv, LAUNCHER_USER_AGENT};
 
 const POINTS_API_BASE_URL: &str =
     "https://mainnet.dkn.dria.co/dashboard/supply/v0/leaderboard/steps";
@@ -31,10 +31,16 @@ pub async fn show_points() -> eyre::Result<()> {
         address.trim_start_matches("0x")
     );
 
-    let res = reqwest::get(&url)
+    let client = reqwest::Client::builder()
+        .user_agent(LAUNCHER_USER_AGENT)
+        .build()
+        .wrap_err("could not create reqwest client")?;
+
+    let res = client
+        .get(&url)
+        .send()
         .await
         .wrap_err("could not make request")?;
-    // println!("Response: {:?}", res.text().await);
 
     let points = res
         .json::<PointsRes>()

src/commands/points.rs

@@ -44,3 +44,7 @@ pub const PROGRESS_BAR_TEMPLATE: &str =
 
 /// Progress bar characters for download progress.
 pub const PROGRESS_BAR_CHARS: &str = "=>-";
+
+/// `UserAgent` header value for the launcher, used for HTTP requests.
+pub const LAUNCHER_USER_AGENT: &str =
+    concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"));

src/utils/mod.rs

@@ -1,11 +1,11 @@
 use eyre::{eyre, Context, Result};
 use libsecp256k1::SecretKey;
-use reqwest::Client;
 
 use crate::utils::crypto::eip191_hash;
 
-const REFERRALS_API_BASE_URL: &str = "https://dkn.dria.co/referral/v0";
-// const REFERRALS_API_BASE_URL: &str = "http://localhost:8080/referral/v0";
+use super::LAUNCHER_USER_AGENT;
+
+const REFERRALS_API_BASE_URL: &str = "https://mainnet.dkn.dria.co/referrals/v0/";
 
 pub struct ReferralsClient {
     base_url: String,
@@ -20,9 +20,14 @@ impl Default for ReferralsClient {
 
 impl ReferralsClient {
     pub fn new(base_url: &str) -> Self {
+        let client = reqwest::Client::builder()
+            .user_agent(LAUNCHER_USER_AGENT)
+            .build()
+            .expect("could not create reqwest client");
+
         Self {
             base_url: base_url.to_string(),
-            client: Client::new(),
+            client,
         }
     }