File: //lib/python3.6/site-packages/awscli/customizations/__pycache__/opsworks.cpython-36.opt-1.pyc
3
�T_mR � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZmZmZ d dlmZ d dlmZ eje�ZdZe jdd�Zd Zd
Z ej d ej �Z ej d �Z ej d
�Z!dZ"dj#� Z$dd� Z%dd� Z&G dd� de�Z'dd� Z(dd� Z)dS )� N)� ClientError)� shlex_quote�urlopen�ensure_text_type)� BasicCommand)�!create_client_from_parsed_globalszOpsWorks-Instance� )Zminutesz/AWS/OpsWorks/z7arn:aws:iam::aws:policy/AWSOpsWorksInstanceRegistrationz ^(?!-)[a-z0-9-]{1,63}(?<!-)$z
^i-[0-9a-f]+$z^\d+\.\d+\.\d+\.\d+$z@http://169.254.169.254/latest/dynamic/instance-identity/documenta�
set -e
umask 007
AGENT_TMP_DIR=$(mktemp -d /tmp/opsworks-agent-installer.XXXXXXXXXXXXXXXX)
curl --retry 5 -L %(agent_installer_url)s | tar xz -C $AGENT_TMP_DIR
cat >$AGENT_TMP_DIR/opsworks-agent-installer/preconfig <<EOF
%(preconfig)s
EOF
exec sudo /bin/sh -c "OPSWORKS_ASSETS_DOWNLOAD_BUCKET=%(assets_download_bucket)s $AGENT_TMP_DIR/opsworks-agent-installer/boot-registration; rm -rf $AGENT_TMP_DIR"
c C s | j dt� d S )Nz building-command-table.opsworks)�register�inject_commands)Zcli� r � /usr/lib/python3.6/opsworks.py�
initialize= s r
c K s t |�| d< d S )Nr )�OpsWorksRegister)Z
command_table�session�kwargsr r r r
A s r
c s e Zd ZdZejd�j� Zdddd�dddd gd
d �d d
dd�dddd�dddd�dddd�dddd�d d d d�d d d!d"�d#d d$d"�d%dd&d'd(d)�g Z� fd*d+�Z d,d-� Z
d.d/� Z d0d1� Z d2d3� Z
d4d5� Zd6d7� Zd8d9� Zd:d;� Zd<d=� Zd>d?� ZedEdAdB��ZedCdD� �Z� ZS )Fr r z�
Registers an EC2 instance or machine with AWS OpsWorks.
Registering a machine using this command will install the AWS OpsWorks
agent on the target machine and register it with an existing OpsWorks
stack.
zstack-idTzZA stack ID. The instance will be registered with the
given stack.)�name�required� help_textzinfrastructure-class�ec2z on-premiseszzSpecifies whether to register an EC2 instance (`ec2`)
or an on-premises instance (`on-premises`).)r r �choicesr zoverride-hostname�hostnamezrThe instance hostname. If not provided, the current
hostname of the machine will be used.)r �destr zoverride-private-ip�
private_ipaA An IP address. If you set this parameter, the given IP
address will be used as the private IP address within
OpsWorks. Otherwise the private IP address will be
determined automatically. Not to be used with EC2
instances.zoverride-public-ip� public_ipa? An IP address. If you set this parameter, the given IP
address will be used as the public IP address within
OpsWorks. Otherwise the public IP address will be
determined automatically. Not to be used with EC2
instances.z override-ssh�sshzmIf you set this parameter, the given command will be
used to connect to the machine.z ssh-username�usernamezXIf provided, this username will be used to connect to
the host.zssh-private-key� private_keyzhIf provided, the given private key file will be used
to connect to the machine.�local�
store_truez�If given, instead of a remote machine, the local
machine will be imported. Cannot be used together
with `target`.)r �actionr zuse-instance-profilezRUse the instance profile instead of creating an IAM
user.�target�?z
[<target>]z�Either the EC2 instance ID or the hostname of the
instance or machine to be registered with OpsWorks.
Cannot be used together with `--local`.)r Zpositional_arg�nargsZsynopsisr c s> t t| �j|� d | _d | _d | _d | _d | _d | _d | _ d S )N)
�superr �__init__�_stack�
_ec2_instance� _prov_params� _use_address�
_use_hostname�
_name_for_iam�
access_key)�selfr )� __class__r r r$ } s zOpsWorksRegister.__init__c C s" | j jd�| _t| j d|�| _d S )N�iam�opsworks)�_session�
create_clientr. r r/ )r, �args�parsed_globalsr r r �_create_clients� s z OpsWorksRegister._create_clientsc C sL | j ||� | j|� | j|� | j|� | j|� | j|� | j|� d S )N)r4 �prevalidate_arguments�retrieve_stack�validate_arguments�determine_details�create_iam_entities�setup_target_machine)r, r2 r3 r r r � _run_main� s
zOpsWorksRegister._run_mainc C s� |j r|j rtd��n|j r.|jr.td��|jrHtj� dkrHtd��|jrb|jsZ|jrbtd��|jdkr�|j rztd��|j
r�td��|jd kr�|j r�td
��|j r�t
j|j �s�td |j ��d S )
zN
Validates command line arguments before doing anything else.
z%One of target or --local is required.z4Arguments target and --local are mutually exclusive.ZLinuxz6Non-Linux instances are not supported by AWS OpsWorks.zYArgument --override-ssh cannot be used together with --ssh-username or --ssh-private-key.r z/--override-private-ip is not supported for EC2.z.--override-public-ip is not supported for EC2.z on-premisesz1--use-instance-profile is only supported for EC2.zxInvalid hostname: '%s'. Hostnames must consist of letters, digits and dashes only and must not start or end with a dash.N)r r �
ValueError�platform�systemr r r �infrastructure_classr r �use_instance_profiler � HOSTNAME_RE�match)r, r2 r r r r5 � s6
z&OpsWorksRegister.prevalidate_argumentsc s~ t jd� | jj� jgd�d d | _| jj| jd d�| _� jdkoL� j �rzt jd� | j
j d| jd d
�}d g i}g �d | jkr�|d j d
| jd gd�� n�j dd� � t
j� j�r̈ jg|d< n@tj� j�r�j � fdd�� � j| _n|d j d� jgd�� �fdd�|jf |�d D �}|�s@td� j ��n0t|�dk�rptd� jdjdd � |D ��f ��|d | _d S ) z�
Retrieves the stack from the API, thereby ensures that it exists.
Provides `self._stack`, `self._prov_params`, `self._use_address`, and
`self._ec2_instance`.
z,Retrieving stack and provisioning parameters)ZStackIdsZStacksr �StackId)rC r z#Retrieving EC2 instance information�Region)Z region_nameZFilters�VpcIdzvpc-id)�NameZValuesc S s d| kS )NrE r )�instancer r r �<lambda>� s z1OpsWorksRegister.retrieve_stack.<locals>.<lambda>Z InstanceIdsc s | j d�� jkp | j d�� jkS )N�PrivateIpAddress�PublicIpAddress)�getr )rG )r2 r r rH � s ztag:Namec s4 g | ],}|d D ] � t � fdd��D ��r� �qqS )� Instancesc 3 s | ]}|� �V qd S )Nr )�.0�c)�ir r � <genexpr>� s z=OpsWorksRegister.retrieve_stack.<locals>.<listcomp>.<genexpr>)�all)rM �r)�
conditions)rO r �
<listcomp>� s z3OpsWorksRegister.retrieve_stack.<locals>.<listcomp>Z Reservationsz&Did not find any instance matching %s.� z)Found multiple instances matching %s: %s.z, c s s | ]}|d V qdS )Z
InstanceIdNr )rM rO r r r rP s z2OpsWorksRegister.retrieve_stack.<locals>.<genexpr>N)�LOG�debugr/ Zdescribe_stacks�stack_idr% Z&describe_stack_provisioning_parametersr' r? r r0 r1 �append�INSTANCE_ID_RErB r �
IP_ADDRESS_REr( �describe_instancesr< �len�joinr& )r, r2 r Z desc_args� instancesr )r2 rS r r6 � sF
z OpsWorksRegister.retrieve_stackc s� � j rB| jj| jd d�d }t� fdd�|D ��rBtd� j ��� jdkr�� jr�tj t
t t �j
� ��d }|| jd kr�td
��d S ) zS
Validates command line arguments using the retrieved information.
rC )rC rL c 3 s | ]}� j j� |d kV qdS )ZHostnameN)r �lower)rM rG )r2 r r rP s z6OpsWorksRegister.validate_arguments.<locals>.<genexpr>z@Invalid hostname: '%s'. Hostnames must be unique within a stack.r �regionrD z1The stack's and the instance's region must match.N)r r/ r\ r% �anyr< r? r �json�loadsr r � IDENTITY_URL�read)r, r2 r_ ra r )r2 r r7 s
z#OpsWorksRegister.validate_argumentsc C s� | j sn|jrn`|jdkr\d| jkr0| jd | _ qnd| jkrRtjd� | jd | _ qntd��n|jdkrn|j| _ |jr�|j| _ |j| _
n&|jr�d| _ t j � | _
nd| _ |j| _
dS )a
Determine details (like the address to connect to and the hostname to
use) from the given arguments and the retrieved data.
Provides `self._use_address` (if not provided already),
`self._use_hostname` and `self._name_for_iam`.
r rJ rI zYInstance does not have a public IP address. Trying to use the private address to connect.z1The instance does not seem to have an IP address.z on-premisesN)
r( r r? r& rV �warnr< r r r) r* �socketZ gethostname)r, r2 r r r r8 s,
z"OpsWorksRegister.determine_detailsc C sR |j rtjd� d| _dS tjd� dt| jd � }y | jj|td� tjd|� W nJ t k
r� } z.|j
j di �j d �d
kr�tjd |� n� W Y dd}~X nX tjd � d
t t| jd �d�t t| j
�d�f }x�td�D ]�}||r�d| nd }y| jj|td� W nN t k
�rX } z0|j
j di �j d �d
k�rFtjd|� n� W Y dd}~X q�X tjd|� P q�W td��tjd� | jj||d� y| jjt|d� W n~ t k
�r } z`|j
j di �j d �dk�rtjdt|� | jjt| j| jd t�|d � tjd t|� n� W Y dd}~X nX tjd t|� tjd � | jj|d!�d" | _dS )#zp
Creates an IAM group, user and corresponding credentials.
Provides `self.access_key`.
z Skipping IAM entity creationNz#Creating the IAM group if necessaryz OpsWorks-%srC )� GroupName�PathzCreated IAM group %s�ErrorZCodeZEntityAlreadyExistsz IAM group %s exists, continuingzCreating an IAM userzOpsWorks-%s-%srF � � z+%s� )�UserNamerj z/IAM user %s already exists, trying another namezCreated IAM user %sz&Couldn't find an unused IAM user name.z3Adding the user to the group and attaching a policy)ri ro )Z PolicyArnro Z AccessDeniedzFUnauthorized to attach policy %s to user %s. Trying to put user policyZArn)Z
PolicyNameZPolicyDocumentro zPut policy %s to user %sz Attached policy %s to user %szCreating an access key)ro Z AccessKey)r@ rV rW r+ �
clean_for_iamr% r. Z create_group�IAM_PATHr ZresponserK � shorten_namer* �rangeZ create_userr< Zadd_user_to_groupZattach_user_policy�IAM_POLICY_ARNZput_user_policy�IAM_USER_POLICY_NAME�_iam_policy_document�IAM_USER_POLICY_TIMEOUTZcreate_access_key)r, r2 Z
group_name�eZ
base_usernameZtry_r r r r r9 G s|
z$OpsWorksRegister.create_iam_entitiesc C sh t | jd | j| j|��| jd d d� }|jrNtjd� tjdd|g� ntjd� | j ||� d S )
zz
Setups the target machine by copying over the credentials and starting
the installation process.
ZAgentInstallerUrl�
Parameters�assets_download_bucket)Zagent_installer_urlZ preconfigrz z Running the installer locallyz/bin/shz-cz6Connecting to the target machine to run the installer.N)
�
REMOTE_SCRIPTr' �
_to_ruby_yaml�_pre_config_documentr rV rW �
subprocess�
check_callr )r, r2 �
remote_scriptr r r r: � s
z%OpsWorksRegister.setup_target_machinec C s8 t j� dkr�z�tjddd�}|j|� |j� |jr<|j}nBd}|jrT|d|j 7 }|jrh|d|j 7 }|d| j 7 }|d 7 }|d|j
7 }t j |d
d � W d t
j|j
� X n�|jr�tjt|j��}n@d
dg}|jr�|jd|jg� |jr�|jd|jg� |j| j � dd|g}|jdjdd� |D ��� t j |� d S )zA
Runs a (sh) script on a remote machine via SSH.
ZWindowsZwtF)�deleteZplinkz -l "%s"z -i "%s"z "%s"z -mT)�shellNr z-ttz-lz-iz/bin/shz-c� c s s | ]}t |�V qd S )N)r )rM Zwordr r r rP � s z'OpsWorksRegister.ssh.<locals>.<genexpr>)r= r> �tempfileZNamedTemporaryFile�write�closer r r r( r r~ r �os�remove�shlex�split�str�extendrY r^ )r, r2 r� Z script_fileZcallZ remote_callr r r r � s8
zOpsWorksRegister.sshc C s� t f d| jd i| jd ��}| jr@| jd |d< | jd |d< | jrP| j|d< |jr`|j|d < |jrp|j|d
< |jd k|d < tj d
|� |S )NrX rC ry Z AccessKeyIdZ
access_key_idZSecretAccessKeyZsecret_access_keyr r r r �importzUsing pre-config: %r)
�dictr% r' r+ r) r r r? rV rW )r, r2 �
parametersr r r r} � s
z%OpsWorksRegister._pre_config_documentNc C sN dd| d�}|d k r8t j j� | }dd|jd�ii|d< |gdd �}tj|�S )
Nzopsworks:RegisterInstanceZAllow)ZActionZEffectZResourceZ DateLessThanzaws:CurrentTimez%Y-%m-%dT%H:%M:%SZZ Conditionz
2012-10-17)Z StatementZVersion)�datetimeZutcnowZstrftimerc �dumps)ZarnZtimeoutZ statementZ valid_untilZpolicy_documentr r r rv � s z%OpsWorksRegister._iam_policy_documentc C s dj dd� t| j� �D ��S )N�
c s s$ | ] \}}d |t j|�f V qdS )z:%s: %sN)rc r� )rM �k�vr r r rP s z1OpsWorksRegister._to_ruby_yaml.<locals>.<genexpr>)r^ �sorted�items)r� r r r r| s
z OpsWorksRegister._to_ruby_yaml)N)�__name__�
__module__� __qualname__�NAME�textwrap�dedent�stripZ DESCRIPTIONZ ARG_TABLEr$ r4 r; r5 r6 r7 r8 r9 r: r r} � staticmethodrv r| �
__classcell__r r )r- r r E sR
'L'\( r c C s t jdd| �S )z9
Cleans a name to fit IAM's naming requirements.
z[^A-Za-z0-9+=,.@_-]+�-)�re�sub)r r r r rp s rp c C sD t | �|kr| S t|d d�\}}| d|| � d | | d� S )z<
Shortens a name to the given number of characters.
� � Nz...)r] �divmod)r Z
max_length�qrR r r r rr s rr )*r� rc Zloggingr� r= r� r� rh r~ r� r� Zbotocore.exceptionsr Z
awscli.compatr r r Z awscli.customizations.commandsr Zawscli.customizations.utilsr Z getLoggerr� rV ru Z timedeltarw rq rt �compile�IrA rZ r[ re �lstripr{ r
r
r rp rr r r r r �<module>
s@
M