CasperSecurity
o
��Jh � @ s� d dl Z d dlZd dlmZmZmZ d dlmZmZm Z m
Z
m Z m Z m
Z
mZ dZg d�ZdZdZdZd Ze �� Ze �e�e��Zed
d efd efd
efdefdefg�Z de fdd�Z de fdd�Z d0dee dee deee ddfdd�Z! d1dedeee ddfdd �Z"d edee fd d �Z#dee fd d!�Z$d"d#� Z%d$ej&fd%d&�Z' 'd2d(edee d)e fd*d+�Z(d(efd,d-�Z)d(ede fd.d/�Z*dS )3� N)�List�
NamedTuple�Optional)�api�apt� event_logger�
exceptions�http�messages�system�utilz
/usr/bin/snap)g �?g �?g @z
proxy.httpz proxy.httpsz/run/snapd.socketz /v2/snaps/{}� SnapPackage�name�version�revision�channel� publisher�returnc C s dt �� v S )z(Returns whether or not snap is installed�snapd)r � get_installed_packages_names� r r �//usr/lib/python3/dist-packages/uaclient/snap.py�is_snapd_installed' s r c C s t dd� t� D ��S )z3Returns whether or not snapd is installed as a snapc s s � | ]}|j d kV qdS )r N�r )�.0�snapr r r � <genexpr>. s � z/is_snapd_installed_as_a_snap.<locals>.<genexpr>)�any�get_installed_snapsr r r r � is_snapd_installed_as_a_snap, s r �
http_proxy� https_proxy� retry_sleepsc C sx t � s
t�d� dS | s|rt�tjjdd�� | r(tj dddd�| �g|d� |r:tj dddd �|�g|d� dS dS )
a
Configure snap to use http and https proxies.
:param http_proxy: http proxy to be used by snap. If None, it will
not be configured
:param https_proxy: https proxy to be used by snap. If None, it will
not be configured
:param retry_sleeps: Optional list of sleep lengths to apply between
retries. Specifying a list of [0.5, 1] tells subp to retry twice
on failure; sleeping half a second before the first retry and 1 second
before the second retry.
z6Skipping configure snap proxy. snapd is not installed.Nr )�service�setr z
proxy.http={}�r" zproxy.https={})
r �LOG�debug�event�infor
�SETTING_SERVICE_PROXY�formatr �subp)r r! r" r r r �configure_snap_proxy1 s
�
��r- �
protocol_typec C s4 t � s
t�d� dS tjdddd�| �g|d� dS )a�
Unset snap configuration settings for http and https proxies.
:param protocol_type: String either http or https
:param retry_sleeps: Optional list of sleep lengths to apply between
retries. Specifying a list of [0.5, 1] tells subp to retry twice
on failure; sleeping half a second before the first retry and 1 second
before the second retry.
z8Skipping unconfigure snap proxy. snapd is not installed.Nr �unsetr zproxy.{}r% )r r& r' r r, r+ )r. r" r r r �unconfigure_snap_proxyV s
�r0 �keyc C s8 zt �ddd| g�\}}|�� W S tjy Y dS w )z�
Gets the config value from snap.
:param protocol: can be any valid snap config option
:return: the value of the snap config option, or None if not set
r �getr N)r r, �stripr �ProcessExecutionError)r1 �out�_r r r �get_config_option_valuek s
�r7 c C sR t �g d��\} }| �� }|dd � }g }|D ]}|�� d }|�t|�� q|S )N)r �listz--color�neverz --unicoder9 � r )r r, �
splitlines�split�append�
get_snap_info)r5 r6 �apps�snaps�liner r r r r x s � r c
C s� t �tjjdd�� z t�t�� � W n tj y. } zt
� dt | �� W Y d } ~ nd } ~ ww z
t
jg d�tjd� W d S tjyH t�� �w )Nzstandard Ubuntur z<Trying to install snapd. Ignoring apt-get update failure: %s)zapt-get�installz --assume-yesr r% )r( r) r
�APT_UPDATING_LISTr+ r �update_sources_list�get_system_sources_filer �UbuntuProErrorr& r' �strr r, � APT_RETRIESr4 �CannotInstallSnapdError)�er r r �
install_snapd� s$ ��� ��rK �progressc
C sv zt jtdddgdd� W d S tjy: } z t�dt|��� �r.t �
d� | � dt j
� n� W Y d }~d S d }~ww ) N�waitr z seed.loadedT��capturezunknown command .*waitz9Detected version of snapd that does not have wait commandr) )r r, �SNAP_CMDr r4 �re�searchrG �lowerr& �warning�emitr
� SNAPD_DOES_NOT_HAVE_WAIT_CMD)rL rJ r r r �run_snapd_wait_cmd� s ����rW Fr �classic_confinement_supportc C s@ t d| g}|r |dg7 }|r|d�|�g7 }tj|dtd� d S )NrB z --classicz --channel={}T)rO r" )rP r+ r r, �SNAP_INSTALL_RETRIES)r r rX �cmdr r r � install_snap� s
�r[ c C s t jtd| gdd� d S )N�refreshTrN )r r, rP �r r r r � refresh_snap� s r^ c C s� t �| �}z5t�td|�}|jdkr9|jdkr)|j�di ��d�dkr)tj | d��|j�di ��d�}tj
|d ��W n t yE t� � �w |j�di �}t
|�d
d �|�d d �|�d
d �|�dd �|�di ��dd �d�S )N�GET�� i� �result�kindzsnap-not-foundr] �message)�errorr � r r r r �username)r r r r r )�SNAPD_SNAPS_APIr+ r �unix_socket_request�SNAPD_SOCKET_PATH�code� json_dictr2 r �SnapNotInstalledError�UnexpectedSnapdAPIError�ConnectionRefusedError�SnapdAPIConnectionRefusedr
)r �url�resp� error_msg� snap_infor r r r> � s.
� � �
�r> )NNN)N)NF)+�loggingrQ �typingr r r �uaclientr r r r r r
r r rP rY �HTTP_PROXY_OPTION�HTTPS_PROXY_OPTIONri rg �get_event_loggerr( � getLogger� replace_top_level_logger_name�__name__r& rG r
�boolr r �floatr- r0 r7 r rK �ProgressWrapperrW r[ r^ r> r r r r �<module> st ( �� ���
�
�&��
�
�
���
�