Merge pull request #60 from ramonasuncion/end-to-end-comp

Add default compile mode to ripec

Author: ramonasuncion

compiler/bin/main.ml

@@ -41,6 +41,26 @@ let parse_file filename =
         else
           let il = Ripe.Codegen.emit_qbe tdecls in
           if !emit_qbe then print_string il
+          else
+            let dir = Filename.dirname !file in
+            let base =
+              Filename.concat dir
+                (Filename.remove_extension (Filename.basename !file))
+            in
+            let tmp_qbe = Filename.temp_file "ripe" ".ssa" in
+            let tmp_asm = Filename.temp_file "ripe" ".s" in
+            let oc = open_out tmp_qbe in
+            output_string oc il;
+            close_out oc;
+            let run cmd =
+              if Sys.command cmd <> 0 then (
+                Printf.eprintf "ripec: command failed: %s\n" cmd;
+                exit 1)
+            in
+            run (Printf.sprintf "qbe -o %s %s" tmp_asm tmp_qbe);
+            run (Printf.sprintf "cc -o %s %s" base tmp_asm);
+            Sys.remove tmp_qbe;
+            Sys.remove tmp_asm
     | exception Ripe.Typechecker.TypeError msg ->
         Printf.eprintf "typecheck error: %s\n" msg;
         exit 1

compiler/lib/codegen.ml

@@ -314,9 +314,10 @@ let emit_func (ctx : ctx) (tfd : T.tfunc_def) =
   (* FIXME: main(): with a trailing colon and no return type syntax error *)
   (* TODO: Create a custom _start. *)
   let is_main = tfd.name = "main" && tfd.ret_ty = TInt I32 in
+  let export_part = if is_main then "export " else "" in
   let ret_part = match tfd.ret_ty with TVoid -> "" | t -> qbe_ty t ^ " " in
   (* TODO: Add export for pub(?) *)
-  emit ctx "function %s$%s(%s) {\n" ret_part tfd.name
+  emit ctx "%sfunction %s$%s(%s) {\n" export_part ret_part tfd.name
     (String.concat ", " params_strs);
   emit ctx "@start\n";