(* Copyright (C) 2004-2005, HELM Team.
 * 
 * This file is part of HELM, an Hypertextual, Electronic
 * Library of Mathematics, developed at the Computer Science
 * Department, University of Bologna, Italy.
 * 
 * HELM is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * HELM is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HELM; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 * MA  02111-1307, USA.
 * 
 * For details, see the HELM World-Wide-Web page,
 * http://helm.cs.unibo.it/
 *)

open Printf

open MatitaTypes

(** {2 Initialization} *)

let arg_spec = [
(*   "-opt", Arg...., "set bla bla bla"; *)
]
let usage =
  sprintf "MatitaC v%s\nUsage: matitac [option ...] file\nOptions:"
    BuildTimeConf.version

let status = ref None

let run_script is eval_function  =
  let status = 
    match !status with
    | None -> assert false
    | Some s -> s
  in
  let slash_n_RE = Pcre.regexp "\\n" in
  let cb status stm = 
    (* dump_status status; *)
    let stm = TacticAstPp.pp_statement stm in
    let stm = Pcre.replace ~rex:slash_n_RE stm in
    let stm = 
      if String.length stm > 50 then
        String.sub stm 0 50 ^ " ..."
      else
        stm
    in
    MatitaLog.debug ("Executing: ``" ^ stm ^ "''")
  in
  try
    eval_function status is cb
  with
  | MatitaEngine.Drop  
  | CicTextualParser2.Parse_error _ as exn -> raise exn
  | exn -> 
      MatitaLog.error (MatitaExcPp.to_string exn);
      raise exn

let fname () =
  let acc = ref [] in
  let add_script fname = acc := fname :: !acc in
  Arg.parse arg_spec add_script usage;
  match !acc with
  | [x] -> x
  | _ -> prerr_endline usage; exit 1

let pp_ocaml_mode () = 
  MatitaLog.message "";
  MatitaLog.message "                      ** Entering Ocaml mode ** ";
  MatitaLog.message "";
  MatitaLog.message "Type 'go ();;' to enter an interactive matitac";
  MatitaLog.message ""
  
let clean_exit n =
 let opt_exit =
  function
     None -> ()
   | Some n -> exit n
 in
  match !status with
     None -> opt_exit n
   | Some status ->
      try
       let baseuri = MatitaTypes.get_string_option !status "baseuri" in
       MatitacleanLib.clean_baseuris ~verbose:false [baseuri];
       opt_exit n
      with MatitaTypes.Option_error("baseuri", "not found") ->
       (* no baseuri ==> nothing to clean yet *)
       opt_exit n
  
let rec interactive_loop () = 
 let str = Stream.of_channel stdin in
  try
    run_script str MatitaEngine.eval_from_stream_greedy
  with 
  | MatitaEngine.Drop -> pp_ocaml_mode ()
  | Sys.Break -> MatitaLog.error "user break!"; interactive_loop ()
  | MatitaTypes.Command_error _ -> interactive_loop ()
  | CicTextualParser2.Parse_error (floc,err) ->
     (* check for EOI *)
     if Stream.peek str = None then
      begin
       print_newline ();
       clean_exit (Some 0)
      end
     else
      let (x, y) = CicAst.loc_of_floc floc in
      MatitaLog.error (sprintf "Parse error at %d-%d: %s" x y err);
      interactive_loop ()
  | exn -> MatitaLog.error (Printexc.to_string exn); interactive_loop ()

let go () =
  Helm_registry.load_from "matita.conf.xml";
  Http_getter.init ();
  MetadataTypes.ownerize_tables (Helm_registry.get "matita.owner");
  status := Some (ref (Lazy.force MatitaEngine.initial_status));
  Sys.catch_break true;
  interactive_loop ()

let clean_exit n =
 match !status with
    None -> exit n
  | Some status ->
     let baseuri = MatitaTypes.get_string_option !status "baseuri" in
      MatitacleanLib.clean_baseuris ~verbose:false [baseuri];
      exit n
  
let dump_moo_to_file file moo =
 let os = open_out (MatitaMisc.obj_file_of_script file) in
 let output s = output_string os s in
 output "(* GENERATED FILE: DO NOT EDIT! *)\n\n";
 List.iter output (List.rev moo);
 close_out os
  
let main ~mode = 
  Helm_registry.load_from "matita.conf.xml";
  Http_getter.init ();
  MetadataTypes.ownerize_tables (Helm_registry.get "matita.owner");
  status := Some (ref (Lazy.force MatitaEngine.initial_status));
  Sys.catch_break true;
  let fname = fname () in
  try
    let time = Unix.time () in
    MatitaLog.message (sprintf "execution of %s started:" fname);
    let is =
      Stream.of_channel
        (match fname with
        | "stdin" -> stdin
        | fname -> open_in fname)
    in
    run_script is MatitaEngine.eval_from_stream;
    let elapsed = Unix.time () -. time in
    let tm = Unix.gmtime elapsed in
    let sec = 
      if tm.Unix.tm_sec > 0 then  (string_of_int tm.Unix.tm_sec ^ "''") else "" 
    in
    let min = 
      if tm.Unix.tm_min > 0 then  (string_of_int tm.Unix.tm_min ^ "' ") else "" 
    in
    let hou = 
      if tm.Unix.tm_hour > 0 then (string_of_int tm.Unix.tm_hour ^ "h ") else "" 
    in
    let proof_status,moo_content_rev = 
      match !status with
      | Some s -> !s.proof_status, !s.moo_content_rev
      | None -> assert false
    in
    if proof_status <> MatitaTypes.No_proof then
     begin
      MatitaLog.error
       "there are still incomplete proofs at the end of the script";
      clean_exit (Some 2)
     end
    else
     begin
       dump_moo_to_file fname moo_content_rev;
       MatitaLog.message 
         (sprintf "execution of %s completed in %s." fname (hou^min^sec));
       exit 0
     end
  with 
  | Sys.Break ->
      MatitaLog.error "user break!";
      if mode = `COMPILER then
        clean_exit (Some ~-1)
      else
        pp_ocaml_mode ()
  | MatitaEngine.Drop ->
      if mode = `COMPILER then 
        clean_exit (Some 1)
      else 
        pp_ocaml_mode ()
  | CicTextualParser2.Parse_error (floc,err) ->
     let (x, y) = CicAst.loc_of_floc floc in
     MatitaLog.error (sprintf "Parse error at %d-%d: %s" x y err);
     if mode = `COMPILER then
       clean_exit (Some 1)
     else
       pp_ocaml_mode ()
  | _ ->
     if mode = `COMPILER then 
       clean_exit (Some 3)
     else 
       pp_ocaml_mode ()