1 files changed, 27 insertions, 6 deletions
diff --git a/app/models/user_custom_attribute.rb b/app/models/user_custom_attribute.rb
index 62614a851c1..559e93be360 100644
--- a/app/models/user_custom_attribute.rb
+++ b/app/models/user_custom_attribute.rb
@@ -6,13 +6,34 @@ class UserCustomAttribute < ApplicationRecord
   validates :user_id, :key, :value, presence: true
   validates :key, uniqueness: { scope: [:user_id] }
 
-  def self.upsert_custom_attributes(custom_attributes)
-    created_at = DateTime.now
-    updated_at = DateTime.now
+  scope :by_key, ->(key) { where(key: key) }
+  scope :by_user_id, ->(user_id) { where(user_id: user_id) }
+  scope :by_updated_at, ->(updated_at) { where(updated_at: updated_at) }
+  scope :arkose_sessions, -> { by_key('arkose_session') }
 
-    custom_attributes.map! do |custom_attribute|
-      custom_attribute.merge({ created_at: created_at, updated_at: updated_at })
+  class << self
+    def upsert_custom_attributes(custom_attributes)
+      created_at = DateTime.now
+      updated_at = DateTime.now
+
+      custom_attributes.map! do |custom_attribute|
+        custom_attribute.merge({ created_at: created_at, updated_at: updated_at })
+      end
+      upsert_all(custom_attributes, unique_by: [:user_id, :key])
+    end
+
+    def sessions
+      return none if blocked_users.empty?
+
+      arkose_sessions
+        .by_user_id(blocked_users.map(&:user_id))
+        .select(:value)
+    end
+
+    private
+
+    def blocked_users
+      by_key('blocked_at').by_updated_at(Date.yesterday.all_day)
     end
-    upsert_all(custom_attributes, unique_by: [:user_id, :key])
   end
 end