Default Sinatra::Default logout to use ::User.

foca · Oct 18, 2010 · 41e2c09 · 41e2c09
1 parent 1a70f87
commit 41e2c09
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/shield/contrib/sinatra/default.rb b/lib/shield/contrib/sinatra/default.rb
@@ -21,6 +21,10 @@ def is_logged_in?
           def ensure_authenticated(model = User)
             super
           end
+
+          def logout(model = User)
+            super
+          end
         end
 
         m.get "/login" do
@@ -37,7 +41,7 @@ def ensure_authenticated(model = User)
         end
 
         m.get "/logout" do
-          logout(User)
+          logout
           redirect settings.shield_redirect_after_logout
         end
       end