From 545e3f07c995d183dc6b4dd7027c3fa4f7871739 Mon Sep 17 00:00:00 2001
From: BarsTiger <kit.barsic@gmail.com>
Date: Wed, 26 Jul 2023 16:50:21 +0300
Subject: [PATCH] set_time for all messages

---
 dragonion_core/proto/web/webmessage/__init__.py   | 4 +++-
 dragonion_core/proto/web/webmessage/message.py    | 4 ----
 dragonion_core/proto/web/webmessage/webmessage.py | 6 ++++++
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/dragonion_core/proto/web/webmessage/__init__.py b/dragonion_core/proto/web/webmessage/__init__.py
index 04bf89d..204d88e 100644
--- a/dragonion_core/proto/web/webmessage/__init__.py
+++ b/dragonion_core/proto/web/webmessage/__init__.py
@@ -1,7 +1,8 @@
 from .webmessage import (
     webmessage_type_literal,
     webmessages_union,
-    WebMessage
+    WebMessage,
+    set_time
 )
 from .server import (
     webmessage_error_message_literal,
@@ -23,6 +24,7 @@ __all__ = [
     'webmessage_type_literal',
     'webmessages_union',
     'WebMessage',
+    'set_time',
 
     'WebMessageMessage',
     'WebBroadcastableMessage',
diff --git a/dragonion_core/proto/web/webmessage/message.py b/dragonion_core/proto/web/webmessage/message.py
index 6e71f56..2e21247 100644
--- a/dragonion_core/proto/web/webmessage/message.py
+++ b/dragonion_core/proto/web/webmessage/message.py
@@ -29,10 +29,6 @@ class WebMessageMessage:
     def decrypt(self, identity: Identity):
         return identity.decrypt(self.message)
 
-    def set_time(self):
-        self.time = datetime.now()
-        return self
-
 
 @dataclass_json
 @dataclass
diff --git a/dragonion_core/proto/web/webmessage/webmessage.py b/dragonion_core/proto/web/webmessage/webmessage.py
index 63ed18a..a4a84c9 100644
--- a/dragonion_core/proto/web/webmessage/webmessage.py
+++ b/dragonion_core/proto/web/webmessage/webmessage.py
@@ -22,6 +22,7 @@ __all__ = [
     'webmessage_type_literal',
     'webmessages_union',
     'WebMessage',
+    'set_time',
 
     'WebMessageMessage',
     'WebBroadcastableMessage',
@@ -87,3 +88,8 @@ class WebMessage:
             "connect_answer": WebConnectionResultMessage.from_json,
             "broadcastable": WebBroadcastableMessage.from_json
         }[_WebAnyMessage.from_json(data).type](data)
+
+
+def set_time(webmessage: webmessages_union):
+    webmessage.time = datetime.now()
+    return webmessage