/
controller_extensions.rb
64 lines (51 loc) · 1.4 KB
/
controller_extensions.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
module TwitterAuth
# These methods borrow HEAVILY from Rick Olsen's
# Restful Authentication. All cleverness props
# go to him, not me.
module ControllerExtensions
def self.included(base)
base.send :helper_method, :current_user, :logged_in?, :authorized?
end
protected
def authentication_failed(message, destination='/')
flash[:error] = message
redirect_to destination
end
def authentication_succeeded(message = 'You have logged in successfully.', destination = '/')
flash[:notice] = message
redirect_to destination
end
def current_user
@current_user ||= User.find_by_id(session[:user_id])
end
def current_user=(new_user)
session[:user_id] = new_user.id
@current_user = new_user
end
def authorized?
!!current_user
end
def login_required
authorized? || access_denied
end
def access_denied
store_location
redirect_to login_path
end
def store_location
session[:return_to] = request.request_uri
end
def redirect_back_or_default(default)
redirect_to(session[:return_to] || default)
session[:return_to] = nil
end
def logged_in?
!!current_user
end
def logout_keeping_session!
@current_user = nil
session[:user_id] = nil
end
end
end
ActionController::Base.send(:include, TwitterAuth::ControllerExtensions)