�
�Hcf. � � � d dl mZmZmZmZ d dlmZmZmZm Z m
Z
d dl m Z m
Z
d dlmZ d dlmZmZ d dlmZ ej* � Z G d� de� Zy )
� )�Any�Dict�Optional�Tuple)�api�contract� event_logger�messages�system)�CanEnableFailure�CanEnableFailureReason)�RepoEntitlement)�AnboxCloudData� anbox_cloud_credentials_file)�MessagingOperationsDictc �F � � e Zd ZdZej
Zej Zej j Z ej Z
dZdZdZdZdZedefd�� Zdeeee f f� fd� Zd ej8 def� fd
� Z d ej8 f� fd � Z d e e e!f d
ee defd�Z"� xZ#S )�AnboxEntitlementz anbox-cloudzubuntu-pro-anbox-cloud.gpgz{}T�Anbox�returnc �D � | j sdt j giS i S )N� post_enable)� access_onlyr
�ANBOX_RUN_INIT_CMD)�selfs �=/usr/lib/python3/dist-packages/uaclient/entitlements/anbox.py� messagingzAnboxEntitlement.messaging s$ � �� � �!�H�$?�$?�#@�A� A��I� c �� �� t �| � � \ }}|s||fS t j � r6| j s*dt t j t j � fS y)NF)TN)
�super�
can_enabler � is_containerr r r
�ONLY_ACCESS_ONLY_SUPPORTEDr
�!ANBOX_FAIL_TO_ENABLE_ON_CONTAINER)r �ret�reason� __class__s �r r zAnboxEntitlement.can_enable% sa �� ��g�(�*� ��V����;� � � � � ��)9�)9�� �*�E�E� �>�>���
� r �progressc � �� t �| � |� }|s|S | j j di � j di � }| j j
d }t
j | j � }|j |d� }t |j di � j di � j dd� |j dd� |j dd� |j d d� �
� }t j |� y ) N� entitlement�
directives� machineTokenz anbox-images�url� �
resourceToken�aptURL�aptKey)�anbox_images_url�anbox_images_resource_token�anbox_cloud_apt_url�anbox_cloud_apt_tokenT) r �_perform_enable�entitlement_cfg�get�cfg�
machine_tokenr �UAContractClient�get_resource_machine_accessr r �write) r r' r$ r* r9 �client�anbox_images_machine_access�anbox_cloud_datar& s �r r5 z AnboxEntitlement._perform_enable6 s� �� ��g�%�h�/����J��)�)�-�-�m�R�@�D�D� �"�
�
� ���.�.�~�>�
��*�*�4�8�8�4��&,�&H�&H� �>�'
�#� *� 8� <� <�]�B� O�
�S� �r�
"�
�S���^�(C�(G�(G� ��)� !+���x�� <�",�.�.��2�">�
�� %�*�*�+;�<�r c �L �� t �| � |� t j � y)NT)r �_perform_disabler �delete)r r' r& s �r rA z!AnboxEntitlement._perform_disableT s �� �
�� ��*�$�+�+�-�r � obligationsr. c � � y)NF� )r rC r. s r �_should_enable_by_defaultz*AnboxEntitlement._should_enable_by_default] s � � r )$�__name__�
__module__� __qualname__�namer
� ANBOX_TITLE�title�ANBOX_DESCRIPTION� description�urls�ANBOX_HOME_PAGE� help_doc_url�ANBOX_HELP_TEXT� help_text�
repo_key_file�
repo_url_tmpl�affordance_check_series�supports_access_only�origin�propertyr r r �boolr r r r �ProgressWrapperr5 rA r �strr rF �
__classcell__)r& s @r r r s� �� � �D� � � �E��,�,�K��=�=�0�0�L��(�(�I�0�M��M� "�� ��
�F�
��2� � �� �E�$��1A�(B�"B� C� �" ��(;�(;� �� �<��)<�)<� �� ��S��>��:B�3�-��
�r r N)�typingr r r r �uaclientr r r r
r �(uaclient.entitlements.entitlement_statusr r
�uaclient.entitlements.repor �uaclient.files.state_filesr r �uaclient.typesr �get_event_logger�eventr rE r r �<module>rf sC �� -� -� B� B�� 7�� 3�%� �%�%�'��N�� Nr |