# frozen_string_literal: true

module Marlowe
  class Railtie < Rails::Railtie # :nodoc:
    initializer "marlowe.configure_rails_initialization" do
      config = app.config

      opts = {
        header: config&.marlowe_header || config&.marlowe_correlation_header,
        handler: config&.marlowe_request_id_handler,
        return: config&.marlowe_return_request_id,
        action_dispatch: config&.marlowe_replace_action_dispatch_request_id
      }.compact

      if opts[:action_dispatch]
        app.middleware.insert_before ActionDispatch::RequestId, Marlowe::Middleware, opts
        app.middleware.delete ActionDispatch::RequestId
      else
        app.middleware.insert_before Rails::Rack::Logger, Marlowe::Middleware, opts
      end
    end

    def app # :nodoc:
      Rails.application
    end
  end
end