module MassiveRecord
  module Rails
    class Railtie < ::Rails::Railtie
      config.massive_record = ActiveSupport::OrderedOptions.new

      initializer "massive_record.logger" do
        MassiveRecord::ORM::Base.logger = ::Rails.logger
      end

      # Make Rails handle RecordNotFound correctly in production
      initializer "massive_record.action_dispatch" do
        ActiveSupport.on_load :action_controller do
          ActionDispatch::ShowExceptions.rescue_responses.update('MassiveRecord::ORM::RecordNotFound' => :not_found)
        end
      end

      # Expose database runtime to controller for logging.
      initializer "massive_record.log_runtime" do
        require "massive_record/rails/controller_runtime"
        ActiveSupport.on_load(:action_controller) do |app|
          include MassiveRecord::Rails::ControllerRuntime
        end
      end

      # Insert IdentityMap's middleware if enabled
      initializer "massive_record.identity_map" do |app|
        if config.massive_record.delete(:identity_map)
          config.app_middleware.insert_after "::ActionDispatch::Callbacks", "MassiveRecord::ORM::IdentityMap::Middleware"
        end
      end

      initializer "massive_record.time_zone_awareness" do
        ActiveSupport.on_load(:massive_record) do
          self.time_zone_aware_attributes = true
          self.default_timezone = :utc
        end
      end

      initializer "massive_record.set_configs" do |app|
        ActiveSupport.on_load(:massive_record) do
          app.config.massive_record.each { |k,v| send("#{k}=", v) }
        end
      end

      config.after_initialize do
        ActiveSupport.on_load(:massive_record) do
          instantiate_observers

          if ::Rails::VERSION::MAJOR >= 3 && ::Rails::VERSION::MINOR >= 1
            ActionDispatch::Reloader.to_prepare do
              MassiveRecord::ORM::Base.instantiate_observers
            end
          else
            ActionDispatch::Callbacks.to_prepare(:massive_record_instantiate_observers) do
              MassiveRecord::ORM::Base.instantiate_observers
            end
          end
        end
      end
    end
  end
end