o

�N�g��@sRdZd
dl
Zd
dlZdd�Zdd�Zdd�Zd	d
�Zddd
�
Zdd�Z	dd�Z
dS)z
Support for Layman
�NcCs"td
dkrt
jj�d�
rdSdS)z;
    Only work on Gentoo systems with layman installed
    �osZGentoo�layman)FzYlayman execution module cannot be loaded: only available on Gentoo with layman installed.)Z
__grains__�salt�utils�path�which�rr�G/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/layman.py�__virtual__	s

r
cCs,d
}d}
td|�
r|Std|
�
r|
SdS)z�
    Find the correct make.conf. Gentoo recently moved the make.conf
    but still supports the old location, using the old location first
    z/etc/make.confz/etc/portage/make.confzfile.file_existsN�
�__salt__)Zold_confZnew_confrrr	�
_get_makeconfs




�r
c
s�t�}
t
��d
|��}td|ddd�}|ddkr"tj�|d�
�
t
�}�s<|r<d	}t�}td
|d�s<td||�
�f
d
d�|D�
}
|
S)a'

    Add the given overlay from the cached remote list to your locally
    installed overlays. Specify 'ALL' to add all overlays from the
    remote list.

    Return a list of the new overlay(s) added:

    CLI Example:

    .. code-block:: bash

        salt '*' layman.add <overlay name>
    zlayman --quietness=0 --add �cmd.run_allF�
y)�python_shell�stdin�retcoder
�stdout� source /var/lib/layman/make.conf�
file.containsrzfile.appendc
�g|]}
|
�v
r|
�qSrr��.0�overlay�
�old_overlaysrr	�
<listcomp>B�zadd.<locals>.<listcomp>��list�
list_localrr�
exceptionsZCommandExecutionErrorr
)r�ret�cmdZadd_attempt�new_overlays�srcline�makeconfrrr	�add#s











r'c
s�t�}
t
�}d
|��}td|dd�}|ddkr!tj�|d�
�
t
���s:d}t�}td	|d
�r:td||d�
�f
d
d�|D�
}
|
S)a


    Remove the given overlay from the your locally installed overlays.
    Specify 'ALL' to remove all overlays.

    Return a list of the overlays(s) that were removed:

    CLI Example:

    .. code-block:: bash

        salt '*' layman.delete <overlay name>
    zlayman --quietness=0 --delete rF�
rrr
rrrrzfile.sed�c
rrrr�
r$rr	rcrzdelete.<locals>.<listcomp>r)rr"rr#Zdelete_attemptr%r&rr*r	�deleteFs












r+�ALLc
Csd
|��}
td|
dd�dkS)a

    Update the specified overlay. Use 'ALL' to synchronize all overlays.
    This is the default if no overlay is specified.

    overlay
        Name of the overlay to sync. (Defaults to 'ALL')

    CLI Example:

    .. code-block:: bash

        salt '*' layman.sync
    zlayman --quietness=0 --sync zcmd.retcodeFr(r
r)rr#rrr	�syncgs

r-cC�,d
}td|dd��
d�
}
dd�|
D�
}|S)z�
    List the locally installed overlays.

    Return a list of installed overlays:

    CLI Example:

    .. code-block:: bash

        salt '*' layman.list_local
    z+layman --quietness=1 --list-local --nocolor�cmd.runFr(�

c
S�(g|]}
t|
�
��
dkr|
�
�d
�qS���
��len�split�r�linerrr	r��(zlist_local.<locals>.<listcomp>�rr7�r#�outr"rrr	r y�


r cCr.)z�
    List all overlays, including remote ones.

    Return a list of available overlays:

    CLI Example:

    .. code-block:: bash

        salt '*' layman.list_all
    z%layman --quietness=1 --list --nocolorr/Fr(r0c
Sr1r2r5r8rrr	r�r:zlist_all.<locals>.<listcomp>r;r<rrr	�list_all�r>r?)
r,)�__doc__Zsalt.exceptionsrZsalt.utils.pathr
r
r'r+r-r r?rrrr	�<module>
s


#
!