summaryrefslogtreecommitdiff
path: root/lib/gitlab_ci_yaml_processor.rb
blob: b6858678eab1c95e8115a4ab9260dcce214a4a08 (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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
class GitlabCiYamlProcessor
  class ValidationError < StandardError;end

  DEFAULT_STAGES = %w(build test deploy)
  DEFAULT_STAGE = 'test'
  ALLOWED_YAML_KEYS = [:before_script, :image, :services, :types, :stages, :variables]
  ALLOWED_JOB_KEYS = [:tags, :script, :only, :except, :type, :image, :services, :allow_failure, :type, :stage]

  attr_reader :before_script, :image, :services, :variables

  def initialize(config)
    @config = YAML.load(config)

    unless @config.is_a? Hash
      raise ValidationError, "YAML should be a hash"
    end

    @config = @config.deep_symbolize_keys

    initial_parsing

    validate!
  end

  def builds_for_stage_and_ref(stage, ref, tag = false)
    builds.select{|build| build[:stage] == stage && process?(build[:only], build[:except], ref, tag)}
  end

  def builds
    @jobs.map do |name, job|
      build_job(name, job)
    end
  end

  def stages
    @stages || DEFAULT_STAGES
  end

  private

  def initial_parsing
    @before_script = @config[:before_script] || []
    @image = @config[:image]
    @services = @config[:services]
    @stages = @config[:stages] || @config[:types]
    @variables = @config[:variables] || {}
    @config.except!(*ALLOWED_YAML_KEYS)

    # anything that doesn't have script is considered as unknown
    @config.each do |name, param|
      raise ValidationError, "Unknown parameter: #{name}" unless param.is_a?(Hash) && param.has_key?(:script)
    end

    unless @config.values.any?{|job| job.is_a?(Hash)}
      raise ValidationError, "Please define at least one job"
    end

    @jobs = {}
    @config.each do |key, job|
      stage = job[:stage] || job[:type] || DEFAULT_STAGE
      @jobs[key] = { stage: stage }.merge(job)
    end
  end

  def process?(only_params, except_params, ref, tag)
    return true if only_params.nil? && except_params.nil?

    if only_params
      return true if tag && only_params.include?("tags")
      return true if !tag && only_params.include?("branches")
      
      only_params.find do |pattern|
        match_ref?(pattern, ref)
      end
    else
      return false if tag && except_params.include?("tags")
      return false if !tag && except_params.include?("branches")

      except_params.each do |pattern|
        return false if match_ref?(pattern, ref)
      end
    end
  end

  def build_job(name, job)
    {
      stage: job[:stage],
      script: "#{@before_script.join("\n")}\n#{normalize_script(job[:script])}",
      tags: job[:tags] || [],
      name: name,
      only: job[:only],
      except: job[:except],
      allow_failure: job[:allow_failure] || false,
      options: {
        image: job[:image] || @image,
        services: job[:services] || @services
      }.compact
    }
  end

  def match_ref?(pattern, ref)
    if pattern.first == "/" && pattern.last == "/"
      Regexp.new(pattern[1...-1]) =~ ref
    else
      pattern == ref
    end
  end

  def normalize_script(script)
    if script.is_a? Array
      script.join("\n")
    else
      script
    end
  end

  def validate!
    unless validate_array_of_strings(@before_script)
      raise ValidationError, "before_script should be an array of strings"
    end

    unless @image.nil? || @image.is_a?(String)
      raise ValidationError, "image should be a string"
    end

    unless @services.nil? || validate_array_of_strings(@services)
      raise ValidationError, "services should be an array of strings"
    end

    unless @stages.nil? || validate_array_of_strings(@stages)
      raise ValidationError, "stages should be an array of strings"
    end

    unless @variables.nil? || validate_variables(@variables)
      raise ValidationError, "variables should be a map of key-valued strings"
    end

    @jobs.each do |name, job|
      validate_job!("#{name} job", job)
    end

    true
  end

  def validate_job!(name, job)
    job.keys.each do |key|
      unless ALLOWED_JOB_KEYS.include? key
        raise ValidationError, "#{name}: unknown parameter #{key}"
      end
    end

    if !job[:script].is_a?(String) && !validate_array_of_strings(job[:script])
      raise ValidationError, "#{name}: script should be a string or an array of a strings"
    end

    if job[:stage]
      unless job[:stage].is_a?(String) && job[:stage].in?(stages)
        raise ValidationError, "#{name}: stage parameter should be #{stages.join(", ")}"
      end
    end

    if job[:image] && !job[:image].is_a?(String)
      raise ValidationError, "#{name}: image should be a string"
    end

    if job[:services] && !validate_array_of_strings(job[:services])
      raise ValidationError, "#{name}: services should be an array of strings"
    end

    if job[:tags] && !validate_array_of_strings(job[:tags])
      raise ValidationError, "#{name}: tags parameter should be an array of strings"
    end

    if job[:only] && !validate_array_of_strings(job[:only])
      raise ValidationError, "#{name}: only parameter should be an array of strings"
    end

    if job[:except] && !validate_array_of_strings(job[:except])
      raise ValidationError, "#{name}: except parameter should be an array of strings"
    end

    if job[:allow_failure] && !job[:allow_failure].in?([true, false])
      raise ValidationError, "#{name}: allow_failure parameter should be an boolean"
    end
  end

  private

  def validate_array_of_strings(values)
    values.is_a?(Array) && values.all? {|tag| tag.is_a?(String)}
  end

  def validate_variables(variables)
    variables.is_a?(Hash) && variables.all? {|key, value| key.is_a?(Symbol) && value.is_a?(String)}
  end
end