summaryrefslogtreecommitdiff
path: root/lib/gitlab/import_export/project/base_task.rb
blob: 356e261e25139b38c759226ac71f7509f262a126 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
# frozen_string_literal: true

module Gitlab
  module ImportExport
    module Project
      class BaseTask
        include Gitlab::WithRequestStore

        def initialize(opts, logger: Logger.new($stdout))
          @project_path = opts.fetch(:project_path)
          @file_path    = opts.fetch(:file_path)
          @namespace    = Namespace.find_by_full_path(opts.fetch(:namespace_path))
          @current_user = User.find_by_username(opts.fetch(:username))
          @logger = logger
        end

        private

        attr_reader :project, :namespace, :current_user, :file_path, :project_path, :logger

        def disable_upload_object_storage
          overwrite_uploads_setting('enabled', false) do
            yield
          end
        end

        def overwrite_uploads_setting(key, value)
          old_value = Settings.uploads.object_store[key]
          Settings.uploads.object_store[key] = value

          yield

        ensure
          Settings.uploads.object_store[key] = old_value
        end

        def success(message)
          logger.info(message)

          true
        end

        def error(message)
          logger.error(message)

          false
        end
      end
    end
  end
end