Edit File: plesk_notifications.cpython-311.pyc
� GhD � � � d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddl m Z ej e � � ZdZd Zg d �Z G d � d e� � ZdS ) u This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <https://www.gnu.org/licenses/>. Copyright © 2019 Cloud Linux Software Inc. This software is also available under ImunifyAV commercial license, see <https://www.imunify360.com/legal/eula> � N)� lru_cache)�Path)� MessageSink)� HooksConfig)�notifierzE/opt/psa/admin/plib/modules/imunify360/scripts/send-notifications.phpz@/opt/imunify360/venv/share/imunify360/scripts/send-notifications)�CUSTOM_SCAN_MALWARE_FOUND�USER_SCAN_MALWARE_FOUND�REALTIME_MALWARE_FOUNDc �\ � e Zd Zd� Z ed�� � defd�� � Zdefd�Zd� Zd� Z d S ) �PleskNotificationsHooksc � �� K � | � � � r2| � � � s | � � � � d{V �� dS dS | � � � � d{V �� dS )zMessageSink methodN)� is_supported� is_applied� add_hooks� remove_hooks)�self�loops �U/opt/imunify360/venv/lib/python3.11/site-packages/imav/plugins/plesk_notifications.py� create_sinkz#PleskNotificationsHooks.create_sink- s� � � � � � � � � � &��?�?�$�$� '��n�n�&�&�&�&�&�&�&�&�&�&�&� '� '� �#�#�%�%� %� %� %� %� %� %� %� %� %� � )�maxsize�returnc � � t t � � � � � o%t t � � � � � S )N)r � SCRIPT_PATH�exists� HOOK_PATH)r s r r z$PleskNotificationsHooks.is_supported5 s3 � ��K� � �'�'�)�)�F�d�9�o�o�.D�.D�.F�.F�Fr c � � t � � � � � }t d� |� di � � � � � D � � � � S )Nc �P � g | ]#\ }}|t v �t |d d v ��$S )�SCRIPT�scripts)�EVENTSr ��.0�event�rules r � <listcomp>z6PleskNotificationsHooks.is_applied.<locals>.<listcomp>< s; � � � � � �E�4��F�?�?� �T�(�^�I� 6�6�"�?�?r �rules)r �get�all�items)r �configs r r z"PleskNotificationsHooks.is_applied9 s_ � � � � �"�"�$�$��� � �#)�:�:�g�r�#:�#:�#@�#@�#B�#B� � � � � � r c � � K � t � � � � � }dd� |� di � � � � � D � � i}d}|d � � � D ]M\ }}t |d d vr3d|d d< |d d � t � � d}�N|r�t � � � |� � t j � � � d {V �� t � d� � d S # t $ r t � d � � Y d S w xY wd S ) Nr( c �, � i | ]\ }}|t v �||��S � �r" r# s r � <dictcomp>z5PleskNotificationsHooks.add_hooks.<locals>.<dictcomp>F �/ � � � � � �E�4��F�?�?� �t�"�?�?r Fr r! T�enabledz%Hooks added and configuration updated�9Notifier is not running, cannot send CONFIG_UPDATED event) r r) r+ r �append�updater �config_updated�logger�info�ConnectionRefusedError�warning�r r, �data�updatedr% r& s r r z!PleskNotificationsHooks.add_hooksC s~ � � � � � � �"�"�$�$�� � � �#)�:�:�g�r�#:�#:�#@�#@�#B�#B�� � � �� �� ��=�.�.�0�0� � �K�E�4���X��y� 9�9�9�,0��X��y�)��X��y�)�0�0��;�;�;� ��� � E� �M�M� � �� &� &� &� E� �-�/�/�/�/�/�/�/�/�/� � � � C�D�D�D�D�D�� *� � � ����O�� � � � � � ���� E� Es �%D �$E�Ec � �� K � t � � � � � }dd� |� di � � � � � D � � i}d}|d � � � D ]j\ }}t |d d v rP|d d � t � � t |d d � � dk |d d< d}�k|r�t � � � |� � t j � � � d {V �� t � d � � d S # t $ r t � d � � Y d S w xY wd S ) Nr( c �, � i | ]\ }}|t v �||��S r/ r0 r# s r r1 z8PleskNotificationsHooks.remove_hooks.<locals>.<dictcomp>` r2 r Fr r! r r3 Tz'Hooks removed and configuration updatedr4 ) r r) r+ r �remove�lenr6 r r7 r8 r9 r: r; r< s r r z$PleskNotificationsHooks.remove_hooks] s� � � � � � � �"�"�$�$�� � � �#)�:�:�g�r�#:�#:�#@�#@�#B�#B�� � � �� �� ��=�.�.�0�0� � �K�E�4��D��N�9� 5�5�5��X��y�)�0�0��;�;�;�,/��X��y�0I�,J�,J�a�,O��X��y�)� ��� � G� �M�M� � �� &� &� &� G� �-�/�/�/�/�/�/�/�/�/� � � � E�F�F�F�F�F�� *� � � ����O�� � � � � � ���� G� Gs �D7 �7$E � E N) �__name__� __module__� __qualname__r r �boolr r r r r/ r r r r , s� � � � � � �&� &� &� �Y�q����G�d� G� G� G� ��G� �D� � � � �E� E� E�4G� G� G� G� Gr r )�__doc__�logging� functoolsr �pathlibr �!defence360agent.contracts.pluginsr � defence360agent.contracts.hooksr �defence360agent.subsysr � getLoggerrC r8 r r r" r r/ r r �<module>rO s� ��� �* ���� � � � � � � � � � � � � :� 9� 9� 9� 9� 9� 7� 7� 7� 7� 7� 7� +� +� +� +� +� +� �� �8� $� $�� L� � O� � � � ��IG� IG� IG� IG� IG�k� IG� IG� IG� IG� IGr