如何临时更改 Ruby 中的 require 路径 ($:)？

Question

我正在为一个复杂的项目使用一堆 Rake 任务做一些技巧，逐渐地一次重构一些块的复杂性。这暴露了前项目维护者留下的奇怪的依赖网络。

我想要做的是将项目中的特定路径添加到 require 的要搜索的路径列表中，即 $:。但是，我只希望在一个特定方法的上下文中搜索该路径。现在我正在做这样的事情：

def foo()
  # Look up old paths, add new special path.
  paths = $:
  $: << special_path

  # Do work ...
  bar()
  baz()
  quux()

  # Reset.
  $:.clear
  $: << paths
end

def bar()
  require '...' # If called from within foo(), will also search special_path.
  ...
end

这显然是一个可怕的黑客行为。有更好的办法吗？

Answer 1

由于 $: 是一个数组，因此您必须小心您所做的事情。您需要获取一份副本（通过dup）并稍后替换。不过，简单地删除您添加的内容更简单：

def foo
  $: << special_path

  # Do work ...
  bar()

ensure
  # Reset.
  $:.delete(special_path)
end

如果没有更多信息，很难知道是否有更好的方法。

Answer 2

require 实际上是一个方法，它是 Kernel#require （它调用 rb_require_safe），所以你至少可以在猴子补丁版本中执行你的黑客攻击。如果你喜欢这样的事情。

• 给原始 require 起别名
• 如果传递了绝对路径，则调用原始 require 方法，
• 否则通过创建绝对路径并调用原始 require 方法来迭代加载路径。

只是为了好玩，我对此进行了快速的抨击，原型如下。这尚未经过充分测试，我还没有检查 rb_require_safe 的语义，您可能还需要查看 #load 和 #includeKernel 模块的猴子补丁。这也许并不完全是可怕的，但它肯定是一种黑客行为。您的电话是否比弄乱全局 $: 变量更好或更坏。

module Kernel

  alias original_require require

  # Just like standard require but takes an
  # optional second argument (a string or an
  # array of strings) for additional directories
  # to search.
  def require(file, more_dirs=[])
    if file =~ /^\// # absolute path
      original_require(file)
    else
      ($: + [ more_dirs ].flatten).each do |dir|
        path = File.join(dir, file)
        begin
          return original_require(path)
        rescue LoadError
        end
      end
      raise LoadError,
        "no such file to load -- #{file}"
    end
  end

end

示例：

require 'mymod'
require 'mymod', '/home/me/lib'
require 'mymod', [ '/home/me/lib', '/home/you/lib' ]