baseimage-tutorial-framework/tfw/components/frontend/frontend_handler.py

40 lines
1.2 KiB
Python
Raw Normal View History

from tfw.internals.networking import Scope
2019-05-26 16:26:33 +00:00
from .message_storage import FrontendMessageStorage
2019-06-27 12:36:18 +00:00
class FrontendHandler:
# TODO: do not store dashboard messages like reloadFrontend
keys = [
'console.read',
'console.write',
'console.showLiveLogs',
'console.rewriteContentWithProcessLogsOnDeploy',
'dashboard.layout',
'dashboard.hideMessages',
'dashboard.terminalMenuItem',
'dashboard.reloadFrontend',
'dashboard.reloadIframe',
'message.config',
'message.queue',
'message.send'
]
2019-05-26 16:26:33 +00:00
def __init__(self):
self.connector = None
self.keys = [*type(self).keys, 'recover']
self._frontend_message_storage = FrontendMessageStorage(type(self).keys)
2019-05-26 16:26:33 +00:00
def send_message(self, message):
self.connector.send_message(message, scope=Scope.WEBSOCKET)
def handle_event(self, message, _):
2019-05-26 16:26:33 +00:00
self._frontend_message_storage.save_message(message)
if message['key'] == 'recover':
self.recover_frontend()
self.send_message(message)
2019-05-26 16:26:33 +00:00
def recover_frontend(self):
for message in self._frontend_message_storage.messages:
self.send_message(message)