Na}{

File Content: `fgw.py`

import contextlib
import logging
from typing import Generator

from defence360agent.api.server import send_message, FGWSendMessgeException
from defence360agent.contracts.config import Core
from defence360agent.contracts.plugins import MessageSink
from defence360agent.internals.global_scope import g
from defence360agent.plugins.client import SendToServerClient
from defence360agent.utils import Scope

logger = logging.getLogger(__name__)


class SendToServerFGW(SendToServerClient, MessageSink):
    SCOPE = Scope.IM360

    @contextlib.contextmanager
    def _get_api(
        self,
    ) -> Generator[send_message.FileBasedGatewayAPI, None, None]:
        api = send_message.FileBasedGatewayAPI(Core.VERSION)

        yield self._set_api_attrs(api)

    async def _send_pending_messages(self) -> None:
        if self._pending.empty():
            return None

        messages = self._pending.pop_all()
        logger.info("Sending %s messages", len(messages))
        with self._get_api() as api:
            try:
                if g.get("DEBUG"):
                    logger.info(
                        "Writing messages to gw folder: %s",
                        list((m[0], m[1][:40], len(m[1])) for m in messages),
                    )
                await api.send_messages(messages)
                if self._pending.qsize() > 0:
                    logger.info(
                        "Still need to send %s messages",
                        self._pending.qsize(),
                    )
            except FGWSendMessgeException:
                self._pending.put_many(messages)

                logger.warning(
                    "Unsuccessful to send %s messages",
                    self._pending.qsize(),
                )

Name	Size	Permission
__pycache__	---	0755
__init__.py	0 bytes	0644
cagefs.py	5265 bytes	0644
cpanel_uploader.py	2777 bytes	0644
export_wblist.py	3854 bytes	0644
fgw.py	1747 bytes	0644
lfd.py	3508 bytes	0644
modsec_ruleset_checker.py	5031 bytes	0644
ossec_rules_checker.py	1803 bytes	0644
pam_manager.py	6829 bytes	0644
php_immunity.py	1438 bytes	0644
remoteip_install.py	1053 bytes	0644
repeater.py	1187 bytes	0644
send_server_config.py	4534 bytes	0644
service_manager.py	4756 bytes	0644
startup_actions.py	2483 bytes	0644
strategy_getter.py	2068 bytes	0644
waf_rules_configurator.py	2535 bytes	0644
whitelist_current_user.py	2817 bytes	0644

File Content: fgw.py

File Content: `fgw.py`