tnargav 1.3.5 → 1.3.6

data/plugins/provisioners/cfengine/cap/debian/cfengine_install.rb

@@ -0,0 +1,19 @@
+module VagrantPlugins
+  module CFEngine
+    module Cap
+      module Debian
+        module CFEngineInstall
+          def self.cfengine_install(machine, config)
+            machine.communicate.tap do |comm|
+              comm.sudo("mkdir -p #{File.dirname(config.deb_repo_file)} && /bin/echo #{config.deb_repo_line} > #{config.deb_repo_file}")
+              comm.sudo("GPGFILE=`tempfile`; wget -O $GPGFILE #{config.repo_gpg_key_url} && apt-key add $GPGFILE; rm -f $GPGFILE")
+
+              comm.sudo("apt-get update")
+              comm.sudo("apt-get install -y #{config.package_name}")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/cfengine/cap/linux/cfengine_installed.rb

@@ -0,0 +1,14 @@
+module VagrantPlugins
+  module CFEngine
+    module Cap
+      module Linux
+        module CFEngineInstalled
+          def self.cfengine_installed(machine)
+            machine.communicate.test(
+              "test -d /var/cfengine && test -x /var/cfengine/bin/cf-agent", sudo: true)
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/cfengine/cap/linux/cfengine_needs_bootstrap.rb

@@ -0,0 +1,34 @@
+require "log4r"
+
+module VagrantPlugins
+  module CFEngine
+    module Cap
+      module Linux
+        module CFEngineNeedsBootstrap
+          def self.cfengine_needs_bootstrap(machine, config)
+            logger = Log4r::Logger.new("vagrant::plugins::cfengine::cap_linux_cfengine_bootstrap")
+
+            machine.communicate.tap do |comm|
+              # We hardcode fixing the permissions on /var/cfengine/ppkeys/, if it exists,
+              # because otherwise CFEngine will fail to bootstrap.
+              if comm.test("test -d /var/cfengine/ppkeys", :sudo => true)
+                logger.debug("Fixing permissions on /var/cfengine/ppkeys")
+                comm.sudo("chmod -R 600 /var/cfengine/ppkeys")
+              end
+
+              logger.debug("Checking if CFEngine is bootstrapped...")
+              bootstrapped = comm.test("test -f /var/cfengine/policy_server.dat", :sudo => true)
+              if bootstrapped && !config.force_bootstrap
+                logger.info("CFEngine already bootstrapped, no need to do it again")
+                return false
+              end
+
+              logger.info("CFEngine needs bootstrap.")
+              return true
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/cfengine/cap/redhat/cfengine_install.rb

@@ -0,0 +1,24 @@
+require "log4r"
+
+module VagrantPlugins
+  module CFEngine
+    module Cap
+      module RedHat
+        module CFEngineInstall
+          def self.cfengine_install(machine, config)
+            logger = Log4r::Logger.new("vagrant::plugins::cfengine::cap_redhat_cfengine_install")
+
+            machine.communicate.tap do |comm|
+              logger.info("Adding the CFEngine repository to #{config.yum_repo_file}")
+              comm.sudo("mkdir -p #{File.dirname(config.yum_repo_file)} && (echo '[cfengine-repository]'; echo 'name=CFEngine Community Yum Repository'; echo 'baseurl=#{config.yum_repo_url}'; echo 'enabled=1'; echo 'gpgcheck=1') > #{config.yum_repo_file}")
+              logger.info("Installing CFEngine Community Yum Repository GPG KEY from #{config.repo_gpg_key_url}")
+              comm.sudo("GPGFILE=$(mktemp) && wget -O $GPGFILE #{config.repo_gpg_key_url} && rpm --import $GPGFILE; rm -f $GPGFILE")
+
+              comm.sudo("yum -y install #{config.package_name}")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/plugins/provisioners/cfengine/config.rb

@@ -0,0 +1,126 @@
+require "pathname"
+
+require "vagrant"
+
+module VagrantPlugins
+  module CFEngine
+    class Config < Vagrant.plugin("2", :config)
+      attr_accessor :am_policy_hub
+      attr_accessor :extra_agent_args
+      attr_accessor :classes
+      attr_accessor :deb_repo_file
+      attr_accessor :deb_repo_line
+      attr_accessor :files_path
+      attr_accessor :force_bootstrap
+      attr_accessor :install
+      attr_accessor :mode
+      attr_accessor :policy_server_address
+      attr_accessor :repo_gpg_key_url
+      attr_accessor :run_file
+      attr_accessor :upload_path
+      attr_accessor :yum_repo_file
+      attr_accessor :yum_repo_url
+      attr_accessor :package_name
+
+      def initialize
+        @am_policy_hub    = UNSET_VALUE
+        @classes          = UNSET_VALUE
+        @deb_repo_file    = UNSET_VALUE
+        @deb_repo_line    = UNSET_VALUE
+        @extra_agent_args = UNSET_VALUE
+        @files_path       = UNSET_VALUE
+        @force_bootstrap  = UNSET_VALUE
+        @install          = UNSET_VALUE
+        @mode             = UNSET_VALUE
+        @policy_server_address = UNSET_VALUE
+        @repo_gpg_key_url = UNSET_VALUE
+        @run_file         = UNSET_VALUE
+        @upload_path      = UNSET_VALUE
+        @yum_repo_file    = UNSET_VALUE
+        @yum_repo_url     = UNSET_VALUE
+        @package_name     = UNSET_VALUE
+      end
+
+      def finalize!
+        @am_policy_hub = false if @am_policy_hub == UNSET_VALUE
+
+        @classes = nil if @classes == UNSET_VALUE
+
+        if @deb_repo_file == UNSET_VALUE
+          @deb_repo_file = "/etc/apt/sources.list.d/cfengine-community.list"
+        end
+
+        if @deb_repo_line == UNSET_VALUE
+          @deb_repo_line = "deb http://cfengine.com/pub/apt $(lsb_release -cs) main"
+        end
+
+        @extra_agent_args = nil if @extra_agent_args == UNSET_VALUE
+
+        @files_path = nil if @files_path == UNSET_VALUE
+
+        @force_bootstrap = false if @force_bootstrap == UNSET_VALUE
+
+        @install = true if @install == UNSET_VALUE
+        @install = @install.to_sym if @install.respond_to?(:to_sym)
+
+        @mode = :bootstrap if @mode == UNSET_VALUE
+        @mode = @mode.to_sym
+
+        @run_file = nil if @run_file == UNSET_VALUE
+
+        @policy_server_address = nil if @policy_server_address == UNSET_VALUE
+
+        if @repo_gpg_key_url == UNSET_VALUE
+          @repo_gpg_key_url = "http://cfengine.com/pub/gpg.key"
+        end
+
+        @upload_path = "/tmp/vagrant-cfengine-file" if @upload_path == UNSET_VALUE
+
+        if @yum_repo_file == UNSET_VALUE
+          @yum_repo_file = "/etc/yum.repos.d/cfengine-community.repo"
+        end
+
+        if @yum_repo_url == UNSET_VALUE
+          @yum_repo_url = "http://cfengine.com/pub/yum/"
+        end
+
+        if @package_name == UNSET_VALUE
+            @package_name = "cfengine-community"
+        end
+      end
+
+      def validate(machine)
+        errors = _detected_errors
+
+        valid_modes = [:bootstrap, :single_run]
+        errors << I18n.t("vagrant.cfengine_config.invalid_mode") if !valid_modes.include?(@mode)
+
+        if @mode == :bootstrap
+          if !@policy_server_address && !@am_policy_hub
+            errors << I18n.t("vagrant.cfengine_config.policy_server_address")
+          end
+        end
+
+        if @classes && !@classes.is_a?(Array)
+          errors << I18n.t("vagrant.cfengine_config.classes_array")
+        end
+
+        if @files_path
+          expanded = Pathname.new(@files_path).expand_path(machine.env.root_path)
+          if !expanded.directory?
+            errors << I18n.t("vagrant.cfengine_config.files_path_not_directory")
+          end
+        end
+
+        if @run_file
+          expanded = Pathname.new(@run_file).expand_path(machine.env.root_path)
+          if !expanded.file?
+            errors << I18n.t("vagrant.cfengine_config.run_file_not_found")
+          end
+        end
+
+        { "CFEngine" => errors }
+      end
+    end
+  end
+end

data/plugins/provisioners/cfengine/plugin.rb

@@ -0,0 +1,42 @@
+require "vagrant"
+
+module VagrantPlugins
+  module CFEngine
+    class Plugin < Vagrant.plugin("2")
+      name "CFEngine Provisioner"
+      description <<-DESC
+      Provisions machines with CFEngine.
+      DESC
+
+      config(:cfengine, :provisioner) do
+        require_relative "config"
+        Config
+      end
+
+      guest_capability("debian", "cfengine_install") do
+        require_relative "cap/debian/cfengine_install"
+        Cap::Debian::CFEngineInstall
+      end
+
+      guest_capability("linux", "cfengine_needs_bootstrap") do
+        require_relative "cap/linux/cfengine_needs_bootstrap"
+        Cap::Linux::CFEngineNeedsBootstrap
+      end
+
+      guest_capability("linux", "cfengine_installed") do
+        require_relative "cap/linux/cfengine_installed"
+        Cap::Linux::CFEngineInstalled
+      end
+
+      guest_capability("redhat", "cfengine_install") do
+        require_relative "cap/redhat/cfengine_install"
+        Cap::RedHat::CFEngineInstall
+      end
+
+      provisioner(:cfengine) do
+        require_relative "provisioner"
+        Provisioner
+      end
+    end
+  end
+end

data/plugins/provisioners/cfengine/provisioner.rb

@@ -0,0 +1,135 @@
+require "log4r"
+require "vagrant"
+
+module VagrantPlugins
+  module CFEngine
+    class Provisioner < Vagrant.plugin("2", :provisioner)
+      def provision
+        @logger = Log4r::Logger.new("vagrant::plugins::cfengine")
+
+        @logger.info("Checking for CFEngine installation...")
+        handle_cfengine_installation
+
+        if @config.files_path
+          @machine.ui.info(I18n.t("vagrant.cfengine_installing_files_path"))
+          install_files(Pathname.new(@config.files_path).expand_path(@machine.env.root_path))
+        end
+
+        handle_cfengine_bootstrap if @config.mode == :bootstrap
+
+        if @config.mode == :single_run
+          # Just let people know
+          @machine.ui.info(I18n.t("vagrant.cfengine_single_run"))
+        end
+
+        if @config.run_file
+          @machine.ui.info(I18n.t("vagrant.cfengine_single_run_execute"))
+          path = Pathname.new(@config.run_file).expand_path(@machine.env.root_path)
+          machine.communicate.upload(path.to_s, @config.upload_path)
+          cfagent("-KI -f #{@config.upload_path}#{cfagent_classes_args}#{cfagent_extra_args}")
+        end
+      end
+
+      protected
+
+      # This runs cf-agent with the given arguments.
+      def cfagent(args, options=nil)
+        options ||= {}
+        command = "/var/cfengine/bin/cf-agent #{args}"
+
+        @machine.communicate.sudo(command, error_check: options[:error_check]) do |type, data|
+          if [:stderr, :stdout].include?(type)
+            # Output the data with the proper color based on the stream.
+            color = type == :stdout ? :green : :red
+            @machine.ui.info(
+              data,
+              :color => color, :new_line => false, :prefix => false)
+          end
+        end
+      end
+
+      # Returns the arguments for the classes configuration if they are
+      # set.
+      #
+      # @return [String]
+      def cfagent_classes_args
+        return "" if !@config.classes
+
+        args = @config.classes.map { |c| "-D#{c}" }.join(" ")
+        return " #{args}"
+      end
+
+      # Extra arguments for calles to cf-agent.
+      #
+      # @return [String]
+      def cfagent_extra_args
+        return "" if !@config.extra_agent_args
+        return " #{@config.extra_agent_args}"
+      end
+
+      # This handles checking if the CFEngine installation needs to
+      # be bootstrapped, and bootstraps if it does.
+      def handle_cfengine_bootstrap
+        @logger.info("Bootstrapping CFEngine...")
+        if !@machine.guest.capability(:cfengine_needs_bootstrap, @config)
+          @machine.ui.info(I18n.t("vagrant.cfengine_no_bootstrap"))
+          return
+        end
+
+        # Needs bootstrap, let's determine the parameters
+        policy_server_address = @config.policy_server_address
+        if !policy_server_address
+          policy_server_address = @machine.guest.capability(:read_ip_address)
+          raise Vagrant::Errors::CFEngineCantAutodetectIP if !policy_server_address
+          @machine.ui.info(I18n.t("vagrant.cfengine_detected_ip", address: policy_server_address))
+        end
+
+        @machine.ui.info(I18n.t("vagrant.cfengine_bootstrapping",
+                                policy_server: policy_server_address))
+        result = cfagent("--bootstrap #{policy_server_address}", error_check: false)
+        raise Vagrant::Errors::CFEngineBootstrapFailed if result != 0
+
+        # Policy hubs need to do additional things before they're ready
+        # to accept agents. Force that run now...
+        if @config.am_policy_hub
+          @machine.ui.info(I18n.t("vagrant.cfengine_bootstrapping_policy_hub"))
+          cfagent("-KI -f /var/cfengine/masterfiles/failsafe.cf#{cfagent_classes_args}")
+          cfagent("-KI #{cfagent_classes_args}#{cfagent_extra_args}")
+        end
+      end
+
+      # This handles verifying the CFEngine installation, installing it
+      # if it was requested, and so on. This method will raise exceptions
+      # if things are wrong.
+      def handle_cfengine_installation
+        if !@machine.guest.capability?(:cfengine_installed)
+          @machine.ui.warn(I18n.t("vagrant.cfengine_cant_detect"))
+          return
+        end
+
+        installed = @machine.guest.capability(:cfengine_installed)
+        if !installed || @config.install == :force
+          raise Vagrant::Errors::CFEngineNotInstalled if !@config.install
+
+          @machine.ui.info(I18n.t("vagrant.cfengine_installing"))
+          @machine.guest.capability(:cfengine_install, @config)
+
+          if !@machine.guest.capability(:cfengine_installed)
+            raise Vagrant::Errors::CFEngineInstallFailed
+          end
+        end
+      end
+
+      # This installs a set of files into the CFEngine folder within
+      # the machine.
+      #
+      # @param [Pathname] local_path
+      def install_files(local_path)
+        @logger.debug("Copying local files to CFEngine: #{local_path}")
+        @machine.communicate.sudo("rm -rf /tmp/cfengine-files")
+        @machine.communicate.upload(local_path.to_s, "/tmp/cfengine-files")
+        @machine.communicate.sudo("cp -R /tmp/cfengine-files/* /var/cfengine")
+      end
+    end
+  end
+end

data/plugins/provisioners/chef/config/base.rb

@@ -0,0 +1,116 @@
+module VagrantPlugins
+  module Chef
+    module Config
+      class Base < Vagrant.plugin("2", :config)
+        attr_accessor :arguments
+        attr_accessor :attempts
+        attr_accessor :binary_path
+        attr_accessor :binary_env
+        attr_accessor :custom_config_path
+        attr_accessor :formatter
+        attr_accessor :http_proxy
+        attr_accessor :http_proxy_user
+        attr_accessor :http_proxy_pass
+        attr_accessor :https_proxy
+        attr_accessor :https_proxy_user
+        attr_accessor :https_proxy_pass
+        attr_accessor :json
+        attr_accessor :log_level
+        attr_accessor :no_proxy
+        attr_accessor :node_name
+        attr_accessor :provisioning_path
+        attr_accessor :run_list
+        attr_accessor :file_cache_path
+        attr_accessor :file_backup_path
+        attr_accessor :verbose_logging
+
+        def initialize
+          super
+
+          @arguments         = UNSET_VALUE
+          @attempts          = UNSET_VALUE
+          @binary_path       = UNSET_VALUE
+          @binary_env        = UNSET_VALUE
+          @custom_config_path = UNSET_VALUE
+          @formatter         = UNSET_VALUE
+          @http_proxy        = UNSET_VALUE
+          @http_proxy_user   = UNSET_VALUE
+          @http_proxy_pass   = UNSET_VALUE
+          @https_proxy       = UNSET_VALUE
+          @https_proxy_user  = UNSET_VALUE
+          @https_proxy_pass  = UNSET_VALUE
+          @log_level         = UNSET_VALUE
+          @no_proxy          = UNSET_VALUE
+          @node_name         = UNSET_VALUE
+          @provisioning_path = UNSET_VALUE
+          @file_cache_path   = UNSET_VALUE
+          @file_backup_path  = UNSET_VALUE
+          @verbose_logging   = UNSET_VALUE
+
+          @json              = {}
+          @run_list          = []
+        end
+
+        def finalize!
+          @arguments         = nil if @arguments == UNSET_VALUE
+          @attempts          = 1 if @attempts == UNSET_VALUE
+          @binary_path       = nil if @binary_path == UNSET_VALUE
+          @binary_env        = nil if @binary_env == UNSET_VALUE
+          @custom_config_path = nil if @custom_config_path == UNSET_VALUE
+          @formatter         = nil if @formatter == UNSET_VALUE
+          @http_proxy        = nil if @http_proxy == UNSET_VALUE
+          @http_proxy_user   = nil if @http_proxy_user == UNSET_VALUE
+          @http_proxy_pass   = nil if @http_proxy_pass == UNSET_VALUE
+          @https_proxy       = nil if @https_proxy == UNSET_VALUE
+          @https_proxy_user  = nil if @https_proxy_user == UNSET_VALUE
+          @https_proxy_pass  = nil if @https_proxy_pass == UNSET_VALUE
+          @log_level         = :info if @log_level == UNSET_VALUE
+          @no_proxy          = nil if @no_proxy == UNSET_VALUE
+          @node_name         = nil if @node_name == UNSET_VALUE
+          @provisioning_path = nil if @provisioning_path == UNSET_VALUE
+          @file_backup_path  = "/var/chef/backup" if @file_backup_path == UNSET_VALUE
+          @file_cache_path   = "/var/chef/cache" if @file_cache_path == UNSET_VALUE
+          @verbose_logging   = false if @verbose_logging == UNSET_VALUE
+
+          # Make sure the log level is a symbol
+          @log_level = @log_level.to_sym
+        end
+
+        def merge(other)
+          super.tap do |result|
+            result.instance_variable_set(:@json, @json.merge(other.json))
+            result.instance_variable_set(:@run_list, (@run_list + other.run_list))
+          end
+        end
+
+        # Just like the normal configuration "validate" method except that
+        # it returns an array of errors that should be merged into some
+        # other error accumulator.
+        def validate_base(machine)
+          errors = _detected_errors
+
+          if @custom_config_path
+            expanded = File.expand_path(@custom_config_path, machine.env.root_path)
+            if !File.file?(expanded)
+              errors << I18n.t("vagrant.config.chef.custom_config_path_missing")
+            end
+          end
+
+          errors
+        end
+
+        # Adds a recipe to the run list
+        def add_recipe(name)
+          name = "recipe[#{name}]" unless name =~ /^recipe\[(.+?)\]$/
+            run_list << name
+        end
+
+        # Adds a role to the run list
+        def add_role(name)
+          name = "role[#{name}]" unless name =~ /^role\[(.+?)\]$/
+            run_list << name
+        end
+      end
+    end
+  end
+end