asciinema.org/app/controllers/application_controller.rb
2013-05-28 11:13:01 +02:00

84 lines
1.7 KiB
Ruby

class NotFound < StandardError; end
class ApplicationController < ActionController::Base
protect_from_forgery
class Unauthorized < Exception; end
class Forbidden < Exception; end
rescue_from ActiveRecord::RecordNotFound, :with => :not_found
rescue_from Unauthorized, :with => :unauthorized
rescue_from Forbidden, :with => :forbidden
helper_method :current_user
def current_user
if session[:user_id]
@current_user ||= User.find_by_id(session[:user_id]).decorate
end
end
def current_user=(user)
if user
@current_user = user
session[:user_id] = user.id
else
@current_user = nil
session[:user_id] = nil
end
end
private
def ensure_authenticated!
raise Unauthorized unless current_user
end
def store_location
session[:return_to] = request.path
end
def get_stored_location
session.delete(:return_to)
end
def redirect_back_or_to(default, options = nil)
path = get_stored_location || default
if options
redirect_to path, options
else
redirect_to path
end
end
def forbidden
if request.xhr?
render :json => "Forbidden", :status => 403
else
redirect_to root_path, :alert => "This action is forbidden"
end
end
def unauthorized
if request.xhr?
render :json => "Unauthorized", :status => 401
else
store_location
redirect_to login_path, :notice => "Please login first"
end
end
def not_found
respond_to do |format|
format.any do
render :text => 'Requested resource not found', :status => 404
end
format.html do
render 'application/not_found', :status => 404, :layout => 'application'
end
end
end
end