start adding option handling for webface

1 files changed, 40 insertions, 2 deletions

diff --git a/src/bin/mt-webface.rs b/src/bin/mt-webface.rs
index 0940ea0..6809f57 100644
--- a/src/bin/mt-webface.rs
+++ b/src/bin/mt-webface.rs
@@ -1,9 +1,13 @@
 
+extern crate getopts;
 extern crate pencil;
 #[macro_use] extern crate log;
 extern crate env_logger;
 
+use std::env;
 use std::collections::BTreeMap;
+use std::process::exit;
+use getopts::Options;
 use pencil::Pencil;
 use pencil::{Request, PencilResult, Response, HTTPError};
 
@@ -43,8 +47,40 @@ fn server_error(_: HTTPError) -> PencilResult {
     Ok(response)
 }
 
+fn print_usage(opts: Options) {
+    let brief = "usage:\tmt-webface [options]";
+    println!("");
+    print!("{}", opts.usage(&brief));
+}
+
 fn main() {
 
+    let args: Vec<String> = env::args().collect();
+
+    let mut opts = Options::new();
+    opts.optflag("h", "help", "print this help menu");
+    opts.optflag("b", "bind", "local IP:port to bind to");
+    opts.optflag("", "version", "print the version");
+
+    let matches = match opts.parse(&args[1..]) {
+        Ok(m) => m,
+        Err(f) => {
+            println!("{}\n", f.to_string());
+            print_usage(opts);
+            exit(-1);
+        }
+    };
+
+    if matches.opt_present("help") {
+        print_usage(opts);
+        return;
+    }
+
+    if matches.opt_present("version") {
+        println!("modelthing {}", env!("CARGO_PKG_VERSION"));
+        return;
+    }
+
     env_logger::init().unwrap();
 
     let mut app = Pencil::new("webface");
@@ -65,6 +101,8 @@ fn main() {
     app.register_template("model_view.html");
     app.get("/model/<model_id:string>", "model", model_view);
 
-    info!("Running on http://localhost:5000/");
-    app.run("127.0.0.1:5000");
+    let bind = matches.opt_str("bind").unwrap_or("127.0.0.1:5000".to_string());
+    let bind_str: &str = &bind;
+    info!("Running on {}", bind);
+    app.run(bind_str);
 }