241 lines
9.2 KiB
C++
241 lines
9.2 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2021
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
#include "td/telegram/MessageReplyInfo.h"
|
|
|
|
#include "td/telegram/ContactsManager.h"
|
|
#include "td/telegram/MessagesManager.h"
|
|
#include "td/telegram/ServerMessageId.h"
|
|
#include "td/telegram/Td.h"
|
|
|
|
#include "td/utils/algorithm.h"
|
|
#include "td/utils/logging.h"
|
|
#include "td/utils/misc.h"
|
|
|
|
namespace td {
|
|
|
|
MessageReplyInfo::MessageReplyInfo(Td *td, tl_object_ptr<telegram_api::messageReplies> &&reply_info, bool is_bot) {
|
|
if (reply_info == nullptr || is_bot || reply_info->channel_id_ == 777) {
|
|
return;
|
|
}
|
|
if (reply_info->replies_ < 0) {
|
|
LOG(ERROR) << "Receive wrong " << to_string(reply_info);
|
|
return;
|
|
}
|
|
reply_count = reply_info->replies_;
|
|
pts = reply_info->replies_pts_;
|
|
|
|
is_comment = reply_info->comments_;
|
|
|
|
if (is_comment) {
|
|
channel_id = ChannelId(reply_info->channel_id_);
|
|
if (!channel_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << channel_id;
|
|
channel_id = ChannelId();
|
|
is_comment = false;
|
|
}
|
|
}
|
|
|
|
if (is_comment) {
|
|
for (const auto &peer : reply_info->recent_repliers_) {
|
|
DialogId dialog_id(peer);
|
|
if (!dialog_id.is_valid()) {
|
|
LOG(ERROR) << "Receive " << dialog_id << " as a recent replier";
|
|
continue;
|
|
}
|
|
if (td::contains(recent_replier_dialog_ids, dialog_id)) {
|
|
LOG(ERROR) << "Receive duplicate " << dialog_id << " as a recent replier";
|
|
continue;
|
|
}
|
|
if (!td->messages_manager_->have_dialog_info(dialog_id)) {
|
|
auto dialog_type = dialog_id.get_type();
|
|
if (dialog_type == DialogType::User) {
|
|
auto replier_user_id = dialog_id.get_user_id();
|
|
if (!td->contacts_manager_->have_min_user(replier_user_id)) {
|
|
LOG(ERROR) << "Have no info about replied " << replier_user_id;
|
|
continue;
|
|
}
|
|
} else if (dialog_type == DialogType::Channel) {
|
|
auto replier_channel_id = dialog_id.get_channel_id();
|
|
auto min_channel = td->contacts_manager_->get_min_channel(replier_channel_id);
|
|
if (min_channel == nullptr) {
|
|
LOG(ERROR) << "Have no info about replied " << replier_channel_id;
|
|
continue;
|
|
}
|
|
replier_min_channels.emplace_back(replier_channel_id, *min_channel);
|
|
} else {
|
|
LOG(ERROR) << "Have no info about replied " << dialog_id;
|
|
continue;
|
|
}
|
|
}
|
|
recent_replier_dialog_ids.push_back(dialog_id);
|
|
if (recent_replier_dialog_ids.size() == MAX_RECENT_REPLIERS) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if ((reply_info->flags_ & telegram_api::messageReplies::MAX_ID_MASK) != 0 &&
|
|
ServerMessageId(reply_info->max_id_).is_valid()) {
|
|
max_message_id = MessageId(ServerMessageId(reply_info->max_id_));
|
|
}
|
|
if ((reply_info->flags_ & telegram_api::messageReplies::READ_MAX_ID_MASK) != 0 &&
|
|
ServerMessageId(reply_info->read_max_id_).is_valid()) {
|
|
last_read_inbox_message_id = MessageId(ServerMessageId(reply_info->read_max_id_));
|
|
}
|
|
if (last_read_inbox_message_id > max_message_id) { // possible if last thread message was deleted after it was read
|
|
max_message_id = last_read_inbox_message_id;
|
|
}
|
|
LOG(DEBUG) << "Parsed " << oneline(to_string(reply_info)) << " to " << *this;
|
|
}
|
|
|
|
bool MessageReplyInfo::need_update_to(const MessageReplyInfo &other) const {
|
|
if (other.is_empty() && !is_empty()) {
|
|
// ignore updates to empty reply info, because we will hide the info ourselves
|
|
// return true;
|
|
}
|
|
if (other.pts < pts) {
|
|
return false;
|
|
}
|
|
return reply_count != other.reply_count || recent_replier_dialog_ids != other.recent_replier_dialog_ids ||
|
|
replier_min_channels.size() != other.replier_min_channels.size() || is_comment != other.is_comment ||
|
|
channel_id != other.channel_id;
|
|
}
|
|
|
|
bool MessageReplyInfo::update_max_message_ids(const MessageReplyInfo &other) {
|
|
return update_max_message_ids(other.max_message_id, other.last_read_inbox_message_id,
|
|
other.last_read_outbox_message_id);
|
|
}
|
|
|
|
bool MessageReplyInfo::update_max_message_ids(MessageId other_max_message_id,
|
|
MessageId other_last_read_inbox_message_id,
|
|
MessageId other_last_read_outbox_message_id) {
|
|
bool result = false;
|
|
if (other_max_message_id > max_message_id) {
|
|
max_message_id = other_max_message_id;
|
|
result = true;
|
|
}
|
|
if (other_last_read_inbox_message_id > last_read_inbox_message_id) {
|
|
last_read_inbox_message_id = other_last_read_inbox_message_id;
|
|
result = true;
|
|
}
|
|
if (other_last_read_outbox_message_id > last_read_outbox_message_id) {
|
|
last_read_outbox_message_id = other_last_read_outbox_message_id;
|
|
result = true;
|
|
}
|
|
if (last_read_inbox_message_id > max_message_id) {
|
|
max_message_id = last_read_inbox_message_id;
|
|
result = true;
|
|
}
|
|
if (last_read_outbox_message_id > max_message_id) {
|
|
max_message_id = last_read_outbox_message_id;
|
|
result = true;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool MessageReplyInfo::add_reply(DialogId replier_dialog_id, MessageId reply_message_id, int diff) {
|
|
CHECK(!is_empty());
|
|
CHECK(diff == +1 || diff == -1);
|
|
|
|
if (diff == -1 && reply_count == 0) {
|
|
return false;
|
|
}
|
|
|
|
reply_count += diff;
|
|
if (is_comment && replier_dialog_id.is_valid()) {
|
|
if (replier_dialog_id.get_type() == DialogType::Channel) {
|
|
// the replier_dialog_id is never min, because it is the sender of a message
|
|
for (auto it = replier_min_channels.begin(); it != replier_min_channels.end(); ++it) {
|
|
if (it->first == replier_dialog_id.get_channel_id()) {
|
|
replier_min_channels.erase(it);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
td::remove(recent_replier_dialog_ids, replier_dialog_id);
|
|
if (diff > 0) {
|
|
recent_replier_dialog_ids.insert(recent_replier_dialog_ids.begin(), replier_dialog_id);
|
|
if (recent_replier_dialog_ids.size() > MAX_RECENT_REPLIERS) {
|
|
recent_replier_dialog_ids.pop_back();
|
|
}
|
|
} else {
|
|
auto max_repliers = static_cast<size_t>(reply_count);
|
|
if (recent_replier_dialog_ids.size() > max_repliers) {
|
|
recent_replier_dialog_ids.resize(max_repliers);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (diff > 0 && reply_message_id > max_message_id) {
|
|
max_message_id = reply_message_id;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool MessageReplyInfo::need_reget(const Td *td) const {
|
|
for (auto &dialog_id : recent_replier_dialog_ids) {
|
|
if (dialog_id.get_type() != DialogType::User && !td->messages_manager_->have_dialog_info(dialog_id)) {
|
|
if (dialog_id.get_type() == DialogType::Channel &&
|
|
td->contacts_manager_->have_min_channel(dialog_id.get_channel_id())) {
|
|
return false;
|
|
}
|
|
LOG(INFO) << "Reget a message because of replied " << dialog_id;
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
td_api::object_ptr<td_api::messageReplyInfo> MessageReplyInfo::get_message_reply_info_object(Td *td) const {
|
|
if (is_empty()) {
|
|
return nullptr;
|
|
}
|
|
|
|
vector<td_api::object_ptr<td_api::MessageSender>> recent_repliers;
|
|
for (auto dialog_id : recent_replier_dialog_ids) {
|
|
auto dialog_type = dialog_id.get_type();
|
|
if (dialog_type == DialogType::User) {
|
|
auto user_id = dialog_id.get_user_id();
|
|
if (td->contacts_manager_->have_min_user(user_id)) {
|
|
recent_repliers.push_back(td_api::make_object<td_api::messageSenderUser>(
|
|
td->contacts_manager_->get_user_id_object(user_id, "get_message_reply_info_object")));
|
|
} else {
|
|
LOG(ERROR) << "Skip unknown replied " << user_id;
|
|
}
|
|
} else {
|
|
if (!td->messages_manager_->have_dialog(dialog_id) &&
|
|
(td->messages_manager_->have_dialog_info(dialog_id) ||
|
|
(dialog_type == DialogType::Channel &&
|
|
td->contacts_manager_->have_min_channel(dialog_id.get_channel_id())))) {
|
|
LOG(INFO) << "Force creation of " << dialog_id;
|
|
td->messages_manager_->force_create_dialog(dialog_id, "get_message_reply_info_object", true);
|
|
}
|
|
if (td->messages_manager_->have_dialog(dialog_id)) {
|
|
recent_repliers.push_back(td_api::make_object<td_api::messageSenderChat>(dialog_id.get()));
|
|
} else {
|
|
LOG(ERROR) << "Skip unknown replied " << dialog_id;
|
|
}
|
|
}
|
|
}
|
|
return td_api::make_object<td_api::messageReplyInfo>(reply_count, std::move(recent_repliers),
|
|
last_read_inbox_message_id.get(),
|
|
last_read_outbox_message_id.get(), max_message_id.get());
|
|
}
|
|
|
|
StringBuilder &operator<<(StringBuilder &string_builder, const MessageReplyInfo &reply_info) {
|
|
if (reply_info.is_comment) {
|
|
return string_builder << reply_info.reply_count << " comments in " << reply_info.channel_id << " by "
|
|
<< reply_info.recent_replier_dialog_ids << " read up to "
|
|
<< reply_info.last_read_inbox_message_id << "/" << reply_info.last_read_outbox_message_id;
|
|
} else {
|
|
return string_builder << reply_info.reply_count << " replies read up to " << reply_info.last_read_inbox_message_id
|
|
<< "/" << reply_info.last_read_outbox_message_id;
|
|
}
|
|
}
|
|
|
|
} // namespace td
|