Add port number flag.

1 files changed, 15 insertions, 2 deletions

diff --git a/src/bin/main.rs b/src/bin/main.rs
index b54c4c0..94bdd71 100644
--- a/src/bin/main.rs
+++ b/src/bin/main.rs
@@ -4,9 +4,22 @@
 extern crate log;
 extern crate systemhttp;
 extern crate env_logger;
+#[macro_use]
+extern crate clap;
 
 fn main() {
+    let matches = clap_app!(
+        systemhttpd =>
+            (version: "0.1")
+            (author: "Kjetil Ørbekk")
+            (about: "A systemd web frontend")
+            (@arg PORT: -p --port +takes_value "Port to serve on"))
+        .get_matches();
+
+    let port = matches.value_of("PORT").unwrap_or("8080")
+        .parse::<u16>().expect("port number");
+
     env_logger::init().unwrap();
-    let _server = systemhttp::server::serve(8080).unwrap();
-    println!("Serving on {}", 8080);
+    let _server = systemhttp::server::serve(port).unwrap();
+    println!("Serving on {}", port);
 }