gab-social/app/models/group.rb
2021-02-16 13:35:19 -05:00

133 lines
4.4 KiB
Ruby

# == 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 :integer
# 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