From f95bcc733090aac6f74ee09531f5acd7029783ff Mon Sep 17 00:00:00 2001
From: Albert Yi <albert@carbonfive.com>
Date: Fri, 17 May 2019 13:47:54 -0700
Subject: [PATCH] move safe_mode from `Thread.current` to `RequestStore`

---
 app/logical/current_user.rb    | 10 +++++-----
 test/unit/current_user_test.rb |  2 +-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/app/logical/current_user.rb b/app/logical/current_user.rb
index e30e44501..6c8b6d6ae 100644
--- a/app/logical/current_user.rb
+++ b/app/logical/current_user.rb
@@ -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)
diff --git a/test/unit/current_user_test.rb b/test/unit/current_user_test.rb
index 2ef0d1d8a..610491a5c 100644
--- a/test/unit/current_user_test.rb
+++ b/test/unit/current_user_test.rb
@@ -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