from dataclasses import dataclass
from typing import Optional, List

from .client import Client
from .room import Room
from .content import ContentBase


@dataclass
class EventBase:
    client: Client
    content: ContentBase
    type: str
    sender: str

    @classmethod
    def from_dict(cls, client: Client, event_dict: dict):
        from .content import content_dispatcher
        if event_dict['type'] == 'm.room.message':
            content_class = content_dispatcher[event_dict['content']['msgtype']]
        else:
            content_class = content_dispatcher[event_dict['type']]

        content_dict = event_dict['content']
        del event_dict['content']

        return cls(
            client=client,
            content=content_class(content_dict),
            **event_dict
        )


@dataclass
class UnsignedData:
    age: int
    redacted_because: Optional[EventBase] = None
    transaction_id: Optional[str] = None
    invite_room_state: Optional[List[EventBase]] = None


@dataclass
class RoomEvent(EventBase):
    event_id: str
    origin_server_ts: int
    unsigned: UnsignedData
    room: Room
    user_id: str


@dataclass
class StateEvent(RoomEvent):
    state_key: str
    age: int
    prev_content: Optional[EventBase] = None


@dataclass
class RedactionEvent(RoomEvent):
    redacts: EventBase


@dataclass
class MessageEvent(RoomEvent):
    pass


@dataclass
class PresenceEvent(EventBase):
    pass