commands.rb in pry-rails-0.2.2

- old
+ new
@@ -70,26 +70,71 @@
       end
 
       def process
         Rails.application.eager_load!
 
-        models = ActiveRecord::Base.descendants.map do |mod|
-          model_string = mod.to_s + "\n"
-          if mod.table_exists?
-            model_string << mod.columns.map { |col| "  #{col.name}: #{col.type.to_s}" }.join("\n")
-          else
-            model_string << "  Table doesn't exist"
-          end
-          mod.reflections.each do |model,ref|
-            model_string << "\n  #{ref.macro.to_s} #{model}"
-            model_string << " through #{ref.options[:through]}" unless ref.options[:through].nil?
-          end
-          model_string
-        end.join("\n")
+        if defined?(ActiveRecord::Base)
+          models = ActiveRecord::Base.descendants.map do |mod|
+            model_string = mod.to_s + "\n"
+            if mod.table_exists?
+              model_string << mod.columns.map { |col| "  #{col.name}: #{col.type.to_s}" }.join("\n")
+            else
+              model_string << "  Table doesn't exist"
+            end
+            mod.reflections.each do |model,ref|
+              model_string << "\n  #{ref.macro.to_s} #{model}"
+              model_string << " through #{ref.options[:through]}" unless ref.options[:through].nil?
+            end
+            model_string
+          end.join("\n")
+        elsif defined?(Mongoid::Document)
+          models = get_files.map do |path|
+            mod = extract_class_name(path)
+            model_string = "\033[1;34m#{mod.to_s}\033[0m\n"
+            begin
+              if mod.constantize.included_modules.include?(Mongoid::Document)
+                model_string << mod.constantize.fields.values.sort_by(&:name).map { |col|
+                  "  #{col.name}: \033[1;33m#{col.options[:type].to_s.downcase}\033[0m"
+                }.join("\n")
+                mod.constantize.relations.each do |model,ref|
+                  model_string << "\n  #{kind_of_relation(ref.relation.to_s)} \033[1;34m#{model}\033[0m"
+                  model_string << ", autosave" if ref.options[:autosave]
+                  model_string << ", autobuild" if ref.options[:autobuild]
+                  model_string << ", validate" if ref.options[:validate]
+                  model_string << ", dependent-#{ref.options[:dependent]}" if ref.options[:dependent]
+                end
+              else
+                model_string << "  Collection doesn't exist"
+              end
+              model_string
 
-        models.gsub!(Regexp.new(opts[:G] || ".", Regexp::IGNORECASE)) { |s| text.red(s) } unless opts[:G].nil?
+            rescue Exception
+              STDERR.puts "Warning: exception #{$!} raised while trying to load model class #{path}"
+            end
+          end.join("\n")
+        end
 
+        models.gsub!(Regexp.new(opts[:G] || ".", Regexp::IGNORECASE)) { |s| text.red(s) } unless opts[:G].nil?
         output.puts models
+      end
+
+      def get_files(prefix ='')
+        Dir.glob(prefix << "app/models/**/*.rb")
+      end
+
+      def extract_class_name(filename)
+        filename.split('/')[2..-1].collect { |i| i.camelize }.join('::').chomp(".rb")
+      end
+
+      def kind_of_relation(string)
+        case string.gsub('Mongoid::Relations::', '')
+        when 'Referenced::Many' then 'has_many'
+        when 'Referenced::One' then 'has_one'
+        when 'Referenced::In' then 'belongs_to'
+        when 'Embedded::Many' then 'embeds_many'
+        when 'Embedded::One' then 'embeds_one'
+        when 'Embedded::In' then 'embedded_in'
+        end
       end
     end
 
     create_command "show-middleware" do
       group "Rails"