Welcome to mirror list, hosted at ThFree Co, Russian Federation.

variables_group_seeder.rb « ci « seeders « gitlab « lib - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: e5c5ee4b75ff9719d53f01ab193b141282ac70e5 (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
# frozen_string_literal: true

module Gitlab
  module Seeders
    module Ci
      class VariablesGroupSeeder
        DEFAULT_SEED_COUNT = 10
        DEFAULT_PREFIX = 'GROUP_VAR_'
        DEFAULT_ENV = '*'

        def initialize(params)
          @group = Group.find_by_name(params[:name])
          @seed_count = params[:seed_count] || DEFAULT_SEED_COUNT
          @environment_scope = params[:environment_scope] || DEFAULT_ENV
          @prefix = params[:prefix] || DEFAULT_PREFIX
        end

        def seed
          if @group.nil?
            warn 'ERROR: Group name is invalid.'
            return
          end

          max_id = group.variables.maximum(:id).to_i
          seed_count.times do
            max_id += 1
            create_ci_variable(max_id)
          end
        end

        private

        attr_reader :environment_scope, :group, :prefix, :seed_count

        def create_ci_variable(id)
          env = environment_scope == 'unique' ? "env_#{id}" : environment_scope
          key = "#{prefix}#{id}"

          if group.variables.by_environment_scope(env).find_by_key(key).present?
            warn "WARNING: Group CI Variable with key '#{key}' already exists. Skipping to next CI variable..."
            return
          end

          group.variables.create(
            environment_scope: env,
            key: key,
            value: SecureRandom.hex(32)
          )
        end
      end
    end
  end
end