(****************************************************************************)
(*                                                                          *)
(*                              Objective Caml                              *)
(*                                                                          *)
(*                            INRIA Rocquencourt                            *)
(*                                                                          *)
(*  Copyright  2006   Institut National de Recherche  en  Informatique et   *)
(*  en Automatique.  All rights reserved.  This file is distributed under   *)
(*  the terms of the GNU Library General Public License, with the special   *)
(*  exception on linking described in LICENSE at the top of the Objective   *)
(*  Caml source tree.                                                       *)
(*                                                                          *)
(****************************************************************************)

(* Authors:
 * - Nicolas Pouillard: initial version
 *)
open YaM
open Format

let getenv var default =
  try Sys.getenv var
  with Not_found ->
    default

let libdir_camlp4 = (getenv "LIBDIR" Camlp4_config.libdir) ^ "/camlp4/."

let bindir = (getenv "BINDIR" Camlp4_config.bindir) ^ "/."

let ocamlrun = "OCAMLRUNPARAM=l=1M ../boot/ocamlrun"

let ocaml = ocamlrun ^ " ../ocaml -I ../stdlib -I ../otherlibs/unix "

let debug_mode =
  (* true *)
  false

let camlp4_modules =
  [
    ocamlrun;
    "./boot/camlp4boot";
  ]
let camlp4_modules =
  if debug_mode then "env STATIC_CAMLP4_DEBUG=\\*" :: camlp4_modules
  else camlp4_modules

let debug_opt x = if debug_mode && Sys.file_exists x then [x] else []
let filter_opt x = if Sys.file_exists x then [x] else []

let camlp4boot = "'" ^ String.concat " " camlp4_modules ^ "'"
let camlp4boot_may_debug mods =
  let camlp4_modules = camlp4_modules @
    debug_opt "./boot/ExceptionTracer.cmo" @
    filter_opt "./boot/Profiler.cmo" @ mods
  in "'" ^ String.concat " " camlp4_modules ^ "'"

let ocamlc =
  best ["../ocamlc.opt", "../ocamlc.opt";
        "../ocamlc",     ocamlrun ^^ "../ocamlc";
        "",              "echo no byte compiler available && false"]
let ocamlopt =
  best ["../ocamlopt.opt", "../ocamlopt.opt";
        "../ocamlopt",     ocamlrun ^^ "../ocamlopt";
        "",                "echo no native compiler available && false"]

let () =
  !options.ocamlc       := ocamlc ^^ " -nostdlib -I ../stdlib";
  !options.ocamlopt     := ocamlopt ^^ " -nostdlib -I ../stdlib";
  !options.ocamldoc     := ocamlrun ^^ "../ocamldoc/ocamldoc";
  !options.ocamlyacc    := ocamlrun ^^ "../boot/ocamlyacc";
  !options.ocamllex     := ocamlrun ^^ "../boot/ocamllex";
  !options.ocamldep     := ocamlrun ^^ "../tools/ocamldep";
  ()

let options_without_camlp4 = new_scope (lazy !options)

let fake_options =
  new_scope (lazy { !options with ocamlc = ref "echo ocamlc";
                                  ocamlopt = ref "echo ocamlopt"})

let () =
  !options.ocaml_Flags ^= "-w Ale -warn-error Ale";
  !options.ocaml_P4     := camlp4boot_may_debug [];
  !options.ocaml_P4_opt := camlp4boot_may_debug ["-D OPT"];
  ()

let options_without_debug () = { (!options) with ocaml_P4     = ref camlp4boot
                                               ; ocaml_P4_opt = ref camlp4boot }

let parsing = "../parsing"
and typing = "../typing"
and toplevel = "../toplevel"
and utils = "../utils"
and dynlink = "../otherlibs/dynlink"
and unix = "../otherlibs/unix"
and build = "build"

let ocaml_Module_with_meta =
  generic_ocaml_Module_extension ".meta.ml"
  (fun _ i o ->
    "if test ! -e"^^o^^
       "|| (ruby --version > /dev/null 2> /dev/null);"^^
    "then ruby ./build/meta.rb"^^i^^">"^^o^^"; else : ; fi")
let ocaml_Module_with_genmap =
  generic_ocaml_Module_extension ".genmap.ml"
   (fun _ i o ->
      "if test ! -e"^^o^^
         "|| (test -e ./camlp4boot.run"^^
             "&& test -e Camlp4Filters/GenerateMap.cmo"^^
             "&& test -e Camlp4Filters/GenerateFold.cmo"^^
             "&& test -e Camlp4Filters/RemoveTrashModule.cmo);"^^
      "then (echo 'module Camlp4FiltersTrash = struct';"^^
            "cat Camlp4/Sig/Camlp4Ast.ml; echo 'end;') > Camlp4/Struct/Camlp4Ast.tmp.ml;"^^
           "(echo '(* Generated file! Do not edit by hand *)';"^^
               "../boot/ocamlrun ./camlp4boot.run"^^
                   "./Camlp4Filters/GenerateMap.cmo"^^
                   "./Camlp4Filters/GenerateFold.cmo"^^
                   "./Camlp4Filters/RemoveTrashModule.cmo -printer OCamlr"^^
                   i^^" -no_comments) >"^^o^^"; else : ; fi")

let misc_modules = [
  ocaml_Module ~o:options_without_camlp4 "build/camlp4_config";
  ocaml_IModule ~includes:[utils] ~o:fake_options "../utils/misc";
  ocaml_IModule ~includes:[utils] ~o:fake_options "../utils/warnings";
  ocaml_IModule ~includes:[parsing; utils] ~o:fake_options "../parsing/linenum";
  ocaml_IModule ~includes:[parsing; utils] ~o:fake_options "../parsing/location";
]

let camlp4_package =
  ocaml_PackageDir "Camlp4" (lazy [
    ocaml_IModule ~includes:[build] "Config";
    ocaml_IModule ~o:(options_without_debug ()) ~impl_flags:"-rectypes" "Debug";
    ocaml_IModule "Options";
    ocaml_PackageDir "Sig" (lazy [
      ocaml_Interface "Id";
      ocaml_Interface ~includes:[parsing] "Loc";
      ocaml_Interface "Error";
      ocaml_Interface "Warning";
      ocaml_Interface "Type";
      ocaml_Interface "Token";
      ocaml_Interface "Lexer";
      ocaml_PackageDir "Grammar" (lazy [
        ocaml_Interface "Action";
        ocaml_Interface "Structure";
        ocaml_Interface "Dynamic";
        ocaml_Interface "Static"
      ]);
      ocaml_IModule "Mapper";
      ocaml_Interface "Ast";
      ocaml_Module "Camlp4Ast";
      ocaml_Interface "Quotation";
      ocaml_Interface "Camlp4Token";
      ocaml_Interface "DynLoader";
      ocaml_Interface "AntiquotSyntax";
      ocaml_Interface "Parser";
      ocaml_Interface "Printer";
      ocaml_Interface "Syntax";
      ocaml_Interface "Camlp4Syntax";
      ocaml_Interface "AstFilters";
      ocaml_Interface "SyntaxExtension";
    ]);
    ocaml_IModule "ErrorHandler";
    ocaml_PackageDir "Struct" (lazy [
      ocaml_IModule ~ext_includes:[parsing] "Loc";
      ocaml_Module  "Warning";
      ocaml_IModule "EmptyError";
      ocaml_IModule "EmptyPrinter";
      ocaml_IModule "Token";
      ocaml_Lexer ~includes:[utils] ~ext_includes:[parsing] ~pp:"" "Lexer";
      ocaml_PackageDir "Grammar" (lazy [
        ocaml_Module "Context";
        ocaml_Module "Structure";
        ocaml_Module "Search";
        ocaml_Module "Tools";
        ocaml_IModule ~impl_flags:"-rectypes" "Print";
        ocaml_Module "Failed";
        ocaml_Module "Parser";
        ocaml_IModule "Fold";
        ocaml_Module "Insert";
        ocaml_Module "Delete";
        ocaml_Module "Entry";
        ocaml_Module "Find";
        ocaml_Module "Dynamic";
        ocaml_Module "Static"
      ]);
      ocaml_Module "Quotation";
      ocaml_IModule ~ext_includes:[dynlink] "DynLoader";
      ocaml_Module_with_genmap ~flags:"-w z -warn-error z" "Camlp4Ast";
      ocaml_IModule "FreeVars";
      ocaml_Module_with_meta "MetaAst";
      ocaml_Module "AstFilters";
      ocaml_IModule ~ext_includes:[parsing] "Camlp4Ast2OCamlAst";
      ocaml_IModule "CommentFilter";
    ]);
    ocaml_Module "OCamlInitSyntax";
    ocaml_IModule "PreCast";
    ocaml_IModule "Register";
    ocaml_PackageDir "Printers" (lazy [
      ocaml_IModule "DumpOCamlAst";
      ocaml_IModule "DumpCamlp4Ast";
      ocaml_IModule "OCaml";
      ocaml_IModule "OCamlr" ~flags:"-w v -warn-error v";
      (* ocaml_IModule "OCamlrr"; *)
    ])
  ])

let camlp4_parsers =
  ocaml_PackageDir "Camlp4Parsers" (lazy [
    ocaml_Module "OCamlr";
    ocaml_Module "OCaml";
    (* ocaml_Module "OCamlrr"; *)
    ocaml_Module "OCamlQuotationBase";
    ocaml_Module "OCamlQuotation";
    ocaml_Module "OCamlRevisedQuotation";
    ocaml_Module "OCamlOriginalQuotation";
    ocaml_Module "OCamlRevisedParser";
    ocaml_Module "OCamlParser";
    ocaml_Module "Grammar";
    ocaml_Module "Macro";
    ocaml_Module ~o:(options_without_debug ()) "Debug";
    ocaml_Module "LoadCamlp4Ast";
  ])

let camlp4_printers =
  ocaml_PackageDir "Camlp4Printers" (lazy [
    ocaml_Module "DumpOCamlAst";
    ocaml_Module "DumpCamlp4Ast";
    ocaml_Module "OCaml";
    ocaml_Module "OCamlr";
    (* ocaml_Module "OCamlrr"; *)
    ocaml_Module "Null";
    ocaml_Module ~includes:[unix] "Auto";
  ])

let camlp4_filters =
  ocaml_PackageDir "Camlp4Filters" (lazy [
    ocaml_Module "ExceptionTracer";
    ocaml_Module "Tracer";
    ocaml_Module "StripLocations";
    ocaml_Module "LiftCamlp4Ast";
    ocaml_Module "GenerateMap";
    ocaml_Module "GenerateFold";
    ocaml_Module "RemoveTrashModule";
    ocaml_Module "Profiler";
  ])

let camlp4_top =
  ocaml_PackageDir "Camlp4Top" (lazy [
    ocaml_Module ~includes:[toplevel; typing; parsing] "Rprint";
    ocaml_Module ~includes:[toplevel; parsing; utils] "Camlp4Top";
  ])

let extensions = [ camlp4_parsers; camlp4_printers; camlp4_filters; camlp4_top ]


let pa_r  = ocaml_Module "Camlp4Parsers/OCamlr"
let pa_o = ocaml_Module "Camlp4Parsers/OCaml"
let pa_q = ocaml_Module "Camlp4Parsers/OCamlQuotation"
let pa_qb = ocaml_Module "Camlp4Parsers/OCamlQuotationBase"
let pa_rq = ocaml_Module "Camlp4Parsers/OCamlRevisedQuotation"
let pa_oq = ocaml_Module "Camlp4Parsers/OCamlOriginalQuotation"
let pa_rp = ocaml_Module "Camlp4Parsers/OCamlRevisedParser"
let pa_op = ocaml_Module "Camlp4Parsers/OCamlParser"
let pa_g  = ocaml_Module "Camlp4Parsers/Grammar"
let pa_macro = ocaml_Module "Camlp4Parsers/Macro"
let pa_debug = ocaml_Module "Camlp4Parsers/Debug"
let pr_dump = ocaml_Module "Camlp4Printers/DumpOCamlAst"
let pr_r = ocaml_Module "Camlp4Printers/OCamlr"
let pr_o = ocaml_Module "Camlp4Printers/OCaml"
let pr_a = ocaml_Module "Camlp4Printers/Auto"
let fi_exc = ocaml_Module "Camlp4Filters/ExceptionTracer"
let fi_tracer = ocaml_Module "Camlp4Filters/Tracer"
let camlp4_bin = ocaml_Module "Camlp4Bin"
let top_rprint = ocaml_Module "Camlp4Top/Rprint"
let top_camlp4_top = ocaml_Module "Camlp4Top/Camlp4Top"
let camlp4Profiler = ocaml_IModule "Camlp4Profiler"

let byte_programs = ref []
let opt_programs = ref []
let byte_libraries = ref []
let opt_libraries = ref []

let special_modules =
  if Sys.file_exists "./boot/Profiler.cmo" then [camlp4Profiler] else []

let mk_camlp4_top_lib name modules =
  byte_libraries += (name ^ ".cma");
  opt_libraries  @= [name ^ ".cmxa"; name ^ ".a"];
  ocaml_Library ~default:`Byte ~libraries:["Camlp4"] ~flags:"-linkall" name
  (special_modules @ modules @ [top_camlp4_top])

let mk_camlp4_bin name modules =
  byte_programs += (name ^ ".run");
  opt_programs  += (name ^ ".opt");
  ocaml_Program ~default:`Byte ~includes:[unix] ~libraries:["unix"; "Camlp4"] ~flags:"-linkall" name
  (special_modules @ modules @ [camlp4_bin])

let mk_camlp4_tool name modules =
  byte_programs += (name ^ ".run");
  opt_programs  += (name ^ ".opt");
  [ocaml_Program ~default:`Byte ~libraries:["Camlp4"] ~flags:"-linkall" name modules]

let mk_camlp4 name modules bin_mods top_mods=
  [mk_camlp4_bin name (modules @ bin_mods);
   mk_camlp4_top_lib name (modules @ top_mods)]

let split c s =
  let rec self acc s =
    try
      let pos = String.rindex s c in
      let x = String.sub s 0 pos
      and y = String.sub s (pos + 1) (String.length s - pos - 1)
      in self (y :: acc) x
    with Not_found -> s :: acc
  in self [] s
let chop_end c s =
  let pos = String.rindex s c in
  String.sub s (pos + 1) (String.length s - pos - 1)
let file ppf f =
  let cin = open_in f in
  let rec loop () =
    pp_print_string ppf (input_line cin);
    fprintf ppf "@\n";
    loop ()
  in try loop () with End_of_file -> ()
let ext_split f = split '.' f
  
  
let print_packed_sources ppf ?(skip = fun _ -> false) package_dir =
  let _ =
  fold_units_sources [package_dir] (fun name sources k (skip, inside) ->
    eprintf "%s: [%s] (%b)@." name (String.concat "; " sources) inside;
    let name = try chop_end '/' name with Not_found -> name in
    let k () = ignore (k (skip, true)) in
    if not inside then k () else (
      if skip name then fprintf ppf "(**/**)@\n" else fprintf ppf "(** *)@\n";
      fprintf ppf "@[<2>module %s " name;
      let (mli, ml, mll, k) =
        List.fold_right
          (fun x (mli, ml, mll, k) ->
            match ext_split x with
            | [_; "mli"] -> (Some x, ml, mll, k)
            | [_; "ml"]  -> (mli, Some x, mll, k)
            | [_; "mll"] -> (mli, ml, Some x, k)
            | [x; "meta"; "ml"] -> (mli, Some (x^".ml"), mll, fun () -> ())
            | [x; "genmap"; "ml"] -> (mli, Some (x^".ml"), mll, fun () -> ())
            | [_; ext] -> failwith ("unknown extension " ^ ext)
            | _ -> failwith ("bad file "^x))
          sources (None, None, None, k) in
      (match (ml, mll, mli) with
      | (None, None, Some mli) -> fprintf ppf "=@ @[<2>struct@\n"
      | (_, _, Some mli) -> fprintf ppf ":@,@[<2>sig@\n%a@]@\nend@\n" file mli;
                            fprintf ppf "=@ @[<2>struct@\n"
      | _ -> fprintf ppf "=@ @[<2>struct@\n");
      (match (ml, mll, mli) with
      | (_, Some mll, _) ->
            fprintf ppf "(*___CAMLP4_LEXER___ %a ___CAMLP4_LEXER___*)@\n();"
            file (String.sub mll 0 (String.length mll - 1))
      | (Some ml, _, _) -> k (); fprintf ppf "%a" file ml
      | (None, _, Some mli) -> k (); fprintf ppf "%a" file mli
      | _ -> if sources <> [] then () else k ());
      fprintf ppf "@]@\nend;@]@\n";
      if skip name then fprintf ppf "(**/**)@\n";
    );
    (skip, inside)
  ) (skip, false) in fprintf ppf "@."

let run l =
  let cmd = String.concat " " l in
  let () = Format.printf "%s@." cmd in
  let st =
    Sys.command cmd
    (* 0 *)
  in
  if st <> 0 then failwith ("Exit: " ^ string_of_int st)

let mkdir l = run ("mkdir" :: "-p" :: l)

let cp src dest = run ["cp"; src; dest]

let sed re str file =
  run ["sed"; "-i"; "-e"; "'s/"^re^"/"^str^"/'"; file]

let try_cp src dest = if Sys.file_exists src then cp src dest

let doc () =
  let revised_to_ocaml f =
    run ["./boot/camlp4boot -printer OCaml -o "^f^".ml -impl "^f^".ml4"] in
  let ocamldoc title fl =
    run (("cd doc && ../../ocamldoc/ocamldoc -v -rectypes -html -I ../../parsing "^
          "-I ../build -I ../../utils -I .. -dump ocamldoc.out -t '"^title^"'") :: fl) in
  let ppf_of_file f = formatter_of_out_channel (open_out f) in
  let skip_struct = function "Struct" -> true | _ -> false in
  print_packed_sources (ppf_of_file "doc/Camlp4.ml4") ~skip:skip_struct camlp4_package;
  print_packed_sources (ppf_of_file "doc/Camlp4Parsers.ml4") camlp4_parsers;
  print_packed_sources (ppf_of_file "doc/Camlp4Printers.ml4") camlp4_printers;
  print_packed_sources (ppf_of_file "doc/Camlp4Filters.ml4") camlp4_filters;
  print_packed_sources (ppf_of_file "doc/Camlp4Top.ml4") camlp4_top;
  revised_to_ocaml "doc/Camlp4";
  sed "(\\*___CAMLP4_LEXER___" "" "doc/Camlp4.ml";
  sed "___CAMLP4_LEXER___\\*)" "" "doc/Camlp4.ml";
  sed "^ *# [0-9]\\+.*$" "" "doc/Camlp4.ml";
  revised_to_ocaml "doc/Camlp4Parsers";
  revised_to_ocaml "doc/Camlp4Printers";
  revised_to_ocaml "doc/Camlp4Filters";
  revised_to_ocaml "doc/Camlp4Top";
  ocamldoc "Camlp4 a Pre-Processor-Pretty-Printer for Objective Caml"
           ["Camlp4.ml"; "Camlp4Parsers.ml"; "Camlp4Printers.ml";
            "Camlp4Filters.ml"; "Camlp4Top.ml"]

let other_objs =
  [
    (* "../utils/misc"; "../parsing/linenum"; "../utils/warnings"; *)
   (* "../parsing/location" *)
   ]
let other_byte_objs = String.concat " " (List.map (fun x -> x ^ ".cmo") other_objs)
let other_opt_objs = String.concat " " (List.map (fun x -> x ^ ".cmx") other_objs)
let all =
 List.flatten [
  [ocaml_Library ~default:`Byte
                ~includes:[dynlink]
                ~byte_flags:("dynlink.cma"^^other_byte_objs) ~opt_flags:other_opt_objs
                ~flags:"-linkall" "Camlp4"
                (misc_modules @ special_modules @ [camlp4_package])];
  [mk_camlp4_bin "camlp4" []];
  mk_camlp4 "camlp4boot"
    [pa_r; pa_qb; pa_q; pa_rp; pa_g; pa_macro; pa_debug; pr_o] [pr_a] [top_rprint];
  mk_camlp4 "camlp4r"
    [pa_r; pa_rp] [pr_a] [top_rprint];
  mk_camlp4 "camlp4rf"
    [pa_r; pa_qb; pa_q; pa_rp; pa_g; pa_macro] [pr_a] [top_rprint];
  mk_camlp4 "camlp4o"
    [pa_r; pa_o; pa_rp; pa_op] [pr_a] [];
  mk_camlp4 "camlp4of"
    [pa_r; pa_qb; pa_q; pa_o; pa_rp; pa_op; pa_g; pa_macro] [pr_a] [];
  mk_camlp4 "camlp4oof"
    [pa_r; pa_o; pa_rp; pa_op; pa_qb; pa_oq; pa_g; pa_macro] [pr_a] [];
  mk_camlp4 "camlp4orf"
    [pa_r; pa_o; pa_rp; pa_op; pa_qb; pa_rq; pa_g; pa_macro] [pr_a] [];
  mk_camlp4_tool "mkcamlp4"
    [ocaml_Module ~o:(options_without_debug ()) "mkcamlp4"];
  mk_camlp4_tool "camlp4prof"
    [camlp4Profiler; ocaml_Module ~o:(options_without_debug ()) "camlp4prof"];
 ] @ extensions


let windows = Sys.os_type = "Win32"

(* X.run -> X.exe || X.run -> X *)
let conv_byte_extension f =
  if windows then
    let c = String.copy f in
    (String.blit c (String.rindex c '.') ".exe" 0 4; c)
  else String.sub f 0 (String.rindex f '.')

(* X.opt -> X.opt.exe || X.opt -> X.opt *)
let conv_opt_extension f =
  if windows then f ^ ".exe" else f

let install_all dir =
  printf "Installing %s@.  " dir;
  run ["find"^^dir^^"-name '*.cmi' -o -name '*.cmo' -exec"^^
       "sh -c 'echo \"  install {}\" ; mkdir -p"^^libdir_camlp4^
       "/`dirname {}`; cp {}"^^libdir_camlp4^"/`dirname {}`' \\;"]


let byte =
  "Camlp4.cmi" ::
  "Camlp4.cma" ::
  "Camlp4Parsers.cmi" ::
  "Camlp4Printers.cmi" ::
  "Camlp4Filters.cmi" ::
  "Camlp4Top.cmi" ::
  "Camlp4Parsers.cmo" ::
  "Camlp4Printers.cmo" ::
  "Camlp4Filters.cmo" ::
  "Camlp4Top.cmo" ::
  !byte_libraries

let opt =
  "Camlp4.cmxa" ::
  "Camlp4.a" ::
  "build/camlp4_config.cmx" ::
  "Camlp4Parsers.cmx" ::
  "Camlp4Printers.cmx" ::
  "Camlp4Filters.cmx" ::
  "Camlp4Top.cmx" ::
  (* !opt_libraries @ *)
  []

let install () =
  mkdir [libdir_camlp4; bindir];
  install_all "Camlp4Parsers";
  install_all "Camlp4Printers";
  install_all "Camlp4Filters";
  install_all "Camlp4Top";
  let cp_bin conv bin =
    if Sys.file_exists bin then cp bin (bindir ^ "/" ^ conv bin) in
  List.iter (fun x -> cp x libdir_camlp4) byte;
  List.iter (fun x -> try_cp x libdir_camlp4) opt;
  List.iter (cp_bin conv_byte_extension) !byte_programs;
  List.iter (cp_bin conv_opt_extension) !opt_programs;
  ()
        (* cp mkcamlp4.sh "$(BINDIR)/mkcamlp4" *)
        (* chmod a+x "$(BINDIR)/mkcamlp4" *)


let byte = byte @ !byte_programs
let opt = opt @ !opt_programs

;;

main ~rebuild:(ocaml ^ "build/build.ml")
 (all @ [
  phony_unit ~depends:byte "all";
  phony_unit ~depends:opt "opt";
  generic_unit ~name:"install" ~targets:["install"]
               ~dependencies:(fun ~native:_ _ -> [])
               ~compile_cmd:(fun _ -> install (); exit 0)
               ();
  generic_unit ~name:"doc" ~targets:["doc"]
               ~dependencies:(fun ~native:_ _ -> [])
               ~compile_cmd:(fun _ -> doc (); exit 0)
               ()


 ])