module Lograge

# File lib/lograge.rb, line 141
def attach_to_action_controller
  Lograge::RequestLogSubscriber.attach_to :action_controller
end

# File lib/lograge.rb, line 45
def before_format(data, payload)
  result = nil
  result = @@before_format.call(data, payload) if @@before_format
  result || data
end

# File lib/lograge.rb, line 31
def custom_options(event)
  if @@custom_options.respond_to?(:call)
    @@custom_options.call(event)
  else
    @@custom_options
  end
end

# File lib/lograge.rb, line 176
def disable_rack_cache_verbose_output
  application.config.action_dispatch.rack_cache[:verbose] = false if rack_cache_hashlike?(application)
end

# File lib/lograge.rb, line 159
def extend_base_controller_class(klass)
  append_payload_method = klass.instance_method(:append_info_to_payload)
  custom_payload_method = lograge_config.custom_payload_method

  klass.send(:define_method, :append_info_to_payload) do |payload|
    append_payload_method.bind(self).call(payload)
    payload[:custom_payload] = custom_payload_method.call(self)
  end
end

# File lib/lograge.rb, line 74
def ignore(test)
  ignore_tests.push(test) if test
end

# File lib/lograge.rb, line 82
def ignore?(event)
  ignore_tests.any? { |ignore_test| ignore_test.call(event) }
end

# File lib/lograge.rb, line 63
def ignore_actions(actions)
  ignore(lambda do |event|
           params = event.payload
           Array(actions).include?("#{params[:controller]}##{params[:action]}")
         end)
end

# File lib/lograge.rb, line 78
def ignore_nothing
  @ignore_tests = []
end

# File lib/lograge.rb, line 70
def ignore_tests
  @ignore_tests ||= []
end

# File lib/lograge.rb, line 180
def keep_original_rails_log
  return if lograge_config.keep_original_rails_log

  require 'lograge/rails_ext/rack/logger'
  Lograge.remove_existing_log_subscriptions
end

# File lib/lograge.rb, line 204
def lograge_config
  application.config.lograge
end

# File lib/lograge.rb, line 187
def rack_cache_hashlike?(app)
  app.config.action_dispatch.rack_cache && app.config.action_dispatch.rack_cache.respond_to?(:[]=)
end

# File lib/lograge.rb, line 97
def remove_existing_log_subscriptions
  ActiveSupport::LogSubscriber.log_subscribers.each do |subscriber|
    case subscriber
    when ActionView::LogSubscriber
      unsubscribe(:action_view, subscriber)
    when ActionController::LogSubscriber
      unsubscribe(:action_controller, subscriber)
    end
  end
end

# File lib/lograge.rb, line 137
def set_formatter
  Lograge.formatter = lograge_config.formatter || Lograge::Formatters::KeyValue.new
end

# File lib/lograge.rb, line 132
def set_ignores
  Lograge.ignore_actions(lograge_config.ignore_actions)
  Lograge.ignore(lograge_config.ignore_custom)
end

# File lib/lograge.rb, line 169
def set_lograge_log_options
  Lograge.logger = lograge_config.logger
  Lograge.custom_options = lograge_config.custom_options
  Lograge.before_format = lograge_config.before_format
  Lograge.log_level = lograge_config.log_level || :info
end

# File lib/lograge.rb, line 119
def setup(app)
  self.application = app
  disable_rack_cache_verbose_output
  keep_original_rails_log

  attach_to_action_controller
  set_lograge_log_options
  setup_custom_payload
  support_deprecated_config # TODO: Remove with version 1.0
  set_formatter
  set_ignores
end

# File lib/lograge.rb, line 145
def setup_custom_payload
  return unless lograge_config.custom_payload_method.respond_to?(:call)

  base_controller_classes = Array(lograge_config.base_controller_class)
  base_controller_classes.map! { |klass| klass.try(:constantize) }
  if base_controller_classes.empty?
    base_controller_classes << ActionController::Base
  end

  base_controller_classes.each do |base_controller_class|
    extend_base_controller_class(base_controller_class)
  end
end

# File lib/lograge.rb, line 194
def support_deprecated_config
  return unless lograge_config.log_format

  legacy_log_format = lograge_config.log_format
  warning = 'config.lograge.log_format is deprecated. Use config.lograge.formatter instead.'
  ActiveSupport::Deprecation.warn(warning, caller)
  legacy_log_format = :key_value if legacy_log_format == :lograge
  lograge_config.formatter = "Lograge::Formatters::#{legacy_log_format.to_s.classify}".constantize.new
end

# File lib/lograge.rb, line 108
def unsubscribe(component, subscriber)
  events = subscriber.public_methods(false).reject { |method| method.to_s == 'call' }
  events.each do |event|
    ActiveSupport::Notifications.notifier.listeners_for("#{event}.#{component}").each do |listener|
      if listener.instance_variable_get('@delegate') == subscriber
        ActiveSupport::Notifications.unsubscribe listener
      end
    end
  end
end