helm/matita/matitacLib.ml

   1 (* Copyright (C) 2004-2005, HELM Team.
   2  *
   3  * This file is part of HELM, an Hypertextual, Electronic
   4  * Library of Mathematics, developed at the Computer Science
   5  * Department, University of Bologna, Italy.
   6  *
   7  * HELM is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation; either version 2
  10  * of the License, or (at your option) any later version.
  11  *
  12  * HELM is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with HELM; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  20  * MA  02111-1307, USA.
  21  *
  22  * For details, see the HELM World-Wide-Web page,
  23  * http://helm.cs.unibo.it/
  24  *)
  25
  26 open Printf
  27
  28 open MatitaTypes
  29
  30 (** {2 Initialization} *)
  31
  32 let arg_spec = [
  33 (*   "-opt", Arg...., "set bla bla bla"; *)
  34 ]
  35 let usage =
  36   sprintf "MatitaC v%s\nUsage: matitac [option ...] file\nOptions:"
  37     BuildTimeConf.version
  38
  39 let status = ref None
  40
  41 let run_script is eval_function  =
  42   let status =
  43     match !status with
  44     | None -> assert false
  45     | Some s -> s
  46   in
  47   let slash_n_RE = Pcre.regexp "\\n" in
  48   let cb status stm =
  49     (* dump_status status; *)
  50     let stm = TacticAstPp.pp_statement stm in
  51     let stm = Pcre.replace ~rex:slash_n_RE stm in
  52     let stm =
  53       if String.length stm > 50 then
  54         String.sub stm 0 50 ^ " ..."
  55       else
  56         stm
  57     in
  58     MatitaLog.debug ("Executing: ``" ^ stm ^ "''")
  59   in
  60   try
  61     eval_function status is cb
  62   with
  63   | MatitaEngine.Drop
  64   | CicTextualParser2.Parse_error _ as exn -> raise exn
  65   | exn ->
  66       MatitaLog.error (MatitaExcPp.to_string exn);
  67       raise exn
  68
  69 let fname () =
  70   let acc = ref [] in
  71   let add_script fname = acc := fname :: !acc in
  72   Arg.parse arg_spec add_script usage;
  73   match !acc with
  74   | [x] -> x
  75   | _ -> prerr_endline usage; exit 1
  76
  77 let pp_ocaml_mode () =
  78   MatitaLog.message "";
  79   MatitaLog.message "                      ** Entering Ocaml mode ** ";
  80   MatitaLog.message "";
  81   MatitaLog.message "Type 'go ();;' to enter an interactive matitac";
  82   MatitaLog.message ""
  83
  84 let clean_exit n =
  85  let opt_exit =
  86   function
  87      None -> ()
  88    | Some n -> exit n
  89  in
  90   match !status with
  91      None -> opt_exit n
  92    | Some status ->
  93       try
  94        let baseuri = MatitaTypes.get_string_option !status "baseuri" in
  95        MatitacleanLib.clean_baseuris ~verbose:false [baseuri];
  96        opt_exit n
  97       with MatitaTypes.Option_error("baseuri", "not found") ->
  98        (* no baseuri ==> nothing to clean yet *)
  99        opt_exit n
 100
 101 let rec interactive_loop () =
 102  let str = Stream.of_channel stdin in
 103   try
 104     run_script str MatitaEngine.eval_from_stream_greedy
 105   with
 106   | MatitaEngine.Drop -> pp_ocaml_mode ()
 107   | Sys.Break -> MatitaLog.error "user break!"; interactive_loop ()
 108   | MatitaTypes.Command_error _ -> interactive_loop ()
 109   | CicTextualParser2.Parse_error (floc,err) ->
 110      (* check for EOI *)
 111      if Stream.peek str = None then
 112       begin
 113        print_newline ();
 114        clean_exit (Some 0)
 115       end
 116      else
 117       let (x, y) = CicAst.loc_of_floc floc in
 118       MatitaLog.error (sprintf "Parse error at %d-%d: %s" x y err);
 119       interactive_loop ()
 120   | exn -> MatitaLog.error (Printexc.to_string exn); interactive_loop ()
 121
 122 let go () =
 123   Helm_registry.load_from "matita.conf.xml";
 124   Http_getter.init ();
 125   MetadataTypes.ownerize_tables (Helm_registry.get "matita.owner");
 126   status := Some (ref (Lazy.force MatitaEngine.initial_status));
 127   Sys.catch_break true;
 128   interactive_loop ()
 129
 130 let dump_moo_to_file file moo =
 131  let os = open_out (MatitaMisc.obj_file_of_script file) in
 132  let output s = output_string os s in
 133  output "(* GENERATED FILE: DO NOT EDIT! *)\n\n";
 134  List.iter output (List.rev moo);
 135  close_out os
 136
 137 let main ~mode =
 138   Helm_registry.load_from "matita.conf.xml";
 139   Http_getter.init ();
 140   MetadataTypes.ownerize_tables (Helm_registry.get "matita.owner");
 141   MatitaDb.create_owner_environment ();
 142   status := Some (ref (Lazy.force MatitaEngine.initial_status));
 143   Sys.catch_break true;
 144   let fname = fname () in
 145   try
 146     let time = Unix.time () in
 147     MatitaLog.message (sprintf "execution of %s started:" fname);
 148     let is =
 149       Stream.of_channel
 150         (match fname with
 151         | "stdin" -> stdin
 152         | fname -> open_in fname)
 153     in
 154     run_script is MatitaEngine.eval_from_stream;
 155     let elapsed = Unix.time () -. time in
 156     let tm = Unix.gmtime elapsed in
 157     let sec =
 158       if tm.Unix.tm_sec > 0 then  (string_of_int tm.Unix.tm_sec ^ "''") else ""
 159     in
 160     let min =
 161       if tm.Unix.tm_min > 0 then  (string_of_int tm.Unix.tm_min ^ "' ") else ""
 162     in
 163     let hou =
 164       if tm.Unix.tm_hour > 0 then (string_of_int tm.Unix.tm_hour ^ "h ") else ""
 165     in
 166     let proof_status,moo_content_rev =
 167       match !status with
 168       | Some s -> !s.proof_status, !s.moo_content_rev
 169       | None -> assert false
 170     in
 171     if proof_status <> MatitaTypes.No_proof then
 172      begin
 173       MatitaLog.error
 174        "there are still incomplete proofs at the end of the script";
 175       clean_exit (Some 2)
 176      end
 177     else
 178      begin
 179        dump_moo_to_file fname moo_content_rev;
 180        MatitaLog.message
 181          (sprintf "execution of %s completed in %s." fname (hou^min^sec));
 182        exit 0
 183      end
 184   with
 185   | Sys.Break ->
 186       MatitaLog.error "user break!";
 187       if mode = `COMPILER then
 188         clean_exit (Some ~-1)
 189       else
 190         pp_ocaml_mode ()
 191   | MatitaEngine.Drop ->
 192       if mode = `COMPILER then
 193         clean_exit (Some 1)
 194       else
 195         pp_ocaml_mode ()
 196   | CicTextualParser2.Parse_error (floc,err) ->
 197      let (x, y) = CicAst.loc_of_floc floc in
 198      MatitaLog.error (sprintf "Parse error at %d-%d: %s" x y err);
 199      if mode = `COMPILER then
 200        clean_exit (Some 1)
 201      else
 202        pp_ocaml_mode ()
 203   | _ ->
 204      if mode = `COMPILER then
 205        clean_exit (Some 3)
 206      else
 207        pp_ocaml_mode ()