fixing heroku rails generator

lib/rails/generators/airbrake/airbrake_generator.rb

@@ -49,7 +49,7 @@ def api_key_expression
     s = if options[:api_key]
       "'#{options[:api_key]}'"
     elsif options[:heroku]
-      "ENV['AIRBRAKE_API_KEY']"
+      heroku_api_key
     end
   end
 
@@ -67,7 +67,7 @@ def determine_api_key
       exit 1
     else
       puts "... Done."
-      puts "Heroku's Airbrake API Key is '#{ENV['AIRBRAKE_API_KEY']}'"
+      puts "Heroku's Airbrake API Key is #{ENV['AIRBRAKE_API_KEY']}"
     end
   end
 
@@ -77,21 +77,19 @@ def heroku_var(var,app_name = nil)
   end
 
   def heroku_api_key
-    heroku_var("AIRBRAKE_API_KEY",options[:app]).split.find {|x| x unless x.blank?}
+    "'#{`heroku config:get AIRBRAKE_API_KEY`.gsub("\n","")}'"
   end
 
   def heroku?
-    options[:heroku] ||
-      system("grep AIRBRAKE_API_KEY config/initializers/airbrake.rb") ||
-      system("grep AIRBRAKE_API_KEY config/environment.rb")
+    options[:heroku] || heroku_api_key.length == 32
   end
 
   def api_key_configured?
     File.exists?('config/initializers/airbrake.rb')
   end
 
   def test_airbrake
-    puts run("rake airbrake:test")
+    puts run("airbrake raise")
   end
 
   def plugin_is_present?