49 lines
1.0 KiB
Python
49 lines
1.0 KiB
Python
from datetime import datetime
|
|
from typing import Any, Literal
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.messages.models import MessageType
|
|
|
|
|
|
RealtimeEventName = Literal[
|
|
"connect",
|
|
"disconnect",
|
|
"send_message",
|
|
"receive_message",
|
|
"typing_start",
|
|
"typing_stop",
|
|
"message_read",
|
|
"message_delivered",
|
|
"error",
|
|
]
|
|
|
|
|
|
class SendMessagePayload(BaseModel):
|
|
chat_id: int
|
|
type: MessageType = MessageType.TEXT
|
|
text: str | None = Field(default=None, max_length=4096)
|
|
temp_id: str | None = None
|
|
|
|
|
|
class ChatEventPayload(BaseModel):
|
|
chat_id: int
|
|
|
|
|
|
class MessageStatusPayload(BaseModel):
|
|
chat_id: int
|
|
message_id: int
|
|
|
|
|
|
class IncomingRealtimeEvent(BaseModel):
|
|
event: Literal["send_message", "typing_start", "typing_stop", "message_read", "message_delivered"]
|
|
payload: dict[str, Any]
|
|
|
|
|
|
class OutgoingRealtimeEvent(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
event: RealtimeEventName
|
|
payload: dict[str, Any]
|
|
timestamp: datetime
|