93 lines
1.7 KiB
Ruby
93 lines
1.7 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
|
|
end
|
|
end
|
|
|
|
def self.as_admin(&block)
|
|
if block_given?
|
|
scoped(User.admins.first, "127.0.0.1", &block)
|
|
else
|
|
self.user = User.admins.first
|
|
self.ip_addr = "127.0.0.1"
|
|
end
|
|
end
|
|
|
|
def self.as_system(&block)
|
|
if block_given?
|
|
scoped(User.system, "127.0.0.1", &block)
|
|
else
|
|
self.user = User.system
|
|
self.ip_addr = "127.0.0.1"
|
|
end
|
|
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.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)
|
|
Thread.current[:safe_mode] = Danbooru.config.enable_safe_mode?(req, CurrentUser.user)
|
|
end
|
|
|
|
def self.method_missing(method, *params, &block)
|
|
if user.respond_to?(method)
|
|
user.__send__(method, *params, &block)
|
|
else
|
|
super
|
|
end
|
|
end
|
|
end
|