30514fb7db
To make the models update correctly in QML: - ListModel and _QtModel merged - Return a ListModelMap QObject from properties instead of a DefaultDict → QVariantMap
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
# Copyright 2019 miruka
|
|
# This file is part of harmonyqml, licensed under GPLv3.
|
|
|
|
from PyQt5.QtCore import QObject
|
|
|
|
from .backend import Backend
|
|
from .client import Client
|
|
from .model.items import User, Room
|
|
|
|
|
|
class SignalManager(QObject):
|
|
def __init__(self, backend: Backend) -> None:
|
|
super().__init__()
|
|
self.backend = backend
|
|
|
|
cm = self.backend.clientManager
|
|
cm.clientAdded.connect(self.onClientAdded)
|
|
cm.clientDeleted.connect(self.onClientDeleted)
|
|
|
|
|
|
def onClientAdded(self, client: Client) -> None:
|
|
self.connectClient(client)
|
|
self.backend.models.accounts.append(User(
|
|
user_id = client.userID,
|
|
display_name = client.userID.lstrip("@").split(":")[0],
|
|
))
|
|
|
|
|
|
def onClientDeleted(self, user_id: str) -> None:
|
|
accs = self.backend.models.accounts
|
|
del accs[accs.indexWhere("user_id", user_id)]
|
|
|
|
|
|
def connectClient(self, client: Client) -> None:
|
|
for sig_name in ("roomInvited", "roomJoined", "roomLeft"):
|
|
sig = getattr(client, sig_name)
|
|
on_sig = getattr(self, f"on{sig_name[0].upper()}{sig_name[1:]}")
|
|
sig.connect(lambda room_id, o=on_sig, c=client: o(c, room_id))
|
|
|
|
|
|
def onRoomInvited(self, client: Client, room_id: str) -> None:
|
|
pass # TODO
|
|
|
|
|
|
def onRoomJoined(self, client: Client, room_id: str) -> None:
|
|
room = client.nio.rooms[room_id]
|
|
name = room.name or room.canonical_alias or room.group_name()
|
|
|
|
self.backend.models.rooms[client.userID].append(Room(
|
|
room_id = room_id,
|
|
display_name = name,
|
|
description = getattr(room, "topic", ""), # FIXME: outside init
|
|
))
|
|
|
|
|
|
def onRoomLeft(self, client: Client, room_id: str) -> None:
|
|
rooms = self.backend.models.rooms[client.userID]
|
|
del rooms[rooms.indexWhere("room_id", room_id)]
|