lib/gitlab/github_import/importer/protected_branches_importer.rb


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
52

# frozen_string_literal: true

module Gitlab
  module GithubImport
    module Importer
      class ProtectedBranchesImporter
        include ParallelScheduling

        # The method that will be called for traversing through all the objects to
        # import, yielding them to the supplied block.
        def each_object_to_import
          repo = project.import_source

          protected_branches = client.branches(repo).select { |branch| branch.protection&.enabled }
          protected_branches.each do |protected_branch|
            object = client.branch_protection(repo, protected_branch.name)
            next if object.nil? || already_imported?(object)

            yield object

            Gitlab::GithubImport::ObjectCounter.increment(project, object_type, :fetched)
            mark_as_imported(object)
          end
        end

        def importer_class
          ProtectedBranchImporter
        end

        def representation_class
          Gitlab::GithubImport::Representation::ProtectedBranch
        end

        def sidekiq_worker_class
          ImportProtectedBranchWorker
        end

        def object_type
          :protected_branch
        end

        def collection_method
          :protected_branches
        end

        def id_for_already_imported_cache(protected_branch)
          protected_branch.name
        end
      end
    end
  end
end