vaguerent/lib/vagrant/provisioners/chef.rb

module Vagrant
  module Provisioners
    # This class is a base class where the common functinality shared between
    # chef-solo and chef-client provisioning are stored. This is **not an actual
    # provisioner**. Instead, {ChefSolo} or {ChefServer} should be used.
    class Chef < Base
      def prepare
        raise ChefError.new(:invalid_provisioner)
      end

      def verify_binary(binary)
        vm.ssh.execute do |ssh|
          # Checks for the existence of chef binary and error if it
          # doesn't exist.
          ssh.exec!("which #{binary}", :error_class => ChefError, :_key => :chef_not_detected, :binary => binary)
        end
      end

      def chown_provisioning_folder
        vm.ssh.execute do |ssh|
          ssh.exec!("sudo mkdir -p #{env.config.chef.provisioning_path}")
          ssh.exec!("sudo chown #{env.config.ssh.username} #{env.config.chef.provisioning_path}")
        end
      end

      def setup_config(template, filename, template_vars)
        config_file = TemplateRenderer.render(template, {
          :log_level => env.config.chef.log_level.to_sym
        }.merge(template_vars))

        vm.ssh.upload!(StringIO.new(config_file), File.join(env.config.chef.provisioning_path, filename))
      end

      def setup_json
        env.ui.info I18n.t("vagrant.provisioners.chef.json")

        # Set up initial configuration
        data = {
          :config => env.config.to_hash,
          :directory => env.config.vm.shared_folders["v-root"][:guestpath],
        }

        # And wrap it under the "vagrant" namespace
        data = { :vagrant => data }

        # Merge with the "extra data" which isn't put under the
        # vagrant namespace by default
        data.merge!(env.config.chef.json)

        json = data.to_json

        vm.ssh.upload!(StringIO.new(json), File.join(env.config.chef.provisioning_path, "dna.json"))
      end
    end

    class Chef < Base
      class ChefError < Errors::VagrantError
        error_namespace("vagrant.provisioners.chef")
      end
    end

    class Chef < Base
      # This is the configuration which is available through `config.chef`
      class ChefConfig < Vagrant::Config::Base
        configures :chef

        # Chef server specific config
        attr_accessor :chef_server_url
        attr_accessor :validation_key_path
        attr_accessor :validation_client_name
        attr_accessor :client_key_path
        attr_accessor :node_name

        # Chef solo specific config
        attr_accessor :cookbooks_path
        attr_accessor :roles_path
        attr_accessor :recipe_url

        # Shared config
        attr_accessor :provisioning_path
        attr_accessor :log_level
        attr_accessor :json

        def initialize
          @validation_client_name = "chef-validator"
          @client_key_path = "/etc/chef/client.pem"
          @node_name = "client"

          @cookbooks_path = ["cookbooks", [:vm, "cookbooks"]]
          @roles_path = []
          @provisioning_path = "/tmp/vagrant-chef"
          @log_level = :info
          @json = {
            :instance_role => "vagrant",
            :run_list => []
          }
        end

        # Returns the run list for the provisioning
        def run_list
          json[:run_list]
        end

        # Sets the run list to the specified value
        def run_list=(value)
          json[:run_list] = value
        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

        def instance_variables_hash
          # Overridden so that the 'json' key could be removed, since its just
          # merged into the config anyways
          result = super
          result.delete("json")
          result
        end

        def validate(errors)
          if top.vm.provisioner == :chef_solo
            # Validate chef solo settings
            errors.add(I18n.t("vagrant.config.chef.cookbooks_path_empty")) if !cookbooks_path || [cookbooks_path].flatten.empty?
          end

          if top.vm.provisioner == :chef_server
            # Validate chef server settings
            errors.add(I18n.t("vagrant.config.chef.server_url_empty")) if !chef_server_url || chef_server_url.strip == ""
            errors.add(I18n.t("vagrant.config.chef.validation_key_path")) if !validation_key_path
          end

          if [:chef_solo, :chef_server].include?(top.vm.provisioner)
            # Validations shared by both chef solo and server
            errors.add(I18n.t("vagrant.config.chef.run_list_empty")) if !run_list || run_list.empty?
          end
        end
      end
    end
  end
end