module T::Private::ClassUtils
def self.replace_method(mod, name, &blk)
on which you can call `bind(...).call(...)` to call the original method, or `restore` to
overriding it (if it is defined by one of mod's ancestors). Returns a ReplacedMethod instance
Replaces a method, either by overwriting it (if it is defined directly on `mod`) or by
def self.replace_method(mod, name, &blk) original_method = mod.instance_method(name) original_visibility = visibility_method_name(mod, name) original_owner = original_method.owner mod.ancestors.each do |ancestor| break if ancestor == mod if ancestor == original_owner # If we get here, that means the method we're trying to replace exists on a *prepended* # mixin, which means in order to supersede it, we'd need to create a method on a new # module that we'd prepend before `ancestor`. The problem with that approach is there'd # be no way to remove that new module after prepending it, so we'd be left with these # empty anonymous modules in the ancestor chain after calling `restore`. # # That's not necessarily a deal breaker, but for now, we're keeping it as unsupported. raise "You're trying to replace `#{name}` on `#{mod}`, but that method exists in a " \ "prepended module (#{ancestor}), which we don't currently support." end end overwritten = original_owner == mod T::Configuration.without_ruby_warnings do T::Private::DeclState.current.without_on_method_added do mod.send(:define_method, name, &blk) if blk.arity < 0 && mod.respond_to?(:ruby2_keywords, true) mod.send(:ruby2_keywords, name) end end end mod.send(original_visibility, name) new_method = mod.instance_method(name) ReplacedMethod.new(mod, original_method, new_method, overwritten, original_visibility) end
def self.visibility_method_name(mod, name)
def self.visibility_method_name(mod, name) d_defined?(name) _method_defined?(name) ethod_defined?(name) ew("undefined method `#{name}` for `#{mod}`")