Add State Generator

src/main.rs

@@ -9,26 +9,41 @@ use pest::Parser;
 mod asm;
 mod generate;
 mod parse;
+mod simulate;
 
 use generate::generate_binary;
 use parse::{parse_asm, AsmParser};
 
 fn main() {
-    let file_name = env::args().nth(1);
+    let sub_cmd = env::args().nth(1);
+    let file_name = env::args().nth(2);
+    let steps = env::args().nth(3);
 
-    let file_content = match file_name {
-        Some(file_name) => {
-            fs::read_to_string(file_name).expect("Could not read the provided asm file")
-        }
-        None => {
-            println!("No input file was provided");
-            return;
-        }
-    };
+    if let Some(sub_cmd) = sub_cmd {
+        let file_content = match file_name {
+            Some(file_name) => {
+                fs::read_to_string(file_name).expect("Could not read the provided asm file")
+            }
+            None => {
+                println!("No input file was provided");
+                return;
+            }
+        };
+
+        let instructions = parse_asm(
+            AsmParser::parse(parse::Rule::program, &file_content).unwrap_or_else(|e| panic!("{}", e)),
+        );
+
+        if sub_cmd == "generate" {
+            let binary = generate_binary(instructions);
+            println!("{}", binary);
+        }
+        else if sub_cmd == "simulate" {
+            if let Some(steps) = steps {
+                let states = simulate::simulate(instructions, steps.parse::<usize>().unwrap());
+                println!("{:#?}", states);
+            }
+        }
+    }
 
-    let instructions = parse_asm(
-        AsmParser::parse(parse::Rule::program, &file_content).unwrap_or_else(|e| panic!("{}", e)),
-    );
-    let binary = generate_binary(instructions);
-    println!("{}", binary);
 }