class Zeitwerk::Loader

# File lib/zeitwerk/loader.rb, line 456
def all_dirs
  dirs = []
  Registry.loaders.each do |loader|
    dirs.concat(loader.dirs)
  end
  dirs.freeze
end

# File lib/zeitwerk/loader.rb, line 428
def eager_load_all
  Registry.loaders.each do |loader|
    begin
      loader.eager_load
    rescue SetupRequired
      # This is fine, we eager load what can be eager loaded.
    end
  end
end

# File lib/zeitwerk/loader.rb, line 442
def eager_load_namespace(mod)
  Registry.loaders.each do |loader|
    begin
      loader.eager_load_namespace(mod)
    rescue SetupRequired
      # This is fine, we eager load what can be eager loaded.
    end
  end
end

# File lib/zeitwerk/loader.rb, line 390
def for_gem(warn_on_extra_files: true)
  called_from = caller_locations(1, 1).first.path
  Registry.loader_for_gem(called_from, namespace: Object, warn_on_extra_files: warn_on_extra_files)
end

# File lib/zeitwerk/loader.rb, line 411
def for_gem_extension(namespace)
  unless namespace.is_a?(Module) # Note that Class < Module.
    raise Zeitwerk::Error, "#{namespace.inspect} is not a class or module object, should be"
  end

  unless real_mod_name(namespace)
    raise Zeitwerk::Error, "extending anonymous namespaces is unsupported"
  end

  called_from = caller_locations(1, 1).first.path
  Registry.loader_for_gem(called_from, namespace: namespace, warn_on_extra_files: false)
end

# File lib/zeitwerk/loader.rb, line 107
def initialize
  super

  @autoloads       = {}
  @inceptions      = Zeitwerk::Cref::Map.new
  @autoloaded_dirs = []
  @to_unload       = {}
  @namespace_dirs  = Zeitwerk::Cref::Map.new
  @shadowed_files  = Set.new
  @setup           = false
  @eager_loaded    = false

  @mutex = Mutex.new
  @dirs_autoload_monitor = Monitor.new

  Registry.loaders.register(self)
end

# File lib/zeitwerk/loader.rb, line 247
def all_expected_cpaths
  result = {}

  actual_roots.each do |root_dir, root_namespace|
    queue = [[root_dir, real_mod_name(root_namespace)]]

    while (dir, cpath = queue.shift)
      result[dir] = cpath

      prefix = cpath == "Object" ? "" : cpath + "::"

      ls(dir) do |basename, abspath, ftype|
        if ftype == :file
          basename.delete_suffix!(".rb")
          result[abspath] = prefix + inflector.camelize(basename, abspath)
        else
          if collapse?(abspath)
            queue << [abspath, cpath]
          else
            queue << [abspath, prefix + inflector.camelize(basename, abspath)]
          end
        end
      end
    end
  end

  result
end

# File lib/zeitwerk/loader.rb, line 277
def cpath_expected_at(path)
  abspath = File.expand_path(path)

  raise Zeitwerk::Error.new("#{abspath} does not exist") unless File.exist?(abspath)

  return unless dir?(abspath) || ruby?(abspath)
  return if ignored_path?(abspath)

  paths = []

  if ruby?(abspath)
    basename = File.basename(abspath, ".rb")
    return if hidden?(basename)

    paths << [basename, abspath]
    walk_up_from = File.dirname(abspath)
  else
    walk_up_from = abspath
  end

  root_namespace = nil

  walk_up(walk_up_from) do |dir|
    break if root_namespace = roots[dir]
    return if ignored_path?(dir)

    basename = File.basename(dir)
    return if hidden?(basename)

    paths << [basename, dir] unless collapse?(dir)
  end

  return unless root_namespace

  if paths.empty?
    real_mod_name(root_namespace)
  else
    cnames = paths.reverse_each.map { cname_for(_1, _2) }

    if root_namespace == Object
      cnames.join("::")
    else
      "#{real_mod_name(root_namespace)}::#{cnames.join("::")}"
    end
  end
end

# File lib/zeitwerk/loader.rb, line 233
def reload
  raise ReloadingDisabledError unless reloading_enabled?
  raise SetupRequired unless @setup

  unload
  recompute_ignored_paths
  recompute_collapse_dirs
  setup
end

# File lib/zeitwerk/loader.rb, line 128
def setup
  mutex.synchronize do
    break if @setup

    actual_roots.each do |root_dir, root_namespace|
      define_autoloads_for_dir(root_dir, root_namespace)
    end

    on_setup_callbacks.each(&:call)

    @setup = true
  end
end

# File lib/zeitwerk/loader.rb, line 362
         def shadowed_file?(file)
  shadowed_files.member?(file)
end

# File lib/zeitwerk/loader.rb, line 154
def unload
  mutex.synchronize do
    raise SetupRequired unless @setup

    # We are going to keep track of the files that were required by our
    # autoloads to later remove them from $LOADED_FEATURES, thus making them
    # loadable by Kernel#require again.
    #
    # Directories are not stored in $LOADED_FEATURES, keeping track of files
    # is enough.
    unloaded_files = Set.new

    autoloads.each do |abspath, cref|
      if cref.autoload?
        unload_autoload(cref)
      else
        # Could happen if loaded with require_relative. That is unsupported,
        # and the constant path would escape unloadable_cpath? This is just
        # defensive code to clean things up as much as we are able to.
        unload_cref(cref)
        unloaded_files.add(abspath) if ruby?(abspath)
      end
    end

    to_unload.each do |abspath, cref|
      unless on_unload_callbacks.empty?
        begin
          value = cref.get
        rescue ::NameError
          # Perhaps the user deleted the constant by hand, or perhaps an
          # autoload failed to define the expected constant but the user
          # rescued the exception.
        else
          run_on_unload_callbacks(cref, value, abspath)
        end
      end

      unload_cref(cref)
      unloaded_files.add(abspath) if ruby?(abspath)
    end

    unless unloaded_files.empty?
      # Bootsnap decorates Kernel#require to speed it up using a cache and
      # this optimization does not check if $LOADED_FEATURES has the file.
      #
      # To make it aware of changes, the gem defines singleton methods in
      # $LOADED_FEATURES:
      #
      #   https://github.com/Shopify/bootsnap/blob/main/lib/bootsnap/load_path_cache/core_ext/loaded_features.rb
      #
      # Rails applications may depend on bootsnap, so for unloading to work
      # in that setting it is preferable that we restrict our API choice to
      # one of those methods.
      $LOADED_FEATURES.reject! { |file| unloaded_files.member?(file) }
    end

    autoloads.clear
    autoloaded_dirs.clear
    to_unload.clear
    namespace_dirs.clear
    shadowed_files.clear

    unregister_inceptions
    unregister_explicit_namespaces

    Registry.autoloads.unregister_loader(self)

    @setup        = false
    @eager_loaded = false
  end
end

# File lib/zeitwerk/loader.rb, line 331
def unloadable_cpath?(cpath)
  unloadable_cpaths.include?(cpath)
end

# File lib/zeitwerk/loader.rb, line 342
def unloadable_cpaths
  to_unload.values.map(&:path)
end

# File lib/zeitwerk/loader.rb, line 350
def unregister
  unregister_inceptions
  unregister_explicit_namespaces
  Registry.loaders.unregister(self)
  Registry.autoloads.unregister_loader(self)
  Registry.unregister_loader(self)
end

# File lib/zeitwerk/loader.rb, line 513
        def autoload_file(cref, file)
  if autoload_path = cref.autoload? || Registry.inceptions.registered?(cref)
    # First autoload for a Ruby file wins, just ignore subsequent ones.
    if ruby?(autoload_path)
      shadowed_files << file
      log("file #{file} is ignored because #{autoload_path} has precedence") if logger
    else
      promote_namespace_from_implicit_to_explicit(dir: autoload_path, file: file, cref: cref)
    end
  elsif cref.defined?
    shadowed_files << file
    log("file #{file} is ignored because #{cref} is already defined") if logger
  else
    define_autoload(cref, file)
  end
end

# File lib/zeitwerk/loader.rb, line 565
        def autoload_path_set_by_me_for?(cref)
  if autoload_path = cref.autoload?
    autoload_path if autoloads.key?(autoload_path)
  else
    inceptions[cref]
  end
end

# File lib/zeitwerk/loader.rb, line 484
        def autoload_subdir(cref, subdir)
  if autoload_path = autoload_path_set_by_me_for?(cref)
    if ruby?(autoload_path)
      # Scanning visited a Ruby file first, and now a directory for the same
      # constant has been found. This means we are dealing with an explicit
      # namespace whose definition was seen first.
      #
      # Registering is idempotent, and we have to keep the autoload pointing
      # to the file. This may run again if more directories are found later
      # on, no big deal.
      register_explicit_namespace(cref)
    end
    # If the existing autoload points to a file, it has to be preserved, if
    # not, it is fine as it is. In either case, we do not need to override.
    # Just remember the subdirectory conforms this namespace.
    namespace_dirs.get_or_set(cref) { [] } << subdir
  elsif !cref.defined?
    # First time we find this namespace, set an autoload for it.
    namespace_dirs.get_or_set(cref) { [] } << subdir
    define_autoload(cref, subdir)
  else
    # For whatever reason the constant that corresponds to this namespace has
    # already been defined, we have to recurse.
    log("the namespace #{cref} already exists, descending into #{subdir}") if logger
    define_autoloads_for_dir(subdir, cref.get)
  end
end

# File lib/zeitwerk/loader.rb, line 547
        def define_autoload(cref, abspath)
  cref.autoload(abspath)

  if logger
    if ruby?(abspath)
      log("autoload set for #{cref}, to be loaded from #{abspath}")
    else
      log("autoload set for #{cref}, to be autovivified from #{abspath}")
    end
  end

  autoloads[abspath] = cref
  Registry.autoloads.register(abspath, self)

  register_inception(cref, abspath) unless cref.autoload?
end

# File lib/zeitwerk/loader.rb, line 466
        def define_autoloads_for_dir(dir, parent)
  ls(dir) do |basename, abspath, ftype|
    if ftype == :file
      basename.delete_suffix!(".rb")
      cref = Cref.new(parent, cname_for(basename, abspath))
      autoload_file(cref, abspath)
    else
      if collapse?(abspath)
        define_autoloads_for_dir(abspath, parent)
      else
        cref = Cref.new(parent, cname_for(basename, abspath))
        autoload_subdir(cref, abspath)
      end
    end
  end
end

# File lib/zeitwerk/loader.rb, line 534
        def promote_namespace_from_implicit_to_explicit(dir:, file:, cref:)
  autoloads.delete(dir)
  Registry.autoloads.unregister(dir)

  log("earlier autoload for #{cref} discarded, it is actually an explicit namespace defined in #{file}") if logger

  # Order matters: When Module#const_added is triggered by the autoload, we
  # don't want the namespace to be registered yet.
  define_autoload(cref, file)
  register_explicit_namespace(cref)
end

# File lib/zeitwerk/loader.rb, line 598
        def raise_if_conflicting_directory(dir)
  MUTEX.synchronize do
    Registry.loaders.each do |loader|
      next if loader == self

      loader.__roots.each_key do |root_dir|
        # Conflicting directories are rare, optimize for the common case.
        next if !dir.start_with?(root_dir) && !root_dir.start_with?(dir)

        dir_slash = dir + "/"
        root_dir_slash = root_dir + "/"
        next if !dir_slash.start_with?(root_dir_slash) && !root_dir_slash.start_with?(dir_slash)

        next if ignores?(root_dir)
        break if loader.__ignores?(dir)

        require "pp" # Needed to have pretty_inspect available.
        raise Error,
          "loader\n\n#{pretty_inspect}\n\nwants to manage directory #{dir}," \
          " which is already managed by\n\n#{loader.pretty_inspect}\n"
      end
    end
  end
end

# File lib/zeitwerk/loader.rb, line 574
        def register_explicit_namespace(cref)
  Registry.explicit_namespaces.register(cref, self)
end

# File lib/zeitwerk/loader.rb, line 584
        def register_inception(cref, abspath)
  inceptions[cref] = abspath
  Registry.inceptions.register(cref, abspath)
end

# File lib/zeitwerk/loader.rb, line 624
        def run_on_unload_callbacks(cref, value, abspath)
  # Order matters. If present, run the most specific one.
  on_unload_callbacks[cref.path]&.each { |c| c.call(value, abspath) }
  on_unload_callbacks[:ANY]&.each { |c| c.call(cref.path, value, abspath) }
end

# File lib/zeitwerk/loader.rb, line 631
        def unload_autoload(cref)
  cref.remove
  log("autoload for #{cref} removed") if logger
end

# File lib/zeitwerk/loader.rb, line 637
        def unload_cref(cref)
  # Let's optimistically remove_const. The way we use it, this is going to
  # succeed always if all is good.
  cref.remove
rescue ::NameError
  # There are a few edge scenarios in which this may happen. If the constant
  # is gone, that is OK, anyway.
else
  log("#{cref} unloaded") if logger
end

# File lib/zeitwerk/loader.rb, line 579
        def unregister_explicit_namespaces
  Registry.explicit_namespaces.unregister_loader(self)
end

# File lib/zeitwerk/loader.rb, line 590
        def unregister_inceptions
  inceptions.each_key do |cref|
    Registry.inceptions.unregister(cref)
  end
  inceptions.clear
end