- (* Command line arguments. *)
- let args = ref [] in
- let directory = ref "." in
- let filename = ref "Goalfile" in
-
- let argspec = [
- "-C", Arg.Set_string directory,
- "directory Change to directory before running";
- "--directory", Arg.Set_string directory,
- "directory Change to directory before running";
- "-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 anon_fun s = args := s :: !args in
- Arg.parse argspec anon_fun usage;
-
- let args = List.rev !args in
- let directory = !directory in
- let filename = !filename in
+ (* Handle the command line. *)
+ let anon_vars, targets = Cmdline.parse () in
+
+ (* Change directory (-C option). *)
+ 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 env Cmdline.prelude_gl_file
+ else env in