require 'pathname'

# Given an array of modules specified by the --modules command line option,
# Parse all of them into an array of usable hash structures.
class PuppetModules
  attr_reader :modules

  def initialize(modules = [])
    @modules = modules
  end

  def list
    return [] unless modules

    modules.collect do |uri|
      git_url, git_ref = uri.split '#'
      folder = Pathname.new(git_url).basename('.git')
      name = folder.to_s.split('-', 2)[1] || folder.to_s
      {
        name: name,
        url: git_url,
        folder: folder.to_s,
        ref: git_ref,
        protocol: git_url.split(':')[0].intern,
      }
    end
  end
end

def install_git_module(mod, hosts)
  # The idea here is that each test can symlink the modules they want from a
  # temporary directory to this location.  This will preserve the global
  # state of the system while allowing individual test cases to quickly run
  # with a module "installed" in the module path.
  moddir = '/opt/puppet-git-repos'
  target = "#{moddir}/#{mod[:name]}"

  step "Clone #{mod[:url]} if needed"
  on hosts, "test -d #{moddir} || mkdir -p #{moddir}"
  on hosts, "test -d #{target} || git clone #{mod[:url]} #{target}"
  step "Update #{mod[:name]} and check out revision #{mod[:ref]}"

  commands = ["cd #{target}",
              'remote rm origin',
              "remote add origin #{mod[:url]}",
              'fetch origin',
              "checkout -f #{mod[:ref]}",
              "reset --hard refs/remotes/origin/#{mod[:ref]}",
              'clean -fdx',]

  on hosts, commands.join(' && git ')
end

def install_scp_module(mod, hosts)
  moddir = '/opt/puppet-git-repos'
  target = "#{moddir}/#{mod[:name]}"

  step "Purge #{target} if needed"
  on hosts, "test -d #{target} && rm -rf #{target} || true"

  step "Copy #{mod[:name]} to hosts"
  scp_to hosts, mod[:url].split(':', 2)[1], target
end

test_name 'Install Puppet Modules'

skip_test 'not testing with puppetserver' unless @options['is_puppetserver']

modules = PuppetModules.new(options[:modules]).list

step 'Masters: Install Puppet Modules'
masters = hosts.select { |host| host['roles'].include? 'master' }

modules.each do |mod|
  if mod[:protocol] == :scp
    install_scp_module(mod, masters)
  else
    install_git_module(mod, masters)
  end
end