o

�N�gU�@srdZd
dl
Z
d
dlZd
dlZe
�e�
ZdZdd�Z	ddd�
Z
d	d
�Zdd�Zd
d�Z
dd�Zdd�Zddd�
ZdS)z�
Support for Varnish

.. versionadded:: 2014.7.0

.. note::

    These functions are designed to work with all implementations of Varnish
    from 3.x onwards
�NZvarnishcCs$tj
j�d
�
rtj
j�d�
rtSdS)z6
    Only load the module if varnish is installed
    �varnishd�
varnishadm)Fz^The varnish execution module failed to load: either varnishd or varnishadm is not in the path.)�salt�utils�path�which�__virtualname__�r	r	�H/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/varnish.py�__virtual__s

rr	cKsFd
|g}|�dd�|
D�
�

t
�dd�|�
�
td|f
ddi
|�
�
S)	a

    Execute varnishadm command
    return the output of the command

    cmd
        The command to run in varnishadm

    params
        Any additional args to add to the command line

    kwargs
        Additional options to pass to the salt cmd.run_all function
    rc
Ssg|]}
|
du
r|
�qS�
Nr	)�.0�paramr	r	r
�
<listcomp>3sz#_run_varnishadm.<locals>.<listcomp>z
Executing: %s�
 zcmd.run_all�python_shellF)�extend�log�debug�join�__salt__)�cmd�params�kwargsr	r	r
�_run_varnishadm$s


rcCs.d
dg}td|dd�}
t
�d|
��d�
}|S)z~
    Return server version from varnishd -V

    CLI Example:

    .. code-block:: bash

        salt '*' varnish.version
    rz-Vzcmd.runF)
rz\(varnish-([^\)]+)\)�
)r�re�search�group)r�out�retr	r	r
�version8s



r!c

Cstd
|g
�ddkS)z
    Add ban to the varnish cache

    CLI Example:

    .. code-block:: bash

        salt '*' varnish.ban ban_expression
    �ban�retcoder
)
r)
Zban_expressionr	r	r
r"Hs
r"c
Cs*td
�
}|dr
dS|d�
d�
dd�S)zx
    List varnish cache current bans

    CLI Example:

    .. code-block:: bash

        salt '*' varnish.ban_list
    zban.listr#F�stdout�

rN)r�split)
r r	r	r
�ban_listUs


r'cCstd
�
S)zm
    Purge the varnish cache

    CLI Example:

    .. code-block:: bash

        salt '*' varnish.purge
    zreq.url ~ .)
r"r	r	r	r
�purgefs
r(cCstd
|t
|
�
g�ddkS)z�
    Set a param in varnish cache

    CLI Example:

    .. code-block:: bash

        salt '*' varnish.param_set param value
    z	param.setr#r
)r�str)r�valuer	r	r
�	param_setss
r+c
Cs^td
|g
�}
|
drdSi}|
d�
d�
D]}t�d|�}|�d�
||�d�
<|r,
|Sq|S)	z}
    Show params of varnish cache

    CLI Example:

    .. code-block:: bash

        salt '*' varnish.param_show param
    z
param.showr#Fr$r%z^(\w+)\s+(.*)$�r)rr&rrr)rr �result�line�
mr	r	r
�
param_show�s







�r0)
r	r)�__doc__�loggingrZsalt.utils.pathr�	getLogger�__name__rrrrr!r"r'r(r+r0r	r	r	r
�<module>
s