Files
danbooru/app/logical/current_user.rb
2014-11-06 17:03:17 -08:00

92 lines
1.6 KiB
Ruby

class CurrentUser
def self.scoped(user, ip_addr = "127.0.0.1")
old_user = self.user
old_ip_addr = self.ip_addr
self.user = user
self.ip_addr = ip_addr
begin
yield
ensure
self.user = old_user
self.ip_addr = old_ip_addr
self.mobile_mode = false
end
end
def self.as_admin(&block)
scoped(User.admins.first, "127.0.0.1", &block)
end
def self.user=(user)
Thread.current[:current_user] = user
end
def self.ip_addr=(ip_addr)
Thread.current[:current_ip_addr] = ip_addr
end
def self.user
Thread.current[:current_user]
end
def self.ip_addr
Thread.current[:current_ip_addr]
end
def self.mobile_mode=(mode)
Thread.current[:mobile_mode] = mode
end
def self.mobile_mode?
Thread.current[:mobile_mode]
end
def self.id
if user.nil?
nil
else
user.id
end
end
def self.name
user.name
end
def self.safe_mode?
Thread.current[:safe_mode]
end
def self.admin_mode?
Thread.current[:admin_mode]
end
def self.without_safe_mode
prev = Thread.current[:safe_mode]
Thread.current[:safe_mode] = false
Thread.current[:admin_mode] = true
yield
ensure
Thread.current[:safe_mode] = prev
Thread.current[:admin_mode] = false
end
def self.set_safe_mode(req)
if req.host =~ /safe/ || req.params[:safe_mode]
Thread.current[:safe_mode] = true
else
Thread.current[:safe_mode] = false
end
end
def self.method_missing(method, *params, &block)
if user.respond_to?(method)
user.__send__(method, *params, &block)
else
super
end
end
end