From 55a773304461a2c261823c6d4bab38b66e95dc07 Mon Sep 17 00:00:00 2001
From: Kjetil Orbekk <kjetil.orbekk@gmail.com>
Date: Sat, 17 Jun 2017 10:17:31 -0400
Subject: Add port number flag.

---
 src/bin/main.rs | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

(limited to 'src')

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);
 }
-- 
cgit v1.2.3