vagrant-pe_build 0.12.0 → 0.13.0

data/lib/pe_build/cap/facts/solaris.rb

@@ -0,0 +1,46 @@
+require_relative 'posix'
+
+# Facts implementation for Solaris guests
+#
+# @since 0.13.0
+class PEBuild::Cap::Facts::Solaris < PEBuild::Cap::Facts::POSIX
+
+  # (see PEBuild::Cap::Facts::Base#os_info)
+  #
+  # Currently returns `family` as `Solaris`.
+  #
+  # @see PEBuild::Cap::Facts::Base#os_info
+  def os_info
+    {
+      'family' => 'Solaris'
+    }
+  end
+
+  # (see PEBuild::Cap::Facts::Base#release_info)
+  #
+  # Reads `/etc/release` and generates a `full` version along with a
+  # `major` component.
+  #
+  # @todo Capture full version string. I.E 11.2, 10u11, etc and add `minor`
+  #   component.
+  #
+  # @see PEBuild::Cap::Facts::Base#release_info
+  def release_info
+    release_file = sudo('cat /etc/release')[:stdout]
+
+    # Cribbed from Facter 2.4.
+    if match = release_file.match(/\s+s(\d+)[sx]?(_u\d+)?.*(?:SPARC|X86)/)
+      version = match.captures.join('')
+    elsif match = release_file.match(/Solaris ([0-9\.]+(?:\s*[0-9\.\/]+))\s*(?:SPARC|X86)/)
+      version = match.captures.first
+    else
+      version = sudo('uname -v')[:stdout]
+    end
+
+    {
+      'major' => version.scan(/\d+/).first,
+      'full'  => version
+    }
+  end
+
+end

data/lib/pe_build/cap/facts/suse.rb

@@ -0,0 +1,38 @@
+require_relative 'posix'
+
+# Facts implementation for SUSE guests
+#
+# @since 0.13.0
+class PEBuild::Cap::Facts::SUSE < PEBuild::Cap::Facts::POSIX
+
+  # (see PEBuild::Cap::Facts::Base#os_info)
+  #
+  # Returns `family` as `SUSE` and `name` as `SLES`.
+  #
+  # @see PEBuild::Cap::Facts::Base#os_info
+  def os_info
+    {
+      'name'   => 'SLES',
+      'family' => 'SUSE'
+    }
+  end
+
+  # (see PEBuild::Cap::Facts::Base#release_info)
+  #
+  # Reads `/etc/SuSE-release` and generates a `full` version along with
+  # `major` and `minor` components.
+  #
+  # @see PEBuild::Cap::Facts::Base#release_info
+  def release_info
+    release_file = sudo('cat /etc/SuSE-release')[:stdout]
+    major = release_file.match(/VERSION\s*=\s*(\d+)/)[1]
+    minor = release_file.match(/PATCHLEVEL\s*=\s*(\d+)/)[1]
+
+    {
+      'major' => major,
+      'minor' => minor,
+      'full'  => [major, minor].join('.')
+    }
+  end
+
+end

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

@@ -0,0 +1,35 @@
+require_relative 'posix'
+
+# Facts implementation for Ubuntu guests
+#
+# @since 0.13.0
+class PEBuild::Cap::Facts::Ubuntu < PEBuild::Cap::Facts::POSIX
+
+  # (see PEBuild::Cap::Facts::Base#os_info)
+  #
+  # Returns `family` as `Debian` and `name` as `Ubuntu`.
+  #
+  # @see PEBuild::Cap::Facts::Base#os_info
+  def os_info
+    {
+      'name'   => 'Ubuntu',
+      'family' => 'Debian'
+    }
+  end
+
+
+  # (see PEBuild::Cap::Facts::Base#release_info)
+  #
+  # Reads `/etc/issue` and generates a `full` version.
+  #
+  # @see PEBuild::Cap::Facts::Base#release_info
+  def release_info
+    release_file = sudo('cat /etc/issue')[:stdout]
+    version = release_file.match(/Ubuntu (\d{2}\.\d{2})/)[1]
+
+    {
+      'full'  => version
+    }
+  end
+
+end

data/lib/pe_build/cap/facts/windows.rb

@@ -0,0 +1,58 @@
+require_relative 'base'
+
+# Facts implementation for Windows guests
+#
+# @since 0.13.0
+class PEBuild::Cap::Facts::Windows < PEBuild::Cap::Facts::Base
+
+  # (see PEBuild::Cap::Facts::Base#architecture)
+  #
+  # Looks at the default pointer size for integers and returns `x86` or `x64`.
+  #
+  # @see PEBuild::Cap::Facts::Base#architecture
+  def architecture
+    sudo('if ([System.IntPtr]::Size -eq 4) { "x86" } else { "x64" }')[:stdout]
+  end
+
+  # (see PEBuild::Cap::Facts::Base#os_info)
+  #
+  # Currently returns `family` as `Windows`.
+  #
+  # @see PEBuild::Cap::Facts::Base#os_info
+  def os_info
+    {
+      'family'  => 'Windows',
+    }
+  end
+
+  # (see PEBuild::Cap::Facts::Base#release_info)
+  #
+  # Queries WMI and generates a `full` version.
+  #
+  # @see PEBuild::Cap::Facts::Base#release_info
+  def release_info
+    version     = sudo('(Get-WmiObject -Class Win32_OperatingSystem).Version')[:stdout]
+    producttype = sudo('(Get-WmiObject -Class Win32_OperatingSystem).Producttype')[:stdout]
+
+    # Cribbed from Facter 2.4.
+    #
+    # NOTE: Currently doesn't support XP/Server 2003 or Windows 10.
+    name = case version
+    when /^6\.3/
+      producttype == 1 ? "8.1" : "2012 R2"
+    when /^6\.2/
+      producttype == 1 ? "8" : "2012"
+    when /^6\.1/
+      producttype == 1 ? "7" : "2008 R2"
+    when /^6\.0/
+      producttype == 1 ? "Vista" : "2008"
+    else
+      version # Default to the raw version number.
+    end
+
+    {
+      'full' => name
+    }
+  end
+
+end

data/lib/pe_build/command.rb

@@ -6,5 +6,6 @@ module PEBuild
     require 'pe_build/command/base'
     require 'pe_build/command/copy'
     require 'pe_build/command/list'
+    require 'pe_build/command/facts'
   end
 end

data/lib/pe_build/command/base.rb

@@ -3,7 +3,7 @@ require 'vagrant'
 class PEBuild::Command::Base < Vagrant.plugin(2, :command)
 
   def self.synopsis
-    'list and download PE installers'
+    'Commands related to PE Installation'
   end
 
   def initialize(argv, env)
@@ -13,6 +13,7 @@ class PEBuild::Command::Base < Vagrant.plugin(2, :command)
     @subcommands = {
       'list'     => PEBuild::Command::List,
       'copy'     => PEBuild::Command::Copy,
+      'facts'    => PEBuild::Command::Facts,
     }
   end
 

data/lib/pe_build/command/facts.rb

@@ -0,0 +1,63 @@
+require 'json'
+
+class PEBuild::Command::Facts < Vagrant.plugin(2, :command)
+
+  def self.synopsis
+    'Load facts from running VMs'
+  end
+
+  def execute
+    argv = parse_options(parser)
+    argv.shift # Remove 'facts' subcommand.
+
+    running = @env.active_machines.map do |name, provider|
+      @env.machine(name, provider)
+    end.select do |vm|
+      begin
+        vm.communicate.ready?
+      rescue Vagrant::Errors::VagrantError
+        # WinRM will raise an error if the VM isn't running instead of
+        # returning false (GH-6356).
+        false
+      end
+    end
+
+    # Filter the list of VMs for inspection down to just those passed on the
+    # command line. Warn if the user passed a VM that was not running.
+    unless argv.empty?
+      running_vms = running.map {|vm| vm.name.to_s}
+      argv.each do |name|
+        @env.ui.warn I18n.t('pebuild.command.facts.vm_not_running', :vm_name => name) unless running_vms.include? name
+      end
+
+      running.select! {|vm| argv.include? vm.name.to_s}
+    end
+
+    running.each do |vm|
+      facts = vm.guest.capability(:pebuild_facts)
+
+      @env.ui.machine('guest-facts', facts, {:target => vm.name.to_s})
+      @env.ui.info(JSON.pretty_generate(facts))
+    end
+
+    return 0
+  end
+
+  private
+
+  def parser
+    OptionParser.new do |o|
+      o.banner = <<-BANNER
+      Usage: vagrant pe-build facts [vm-name]
+      BANNER
+
+      o.separator ''
+
+      o.on('-h', '--help', 'Display this help') do
+        puts o
+        exit(0)
+      end
+    end
+  end
+
+end

data/lib/pe_build/config.rb

@@ -4,5 +4,6 @@ module PEBuild
   module Config
     require 'pe_build/config/global'
     require 'pe_build/config/pe_bootstrap'
+    require 'pe_build/config/pe_agent'
   end
 end

data/lib/pe_build/config/pe_agent.rb

@@ -0,0 +1,120 @@
+# Configuration for PE Agent provisioners
+#
+# @since 0.13.0
+class PEBuild::Config::PEAgent < Vagrant.plugin('2', :config)
+  # The minimum PE Version supported by this provisioner.
+  MINIMUM_VERSION    = '2015.2.0'
+
+  # @!attribute [rw] autosign
+  #   If true, and {#master_vm} is set, the agent's certificate will be signed
+  #   on the master VM.
+  #
+  #   @return [true, false] Defaults to `true` if {#master_vm} is set,
+  #     otherwise `false`.
+  attr_accessor :autosign
+
+  # @!attribute [rw] autopurge
+  #   If true, and {#master_vm} is set, the agent's certificate and data will
+  #   be purged from the master VM if the agent is destroyed by Vagrant.
+  #
+  #   @return [true, false] Defaults to `true` if {#master_vm} is set,
+  #     otherwise `false`.
+  attr_accessor :autopurge
+
+  # @!attribute master
+  #   @return [String] The DNS hostname of the Puppet master for this node.
+  #     If {#master_vm} is set, the hostname of that machine will be used
+  #     as a default.
+  attr_accessor :master
+
+  # @!attribute master_vm
+  #   @return [String] The name of a Vagrant VM to use as the master.
+  attr_accessor :master_vm
+
+  # @!attribute version
+  #   @return [String] The version of PE to install. May be either a version
+  #   string of the form `x.y.x[-optional-arbitrary-stuff]` or the string
+  #   `current`. Defaults to `current`.
+  attr_accessor :version
+
+  def initialize
+    @autosign      = UNSET_VALUE
+    @autopurge     = UNSET_VALUE
+    @master        = UNSET_VALUE
+    @master_vm     = UNSET_VALUE
+    @version       = UNSET_VALUE
+  end
+
+  def finalize!
+    @master        = nil if @master == UNSET_VALUE
+    @master_vm     = nil if @master_vm == UNSET_VALUE
+    @autosign      = (not @master_vm.nil?) if @autosign  == UNSET_VALUE
+    @autopurge     = (not @master_vm.nil?) if @autopurge == UNSET_VALUE
+    @version       = 'current' if @version == UNSET_VALUE
+  end
+
+  def validate(machine)
+    errors = _detected_errors
+
+    if @master.nil? && @master_vm.nil?
+      errors << I18n.t('pebuild.config.pe_agent.errors.no_master')
+    end
+
+    validate_master_vm!(errors, machine)
+    validate_version!(errors, machine)
+
+    {'pe_agent provisioner' => errors}
+  end
+
+  private
+
+  def validate_master_vm!(errors, machine)
+    if (not @master_vm.nil?) && (not machine.env.machine_names.include?(@master_vm.intern))
+      errors << I18n.t(
+        'pebuild.config.pe_agent.errors.master_vm_not_defined',
+        :vm_name  => @master_vm
+      )
+    end
+
+    if @autosign && @master_vm.nil?
+      errors << I18n.t(
+        'pebuild.config.pe_agent.errors.master_vm_required',
+        :setting  => 'autosign'
+      )
+    end
+
+    if @autopurge && @master_vm.nil?
+      errors << I18n.t(
+        'pebuild.config.pe_agent.errors.master_vm_required',
+        :setting  => 'autopurge'
+      )
+    end
+  end
+
+  def validate_version!(errors, machine)
+    pe_version_regex = %r[\d+\.\d+\.\d+[\w-]*]
+
+    if @version.kind_of? String
+      return if version == 'current'
+      if version.match(pe_version_regex)
+        unless PEBuild::Util::VersionString.compare(@version, MINIMUM_VERSION) > 0
+          errors << I18n.t(
+            'pebuild.config.pe_agent.errors.version_too_old',
+            :version         => @version,
+            :minimum_version => MINIMUM_VERSION
+          )
+        end
+
+        return
+      end
+    end
+
+    # If we end up here, the version was not a string that matched 'current' or
+    # the regex. Mutate the error array.
+    errors << I18n.t(
+      'pebuild.config.pe_agent.errors.malformed_version',
+      :version       => @version,
+      :version_class => @version.class
+    )
+  end
+end

data/lib/pe_build/config_builder.rb

@@ -2,5 +2,6 @@ module PEBuild
   module ConfigBuilder
     require 'pe_build/config_builder/global'
     require 'pe_build/config_builder/pe_bootstrap'
+    require 'pe_build/config_builder/pe_agent'
   end
 end

data/lib/pe_build/config_builder/pe_agent.rb

@@ -0,0 +1,44 @@
+require 'config_builder/model'
+
+# @since 0.13.0
+class PEBuild::ConfigBuilder::PEAgent < ::ConfigBuilder::Model::Base
+  # @!attribute [rw] autosign
+  #   If true, and {#master_vm} is set, the agent's certificate will be signed
+  #   on the master VM.
+  #
+  #   @return [true, false] Defaults to `true` if {#master_vm} is set,
+  #     otherwise `false`.
+  def_model_attribute :autosign
+  # @!attribute [rw] autopurge
+  #   If true, and {#master_vm} is set, the agent's certificate and data will
+  #   be purged from the master VM if the agent is destroyed by Vagrant.
+  #
+  #   @return [true, false] Defaults to `true` if {#master_vm} is set,
+  #     otherwise `false`.
+  def_model_attribute :autopurge
+  # @!attribute master
+  #   @return [String] The DNS hostname of the Puppet master for this node.
+  def_model_attribute :master
+  # @!attribute master_vm
+  #   @return [String] The name of a Vagrant VM to use as the master.
+  def_model_attribute :master_vm
+  # @!attribute version
+  #   @return [String] The version of PE to install. May be either a version
+  #   string of the form `x.y.x[-optional-arbitrary-stuff]` or the string
+  #   `current`. Defaults to `current`.
+  def_model_attribute :version
+
+  def to_proc
+    Proc.new do |vm_config|
+      vm_config.provision :pe_agent do |config|
+        with_attr(:autosign)     {|val| config.autosign     = val }
+        with_attr(:autopurge)    {|val| config.autopurge    = val }
+        with_attr(:master)       {|val| config.master       = val }
+        with_attr(:master_vm)    {|val| config.master_vm    = val }
+        with_attr(:version)      {|val| config.version      = val }
+      end
+    end
+  end
+
+  ::ConfigBuilder::Model::Provisioner.register('pe_agent', self)
+end