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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# encoding: UTF-8
class PryRails::ShowModels < Pry::ClassCommand
match "show-models"
group "Rails"
description "Show all models."
def options(opt)
opt.banner unindent <<-USAGE
Usage: show-models
show-models displays the current Rails app's models.
USAGE
opt.on :G, "grep", "Filter output by regular expression", :argument => true
end
def process
Rails.application.eager_load!
@formatter = PryRails::ModelFormatter.new
display_activerecord_models
display_mongoid_models
end
def display_activerecord_models
return unless defined?(ActiveRecord::Base)
models = ActiveRecord::Base.descendants
models.sort_by(&:to_s).each do |model|
print_unless_filtered @formatter.format_active_record(model)
end
end
def display_mongoid_models
return unless defined?(Mongoid::Document)
models = []
ObjectSpace.each_object do |o|
# If this is deprecated, calling any methods on it will emit a warning,
# so just back away slowly.
next if ActiveSupport::Deprecation::DeprecationProxy === o
is_model = false
begin
is_model = o.class == Class && o.ancestors.include?(Mongoid::Document)
rescue
# If it's a weird object, it's not what we want anyway.
end
models << o if is_model
end
models.sort_by(&:to_s).each do |model|
print_unless_filtered @formatter.format_mongoid(model)
end
end
def print_unless_filtered(str)
if opts.present?(:G)
return unless str =~ grep_regex
str = colorize_matches(str) # :(
end
output.puts str
end
def colorize_matches(string)
if Pry.color
string.to_s.gsub(grep_regex) { |s| "\e[7m#{s}\e[27m" }
else
string
end
end
def grep_regex
@grep_regex ||= Regexp.new(opts[:G], Regexp::IGNORECASE)
end
end
PryRails::Commands.add_command PryRails::ShowModels
|
