1 files changed, 19 insertions, 0 deletions

diff --git a/main.ml b/main.ml
index 1bb6704..2f8d57c 100644
--- a/main.ml
+++ b/main.ml
@@ -9,3 +9,22 @@ open Instptr
 open Reg
 open Urm
 
+let exec_with_resource func filename =
+  let file = open_in filename in
+  let res = func file in
+  close_in file; res
+
+let read_prgm = exec_with_resource (fun f -> string_of_file f |> program_of_string)
+let read_regs = exec_with_resource (fun f -> string_of_file f |> regs_of_string)
+let run run_func prgm regs = urm_mk prgm regs |> run_func |> regs_string |> print_endline
+
+let run_mode_of_string = function
+  | "run" -> urm_run
+  | "trace" -> urm_run_trace
+  | _ -> failwith "Invalid run mode"
+
+let () = match Sys.argv with
+  | [| _; "run-tests" |] -> () (* handled in test files *)
+  | [| _; mode; prgm; regs |] -> run (run_mode_of_string mode) (read_prgm prgm) (read_regs regs)
+  | _ -> print_endline "Usage: urm <run-tests | run <prgmfile> <regfile> | trace <prgmfile> <regfile>>"
+