1 files changed, 11 insertions, 5 deletions
diff --git a/src/main.rs b/src/main.rs
index 88fa2ab..c3431cd 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,10 +1,16 @@
+mod cli;
 mod dice;
 
 fn main() {
-    const TEST_FORMULA : &str = "3d6";
-    println!("Rolling {}:", TEST_FORMULA);
-    match dice::roll_formula(TEST_FORMULA) {
-        Some(result) => println!("{}", result),
-        None => eprintln!("Error: Invalid roll formula, or die roll otherwise failed.")
+    let matches = cli::cli().get_matches();
+
+    match matches.subcommand() {
+        Some(("roll", sub_matches)) => {
+            match dice::roll_formula(sub_matches.get_one::<String>("FORMULA").expect("required")) {
+                Some(result) => println!("Rolled: {}", result),
+                None => eprintln!("Error: Invalid roll formula or calculation failed.")
+            }
+        }
+        _ => unreachable!()
     }
 }