X-Git-Url: http://git.annexia.org/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain.ml;h=7e1c2f0e8b8366783d48e8d6c5addf8fa9ae5cad;hb=156de1e0df4aace1a42957491118cc2174d70c6a;hp=df4fb43a07c41b0d3594e96b81629f4d9cc9d33b;hpb=a48b05d35f0646322e8178ff10f8ed7af3e739aa;p=goals.git diff --git a/src/main.ml b/src/main.ml index df4fb43..7e1c2f0 100644 --- a/src/main.ml +++ b/src/main.ml @@ -19,24 +19,59 @@ open Printf +open Utils + (* See comment in parser.mly. *) let () = Parser.lexer_read := Some Lexer.read; Parser.eval_substitute := Some Eval.substitute let main () = + (* Handle the command line. *) + let anon_vars, targets = Cmdline.parse () in + (* Change directory (-C option). *) - Sys.chdir Cmdline.directory; + Sys.chdir (Cmdline.directory ()); + + (* Create a temporary directory which is always cleaned up at exit. *) + let tmpdir = + let temp_dir = try Unix.getenv "TMPDIR" with Not_found -> "/var/tmp" in + let t = Filename.temp_file ~temp_dir "goals" ".d" in + Unix.unlink t; + Unix.mkdir t 0o700; + at_exit ( + fun () -> + let cmd = sprintf "rm -rf %s" (Filename.quote t) in + ignore (Sys.command cmd) + ); + t in + + (* 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 "tmpdir" (Ast.EConstant (Ast.noloc, Ast.CString tmpdir)) env 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 *) (* Parse the prelude. *) let env = - if Cmdline.use_prelude then - Parse.parse_goalfile Ast.Env.empty Cmdline.prelude_gl_file - else - Ast.Env.empty in + if Cmdline.use_prelude () then + Parse.parse_goalfile env Cmdline.prelude_gl_file + else env in (* Parse the input file. *) - let env = Parse.parse_goalfile env Cmdline.input_file in + let env = Parse.parse_goalfile env (Cmdline.input_file ()) in (* Parse the command line assignments. *) let env = @@ -44,24 +79,33 @@ let main () = fun env (name, expr) -> let expr = Parse.parse_expr "commandline" expr in Ast.Env.add name expr env - ) env Cmdline.anon_vars in + ) env anon_vars in (* Parse the target expressions. *) - let targets = List.map (Parse.parse_expr "commandline") Cmdline.targets in + let targets = List.map (Parse.parse_expr "commandline") targets in (* If no target was set on the command line, use "all ()". *) let targets = if targets <> [] then targets else [Ast.ECall (Ast.noloc, "all", [])] in - if Cmdline.debug_flag then + if Cmdline.debug_flag () then Ast.print_env stderr env; - (* Run the target expressions. *) - Run.run_targets env targets + (* Construct the dependency DAG with the root(s) being the targets. *) + let dag = Deps.create env targets in + + (* 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 fail_job job = Deps.fail_job state job in + let string_of_job job = Deps.string_of_job job in + Jobs.run next_job retire_job fail_job string_of_job let () = try main () with - Failure msg | Sys_error msg -> - prerr_endline ("error: " ^ msg); exit 1 + | Failure msg | Sys_error msg -> + prerr_endline ("*** error: " ^ msg); + exit 1