1 files changed, 64 insertions, 0 deletions
diff --git a/app/workers/gitlab/github_gists_import/start_import_worker.rb b/app/workers/gitlab/github_gists_import/start_import_worker.rb
new file mode 100644
index 00000000000..33c91611719
--- /dev/null
+++ b/app/workers/gitlab/github_gists_import/start_import_worker.rb
@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module GithubGistsImport
+    class StartImportWorker # rubocop:disable Scalability/IdempotentWorker
+      include ApplicationWorker
+
+      data_consistency :always
+      queue_namespace :github_gists_importer
+      feature_category :importers
+
+      sidekiq_options dead: false, retry: 5
+
+      worker_has_external_dependencies!
+
+      sidekiq_retries_exhausted do |msg, _|
+        Gitlab::GithubGistsImport::Status.new(msg['args'][0]).fail!
+
+        user = User.find(msg['args'][0])
+        Gitlab::GithubImport::PageCounter.new(user, :gists, 'github-gists-importer').expire!
+      end
+
+      def perform(user_id, encrypted_token)
+        logger.info(structured_payload(user_id: user_id, message: 'starting importer'))
+
+        user = User.find(user_id)
+        decrypted_token = Gitlab::CryptoHelper.aes256_gcm_decrypt(encrypted_token)
+        result = Gitlab::GithubGistsImport::Importer::GistsImporter.new(user, decrypted_token).execute
+
+        if result.success?
+          schedule_finish_worker(user_id, result.waiter)
+        elsif result.next_attempt_in
+          schedule_next_attempt(result.next_attempt_in, user_id, encrypted_token)
+        else
+          log_error_and_raise!(user_id, result.error)
+        end
+      end
+
+      private
+
+      def schedule_finish_worker(user_id, waiter)
+        logger.info(structured_payload(user_id: user_id, message: 'importer finished'))
+
+        Gitlab::GithubGistsImport::FinishImportWorker.perform_async(user_id, waiter.key, waiter.jobs_remaining)
+      end
+
+      def schedule_next_attempt(next_attempt_in, user_id, encrypted_token)
+        logger.info(structured_payload(user_id: user_id, message: 'rate limit reached'))
+
+        self.class.perform_in(next_attempt_in, user_id, encrypted_token)
+      end
+
+      def log_error_and_raise!(user_id, error)
+        logger.error(structured_payload(user_id: user_id, message: 'import failed', 'error.message': error.message))
+
+        raise(error)
+      end
+
+      def logger
+        Gitlab::GithubImport::Logger
+      end
+    end
+  end
+end