| Server IP : 188.40.95.74 / Your IP : 216.73.216.142 Web Server : Apache System : Linux cp01.striminghost.net 3.10.0-1160.119.1.el7.tuxcare.els13.x86_64 #1 SMP Fri Nov 22 06:29:45 UTC 2024 x86_64 User : vlasotin ( 1054) PHP Version : 5.6.40 Disable Function : NONE MySQL : ON | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/ |
Upload File : |
import logging
from defence360agent.contracts.config import MyImunifyConfig
from defence360agent.contracts.messages import MessageType
from defence360agent.contracts.plugins import (
MessageSink,
MessageSource,
expect,
)
from defence360agent.myimunify.model import update_users_protection
from defence360agent.subsys.panels import hosting_panel
from defence360agent.subsys.persistent_state import load_state, save_state
logger = logging.getLogger(__name__)
class MyImunifyPlugin(MessageSink, MessageSource):
def __init__(self):
self._previous_myimunify_status = (
load_state("MyImunifyPlugin").get("myimunify_enabled")
or MyImunifyConfig.ENABLED
)
self._loop = None
self._sink = None
async def shutdown(self):
save_state(
"MyImunifyPlugin",
{"myimunify_enabled": self._previous_myimunify_status},
)
async def create_sink(self, loop):
pass
async def create_source(self, loop, sink):
self._loop = loop
self._sink = sink
async def _update_myimunify_users(self):
existing_users = await hosting_panel.HostingPanel().get_users()
await update_users_protection(self._sink, existing_users, False)
@expect(MessageType.ConfigUpdate)
async def on_config_update(self, message: MessageType.ConfigUpdate):
myimunify_enabled = MyImunifyConfig.ENABLED
previous_status = self._previous_myimunify_status
# We're also triggering additional MessageType.ConfigUpdate messages
# so we must update previous_status before triggering new one
self._previous_myimunify_status = myimunify_enabled
if myimunify_enabled and not previous_status:
await self._update_myimunify_users()
if myimunify_enabled != previous_status:
await hosting_panel.HostingPanel().switch_ui_config(
myimunify_enabled=myimunify_enabled
)