let debug = false
let ignore_exc = false

let indent = ref 0;;

let logger =
    let do_indent () = String.make !indent ' ' in  
    (function 
    | `Start_type_checking s -> ();
        if debug then
         prerr_endline (do_indent () ^ "Start: " ^ NUri.string_of_uri s); 
        incr indent

    | `Type_checking_completed s ->  ();
        decr indent;
        if debug then
         prerr_endline (do_indent () ^ "End: " ^ NUri.string_of_uri s))
;;

let _ =
  CicParser.impredicative_set := false;
  NCicTypeChecker.set_logger logger;
  NCicPp.set_ppterm NCicPp.trivial_pp_term;
  Helm_registry.load_from "conf.xml";
  let alluris = 
    try
      let s = Sys.argv.(1) in
      if s = "-alluris" then
       begin
        let uri_re = Str.regexp ".*\\(ind\\|con\\)$" in
        let uris = Http_getter.getalluris () in
        let alluris = List.filter (fun u -> Str.string_match uri_re u 0) uris in
        let oc = open_out "alluris.txt" in
        List.iter (fun s -> output_string oc (s^"\n")) alluris;
        close_out oc; 
        []
       end
      else [s]
    with Invalid_argument _ -> 
      let r = ref [] in
      let ic = open_in "alluris.txt" in
      try while true do r := input_line ic :: !r; done; []
      with _ -> List.rev !r
  in
  let alluris = 
    HExtlib.filter_map
      (fun u -> try Some (UriManager.uri_of_string u) with _ -> None) alluris 
  in
  (* brutal *)
  prerr_endline "computing graphs to load...";
  let dbd = HSql.quick_connect (LibraryDb.parse_dbd_conf ()) in
  MetadataTypes.ownerize_tables (Helm_registry.get "matita.owner");
  let uniq l = 
     HExtlib.list_uniq (List.sort UriManager.compare l)
  in
  let who_uses u = 
    uniq (List.map (fun (uri,_) -> UriManager.strip_xpointer uri)
     (MetadataDeps.inverse_deps ~dbd u)) in
  let roots_alluris = 
   let rec fix acc l = 
    let acc, todo = 
     List.fold_left (fun (acc,todo) x ->
        let w = who_uses x in
        if w = [] then (x::acc,todo) else (acc,uniq (todo@w)))
     (acc,[]) l
    in
    if todo = [] then uniq acc else fix acc todo
   in
   (fix [] alluris)
  in
(*
 BARO!
  let roots_alluris = alluris in *)
  prerr_endline "generating Coq graphs...";
  (* per barare *)
   let roots_alluris = alluris in
  (* /per barare *)
  CicEnvironment.set_trust (fun _ -> false);
  List.iter (fun u ->
          prerr_endline (" - " ^ UriManager.string_of_uri u);
          try
            ignore(CicTypeChecker.typecheck u);
          with 
          | CicTypeChecker.AssertFailure s
          | CicTypeChecker.TypeCheckerFailure s -> prerr_endline (Lazy.force s)
    ) roots_alluris;
  prerr_endline "loading...";
  List.iter 
    (fun u -> 
       prerr_endline ("  - "^UriManager.string_of_uri u);
       try NCicEnvironment.load_graph u with exn -> ())
    roots_alluris;
  prerr_endline "finished....";
  CicUniv.do_rank (NCicEnvironment.get_graph ());
  prerr_endline "caching objects";
  prerr_endline "ranked....";
  HExtlib.profiling_enabled := false;
  List.iter (fun uu ->
     let uu= NUri.nuri_of_ouri uu in
     indent := 0;
     logger (`Start_type_checking uu);
    let _,o = NCicEnvironment.get_obj uu in
    try 
      NCicTypeChecker.typecheck_obj o;
      logger (`Type_checking_completed uu);
    with 
    | NCicTypeChecker.AssertFailure s 
    | NCicTypeChecker.TypeCheckerFailure s as e -> 
       prerr_endline ("######### " ^ Lazy.force s);
       if not ignore_exc then raise e
    | CicEnvironment.Object_not_found s -> 
       prerr_endline ("Obj not found: " ^ UriManager.string_of_uri s)
    )
    alluris;
  NCicEnvironment.invalidate ();
  Gc.compact ();
  HExtlib.profiling_enabled := true;
  NCicTypeChecker.set_logger (fun _ -> ());
  prerr_endline "typechecking, first with the new and then with the old kernel";
  let prima = Unix.gettimeofday () in
  List.iter 
    (fun u ->
       let u= NUri.nuri_of_ouri u in
      indent := 0;
      NCicTypeChecker.typecheck_obj (snd (NCicEnvironment.get_obj u)))
    alluris;
  let dopo = Unix.gettimeofday () in
  Gc.compact ();
  let dopo2 = Unix.gettimeofday () in
  Printf.eprintf "NEW typing: %3.2f, gc: %3.2f\n%!" (dopo -. prima) (dopo2 -.  dopo);
  CicEnvironment.invalidate ();
  Gc.compact ();
  let prima = Unix.gettimeofday () in
  List.iter (fun u -> ignore (CicTypeChecker.typecheck u)) alluris;
  let dopo = Unix.gettimeofday () in
  Gc.compact ();
  let dopo2 = Unix.gettimeofday () in
  Printf.eprintf "OLD typing: %3.2f, gc: %3.2f\n%!" (dopo -. prima) (dopo2 -. dopo)
;;