# frozen_string_literal: true module Members module BulkCreateUsers extend ActiveSupport::Concern included do class << self def add_users(source, users, access_level, current_user: nil, expires_at: nil, tasks_to_be_done: [], tasks_project_id: nil) return [] unless users.present? emails, users, existing_members = parse_users_list(source, users) Member.transaction do (emails + users).map! do |user| new(source, user, access_level, existing_members: existing_members, current_user: current_user, expires_at: expires_at, tasks_to_be_done: tasks_to_be_done, tasks_project_id: tasks_project_id) .execute end end end private def parse_users_list(source, list) emails = [] user_ids = [] users = [] existing_members = {} list.each do |item| case item when User users << item when Integer user_ids << item when /\A\d+\Z/ user_ids << item.to_i when Devise.email_regexp emails << item end end if user_ids.present? # we should handle the idea of existing members where users are passed as users - https://gitlab.com/gitlab-org/gitlab/-/issues/352617 # the below will automatically discard invalid user_ids users.concat(User.id_in(user_ids)) # helps not have to perform another query per user id to see if the member exists later on when fetching existing_members = source.members_and_requesters.where(user_id: user_ids).index_by(&:user_id) # rubocop:disable CodeReuse/ActiveRecord end users.uniq! # de-duplicate just in case as there is no controlling if user records and ids are sent multiple times [emails, users, existing_members] end end end def initialize(source, user, access_level, **args) super @existing_members = args[:existing_members] || (raise ArgumentError, "existing_members must be included in the args hash") end private attr_reader :existing_members def find_or_initialize_member_by_user existing_members[user.id] || source.members.build(user_id: user.id) end end end