#
# Copyright (C) 2012 mkristian
#
# Permission is hereby granted, free of charge, to any person obtaining a copy of
# this software and associated documentation files (the "Software"), to deal in
# the Software without restriction, including without limitation the rights to
# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
# the Software, and to permit persons to whom the Software is furnished to do so,
# subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#
require 'ixtlan/optimistic/stale_check'
module Ixtlan
  module Optimistic
    module DataMapper

      def self.included(base)
        base.extend StaleCheck
        base.class_eval do

          def self.optimistic_get(updated_at, *args) 
            result = get( *args )
            if result
              __check( updated_at )
              __check_stale( updated_at, result )
            end
          end
 
          def self.optimistic_get!(updated_at, *args)
            result = get!( *args )
            __check( updated_at )
            __check_stale( updated_at, result )
          end
        end
      end

      ::DataMapper::Model.append_inclusions( Ixtlan::Optimistic::DataMapper )
      ::DataMapper::Associations::OneToMany::Collection.send( :include,
                                                              Ixtlan::Optimistic::DataMapper )
      ::DataMapper::Associations::ManyToMany::Collection.send( :include,
                                                               Ixtlan::Optimistic::DataMapper )
    end
  end
end