X-Git-Url: http://git.annexia.org/?a=blobdiff_plain;f=src%2Fmain.ml;h=2b74a4f42ca2e5226b81b55c8c9a487621797dae;hb=881b2e9b7bad0da8f44418e9e6558710db5ce690;hp=c23b70aa641afcef8fed1e9c1102490c9b2e4139;hpb=315ceff80b09b76959a067fa2f3288f22d2e55da;p=goals.git

diff --git a/src/main.ml b/src/main.ml
index c23b70a..2b74a4f 100644
--- a/src/main.ml
+++ b/src/main.ml
@@ -21,100 +21,64 @@ open Printf
 
 open Utils
 
-(* See also "let id" in [lexer.mll]. *)
-let var_regexp =
-  Str.regexp "\\([a-zA-Z_][a-zA-Z0-9_]*\\)[ \t]*=[ \t]*\\(.*\\)"
-
-let usage =
-  "\
-goals: Build software.
-
- goals [-f Goalfile] ['var=value' ...] ['target' ...]
-
-For detailed help see goals(1).
-
-Options:"
-
-let print_version () =
-  printf "%s %s\n" Config.package_name Config.package_version;
-  exit 0
+(* See comment in parser.mly. *)
+let () =
+  Parser.lexer_read := Some Lexer.read;
+  Parser.eval_substitute := Some Eval.substitute
 
 let main () =
-  (* Get stdlib directory. *)
-  let datadir =
-    try Sys.getenv "GOALS_DATADIR" with Not_found -> Config.datadir in
-  let stdlibdir = datadir // "stdlib" in
-  let prelude_gl = stdlibdir // "prelude.gl" in
-  if not (is_directory stdlibdir) || not (Sys.file_exists prelude_gl) then
-    failwithf "%s: cannot find the standard library directory, expected %s.  If the standard library directory is in a non-standard location then set GOALS_DATADIR.  If you can trying to run goals from the build directory then use ‘./run goals ...’"
-      Sys.executable_name stdlibdir;
-
-  (* 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";
-    "-V",          Arg.Unit print_version,
-                   " Print version and exit";
-    "--version",   Arg.Unit print_version,
-                   " Print version and exit";
-  ] in
-  let argspec = Arg.align argspec in
-  let anon_fun s = args := s :: !args in
-  Arg.parse argspec anon_fun usage;
+  (* Change directory (-C option). *)
+  Sys.chdir Cmdline.directory;
 
-  let args = List.rev !args in
-  let directory = !directory in
-  let filename = !filename 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 "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
 
   (* Parse the input file. *)
-  let env = Parse.parse_goalfile filename in
+  let env = Parse.parse_goalfile env Cmdline.input_file in
 
-  (* Now we've read the input, change directory. *)
-  Sys.chdir directory;
+  (* 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 Cmdline.anon_vars in
 
-  (* Parse the command line anon args.  Each parameter has the
-   * form "name=<expr>" to assign a value to a variable, or
-   * "<expr>" to indicate a target to build.
-   *)
-  let targets = ref [] in
-  let env = ref env in
-  List.iter (
-    fun arg ->
-      if Str.string_match var_regexp arg 0 then (
-        (* assignment *)
-        let name = Str.matched_group 1 arg in
-        let expr = Parse.parse_cli_expr (Str.matched_group 2 arg) in
-        env := Ast.Env.add name expr !env
-      )
-      else (
-        (* target *)
-        let expr = Parse.parse_cli_expr arg in
-        targets := expr :: !targets
-      )
-  ) args;
-  let targets = List.rev !targets and env = !env in
+  (* Parse the target expressions. *)
+  let targets = List.map (Parse.parse_expr "commandline") Cmdline.targets in
 
   (* If no target was set on the command line, use "all ()". *)
   let targets =
     if targets <> [] then targets
-    else [Ast.ECallGoal (Ast.noloc, "all", [])] in
+    else [Ast.ECall (Ast.noloc, "all", [])] in
 
-  Ast.print_env stderr env;
+  if Cmdline.debug_flag then
+    Ast.print_env stderr env;
 
-  (* Evaluate the target expressions in turn. *)
-  Eval.evaluate_targets env targets
+  (* Run the target expressions. *)
+  Run.run_targets env targets
 
 let () =
   try main ()
   with
-    Failure msg -> prerr_endline msg; exit 1
+    Failure msg | Sys_error msg ->
+      prerr_endline ("error: " ^ msg); exit 1