3

-�_g��@s:dd
lZdd
l
Z
ddlmZmZmZ
Gdd�dee�Zd
S)�N)�Plugin�UbuntuPlugin�	PluginOptc	@sleZ
dZd
ZdZdZdZdZedde	d	d
�edde
d
d
�edde
dd
�edde
dd
�gZdd�Zdd�Z
dS)�Jujua3The Juju plugin is aimed at collecting Juju-related logs,
    configurations, and controller/model state(s).

    Logs and agent configuration information (/var/log/juju and /var/lib/juju)
    is collected by default since these are useful for troubleshooting.

    The Juju state collection is disabled by default and can be enabled with
    the 'juju-state' option. Collecting Juju state is safe in theory, but it
    does act on the live controller(s)/model(s) and is therefore optional.

    The default Juju state collection collects all controllers and models that
    the 'juju-user' (default=ubuntu) has access to.

    Specific controllers or models can be collected using the 'controllers'
    and 'models' options.

    Important: the string list is whitespace delimited, not colon delimited
    (sos plugin standard). This is due to the underlying Juju CLI accepting
    specific models in the format 'controller:model' and whitespaces are not
    allowed in either controller and model names.

    Example: models="controller_a:model_x controller_b:model_y"
    zJuju orchestration toolZjuju�virt�sysmgmt�
/var/log/jujuz
juju-stateFz Include Juju state in the report)�defaultZval_typeZdescz	juju-userZubuntuzJuju client user.�controllers�z[Collect Juju state for specified controllers. Uses a             whitespace delimited list.�modelszVCollect Juju state for specified models. Uses a whitespace             delimited list.c
Cs2x&|jd
�
D]}
|j
|
�

|j|
�

qW|jdddddddd	d
dg
�

|jd�

|jd
dgdd�
|jd�
r�|jdddg�

|jd�

n
|jd�

|jd�
s�dS|jd�
}ytj	|�

Wn&t
k
r�


|jd|�d��

dSX|jd�
o�|jd�
�
r|jd�

dS|jd|d�}|d d!k�
rht
|jd�
jd"�
�
}|�
rL|d#h
k�
rL|}nt
tj|d$�
dj��
}ni}|jd�
�
r�x�|jd�
jd"�
D] }d%|�d&�}|j||d�
�
q�Wn~x||D]t}|jd'|�d&�|d�}	|	d d!k�
r�tj|	d$�
d}
x6|
D].}|d(}d%|�d)|�d&�}|j||d�
�
q�W�
q�WdS)*Nzjuju*Zjuju_engine_reportZjuju_goroutinesZjuju_heap_profileZjuju_leasesZjuju_metricsZjuju_pubsub_reportZjuju_presence_reportZjuju_statepool_reportZjuju_statetracker_reportZjuju_unit_statusz!/var/lib/juju/agents/*/agent.confz/var/log/juju*z/var/lib/juju*T)
�	recursiveZall_logsz
/var/log/jujuz
/var/lib/jujuz/var/lib/juju/**/.*z/var/lib/juju/kvmz/var/log/juju/*.logz
juju-statez	juju-userzUser "z4" does not exist, will not collect Juju information.r
rzWOptions: controllers, models are mutually exclusive. Will not collect Juju information.zjuju controllers --format=json)
ZrunasZstatusr
�
 r�outputzjuju status -m z --format=jsonzjuju models --all -c z
short-name�
:)Zget_service_namesZadd_journalZadd_service_statusZadd_cmd_outputZ
add_copy_specZadd_dir_listingZ
get_optionZadd_forbidden_path�pwd�getpwnam�KeyErrorZ	_log_warnZcollect_cmd_output�set�split�json�loads�keysZexec_cmd)�selfZserviceZ	juju_userZcontrollers_jsonZdesired_controllersr
ZmodelZcommandZ
controllerZmodels_jsonrZ
short_name�r�/usr/lib/python3.6/juju.py�setupOs�


















































z
Juju.setupc
CsLd
}
ddddg}ddj|�
�d�}d	}|j
|
||�
|j|
�

|jd
�

dS)Nz/var/lib/juju/agents/*ZsharedsecretZapipasswordZoldpasswordZ
statepasswordz(^\s*(�
|z
)\s*:\s*)(.*)z\1*********zjuju controllers)�joinZdo_path_regex_subZdo_file_private_subZdo_cmd_private_sub)rZagents_pathZprotect_keysZ
keys_regexZ	sub_regexrrr�postproc�s







z
Juju.postprocN)rr)
r)�__name__�
__module__�__qualname__�__doc__Z
short_descZplugin_nameZprofiles�filesr�bool�strZoption_listrrrrrrrs6















mr)rrZsos.report.pluginsrrrrrrrr�<module>s