1 files changed, 58 insertions, 0 deletions
diff --git a/app/workers/gitlab/github_import/stage/import_attachments_worker.rb b/app/workers/gitlab/github_import/stage/import_attachments_worker.rb
new file mode 100644
index 00000000000..e9086dca503
--- /dev/null
+++ b/app/workers/gitlab/github_import/stage/import_attachments_worker.rb
@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module GithubImport
+    module Stage
+      class ImportAttachmentsWorker # rubocop:disable Scalability/IdempotentWorker
+        include ApplicationWorker
+
+        data_consistency :always
+
+        sidekiq_options retry: 5
+        include GithubImport::Queue
+        include StageMethods
+
+        # client - An instance of Gitlab::GithubImport::Client.
+        # project - An instance of Project.
+        def import(client, project)
+          return skip_to_next_stage(project) if feature_disabled?(project)
+
+          waiters = importers.each_with_object({}) do |importer, hash|
+            waiter = start_importer(project, importer, client)
+            hash[waiter.key] = waiter.jobs_remaining
+          end
+          move_to_next_stage(project, waiters)
+        end
+
+        private
+
+        # For future issue/mr/milestone/etc attachments importers
+        def importers
+          [::Gitlab::GithubImport::Importer::ReleasesAttachmentsImporter]
+        end
+
+        def start_importer(project, importer, client)
+          info(project.id, message: "starting importer", importer: importer.name)
+          importer.new(project, client).execute
+        end
+
+        def skip_to_next_stage(project)
+          info(project.id, message: "skipping importer", importer: 'Attachments')
+          move_to_next_stage(project)
+        end
+
+        def move_to_next_stage(project, waiters = {})
+          AdvanceStageWorker.perform_async(
+            project.id,
+            waiters,
+            :protected_branches
+          )
+        end
+
+        def feature_disabled?(project)
+          Feature.disabled?(:github_importer_attachments_import, project.group, type: :ops)
+        end
+      end
+    end
+  end
+end