module Librarian::Puppet::Simple::Installer

# File lib/librarian/puppet/simple/installer.rb, line 23
def install!
  each_module do |repo|

    print_verbose "\n##### processing module #{repo[:name]}..."

    module_dir = File.join(module_path, repo[:name])

    unless File.exists?(module_dir)
      case
      when repo[:git]
        install_git module_path, repo[:name], repo[:git], repo[:ref]
      when repo[:tarball]
        install_tarball module_path, repo[:name], repo[:tarball]
      else
        abort('only the :git and :tarball provider are currently supported')
      end
    else
      print_verbose "\nModule #{repo[:name]} already installed in #{module_path}"
    end
  end
end

# File lib/librarian/puppet/simple/installer.rb, line 48
def install_git(module_path, module_name, repo, ref = nil)
  module_dir = File.join(module_path, module_name)

  Dir.chdir(module_path) do
    print_verbose "cloning #{repo}"
    system_cmd("git clone #{repo} #{module_name}")
    Dir.chdir(module_dir) do
      system_cmd('git branch -r')
      system_cmd("git checkout #{ref}") if ref
    end
  end
end

# File lib/librarian/puppet/simple/installer.rb, line 61
def install_tarball(module_path, module_name, remote_tarball)
  Dir.mktmpdir do |tmp|
    temp_file = File.join(tmp,"downloaded_module.tar.gz")
    File.open(temp_file, "w+b") do |saved_file|
      print_verbose "Downloading #{remote_tarball}..."
      open(remote_tarball, 'rb') do |read_file|
        saved_file.write(read_file.read)
      end
      saved_file.rewind

      target_directory = File.join(module_path, module_name)
      print_verbose "Extracting #{remote_tarball} to #{target_directory}..."
      unzipped_target = ungzip(saved_file)
      tarfile_full_name = untar(unzipped_target, module_path)
      FileUtils.mv File.join(module_path, tarfile_full_name), target_directory
    end
  end
end

# File lib/librarian/puppet/simple/installer.rb, line 82
def ungzip(tarfile)
  z = Zlib::GzipReader.new(tarfile)
  unzipped = StringIO.new(z.read)
  z.close
  unzipped
end

# File lib/librarian/puppet/simple/installer.rb, line 91
def untar(io, destination)
  tarfile_full_name = nil
  Gem::Package::TarReader.new io do |tar|
    tar.each do |tarfile|
      tarfile_full_name ||= tarfile.full_name
      destination_file = File.join destination, tarfile.full_name
      if tarfile.directory?
        FileUtils.mkdir_p destination_file
      else
        destination_directory = File.dirname(destination_file)
        FileUtils.mkdir_p destination_directory unless File.directory?(destination_directory)
        File.open destination_file, "wb" do |f|
          f.write tarfile.read
        end
      end
    end
  end
  tarfile_full_name
end