require 'rubygems'

# gem 'dm-core', '=0.9.6'
require 'dm-core'

#
# Stolen from http://github.com/sam/dm-more/tree/master/dm-timestamps/lib/dm-timestamps.rb
#

module DataMapper
  module Timestamp
    TIMESTAMP_PROPERTIES = {
      :updated_at => lambda { |r| r.updated_at = DateTime.now },
      :updated_on => lambda { |r| r.updated_on = Date.today   },
      :updated_by => lambda { |r| r.updated_by = IMW::USER_INFO[:id] },
      :created_at => lambda { |r| r.created_at = DateTime.now            if r.new_record? && r.created_at.nil? },
      :created_on => lambda { |r| r.created_on = Date.today              if r.new_record? && r.created_on.nil?},
      :created_by => lambda { |r| r.created_by = IMW::USER_INFO[:id]     if r.new_record? && r.created_by.blank?},
    }

    def self.included(model)
      model.before :save, :set_timestamp_properties
    end

    private

    def set_timestamp_properties
      if dirty?
        self.class.properties.slice(*TIMESTAMP_PROPERTIES.keys).compact.each do |property|
          TIMESTAMP_PROPERTIES[property.name][self] unless attribute_dirty?(property.name)
        end
      end
    end
  end # module Timestamp

  Resource::append_inclusions Timestamp
end