require 'justlogging'

module Justlogging
  class Railtie < Rails::Railtie

    initializer "justlogging.configure_rails_initialization" do |app|
      app.middleware.insert_before ActionDispatch::RemoteIp,
        Justlogging::Middleware

      Justlogging.subscriber = ActiveSupport::Notifications.subscribe do |*args|
        if Justlogging::Transaction.current
          event = ActiveSupport::Notifications::Event.new(*args)
          if event.name == 'process_action.action_controller'
            Justlogging::Transaction.current.set_log_entry(event)
          else
            Justlogging::Transaction.current.add_event(event)
          end
        end
      end
    end

  end
end