gab-social/app/models/group.rb

# == Schema Information
#
# Table name: groups
#
#  id                       :bigint(8)        not null, primary key
#  account_id               :bigint(8)
#  title                    :string           not null
#  description              :string           not null
#  cover_image_file_name    :string
#  cover_image_content_type :string
#  cover_image_file_size    :bigint(8)
#  cover_image_updated_at   :datetime
#  is_nsfw                  :boolean          default(FALSE), not null
#  is_featured              :boolean          default(FALSE), not null
#  is_archived              :boolean          default(FALSE), not null
#  created_at               :datetime         not null
#  updated_at               :datetime         not null
#  member_count             :integer          default(0)
#  slug                     :text
#  is_private               :boolean          default(FALSE)
#  is_visible               :boolean          default(TRUE)
#  tags                     :string           default([]), is an Array
#  password                 :string
#  group_category_id        :integer
#

class Group < ApplicationRecord
  self.ignored_columns = ["group_categories_id"]

  include Paginable
  include GroupInteractions
  include GroupCoverImage

  PER_ACCOUNT_LIMIT_PRO = 100
  PER_ACCOUNT_LIMIT_NORMAL = 10

  belongs_to :account, optional: true

  has_many :group_accounts, inverse_of: :group, dependent: :destroy
  has_many :accounts, through: :group_accounts

  has_many :group_join_requests, inverse_of: :group, dependent: :destroy
  has_many :join_requests, source: :account, through: :group_join_requests

  has_many :group_pinned_statuses, inverse_of: :group, dependent: :destroy
  has_many :pinned_statuses, source: :status, through: :group_pinned_statuses

  has_many :group_removed_accounts, inverse_of: :group, dependent: :destroy
  has_many :removed_accounts, source: :account, through: :group_removed_accounts

  belongs_to :group_categories, optional: true, foreign_key: 'group_category_id'

  validates :title, presence: true
  validates :description, presence: true

  validates_each :account_id, on: :create do |record, _attr, value|
    account = Account.find(value)
    limit = account.is_pro ? PER_ACCOUNT_LIMIT_PRO : PER_ACCOUNT_LIMIT_NORMAL
    record.errors.add(:base, "You have reached the limit for group creation.") if Group.where(account_id: value).count >= limit
  end

  before_save :set_slug
  before_save :set_password
  before_destroy :clean_feed_manager
  after_create :add_owner_to_accounts

  class << self
    def search_for(term, offset = 0)
      pattern = sanitize_sql_like(term.strip) + '%'

      Group.where('lower(title) like lower(?) AND is_archived=false AND is_visible=true', pattern)
        .order('member_count DESC')
        .limit(25)
        .offset(offset)
    end

    def search_for_members(group, term, limit)
      pattern = '%' + sanitize_sql_like(term.strip) + '%'
      group.accounts.where("LOWER(username) LIKE LOWER(?)", pattern).limit(limit)
    end

    def search_for_removed_accounts(group, term, limit)
      pattern = '%' + sanitize_sql_like(term.strip) + '%'
      group.removed_accounts.where("LOWER(username) LIKE LOWER(?)", pattern).limit(limit)
    end
  end

  def has_password?
    return !!self.password && self.password.gsub(/\s+/, "").length > 1 && self.password.to_s != "null"
  end

  private

  def set_password
    if password.nil? || !password || password.gsub(/\s+/, "").length <= 1 || password == "null"
      nil
    else
      password
    end
  end

  def set_slug
    if !slug.nil? && self.member_count > 50 && self.slug.nil?
      self.slug = slug.downcase.strip.gsub(' ', '-').gsub(/[^\w-]/, '')
    elsif !slug.nil? && self.member_count < 50 && self.slug.nil?
      self.slug = nil
    else
      self.slug = self.slug
    end
  end

  def add_owner_to_accounts
    group_accounts << GroupAccount.new(account: account, role: :admin, write_permissions: true)
  end

  def clean_feed_manager
    Redis.current.with do |conn|
      reblog_key       = FeedManager.instance.key(:group, id, 'reblogs')
      reblogged_id_set = conn.zrange(reblog_key, 0, -1)

      conn.pipelined do
        conn.del(FeedManager.instance.key(:group, id))
        conn.del(reblog_key)

        reblogged_id_set.each do |reblogged_id|
          reblog_set_key = FeedManager.instance.key(:group, id, "reblogs:#{reblogged_id}")
          conn.del(reblog_set_key)
        end
      end
    end
  end
end