(*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *)

open Scope

module LookupMode = struct
  type t =
    | ForValue
    | ForType
    | ForTypeof
end

module type S = sig
  type scope

  type t = scope list

  val new_env : bool

  val in_toplevel_scope : unit -> bool

  val in_global_scope : unit -> bool

  val peek_env : unit -> t

  val clone_env : t -> t

  val string_of_env : Context.t -> t -> string

  val var_scope_kind : unit -> Scope.var_scope_kind

  val in_async_scope : unit -> bool

  val in_generator_scope : unit -> bool

  val in_predicate_scope : unit -> bool

  val find_entry :
    Context.t -> Reason.name -> ?desc:Reason.reason_desc -> ALoc.t -> Scope.t * Entry.t

  val push_var_scope : Scope.t -> unit

  val pop_var_scope : unit -> unit

  val in_lex_scope : (unit -> 'a) -> 'a

  val env_depth : unit -> int

  val trunc_env : int -> unit

  val init_env : ?exclude_syms:NameUtils.Set.t -> Context.t -> Scope.t -> unit

  val update_env : ALoc.t -> t -> unit

  val save_excluded_symbols : unit -> NameUtils.Set.t

  val restore_excluded_symbols : NameUtils.Set.t -> unit

  (***)

  val bind_class :
    Context.t ->
    ALoc.id ->
    Type.Properties.id ->
    Type.Properties.id ->
    Type.Properties.id ->
    Type.Properties.id ->
    unit

  val bind_var :
    ?state:State.t -> Context.t -> string -> Type.annotated_or_inferred -> ALoc.t -> unit

  val bind_let :
    ?state:State.t -> Context.t -> string -> Type.annotated_or_inferred -> ALoc.t -> unit

  val bind_implicit_let :
    ?state:State.t ->
    Entry.let_binding_kind ->
    Context.t ->
    Reason.name ->
    Type.annotated_or_inferred ->
    ALoc.t ->
    unit

  val bind_fun : ?state:State.t -> Context.t -> Reason.name -> Type.t -> ALoc.t -> unit

  val bind_implicit_const :
    ?state:State.t ->
    Entry.const_binding_kind ->
    Context.t ->
    string ->
    Type.annotated_or_inferred ->
    ALoc.t ->
    unit

  val bind_const :
    ?state:State.t -> Context.t -> string -> Type.annotated_or_inferred -> ALoc.t -> unit

  val bind_import : Context.t -> string -> Type.t -> ALoc.t -> unit

  val bind_type : ?state:State.t -> Context.t -> string -> Type.t -> ALoc.t -> unit

  val bind_this_tparam : state:State.t -> Context.t -> Type.t -> ALoc.t -> unit

  val bind_import_type : Context.t -> string -> Type.t -> ALoc.t -> unit

  val bind_declare_var : Context.t -> Reason.name -> Type.t -> ALoc.t -> unit

  val bind_declare_fun : Context.t -> predicate:bool -> Reason.name -> Type.t -> ALoc.t -> unit

  val declare_let : Context.t -> Reason.name -> ALoc.t -> unit

  val declare_implicit_let : Entry.let_binding_kind -> Context.t -> Reason.name -> ALoc.t -> unit

  val declare_const : Context.t -> Reason.name -> ALoc.t -> unit

  val declare_implicit_const :
    Entry.const_binding_kind -> Context.t -> Reason.name -> ALoc.t -> unit

  val init_var :
    Context.t -> use_op:Type.use_op -> Reason.name -> has_anno:bool -> Type.t -> ALoc.t -> unit

  val init_let :
    Context.t -> use_op:Type.use_op -> Reason.name -> has_anno:bool -> Type.t -> ALoc.t -> unit

  val init_implicit_let :
    Entry.let_binding_kind ->
    Context.t ->
    use_op:Type.use_op ->
    Reason.name ->
    has_anno:bool ->
    Type.t ->
    ALoc.t ->
    unit

  val init_fun : Context.t -> use_op:Type.use_op -> Reason.name -> Type.t -> ALoc.t -> unit

  val init_const :
    Context.t -> use_op:Type.use_op -> Reason.name -> has_anno:bool -> Type.t -> ALoc.t -> unit

  val init_implicit_const :
    Entry.const_binding_kind ->
    Context.t ->
    use_op:Type.use_op ->
    Reason.name ->
    has_anno:bool ->
    Type.t ->
    ALoc.t ->
    unit

  val init_type : Context.t -> string -> Type.t -> ALoc.t -> unit

  val init_import : lookup_mode:LookupMode.t -> Context.t -> Reason.name -> ALoc.t -> Type.t -> unit

  val pseudo_init_declared_type : Context.t -> string -> ALoc.t -> unit

  val is_provider : Context.t -> ALoc.t -> bool

  val install_provider : Context.t -> Type.t -> Reason.name -> ALoc.t -> unit

  val local_scope_entry_exists : Context.t -> ALoc.t -> string -> bool

  val is_global_var : Context.t -> string -> ALoc.t -> bool

  val get_current_env_refi : Key.t -> Scope.refi_binding option

  val get_class_entries : unit -> Type.class_binding list

  val get_var : ?lookup_mode:LookupMode.t -> Context.t -> string -> ALoc.t -> Type.t

  val get_internal_var : Context.t -> string -> ALoc.t -> Type.t

  val get_var_annotation : Context.t -> Reason.name -> ALoc.t -> Type.t option

  val get_var_declared_type :
    ?lookup_mode:LookupMode.t -> Context.t -> Reason.name -> ALoc.t -> Type.t

  val constraining_type : default:Type.t -> Context.t -> Reason.name -> ALoc.t -> Type.t

  val unify_declared_type :
    ?lookup_mode:LookupMode.t -> Context.t -> Reason.name -> ALoc.t -> Type.t -> unit

  val unify_declared_fun_type : Context.t -> Reason.name -> ALoc.t -> Type.t -> unit

  val var_ref :
    ?lookup_mode:LookupMode.t ->
    Context.t ->
    ?desc:Reason.reason_desc ->
    Reason.name ->
    ALoc.t ->
    Type.t

  val query_var :
    ?lookup_mode:LookupMode.t ->
    Context.t ->
    Reason.name ->
    ?desc:Reason.reason_desc ->
    ALoc.t ->
    Type.t

  val set_var : Context.t -> use_op:Type.use_op -> string -> Type.t -> ALoc.t -> unit

  val set_internal_var : Context.t -> string -> Type.t -> ALoc.t -> unit

  val set_expr : Context.t -> Key.t -> ALoc.t -> refined:Type.t -> original:Type.t -> unit

  val refine_expr : Key.t -> ALoc.t -> Type.t -> Type.t -> int * Key.t * Changeset.op

  val refine_with_preds :
    Context.t -> ALoc.t -> Type.predicate Key_map.t -> Type.t Key_map.t -> Changeset.t

  val in_refined_env :
    Context.t -> ALoc.t -> Type.predicate Key_map.t -> Type.t Key_map.t -> (unit -> 'a) -> 'a

  val merge_env : Context.t -> ALoc.t -> t * t * t -> Changeset.t -> unit

  val widen_env : Context.t -> ALoc.t -> unit

  val copy_env : Context.t -> ALoc.t -> t * t -> Changeset.t -> unit

  val havoc_all : unit -> unit

  val reset_current_activation : ALoc.t -> unit

  val havoc_vars : Changeset.t -> unit

  val havoc_heap_refinements : unit -> unit

  val havoc_local_refinements : ?all:bool -> Context.t -> unit

  val havoc_heap_refinements_with_propname : private_:bool -> string -> unit

  val get_refinement : Context.t -> Key.t -> ALoc.t -> Type.t option

  val record_expression_type_if_needed : Context.t -> ALoc.t -> Type.t -> unit

  val discriminant_after_negated_cases :
    Context.t ->
    ALoc.t ->
    (Reason.name * Key.proj list) option ->
    (ALoc.t, ALoc.t) Flow_ast.Expression.t ->
    Type.t option

  val valid_declaration_check : Context.t -> Reason.name -> ALoc.t -> unit
end
