vagrant-pe_build 0.2.0 → 0.3.0

data/lib/pe_build/plugin.rb

@@ -6,33 +6,69 @@ if Vagrant::VERSION < "1.1.0"
 end
 
 module PEBuild
-class Plugin < Vagrant.plugin('2')
+  class Plugin < Vagrant.plugin('2')
 
-  name 'pe_build'
+    name 'pe_build'
 
-  description <<-DESC
-  This plugin adds commands and provisioners to automatically install Puppet
-  Enterprise on Vagrant guests.
-  DESC
+    description <<-DESC
+    This plugin adds commands and provisioners to automatically install Puppet
+    Enterprise on Vagrant guests.
+    DESC
 
-  config(:pe_bootstrap, :provisioner) do
-    require_relative 'config/pe_bootstrap'
-    PEBuild::Config::PEBootstrap
-  end
+    # User facing plugin configuration
 
-  config(:pe_build) do
-    require_relative 'config/global'
-    PEBuild::Config::Global
-  end
+    config(:pe_bootstrap, :provisioner) do
+      require_relative 'config/pe_bootstrap'
+      PEBuild::Config::PEBootstrap
+    end
 
-  provisioner(:pe_bootstrap) do
-    require_relative 'provisioner/pe_bootstrap'
-    PEBuild::Provisioner::PEBootstrap
-  end
+    config(:pe_build) do
+      require_relative 'config/global'
+      PEBuild::Config::Global
+    end
+
+    provisioner(:pe_bootstrap) do
+      require_relative 'provisioner/pe_bootstrap'
+      PEBuild::Provisioner::PEBootstrap
+    end
+
+    command(:'pe-build') do
+      require_relative 'command'
+      PEBuild::Command::Base
+    end
+
+    # Guest capabilities for installing PE
+
+    guest_capability('debian', 'detect_installer') do
+      require_relative 'cap'
+      PEBuild::Cap::DetectInstaller::Debian
+    end
 
-  command(:'pe-build') do
-    require_relative 'command'
-    PEBuild::Command
+    guest_capability('redhat', 'detect_installer') do
+      require_relative 'cap'
+      PEBuild::Cap::DetectInstaller::Redhat
+    end
+
+    guest_capability('ubuntu', 'detect_installer') do
+      require_relative 'cap'
+      PEBuild::Cap::DetectInstaller::Ubuntu
+    end
+
+    guest_capability('linux', 'run_install') do
+      require_relative 'cap'
+      PEBuild::Cap::RunInstall::POSIX
+    end
+
+    guest_capability('solaris', 'run_install') do
+      require_relative 'cap'
+      PEBuild::Cap::RunInstall::POSIX
+    end
+
+    # internal action hooks
+
+    action_hook('PE Build: initialize build dir') do |hook|
+      require 'pe_build/action'
+      hook.prepend PEBuild::Action::PEBuildDir
+    end
   end
 end
-end

data/lib/pe_build/provisioner/pe_bootstrap.rb

@@ -5,198 +5,165 @@ require 'pe_build/util/config'
 
 require 'log4r'
 require 'fileutils'
-require 'erb'
 
 module PEBuild
-module Provisioner
+  module Provisioner
+    class PEBootstrap < Vagrant.plugin('2', :provisioner)
 
-class PEBootstrap < Vagrant.plugin('2', :provisioner)
+      require 'pe_build/provisioner/pe_bootstrap/answers_file'
 
-  # @!attribute [r] work_dir
-  #   @return [String] The path to the machine pe_build working directory
+      # @!attribute [r] work_dir
+      #   @return [String] The path to the machine pe_build working directory
 
-  attr_reader :work_dir
+      attr_reader :work_dir
 
-  # @!attribute [r] answer_dir
-  #   @return [String] The path to the default answer file template dir
-  attr_reader :answer_dir
+      # @!attribute [r] answer_dir
+      #   @return [String] The path to the default answer file template dir
+      attr_reader :answer_dir
 
-  # @!attribute [r] answer_file
-  #   @return [String] The path to the answer file for this machine.
-  attr_reader :answer_file
+      # @!attribute [r] answer_file
+      #   @return [String] The path to the answer file for this machine.
+      attr_reader :answer_file
 
-  def initialize(machine, config)
-    super
+      def initialize(machine, config)
+        super
 
-    @logger = Log4r::Logger.new('vagrant::provisioners::pe_bootstrap')
+        @logger = Log4r::Logger.new('vagrant::provisioners::pe_bootstrap')
 
-    @work_dir    = File.join(@machine.env.root_path, '.pe_build')
-    @answer_dir  = File.join(work_dir, 'answers')
-  end
-
-  # Instantiate all working directory content and stage the PE installer.
-  #
-  # @param root_config [Object] ???
-  # @return [void]
-  def configure(root_config)
-    late_config_merge(root_config)
-
-    unless File.directory? work_dir
-      FileUtils.mkdir_p work_dir
-    end
-
-    unless File.directory? answer_dir
-      FileUtils.mkdir_p answer_dir
-    end
-
-    archive = PEBuild::Archive.new(@config.filename, @machine.env)
-    archive.version = @config.version
-
-    archive.download_from(@config.download_root)
-    archive.unpack_to(@work_dir)
-  end
-
-  def provision
-    prepare_answers_file
-
-    [:base, @config.role].each do |rolename|
-      process_step rolename, :pre
-    end
-
-    perform_installation
+        @work_dir   = File.join(@machine.env.root_path.join, PEBuild::WORK_DIR)
+        @answer_dir = File.join(work_dir, 'answers')
+      end
 
-    if @config.relocate_manifests
-      relocate_installation
-    end
+      # Instantiate all working directory content and stage the PE installer.
+      #
+      # @param root_config [Object] ???
+      # @return [void]
+      def configure(root_config)
+        late_config_merge(root_config)
 
-    [:base, @config.role].each do |rolename|
-      process_step rolename, :post
-    end
-  end
-
-  private
+        unless File.directory? work_dir
+          FileUtils.mkdir_p work_dir
+        end
+      end
 
-  def late_config_merge(root_config)
-    provision = @config
-    global    = root_config.pe_build
+      def provision
+        prepare_answers_file
+        load_archive
+        fetch_installer
 
-    # We don't necessarily know if the configs have been merged. If a config
-    # is being used for default values and was never directly touched then it
-    # may have bad values, so we re-finalize everything. This may not be
-    # generally safe but inside of this plugin it should be ok.
-    provision.finalize!
-    global.finalize!
+        [:base, @config.role].each { |rolename| process_step rolename, :pre }
 
-    merged = PEBuild::Util::Config.local_merge(provision, global)
+        perform_installation
+        relocate_installation if @config.relocate_manifests
 
-    @config = merged
-  end
+        [:base, @config.role].each { |rolename| process_step rolename, :post }
+      end
 
-  def generate_answers
-    if @config.answer_file
-      template_path = @config.answer_file
-    else
-      default_template_path = File.join(PEBuild.template_dir, 'answers', "#{@config.role}.txt.erb")
-      template_path = default_template_path
-    end
-    @machine.ui.info "Using #{template_path} as answers template"
-    template = File.read(template_path)
-    str = ERB.new(template).result(binding)
-  end
+      private
 
-  def prepare_answers_file
-    str = generate_answers
+      def late_config_merge(root_config)
+        provision = @config
+        global    = root_config.pe_build
 
-    dest_file = File.join(@answer_dir, "#{@machine.name}.txt")
+        # We don't necessarily know if the configs have been merged. If a config
+        # is being used for default values and was never directly touched then it
+        # may have bad values, so we re-finalize everything. This may not be
+        # generally safe but inside of this plugin it should be ok.
+        provision.finalize!
+        global.finalize!
 
-    @machine.ui.info "Writing answers file to #{dest_file}"
-    File.open(dest_file, "w") do |file|
-      file.write(str)
-    end
-  end
+        merged = PEBuild::Util::Config.local_merge(provision, global)
 
-  def process_step(role, stepname)
+        @config = merged
+      end
 
-    if role != :base && config.step[stepname]
-      if File.file? config.step[stepname]
-        script_list = [*config.step[stepname]]
-      else
-        script_list = []
-        @machine.ui.warn "Cannot find defined step for #{role}/#{stepname.to_s} at \'#{config.step[stepname]}\'"
+      def prepare_answers_file
+        af = AnswersFile.new(@machine, @config, @work_dir)
+        af.generate
       end
-    else
-      # We do not have a user defined step for this role or we're processing the :base step
-      script_dir  = File.join(PEBuild.source_root, 'bootstrap', role.to_s, stepname.to_s)
-      script_list = Dir.glob("#{script_dir}/*")
-    end
 
-    if script_list.empty?
-      @logger.info "No steps for #{role}/#{stepname}", :color => :cyan
-    end
+      def load_archive
+        if @config.suffix == :detect
+          filename = @machine.guest.capability('detect_installer', @config.version)
+        else
+          filename = @config.filename
+        end
 
-    script_list.each do |template_path|
-      # A message to show which step's action is running
-      @machine.ui.info "Running action for #{role}/#{stepname}"
-      template = File.read(template_path)
-      contents = ERB.new(template).result(binding)
+        @archive = PEBuild::Archive.new(filename, @machine.env)
+        @archive.version = @config.version
+      end
 
-      on_remote contents
-    end
-  end
+      # @todo panic if @config.download_root is undefined
+      def fetch_installer
+        uri = @config.download_root
+        @archive.fetch(@config.download_root)
+        @archive.unpack_to(@work_dir)
+      end
 
-  # Determine the proper invocation of the PE installer
-  def installer_cmd
-    root = "/vagrant/.pe_build"
+      def process_step(role, stepname)
 
-    installer_dir = "puppet-enterprise-#{@config.version}-#{@config.suffix}"
-    installer     = "puppet-enterprise-installer"
+        if role != :base && config.step[stepname]
+          if File.file? config.step[stepname]
+            script_list = [*config.step[stepname]]
+          else
+            script_list = []
+            @machine.ui.warn "Cannot find defined step for #{role}/#{stepname.to_s} at \'#{config.step[stepname]}\'"
+          end
+        else
+          # We do not have a user defined step for this role or we're processing the :base step
+          script_dir  = File.join(PEBuild.source_root, 'bootstrap', role.to_s, stepname.to_s)
+          script_list = Dir.glob("#{script_dir}/*")
+        end
 
-    answers     = "#{root}/answers/#{@machine.name}.txt"
-    log_file    = "/root/puppet-enterprise-installer-#{Time.now.strftime('%s')}.log"
+        if script_list.empty?
+          @logger.info "No steps for #{role}/#{stepname}"
+        end
 
-    cmd = File.join(root, installer_dir, installer)
+        script_list.each do |template_path|
+          # A message to show which step's action is running
+          @machine.ui.info "Running action for #{role}/#{stepname}"
+          template = File.read(template_path)
+          contents = ERB.new(template).result(binding)
 
-    @installer_cmd = "#{cmd} -a #{answers} -l #{log_file}"
-  end
+          on_remote contents
+        end
+      end
 
-  def perform_installation
-    if @machine.communicate.test('test -f /opt/puppet/pe_version')
-      @machine.ui.warn "Puppet Enterprise is already installed, skipping installation.",
-        :name  => @machine.name
-    else
-      on_remote installer_cmd
-      @machine.ui.info "Scheduling puppet run to prime pe_mcollective"
-      on_remote "echo '/opt/puppet/bin/puppet agent -t' | at next minute"
-    end
-  end
+      def perform_installation
+        if @machine.communicate.test('test -f /opt/puppet/pe_version')
+          @machine.ui.warn I18n.t('pebuild.provisioner.pe_bootstrap.already_installed'),
+            :name  => @machine.name
+        else
+          @machine.guest.capability('run_install', @config, @archive)
+        end
+      end
 
-  # Modify the PE puppet master config to use alternate /manifests and /modules
-  #
-  # Manifests and modules need to be mounted on the master via shared folders,
-  # but the default /vagrant mount has permissions and ownership that conflicts
-  # with the puppet master process and the pe-puppet user. Those directories
-  # need to be mounted with permissions like 'fmode=644,dmode=755,fmask=022,dmask=022'
-  #
-  def relocate_installation
-    script_path = File.join(PEBuild.template_dir, 'scripts', 'relocate_installation.sh')
-    script = File.read script_path
-    on_remote script
-  end
+      # Modify the PE puppet master config to use alternate /manifests and /modules
+      #
+      # Manifests and modules need to be mounted on the master via shared folders,
+      # but the default /vagrant mount has permissions and ownership that conflicts
+      # with the puppet master process and the pe-puppet user. Those directories
+      # need to be mounted with permissions like 'fmode=644,dmode=755,fmask=022,dmask=022'
+      #
+      def relocate_installation
+        script_path = File.join(PEBuild.template_dir, 'scripts', 'relocate_installation.sh')
+        script = File.read script_path
+        on_remote script
+      end
 
-  def on_remote(cmd)
-    @machine.communicate.sudo(cmd) do |type, data|
-      if type == :stdout
-        if @config.verbose
-          @machine.ui.info(data.chomp, :color => :green, :prefix => true)
-        else
-          @machine.ui.info('.', :color => :green)
+      def on_remote(cmd)
+        @machine.communicate.sudo(cmd) do |type, data|
+          if type == :stdout
+            if @config.verbose
+              @machine.ui.info(data.chomp, :color => :green, :prefix => true)
+            else
+              @machine.ui.info('.', :color => :green)
+            end
+          else
+            @machine.ui.info(data.chomp, :color => :red, :prefix => true)
+          end
         end
-      else
-        @machine.ui.info(data.chomp, :color => :red, :prefix => true)
       end
     end
   end
 end
-
-end
-end

data/lib/pe_build/provisioner/pe_bootstrap/answers_file.rb

@@ -0,0 +1,49 @@
+require 'pe_build/release'
+
+require 'erb'
+
+class PEBuild::Provisioner::PEBootstrap::AnswersFile
+
+  # @param machine [Vagrant::Machine]
+  # @param config [Object < Vagrant.plugin('2', :config)]
+  # @param work_dir [String]
+  def initialize(machine, config, work_dir)
+    @machine, @config = machine, config
+    @work_dir = Pathname.new(work_dir)
+
+    @logger = Log4r::Logger.new('vagrant::provisioner::pe_bootstrap::answers_file')
+
+    @answer_dir  = @work_dir.join('answers')
+    @output_file = @answer_dir.join "#{@machine.name}.txt"
+
+    set_template_path
+  end
+
+  def generate
+    @logger.info "Writing answers file for #{@machine.inspect} to #{@output_file}"
+    @answer_dir.mkpath unless @answer_dir.exist?
+
+    @output_file.open('w') { |fh| fh.write(render_answers) }
+  end
+
+  private
+
+  def set_template_path
+    if @config.answer_file
+      @template = @config.answer_file
+      mode = 'explicit'
+    else
+      release_info = PEBuild::Release[@config.version]
+
+      @template = release_info.answer_file(@config.role)
+      mode = 'default'
+    end
+
+    @logger.info "Using #{mode} answers file template #{@template} for #{@machine.inspect}"
+  end
+
+  def render_answers
+    template_data = File.read(@template)
+    ERB.new(template_data).result(binding)
+  end
+end