main.rs: Add port command line option

Provide users with an option to change the default port the program
listens on.

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index c6e7054..098a8bb 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -45,6 +45,7 @@ fn main() {
 
     let mut opts = Options::new();
     opts.optopt("b", "browser", "set browser", "BROWSER");
+    opts.optopt("p", "port", "set port number", "PORT");
     opts.optflag("h", "help", "print this help menu");
 
     let opt_matches = match opts.parse(&args[1..]) {
@@ -65,6 +66,11 @@ fn main() {
         },
     };
 
+    let port: u16 = match opt_matches.opt_str("p") {
+        Some(p) => p.parse().unwrap(),
+        None => 37705,
+    };
+
     open_stream().unwrap_or_else(|e| {
         writeln!(io::stderr(), "{}", e)
             .expect("Failed printing to stderr");