gitlab-org--gitlab-foss/scripts/changed-feature-flags

#!/usr/bin/env ruby
# frozen_string_literal: true

require 'yaml'
require 'optparse'
require 'pathname'
require_relative 'api/default_options'

# This script returns the desired feature flag state as a comma-separated string for the feature flags in the specified files.
# Each desired feature flag state is specified as 'feature-flag=state'. This allows us to run package-and-qa with the
# feature flag set to the desired state.
#
# For example, if the specified files included `config/feature_flags/development/ci_yaml_limit_size.yml` and the desired
# state as specified by the second argument was enabled, the value returned would be `ci_yaml_limit_size=enabled`

class GetFeatureFlagsFromFiles
  def initialize(options)
    @files = options.delete(:files)
    @state = options.delete(:state)

    abort("ERROR: Please specify the directory containing MR diffs.") if @files.to_s.empty?
  end

  # Gets feature flags from definition files or diffs of deleted defition files
  #
  # @return [String] a comma-separated list of feature flags and their desired state
  def extracted_flags
    flags_list = diffs_dir.glob('**/*').each_with_object([]) do |file_path, flags|
      ff_yaml = ff_yaml_for_file(file_path)
      next if ff_yaml.nil?
      break [] if ff_yaml.empty?

      flags << ff_yaml['name']
    end
    flags_list = flags_list.map { |flag| "#{flag}=#{state}" } unless state.to_s.empty?
    flags_list.join(',')
  end

  # Loads the YAML feature flag definition based on a diff of the definition file. The definition is loaded from the
  # definition file itself, or from a diff of the deleted definition file.
  #
  # @param [Pathname] path the path to the diff
  # @return [Hash] a hash containing the YAML data for the feature flag definition
  def ff_yaml_for_file(path)
    return unless File.expand_path(path).to_s =~ %r{/feature_flags/(development|ops)/.*\.yml}

    if path.to_s.end_with?('yml.deleted.diff')
      # Ignore deleted feature flag definitions if we want to enable/disable existing flags.
      return if state != 'deleted'

      yaml_from_deleted_diff(path)
    else
      # If we want deleted definition files but find one that wasn't deleted, we return immediately to
      # because non-deleted flags are tested in separate jobs from deleted flags, so we don't need to run
      # a job with just deleted flags.
      return [] if state == 'deleted'

      yaml_from_file(path, diffs_dir)
    end
  end

  private

  attr_reader :files, :state

  # The absolute path to the directory of diffs
  #
  # @return [String]
  def diffs_dir
    @diffs_dir ||= Pathname.new(files).expand_path
  end

  # Loads the YAML feature flag definition from a file corresponding to a diff of the definition file.
  #
  # @param [Pathname] file_path the path to the diff
  # @param [Pathname] diffs_dir the path to the diffs directory
  # @return [Hash] a hash containing the YAML data from the feature flag definition file corresponding to the diff
  def yaml_from_file(file_path, diffs_dir)
    real_file_path = File.join(Dir.pwd, file_path.to_s.delete_prefix(diffs_dir.to_s)).delete_suffix('.diff')
    YAML.safe_load(File.read(real_file_path))
  end

  # Loads the YAML feature flag definition from a diff of the deleted feature flag definition file.
  #
  # @param [Pathname] file_path the path of the diff
  # @return [Hash] a hash containing the YAML data for the feature flag definition from the diff
  def yaml_from_deleted_diff(file_path)
    cleaned_diff = File.read(file_path).gsub(/^[^a-z]+/, '')
    YAML.safe_load(cleaned_diff)
  end
end

if $0 == __FILE__
  options = API::DEFAULT_OPTIONS.dup

  OptionParser.new do |opts|
    opts.on("-f", "--files FILES", String, "A directory containing diffs including feature flag definition change diffs") do |value|
      options[:files] = value
    end

    opts.on("-s", "--state STATE", String,
      "The desired state of the feature flags (enabled or disabled). If not specified the output will only list the feature flags."
    ) do |value|
      options[:state] = value
    end

    opts.on("-h", "--help", "Prints this help") do
      puts opts
      exit
    end
  end.parse!

  puts GetFeatureFlagsFromFiles.new(options).extracted_flags
end