Support compilation of multiple files

1 files changed, 3 insertions, 2 deletions

diff --git a/xacml-proofs/compiler/bin/main.ml b/xacml-proofs/compiler/bin/main.ml
index 3a2073e..946d65d 100644
--- a/xacml-proofs/compiler/bin/main.ml
+++ b/xacml-proofs/compiler/bin/main.ml
@@ -2,6 +2,7 @@ open Core
 open Compiler
 
 let () =
-  let file_name = Array.get (Sys.get_argv ()) 1 in
-  let policy_code = In_channel.read_all file_name in
+  let argv = Sys.get_argv () in
+  let file_names = Array.sub argv ~pos:1 ~len:(Array.length argv - 1) |> Array.to_list in
+  let policy_code = List.map file_names ~f:In_channel.read_all in
   compile' policy_code |> Stdio.print_endline