require 'yell'
require 'toadhopper'
class AirbrakeAdapter < Yell::Adapters::Base
  setup do |options|
    @env = options[:environment]
    @airbrake = Toadhopper(options[:api_key])
  end

  write do |event|
    if event.message.respond_to?(:backtrace)
      @airbrake.post! event.message, :framework_env => @env
    else
      begin
        raise event.message
      rescue => e
        @airbrake.post! e, :framework_env => @env
      end
    end
  end
end

# Register the newly written adapter with Yell
Yell::Adapters.register :airbrake, AirbrakeAdapter