mongoid-erd 0.0.3 → 0.0.5

data/examples/models/hello.rb

@@ -3,6 +3,7 @@
 class Hello # erd{color: red} Hello Class
   field :name, type:String # erd Name Field
   field :link, type:String # -> World
+  as_enum :gender, :male => 1, :female => 2  # erd: 性别
 
   def say # erd say hello world
 

data/lib/mongoid_erd.rb

@@ -134,9 +134,9 @@ class MongoidErd
           field_.name = "#{rel_} (as #{as_})" if as_
           self.parse_erd field_, line # parse erd attr and label
           crt_model.fields << field_ 
-          unless %w[belongs_to embedded_in].include? rel_
-            field_.edge = [name_, '', {label: rel_}]
-          end
+          #if %w[belongs_to embedded_in embeds_one has_one].include? rel_
+          field_.edge = [name_, '', {label: rel_, arrowhead: 'onormal'}]
+          #end
         end
         
         # common extension field
@@ -154,6 +154,15 @@ class MongoidErd
           self.parse_erd field_, line # parse erd attr and label
           crt_model.fields << field_ 
         end
+
+        if /\s*as_enum\s+\:(?<name_>\w+)\s*\,\s*(?<enum_>[^#]+)/ =~ line
+          field_ = Fields.new
+          field_.name = name_
+          field_.type = "[ENUM] " + enum_ 
+          self.parse_erd field_, line # parse erd attr and label
+          crt_model.fields << field_ 
+        end
+
       end # open and parse one file
 
       # assign attributes at the last moment

data/mongoid-erd.gemspec

@@ -1,6 +1,6 @@
 $LOAD_PATH.unshift File.expand_path('../lib', __FILE__)
 
-Gem::Specification.new 'mongoid-erd', '0.0.3' do |s|
+Gem::Specification.new 'mongoid-erd', '0.0.5' do |s|
   s.executables       << "merd"
   s.description       = "Create model diagram graph (ERD graph) in graphviz's dot language"
   s.summary           = "Mongoid ERD diagram creator"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongoid-erd
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
   prerelease: 
 platform: ruby
 authors: