lib/toadhopper/capistrano.rb in toadhopper-2.0 vs lib/toadhopper/capistrano.rb in toadhopper-2.1

@@ -7,13 +7,19 @@
 
   namespace :deploy do
     desc 'Notify Airbrake of the deployment'
     task :notify_airbrake, :except => {:no_release => true} do
       framework_env = fetch(:rails_env, fetch(:airbrake_env, 'production'))
-      api_key = fetch(:airbrake_api_key) || fetch(:hoptoad_api_key)
+      api_key = fetch(:hoptoad_api_key, nil) || fetch(:airbrake_api_key)
+      host_options = {
+        :notify_host  => fetch(:airbrake_notify_host, nil),
+        :error_url    => fetch(:airbrake_error_url, nil),
+        :deploy_url   => fetch(:airbrake_deploy_url, nil),
+        :transport    => fetch(:airbrake_transport, nil),
+      }
       puts 'Notifying Airbrake of deploy'
-      options = {:framework_env => framework_env, :scm_revision => current_revision, :scm_repository => repository}
-      Toadhopper(api_key).deploy!(options)
+      deploy_options = {:framework_env => framework_env, :scm_revision => current_revision, :scm_repository => repository}
+      Toadhopper.new(api_key, host_options).deploy!(deploy_options)
       puts 'Airbrake notification complete'
     end
   end
 end