remove require io/console from find_method.rb
This commit is contained in:
John Mair
parent
053d91051a
commit
5d3ada6b29
|
|
@ -1,115 +1,114 @@
|
|||
require "io/console"
|
||||
|
||||
class Pry
|
||||
module DefaultCommands
|
||||
FindMethod = Pry::CommandSet.new do
|
||||
|
||||
|
||||
create_command "find-method" do
|
||||
|
||||
group "Context"
|
||||
|
||||
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
|
||||
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
|
||||
end
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue