vagrant-pe_build 0.2.0 → 0.3.0

data/lib/pe_build/cap/detect_installer/posix.rb

@@ -0,0 +1,73 @@
+# Provide an abstract base class for detecting the installer name on
+# POSIX systems.
+#
+# @abstract
+# @protected
+class PEBuild::Cap::DetectInstaller::POSIX < PEBuild::Cap::DetectInstaller::Base
+
+  # @!method name
+  #   @abstract
+  #   @return [String] The name of the distribution
+
+  # @!method release_file
+  #   @abstract
+  #   @return [String] The file to use as the release file for the guest
+
+  # @!method release_file_format
+  #   @abstract
+  #   @return [Regexp] A regular expression with one capture that parses the distro version
+
+  # @!method supported_releases
+  #   @abstract
+  #   @return [Array<String>] All supported releases for the distribution
+
+
+  def detect
+    dist_version = parse_release_file
+
+    unless supported_releases.include? dist_version
+      raise "#{self.class.name} release #{dist_version} not supported"
+    end
+
+    "puppet-enterprise-#{@version}-#{name}-#{dist_version}-#{arch}.#{ext}"
+  end
+
+  def arch
+    content = ""
+    @machine.communicate.execute("uname -m") do |type, data|
+      raise "Could not run 'uname -m' on #{@machine}: got #{data}" if type == :stderr
+      content << data.chomp
+    end
+
+    content = 'i386' if content.match /i\d86/
+
+    content
+  end
+
+  def ext
+    'tar.gz'
+  end
+
+  private
+
+  def release_content
+    content = ""
+
+    @machine.communicate.execute("cat #{release_file}") do |type, data|
+      raise "Could not read #{release_file} on #{@machine}: got #{data}" if type == :stderr
+      content << data
+    end
+
+    content
+  end
+
+  def parse_release_file
+    matchdata = release_content.match(release_file_format)
+
+    if matchdata.nil? or matchdata[1].nil?
+      raise "#{self.class.name} could not determine release value: content #{release_content.inspect} did not match #{release_file_format}"
+    end
+
+    matchdata[1]
+  end
+end

data/lib/pe_build/cap/detect_installer/redhat.rb

@@ -0,0 +1,18 @@
+class PEBuild::Cap::DetectInstaller::Redhat < PEBuild::Cap::DetectInstaller::POSIX
+
+  def name
+    'el'
+  end
+
+  def release_file
+    '/etc/redhat-release'
+  end
+
+  def release_file_format
+    %r[release (\d+)\.\d+]
+  end
+
+  def supported_releases
+    %w[5 6]
+  end
+end

data/lib/pe_build/cap/detect_installer/ubuntu.rb

@@ -0,0 +1,23 @@
+class PEBuild::Cap::DetectInstaller::Ubuntu < PEBuild::Cap::DetectInstaller::POSIX
+
+  def name
+    'ubuntu'
+  end
+
+  def release_file
+    '/etc/issue'
+  end
+
+  def release_file_format
+    %r[Ubuntu (\d{2}\.\d{2})]
+  end
+
+  def supported_releases
+    %w[10.04 12.04]
+  end
+
+  def arch
+    retval = super
+    (retval == 'x86_64') ? 'amd64' : retval
+  end
+end

data/lib/pe_build/cap/run_install/posix.rb

@@ -0,0 +1,26 @@
+require 'pe_build/on_machine'
+class PEBuild::Cap::RunInstall::POSIX
+
+  extend PEBuild::OnMachine
+
+  def self.run_install(machine, config, archive)
+
+    root = File.join('/vagrant', PEBuild::WORK_DIR)
+
+    cmd_path = []
+    cmd_path << root
+
+    cmd_path << archive.installer_dir
+    cmd_path << "puppet-enterprise-installer"
+
+    cmd     = File.join(cmd_path)
+    answers = File.join(root, 'answers', "#{machine.name}.txt")
+
+    argv = "#{cmd} -a #{answers}"
+
+    on_machine(machine, argv)
+
+    machine.ui.info I18n.t('pebuild.provisioner.pe_bootstrap.scheduling_run')
+    machine.communicate.sudo("echo '/opt/puppet/bin/puppet agent -t' | at next minute")
+  end
+end

data/lib/pe_build/command.rb

@@ -1,68 +1,11 @@
 require 'vagrant'
+require 'optparse'
 
 module PEBuild
-class Command < Vagrant.plugin(2, :command)
-
-  def initialize(argv, env)
-    super
-
-    @main_args, @subcommand, @sub_args = split_main_and_subcommand(argv)
-
-    @subcommands = Vagrant::Registry.new
-
-    register_subcommands
-  end
-
-  def execute
-    if @subcommand and (klass = @subcommands.get(@subcommand))
-      klass.new(@argv, @env).execute
-    elsif @subcommand
-      raise "Unrecognized subcommand #{@subcommand}"
-    else
-      print_help
-    end
-  end
-
-  private
-
-  def register_subcommands
-    @subcommands.register('copy') do
-      require_relative 'command/copy'
-      PEBuild::Command::Copy
-    end
-
-    @subcommands.register('download') do
-      require_relative 'command/download'
-      PEBuild::Command::Download
-    end
-
-    @subcommands.register('list') do
-      require_relative 'command/list'
-      PEBuild::Command::List
-    end
+  module Command
+    require 'pe_build/command/base'
+    require 'pe_build/command/copy'
+    require 'pe_build/command/list'
+    require 'pe_build/command/download'
   end
-
-  def print_help
-    cmd = 'vagrant pe-build'
-    opts = OptionParser.new do |opts|
-      opts.banner = "Usage: #{cmd} <command> [<args>]"
-      opts.separator ""
-      opts.separator "Available subcommands:"
-
-      # Add the available subcommands as separators in order to print them
-      # out as well.
-      keys = []
-      @subcommands.each { |key, value| keys << key.to_s }
-
-      keys.sort.each do |key|
-        opts.separator "     #{key}"
-      end
-
-      opts.separator ""
-      opts.separator "For help on any individual command run `#{cmd} COMMAND -h`"
-    end
-
-    @env.ui.info(opts.help, :prefix => false)
-  end
-end
 end

data/lib/pe_build/command/base.rb

@@ -0,0 +1,55 @@
+require 'vagrant'
+
+class PEBuild::Command::Base < Vagrant.plugin(2, :command)
+
+  def initialize(argv, env)
+    super
+    split_argv
+
+    @subcommands = {
+      'list'     => PEBuild::Command::List,
+      'download' => PEBuild::Command::Download,
+      'copy'     => PEBuild::Command::Copy,
+    }
+  end
+
+  def execute
+    if @subcommand
+      execute_subcommand
+    else
+      print_help
+    end
+  end
+
+  private
+
+  def split_argv
+    @main_args, @subcommand, @sub_args = split_main_and_subcommand(@argv)
+  end
+
+  def execute_subcommand
+    if (klass = @subcommands[@subcommand])
+      klass.new(@argv, @env).execute
+    else
+      raise "Unrecognized subcommand #{@subcommand}"
+    end
+  end
+
+  def print_help
+    cmd = 'vagrant pe-build'
+    opts = OptionParser.new do |opts|
+      opts.banner = "Usage: #{cmd} <command> [<args>]"
+      opts.separator ""
+      opts.separator "Available subcommands:"
+
+      @subcommands.keys.sort.each do |key|
+        opts.separator "     #{key}"
+      end
+
+      opts.separator ""
+      opts.separator "For help on any individual command run `#{cmd} COMMAND -h`"
+    end
+
+    @env.ui.info(opts.help, :prefix => false)
+  end
+end

data/lib/pe_build/command/copy.rb

@@ -1,16 +1,31 @@
-require 'vagrant'
-require 'optparse'
 require 'pe_build/archive'
 
-module PEBuild
-class Command
-class Copy < Vagrant.plugin(2, :command)
+class PEBuild::Command::Copy < Vagrant.plugin(2, :command)
+
+  def initialize(argv, env)
+    super
+    @options = {}
+  end
 
   def execute
+    argv = parse_options(parser)
+
+    filename = File.basename(argv.last)
+    src_dir  = File.dirname(argv.last)
+
+    archive = PEBuild::Archive.new(filename, @env)
+    archive.version = @options[:version]
+
+    uri = URI.parse src_dir
+    archive.fetch(uri)
+
+    @env.ui.info "pe-build: #{archive} has been added and is ready for use!", :prefix => true
+  end
 
-    options = {}
+  private
 
-    parser = OptionParser.new do |o|
+  def parser
+    OptionParser.new do |o|
       o.banner = "Usage: vagrant pe-build copy path/to/installer.tar.gz"
       o.separator ''
 
@@ -18,21 +33,5 @@ class Copy < Vagrant.plugin(2, :command)
         options[:version] = val
       end
     end
-
-    argv = parse_options(parser)
-    fpath = argv.last
-
-    basename = File.basename(fpath)
-    dirname  = File.dirname(fpath)
-
-    #unless options[:version]
-    #  raise Vagrant::Errors::CLIInvalidUsage, :help => parser.help.chomp
-    #end
-
-    archive = PEBuild::Archive.new(fpath, @env)
-    archive.version = options[:version]
-    archive.copy_from(dirname)
   end
 end
-end
-end

data/lib/pe_build/command/download.rb

@@ -1,10 +1,6 @@
-require 'vagrant'
-require 'optparse'
 require 'pe_build/archive'
 
-module PEBuild
-class Command
-class Download < Vagrant.plugin(2, :command)
+class PEBuild::Command::Download < Vagrant.plugin(2, :command)
 
   def execute
 
@@ -30,10 +26,10 @@ class Download < Vagrant.plugin(2, :command)
       raise Vagrant::Errors::CLIInvalidUsage, :help => parser.help.chomp
     end
 
+    uri = URI.parse(options[:dir])
+
     archive = PEBuild::Archive.new(filename, @env)
     archive.version = options[:version]
-    archive.download_from(options[:dir])
+    archive.fetch(options[:dir])
   end
 end
-end
-end

data/lib/pe_build/command/list.rb

@@ -1,9 +1,6 @@
-require 'vagrant'
 require 'pe_build/archive_collection'
 
-module PEBuild
-class Command
-class List < Vagrant.plugin(2, :command)
+class PEBuild::Command::List < Vagrant.plugin(2, :command)
   def execute
     archive_dir = PEBuild.archive_directory(@env)
 
@@ -18,5 +15,3 @@ class List < Vagrant.plugin(2, :command)
     end
   end
 end
-end
-end

data/lib/pe_build/config/global.rb

@@ -2,6 +2,7 @@ require 'uri'
 require 'vagrant'
 
 require 'pe_build/config_default'
+require 'pe_build/transfer'
 
 module PEBuild
 module Config
@@ -45,7 +46,8 @@ class Global < Vagrant.plugin('2', :config)
   include PEBuild::ConfigDefault
 
   def finalize!
-    set_default :@suffix,   'all'
+    set_default :@suffix, :detect
+
     #set_default :@version,  DEFAULT_PE_VERSION
 
     set_default :@download_root, nil
@@ -66,9 +68,16 @@ class Global < Vagrant.plugin('2', :config)
       errors << "version only accepts a string, got #{@version.class}"
     end
 
-    uri = URI.parse(@download_root) rescue nil
-    unless @download_root == UNSET_VALUE or @download_root.nil? or File.directory?(File.expand_path(@download_root)) or uri.kind_of? (URI::HTTP||URI::HTTPS) or uri.kind_of? URI::FTP
-      errors << "download_root must be valid URL or present local file path, got #{@download_root}"
+    if @download_root and @download_root != UNSET_VALUE
+      begin
+        uri = URI.parse(@download_root)
+
+        if PEBuild::Transfer::IMPLEMENTATIONS[uri.scheme].nil?
+          errors << "No handlers available for URI scheme #{uri.scheme}"
+        end
+      rescue URI::InvalidURIError
+        errors << 'download_root must be a valid URL or nil'
+      end
     end
 
     {"PE Build global config" => errors}

data/lib/pe_build/config/pe_bootstrap.rb

@@ -77,6 +77,10 @@ class PEBootstrap < PEBuild::Config::Global
   def validate(machine)
     h = super
 
+    unless @step.empty?
+      machine.ui.warn "pe_bootstrap: explicit steps are deprecated and will be removed soon."
+    end
+
     errors = []
     if @version == UNSET_VALUE and global_config_from(machine).pe_build.version == UNSET_VALUE
       errors << "Version must be set on provisioner when unset globally"

data/lib/pe_build/idempotent.rb

@@ -7,7 +7,7 @@ module Idempotent
     desc ||= fpath
 
     if File.exist? fpath
-      @env.ui.warn "#{desc} is already present.", :prefix => true
+      @logger.info "#{desc} is already present."
     else
       yield
     end

data/lib/pe_build/on_machine.rb

@@ -0,0 +1,10 @@
+module PEBuild
+  module OnMachine
+    def on_machine(machine, cmd)
+      machine.communicate.sudo(cmd) do |type, data|
+        color = (type == :stdout) ? :green : :red
+        machine.ui.info(data.chomp, :color => color, :prefix => true)
+      end
+    end
+  end
+end