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(user, &block) scoped(user, &block) 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.root_url Thread.current[:current_root_url] || "http://#{Danbooru.config.hostname}/" end def self.root_url=(root_url) Thread.current[:current_root_url] = root_url 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) user.__send__(method, *params, &block) end end