|
|
import base64 |
|
|
from datetime import datetime |
|
|
from uuid import UUID |
|
|
|
|
|
from app.api.messages.schemas import Message |
|
|
from app.api.participants.schemas import Participant |
|
|
from app.core.config import ( |
|
|
ENCODING_ASCII, |
|
|
ENCODING_UTF8, |
|
|
MESSAGE_EXPECTED_PARTS, |
|
|
MESSAGE_PARSE_EMPTY_ERROR, |
|
|
MESSAGE_PARSE_INVALID_ERROR, |
|
|
MESSAGE_SEPARATOR, |
|
|
TABLE_PARTICIPANTS, |
|
|
TABLE_TOPICS, |
|
|
) |
|
|
from app.core.link import Link, TopicLink, ParticipantLink |
|
|
|
|
|
|
|
|
class MessageLineParser: |
|
|
SEPARATOR = MESSAGE_SEPARATOR |
|
|
EXPECTED_PARTS = MESSAGE_EXPECTED_PARTS |
|
|
|
|
|
@staticmethod |
|
|
def _decode_content(encoded: str) -> str: |
|
|
return base64.b64decode(encoded.encode(ENCODING_ASCII)).decode(ENCODING_UTF8) |
|
|
|
|
|
@staticmethod |
|
|
def _encode_content(value: str) -> str: |
|
|
return base64.b64encode(value.encode(ENCODING_UTF8)).decode(ENCODING_ASCII) |
|
|
|
|
|
@classmethod |
|
|
def parse(cls, line: str) -> Message: |
|
|
raw = line.strip() |
|
|
if not raw: |
|
|
raise ValueError(MESSAGE_PARSE_EMPTY_ERROR) |
|
|
|
|
|
parts = raw.split(cls.SEPARATOR) |
|
|
if len(parts) != cls.EXPECTED_PARTS: |
|
|
raise ValueError(MESSAGE_PARSE_INVALID_ERROR) |
|
|
|
|
|
( |
|
|
message_id, |
|
|
topic_raw, |
|
|
order_in_topic, |
|
|
participant_raw, |
|
|
created_at_value, |
|
|
encoded_content, |
|
|
) = parts |
|
|
|
|
|
return Message( |
|
|
id=UUID(message_id), |
|
|
topic_id=Link.from_raw(topic_raw), |
|
|
order_in_topic=int(order_in_topic), |
|
|
participant_id=Link.from_raw(participant_raw), |
|
|
created_at=datetime.fromisoformat(created_at_value), |
|
|
content=cls._decode_content(encoded_content), |
|
|
) |
|
|
|
|
|
@classmethod |
|
|
def serialize(cls, message: Message) -> str: |
|
|
return cls.SEPARATOR.join( |
|
|
[ |
|
|
str(message.id), |
|
|
message.topic_id.as_path(), |
|
|
str(message.order_in_topic), |
|
|
message.participant_id.as_path(), |
|
|
message.created_at.isoformat(), |
|
|
cls._encode_content(message.content), |
|
|
] |
|
|
) |
|
|
|
|
|
|
