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