# frozen_string_literal: true
require 'open3'
require 'tmpdir'

def create_facter_gem
  temp_dir = Dir.mktmpdir
  Dir.chdir(temp_dir) do
    download_and_build_facter_ng

    facter_repo_dir = Pathname.new("#{temp_dir}/facter-ng")
    facter_gem_path = Dir.entries(facter_repo_dir).select { |file| file =~ /facter-ng-[0-9]+.[0-9]+.[0-9]+(.pre)?.gem/ }
    File.join(facter_repo_dir, facter_gem_path)
  end
end

def download_and_build_facter_ng
  branch_name = ENV['FACTER_4_BRANCH']
  puts "Cloning branch #{branch_name}"
  Open3.capture2('echo $PATH')

  Open3.capture2("git clone https://github.com/puppetlabs/facter-ng.git &&" \
  'cd facter-ng &&' \
  'git fetch &&' \
  "git reset --hard origin/#{branch_name}")

  Dir.chdir('facter-ng') do
    puts "Latest commit on branch #{branch_name}"
    output, _stderr = Open3.capture2('git log -1')
    puts output

    Open3.capture2('gem build agent/facter-ng.gemspec')
  end
end

def install_facter_gem(agent, facter_gem_path)
  home_dir = on(agent, 'pwd').stdout.chop
  gem_c = gem_command(agent)

  scp_to(agent, facter_gem_path, home_dir)

  on(agent, "#{gem_c} uninstall facter-ng")
  on(agent, "#{gem_c} install -f facter-ng-*.gem")
end

test_name 'Setup for Facter NG' do
  windows_puppet_bin_path = '/cygdrive/c/Program\ Files/Puppet\ Labs/Puppet/bin'
  linux_puppet_bin_path = '/opt/puppetlabs/puppet/bin'
  set_facter_ng_command = 'puppet config set facterng true'

  puts 'Setting run with facter ng if environment variable FACTER_NG is true.'
  puts "FACTER_NG is #{ENV["FACTER_NG"]}."
  puts "FACTER_NG branch is #{ENV['FACTER_4_BRANCH']}"

  if ENV["FACTER_NG"] == 'true'
    puts 'Cloning facter ng repository and creating gem file.'
    facter_gem_path = create_facter_gem

    agents.each do |agent|
      puts 'Installing Facter NG on agent.'
      install_facter_gem(agent, facter_gem_path)

      puts 'Renaming facter to facter-original and facter-ng to facter.'
      if agent['platform'] =~ /windows/
        on(agent, "cmd /c #{set_facter_ng_command}")
        on(agent, "cd #{windows_puppet_bin_path} && mv facter-ng.bat facter.bat")
      else
        on(agent, "#{set_facter_ng_command} ")
        on(agent, "cd #{linux_puppet_bin_path} && mv facter-ng facter")
      end

      puts 'FACTER VERSION'
      on(agent, "facter -v")
    end
  end
end