Fix bug caused by invalid token

seven1m · Oct 31, 2014 · 604f6ac · 604f6ac
1 parent a6e838a
commit 604f6ac
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/app.rb b/app.rb
@@ -52,9 +52,15 @@
     installer.size = params[:size]
     session[:config] = installer.as_json
     if installer.auth_token = session[:token]
-      installer.go!
-      session[:config] = installer.as_json
-      redirect '/status'
+      begin
+        installer.go!
+      rescue RestClient::Unauthorized
+        session[:token] = nil
+        redirect "https://cloud.digitalocean.com/v1/oauth/authorize?response_type=code&client_id=#{CLIENT_ID}&redirect_uri=#{CALLBACK_URL}&scope=read+write"
+      else
+        session[:config] = installer.as_json
+        redirect '/status'
+      end
     else
       redirect "https://cloud.digitalocean.com/v1/oauth/authorize?response_type=code&client_id=#{CLIENT_ID}&redirect_uri=#{CALLBACK_URL}&scope=read+write"
     end