ruby-code-reader.rb

ruby-code-reader.rb

class Code
  class << self
    def print(clazz, class_method: nil, instance_method: nil)
      method = class_method ? clazz.method(class_method) : clazz.instance_method(instance_method)
      puts read_method(*method.source_location)
    end

    private

    def read_method(path, line)
      lines = File.readlines(path)[(line-1)..-1]
      padding = lines.first.scan(/^(\s+)/).first.first
      index = lines.index { |line| line.scan(/^#{padding}end?$/).first } || -1
      lines[0..index]
    end
  end
end