From a57a2bcf4a2c29519d553277e4439790ca443cc7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rafael=20Mendon=C3=A7a=20Fran=C3=A7a?=
 <rafaelmfranca@gmail.com>
Date: Mon, 20 Jan 2014 21:59:20 -0200
Subject: Make enum feature work with dirty methods

To make this possible we have to override the save_changed_attribute
hook.
---
 activerecord/lib/active_record/enum.rb | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

(limited to 'activerecord/lib/active_record/enum.rb')

diff --git a/activerecord/lib/active_record/enum.rb b/activerecord/lib/active_record/enum.rb
index 3deb2d65f8..06e87cf854 100644
--- a/activerecord/lib/active_record/enum.rb
+++ b/activerecord/lib/active_record/enum.rb
@@ -63,6 +63,12 @@ module ActiveRecord
   #
   #   Conversation.where("status <> ?", Conversation.statuses[:archived])
   module Enum
+    DEFINED_ENUMS = [] # :nodoc:
+
+    def enum_attribute?(attr_name) # :nodoc:
+      DEFINED_ENUMS.include?(attr_name.to_sym)
+    end
+
     def enum(definitions)
       klass = self
       definitions.each do |name, values|
@@ -70,6 +76,8 @@ module ActiveRecord
         enum_values = ActiveSupport::HashWithIndifferentAccess.new
         name        = name.to_sym
 
+        DEFINED_ENUMS.unshift name
+
         # def self.statuses statuses end
         klass.singleton_class.send(:define_method, name.to_s.pluralize) { enum_values }
 
@@ -114,7 +122,16 @@ module ActiveRecord
     private
       def _enum_methods_module
         @_enum_methods_module ||= begin
-          mod = Module.new
+          mod = Module.new do
+            def save_changed_attribute(attr_name, value)
+              if self.class.enum_attribute?(attr_name)
+                old = clone_attribute_value(:read_attribute, attr_name)
+                changed_attributes[attr_name] = self.class.public_send(attr_name.pluralize).key old
+              else
+                super
+              end
+            end
+          end
           include mod
           mod
         end
-- 
cgit v1.2.3