lib/heroku_deploy/tasks.rb in heroku_deploy-0.0.13 vs lib/heroku_deploy/tasks.rb in heroku_deploy-0.0.14

@@ -78,16 +78,20 @@
 
     puts ""
     puts "Creating #{staging_app} Heroku app"
     puts ""
     `heroku app:create #{staging_app}`
+    `heroku config:add RACK_ENV=staging --app #{staging_app}`
+    `git remote rm heroku`
     `heroku addons:add bundles:single --app #{staging_app}`
 
     puts ""
     puts "Creating #{production_app} Heroku app"
     puts ""
     `heroku app:create #{production_app}`
+    `heroku config:add RACK_ENV=production --app #{production_app}`
+    `git remote rm heroku`
     `heroku addons:add bundles:single --app #{production_app}`
 
     puts ""
     puts "Setup Complete!"
     puts ""
@@ -173,14 +177,15 @@
     `heroku bundles:destroy #{old_bundle} --app #{app}`
     puts "Old Bundle Destroyed"
     `heroku bundles:capture backup-#{timestamp} --app #{app}`
     puts "New Bundle Captured on Heroku: backup-#{timestamp}"
 
-    puts "Waiting for Bundle to become available..."
+    print "Waiting for Bundle to become available..."
     while bundle_not_yet_captured?( app ) do
       print "."
     end
-
+    puts ""
+    
     if bundle_captured?( app )
       puts "New Bundle Ready For Download"
 
       `heroku bundles:download backup-#{timestamp} --app #{app}`
       `mv #{app}.tar.gz #{app}-#{timestamp}.tar.gz`
\ No newline at end of file