require 'action_dispatch'
require 'exception_notifier/notifier'

class ExceptionNotifier
  class << self
    def default_ignore_exceptions
      [].tap do |exceptions|
        exceptions << ActiveRecord::RecordNotFound if defined? ActiveRecord
        exceptions << AbstractController::ActionNotFound if defined? AbstractController
        exceptions << ActionController::RoutingError if defined? ActionController
      end
    end
  
    def background_options
      @background_options ||= Rails.application.middleware.find { |klass| klass == ExceptionNotifier }.args.first rescue {}
    end
  
    def with(options = {}, &block)
      block.call
    rescue Exception => exception
      unless Array.wrap(background_options[:ignore_exceptions]).include?(exception.class)
        Notifier.background_exception_notification(exception, options).deliver
      end
      raise exception
    end
  end

  def initialize(app, options = {})
    @app, @options = app, options
    @options[:ignore_exceptions] ||= self.class.default_ignore_exceptions
  end

  def call(env)
    @app.call(env)
  rescue Exception => exception
    options = (env['exception_notifier.options'] ||= {})
    options.reverse_merge!(@options)

    unless Array.wrap(options[:ignore_exceptions]).include?(exception.class)
      Notifier.exception_notification(env, exception).deliver
      env['exception_notifier.delivered'] = true
    end

    raise exception
  end
end