about summary refs log tree commit diff
path: root/app/models
diff options
context:
space:
mode:
authormultiple creatures <dev@multiple-creature.party>2019-02-12 06:56:36 +0000
committermultiple creatures <dev@multiple-creature.party>2019-05-21 03:16:21 -0500
commit1d5da3990200f8e1b4db64eab5dff7691dc93353 (patch)
tree0491c8525047616c26d83e42a31ba0fa6326923a /app/models
parentd6738df0833dcbead2e6d80e137f5521f75dc464 (diff)
allow overriding visibility/sensitivity by domain
Diffstat (limited to 'app/models')
-rw-r--r--app/models/status.rb11
1 files changed, 11 insertions, 0 deletions
diff --git a/app/models/status.rb b/app/models/status.rb
index ce642f9bf..157986e24 100644
--- a/app/models/status.rb
+++ b/app/models/status.rb
@@ -37,6 +37,8 @@ class Status < ApplicationRecord
 
   LOCAL_DOMAINS = ENV.fetch('LOCAL_DOMAINS', '').chomp.split(/\.?\s+/).freeze
   LOCAL_URIS = LOCAL_DOMAINS.map { |domain| "https://#{domain}/%" }.freeze
+  FORCE_SENSITIVE = ENV.fetch('FORCE_SENSITIVE', '').chomp.split(/\.?\s+/).freeze
+  FORCE_UNLISTED = ENV.fetch('FORCE_UNLISTED', '').chomp.split(/\.?\s+/).freeze
 
   # If `override_timestamps` is set at creation time, Snowflake ID creation
   # will be based on current time instead of `created_at`
@@ -293,6 +295,8 @@ class Status < ApplicationRecord
 
   after_create :set_poll_id
 
+  after_find :limit_visibility
+
   class << self
     def selectable_visibilities
       visibilities.keys - %w(direct limited)
@@ -547,6 +551,13 @@ class Status < ApplicationRecord
     self.sensitive  = false if sensitive.nil?
   end
 
+  def limit_visibility
+    return unless has_attribute?(:uri) && !uri.nil?
+    domain = Addressable::URI.parse(uri).host
+    self.sensitive = true if domain.in?(FORCE_SENSITIVE)
+    self.visibility = :unlisted if domain.in?(FORCE_UNLISTED)
+  end
+
   def set_locality
     if account.domain.nil? && !attribute_changed?(:local_only)
       self.local_only = marked_local_only?