mirror of
https://github.com/discourse/discourse.git
synced 2024-12-16 09:05:47 +08:00
115 lines
2.9 KiB
Ruby
115 lines
2.9 KiB
Ruby
require_dependency 'notification_levels'
|
|
|
|
class GroupUser < ActiveRecord::Base
|
|
belongs_to :group, counter_cache: "user_count"
|
|
belongs_to :user
|
|
|
|
after_save :update_title
|
|
after_destroy :remove_title
|
|
|
|
after_save :set_primary_group
|
|
after_destroy :remove_primary_group, :recalculate_trust_level
|
|
|
|
before_create :set_notification_level
|
|
after_save :grant_trust_level
|
|
|
|
def self.notification_levels
|
|
NotificationLevels.all
|
|
end
|
|
|
|
protected
|
|
|
|
def set_notification_level
|
|
self.notification_level = group&.default_notification_level || 3
|
|
end
|
|
|
|
def set_primary_group
|
|
if group.primary_group
|
|
self.class.exec_sql("
|
|
UPDATE users
|
|
SET primary_group_id = :id
|
|
WHERE id = :user_id",
|
|
id: group.id, user_id: user_id
|
|
)
|
|
end
|
|
end
|
|
|
|
def remove_primary_group
|
|
self.class.exec_sql("
|
|
UPDATE users
|
|
SET primary_group_id = NULL
|
|
WHERE id = :user_id AND primary_group_id = :id",
|
|
id: group.id, user_id: user_id
|
|
)
|
|
end
|
|
|
|
def remove_title
|
|
if group.title.present?
|
|
self.class.exec_sql("
|
|
UPDATE users SET title = NULL
|
|
WHERE title = :title AND id = :id",
|
|
id: user_id, title: group.title
|
|
)
|
|
end
|
|
end
|
|
|
|
def update_title
|
|
if group.title.present?
|
|
self.class.exec_sql("
|
|
UPDATE users SET title = :title
|
|
WHERE (title IS NULL OR title = '') AND id = :id",
|
|
id: user_id, title: group.title
|
|
)
|
|
end
|
|
end
|
|
|
|
def grant_trust_level
|
|
return if group.grant_trust_level.nil?
|
|
|
|
if (user.group_locked_trust_level || 0) < group.grant_trust_level
|
|
user.update!(group_locked_trust_level: group.grant_trust_level)
|
|
end
|
|
|
|
TrustLevelGranter.grant(group.grant_trust_level, user)
|
|
end
|
|
|
|
def recalculate_trust_level
|
|
return if group.grant_trust_level.nil?
|
|
|
|
# Find the highest level of the user's remaining groups
|
|
highest_level = GroupUser
|
|
.where(user_id: user.id)
|
|
.includes(:group)
|
|
.maximum("groups.grant_trust_level")
|
|
|
|
if highest_level.nil?
|
|
# If the user no longer has a group with a trust level,
|
|
# unlock them, start at 0 and consider promotions.
|
|
user.update!(group_locked_trust_level: nil)
|
|
Promotion.recalculate(user)
|
|
else
|
|
user.update!(group_locked_trust_level: highest_level)
|
|
user.change_trust_level!(highest_level)
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
# == Schema Information
|
|
#
|
|
# Table name: group_users
|
|
#
|
|
# id :integer not null, primary key
|
|
# group_id :integer not null
|
|
# user_id :integer not null
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
# owner :boolean default(FALSE), not null
|
|
# notification_level :integer default(2), not null
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_group_users_on_group_id_and_user_id (group_id,user_id) UNIQUE
|
|
# index_group_users_on_user_id_and_group_id (user_id,group_id) UNIQUE
|
|
#
|