cf2745ebfc
a. Instead of executing a bash string, write out a temporary file
and then just execute '/bin/sh' on that file with the right arguments instead.
2. Rename util.SilentTemporaryFile to util.ExtendedTemporaryFile and update the usages
of the previous name accordingly, this better reflects what this temp file is.
3. More teenie pep8 line length fixings
130 lines
5.2 KiB
Python
130 lines
5.2 KiB
Python
# vi: ts=4 expandtab
|
|
#
|
|
# Copyright (C) 2012 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# Author: Avishai Ish-Shalom <avishai@fewbytes.com>
|
|
# Author: Mike Moulton <mike@meltmedia.com>
|
|
# Author: Juerg Haefliger <juerg.haefliger@hp.com>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 3, as
|
|
# published by the Free Software Foundation.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import json
|
|
import os
|
|
|
|
from cloudinit import templater
|
|
from cloudinit import util
|
|
|
|
RUBY_VERSION_DEFAULT = "1.8"
|
|
|
|
|
|
def handle(name, cfg, cloud, log, _args):
|
|
|
|
# If there isn't a chef key in the configuration don't do anything
|
|
if 'chef' not in cfg:
|
|
log.debug(("Skipping module named %s,"
|
|
" no 'chef' key in configuration"), name)
|
|
return
|
|
chef_cfg = cfg['chef']
|
|
|
|
# Ensure the chef directories we use exist
|
|
c_dirs = [
|
|
'/etc/chef',
|
|
'/var/log/chef',
|
|
'/var/lib/chef',
|
|
'/var/cache/chef',
|
|
'/var/backups/chef',
|
|
'/var/run/chef',
|
|
]
|
|
for d in c_dirs:
|
|
util.ensure_dir(cloud.paths.join(False, d))
|
|
|
|
# Set the validation key based on the presence of either 'validation_key'
|
|
# or 'validation_cert'. In the case where both exist, 'validation_key'
|
|
# takes precedence
|
|
for key in ('validation_key', 'validation_cert'):
|
|
if key in chef_cfg and chef_cfg[key]:
|
|
v_fn = cloud.paths.join(False, '/etc/chef/validation.pem')
|
|
util.write_file(v_fn, chef_cfg[key])
|
|
break
|
|
|
|
# Create the chef config from template
|
|
template_fn = cloud.get_template_filename('chef_client.rb')
|
|
if template_fn:
|
|
iid = str(cloud.datasource.get_instance_id())
|
|
params = {
|
|
'server_url': chef_cfg['server_url'],
|
|
'node_name': util.get_cfg_option_str(chef_cfg, 'node_name', iid),
|
|
'environment': util.get_cfg_option_str(chef_cfg, 'environment',
|
|
'_default'),
|
|
'validation_name': chef_cfg['validation_name']
|
|
}
|
|
out_fn = cloud.paths.join(False, '/etc/chef/client.rb')
|
|
templater.render_to_file(template_fn, out_fn, params)
|
|
else:
|
|
log.warn("No template found, not rendering to /etc/chef/client.rb")
|
|
|
|
# set the firstboot json
|
|
initial_json = {}
|
|
if 'run_list' in chef_cfg:
|
|
initial_json['run_list'] = chef_cfg['run_list']
|
|
if 'initial_attributes' in chef_cfg:
|
|
initial_attributes = chef_cfg['initial_attributes']
|
|
for k in list(initial_attributes.keys()):
|
|
initial_json[k] = initial_attributes[k]
|
|
firstboot_fn = cloud.paths.join(False, '/etc/chef/firstboot.json')
|
|
util.write_file(firstboot_fn, json.dumps(initial_json))
|
|
|
|
# If chef is not installed, we install chef based on 'install_type'
|
|
if not os.path.isfile('/usr/bin/chef-client'):
|
|
install_type = util.get_cfg_option_str(chef_cfg, 'install_type',
|
|
'packages')
|
|
if install_type == "gems":
|
|
# this will install and run the chef-client from gems
|
|
chef_version = util.get_cfg_option_str(chef_cfg, 'version', None)
|
|
ruby_version = util.get_cfg_option_str(chef_cfg, 'ruby_version',
|
|
RUBY_VERSION_DEFAULT)
|
|
install_chef_from_gems(cloud.distro, ruby_version, chef_version)
|
|
# and finally, run chef-client
|
|
log.debug('Running chef-client')
|
|
util.subp(['/usr/bin/chef-client',
|
|
'-d', '-i', '1800', '-s', '20'], capture=False)
|
|
elif install_type == 'packages':
|
|
# this will install and run the chef-client from packages
|
|
cloud.distro.install_packages(('chef',))
|
|
else:
|
|
log.warn("Unknown chef install type %s", install_type)
|
|
|
|
|
|
def get_ruby_packages(version):
|
|
# return a list of packages needed to install ruby at version
|
|
pkgs = ['ruby%s' % version, 'ruby%s-dev' % version]
|
|
if version == "1.8":
|
|
pkgs.extend(('libopenssl-ruby1.8', 'rubygems1.8'))
|
|
return pkgs
|
|
|
|
|
|
def install_chef_from_gems(ruby_version, chef_version, distro):
|
|
distro.install_packages(get_ruby_packages(ruby_version))
|
|
if not os.path.exists('/usr/bin/gem'):
|
|
util.sym_link('/usr/bin/gem%s' % ruby_version, '/usr/bin/gem')
|
|
if not os.path.exists('/usr/bin/ruby'):
|
|
util.sym_link('/usr/bin/ruby%s' % ruby_version, '/usr/bin/ruby')
|
|
if chef_version:
|
|
util.subp(['/usr/bin/gem', 'install', 'chef',
|
|
'-v %s' % chef_version, '--no-ri',
|
|
'--no-rdoc', '--bindir', '/usr/bin', '-q'], capture=False)
|
|
else:
|
|
util.subp(['/usr/bin/gem', 'install', 'chef',
|
|
'--no-ri', '--no-rdoc', '--bindir',
|
|
'/usr/bin', '-q'], capture=False)
|