115 lines
4.2 KiB
Ruby
115 lines
4.2 KiB
Ruby
require 'tempfile'
|
|
|
|
require "vagrant/util/counter"
|
|
require "vagrant/util/template_renderer"
|
|
|
|
module VagrantPlugins
|
|
module Chef
|
|
module Provisioner
|
|
# This class is a base class where the common functionality shared between
|
|
# chef-solo and chef-client provisioning are stored. This is **not an actual
|
|
# provisioner**. Instead, {ChefSolo} or {ChefServer} should be used.
|
|
class Base < Vagrant.plugin("2", :provisioner)
|
|
class ChefError < Vagrant::Errors::VagrantError
|
|
error_namespace("vagrant.provisioners.chef")
|
|
end
|
|
|
|
include Vagrant::Util::Counter
|
|
|
|
def initialize(machine, config)
|
|
super
|
|
|
|
config.provisioning_path ||= "/tmp/vagrant-chef-#{get_and_update_counter(:provisioning_path)}"
|
|
end
|
|
|
|
def verify_binary(binary)
|
|
# Checks for the existence of chef binary and error if it
|
|
# doesn't exist.
|
|
@machine.communicate.sudo(
|
|
"which #{binary}",
|
|
:error_class => ChefError,
|
|
:error_key => :chef_not_detected,
|
|
:binary => binary)
|
|
end
|
|
|
|
# Returns the path to the Chef binary, taking into account the
|
|
# `binary_path` configuration option.
|
|
def chef_binary_path(binary)
|
|
return binary if !@config.binary_path
|
|
return File.join(@config.binary_path, binary)
|
|
end
|
|
|
|
def chown_provisioning_folder
|
|
@machine.communicate.tap do |comm|
|
|
comm.sudo("mkdir -p #{@config.provisioning_path}")
|
|
comm.sudo("chown #{@machine.ssh_info[:username]} #{@config.provisioning_path}")
|
|
end
|
|
end
|
|
|
|
def setup_config(template, filename, template_vars)
|
|
# If we have custom configuration, upload it
|
|
remote_custom_config_path = nil
|
|
if @config.custom_config_path
|
|
expanded = File.expand_path(
|
|
@config.custom_config_path, @machine.env.root_path)
|
|
remote_custom_config_path = File.join(
|
|
config.provisioning_path, "custom-config.rb")
|
|
|
|
@machine.communicate.upload(expanded, remote_custom_config_path)
|
|
end
|
|
|
|
config_file = Vagrant::Util::TemplateRenderer.render(template, {
|
|
:custom_configuration => remote_custom_config_path,
|
|
:file_cache_path => @config.file_cache_path,
|
|
:file_backup_path => @config.file_backup_path,
|
|
:log_level => @config.log_level.to_sym,
|
|
:verbose_logging => @config.verbose_logging,
|
|
:http_proxy => @config.http_proxy,
|
|
:http_proxy_user => @config.http_proxy_user,
|
|
:http_proxy_pass => @config.http_proxy_pass,
|
|
:https_proxy => @config.https_proxy,
|
|
:https_proxy_user => @config.https_proxy_user,
|
|
:https_proxy_pass => @config.https_proxy_pass,
|
|
:log_level => @config.log_level.to_sym,
|
|
:no_proxy => @config.no_proxy,
|
|
:formatter => @config.formatter
|
|
}.merge(template_vars))
|
|
|
|
# Create a temporary file to store the data so we
|
|
# can upload it
|
|
temp = Tempfile.new("vagrant")
|
|
temp.write(config_file)
|
|
temp.close
|
|
|
|
remote_file = File.join(config.provisioning_path, filename)
|
|
@machine.communicate.tap do |comm|
|
|
comm.sudo("rm #{remote_file}", :error_check => false)
|
|
comm.upload(temp.path, remote_file)
|
|
end
|
|
end
|
|
|
|
def setup_json
|
|
@machine.env.ui.info I18n.t("vagrant.provisioners.chef.json")
|
|
|
|
# Get the JSON that we're going to expose to Chef
|
|
json = @config.json
|
|
json[:run_list] = @config.run_list if !@config.run_list.empty?
|
|
json = JSON.pretty_generate(json)
|
|
|
|
# Create a temporary file to store the data so we
|
|
# can upload it
|
|
temp = Tempfile.new("vagrant")
|
|
temp.write(json)
|
|
temp.close
|
|
|
|
remote_file = File.join(@config.provisioning_path, "dna.json")
|
|
@machine.communicate.tap do |comm|
|
|
comm.sudo("rm #{remote_file}", :error_check => false)
|
|
comm.upload(temp.path, remote_file)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|