PyCS/pycs/frontend/WebServer.py

from os.path import exists

import eventlet
import socketio

from pycs.ApplicationStatus import ApplicationStatus
from pycs.frontend.FileProvider import FileProvider


class WebServer:
    def __init__(self, app_status: ApplicationStatus):
        # initialize web server
        # TODO svg files need the correct mime type to be displayed correctly
        if exists('webui/index.html'):
            print('production build')
            files = FileProvider(app_status)
            self.__sio = socketio.Server()
            self.__app = socketio.WSGIApp(self.__sio, files, static_files={'/': 'webui/'})
        elif exists('webui/dist/index.html'):
            print('production build')
            files = FileProvider(app_status)
            self.__sio = socketio.Server()
            self.__app = socketio.WSGIApp(self.__sio, files, static_files={'/': 'webui/dist/'})
        else:
            print('development build')
            files = FileProvider(app_status, cors=True)
            self.__sio = socketio.Server(cors_allowed_origins='*')
            self.__app = socketio.WSGIApp(self.__sio, files)

        # save every change in application status and send it to the client
        app_status.subscribe(self.__update_application_status, immediate=True)

        # define events
        @self.__sio.event
        def connect(id, msg):
            self.__sio.emit('app_status', self.__status, to=id)

        @self.__sio.on('set')
        def set_status(id, msg):
            path = msg['path'].split('/')
            value = msg['value']

            obj = app_status
            for p in path[:-1]:
                obj = obj[p]

            obj[path[-1]] = value

        @self.__sio.on('add')
        def add_status(id, msg):
            path = msg['path'].split('/')
            value = msg['value']

            obj = app_status
            for p in path[:-1]:
                obj = obj[p]

            obj[path[-1]].append(value)

        # finally start web server
        host = app_status['settings']['frontend']['host']
        port = app_status['settings']['frontend']['port']

        eventlet.wsgi.server(eventlet.listen((host, port)), self.__app)

    def __update_application_status(self, status):
        self.__status = status
        self.__sio.emit('app_status', status)