enumerate_it 1.1.1 → 1.2.0

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    enumerate_it (1.1.1)
+    enumerate_it (1.2.0)
       activesupport (>= 3.0.0)
 
 GEM

data/README.rdoc

@@ -83,7 +83,7 @@ This will create some nice stuff:
 
 * You can retrieve the symbol used to declare a specific enumeration value:
 
-   RelationshipStatus.key_for(RelationshioStatus::MARRIED) # :married
+   RelationshipStatus.key_for(RelationshipStatus::MARRIED) # :married
 
 * You can iterate over the list of the enumeration's values:
 
@@ -197,6 +197,44 @@ This will create:
   p.relationship_status_married? #=> true
   p.relationship_status_divoced? #=> false
 
+* You can define polymorphic behavior for the enum values, so you can define a class for each of
+ them:
+
+  class RelationshipStatus < EnumerateIt::Base
+    associate_values :married, :single
+
+    class Married
+      def saturday_night
+        "At home with the kids"
+      end
+    end
+
+    class Single
+      def saturday_night
+        "Party Hard!"
+      end
+    end
+  end
+
+  class Person < ActiveRecord::Base
+    has_enumeration_for :relationship_status, :with => RelationshipStatus, :create_helpers => { :polymorphic => true }
+  end
+
+  p = Person.new
+  p.relationship_status = RelationshipStatus::MARRIED
+  p.relationship_status_object.saturday_night # => "At home with the kids"
+
+  p.relationship_status = RelationshipStatus::SINGLE
+  p.relationship_status_object.saturday_night # => "Party Hard!"
+
+  You can also change the suffix '_object', using the :suffix option:
+
+  class Person < ActiveRecord::Base
+    has_enumeration_for :relationship_status, :with => RelationshipStatus, :create_helpers => { :polymorphic => { :suffix => "_mode" } }
+  end
+
+  p.relationship_status_mode.saturday_night
+
 * The :create_helpers also creates some mutator helper methods, that can be used to change the attribute's value.
 
    class Person < ActiveRecord::Base

data/lib/enumerate_it.rb

@@ -170,6 +170,44 @@
 #  p.relationship_status_married? #=> true
 #  p.relationship_status_divoced? #=> false
 #
+# - You can define polymorphic behavior for the enum values, so you can define a class for each of
+# them:
+#
+# class RelationshipStatus < EnumerateIt::Base
+#   associate_values :married, :single
+#
+#   class Married
+#     def saturday_night
+#       "At home with the kids"
+#     end
+#   end
+#
+#   class Single
+#     def saturday_night
+#       "Party Hard!"
+#     end
+#   end
+# end
+#
+#  class Person < ActiveRecord::Base
+#    has_enumeration_for :relationship_status, :with => RelationshipStatus, :create_helpers => { :polymorphic => true }
+#  end
+#
+#  p = Person.new
+#  p.relationship_status = RelationshipStatus::MARRIED
+#  p.relationship_status_object.saturday_night # => "At home with the kids"
+#
+#  p.relationship_status = RelationshipStatus::SINGLE
+#  p.relationship_status_object.saturday_night # => "Party Hard!"
+#
+# You can also change the suffix '_object', using the :suffix option:
+#
+#  class Person < ActiveRecord::Base
+#    has_enumeration_for :relationship_status, :with => RelationshipStatus, :create_helpers => { :polymorphic => { :suffix => "_mode" } }
+#  end
+#
+#  p.relationship_status_mode.saturday_night
+#
 # - If you pass the :create_scopes option as 'true', it will create a scope method for each enumeration option (this option defaults to false):
 #
 #   class Person < ActiveRecord::Base

data/lib/enumerate_it/class_methods.rb

@@ -12,6 +12,7 @@ module EnumerateIt
       if options[:create_helpers]
         create_helper_methods options[:with], attribute, options[:create_helpers]
         create_mutator_methods options[:with], attribute, options[:create_helpers]
+        create_polymorphic_methods options[:with], attribute, options[:create_helpers]
       end
 
       if options[:create_scopes]
@@ -66,6 +67,21 @@ module EnumerateIt
       end
     end
 
+    def create_polymorphic_methods(klass, attribute_name, helpers)
+      return unless helpers.is_a?(Hash) && helpers[:polymorphic]
+      options = helpers[:polymorphic]
+      suffix = options.is_a?(Hash) && options[:suffix]
+      suffix ||= "_object"
+
+      class_eval do
+        define_method "#{attribute_name}#{suffix}" do
+          value = self.public_send(attribute_name)
+
+          klass.const_get(klass.key_for(value).to_s.camelize).new if value
+        end
+      end
+    end
+
     def define_enumeration_class(attribute, options)
       if options[:with].nil?
         inner_enum_class_name = attribute.to_s.camelize.to_sym

data/lib/enumerate_it/version.rb

@@ -1,3 +1,3 @@
 module EnumerateIt
-  VERSION = "1.1.1"
+  VERSION = "1.2.0"
 end

data/spec/enumerate_it_spec.rb

@@ -185,6 +185,52 @@ describe EnumerateIt do
         target.foobar.should == TestEnumeration::VALUE_3
       end
     end
+
+    context "with :polymorphic option" do
+      before :each do
+        class Polymorphic
+          extend EnumerateIt
+          attr_accessor :foo
+          has_enumeration_for :foo, :with => PolymorphicEnum, :create_helpers => { :polymorphic => true }
+        end
+      end
+
+      it "calls methods on the enum constants' objects" do
+        target = Polymorphic.new
+        target.foo = PolymorphicEnum::NORMAL
+
+        target.foo_object.print("Gol").should == "I'm Normal: Gol"
+
+        target.foo = PolymorphicEnum::CRAZY
+
+        target.foo_object.print("Gol").should == "Whoa!: Gol"
+      end
+
+      it "returns nil if foo is not set" do
+        target = Polymorphic.new
+
+        target.foo_object.should be_nil
+      end
+
+      context "and :suffix" do
+        before :each do
+          class Polymorphic
+            has_enumeration_for :foo, :with => PolymorphicEnum, :create_helpers => { :polymorphic => { :suffix => "_strategy" } }
+          end
+        end
+
+        it "calls methods on the enum constants' objects" do
+          target = Polymorphic.new
+          target.foo = PolymorphicEnum::NORMAL
+
+          target.foo_strategy.print("Gol").should == "I'm Normal: Gol"
+
+          target.foo = PolymorphicEnum::CRAZY
+
+          target.foo_strategy.print("Gol").should == "Whoa!: Gol"
+        end
+      end
+    end
   end
 
   describe "using the :create_scopes option" do

data/spec/support/test_classes.rb

@@ -33,6 +33,22 @@ class Foobar < EnumerateIt::Base
   )
 end
 
+class PolymorphicEnum < EnumerateIt::Base
+  associate_values :normal, :crazy
+
+  class Normal
+    def print(msg)
+      "I'm Normal: #{msg}"
+    end
+  end
+
+  class Crazy
+    def print(msg)
+      "Whoa!: #{msg}"
+    end
+  end
+end
+
 class BaseClass
   extend EnumerateIt
   has_enumeration_for :foobar, :with => TestEnumeration

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: enumerate_it
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-03 00:00:00.000000000 Z
+date: 2013-08-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -158,7 +158,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -817288258794681256
+      hash: 1153526174871300187
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -167,10 +167,10 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -817288258794681256
+      hash: 1153526174871300187
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.25
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: Ruby Enumerations