debian-mirror-gitlab/lib/gitlab/ci/config/yaml.rb

# 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!
            else
              ::Gitlab::Config::Loader::Yaml
                .new(content, additional_permitted_classes: AVAILABLE_TAGS)
                .load!
            end
          end

          def to_result
            Yaml::Result.new(config: load!, error: nil)
          rescue ::Gitlab::Config::Loader::FormatError => e
            Yaml::Result.new(error: e)
          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).to_result.then do |result|
              ##
              # raise an error for backwards compatibility
              #
              raise result.error unless result.valid?

              result.content
            end
          end

          def load_result!(content, project: nil)
            Loader.new(content, project: project).to_result
          end
        end
      end
    end
  end
end