mirror of
https://github.com/avatao-content/baseimage-tutorial-framework
synced 2024-11-17 06:01:33 +00:00
34 lines
1020 B
Python
34 lines
1020 B
Python
from tfw.internals.networking import Scope
|
|
|
|
from .message_storage import FrontendMessageStorage
|
|
|
|
|
|
class FrontendProxyHandler:
|
|
keys = ['console', 'dashboard', 'frontend', 'message', 'ide.read', 'deploy.finish']
|
|
|
|
def __init__(self):
|
|
self.connector = None
|
|
self._frontend_message_storage = FrontendMessageStorage()
|
|
|
|
def send_message(self, message):
|
|
self.connector.send_message(message, scope=Scope.WEBSOCKET)
|
|
|
|
def handle_event(self, message, _):
|
|
self._frontend_message_storage.save_message(message)
|
|
if message['key'] == 'frontend.ready':
|
|
self.recover_frontend()
|
|
return
|
|
if self._filter_message(message):
|
|
self.send_message(message)
|
|
|
|
@staticmethod
|
|
def _filter_message(message):
|
|
return not message['key'].startswith((
|
|
'ide',
|
|
'message.queue'
|
|
))
|
|
|
|
def recover_frontend(self):
|
|
for message in self._frontend_message_storage.messages:
|
|
self.send_message(message)
|