1 files changed, 66 insertions, 0 deletions
diff --git a/lib/gitlab/github_import/importer/collaborator_importer.rb b/lib/gitlab/github_import/importer/collaborator_importer.rb
new file mode 100644
index 00000000000..9a90ea5a4ed
--- /dev/null
+++ b/lib/gitlab/github_import/importer/collaborator_importer.rb
@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module GithubImport
+    module Importer
+      class CollaboratorImporter
+        attr_reader :collaborator, :project, :client, :members_finder
+
+        # collaborator - An instance of `Gitlab::GithubImport::Representation::Collaborator`
+        # project - An instance of `Project`
+        # client - An instance of `Gitlab::GithubImport::Client`
+        def initialize(collaborator, project, client)
+          @collaborator = collaborator
+          @project = project
+          @client = client
+          @members_finder = ::MembersFinder.new(project, project.creator)
+        end
+
+        def execute
+          user_finder = GithubImport::UserFinder.new(project, client)
+          user_id = user_finder.user_id_for(collaborator)
+          return if user_id.nil?
+
+          membership = existing_user_membership(user_id)
+          access_level = map_access_level
+          return if membership && membership[:access_level] >= map_access_level
+
+          create_membership!(user_id, access_level)
+        end
+
+        private
+
+        def existing_user_membership(user_id)
+          members_finder.execute.find_by_user_id(user_id)
+        end
+
+        def map_access_level
+          access_level =
+            case collaborator[:role_name]
+            when 'read' then Gitlab::Access::GUEST
+            when 'triage' then Gitlab::Access::REPORTER
+            when 'write' then Gitlab::Access::DEVELOPER
+            when 'maintain' then Gitlab::Access::MAINTAINER
+            when 'admin' then Gitlab::Access::OWNER
+            end
+          return access_level if access_level
+
+          raise(
+            ::Gitlab::GithubImport::ObjectImporter::NotRetriableError,
+            "Unknown GitHub role: #{collaborator[:role_name]}"
+          )
+        end
+
+        def create_membership!(user_id, access_level)
+          ::ProjectMember.create!(
+            source: project,
+            access_level: access_level,
+            user_id: user_id,
+            member_namespace_id: project.project_namespace_id,
+            created_by_id: project.creator_id
+          )
+        end
+      end
+    end
+  end
+end