�
��>f� � �� � d dl Z d dlmZmZmZ d dlmZmZmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ d dlmZ d d lmZ G d
� d ej. � Z G d � d
e� Z G d� de� Zy)� N)�Tuple�Type�Union)�api�messages�system)� APT_KEYS_DIR�DEB822_REPO_FILE_CONTENT� KEYRINGS_DIR)�ESM_APT_ROOTDIR)�repo)�
UAEntitlement)�CanDisableFailure)�set_filename_extensionc � � � e Zd Zej j
Zedee e
df fd�� Z de j def� fd� Zdd�Zdd�Z� xZS ) �ESMBaseEntitlement�return.c � � ddl m}m} ||fS )Nr )�ROSEntitlement�ROSUpdatesEntitlement)�uaclient.entitlements.rosr r )�selfr r s �;/usr/lib/python3/dist-packages/uaclient/entitlements/esm.py�dependent_servicesz%ESMBaseEntitlement.dependent_services s � �
�
&�~�6�6� �progressc �z �� ddl m} t �| �
|� }|r" || j � | j � |S �Nr )�update_motd_messages)� uaclient.timer.update_messagingr �super�_perform_enable�cfg�disable_local_esm_repo)r r r �enable_performed� __class__s �r r" z"ESMBaseEntitlement._perform_enable s7 �� �H� �7�2�8�<�� � ���� *� � '� '� )� � r c
�� � t j � j }| j j d� sJ �| j t d� d }t
t j j t | j z � d� }t
t j j t | j z � d� }t j j |� s t j j |� ry dj |�� }dj ||�� }t j |t j ||t | j" d� � � y )
Nzesm-�sources�listz$https://esm.ubuntu.com/{name}/ubuntu)�namez0{series}-{name}-security {series}-{name}-updates)�seriesr* � )�url�suites� keyrings_dir� keyring_file�deb_src)r �get_release_infor+ r* �
startswith�lenr �os�path�normpathr � repo_file�exists�format�
write_filer
r �
repo_key_file)r r+ �esm_name�sources_repo_filename�list_repo_filename�esm_urlr. s r �setup_local_esm_repoz'ESMBaseEntitlement.setup_local_esm_repo" s. � ��(�(�*�1�1���y�y�#�#�F�+�+�+��9�9�S��[�]�+�� 6� �G�G� � � �$�.�.�0�
�
� !
� � 4� �G�G� � � �$�.�.�0�
�
�
�� �7�7�>�>�/� 0�B�G�G�N�N� �5
�
�8�?�?�X�?�N��C�J�J��� K�
��
��� !� $� +� +�� � )� !� /� /��
�
r c �x � t j j t t z | j
z � }t
j |� t j j t | j z � }t
j t |d� � t
j t |d� � y )Nr( r) )
r5 r6 r7 r r r< r �ensure_file_absentr8 r )r r0 �
repo_filenames r r$ z)ESMBaseEntitlement.disable_local_esm_repoK s� � ��w�w�'�'� �l� *�T�-?�-?� ?�
� � �!�!�,�/����(�(� �d�n�n� ,�
�
� �!�!� "�=�)� <�
� �!�!� "�=�&� 9�
r )r N)�__name__�
__module__� __qualname__r �urls�
ESM_HOME_PAGE� help_doc_url�propertyr r r r r �ProgressWrapper�boolr" rA r$ �
__classcell__�r& s @r r r
s^ �� ��=�=�.�.�L�
�7�E�$�}�*=�s�*B�$C� 7� �7� ��(;�(;� �� �'
�R
r r c � � � e Zd ZdZdZej Zej Z ej Z dZ de
j deeedef f f� fd� Z� xZS )�ESMAppsEntitlement�
UbuntuESMAppszesm-appszubuntu-pro-esm-apps.gpgr r Nc � �� ddl m} t �| �
|� \ }}|r6 || j � t j � r| j � ||fS r )r r r! �disabler# r �is_current_series_ltsrA �r r r �disable_performed�failr& s �r rT zESMAppsEntitlement.disablee sO �� � I�"'�'�/�(�";� ��4� � ���� *��+�+�-��)�)�+� �$�&�&r )rE rF rG �originr* r �ESM_APPS_TITLE�title�ESM_APPS_DESCRIPTION� description�ESM_APPS_HELP_TEXT� help_textr< r rL r rM r r rT rN rO s @r rQ rQ ] sh �� �
�F� �D� � #� #�E��/�/�K��+�+�I�-�M�
'��+�+�
'� �t�U�4�!2�2�3�3� 4�
'�
'r rQ c � � � e Zd ZdZdZej Zej Z ej Z dZ de
j deeedef f f� fd� Z� xZS )�ESMInfraEntitlementz esm-infra� UbuntuESMzubuntu-pro-esm-infra.gpgr r Nc � �� ddl m} t �| �
|� \ }}|r6 || j � t j � r| j � ||fS r )r r r! rT r# r � is_current_series_active_esmrA rV s �r rT zESMInfraEntitlement.disablez sO �� � I�"'�'�/�(�";� ��4� � ���� *��2�2�4��)�)�+� �$�&�&r )rE rF rG r* rY r �ESM_INFRA_TITLEr[ �ESM_INFRA_DESCRIPTIONr] �ESM_INFRA_HELP_TEXTr_ r< r rL r rM r r rT rN rO s @r ra ra r sh �� � �D�
�F� � $� $�E��0�0�K��,�,�I�.�M�
'��+�+�
'� �t�U�4�!2�2�3�3� 4�
'�
'r ra )r5 �typingr r r �uaclientr r r � uaclient.aptr r
r �uaclient.defaultsr �uaclient.entitlementsr
�uaclient.entitlements.baser �(uaclient.entitlements.entitlement_statusr �
uaclient.utilr �RepoEntitlementr rQ ra � r r �<module>rr sX �� � %� %� *� *� M� M� -� &� 4� F� 0�M
��-�-� M
�`'�+� '�*'�,� 'r |