Mini Shell
3
C��e�2 � @ s d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
mZ d dlmZ d dlmZ ejdd � ejd
d � d d lmZmZmZmZmZ d d
lmZ ee �Z dgZ G dd� de �Z G dd� de�Z!G dd� de!�Z"G dd� de!�Z#dS )� N)�ABC�abstractmethod)�get_module_logger)�_)�GError� VariantType�Variant�Bytes)� progressQ)�FlatpakInstallErrorZFlatpakz1.0ZGioz2.0)� Transaction� Installation�Remote�TransactionOperationType�TransactionErrorDetails)�File�FlatpakPayloadc @ s� e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z e
dd� �Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zed&d'� �Zd(S ))r z9Main class to handle flatpak installation and management.ZAnacondazfile:///flatpak/repoc C s || _ d| _d| _dS )a� Create and initialize this class.
This flatpak implementation works on a repository stored in the stage2 image specifically
for the SilverBlue image. It will be used from the ostree payload after the installation.
This is a temporal solution for SilverBlue use-case. It will be extended as full featured
payload in the future.
:param sysroot: path to the system root
:type sysroot: str
N)�_sysroot�_remote_refs_list� _transaction)�selfZsysroot� r � /usr/lib64/python3.6/flatpak.py�__init__2 s zFlatpakPayload.__init__c C s t jj| jd�}| j|� dS )z�Create flatpak objects and set them to install to the result system.
This call will re-initialize current settings so everything set here will be cleaned.
zvar/lib/flatpakN)�os�path�joinr �initialize_with_path)r � target_pathr r r �initialize_with_system_pathB s z*FlatpakPayload.initialize_with_system_pathc C sD t jd|� | j| j| jd�}| j||�}| j|�| _t|�| _ dS )a> Create flatpak objects and set them for the given target path.
The initialization process will create a repository on the given path.
This call will re-initialize current settings so everything set here will be cleaned.
:param str target_path: path where we want to install flatpaks
z Configure flatpak for path %sFN)
�log�debug�_create_flatpak_remote�LOCAL_REMOTE_NAME�LOCAL_REMOTE_PATH� _create_flatpak_installation�_create_flatpak_transactionr �RemoteRefsListr )r r �remote� installationr r r r J s
z#FlatpakPayload.initialize_with_pathc C s" t j|�}|j|� |j|� |S )N)r �newZset_gpg_verifyZset_url)r �namer Z
gpg_verifyr( r r r r" [ s
z%FlatpakPayload._create_flatpak_remotec C s* t j|�}tj|dd �}|j|dd � |S )NF)r Z new_for_pathr
�
add_remote)r r( r � install_pathr) r r r r% b s
z+FlatpakPayload._create_flatpak_installationc C s8 t j|�}|jd| j� |jd| j� |jd| j� |S )NZ
new_operationZoperation_doneZoperation_error)r Znew_for_installationZconnect�_operation_started_callback�_operation_stopped_callback�_operation_error_callback)r r) � transactionr r r r& i s
z*FlatpakPayload._create_flatpak_transactionc C sR t jd� | jrH| jj� j� }|j� }tjj|�rHt jd|� tj |� d| _dS )z�Clean the current repository and settings.
One of the initialize methods have to be called before the flatpak object can be
used again.
z Cleaning up flatpak repositoryz Removing flatpak repository %sN)
r r! r �get_installation�get_pathr r �exists�shutilZrmtree)r r r r r �cleanupq s
zFlatpakPayload.cleanupc C s t jj| jdd� �S )zQTest if flatpak installation source is available.
:return: bool
� N)r r �isdirr$ )�clsr r r � is_available� s zFlatpakPayload.is_availablec C s
| j j� S )z�Get required size to install all the flatpaks.
:returns: bytes required to install all flatpaks in the remote
:rtype: int
)r �get_sum_installation_size)r r r r �get_required_size� s z FlatpakPayload.get_required_sizec C s8 t jd||� | j||d�}| jj� }|j|dd� dS )z�Add a new remote to the existing installation.
:param str name: name of the remote
:param str url: url pointing to the remote (use file:// for local paths)
z"Adding a new flatpak remote %s: %sTN)r r! r" r r2 r, )r r+ Zurlr( r) r r r r, � s
zFlatpakPayload.add_remotec C sP t jd|� | jj� }x4|j� D ](}|j� |kr |j|d� t jd|� q W dS )znRemove remote from the existing installation.
:param str name: Name of the remote to remove.
z Removing a flatpak remote %sNzFlatpak remote %s removed)r r! r r2 Z list_remotesZget_name�
remove_remote)r r+ r) r( r r r r= � s
z FlatpakPayload.remove_remotec C sJ | j � y| jj� W n. tk
rD } ztt|���W Y dd}~X nX dS )z-Install all the refs contained on the remote.N)�_stuff_refs_to_transactionr Zrunr r �str)r Zexnr r r � install_all� s
zFlatpakPayload.install_allc C s* x$| j j� D ]}| jj| j|d � q W d S )N)r �get_refs_full_formatr Z add_installr# )r �refr r r r> � s z)FlatpakPayload._stuff_refs_to_transactionc C s t | jj� �}|j|� dS )ziReplace remote on all the installed refs.
:param str new_remote: name of the new remote
N)�InstalledRefsListr r2 � replace_installed_refs_remote)r Z
new_remoteZinstalled_refs_listr r r rD � s z,FlatpakPayload.replace_installed_refs_remotec C s* | j |d� tjtd�d|j� i � dS )a� Start of the new operation.
:param transaction: the main transaction object
:type transaction: Flatpak.Transaction instance
:param operation: object describing the operation
:type operation: Flatpak.TransactionOperation instance
:param progress: object providing progess of the operation
:type progress: Flatpak.TransactionProgress instance
ZstartedzInstalling %(flatpak_name)sZ flatpak_nameN)�_log_operationr
Z send_messager �get_ref)r r1 � operationZprogressr r r r. � s
z*FlatpakPayload._operation_started_callbackc C s | j |d� dS )a� Existing operation ended.
:param transaction: the main transaction object
:type transaction: Flatpak.Transaction instance
:param operation: object describing the operation
:type operation: Flatpak.TransactionOperation instance
:param str commit: operation was committed this is a commit id
:param result: object containing details about the result of the operation
:type result: Flatpak.TransactionResult instance
ZstoppedN)rE )r r1 rG Zcommit�resultr r r r/ � s z*FlatpakPayload._operation_stopped_callbackc C s& | j |d� tjd|j|tjk� dS )a Process error raised by the flatpak operation.
:param transaction: the main transaction object
:type transaction: Flatpak.Transaction instance
:param operation: object describing the operation
:type operation: Flatpak.TransactionOperation instance
:param error: object containing error description
:type error: GLib.Error instance
:param details: information if the error was fatal
:type details: int value of Flatpak.TransactionErrorDetails
ZfailedzCFlatpak installation failed with message: '%s' -- error is fatal %sN)rE r �error�messager ZFATAL)r r1 rG rI Zdetailsr r r r0 � s z(FlatpakPayload._operation_error_callbackc C s& t j| j� �}tjd|| j� |� d S )Nz(Flatpak operation: %s of ref %s state %s)r Z to_stringZget_operation_typer r! rF )rG �stateZoperation_type_strr r r rE � s z FlatpakPayload._log_operationN)�__name__�
__module__� __qualname__�__doc__r# r$ r r r r" r% r& r6 � classmethodr: r< r, r= r@ r> rD r. r/ r0 � staticmethodrE r r r r r , s*
c @ s4 e Zd Zdd� Zedd� �Zedd� �Zdd� Zd S )
� BaseRefsListc C s || _ g | _dS )z�Load all flatpak refs from the installed system.
Makes easier operations above the refs.
:param installation: flatpak installation instance with remotes attached
:type installation: Flatpak.Installation instance
N)�
_installation�_refs)r r) r r r r � s zBaseRefsList.__init__c C s | j s| j� | j S )z'Get list of installed application refs.)rT �
_load_refs)r r r r �refs s zBaseRefsList.refsc C s d S )Nr )r r r r rU
s zBaseRefsList._load_refsc C s& g }x | j D ]}|j|j� � q W |S )an Get list of refs in full format.
This formatting is used for example for installation.
The format looks like:
[app|runtime]/ref/arch/branch
examples:
runtime/org.videolan.VLC.Plugin.bdj/x86_64/3-18.08
app/org.gnome.Gnote/x86_64/stable
:return: list of refs in the full format
:rtype: [str]
)rV �appendZ
format_ref)r rH rB r r r rA s z!BaseRefsList.get_refs_full_formatN) rL rM rN r �propertyrV r rU rA r r r r rR � s rR c @ s e Zd Zdd� Zdd� ZdS )r' c C s | j jtjd�| _dS )a Load remote application references.
This will load the list just once. We can do that because we support only one repository
on the fixed place right now. This have to be re-implemented when there will be a proper
flatpak support.
N)rS Zlist_remote_refs_syncr r# rT )r r r r rU ' s zRemoteRefsList._load_refsc C s$ d}x| j D ]}||j� }q W |S )z�Get sum of the installation size for all the flatpaks.
:return: sum of bytes of the installation size of the all flatpaks
:rtype: int
r )rV Zget_installed_size)r Zsize_sumrB r r r r; 2 s z(RemoteRefsList.get_sum_installation_sizeN)rL rM rN rU r; r r r r r' % s r' c @ s e Zd ZdZdd� Zdd� ZdS )rC z (ssasta{sv})c C s | j j� | _d S )N)rS Zlist_installed_refsrT )r r r r rU C s z InstalledRefsList._load_refsc s� | j j� }|j� }t| j�}x�| j� D ]�}tjj||d�}t|d��}|j � }W dQ R X t
j |t |�d�� � fdd�t
� j� �D �}t
d|�|d< t
j|� } | j� j� }
t|d ��}|j|
� W dQ R X q&W dS )
z�Replace remote for all the refs.
:param str new_remote_name: the remote name which will be used instead of the current one
z
active/deploy�rbNFc s g | ]}� j |��qS r )Zget_child_value)�.0�i)�variantr r �
<listcomp>W s zCInstalledRefsList.replace_installed_refs_remote.<locals>.<listcomp>�sr �wb)rS r3 r �#FLATPAK_DEPLOY_DATA_GVARIANT_STRINGrA r r r �open�readr Znew_from_bytesr �rangeZ
n_childrenZ new_tupleZget_data_as_bytes�get_data�write) r Znew_remote_namer- Z variant_typerB Z deploy_path�fZcontentZchildrenZ new_variantZ
serializedr )r\ r rD F s
z/InstalledRefsList.replace_installed_refs_remoteN)rL rM rN r` rU rD r r r r rC ? s rC )$r r5 Zgi�abcr r Zpyanaconda.anaconda_loggersr Zpyanaconda.core.i18nr Zpyanaconda.core.glibr r r r Zpyanaconda.progressr
Zpyanaconda.payload.errorsr Zrequire_versionZgi.repository.Flatpakr r
r r r Zgi.repository.Gior rL r �__all__�objectr rR r' rC r r r r �<module> s& I1