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
|
# frozen_string_literal: true
module Gitlab
module Ci
class Config
module Yaml
class Loader
include Gitlab::Utils::StrongMemoize
AVAILABLE_TAGS = [Config::Yaml::Tags::Reference].freeze
MAX_DOCUMENTS = 2
def initialize(content, inputs: {}, variables: [])
@content = content
@inputs = inputs
@variables = variables
end
def load
yaml_result = load_uninterpolated_yaml
return yaml_result unless yaml_result.valid?
interpolator = Interpolation::Interpolator.new(yaml_result, inputs, variables)
interpolator.interpolate!
if interpolator.valid?
# This Result contains only the interpolated config and does not have a header
Yaml::Result.new(config: interpolator.to_hash, error: nil, interpolated: interpolator.interpolated?)
else
Yaml::Result.new(error: interpolator.error_message, interpolated: interpolator.interpolated?)
end
end
def load_uninterpolated_yaml
Yaml::Result.new(config: load_yaml!, error: nil)
rescue ::Gitlab::Config::Loader::FormatError => e
Yaml::Result.new(error: e.message, error_class: e)
end
private
attr_reader :content, :inputs, :variables
def load_yaml!
ensure_custom_tags
::Gitlab::Config::Loader::MultiDocYaml.new(
content,
max_documents: MAX_DOCUMENTS,
additional_permitted_classes: AVAILABLE_TAGS,
reject_empty: true
).load!
end
def ensure_custom_tags
@ensure_custom_tags ||= begin
AVAILABLE_TAGS.each { |klass| Psych.add_tag(klass.tag, klass) }
true
end
end
end
end
end
end
end
|
