mirror of
https://github.com/pry/pry.git
synced 2022-11-09 12:35:05 -05:00
114 lines
3.5 KiB
Ruby
114 lines
3.5 KiB
Ruby
class Pry
|
|
module DefaultCommands
|
|
FindMethod = Pry::CommandSet.new do
|
|
|
|
|
|
create_command "find-method" do
|
|
|
|
group "Context"
|
|
|
|
description "Recursively search for a method within a Class/Module or the current namespace. find-method [-n | -c] METHOD [NAMESPACE]"
|
|
|
|
def options(opti)
|
|
opti.on :n, :name, "Search for a method by name"
|
|
opti.on :c, :content, "Search for a method based on content in Regex form"
|
|
end
|
|
|
|
def process
|
|
return if args.size < 1
|
|
pattern = ::Regexp.new args[0]
|
|
if args[1]
|
|
klass = target.eval(args[1])
|
|
if !klass.is_a?(Module)
|
|
klass = klass.class
|
|
end
|
|
else
|
|
klass = eval target_self.pretty_inspect
|
|
end
|
|
if opts.name?
|
|
to_put = name_search(pattern, klass)
|
|
elsif opts.content?
|
|
to_put = content_search(pattern, klass)
|
|
else
|
|
to_put = name_search(pattern, klass)
|
|
end
|
|
1
|
|
if to_put.flatten == []
|
|
puts "\e[32;1mNo Methods Matched\e[0m"
|
|
else
|
|
puts "\e[1mMethods Matched\e[0m"
|
|
puts "--"
|
|
puts to_put
|
|
end
|
|
|
|
end
|
|
|
|
private
|
|
|
|
def puts(item)
|
|
output.puts item
|
|
end
|
|
|
|
def content_search(pattern, klass, current=[])
|
|
return unless(klass.is_a? Module)
|
|
return if current.include? klass
|
|
current << klass
|
|
meths = []
|
|
(Pry::Method.all_from_class(klass) + Pry::Method.all_from_obj(klass)).uniq.each do |meth|
|
|
begin
|
|
if meth.source =~ pattern && !meth.alias?
|
|
header = "#{klass}##{meth.name}: "
|
|
meths << header + colorize_code((meth.source.split(/\n/).select {|x| x =~ pattern }).join("\n#{' ' * header.length}"))
|
|
end
|
|
rescue Exception
|
|
next
|
|
rescue Pry::CommandError
|
|
next
|
|
end
|
|
end
|
|
klass.constants.each do |klazz|
|
|
meths += ((res = content_search(pattern, klass.const_get(klazz), current)) ? res : [])
|
|
end
|
|
return meths.uniq.flatten
|
|
end
|
|
|
|
def name_search(regex, klass, current=[])
|
|
return unless(klass.is_a? Module)
|
|
return if current.include? klass
|
|
current << klass
|
|
header = "\e[1;34;4m#{klass.name}\e[0;24m:"
|
|
meths = []
|
|
(Pry::Method.all_from_class(klass) + Pry::Method.all_from_obj(klass)).uniq.each do |x|
|
|
if x.name =~ regex
|
|
meths << " #{x.name}"
|
|
begin
|
|
if x.alias?
|
|
meths[-1] += "#A|#{x.original_name}" if x.original_name
|
|
end
|
|
rescue Exception
|
|
end
|
|
end
|
|
|
|
end
|
|
max = meths.map(&:length).max
|
|
meths.map! do |x|
|
|
if x =~ /#{"#A"}/
|
|
x = x.sub!("#A|", ((' ' * ((max - x.length) + 3)) + "\e[33m(Alias of ")) + ")\e[0m"
|
|
end
|
|
x
|
|
end
|
|
meths.unshift header if meths.size > 0
|
|
klass.constants.each do |x|
|
|
begin
|
|
meths << ((res = name_search(regex, klass.const_get(x), current)) ? res : [])
|
|
rescue Exception
|
|
next
|
|
end
|
|
end
|
|
return meths.uniq.flatten
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|