feat: server cli args

Cargo.toml

@@ -13,6 +13,7 @@ path = "src/main.rs"
 [dependencies]
 anyhow = "1.0.65"
 axum = "0.5.16"
+clap = { version = "4.0.10", features = ["derive"] }
 hyper = { version = "0.14.20", features = ["full"] }
 serde = { version = "1.0.145", features = ["serde_derive"] }
 tokio = { version = "1.21.2", features = ["full"] }

README.md

@@ -1,7 +1,8 @@
 # Frontman
 
 ## TODO
-- [ ] Read/parse proxy config from file?
+- [x] Read/parse proxy config from file?
+- [x] Server CLI args (config file path, listening port etc)
 - [ ] Load balance requests to 1 or more origin servers (of the same backend), supporting IP/DNS
 - [ ] Anyhow for result?
 - [ ] proper error responses

src/args.rs

@@ -0,0 +1,11 @@
+use clap::Parser;
+
+#[derive(Parser, Debug)]
+#[command(author, version, about, long_about = None)]
+pub struct ServerArgs {
+    #[arg(short, long, default_value_t = String::from("frontman.toml"))]
+    pub config_path: String,
+
+    #[arg(short, long, default_value_t = 3000)]
+    pub port: u16,
+}

src/main.rs

@@ -2,23 +2,32 @@
 use std::net::SocketAddr;
 
 use axum::{routing::any, Router};
+use clap::Parser;
 use tower_http::trace::TraceLayer;
 use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
 
+mod args;
 mod config;
 mod proxy;
 
 #[tokio::main]
 async fn main() {
+    let args = args::ServerArgs::parse();
+
     tracing_subscriber::registry()
         .with(tracing_subscriber::EnvFilter::new(
             std::env::var("RUST_LOG").unwrap_or_else(|_| "frontman=debug,tower_http=debug".into()),
         ))
         .with(tracing_subscriber::fmt::layer())
         .init();
 
-    let config_file =
-        std::fs::read_to_string("frontman.toml").expect("Should have been able to read the file");
+    let config_file = match std::fs::read_to_string(&args.config_path) {
+        Ok(file) => file,
+        Err(error) => panic!(
+            "Couldn't read config file at {}: {:?}",
+            args.config_path, error
+        ),
+    };
     let config: config::Config = toml::from_str(&config_file).unwrap();
 
     tracing::debug!("Starting server with configuration: {:?}", config);
@@ -34,7 +43,7 @@ async fn main() {
         .route("/*path", any(handler))
         .layer(TraceLayer::new_for_http());
 
-    let addr = SocketAddr::from(([0, 0, 0, 0], 3000));
+    let addr = SocketAddr::from(([0, 0, 0, 0], args.port));
     tracing::debug!("Server listening on {}", addr);
     axum::Server::bind(&addr)
         .serve(app.into_make_service())