diff --git a/lib/impressionist/engine.rb b/lib/impressionist/engine.rb index c02231a..ffa1244 100644 --- a/lib/impressionist/engine.rb +++ b/lib/impressionist/engine.rb @@ -3,8 +3,15 @@ require "rails" module Impressionist class Engine < Rails::Engine - initializer 'impressionist.extend_ar' do + initializer 'impressionist.extend_ar' do |app| ActiveRecord::Base.extend Impressionist::Impressionable end + + initializer 'impressionist.controller' do + ActiveSupport.on_load(:action_controller) do + include ImpressionistController::InstanceMethods + extend ImpressionistController::ClassMethods + end + end end end \ No newline at end of file