Merge remote-tracking branch 'parent/main' into upstream-20240117

app/models/concerns/status/visibility.rb

@@ -0,0 +1,99 @@
+# frozen_string_literal: true
+
+module Status::Visibility
+  extend ActiveSupport::Concern
+
+  included do
+    enum :visibility,
+         { public: 0, unlisted: 1, private: 2, direct: 3, limited: 4, public_unlisted: 10, login: 11 },
+         suffix: :visibility,
+         validate: true
+    enum :searchability,
+         { public: 0, private: 1, direct: 2, limited: 3, unsupported: 4, public_unlisted: 10 },
+         suffix: :searchability
+    enum :limited_scope,
+         { none: 0, mutual: 1, circle: 2, personal: 3, reply: 4 },
+         suffix: :limited
+
+    scope :unset_searchability, -> { where(searchability: nil, reblog_of_id: nil) }
+    scope :distributable_visibility, -> { where(visibility: %i(public unlisted)) }
+    scope :distributable_visibility_for_anonymous, -> { where(visibility: %i(public public_unlisted unlisted)) }
+    scope :list_eligible_visibility, -> { where(visibility: %i(public unlisted private)) }
+    scope :not_direct_visibility, -> { where.not(visibility: :direct) }
+
+    validates :visibility, exclusion: { in: %w(direct limited) }, if: :reblog?
+
+    before_validation :set_visibility, unless: :visibility?
+    before_validation :set_searchability
+  end
+
+  class_methods do
+    def selectable_visibilities
+      selectable_all_visibilities - %w(mutual circle reply direct)
+    end
+
+    def selectable_all_visibilities
+      vs = %w(public public_unlisted login unlisted private mutual circle reply direct)
+      vs -= %w(public_unlisted) unless Setting.enable_public_unlisted_visibility
+      vs -= %w(public) unless Setting.enable_public_visibility
+      vs
+    end
+
+    def selectable_reblog_visibilities
+      %w(unset) + selectable_visibilities
+    end
+
+    def all_visibilities
+      visibilities.keys
+    end
+
+    def selectable_searchabilities
+      ss = searchabilities.keys - %w(unsupported)
+      ss -= %w(public_unlisted) unless Setting.enable_public_unlisted_visibility
+      ss
+    end
+
+    def selectable_searchabilities_for_search
+      searchabilities.keys - %w(public_unlisted unsupported)
+    end
+
+    def all_searchabilities
+      searchabilities.keys - %w(unlisted login unsupported)
+    end
+  end
+
+  def hidden?
+    !distributable?
+  end
+
+  def distributable?
+    public_visibility? || unlisted_visibility? || public_unlisted_visibility?
+  end
+
+  alias sign? distributable?
+
+  private
+
+  def set_visibility
+    self.visibility ||= reblog.visibility if reblog?
+    self.visibility ||= visibility_from_account
+  end
+
+  def visibility_from_account
+    account.locked? ? :private : :public
+  end
+
+  def set_searchability
+    return if searchability.nil?
+
+    self.searchability = if %w(public public_unlisted login unlisted).include?(visibility)
+                           searchability
+                         elsif ['limited', 'direct'].include?(visibility)
+                           searchability == 'limited' ? :limited : :direct
+                         elsif visibility == 'private'
+                           ['public', 'public_unlisted'].include?(searchability) ? :private : searchability
+                         else
+                           :direct
+                         end
+  end
+end