orange-cloudfoundry/cf-ops-automation

require 'yaml'

class ConcoursePipelineResourceConfigGenerator
  def initialize(base_dir = ".", config_dir = '', templates_dir = '', output_dir = '')
    @pipelines = { 'pipelines' => [] }
    @pipelines_base_dir = base_dir
    @config_dir = config_dir
    @templates_dir = templates_dir
    @output_dir = output_dir
  end

  def execute
    validate_dir
    puts "execute"
    teams = list_teams
    puts "teams: #{teams}"
    teams.each do |team_name|
      puts "processing team #{team_name}"
      root_deployments = list_root_deployments(team_name)
      root_deployments.each do |root_deployment_name|
        puts "processing root_deployment #{root_deployment_name}"
        pipelines = list_pipelines(root_deployment_name, team_name)
        pipelines.each do |pipeline_config_filename|
          puts "processing #{pipeline_config_filename}"
          pipeline_name = File.basename(pipeline_config_filename, '.yml')
          vars_files = generate_vars_files(pipeline_name, root_deployment_name)
          add_pipeline(pipeline_name, team_name, pipeline_config_filename, vars_files)
        end
      end
    end
    @pipelines['pipelines'] = @pipelines['pipelines'].sort_by { |pipeline| pipeline['name'] }
    write_yaml
  end

  private

  def write_yaml
    puts "pipelines:"
    puts @pipelines.to_yaml
    pipeline_config_file = File.join(@output_dir, 'pipelines-definitions.yml')
    File.open(pipeline_config_file, 'w') { |file| file.write(@pipelines.to_yaml) }
  end

  def list_teams
    Dir[File.join(@pipelines_base_dir, '*')].select { |item| File.directory?(item) }.map { |path| File.basename(path) }
  end

  def list_root_deployments(team = 'main')
    Dir[File.join(@pipelines_base_dir, team, '*')].select { |item| File.directory?(item) }.map { |path| File.basename(path) }
  end

  def list_pipelines(root_deployment, team = 'main')
    Dir[File.join(@pipelines_base_dir, team, root_deployment, '*')].select { |item| File.file?(item) }
  end

  def add_pipeline(name, team, config, vars_files)
    pipeline = {}
    pipeline['name'] = name
    pipeline['team'] = team || 'main'
    pipeline['config_file'] = config
    pipeline['vars_files'] = vars_files || []
    @pipelines['pipelines'] << pipeline
  end

  def generate_vars_files(pipeline_name, root_deployment)
    puts " INFO - ignores credentials include 'pipeline' or 'generated'"
    vars_files = Dir[File.join(@config_dir, 'credentials-*.yml')].reject { |file_path| filter_credentials_file(file_path) }&.sort
    config_file_suffix = pipeline_name.gsub('-generated', '')
    config_file_suffix += '-pipeline' unless config_file_suffix.end_with?('-pipeline')
    current_pipeline_config_file = File.join(@config_dir, "credentials-#{config_file_suffix}.yml")
    puts "INFO - checking existence of #{current_pipeline_config_file}"
    vars_files << current_pipeline_config_file if File.exist?(current_pipeline_config_file)
    versions_file = File.join(@templates_dir, root_deployment, "root-deployment.yml")
    if File.exist?(versions_file)
      vars_files << versions_file
    else
      puts "WARNING: missing version file: #{versions_file} for #{root_deployment}"
    end
    vars_files
  end

  def validate_dir
    error_message = ''
    error_message << "\nPipelines directory does not exist: #{@pipelines_base_dir}" unless File.exist?(@pipelines_base_dir)
    error_message << "\nConfig directory does not exist: #{@config_dir}" unless File.exist?(@config_dir)
    error_message << "\nTemplates directory does not exist: #{@templates_dir}" unless File.exist?(@templates_dir)
    error_message << "\nOutput directory does not exist: #{@output_dir}" unless File.exist?(@output_dir)
    raise error_message unless error_message.empty?
  end

  def filter_credentials_file(file_path)
    File.basename(file_path).include?('pipeline') || File.basename(file_path).include?('generated')
  end
end