module CassandraObject
  module Dirty
    extend ActiveSupport::Concern
    include ActiveModel::Dirty

    # Attempts to +save+ the record and clears changed attributes if successful.
    def save(*) #:nodoc:
      if status = super
        @previously_changed = changes
        @changed_attributes.clear
      end
      status
    end

    # Attempts to <tt>save!</tt> the record and clears changed attributes if successful.
    def save!(*) #:nodoc:
      super.tap do
        @previously_changed = changes
        @changed_attributes.clear
      end
    end

    # <tt>reload</tt> the record and clears changed attributes.
    def reload
      super.tap do
        @previously_changed.try :clear
        @changed_attributes.clear
      end
    end

    def write_attribute(name, value)
      name = name.to_s
      unless attribute_changed?(name)
        old = read_attribute(name)
        changed_attributes[name] = old if old != value
      end
      super
    end
  end
end