don't say following succeeded if it failed

toypaj · Mar 14, 2011 · a009b70 · a009b70
1 parent ed1c37e
commit a009b70
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/rstatus.rb b/rstatus.rb
@@ -185,10 +185,13 @@ class Rstatus < Sinatra::Base
     end
 
     # then follow them!
-    current_user.follow! @user
-
-    flash[:notice] = "Now following #{params[:name]}."
-    redirect "/users/#{@user.username}"
+    unless  current_user.follow! @user
+      flash[:notice] = "The was a problem following #{params[:name]}."
+      redirect "/users/#{@user.username}"
+    else
+      flash[:notice] = "Now following #{params[:name]}."
+      redirect "/users/#{@user.username}"
+    end
   end
 
   #this lets you unfollow a user