From 41e2c09d8a08586c806c0826958221ca4af259f8 Mon Sep 17 00:00:00 2001
From: Cyril David <cyx.ucron@gmail.com>
Date: Mon, 18 Oct 2010 15:29:58 +0800
Subject: [PATCH] Default Sinatra::Default logout to use ::User.

---
 lib/shield/contrib/sinatra/default.rb | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lib/shield/contrib/sinatra/default.rb b/lib/shield/contrib/sinatra/default.rb
index 1096dc7..b91f162 100644
--- 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