/* Copyright (C) 2020 Sebastian J. Wolf and other contributors This file is part of Fernschreiber. Fernschreiber is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. Fernschreiber is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with Fernschreiber. If not, see . */ #include "tdlibreceiver.h" #define DEBUG_MODULE TDLibReceiver #include "debuglog.h" namespace { const QString ID("id"); const QString LIST("list"); const QString CHAT_ID("chat_id"); const QString USER_ID("user_id"); const QString MESSAGE_ID("message_id"); const QString MESSAGE_IDS("message_ids"); const QString MESSAGE("message"); const QString MESSAGES("messages"); const QString TITLE("title"); const QString NAME("name"); const QString VALUE("value"); const QString POSITION("position"); const QString POSITIONS("positions"); const QString PHOTO("photo"); const QString ORDER("order"); const QString BASIC_GROUP("basic_group"); const QString SUPERGROUP("supergroup"); const QString LAST_MESSAGE("last_message"); const QString TOTAL_COUNT("total_count"); const QString UNREAD_COUNT("unread_count"); const QString LAST_READ_INBOX_MESSAGE_ID("last_read_inbox_message_id"); const QString LAST_READ_OUTBOX_MESSAGE_ID("last_read_outbox_message_id"); const QString SECRET_CHAT("secret_chat"); const QString TYPE("@type"); const QString EXTRA("@extra"); const QString TYPE_CHAT_POSITION("chatPosition"); const QString TYPE_CHAT_LIST_MAIN("chatListMain"); } static QString getChatPositionOrder(const QVariantMap &position) { if (position.value(TYPE).toString() == TYPE_CHAT_POSITION && position.value(LIST).toMap().value(TYPE) == TYPE_CHAT_LIST_MAIN) { return position.value(ORDER).toString(); } return QString(); } static QString findChatPositionOrder(const QVariantList &positions) { const int n = positions.count(); for (int i = 0; i < n; i++) { const QString order(getChatPositionOrder(positions.at(i).toMap())); if (!order.isEmpty()) { return order; } } return QString(); } TDLibReceiver::TDLibReceiver(void *tdLibClient, QObject *parent) : QThread(parent) { this->tdLibClient = tdLibClient; this->isActive = true; handlers.insert("updateOption", &TDLibReceiver::processUpdateOption); handlers.insert("updateAuthorizationState", &TDLibReceiver::processUpdateAuthorizationState); handlers.insert("updateConnectionState", &TDLibReceiver::processUpdateConnectionState); handlers.insert("updateUser", &TDLibReceiver::processUpdateUser); handlers.insert("updateUserStatus", &TDLibReceiver::processUpdateUserStatus); handlers.insert("updateFile", &TDLibReceiver::processUpdateFile); handlers.insert("file", &TDLibReceiver::processFile); handlers.insert("updateNewChat", &TDLibReceiver::processUpdateNewChat); handlers.insert("updateUnreadMessageCount", &TDLibReceiver::processUpdateUnreadMessageCount); handlers.insert("updateUnreadChatCount", &TDLibReceiver::processUpdateUnreadChatCount); handlers.insert("updateChatLastMessage", &TDLibReceiver::processUpdateChatLastMessage); handlers.insert("updateChatOrder", &TDLibReceiver::processUpdateChatOrder); handlers.insert("updateChatPosition", &TDLibReceiver::processUpdateChatPosition); handlers.insert("updateChatReadInbox", &TDLibReceiver::processUpdateChatReadInbox); handlers.insert("updateChatReadOutbox", &TDLibReceiver::processUpdateChatReadOutbox); handlers.insert("updateBasicGroup", &TDLibReceiver::processUpdateBasicGroup); handlers.insert("updateSupergroup", &TDLibReceiver::processUpdateSuperGroup); handlers.insert("updateChatOnlineMemberCount", &TDLibReceiver::processChatOnlineMemberCountUpdated); handlers.insert("messages", &TDLibReceiver::processMessages); handlers.insert("updateNewMessage", &TDLibReceiver::processUpdateNewMessage); handlers.insert("message", &TDLibReceiver::processMessage); handlers.insert("updateMessageSendSucceeded", &TDLibReceiver::processMessageSendSucceeded); handlers.insert("updateActiveNotifications", &TDLibReceiver::processUpdateActiveNotifications); handlers.insert("updateNotificationGroup", &TDLibReceiver::processUpdateNotificationGroup); handlers.insert("updateChatNotificationSettings", &TDLibReceiver::processUpdateChatNotificationSettings); handlers.insert("updateMessageContent", &TDLibReceiver::processUpdateMessageContent); handlers.insert("updateDeleteMessages", &TDLibReceiver::processUpdateDeleteMessages); handlers.insert("chats", &TDLibReceiver::processChats); handlers.insert("chat", &TDLibReceiver::processChat); handlers.insert("updateRecentStickers", &TDLibReceiver::processUpdateRecentStickers); handlers.insert("stickers", &TDLibReceiver::processStickers); handlers.insert("updateInstalledStickerSets", &TDLibReceiver::processUpdateInstalledStickerSets); handlers.insert("stickerSets", &TDLibReceiver::processStickerSets); handlers.insert("stickerSet", &TDLibReceiver::processStickerSet); handlers.insert("chatMembers", &TDLibReceiver::processChatMembers); handlers.insert("userFullInfo", &TDLibReceiver::processUserFullInfo); handlers.insert("updateUserFullInfo", &TDLibReceiver::processUpdateUserFullInfo); handlers.insert("basicGroupFullInfo", &TDLibReceiver::processBasicGroupFullInfo); handlers.insert("updateBasicGroupFullInfo", &TDLibReceiver::processUpdateBasicGroupFullInfo); handlers.insert("supergroupFullInfo", &TDLibReceiver::processSupergroupFullInfo); handlers.insert("updateSupergroupFullInfo", &TDLibReceiver::processUpdateSupergroupFullInfo); handlers.insert("userProfilePhotos", &TDLibReceiver::processUserProfilePhotos); handlers.insert("updateChatPermissions", &TDLibReceiver::processUpdateChatPermissions); handlers.insert("updateChatPhoto", &TDLibReceiver::processUpdateChatPhoto); handlers.insert("updateChatTitle", &TDLibReceiver::processUpdateChatTitle); handlers.insert("updateChatPinnedMessage", &TDLibReceiver::processUpdateChatPinnedMessage); handlers.insert("users", &TDLibReceiver::processUsers); handlers.insert("error", &TDLibReceiver::processError); handlers.insert("ok", &TDLibReceiver::nop); handlers.insert("secretChat", &TDLibReceiver::processSecretChat); handlers.insert("updateSecretChat", &TDLibReceiver::processUpdateSecretChat); } void TDLibReceiver::setActive(bool active) { if (active) { LOG("Activating receiver loop..."); } else { LOG("Deactivating receiver loop, this may take a while..."); } this->isActive = active; } void TDLibReceiver::receiverLoop() { LOG("Starting receiver loop"); const double WAIT_TIMEOUT = 5.0; while (this->isActive) { const char *result = td_json_client_receive(this->tdLibClient, WAIT_TIMEOUT); if (result) { QJsonDocument receivedJsonDocument = QJsonDocument::fromJson(QByteArray(result)); VERBOSE("Raw result:" << receivedJsonDocument.toJson(QJsonDocument::Indented).constData()); processReceivedDocument(receivedJsonDocument); } } LOG("Stopping receiver loop"); } void TDLibReceiver::processReceivedDocument(const QJsonDocument &receivedJsonDocument) { QVariantMap receivedInformation = receivedJsonDocument.object().toVariantMap(); QString objectTypeName = receivedInformation.value(TYPE).toString(); Handler handler = handlers.value(objectTypeName); if (handler) { (this->*handler)(receivedInformation); } else { LOG("Unhandled object type" << objectTypeName); } } void TDLibReceiver::processUpdateOption(const QVariantMap &receivedInformation) { const QString currentOption = receivedInformation.value(NAME).toString(); const QVariant value = receivedInformation.value(VALUE).toMap().value(VALUE); if (currentOption == "version") { QString detectedVersion = value.toString(); LOG("TD Lib version detected: " << detectedVersion); emit versionDetected(detectedVersion); } else { LOG("Option updated: " << currentOption << value); emit optionUpdated(currentOption, value); } } void TDLibReceiver::processUpdateAuthorizationState(const QVariantMap &receivedInformation) { QString authorizationState = receivedInformation.value("authorization_state").toMap().value(TYPE).toString(); LOG("Authorization state changed: " << authorizationState); emit authorizationStateChanged(authorizationState, receivedInformation); } void TDLibReceiver::processUpdateConnectionState(const QVariantMap &receivedInformation) { QString connectionState = receivedInformation.value("state").toMap().value(TYPE).toString(); LOG("Connection state changed: " << connectionState); emit connectionStateChanged(connectionState); } void TDLibReceiver::processUpdateUser(const QVariantMap &receivedInformation) { QVariantMap userInformation = receivedInformation.value("user").toMap(); VERBOSE("User was updated: " << userInformation.value("username").toString() << userInformation.value("first_name").toString() << userInformation.value("last_name").toString()); emit userUpdated(userInformation); } void TDLibReceiver::processUpdateUserStatus(const QVariantMap &receivedInformation) { const QString userId = receivedInformation.value(USER_ID).toString(); QVariantMap userStatusInformation = receivedInformation.value("status").toMap(); VERBOSE("User status was updated: " << receivedInformation.value(USER_ID).toString() << userStatusInformation.value(TYPE).toString()); emit userStatusUpdated(userId, userStatusInformation); } void TDLibReceiver::processUpdateFile(const QVariantMap &receivedInformation) { const QVariantMap fileInformation = receivedInformation.value("file").toMap(); LOG("File was updated: " << fileInformation.value(ID).toString()); emit fileUpdated(fileInformation); } void TDLibReceiver::processFile(const QVariantMap &receivedInformation) { LOG("File was updated: " << receivedInformation.value(ID).toString()); emit fileUpdated(receivedInformation); } void TDLibReceiver::processUpdateNewChat(const QVariantMap &receivedInformation) { const QVariantMap chatInformation = receivedInformation.value("chat").toMap(); LOG("New chat discovered: " << chatInformation.value(ID).toString() << chatInformation.value(TITLE).toString()); emit newChatDiscovered(chatInformation); } void TDLibReceiver::processUpdateUnreadMessageCount(const QVariantMap &receivedInformation) { QVariantMap messageCountInformation; messageCountInformation.insert("chat_list_type", receivedInformation.value("chat_list").toMap().value(TYPE)); messageCountInformation.insert(UNREAD_COUNT, receivedInformation.value(UNREAD_COUNT)); messageCountInformation.insert("unread_unmuted_count", receivedInformation.value("unread_unmuted_count")); LOG("Unread message count updated: " << messageCountInformation.value("chat_list_type").toString() << messageCountInformation.value(UNREAD_COUNT).toString()); emit unreadMessageCountUpdated(messageCountInformation); } void TDLibReceiver::processUpdateUnreadChatCount(const QVariantMap &receivedInformation) { QVariantMap chatCountInformation; chatCountInformation.insert("chat_list_type", receivedInformation.value("chat_list").toMap().value(TYPE)); chatCountInformation.insert("marked_as_unread_count", receivedInformation.value("marked_as_unread_count")); chatCountInformation.insert("marked_as_unread_unmuted_count", receivedInformation.value("marked_as_unread_unmuted_count")); chatCountInformation.insert(TOTAL_COUNT, receivedInformation.value(TOTAL_COUNT)); chatCountInformation.insert(UNREAD_COUNT, receivedInformation.value(UNREAD_COUNT)); chatCountInformation.insert("unread_unmuted_count", receivedInformation.value("unread_unmuted_count")); LOG("Unread chat count updated: " << chatCountInformation.value("chat_list_type").toString() << chatCountInformation.value(UNREAD_COUNT).toString()); emit unreadChatCountUpdated(chatCountInformation); } void TDLibReceiver::processUpdateChatLastMessage(const QVariantMap &receivedInformation) { const QString chat_id(receivedInformation.value(CHAT_ID).toString()); QString order; if (receivedInformation.contains(POSITIONS)) { order = findChatPositionOrder(receivedInformation.value(POSITIONS).toList()); } else { order = receivedInformation.value(ORDER).toString(); } const QVariantMap lastMessage = receivedInformation.value(LAST_MESSAGE).toMap(); LOG("Last message of chat" << chat_id << "updated, order" << order << "type" << lastMessage.value(TYPE).toString()); emit chatLastMessageUpdated(chat_id, order, lastMessage); } void TDLibReceiver::processUpdateChatOrder(const QVariantMap &receivedInformation) { const QString chat_id(receivedInformation.value(CHAT_ID).toString()); const QString order(receivedInformation.value(ORDER).toString()); LOG("Chat order updated for ID" << chat_id << "to" << order); emit chatOrderUpdated(chat_id, order); } void TDLibReceiver::processUpdateChatPosition(const QVariantMap &receivedInformation) { const QString chat_id(receivedInformation.value(CHAT_ID).toString()); const QString order(receivedInformation.value(POSITION).toMap().value(ORDER).toString()); LOG("Chat position updated for ID" << chat_id << "new order" << order); emit chatOrderUpdated(chat_id, order); } void TDLibReceiver::processUpdateChatReadInbox(const QVariantMap &receivedInformation) { const QString chat_id(receivedInformation.value(CHAT_ID).toString()); const QString unread_count(receivedInformation.value(UNREAD_COUNT).toString()); LOG("Chat read information updated for" << chat_id << "unread count:" << unread_count); emit chatReadInboxUpdated(chat_id, receivedInformation.value(LAST_READ_INBOX_MESSAGE_ID).toString(), unread_count.toInt()); } void TDLibReceiver::processUpdateChatReadOutbox(const QVariantMap &receivedInformation) { const QString chat_id(receivedInformation.value(CHAT_ID).toString()); const QString last_read_outbox_message_id(receivedInformation.value(LAST_READ_OUTBOX_MESSAGE_ID).toString()); LOG("Sent messages read information updated for" << chat_id << "last read message ID:" << last_read_outbox_message_id); emit chatReadOutboxUpdated(chat_id, last_read_outbox_message_id); } void TDLibReceiver::processUpdateBasicGroup(const QVariantMap &receivedInformation) { const QVariantMap basicGroup(receivedInformation.value(BASIC_GROUP).toMap()); const qlonglong basicGroupId = basicGroup.value(ID).toLongLong(); LOG("Basic group information updated for " << basicGroupId); emit basicGroupUpdated(basicGroupId, basicGroup); } void TDLibReceiver::processUpdateSuperGroup(const QVariantMap &receivedInformation) { const QVariantMap supergroup(receivedInformation.value(SUPERGROUP).toMap()); const qlonglong superGroupId = supergroup.value(ID).toLongLong(); LOG("Super group information updated for " << superGroupId); emit superGroupUpdated(superGroupId, supergroup); } void TDLibReceiver::processChatOnlineMemberCountUpdated(const QVariantMap &receivedInformation) { const QString chatId = receivedInformation.value(CHAT_ID).toString(); LOG("Online member count updated for chat " << chatId); emit chatOnlineMemberCountUpdated(chatId, receivedInformation.value("online_member_count").toInt()); } void TDLibReceiver::processMessages(const QVariantMap &receivedInformation) { LOG("Received new messages, amount: " << receivedInformation.value(TOTAL_COUNT).toString()); emit messagesReceived(receivedInformation.value(MESSAGES).toList(), receivedInformation.value(TOTAL_COUNT).toInt()); } void TDLibReceiver::processUpdateNewMessage(const QVariantMap &receivedInformation) { const QVariantMap message = receivedInformation.value(MESSAGE).toMap(); const QString chatId = message.value(CHAT_ID).toString(); LOG("Received new message for chat " << chatId); emit newMessageReceived(chatId, message); } void TDLibReceiver::processMessage(const QVariantMap &receivedInformation) { const QString chatId = receivedInformation.value(CHAT_ID).toString(); const QString messageId = receivedInformation.value(ID).toString(); LOG("Received message " << chatId << messageId); emit messageInformation(messageId, receivedInformation); } void TDLibReceiver::processMessageSendSucceeded(const QVariantMap &receivedInformation) { const QString oldMessageId = receivedInformation.value("old_message_id").toString(); const QVariantMap message = receivedInformation.value(MESSAGE).toMap(); const QString messageId = message.value(ID).toString(); LOG("Message send succeeded " << messageId << oldMessageId); emit messageSendSucceeded(messageId, oldMessageId, message); } void TDLibReceiver::processUpdateActiveNotifications(const QVariantMap &receivedInformation) { LOG("Received active notification groups"); emit activeNotificationsUpdated(receivedInformation.value("groups").toList()); } void TDLibReceiver::processUpdateNotificationGroup(const QVariantMap &receivedInformation) { LOG("Received updated notification group"); emit notificationGroupUpdated(receivedInformation); } void TDLibReceiver::processUpdateNotification(const QVariantMap &receivedInformation) { LOG("Received notification update"); emit notificationUpdated(receivedInformation); } void TDLibReceiver::processUpdateChatNotificationSettings(const QVariantMap &receivedInformation) { const QString chatId = receivedInformation.value(CHAT_ID).toString(); LOG("Received new notification settings for chat " << chatId); emit chatNotificationSettingsUpdated(chatId, receivedInformation.value("notification_settings").toMap()); } void TDLibReceiver::processUpdateMessageContent(const QVariantMap &receivedInformation) { const QString chatId = receivedInformation.value(CHAT_ID).toString(); const QString messageId = receivedInformation.value(MESSAGE_ID).toString(); LOG("Message content updated " << chatId << messageId); emit messageContentUpdated(chatId, messageId, receivedInformation.value("new_content").toMap()); } void TDLibReceiver::processUpdateDeleteMessages(const QVariantMap &receivedInformation) { const QString chatId = receivedInformation.value(CHAT_ID).toString(); const QVariantList messageIds = receivedInformation.value(MESSAGE_IDS).toList(); LOG("Some messages were deleted " << chatId); emit messagesDeleted(chatId, messageIds); } void TDLibReceiver::processChats(const QVariantMap &receivedInformation) { emit chats(receivedInformation); } void TDLibReceiver::processChat(const QVariantMap &receivedInformation) { emit chat(receivedInformation); } void TDLibReceiver::processUpdateRecentStickers(const QVariantMap &receivedInformation) { LOG("Recent stickers updated"); emit recentStickersUpdated(receivedInformation.value("sticker_ids").toList()); } void TDLibReceiver::processStickers(const QVariantMap &receivedInformation) { LOG("Received some stickers..."); emit stickers(receivedInformation.value("stickers").toList()); } void TDLibReceiver::processUpdateInstalledStickerSets(const QVariantMap &receivedInformation) { LOG("Recent sticker sets updated"); emit installedStickerSetsUpdated(receivedInformation.value("sticker_set_ids").toList()); } void TDLibReceiver::processStickerSets(const QVariantMap &receivedInformation) { LOG("Received some sticker sets..."); emit stickerSets(receivedInformation.value("sets").toList()); } void TDLibReceiver::processStickerSet(const QVariantMap &receivedInformation) { LOG("Received a sticker set..."); emit stickerSet(receivedInformation); } void TDLibReceiver::processChatMembers(const QVariantMap &receivedInformation) { LOG("Received super group members"); const QString extra = receivedInformation.value(EXTRA).toString(); emit chatMembers(extra, receivedInformation.value("members").toList(), receivedInformation.value(TOTAL_COUNT).toInt()); } void TDLibReceiver::processUserFullInfo(const QVariantMap &receivedInformation) { LOG("Received UserFullInfo"); emit userFullInfo(receivedInformation); } void TDLibReceiver::processUpdateUserFullInfo(const QVariantMap &receivedInformation) { LOG("Received UserFullInfoUpdate"); emit userFullInfoUpdated(receivedInformation.value(USER_ID).toString(), receivedInformation.value("user_full_info").toMap()); } void TDLibReceiver::processBasicGroupFullInfo(const QVariantMap &receivedInformation) { LOG("Received BasicGroupFullInfo"); const QString groupId = receivedInformation.value(EXTRA).toString(); emit basicGroupFullInfo(groupId, receivedInformation); } void TDLibReceiver::processUpdateBasicGroupFullInfo(const QVariantMap &receivedInformation) { LOG("Received BasicGroupFullInfoUpdate"); const QString groupId = receivedInformation.value("basic_group_id").toString(); emit basicGroupFullInfoUpdated(groupId, receivedInformation.value("basic_group_full_info").toMap()); } void TDLibReceiver::processSupergroupFullInfo(const QVariantMap &receivedInformation) { LOG("Received SuperGroupFullInfoUpdate"); const QString groupId = receivedInformation.value(EXTRA).toString(); emit supergroupFullInfo(groupId, receivedInformation); } void TDLibReceiver::processUpdateSupergroupFullInfo(const QVariantMap &receivedInformation) { LOG("Received SuperGroupFullInfoUpdate"); const QString groupId = receivedInformation.value("supergroup_id").toString(); emit supergroupFullInfoUpdated(groupId, receivedInformation.value("supergroup_full_info").toMap()); } void TDLibReceiver::processUserProfilePhotos(const QVariantMap &receivedInformation) { const QString extra = receivedInformation.value(EXTRA).toString(); emit userProfilePhotos(extra, receivedInformation.value("photos").toList(), receivedInformation.value(TOTAL_COUNT).toInt()); } void TDLibReceiver::processUpdateChatPermissions(const QVariantMap &receivedInformation) { emit chatPermissionsUpdated(receivedInformation.value(CHAT_ID).toString(), receivedInformation.value("permissions").toMap()); } void TDLibReceiver::processUpdateChatPhoto(const QVariantMap &receivedInformation) { const qlonglong chatId = receivedInformation.value(CHAT_ID).toLongLong(); LOG("Photo updated for chat" << chatId); emit chatPhotoUpdated(chatId, receivedInformation.value(PHOTO).toMap()); } void TDLibReceiver::processUpdateChatTitle(const QVariantMap &receivedInformation) { LOG("Received UpdateChatTitle"); emit chatTitleUpdated(receivedInformation.value(CHAT_ID).toString(), receivedInformation.value(TITLE).toString()); } void TDLibReceiver::processUpdateChatPinnedMessage(const QVariantMap &receivedInformation) { LOG("Received UpdateChatPinnedMessage"); emit chatPinnedMessageUpdated(receivedInformation.value(CHAT_ID).toLongLong(), receivedInformation.value("pinned_message_id").toLongLong()); } void TDLibReceiver::processUsers(const QVariantMap &receivedInformation) { LOG("Received Users"); emit usersReceived(receivedInformation.value(EXTRA).toString(), receivedInformation.value("user_ids").toList(), receivedInformation.value(TOTAL_COUNT).toInt()); } void TDLibReceiver::processError(const QVariantMap &receivedInformation) { LOG("Received an error"); emit errorReceived(receivedInformation.value("code").toInt(), receivedInformation.value(MESSAGE).toString()); } void TDLibReceiver::nop(const QVariantMap &) { } void TDLibReceiver::processSecretChat(const QVariantMap &receivedInformation) { LOG("Received a secret chat"); emit secretChat(receivedInformation.value(ID).toString(), receivedInformation); } void TDLibReceiver::processUpdateSecretChat(const QVariantMap &receivedInformation) { LOG("A secret chat was updated"); QVariantMap updatedSecretChat = receivedInformation.value(SECRET_CHAT).toMap(); emit secretChatUpdated(updatedSecretChat.value(ID).toString(), updatedSecretChat); }