blob: 9a0364e3eba3ed70f548b530ee3a7cd385ded50d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
# frozen_string_literal: true
require_relative "nop"
require_relative "../source_finder"
require_relative "../pager"
require_relative "../color"
module IRB
module ExtendCommand
class ShowSource < Nop
category "Context"
description "Show the source code of a given method or constant."
class << self
def transform_args(args)
# Return a string literal as is for backward compatibility
if args.empty? || string_literal?(args)
args
else # Otherwise, consider the input as a String for convenience
args.strip.dump
end
end
end
def execute(str = nil)
unless str.is_a?(String)
puts "Error: Expected a string but got #{str.inspect}"
return
end
if str.include? " -s"
str, esses = str.split(" -")
s_count = esses.count("^s").zero? ? esses.size : 1
source = SourceFinder.new(@irb_context).find_source(str, s_count)
else
source = SourceFinder.new(@irb_context).find_source(str)
end
if source
show_source(source)
elsif s_count
puts "Error: Couldn't locate a super definition for #{str}"
else
puts "Error: Couldn't locate a definition for #{str}"
end
nil
end
private
def show_source(source)
file_content = IRB::Color.colorize_code(File.read(source.file))
code = file_content.lines[(source.first_line - 1)...source.last_line].join
content = <<~CONTENT
#{bold("From")}: #{source.file}:#{source.first_line}
#{code}
CONTENT
Pager.page_content(content)
end
def bold(str)
Color.colorize(str, [:BOLD])
end
end
end
end
|
