@@ -4,16 +4,17 @@ gem 'dm-core', '>=0.9.10' require 'dm-core' module DataMapper - module Userstamp - module Stamper def self.included(base) base.extend(ClassMethods) end - + module ClassMethods - + def userstamp_class + User + end + def current_user=(user) Thread.current["#{self.to_s.downcase}_#{self.object_id}_stamper"] = user end @@ -21,15 +22,16 @@ module DataMapper def current_user Thread.current["#{self.to_s.downcase}_#{self.object_id}_stamper"] end - end - end - - + + def self.userstamp_class + User + end + USERSTAMP_PROPERTIES = { - :created_by_id => lambda { |r| r.created_by_id = User.current_user.id if User.current_user && r.new_record? && r.created_by_id.nil? }, - :updated_by_id => lambda { |r| r.updated_by_id = User.current_user.id if User.current_user} + :created_by_id => lambda { |r| r.created_by_id = userstamp_class.current_user.id if userstamp_class.current_user && r.new_record? && r.created_by_id.nil? }, + :updated_by_id => lambda { |r| r.updated_by_id = userstamp_class.current_user.id if userstamp_class.current_user} } def self.included(model) @@ -44,6 +46,6 @@ module DataMapper end end end - + Resource::append_inclusions Userstamp end \ No newline at end of file lib/dm-userstamp.rb 3f7cc0d20465c49945fe417dcce6cac59d123b74 Michael S. Klishin michael@novemberain.com http://github.com/rlivsey/dm-userstamp/commit/e01eee51bbf65c8cc91a44b79b3406f5dfa645bf e01eee51bbf65c8cc91a44b79b3406f5dfa645bf 2009-06-12T03:40:59-07:00 2009-03-25T14:42:53-07:00 Let resources override userstamp class Signed-off-by: Richard Livsey <github@livsey.org> d2ed3eb0b644134f8aabed4c17eac48bf9b88c2b Richard Livsey github@livsey.org