blob: 10463ebf08ba6e2560bd0437cbf200cb7306d889 (
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
|
# frozen_string_literal: true
require_relative "nop"
require_relative "../source_finder"
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
source = SourceFinder.new(@irb_context).find_source(str)
if source
show_source(source)
else
puts "Error: Couldn't locate a definition for #{str}"
end
nil
end
private
def show_source(source)
puts
puts "#{bold("From")}: #{source.file}:#{source.first_line}"
puts
code = IRB::Color.colorize_code(File.read(source.file))
puts code.lines[(source.first_line - 1)...source.last_line].join
puts
end
def bold(str)
Color.colorize(str, [:BOLD])
end
end
end
end
|