134 lines
4.2 KiB
Python
134 lines
4.2 KiB
Python
# Copyright 2019 miruka
|
|
# This file is part of harmonyqml, licensed under GPLv3.
|
|
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
from typing import Dict, Sequence, Set
|
|
|
|
from PyQt5.QtCore import QObject, pyqtProperty, pyqtSlot
|
|
|
|
from .html_filter import HtmlFilter
|
|
from .model.items import RoomEvent
|
|
from .model.qml_models import QMLModels
|
|
from .pyqt_future import futurize
|
|
|
|
|
|
class Backend(QObject):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self.pool: ThreadPoolExecutor = ThreadPoolExecutor(max_workers=6)
|
|
|
|
self._queried_displaynames: Dict[str, str] = {}
|
|
|
|
self.past_tokens: Dict[str, str] = {}
|
|
self.fully_loaded_rooms: Set[str] = set()
|
|
|
|
from .client_manager import ClientManager
|
|
self._client_manager: ClientManager = ClientManager(self)
|
|
self._models: QMLModels = QMLModels()
|
|
self._html_filter: HtmlFilter = HtmlFilter()
|
|
|
|
from .signal_manager import SignalManager
|
|
self._signal_manager: SignalManager = SignalManager(self)
|
|
|
|
self.clientManager.configLoad()
|
|
|
|
|
|
@pyqtProperty("QVariant", constant=True)
|
|
def clientManager(self):
|
|
return self._client_manager
|
|
|
|
@pyqtProperty("QVariant", constant=True)
|
|
def models(self):
|
|
return self._models
|
|
|
|
@pyqtProperty("QVariant", constant=True)
|
|
def htmlFilter(self):
|
|
return self._html_filter
|
|
|
|
|
|
@pyqtSlot(str, result="QVariant")
|
|
@pyqtSlot(str, bool, result="QVariant")
|
|
@futurize()
|
|
def getUserDisplayName(self, user_id: str, can_block: bool = True) -> str:
|
|
if user_id in self._queried_displaynames:
|
|
return self._queried_displaynames[user_id]
|
|
|
|
for client in self.clientManager.clients.values():
|
|
for room in client.nio.rooms.values():
|
|
displayname = room.user_name(user_id)
|
|
|
|
if displayname:
|
|
return displayname
|
|
|
|
return self._query_user_displayname(user_id) if can_block else user_id
|
|
|
|
|
|
def _query_user_displayname(self, user_id: str) -> str:
|
|
client = next(iter(self.clientManager.clients.values()))
|
|
response = client.net.talk(client.nio.get_displayname, user_id)
|
|
displayname = getattr(response, "displayname", "") or user_id
|
|
|
|
self._queried_displaynames[user_id] = displayname
|
|
return displayname
|
|
|
|
|
|
@pyqtSlot(str, result=float)
|
|
def hueFromString(self, string: str) -> float:
|
|
# pylint:disable=no-self-use
|
|
return sum((ord(char) * 99 for char in string)) % 360 / 360
|
|
|
|
|
|
@pyqtSlot(str)
|
|
@pyqtSlot(str, int)
|
|
def loadPastEvents(self, room_id: str, limit: int = 100) -> None:
|
|
if not room_id in self.past_tokens:
|
|
return # Initial sync not done yet
|
|
|
|
if room_id in self.fully_loaded_rooms:
|
|
return
|
|
|
|
for client in self.clientManager.clients.values():
|
|
if room_id in client.nio.rooms:
|
|
client.loadPastEvents(
|
|
room_id, self.past_tokens[room_id], limit
|
|
)
|
|
break
|
|
else:
|
|
raise ValueError(f"Room not found in any client: {room_id}")
|
|
|
|
|
|
@pyqtSlot()
|
|
@pyqtSlot(list)
|
|
def pdb(self, additional_data: Sequence = ()) -> None:
|
|
# pylint: disable=all
|
|
ad = additional_data
|
|
cm = self.clientManager
|
|
cl = self.clientManager.clients
|
|
m = self.models
|
|
|
|
tcl = lambda user: cl[f"@test_{user}:matrix.org"]
|
|
|
|
import pdb
|
|
from PyQt5.QtCore import pyqtRemoveInputHook
|
|
pyqtRemoveInputHook()
|
|
pdb.set_trace()
|
|
|
|
|
|
@pyqtSlot("QVariant", str, result=bool)
|
|
def EventIsOurProfileChanged(self, event: RoomEvent, account_id) -> bool:
|
|
# pylint: disable=unused-self
|
|
info = event.dict.get("content")
|
|
previous = event.dict.get("prev_content")
|
|
|
|
return (
|
|
event.type == "RoomMemberEvent" and
|
|
event.dict["sender"] == account_id and
|
|
bool(info) and
|
|
bool(previous) and
|
|
info["membership"] == previous["membership"] and
|
|
(
|
|
info.get("displayname") != previous.get("displayname") or
|
|
info.get("avatar_url") != previous.get("avatar_url")
|
|
)
|
|
)
|