summaryrefslogtreecommitdiff
path: root/app/models/bulk_imports/stage.rb
blob: 050c2c76ce857aeddbf737f73b8eddc59b27f565 (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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# frozen_string_literal: true

module BulkImports
  class Stage
    include Singleton

    CONFIG = {
      group: {
        pipeline: BulkImports::Groups::Pipelines::GroupPipeline,
        stage: 0
      },
      subgroups: {
        pipeline: BulkImports::Groups::Pipelines::SubgroupEntitiesPipeline,
        stage: 1
      },
      members: {
        pipeline: BulkImports::Groups::Pipelines::MembersPipeline,
        stage: 1
      },
      labels: {
        pipeline: BulkImports::Groups::Pipelines::LabelsPipeline,
        stage: 1
      },
      milestones: {
        pipeline: BulkImports::Groups::Pipelines::MilestonesPipeline,
        stage: 1
      },
      badges: {
        pipeline: BulkImports::Groups::Pipelines::BadgesPipeline,
        stage: 1
      },
      finisher: {
        pipeline: BulkImports::Groups::Pipelines::EntityFinisher,
        stage: 2
      }
    }.freeze

    def self.pipelines
      instance.pipelines
    end

    def self.pipeline_exists?(name)
      pipelines.any? do |(_, pipeline)|
        pipeline.to_s == name.to_s
      end
    end

    def pipelines
      @pipelines ||= config
        .values
        .sort_by { |entry| entry[:stage] }
        .map do |entry|
          [entry[:stage], entry[:pipeline]]
        end
    end

    private

    def config
      @config ||= CONFIG
    end
  end
end

::BulkImports::Stage.prepend_if_ee('::EE::BulkImports::Stage')