class T::Private::ClassUtils::ReplacedMethod

def bind(obj)
  @old_method.bind(obj)
end

def initialize(mod, old_method, new_method, overwritten, visibility)
  if old_method.name != new_method.name
    raise "Method names must match. old=#{old_method.name} new=#{new_method.name}"
  end
  @mod = mod
  @old_method = old_method
  @new_method = new_method
  @overwritten = overwritten
  @name = old_method.name
  @visibility = visibility
  @restored = false
end

def restore
  # The check below would also catch this, but this makes the failure mode much clearer
  if @restored
    raise "Method '#{@name}' on '#{@mod}' was already restored"
  end
  if @mod.instance_method(@name) != @new_method
    raise "Trying to restore #{@mod}##{@name} but the method has changed since the call to replace_method"
  end
  @restored = true
  if @overwritten
    # The original method was overwritten. Overwrite again to restore it.
    @mod.send(:define_method, @old_method.name, @old_method) # rubocop:disable PrisonGuard/UsePublicSend
  else
    # The original method was in an ancestor. Restore it by removing the overriding method.
    @mod.send(:remove_method, @old_method.name) # rubocop:disable PrisonGuard/UsePublicSend
  end
  # Restore the visibility. Note that we need to do this even when we call remove_method
  # above, because the module may have set custom visibility for a method it inherited.
  @mod.send(@visibility, @old_method.name) # rubocop:disable PrisonGuard/UsePublicSend
  nil
end

def to_s
  @old_method.to_s
end