# 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.

class DirMock < Dir
    @@exist_invocation_params = []
    @@mocked_existing_dirs = []

    @@glob_invocation = []
    @@mocked_existing_globs = {}

    @@pwd = nil

    # Monkey patched exists? method.
    # It is used also by the test runner, so it can't start monkey patched
    # To use this, invoke the `is_testing` method before starting your test.
    # Remember to invoke `reset` after the test.
    def self.exist?(path)
        @@exist_invocation_params.push(path)
        return @@mocked_existing_dirs.include?(path)
    end

    # Getter for the `exist_invocation_params` to check that the exist method
    # is invoked with the right parameters
    def self.exist_invocation_params()
        return @@exist_invocation_params
    end

    # Set the list of dirs the test must return as existing
    def self.mocked_existing_dirs(dirs)
        @@mocked_existing_dirs = dirs
    end

    # Set what the `glob` function should return
    def self.mocked_existing_globs(globs, path)
        @@mocked_existing_globs[path] = globs
    end

    def self.glob_invocation
        return @@glob_invocation
    end

    def self.glob(path)
        @@glob_invocation.push(path)
        return @@mocked_existing_globs[path] != nil ? @@mocked_existing_globs[path] : []
    end

    def self.remove_mocked_paths(path)
        @@mocked_existing_globs = @@mocked_existing_globs.select { |k, v| v != path }
    end

    def self.set_pwd(pwd)
        @@pwd = pwd
    end

    def self.pwd
        if @@pwd != nil
            return @@pwd
        end
        return pwd
    end

    # Resets all the settings for the File mock
    def self.reset()
        @@pwd = nil
        @@mocked_existing_dirs = []
        @@exist_invocation_params = []
        @@glob_invocation = []
        @@mocked_existing_globs = {}
    end
end