cli args

src/main.rs

@@ -2,6 +2,7 @@
 extern crate rocket;
 #[macro_use]
 extern crate lazy_static;
+use clap::Parser;
 use markov::Chain;
 use std::{
     fs::File,
@@ -10,6 +11,7 @@ use std::{
 
 lazy_static! {
     static ref CHAIN: Chain<String> = build_chain();
+    static ref ARGS: Args = Args::parse();
 }
 
 #[get("/")]
@@ -19,17 +21,32 @@ fn index() -> String {
 
 #[launch]
 fn rocket() -> _ {
-    rocket::build().mount("/", routes![index])
+    let mut config = rocket::Config::default();
+    config.port = ARGS.port;
+    rocket::custom(config).mount("/", routes![index])
+    //rocket::build().mount("/", routes![index])
 }
 
 fn build_chain() -> Chain<String> {
-    println!("Building Chain...");
+    println!("Building Chain from \"{}\"...", ARGS.input);
     let mut new_chain = Chain::new();
     // get input from input file
-    let file = File::open("./input").expect("No input file!");
+    let file = File::open(&ARGS.input).expect("No input file!");
     let reader = BufReader::new(file);
     for line in reader.lines() {
         new_chain.feed_str(&line.unwrap());
     }
     new_chain
 }
+
+#[derive(Parser, Debug)]
+#[clap(about, version, author)]
+struct Args {
+    // Input file to make markov chain from
+    #[clap(short, long)]
+    input: String,
+
+    // Port number for server
+    #[clap(short, long, default_value_t = 8000)]
+    port: u16,
+}