Set current_user on all models in before_* callback

gistfile1.rb

# lib/user_stamp.rb
module UserStamp
  def self.included(klass)
    callbacks = [:before_validation, :before_save, :before_create, :before_update]

    callbacks.each do |callback|
      klass.send(callback) do
        return unless respond_to?(:updated_by) || updated_by.nil?
	self.updated_by = Thread.current[:current_user]
      end
    end
  end
end

# config/initializers/add_user_stamp.rb
require 'user_stamp'

Dir[Rails.root + "app/models/**/*.rb"].each do |path|
  require path

  klass = path.gsub(/^.+\//,'').gsub(/\.rb/,'').camelize.constantize
  klass.class_eval { include UserStamp }
end

# app/controllers/application_controller.rb
class ApplicationController < ActionController::Base
  protect_from_forgery
  before_filter :set_current_user

  private

  def current_user
    # ...
  end

  def set_current_user
    Thread.current[:current_user] = current_user
  end
end