3
\�me7 �
@ s
U d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z m Z m
Z
mZ ddlmZ ddlmZ ddlmZmZ dd lmZ dd
lmZ d Zed d
ddddg�Z ed g�Z dZ dZ dZ dZ!dZ"dZ#ddde dde#ddd d dd � Z$e%d g�Z&e%d d!d"d#d$d%d&g�Z'e%ej(e$j)� e&e'd'd(d)d*d+g��Z*d,Z+d-Z,ed.d/d0d1d2g�Z-eZ.d3gZ/ej0e1�Z2d4d5d6ed7�e/ed8�ge.d9gd:�Z3e3ee3�Z d;d<� Z4d=d>� Z5e6eee7dd?�d@dA�Z8dBdC� Z9ee7dD�dEdF�Z:dRedG�dHdI�Z;edJ�dKdL�Z<ee6 dM�dNdO�Z=dPdQ� Z>dS )Sz7Chef: module that configures, starts and installs chef.� N)�dedent)�List)�subp�
temp_utils� templater�
url_helper�util)�Cloud)�Config)�
MetaSchema� get_meta_doc)�Distro)�
PER_ALWAYSz1.8z /etc/chefz
/var/log/chefz
/var/lib/chefz/var/cache/chefz/var/backups/chefz
/var/run/chefz#https://www.chef.io/chef/install.sh� z/etc/chef/validation.pemz#/etc/chef/encrypted_data_bag_secret�_defaultz/etc/chef/firstboot.jsonz :verify_nonez:infoz/var/log/chef/client.logz/etc/chef/client.pemz/var/run/chef/client.pidT) Zssl_verify_modeZ log_level� log_location�validation_key�validation_cert�
client_key� json_attribs�file_cache_pathZfile_backup_path�pid_file� show_time�encrypted_data_bag_secretr r r r r r r r �
server_url� node_name� environment�validation_nameZ chef_licensez/etc/chef/client.rbz/usr/bin/chef-clientz-dz-iZ1800z-sZ20�allZcc_chefZChefz0module that configures, starts and installs chefa This module enables chef to be installed (from packages,
gems, or from omnibus). Before this occurs, chef configuration is
written to disk (validation.pem, client.pem, firstboot.json,
client.rb), and required directories are created (/etc/chef and
/var/log/chef and so-on). If configured, chef will be
installed and started in either daemon or non-daemon mode.
If run in non-daemon mode, post run actions are executed to do
finishing activities such as removing validation.pem.a�
chef:
directories:
- /etc/chef
- /var/log/chef
validation_cert: system
install_type: omnibus
initial_attributes:
apache:
prefork:
maxclients: 100
keepalive: off
run_list:
- recipe[apache2]
- role[db]
encrypted_data_bag_secret: /etc/chef/encrypted_data_bag_secret
environment: _default
log_level: :auto
omnibus_url_retries: 2
server_url: https://chef.yourorg.com:4000
ssl_verify_mode: :verify_peer
validation_name: yourorg-validator�chef)�id�name�title� description�distrosZexamples� frequencyZactivate_by_schema_keysc C s. t j| ddd�}|r*tjjt�r*tjt� d S )NZdelete_validation_post_execF)�default)r �get_cfg_option_bool�os�path�isfile�CHEF_VALIDATION_PEM_PATH�unlink)�chef_cfgZ
delete_pem� r. � /usr/lib/python3.6/cc_chef.py�
post_run_chef� s r0 c C s� t j� }xf|j� D ]Z\}}|tkr0tjd|� q|d krBd ||< q|tkr\tj||�||< qtj ||�||< qW |j
tj � tj |d| d�tj |ddd�|d |d d�� |S ) Nz'Skipping unknown chef template key '%s'r )r& r r r r )Z generated_byr r r r ) �CHEF_RB_TPL_DEFAULTS�copy�items�CHEF_RB_TPL_KEYS�LOG�debug�CHEF_RB_TPL_BOOL_KEYSr r' �get_cfg_option_str�updateZ make_header)�iidr- �params�k�vr. r. r/ �get_template_params� s&
r> )r! �cfg�cloud�args�returnc C s d|krt jd| � dS |d }tj|d�}|s8tt�}x tj|t�D ]}tj |� qFW |j
dt �}|j
d�}|r�|dkr�tj ||� nt
jj|�s�t jd|� |jd �} | �r t|jj� �}
t|
|�} t� } x2| j� D ]&\}
}|
tko�|r�| jt
jj|�� q�W tj| � t j | t | � n t jd
t � tj |d t d �}|�sJt j!d
� nbi }d|k�rd|d |d< d|k�r�|d }x"t|j"� �D ]}
||
||
< �q�W tj |t#j$|�� tj%|ddd �}t&j't(�}| �s�|�r�t)||�}n |�r�tj%|ddd �}nd}|�rt*|� t+|� dS )z'Handler method activated by cloud-init.r z8Skipping module named %s, no 'chef' key in configurationNZ directoriesr r �systemzWchef validation_cert provided as 'system', but validation_key path '%s' does not exist.zchef_client.rbz&No template found, not rendering to %sZfirstboot_path)r& z7First boot path empty, not writing first boot json fileZrun_list�initial_attributes�
force_installF�exec),r5 r6 r Zget_cfg_option_list�list� CHEF_DIRS� itertools�chain�REQUIRED_CHEF_DIRSZ
ensure_dir�getr+ �
write_filer( r) r* �warningZget_template_filename�strZ
datasourceZget_instance_idr> �setr3 �CHEF_RB_TPL_PATH_KEYS�add�dirnameZ ensure_dirsr Zrender_to_file� CHEF_RB_PATHr8 � CHEF_FB_PATH�info�keys�json�dumpsr' r Zis_exe�CHEF_EXEC_PATH� install_chef�run_chefr0 )r! r? r@ rA r- Z chef_dirs�dZ vkey_pathZvcertZ template_fnr: r; Z param_pathsr< r= Z fb_filenameZ initial_jsonrD rE Z installed�runr. r. r/ �handle� sj
r_ c C s� t jd� tg}d| krl| d }t|ttf�r:|j|� qvt|t�rP|j|� qvt j dt
|�� |jt � n
|jt � t j |dd� d S )NzRunning chef-clientZexec_argumentszRUnknown type %s provided for chef 'exec_arguments' expected list, tuple, or stringF)�capture)
r5 r6 rZ �
isinstancerG �tuple�extendrO �appendrN �type�CHEF_EXEC_DEF_ARGSr )r- �cmdZcmd_argsr. r. r/ r\ s
r\ )�distrorA c K sp |j � }|jdd�}tj|j� dd��@}tjj||�}|jd|� t j
|| dd� t j f d|i|��S Q R X d S )
a Write blob to a tempfile, and call subp with args, kwargs. Then cleanup.
'basename' as a kwarg allows providing the basename for the file.
The 'args' argument to subp will be updated with the full path to the
filename as the first argument.
�basenameZ subp_blobT)�dirZ needs_exer i� )�moderA N) r2 �popr ZtempdirZget_tmp_exec_pathr( r) �join�insertr rM r )�blobrh rA �kwargsri ZtmpdZtmpfr. r. r/ �subp_blob_in_tempfile+ s rq )rh c C sP |dkr t }|dkrt}|dkr&g }nd|g}tj||d�j}t| ||ddd�S )ag Install an omnibus unified package from url.
@param url: URL where blob of chef content may be downloaded. Defaults to
OMNIBUS_URL.
@param retries: Number of retries to perform when attempting to read url.
Defaults to OMNIBUS_URL_RETRIES
@param omnibus_version: Optional version string to require for omnibus
install.
Nz-v)�url�retrieszchef-omnibus-installF)rh ro rA ri r` )� OMNIBUS_URL�OMNIBUS_URL_RETRIESr Zreadurl�contentsrq )rh rr rs �omnibus_versionrA Zcontentr. r. r/ �install_chef_from_omnibus> s rx )r@ c C s� t j|dd�}t j|ddd�}|dkrbt j|dd �}t j|dt�}t||| j� t j|dd d�}n`|dkrz| jjd
g� nH|d kr�t j|d �}t| jt j|d
�t j|d�|d� nt j
d|� d}|S )N� install_typeZpackagesrF F)r& Zgems�version� ruby_versionTr Zomnibusrw Z omnibus_urlZomnibus_url_retries)rh rr rs rw z Unknown chef install type '%s') r r8 r' �RUBY_VERSION_DEFAULT�install_chef_from_gemsrh �install_packagesrx Zget_cfg_option_intr5 rN )r@ r- ry r^ � chef_versionr{ rw r. r. r/ r[ ] s*
r[ )rB c C s& d| d| g}| dkr"|j d� |S )Nzruby%sz
ruby%s-devz1.8�libopenssl-ruby1.8� rubygems1.8)r� r� )rc )rz Zpkgsr. r. r/ �get_ruby_packages~ s
r� c
C s� |j t| �� tjjd�s*tjd| d� tjjd�sFtjd| d� |rptjdddd| dd d
d d g d
d� n tjddddd d
d d gd
d� d S )Nz /usr/bin/gemz/usr/bin/gem%sz
/usr/bin/rubyz/usr/bin/ruby%sZinstallr z-v %sz--no-riz --no-rdocz--bindirz/usr/binz-qF)r` )r~ r� r( r) �existsr Zsym_linkr )r{ r rh r. r. r/ r} � s6
r} )NNN)?�__doc__rI rX Zloggingr( �textwrapr Ztypingr Z cloudinitr r r r r Zcloudinit.cloudr Zcloudinit.configr
Zcloudinit.config.schemar r Zcloudinit.distrosr
Zcloudinit.settingsr r| rb rH rK rt ru r+ Z CHEF_ENCRYPTED_DATA_BAG_PATHZCHEF_ENVIRONMENTrU r1 � frozensetr7 rQ rJ rW r4 rT rZ rf r% r$ Z getLogger�__name__r5 �metar0 r> rO rG r_ r\ rq rx r[ r� r} r. r. r. r/ �<module> s�
#R !