X-Git-Url: http://git.annexia.org/?a=blobdiff_plain;f=src%2Fmain.ml;h=7e1c2f0e8b8366783d48e8d6c5addf8fa9ae5cad;hb=98b795ddf06271fa8018edcd0bd15960871828fd;hp=2e7ac44a0cc0b08a01fde2dd7b5834c5e9bee0a2;hpb=976bb1b35d77f3058df3c25c1e2a4767147d606b;p=goals.git diff --git a/src/main.ml b/src/main.ml index 2e7ac44..7e1c2f0 100644 --- a/src/main.ml +++ b/src/main.ml @@ -19,44 +19,93 @@ 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_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 = List.fold_left ( fun env (name, expr) -> - let expr = Parse.parse_cli_expr expr in + 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_cli_expr 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