run.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License version 3 for
# more details.
#
# You should have received a copy of the GNU General Public License version 3
# along with this program; if not, write to the Free Software Foundation, Inc., 51
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# (c) 2023 Onlime GmbH
#
import os
import sentry_sdk
import socket
import threading
from app.conf import Config
from app.handler import Handler
from app.logging import Logging
from app.db import DbConnectionPool


class Daemon:

    def __init__(self) -> None:
        self.conf = Config()
        self.logger = Logging.get_logger(self.conf)
        self.db_pool = DbConnectionPool(self.conf)
        # TODO: Improve socket configuration parsing
        self.socket_conf = self.conf.get_array('SOCKET', '127.0.0.1,10033')
        self.init_sentry()
        self.run()

    def init_sentry(self) -> None:
        sentry_dsn = self.conf.get('SENTRY_DSN')
        if sentry_dsn:
            sentry_sdk.init(
                dsn=sentry_dsn,
                environment=self.conf.get('SENTRY_ENVIRONMENT', 'dev'),
            )
            # In addition to (auto-detected) server_name, provide FQDN as tag
            sentry_sdk.set_tag('fqdn', socket.getfqdn())

    def run(self) -> None:
        """Run server"""
        self.open_socket()
        while True:
            try:
                conn, addr = self.socket.accept()
                threading.Thread(target=Handler, args=(conn, addr, self.conf, self.logger, self.db_pool)).start()
            except KeyboardInterrupt:
                break
        self.close_socket()

    def open_socket(self) -> None:
        """Open socket for communications"""
        socket_conf = self.socket_conf
        if len(socket_conf) == 1:
            # TODO: Create socket directory if it does not exist
            socket_path = socket_conf[0]
            try:
                os.remove(socket_path)
            except OSError:
                if os.path.exists(socket_path):  # pragma: no cover
                    raise
            self.bind_socket(socket.AF_UNIX, socket_path)
            os.chmod(socket_path, 0o666)
        else:
            host, port = socket_conf[0], int(socket_conf[1])
            if ':' in host:
                self.bind_socket(socket.AF_INET6, (host, port))
            elif '.' in host:
                self.bind_socket(socket.AF_INET, (host, port))
            else:
                raise ValueError('Invalid socket configuration')
        self.socket.listen(5)

    def bind_socket(self, family: int, address):
        """Bind socket"""
        self.socket = socket.socket(family, socket.SOCK_STREAM)
        self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.socket.bind(address)

    def close_socket(self) -> None:
        """Close socket"""
        self.socket.close()
        socket_conf = self.socket_conf
        if len(socket_conf) == 1:
            try:
                os.remove(socket_conf[0])
            except OSError as error:
                self.logger.error('run.py - Error removing socket file: %s', error)


if __name__ == '__main__':  # pragma: no cover
    Daemon()