feat: properly add host support, fix env support

Cargo.toml

@@ -15,7 +15,7 @@
 
 [package]
 name    = "sentry-worker"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 authors = ["Noel <[email protected]>"]
 

src/config.rs

@@ -44,7 +44,10 @@ fn env_to_config() -> Config {
         sentry_dsn,
         port: if port.is_some() {
             let p = port.unwrap();
-            Some(p.parse::<u16>().unwrap())
+            Some(
+                p.parse::<u16>()
+                    .expect("Unable to parse String -> u16 for http port."),
+            )
         } else {
             None
         },
@@ -53,6 +56,12 @@ fn env_to_config() -> Config {
 }
 
 pub static CONFIG: Lazy<Config> = Lazy::new(|| {
-    let contents = read_to_string("config.toml").expect("Unable to open config.toml :(");
-    toml::from_str(&contents).unwrap_or(env_to_config())
+    let res = read_to_string("config.toml");
+
+    match res {
+        Ok(contents) => {
+            toml::from_str::<Config>(&contents).expect("Unable to parse 'config.toml' contents.")
+        }
+        Err(_) => env_to_config(),
+    }
 });

src/main.rs

@@ -17,7 +17,7 @@ use ansi_term::Colour::RGB;
 use chrono::Local;
 use fern::Dispatch;
 use log::{debug, info};
-use std::{collections::HashMap, env::var, thread};
+use std::{collections::HashMap, env::var, net::SocketAddr, thread};
 use warp::Filter;
 
 mod config;
@@ -41,15 +41,24 @@ async fn main() -> Result<()> {
         .and(warp::body::json::<HashMap<String, serde_json::Value>>())
         .and_then(routes::sentry);
 
-    let port = config::CONFIG.port.clone();
+    let port = config::CONFIG.port.clone().unwrap_or(3939);
     let routes = warp::any()
         .and(index.or(sentry))
         .with(warp::log("sentry::http"));
 
-    warp::serve(routes)
-        .run(([0, 0, 0, 0], port.unwrap_or(3939)))
-        .await;
+    // Check if we can parse the config host to a SocketAddr.
+    let host = config::CONFIG.host.clone();
+    let addr = if let Some(h) = host {
+        format!("{}:{}", h, port)
+            .parse::<SocketAddr>()
+            .expect(format!("Unable to parse host '{}' -> SocketAddr.", h).as_str())
+    } else {
+        format!("{}:{}", "0.0.0.0", port.to_string())
+            .parse::<SocketAddr>()
+            .unwrap()
+    };
 
+    warp::serve(routes).run(addr).await;
     Ok(())
 }