Add clap to the main server with --host flag

Author: ekzhang

Cargo.lock

@@ -6,6 +6,7 @@ edition = "2021"
 [dependencies]
 anyhow = "1.0.53"
 axum = "0.4.5"
+clap = { version = "3.1.6", features = ["derive"] }
 hyper = { version = "0.14.17", features = ["full"] }
 sshx-core = { path = "../sshx-core" }
 tokio = { version = "1.16.1", features = ["full"] }

crates/sshx-server/Cargo.toml

@@ -1,12 +1,30 @@
+use std::net::SocketAddr;
+
 use anyhow::Result;
+use clap::Parser;
 use sshx_server::make_server_bind;
 use tokio::signal::unix::{signal, SignalKind};
 
+/// The sshx server CLI interface.
+#[derive(Parser, Debug)]
+#[clap(author, version, about, long_about = None)]
+struct Args {
+    /// Specify port to listen on.
+    #[clap(long, default_value_t = 8051)]
+    port: u16,
+
+    /// Whether to expose the server on all network interfaces.
+    #[clap(long)]
+    host: bool,
+}
+
 #[tokio::main]
 async fn main() -> Result<()> {
     tracing_subscriber::fmt::init();
 
-    let addr = "[::1]:8051".parse()?;
+    let args = Args::parse();
+    let host = if args.host { "::" } else { "::1" };
+    let addr = SocketAddr::new(host.parse()?, args.port);
 
     let mut sigterm = signal(SignalKind::terminate())?;
     let mut sigint = signal(SignalKind::interrupt())?;