(**
 * Copyright (c) 2013-present, Facebook, Inc.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *)

open Utils_js

(****************** shared context heap *********************)

module SigContextHeap = SharedMem_js.WithCache (File_key) (struct
  type t = Context.sig_t
  let prefix = Prefix.make()
  let description = "SigContext"
end)

let master_sig: Context.sig_t option option ref = ref None

let add_sig_context = Expensive.wrap SigContextHeap.add

let add_sig ~audit cx =
  let cx_file = Context.file cx in
  if cx_file = File_key.Builtins then master_sig := None;
  add_sig_context ~audit cx_file (Context.sig_cx cx)

let find_sig file =
  let cx_opt =
    if file = File_key.Builtins then
      match !master_sig with
      | Some cx_opt -> cx_opt
      | None ->
        let cx_opt = SigContextHeap.get file in
        master_sig := Some cx_opt;
        cx_opt
    else SigContextHeap.get file
  in
  match cx_opt with
  | Some cx -> cx
  | None -> raise (Key_not_found ("SigContextHeap", File_key.to_string file))

module SigHashHeap = SharedMem_js.NoCache (File_key) (struct
  type t = Xx.hash
  let prefix = Prefix.make()
  let description = "SigHash"
end)

module LeaderHeap = SharedMem_js.WithCache (File_key) (struct
  type t = File_key.t
  let prefix = Prefix.make()
  let description = "Leader"
end)

let find_leader file =
  match LeaderHeap.get file with
  | Some leader -> leader
  | None -> raise (Key_not_found ("LeaderHeap", (File_key.to_string file)))

(* While merging, we must keep LeaderHeap, SigContextHeap, and SigHashHeap in
   sync, sometimes creating new entries and sometimes reusing old entries. *)

(* Add a sig only if it has not changed meaningfully, and return the result of
   that check. *)
let add_merge_on_diff ~audit leader_cx component_files xx =
  let leader_f = Context.file leader_cx in
  let diff = match SigHashHeap.get_old leader_f with
  | None -> true
  | Some xx_old ->
    File_key.check_suffix leader_f Files.flow_ext || xx <> xx_old
  in
  if diff then (
    Nel.iter (fun f -> LeaderHeap.add f leader_f) component_files;
    add_sig_context ~audit leader_f (Context.sig_cx leader_cx);
    SigHashHeap.add leader_f xx;
  )

let add_merge_on_exn ~audit ~options component =
  let leader_f = Nel.hd component in
  let sig_cx = Context.make_sig () in
  let cx =
    let metadata = Context.metadata_of_options options in
    let module_ref = Files.module_ref leader_f in
    Context.make sig_cx metadata leader_f module_ref
  in
  let module_refs = List.map (fun f ->
    let module_ref = Files.module_ref f in
    let module_t = Type.Locationless.AnyT.t in
    Context.add_module cx module_ref module_t;
    LeaderHeap.add f leader_f;
    module_ref
  ) (Nel.to_list component) in
  let xx = Merge_js.ContextOptimizer.sig_context cx module_refs in
  add_sig_context ~audit leader_f sig_cx;
  SigHashHeap.add leader_f xx

let sig_hash_changed f =
  match SigHashHeap.get f with
  | None -> false
  | Some xx ->
    match SigHashHeap.get_old f with
    | None -> true
    | Some xx_old ->
      File_key.check_suffix f Files.flow_ext || xx <> xx_old

let oldify_merge_batch files =
  LeaderHeap.oldify_batch files;
  SigContextHeap.oldify_batch files;
  SigHashHeap.oldify_batch files

let remove_merge_batch files =
  LeaderHeap.remove_batch files;
  SigContextHeap.remove_batch files;
  SigHashHeap.remove_batch files

let remove_old_merge_batch files =
  LeaderHeap.remove_old_batch files;
  SigContextHeap.remove_old_batch files;
  SigHashHeap.remove_old_batch files

let revive_merge_batch files =
  LeaderHeap.revive_batch files;
  SigContextHeap.revive_batch files;
  SigHashHeap.revive_batch files
