wo4hI ddlZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZddlmZejeZGd d e ZdS) N) HookEvents) EventHook)ValidationError) RootEndpointsbind) HooksConfig)notifier) LicenseCLNc@eZdZddZedddZedddZeddd Zedd d Zed d dZ ed dddZ ed dddZ dS)HooksEndpointsNcv|tjvr(||kr$td|dSdS)Nz "{}" is not valid event for hook)rEVENTSrformat)selfeventextras U/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/simple_rpc/hooks.py _check_eventzHooksEndpoints._check_eventsG  ) ) )eunn!299%@@  * )nnhookaddcK||tj||}|s#td||d|d<d|iS)NrpathzUnable to add hook "{} {}" registeredstatusitemsrradd_hookrrrrrresults rhook_addzHooksEndpoints.hook_addss %   #%d;;; !,33E4@@ (x  rdeletecK||tj||}|s#td||d|d<d|iS)NrzUnable to delete hook "{} {}" unregisteredrr)rr delete_hookrrr s r hook_deletezHooksEndpoints.hook_delete"ss %   &U>>> !/66udCC *x  rlistcbK||dtj|}d|iS)Nallr)rr list_events)rrr!s r hook_listzHooksEndpoints.hook_list-s7 %'''&u--  rz add-nativecK||tj||d}|s#td||d|d<d|iS)NT)rrnativez!Unable to add native hook "{} {}"rrrrr s rhook_add_nativezHooksEndpoints.hook_add_native3su %   #%d4HHH !3::5$GG (x  rznotifications-configshowcJKdtiS)Nr)rget)rs rr0zHooksEndpoints.show>s!**,,--rupdatec.Ktjrtd|r|d}tj|}t |tjd{V| d{VS)N*This action is not allowed in demo versionr) r is_demorjsonloadsrr3r config_updatedr0)rrdatanew_datas rr3zHooksEndpoints.updateBs     P!"NOO O  8D:d## X&&&%'''''''''YY[[       rpatchcKtjrtdt|t jd{V|d{VS)Nr5)r r6rrr3r r9r0)rr:s r update_uizHooksEndpoints.update_uiMs     P!"NOO O T"""%'''''''''YY[[       r)N)NN) __name__ __module__ __qualname__rrr"r'r,r/r0r3r>rrr r sM  T&%!!! T&(!!! T&&!!!  T&,!! ! T &))..*). T (++!!!,+! T '**!!!+*!!!rr )loggingr7 defence360agent.contracts.configr defence360agent.model.event_hookrdefence360agent.rpc_toolsr defence360agent.rpc_tools.lookuprrdefence360agent.contracts.hooksrdefence360agent.subsysr !defence360agent.contracts.licenser getLoggerr?loggerr rBrrrMs 777777666666555555@@@@@@@@777777++++++888888  8 $ $C!C!C!C!C!]C!C!C!C!C!r