moment/harmonyqml/backend/signal_manager.py

# Copyright 2019 miruka
# This file is part of harmonyqml, licensed under GPLv3.

from threading import Lock
from typing import Any, Deque, Dict, List, Optional, Sequence

from PyQt5.QtCore import QDateTime, QObject, pyqtBoundSignal

import nio
from nio.rooms import MatrixRoom

from .backend import Backend
from .client import Client
from .model.items import Room, RoomEvent, User

Inviter   = Optional[Dict[str, str]]
LeftEvent = Optional[Dict[str, str]]


class SignalManager(QObject):
    _lock: Lock = Lock()

    def __init__(self, backend: Backend) -> None:
        super().__init__(parent=backend)
        self.backend = backend

        self.last_room_events:    Deque[str] = Deque(maxlen=1000)
        self._events_in_transfer: int        = 0

        cm = self.backend.clientManager
        cm.clientAdded.connect(self.onClientAdded)
        cm.clientDeleted.connect(self.onClientDeleted)


    def onClientAdded(self, client: Client) -> None:
        print(client)
        self.connectClient(client)
        self.backend.models.accounts.append(User(
            userId      = client.userId,
            displayName = self.backend.getUserDisplayName(client.userId),
        ))


    def onClientDeleted(self, user_id: str) -> None:
        accs = self.backend.models.accounts
        del accs[accs.indexWhere("userId", user_id)]


    def connectClient(self, client: Client) -> None:
        for name in dir(client):
            attr = getattr(client, name)

            if isinstance(attr, pyqtBoundSignal):
                def onSignal(*args, name=name) -> None:
                    func = getattr(self, f"on{name[0].upper()}{name[1:]}")
                    func(client, *args)

                attr.connect(onSignal)


    def onRoomInvited(self,
                      client:  Client,
                      room_id: str,
                      inviter: Inviter = None) -> None:

        self._add_room(client, room_id, client.nio.invited_rooms[room_id],
                       "Invites", inviter=inviter)


    def onRoomJoined(self, client: Client, room_id: str) -> None:
        self._add_room(client, room_id, client.nio.rooms[room_id], "Rooms")


    def onRoomLeft(self,
                   client:     Client,
                   room_id:    str,
                   left_event: LeftEvent = None) -> None:

        self._add_room(client, room_id, client.nio.rooms.get(room_id), "Left",
                       left_event=left_event)


    def _move_room(self, account_id: str, room_id: str) -> None:
        def get_newest_event_date_time(room_id: str) -> QDateTime:
            for ev in self.backend.models.roomEvents[room_id]:
                if not self.backend.EventIsOurProfileChanged(ev, account_id):
                    return ev.dateTime

            return QDateTime.fromMSecsSinceEpoch(0)

        rooms_model = self.backend.models.rooms[account_id]
        room_index  = rooms_model.indexWhere("roomId", room_id)
        category    = rooms_model[room_index].category
        timestamp   = get_newest_event_date_time(room_id)

        def get_index(put_before_categories: Sequence[str],
                      put_after_categories: Sequence[str]) -> int:
            for i, room in enumerate(rooms_model):
                if room.category not in put_after_categories and \
                   (room.category in put_before_categories or
                    timestamp >= get_newest_event_date_time(room.roomId)):
                    return i

            return len(rooms_model) - 1

        to = 0

        if category == "Invites":
            to = get_index(["Rooms", "Left"], [])

        if category == "Rooms":
            to = get_index(["Left"], ["Invites"])

        elif category == "Left":
            to = get_index([], ["Invites", "Rooms", "Left"])

        rooms_model.move(room_index, to)


    def _add_room(self,
                  client:     Client,
                  room_id:    str,
                  room:       MatrixRoom,
                  category:   str,
                  inviter:    Inviter   = None,
                  left_event: LeftEvent = None) -> None:

        if (inviter and left_event):
            raise ValueError()

        model     = self.backend.models.rooms[client.userId]
        no_update = []

        def get_displayname() -> Optional[str]:
            if not room:
                no_update.append("displayName")
                return room_id

            name = room.name or room.canonical_alias
            if name:
                return name

            name = room.group_name()
            return None if name == "Empty room?" else name

        item = Room(
            roomId      = room_id,
            category    = category,
            displayName = get_displayname(),
            topic       = room.topic if room else "",
            inviter     = inviter,
            leftEvent   = left_event,
            no_update   = no_update,
        )

        model.updateOrAppendWhere("roomId", room_id, item)
        with self._lock:
            self._move_room(client.userId, room_id)


    def onRoomSyncPrevBatchTokenReceived(self,
                                         _:       Client,
                                         room_id: str,
                                         token:   str) -> None:

        if room_id not in self.backend.past_tokens:
            self.backend.past_tokens[room_id] = token


    def onRoomPastPrevBatchTokenReceived(self,
                                         _:       Client,
                                         room_id: str,
                                         token:   str) -> None:

        if self.backend.past_tokens[room_id] == token:
            self.backend.fully_loaded_rooms.add(room_id)

        self.backend.past_tokens[room_id] = token


    def onRoomEventReceived(self,
                            client:  Client,
                            room_id: str,
                            etype:   str,
                            edict:   Dict[str, Any]) -> None:
        def process() -> None:
            # Prevent duplicate events in models due to multiple accounts
            if edict["event_id"] in self.last_room_events:
                return

            self.last_room_events.appendleft(edict["event_id"])

            model     = self.backend.models.roomEvents[room_id]
            date_time = QDateTime\
                        .fromMSecsSinceEpoch(edict["server_timestamp"])
            new_event = RoomEvent(type=etype, dateTime=date_time, dict=edict)

            if etype == "RoomCreateEvent":
                print(room_id, "ff")
                self.backend.fully_loaded_rooms.add(room_id)

            if self._events_in_transfer:
                local_echoes_met: int           = 0
                update_at:       Optional[int] = None

                # Find if any locally echoed event corresponds to new_event
                for i, event in enumerate(model):
                    if not event.isLocalEcho:
                        continue

                    sb     = (event.dict["sender"], event.dict["body"])
                    new_sb = (new_event.dict["sender"], new_event.dict["body"])

                    if sb == new_sb:
                        # The oldest matching local echo shall be replaced
                        update_at = max(update_at or 0, i)

                    local_echoes_met += 1
                    if local_echoes_met >= self._events_in_transfer:
                        break

                if update_at is not None:
                    model.update(update_at, new_event)
                    self._events_in_transfer -= 1
                    return

            for i, event in enumerate(model):
                if event.isLocalEcho:
                    continue

                # Model is sorted from newest to oldest message
                if new_event.dateTime > event.dateTime:
                    model.insert(i, new_event)
                    return

            model.append(new_event)

        with self._lock:
            process()
            self._move_room(client.userId, room_id)


    def onRoomTypingUsersUpdated(self,
                                 client:  Client,
                                 room_id: str,
                                 users:   List[str]) -> None:

        rooms = self.backend.models.rooms[client.userId]
        rooms[rooms.indexWhere("roomId", room_id)].typingUsers = users


    def onMessageAboutToBeSent(self,
                               client:  Client,
                               room_id: str,
                               content: Dict[str, str]) -> None:

        with self._lock:
            timestamp = QDateTime.currentMSecsSinceEpoch()
            model     = self.backend.models.roomEvents[room_id]
            nio_event = nio.events.RoomMessage.parse_event({
                "event_id":         "",
                "sender":           client.userId,
                "origin_server_ts": timestamp,
                "content":          content,
            })
            event = RoomEvent(
                type        = type(nio_event).__name__,
                dateTime    = QDateTime.fromMSecsSinceEpoch(timestamp),
                dict        = nio_event.__dict__,
                isLocalEcho = True,
            )
            model.insert(0, event)
            self._events_in_transfer += 1

            self._move_room(client.userId, room_id)


    def onRoomAboutToBeForgotten(self, client: Client, room_id: str) -> None:
        with self._lock:
            rooms = self.backend.models.rooms[client.userId]
            del rooms[rooms.indexWhere("roomId", room_id)]

            self.backend.models.roomEvents[room_id].clear()