enum_field 0.1.4 → 1.0.0

data/README.rdoc

@@ -80,6 +80,37 @@ You have some +AR+ like methods in enum classes
   PhoneType.find(2) == PhoneType.commercial
   PhoneType.find(123456)  # will raise
 
+It also mimics has_many :through behavior, for cases such as:
+
+  class Role
+    define_enum do |builder|
+      builder.member :admin
+      builder.member :manager
+      builder.member :employee
+    end
+  end
+
+  class User
+    has_many_enumerated_attributes :roles, :through  => UserRole
+  end
+
+  class UserRole < ActiveRecord::Base
+    belongs_to :user
+    enumerated_attribute :role
+  end
+
+  user = User.create
+  user.role = [Role.manager, Role.admin]
+  user.roles.include?(Role.admin)         #will be true
+  user.roles.include?(Role.manager)       #will be true
+  user.roles.include?(Role.employee)      #will be false
+  user.role_ids.include?(Role.manager.id) #will be true
+  user.role_ids = [Role.employee.id]
+  user.roles.include?(Role.employee)      #will be true
+  user.roles.include?(Role.admin)         #will be false
+
+
+
 == REQUIREMENTS:
 
 * activerecord

data/lib/enum_field/enumerated_attribute.rb

@@ -11,6 +11,24 @@ module EnumField
 # These added methods expect an object of class Role, and Role should provide a find class method.
 # You could get this by using Enumerated mixin
 #
+# Similar to has_many :through in AR, it creates reader and writter methods to get enumerated atributes going through an
+# intermediate model.
+# For instance:
+# <tt>
+# class User
+#   has_many_enumerated_attributes :roles, :through  => UserRole
+# end
+#
+# class UserRole < ActiveRecord::Base
+#   belongs_to :user
+#   enumerated_attribute :role
+# end
+# </tt>
+# This assumes a Role class, the UserRole AR class is there just to persist the one-to-many relationship.
+# This will define roles, roles=, role_ids and role_ids= methods in User class
+
+
+
 module EnumeratedAttribute
   # Define an enumerated field of the AR class
   # * +name_attribute+: the name of the field that will be added, for instance +role+
@@ -30,6 +48,44 @@ module EnumeratedAttribute
       write_attribute(id_attribute, value ? value.id : nil)
     end
   end
+
+  alias belongs_to_enumerated_attribute enumerated_attribute
+
+  # Define a one-to-many association between an AR class and the enumerated
+  # * +association+: the name of the one-to-many association, for instance +roles+
+  # * +options+: Valid options are:
+  #    * +through+ : the name of the AR class needed to persist the one-to-many association.
+  #       Defaults to AR class in camelcase form plus enumerated class in camelcase form.
+  #    * +class+:    the enumerated class that will be instantiated +n+ times when +association+ method is called.
+  #       Defaults to +association+ in singular camelcase form.
+  def has_many_enumerated_attributes(association, options = {})
+    enum_attr = association.to_s.singularize
+    klass     = options[:class]   || enum_attr.camelcase.constantize
+    through   = options[:through] || (self.name + klass.name)
+    self_attribute = self.name.demodulize.underscore
+    association_ids = association.to_s.singularize + '_ids'
+    has_many_aux = through.demodulize.underscore.pluralize
+
+    has_many has_many_aux, {:class_name => through, :dependent => :destroy}
+
+    define_method(association) do
+      self.send(has_many_aux).map(&enum_attr.to_sym)
+    end
+
+    define_method(association.to_s + '=') do |values|
+      self.send(has_many_aux + '=', values.map{|g| through.constantize.new(self_attribute => self, enum_attr => g)})
+    end
+
+    define_method(association_ids) do
+       self.send(association).map(&:id)
+    end
+
+    define_method(association_ids + '=') do |values|
+      self.send(has_many_aux + '=', values.map{|g| g.to_i unless g.blank?}.compact.map{|g_id| through.constantize.new(self_attribute => self, enum_attr + '_id' => g_id) })
+    end
+
+  end
+
 end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: enum_field
 version: !ruby/object:Gem::Version 
-  version: 0.1.4
+  version: 1.0.0
 platform: ruby
 authors: 
 - "Sebasti\xC3\xA1n Bernardo Galkin"