Contents

# encoding: UTF-8
module MongoMapper
  module Plugins
    module Dirty
      extend ActiveSupport::Concern

      include ::ActiveModel::Dirty

      module InstanceMethods
        def initialize(*)
          # never register initial id assignment as a change
          super.tap { changed_attributes.delete('_id') }
        end

        def initialize_from_database(*)
          super.tap { changed_attributes.clear }
        end

        def save(*)
          clear_changes { super }
        end

        def save!(*)
          clear_changes { super }
        end

        def reload(*)
          super.tap { clear_changes }
        end

        protected

        def attribute_method?(attr)
          # This overrides ::ActiveSupport::Dirty#attribute_method? to allow attributes to be any key
          # in the attributes hash ( default ) or any key defined on the model that may not yet have
          # had a value stored in the attributes collection.
          super || key_names.include?(attr)
        end

        def clear_changes
          previous = changes
          (block_given? ? yield : true).tap do |result|
            unless result == false #failed validation; nil is OK.
              @previously_changed = previous
              changed_attributes.clear
            end
          end
        end

        private

        def write_key(key, value)
          key = key.to_s
          old = read_key(key)
          attribute_will_change!(key) if attribute_should_change?(key, old, value)
          changed_attributes.delete(key) unless attribute_value_changed?(key, attribute_was(key), value)
          super(key, value)
        end

        def attribute_should_change?(key, old, value)
          attribute_changed?(key) == false && attribute_value_changed?(key, old, value)
        end

        def attribute_value_changed?(key_name, old, value)
          value = nil if keys[key_name.to_s].number? && value.blank?
          old != value
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
mongo_mapper-0.9.2	lib/mongo_mapper/plugins/dirty.rb
mongo_mapper-0.9.1	lib/mongo_mapper/plugins/dirty.rb