Merge pull request #1 from m-misiura/ip_address_parsing

🚧 removing the bind address being hardcoded to `127.0.0.1`

Author: resoluteCoder

src/main.rs

@@ -1,14 +1,11 @@
-use std::{env, net::SocketAddr};
-
+use std::{env, net::{SocketAddr, IpAddr}};
 use axum::{
     routing::{get, post},
     Router,
 };
 use tower_http::trace::{self, TraceLayer};
 use tracing::Level;
-
 mod detectors;
-
 use detectors::handle_text_contents;
 
 #[tokio::main]
@@ -17,7 +14,8 @@ async fn main() {
         .with_target(false)
         .compact()
         .init();
-
+    
+    // Get port from environment variable or use default
     let mut http_port = 8080;
     if let Ok(port) = env::var("HTTP_PORT") {
         match port.parse::<u16>() {
@@ -26,6 +24,9 @@ async fn main() {
         }
     }
 
+    // Get host from environment variable or use default
+    let host = env::var("HOST").unwrap_or_else(|_| "0.0.0.0".to_string());
+    
     let app = Router::new()
         .route("/health", get(|| async { "Hello, World!" }))
         .route("/api/v1/text/contents", post(handle_text_contents))
@@ -35,9 +36,10 @@ async fn main() {
                 .on_response(trace::DefaultOnResponse::new().level(Level::INFO)),
         );
 
-    let addr = SocketAddr::from(([127, 0, 0, 1], http_port));
+    let ip: IpAddr = host.parse().expect("Failed to parse host IP address");
+    let addr = SocketAddr::from((ip, http_port));
+    
     tracing::info!("listening on {}", addr);
-
     let listener = tokio::net::TcpListener::bind(addr).await.unwrap();
     axum::serve(listener, app).await.unwrap();
-}
+}