Make 'status' dynamic with HEROKU_HOST

heroku · May 17, 2012 · 8584df1 · 8584df1
1 parent fe54493
commit 8584df1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/heroku/command/status.rb b/lib/heroku/command/status.rb
@@ -4,12 +4,15 @@
 #
 class Heroku::Command::Status < Heroku::Command::Base
 
+
+
   # status
   #
   # display current status of Heroku platform
   #
   def index
-    status = json_decode(heroku.get("https://status.heroku.com/status.json"))
+    heroku_host = ENV['HEROKU_HOST'] || "heroku.com"
+    status = json_decode(heroku.get("https://status.#{heroku_host}/status.json"))
 
     display('')
     if status.values.all? {|value| value == 'green'}
@@ -18,7 +21,7 @@ def index
       status.each do |key, value|
         display("#{key}: #{value}")
       end
-      response = heroku.xml(heroku.get("https://status.heroku.com/feed"))
+      response = heroku.xml(heroku.get("https://status.#{heroku_host}/feed"))
       entry = response.elements.to_a("//entry").first
       display('')
       display(entry.elements['title'].text)