2010-07-08 20:48:01 -07:00

77 lines
2.5 KiB
Ruby

module Vagrant
module Provisioners
# This class implements provisioning via chef-client, allowing provisioning
# with a chef server.
class ChefServer < Chef
def prepare
if env.config.chef.validation_key_path.nil?
action_env.error!(:chef_server_validation_key_required)
elsif !File.file?(validation_key_path)
action_env.error!(:chef_server_validation_key_doesnt_exist)
elsif env.config.chef.chef_server_url.nil?
action_env.error!(:chef_server_url_required)
end
end
def provision!
verify_binary("chef-client")
chown_provisioning_folder
create_client_key_folder
upload_validation_key
setup_json
setup_server_config
run_chef_client
end
def create_client_key_folder
logger.info "Creating folder to hold client key..."
path = Pathname.new(env.config.chef.client_key_path)
vm.ssh.execute do |ssh|
ssh.exec!("sudo mkdir -p #{path.dirname}")
end
end
def upload_validation_key
logger.info "Uploading chef client validation key..."
vm.ssh.upload!(validation_key_path, guest_validation_key_path)
end
def setup_server_config
setup_config("chef_server_client", "client.rb", {
:node_name => env.config.chef.node_name,
:chef_server_url => env.config.chef.chef_server_url,
:validation_client_name => env.config.chef.validation_client_name,
:validation_key => guest_validation_key_path,
:client_key => env.config.chef.client_key_path
})
end
def run_chef_client
command = "cd #{env.config.chef.provisioning_path} && sudo chef-client -c client.rb -j dna.json"
logger.info "Running chef-client..."
vm.ssh.execute do |ssh|
ssh.exec!(command) do |channel, type, data|
# TODO: Very verbose. It would be easier to save the data and only show it during
# an error, or when verbosity level is set high
if type == :exit_status
ssh.check_exit_status(data, command)
else
logger.info("#{data}: #{type}")
end
end
end
end
def validation_key_path
File.expand_path(env.config.chef.validation_key_path, env.root_path)
end
def guest_validation_key_path
File.join(env.config.chef.provisioning_path, "validation.pem")
end
end
end
end