move safe_mode from Thread.current to RequestStore

This commit is contained in:
Albert Yi
2019-05-17 13:47:54 -07:00
parent e51d77b065
commit f95bcc7330
2 changed files with 6 additions and 6 deletions

View File

@@ -79,7 +79,7 @@ class CurrentUser
end
def self.safe_mode?
Thread.current[:safe_mode]
RequestStore[:safe_mode]
end
def self.admin_mode?
@@ -87,17 +87,17 @@ class CurrentUser
end
def self.without_safe_mode
prev = Thread.current[:safe_mode]
Thread.current[:safe_mode] = false
prev = RequestStore[:safe_mode]
RequestStore[:safe_mode] = false
Thread.current[:admin_mode] = true
yield
ensure
Thread.current[:safe_mode] = prev
RequestStore[: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)
RequestStore[:safe_mode] = Danbooru.config.enable_safe_mode?(req, CurrentUser.user)
end
def self.method_missing(method, *params, &block)

View File

@@ -7,7 +7,7 @@ class CurrentUserTest < ActiveSupport::TestCase
end
teardown do
Thread.current[:safe_mode] = false
RequestStore[:safe_mode] = false
end
context ".safe_mode?" do