'N d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
mZ d dlmZ ej e Z G d d
e
ZdS ) N)
HookEvents) HooksConfig)
LicenseCLN) EventHook)ValidationError)
RootEndpointsbind)notifierc @ e Zd ZddZ edd d Z edd d Z edd d Z edd
d Z ed d
d Z ed d dd Z
ed d dd Z dS )HooksEndpointsNc v |t j vr(||k r$t d | d S d S )Nz "{}" is not valid event for hook)r EVENTSr format)selfeventextras U/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/simple_rpc/hooks.py _check_eventzHooksEndpoints._check_event sG
) ) )eunn!299%@@
* )nn hookaddc K | | t j || }|s#t d || d|d< d|iS )Nr pathzUnable to add hook "{} {}"
registeredstatusitemsr r add_hookr r r r r results r hook_addzHooksEndpoints.hook_add ss % #%d;;; !,33E4@@
(x r deletec K | | t j || }|s#t d || d|d< d|iS )Nr z Unable to delete hook "{} {}" unregisteredr r )r r delete_hookr r r s r hook_deletezHooksEndpoints.hook_delete! ss % &U>>> !/66udCC
*x r listc b K | |d t j | }d|iS )Nallr )r r list_events)r r r! s r hook_listzHooksEndpoints.hook_list, s7 %'''&u-- r z
add-nativec K | | t j ||d }|s#t d || d|d< d|iS )NT)r r nativez!Unable to add native hook "{} {}"r r r r r s r hook_add_nativez HooksEndpoints.hook_add_native2 su % #%d4HHH !3::5$GG
(x r znotifications-configshowc J K dt iS )Nr )r get)r s r r0 zHooksEndpoints.show= s! **,,--r updatec . K t j rt d |r|d }t j | }t | t j d {V | d {V S )N*This action is not allowed in demo versionr )
r is_demor jsonloadsr r3 r
config_updatedr0 )r r datanew_datas r r3 zHooksEndpoints.updateA s P!"NOO O 8D:d##
X&&&%'''''''''YY[[ r patchc K t j rt d t | t j d {V | d {V S )Nr5 )r r6 r r r3 r
r9 r0 )r r: s r update_uizHooksEndpoints.update_uiL s P!"NOO O
T"""%'''''''''YY[[ r )N)NN) __name__
__module__ __qualname__r r r" r' r, r/ r0 r3 r> r r r r sM
T&%! ! !
T&(! ! !
T&&! ! !
T&, ! ! !
T
&)). . *).
T
(++! ! ! ,+!
T
'**! ! ! +*! ! !r r )r7 logging defence360agent.contracts.configr defence360agent.contracts.hooksr !defence360agent.contracts.licenser defence360agent.model.event_hookr defence360agent.rpc_toolsr defence360agent.rpc_tools.lookupr r defence360agent.subsysr
getLoggerr? loggerr rB r r rM s 7 7 7 7 7 7 7 7 7 7 7 7 8 8 8 8 8 8 6 6 6 6 6 6 5 5 5 5 5 5 @ @ @ @ @ @ @ @ + + + + + + 8 $ $C! C! C! C! C!] C! C! C! C! C!r