class Airbrake::AirbrakeLogger

def airbrake_level=(level)
  if level < Logger::WARN
    raise "Airbrake severity level #{level} is not allowed. " \
          "Minimum allowed level is #{Logger::WARN}"
  end
  @airbrake_level = level
end

def error(progname = nil, &block)
  notify_airbrake(Logger::ERROR, progname)
  super
end

def fatal(progname = nil, &block)
  notify_airbrake(Logger::FATAL, progname)
  super
end

def initialize(logger)
  super
  __setobj__(logger)
  @airbrake_notifier = Airbrake
  self.level = logger.level
end

def level=(value)
  self.airbrake_level = value < Logger::WARN ? Logger::WARN : value
  super
end

def normalize_severity(severity)
  (case severity
   when Logger::WARN then 'warning'
   when Logger::ERROR, Logger::UNKNOWN then 'error'
   when Logger::FATAL then 'critical'
   else
     raise "Unknown airbrake severity: #{severity}"
   end).freeze
end

def notify_airbrake(severity, progname)
  return if severity < @airbrake_level || !@airbrake_notifier
  @airbrake_notifier.notify(progname) do |notice|
    # Get rid of unwanted internal Logger frames. Examples:
    # * /ruby-2.4.0/lib/ruby/2.4.0/logger.rb
    # * /gems/activesupport-4.2.7.1/lib/active_support/logger.rb
    backtrace = notice[:errors].first[:backtrace]
    notice[:errors].first[:backtrace] =
      backtrace.drop_while { |frame| frame[:file] =~ %r{/logger.rb\z} }
    notice[:context][:component] = 'log'
    notice[:context][:severity] = normalize_severity(severity)
  end
end

def unknown(progname = nil, &block)
  notify_airbrake(Logger::UNKNOWN, progname)
  super
end

def warn(progname = nil, &block)
  notify_airbrake(Logger::WARN, progname)
  super
end