pry-rails 0.2.1 → 0.2.2

data/lib/pry-rails/commands.rb

@@ -72,24 +72,69 @@ module PryRails
       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
+
+            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

data/lib/pry-rails/version.rb

@@ -1,5 +1,5 @@
 # encoding: UTF-8
 
 module PryRails
-  VERSION = "0.2.1"
+  VERSION = "0.2.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pry-rails
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-10 00:00:00.000000000 Z
+date: 2012-09-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pry