blob: 31efe6ab6d98042c1af12f13600b9b15496bcb26 (
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
|
# frozen_string_literal: true
module Gitlab
module Ci
class Config
module Yaml
AVAILABLE_TAGS = [Config::Yaml::Tags::Reference].freeze
MAX_DOCUMENTS = 2
class Loader
def initialize(content, project: nil)
@content = content
@project = project
end
def load!
ensure_custom_tags
if project.present? && ::Feature.enabled?(:ci_multi_doc_yaml, project)
Gitlab::Config::Loader::MultiDocYaml.new(
content,
max_documents: MAX_DOCUMENTS,
additional_permitted_classes: AVAILABLE_TAGS
).load!.first
else
Gitlab::Config::Loader::Yaml.new(content, additional_permitted_classes: AVAILABLE_TAGS).load!
end
end
private
attr_reader :content, :project
def ensure_custom_tags
@ensure_custom_tags ||= begin
AVAILABLE_TAGS.each { |klass| Psych.add_tag(klass.tag, klass) }
true
end
end
end
class << self
def load!(content, project: nil)
Loader.new(content, project: project).load!
end
end
end
end
end
end
|