gitlab-org--gitlab-foss/lib/gitlab/ci/config/entry/variable.rb

# frozen_string_literal: true

module Gitlab
  module Ci
    class Config
      module Entry
        ##
        # Entry that represents a CI/CD variable.
        #
        class Variable < ::Gitlab::Config::Entry::Simplifiable
          strategy :SimpleVariable, if: -> (config) { SimpleVariable.applies_to?(config) }
          strategy :ComplexVariable, if: -> (config) { ComplexVariable.applies_to?(config) }
          strategy :ComplexArrayVariable, if: -> (config) { ComplexArrayVariable.applies_to?(config) }

          class SimpleVariable < ::Gitlab::Config::Entry::Node
            include ::Gitlab::Config::Entry::Validatable

            class << self
              def applies_to?(config)
                Gitlab::Config::Entry::Validators::AlphanumericValidator.validate(config)
              end
            end

            validations do
              validates :key, alphanumeric: true
              validates :config, alphanumeric: true
            end

            def value
              @config.to_s
            end

            def value_with_data
              { value: @config.to_s }
            end

            def value_with_prefill_data
              value_with_data
            end
          end

          class ComplexVariable < ::Gitlab::Config::Entry::Node
            include ::Gitlab::Config::Entry::Validatable

            class << self
              def applies_to?(config)
                config.is_a?(Hash) && !config[:value].is_a?(Array)
              end
            end

            validations do
              validates :key, alphanumeric: true
              validates :config_value, alphanumeric: true, allow_nil: false, if: :config_value_defined?
              validates :config_description, alphanumeric: true, allow_nil: false, if: :config_description_defined?
              validates :config_expand, boolean: true,
                                        allow_nil: false,
                                        if: -> { ci_raw_variables_in_yaml_config_enabled? && config_expand_defined? }

              validate do
                allowed_value_data = Array(opt(:allowed_value_data))

                if allowed_value_data.any?
                  extra_keys = config.keys - allowed_value_data

                  errors.add(:config, "uses invalid data keys: #{extra_keys.join(', ')}") if extra_keys.present?
                else
                  errors.add(:config, "must be a string")
                end
              end
            end

            def value
              config_value.to_s
            end

            def value_with_data
              if ci_raw_variables_in_yaml_config_enabled?
                {
                  value: value,
                  raw: (!config_expand if config_expand_defined?)
                }.compact
              else
                {
                  value: value
                }.compact
              end
            end

            def value_with_prefill_data
              value_with_data.merge(
                description: config_description
              ).compact
            end

            def config_value
              @config[:value]
            end

            def config_description
              @config[:description]
            end

            def config_expand
              @config[:expand]
            end

            def config_value_defined?
              config.key?(:value)
            end

            def config_description_defined?
              config.key?(:description)
            end

            def config_expand_defined?
              config.key?(:expand)
            end

            def ci_raw_variables_in_yaml_config_enabled?
              YamlProcessor::FeatureFlags.enabled?(:ci_raw_variables_in_yaml_config)
            end
          end

          class ComplexArrayVariable < ComplexVariable
            include ::Gitlab::Config::Entry::Validatable

            class << self
              def applies_to?(config)
                config.is_a?(Hash) && config[:value].is_a?(Array)
              end
            end

            validations do
              validates :config_value, array_of_strings: true, allow_nil: false, if: :config_value_defined?

              validate do
                next if opt(:allow_array_value)

                errors.add(:config, 'value must be an alphanumeric string')
              end
            end

            def value
              config_value.first
            end

            def value_with_prefill_data
              super.merge(
                value_options: config_value
              ).compact
            end
          end

          class UnknownStrategy < ::Gitlab::Config::Entry::Node
            def errors
              ["variable definition must be either a string or a hash"]
            end
          end
        end
      end
    end
  end
end