<reponame>Ali-aqrabawi/piat
import socketserver

from piat.parsers.syslog import SyslogMsg
from piat.utils.threads import ThreadsManager
from piat.utils.decorators import restart_on_failure
from piat.utils.logger import get_logger

LOGGER = get_logger(__name__)


class _SyslogHandler(socketserver.BaseRequestHandler):
    """ Hanlder Syslog Messages """
    callbacks = []

    def handle(self):
        """ Syslog Msg Hanlder"""
        data = bytes.decode(self.request[0].strip())
        ip = self.client_address[0]

        LOGGER.debug("received syslog message from %s, msg=%s", ip, data)

        syslog_msg = SyslogMsg.create_msg(ip, data)

        threads_mgr = ThreadsManager()

        for callback in self.callbacks:
            threads_mgr.add(callback, args=[syslog_msg, ])

        threads_mgr.start()


class SyslogServer:
    """ Syslog Server API"""

    def __init__(self, callbacks, port=514):
        self._callbacks = callbacks
        self._port = port
        self._setup()
        LOGGER.info("registered %r callbacks to Syslog server" % ([func.__name__ for func in self._callbacks]))

    def _setup(self):
        """ Setup Server"""
        assert isinstance(self._callbacks, list), \
            "callbacks should be list of functions type not %s" % type(self._callbacks)
        _SyslogHandler.callbacks = self._callbacks
        self.server = socketserver.UDPServer(('0.0.0.0', self._port), _SyslogHandler)

    @restart_on_failure
    def start(self):
        """ Start Server """
        LOGGER.info("Syslog service Started...")
        self.server.serve_forever(poll_interval=0.5)

import socketserver

from piat.parsers.syslog import SyslogMsg
from piat.utils.threads import ThreadsManager
from piat.utils.decorators import restart_on_failure
from piat.utils.logger import get_logger

LOGGER = get_logger(__name__)


class _SyslogHandler(socketserver.BaseRequestHandler):
    """ Hanlder Syslog Messages """
    callbacks = []

    def handle(self):
        """ Syslog Msg Hanlder"""
        data = bytes.decode(self.request[0].strip())
        ip = self.client_address[0]

        LOGGER.debug("received syslog message from %s, msg=%s", ip, data)

        syslog_msg = SyslogMsg.create_msg(ip, data)

        threads_mgr = ThreadsManager()

        for callback in self.callbacks:
            threads_mgr.add(callback, args=[syslog_msg, ])

        threads_mgr.start()


class SyslogServer:
    """ Syslog Server API"""

    def __init__(self, callbacks, port=514):
        self._callbacks = callbacks
        self._port = port
        self._setup()
        LOGGER.info("registered %r callbacks to Syslog server" % ([func.__name__ for func in self._callbacks]))

    def _setup(self):
        """ Setup Server"""
        assert isinstance(self._callbacks, list), \
            "callbacks should be list of functions type not %s" % type(self._callbacks)
        _SyslogHandler.callbacks = self._callbacks
        self.server = socketserver.UDPServer(('0.0.0.0', self._port), _SyslogHandler)

    @restart_on_failure
    def start(self):
        """ Start Server """
        LOGGER.info("Syslog service Started...")
        self.server.serve_forever(poll_interval=0.5)
Hanlder Syslog Messages Syslog Msg Hanlder Syslog Server API Setup Server Start Server