Password in WebConnectionMessage, avatar in WebMessageMessage

This commit is contained in:
BarsTiger
2023-07-24 23:31:29 +03:00
parent 813c55bb91
commit 265705e391
2 changed files with 4 additions and 0 deletions

View File

@@ -14,6 +14,7 @@ class WebConnectionMessage:
""" """
username: str username: str
public_key: bytes public_key: bytes
password: str
type: Final = "connect" type: Final = "connect"

View File

@@ -19,6 +19,7 @@ class WebMessageMessage:
:param message: Encrypted b64-encoded message :param message: Encrypted b64-encoded message
""" """
username: str username: str
avatar: str
message: bytes message: bytes
type: Final = "message" type: Final = "message"
@@ -47,6 +48,7 @@ class WebBroadcastableBuilder:
:param keys: Dict with public keys in format username:public_key :param keys: Dict with public keys in format username:public_key
""" """
from_user: str from_user: str
avatar: str
message_content: str message_content: str
keys: dict[str, bytes] keys: dict[str, bytes]
@@ -60,6 +62,7 @@ class WebBroadcastableBuilder:
self.broadcastable.messages[username] = WebMessageMessage( self.broadcastable.messages[username] = WebMessageMessage(
username=self.from_user, username=self.from_user,
avatar=self.avatar,
message=base64.urlsafe_b64encode(public_key.encrypt( message=base64.urlsafe_b64encode(public_key.encrypt(
self.message_content.encode(), self.message_content.encode(),
padding=padding.OAEP( padding=padding.OAEP(