Split implementation into dependency analysis and traversal.

[goals.git] / src / main.ml

diff --git a/src/main.ml b/src/main.ml
index be1baff..ba4a195 100644 (file)
--- a/src/main.ml
+++ b/src/main.ml
@@ -21,59 +21,75 @@ open Printf
 
 open Utils
 
-let usage =
-  "\
-goals: Build software.
+(* See comment in parser.mly. *)
+let () =
+  Parser.lexer_read := Some Lexer.read;
+  Parser.eval_substitute := Some Eval.substitute
 
- goals [-f Goalfile] ['var = value' ...] ['target' ...]
+let main () =
+  (* Handle the command line. *)
+  let anon_vars, targets = Cmdline.parse () in
 
-For detailed help see goals(1).
+  (* Change directory (-C option). *)
+  Sys.chdir (Cmdline.directory ());
 
-Options:"
+  (* Create the initial environment, containing the system environment
+   * and a few other standard strings.
+   *)
+  let env =
+    Array.fold_left (
+      fun env environ ->
+        let k, v = split "=" environ in
+        Ast.Env.add k (Ast.EConstant (Ast.noloc, Ast.CString v)) env
+    ) Ast.Env.empty (Unix.environment ()) in
+  let env =
+    Ast.Env.add "stdlib"
+      (Ast.EConstant (Ast.noloc, Ast.CString Cmdline.stdlibdir))
+      env in
+  (*let env =
+    if Cmdline.debug_flag then Ast.Env.add "debug" (Ast.EConstant (noloc, Ast.CBool true)) env else env in *)
 
-let main () =
-  (* Command line arguments. *)
-  let filename = ref "Goalfile" in
+  (* Parse the prelude. *)
+  let env =
+    if Cmdline.use_prelude () then
+      Parse.parse_goalfile env Cmdline.prelude_gl_file
+    else env in
 
-  let argspec = [
-    "-f",        Arg.Set_string filename,
-                 "filename Set name of Goalfile";
-    "--file",    Arg.Set_string filename,
-                 "filename Set name of Goalfile";
-  ] in
-  let argspec = Arg.align argspec in
-  let args = ref [] in
-  let anon_fun s = args := s :: !args in
-  Arg.parse argspec anon_fun usage;
+  (* Parse the input file. *)
+  let env = Parse.parse_goalfile env (Cmdline.input_file ()) in
 
-  (*let args = List.rev !args in*)
-  let filename = !filename in
+  (* Parse the command line assignments. *)
+  let env =
+    List.fold_left (
+      fun env (name, expr) ->
+        let expr = Parse.parse_expr "commandline" expr in
+        Ast.Env.add name expr env
+    ) env anon_vars in
 
-  (* Parse the input file. *)
-  let file = Parse.parse_goalfile filename in
+  (* Parse the target expressions. *)
+  let targets = List.map (Parse.parse_expr "commandline") targets in
 
-  Ast.print_file stdout file;
+  (* If no target was set on the command line, use "all ()". *)
+  let targets =
+    if targets <> [] then targets
+    else [Ast.ECall (Ast.noloc, "all", [])] in
 
-  (* Find the target(s) to execute first. *)
-  let initial_targets = ref [] in
-  (* XXX Parse command line anon args here. XXX *)
+  if Cmdline.debug_flag () then
+    Ast.print_env stderr env;
 
-  (* If no initial target set on the command line, find
-   * the first goal in the file.
-   *)
-  List.iter (
-    function
-    | Ast.Goal (name, [], _, _, _) ->
-       if !initial_targets = [] then
-         initial_targets := Ast.ECall (name, []) :: !initial_targets
-    | Ast.Goal (name, _, _, _, _) ->
-       if !initial_targets = [] then
-         failwithf "%s: first target ‘%s’ has parameters and so cannot be used as the default target"
-           filename name
-    | _ -> ()
-  ) file;
+  (* Construct the dependency DAG with the root(s) being the targets. *)
+  let dag = Deps.create env targets in
 
-  let initial_targets = List.rev !initial_targets in
-  ignore initial_targets
+  (* Run the jobs. *)
+  let state = Deps.new_state dag Run.goal_runner Run.exists_runner in
+  let next_job () = Deps.next_job state in
+  let retire_job job = Deps.retire_job state job in
+  let string_of_job job = Deps.string_of_job job in
+  Jobs.run next_job retire_job string_of_job
 
-let () = main ()
+let () =
+  try main ()
+  with
+  | Failure msg | Sys_error msg ->
+     prerr_endline ("*** error: " ^ msg);
+     exit 1