81 lines
1.7 KiB
Ruby
81 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class GroupMember < Member
|
|
include FromUnion
|
|
|
|
SOURCE_TYPE = 'Namespace'
|
|
|
|
belongs_to :group, foreign_key: 'source_id'
|
|
|
|
delegate :update_two_factor_requirement, to: :user
|
|
|
|
# Make sure group member points only to group as it source
|
|
default_value_for :source_type, SOURCE_TYPE
|
|
validates :source_type, format: { with: /\ANamespace\z/ }
|
|
default_scope { where(source_type: SOURCE_TYPE) }
|
|
|
|
scope :of_groups, ->(groups) { where(source_id: groups.select(:id)) }
|
|
scope :count_users_by_group_id, -> { joins(:user).group(:source_id).count }
|
|
scope :of_ldap_type, -> { where(ldap: true) }
|
|
|
|
after_create :update_two_factor_requirement, unless: :invite?
|
|
after_destroy :update_two_factor_requirement, unless: :invite?
|
|
|
|
def self.access_level_roles
|
|
Gitlab::Access.options_with_owner
|
|
end
|
|
|
|
def self.access_levels
|
|
Gitlab::Access.sym_options_with_owner
|
|
end
|
|
|
|
def group
|
|
source
|
|
end
|
|
|
|
# Because source_type is `Namespace`...
|
|
def real_source_type
|
|
'Group'
|
|
end
|
|
|
|
def notifiable_options
|
|
{ group: group }
|
|
end
|
|
|
|
private
|
|
|
|
def send_invite
|
|
run_after_commit_or_now { notification_service.invite_group_member(self, @raw_invite_token) }
|
|
|
|
super
|
|
end
|
|
|
|
def post_create_hook
|
|
run_after_commit_or_now { notification_service.new_group_member(self) }
|
|
|
|
super
|
|
end
|
|
|
|
def post_update_hook
|
|
if saved_change_to_access_level?
|
|
run_after_commit { notification_service.update_group_member(self) }
|
|
end
|
|
|
|
super
|
|
end
|
|
|
|
def after_accept_invite
|
|
notification_service.accept_group_invite(self)
|
|
update_two_factor_requirement
|
|
|
|
super
|
|
end
|
|
|
|
def after_decline_invite
|
|
notification_service.decline_group_invite(self)
|
|
|
|
super
|
|
end
|
|
end
|
|
|
|
GroupMember.prepend_if_ee('EE::GroupMember')
|