84 lines
2.2 KiB
Ruby
84 lines
2.2 KiB
Ruby
class ApplicationController < ActionController::Base
|
|
protect_from_forgery
|
|
helper :pagination
|
|
before_filter :set_current_user
|
|
after_filter :reset_current_user
|
|
before_filter :initialize_cookies
|
|
before_filter :set_title
|
|
before_filter :set_started_at_session
|
|
layout "default"
|
|
|
|
rescue_from User::PrivilegeError, :with => :access_denied
|
|
rescue_from Danbooru::Paginator::PaginationError, :with => :render_pagination_limit
|
|
|
|
protected
|
|
def rescue_exception(exception)
|
|
@exception = exception
|
|
render :action => "static/error", :status => 500
|
|
end
|
|
|
|
def render_pagination_limit
|
|
@error_message = "You can view up to 1,000 pages. Please narrow your search terms."
|
|
render :action => "static/error", :status => 410
|
|
end
|
|
|
|
def access_denied
|
|
previous_url = params[:url] || request.fullpath
|
|
|
|
respond_to do |fmt|
|
|
fmt.html do
|
|
if request.get?
|
|
redirect_to new_session_path(:url => previous_url), :notice => "Access denied"
|
|
else
|
|
redirect_to new_session_path, :notice => "Access denied"
|
|
end
|
|
end
|
|
fmt.xml do
|
|
render :xml => {:success => false, :reason => "access denied"}.to_xml(:root => "response"), :status => 403
|
|
end
|
|
fmt.json do
|
|
render :json => {:success => false, :reason => "access denied"}.to_json, :status => 403
|
|
end
|
|
end
|
|
end
|
|
|
|
def set_current_user
|
|
session_loader = SessionLoader.new(session, cookies, request)
|
|
session_loader.load
|
|
end
|
|
|
|
def reset_current_user
|
|
CurrentUser.user = nil
|
|
CurrentUser.ip_addr = nil
|
|
end
|
|
|
|
def set_started_at_session
|
|
if session[:started_at].blank?
|
|
session[:started_at] = Time.now
|
|
end
|
|
end
|
|
|
|
%w(member banned privileged platinum contributor janitor moderator admin).each do |level|
|
|
define_method("#{level}_only") do
|
|
if CurrentUser.user.__send__("is_#{level}?")
|
|
true
|
|
else
|
|
access_denied()
|
|
false
|
|
end
|
|
end
|
|
end
|
|
|
|
def initialize_cookies
|
|
if CurrentUser.user.is_anonymous?
|
|
cookies["blacklisted_tags"] = ""
|
|
else
|
|
cookies["blacklisted_tags"] = CurrentUser.user.blacklisted_tags
|
|
end
|
|
end
|
|
|
|
def set_title
|
|
@page_title = Danbooru.config.app_name + "/#{params[:controller]}"
|
|
end
|
|
end
|