mirror of
https://github.com/pry/pry.git
synced 2022-11-09 12:35:05 -05:00
remove require io/console from find_method.rb
This commit is contained in:
John Mair
parent
053d91051a
commit
5d3ada6b29
1 changed files with 106 additions and 107 deletions
|
|
@ -1,115 +1,114 @@
|
||||||
require "io/console"
|
|
||||||
|
|
||||||
class Pry
|
class Pry
|
||||||
module DefaultCommands
|
module DefaultCommands
|
||||||
FindMethod = Pry::CommandSet.new do
|
FindMethod = Pry::CommandSet.new do
|
||||||
|
|
||||||
|
|
||||||
|
create_command "find-method" do
|
||||||
|
|
||||||
|
group "Context"
|
||||||
|
|
||||||
create_command "find-method" do
|
description "Recursively search for a method within a Class/Module or the current namespace. find-method [-n | -c] METHOD [NAMESPACE]"
|
||||||
|
|
||||||
group "Context"
|
def options(opti)
|
||||||
|
opti.on :n, :name, "Search for a method by name"
|
||||||
description "Recursively search for a method within a Class/Module or the current namespace. find-method [-n | -c] METHOD [NAMESPACE]"
|
opti.on :c, :content, "Search for a method based on content in Regex form"
|
||||||
|
|
||||||
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
|
||||||
|
|
||||||
|
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
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue