Rakefile

# frozen_string_literal: true

require 'bundler/gem_tasks'
require 'rubocop/rake_task'
require 'fileutils'
require 'open3'
require 'pwsh/version'
require 'rspec/core/rake_task'
require 'yard'

RSpec::Core::RakeTask.new(:spec) do |t|
  t.pattern = 'spec/unit/**/*_spec.rb'
end
task default: :spec

namespace :spec do
  desc 'Run RSpec code examples with coverage collection'
  task :coverage do
    ENV['COVERAGE'] = 'yes'
    Rake::Task['spec'].execute
  end
end

YARD::Rake::YardocTask.new do |t|
end
# Used in vendor_dsc_module
TAR_LONGLINK = '././@LongLink'

# Vendor a Puppetized DSC Module to spec/fixtures/modules.
#
# This is necessary because `puppet module install` fails on modules with
# long file paths, like xpsdesiredstateconfiguration
#
# @param command [String] command to execute.
# @return [Object] the standard out stream.
def vendor_dsc_module(name, version, destination)
  require 'open-uri'
  require 'rubygems/package'
  require 'zlib'

  module_uri = "https://forge.puppet.com/v3/files/dsc-#{name}-#{version}.tar.gz"
  tar_gz_archive = File.expand_path("#{name}.tar.gz", ENV['TEMP'])

  # Download the archive from the forge
  File.open(tar_gz_archive, 'wb') do |file|
    file.write(URI.open(module_uri).read) # rubocop:disable Security/Open
  end

  # Unzip to destination
  # Taken directly from StackOverflow:
  # - https://stackoverflow.com/a/19139114
  Gem::Package::TarReader.new(Zlib::GzipReader.open(tar_gz_archive)) do |tar|
    dest = nil
    tar.each do |entry|
      if entry.full_name == TAR_LONGLINK
        dest = File.join(destination, entry.read.strip)
        next
      end
      dest ||= File.join(destination, entry.full_name)
      if entry.directory?
        File.delete(dest) if File.file?(dest)
        FileUtils.mkdir_p(dest, mode: entry.header.mode, verbose: false)
      elsif entry.file?
        FileUtils.rm_rf(dest) if File.directory?(dest)
        File.open(dest, 'wb') do |f|
          f.print(entry.read)
        end
        FileUtils.chmod(entry.header.mode, dest, verbose: false)
      elsif entry.header.typeflag == '2' # Symlink!
        File.symlink(entry.header.linkname, dest)
      end
      dest = nil
    end
  end

  # Rename folder to just the module name, as needed by Puppet
  Dir.glob("#{destination}/*#{name}*").each do |existing_folder|
    new_folder = File.expand_path(name, destination)
    FileUtils.mv(existing_folder, new_folder)
  end
end

# Ensure that winrm is configured on the target system.
#
# @return [Object] The result of the command execution.
def configure_winrm
  return unless Gem.win_platform?

  command = 'pwsh.exe -NoProfile -NonInteractive -NoLogo -ExecutionPolicy Bypass -File "spec/acceptance/support/setup_winrm.ps1"'
  system(command)
rescue StandardError => e
  puts "Failed to configure WinRM: #{e}"
  exit 1
end

RSpec::Core::RakeTask.new(:acceptance) do |t|
  t.pattern = 'spec/acceptance/dsc/*.rb'
  end

namespace :acceptance do
  desc 'Prep for running DSC acceptance tests'
  task :spec_prep do
    # Create the modules fixture folder, if needed
    modules_folder = File.expand_path('spec/fixtures/modules', File.dirname(__FILE__))
    FileUtils.mkdir_p(modules_folder) unless Dir.exist?(modules_folder)
    # symlink the parent folder to the modules folder for puppet
    symlink_path = File.expand_path('pwshlib', modules_folder)
    File.symlink(File.dirname(__FILE__), symlink_path) unless Dir.exist?(symlink_path)
    # Install each of the required modules for acceptance testing
    # Note: This only works for modules in the dsc namespace on the forge.
    puppetized_dsc_modules = [
      { name: 'powershellget', version: '2.2.5-0-1' },
      { name: 'jeadsc', version: '0.7.2-0-3' },
      { name: 'xpsdesiredstateconfiguration', version: '9.1.0-0-1' },
      { name: 'xwebadministration', version: '3.2.0-0-2' },
      { name: 'accesscontroldsc', version: '1.4.1-0-3' }
    ]
    puppetized_dsc_modules.each do |puppet_module|
      next if Dir.exist?(File.expand_path(puppet_module[:name], modules_folder))

      vendor_dsc_module(puppet_module[:name], puppet_module[:version], modules_folder)
    end

    # Configure WinRM for acceptance tests
    configure_winrm
  end
end

task :acceptance => 'acceptance:spec_prep'