diff --git a/dragonion_server/modules/server/handlers/managers/connection.py b/dragonion_server/modules/server/handlers/managers/connection.py index fe75be6..ab2622b 100644 --- a/dragonion_server/modules/server/handlers/managers/connection.py +++ b/dragonion_server/modules/server/handlers/managers/connection.py @@ -1,6 +1,7 @@ from attrs import define from fastapi import WebSocket from dragonion_core.proto.web.webmessage import ( + set_time, webmessages_union, webmessage_error_message_literal, WebErrorMessage @@ -20,7 +21,7 @@ class Connection(object): :param obj: Should be some type of WebMessage :return: """ - await self.ws.send_text(obj.to_json()) + await self.ws.send_text(set_time(obj).to_json()) async def send_error( self, diff --git a/dragonion_server/modules/server/handlers/managers/room.py b/dragonion_server/modules/server/handlers/managers/room.py index fcca882..b84c268 100644 --- a/dragonion_server/modules/server/handlers/managers/room.py +++ b/dragonion_server/modules/server/handlers/managers/room.py @@ -8,6 +8,7 @@ from json.decoder import JSONDecodeError from dragonion_core.proto.web.webmessage import ( webmessages_union, + set_time, WebMessageMessage, WebBroadcastableMessage, WebNotificationMessage, @@ -37,9 +38,9 @@ class Room(object): await ws.receive_text() ) except JSONDecodeError: - await ws.send_text(WebErrorMessage( + await ws.send_text(set_time(WebErrorMessage( 'invalid_webmessage' - ).to_json()) + )).to_json()) await ws.close(reason='invalid_webmessage') return