andreacavalli/tdlib-fork
Archived
1
0
Fork 0
Code Pull Requests Releases Activity

Do not add sponsored chat to ordered_dialogs_.

Browse Source 
GitOrigin-RevId: b0b9f90f2018bbe4f32597cd5b4077056ac901e7
This commit is contained in:
levlam 2020-03-20 01:15:15 +03:00
parent 3caaddc98d
commit 61d0b113d9
2 changed files with 136 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

207
td/telegram/MessagesManager.cpp
View File

@ -9194,13 +9194,7 @@ void MessagesManager::delete_all_dialog_messages(Dialog *d, bool remove_from_dia
send_update_chat_last_message(d, "delete_all_dialog_messages"); send_update_chat_last_message(d, "delete_all_dialog_messages");
} }
if (remove_from_dialog_list) { if (remove_from_dialog_list) {
if (d->pinned_order != DEFAULT_ORDER) { set_dialog_order(d, DEFAULT_ORDER, true, false, "delete_all_dialog_messages 1");
set_dialog_is_pinned(d, false);
}
int64 new_order = d->dialog_id == sponsored_dialog_id_ && d->folder_id == FolderId::main() ? SPONSORED_DIALOG_ORDER
: DEFAULT_ORDER;
set_dialog_order(d, new_order, true, false, "delete_all_dialog_messages 1");
} else { } else {
update_dialog_pos(d, "delete_all_dialog_messages 2"); update_dialog_pos(d, "delete_all_dialog_messages 2");
} }
@ -9643,15 +9637,24 @@ bool MessagesManager::need_unread_counter(int64 dialog_order) {
return dialog_order != DEFAULT_ORDER; return dialog_order != DEFAULT_ORDER;
} }
int32 MessagesManager::get_dialog_total_count(const DialogList &list) { int32 MessagesManager::get_dialog_total_count(const DialogList &list) const {
int32 sponsored_dialog_count = 0;
if (sponsored_dialog_id_.is_valid() && list.folder_id == FolderId::main()) {
auto d = get_dialog(sponsored_dialog_id_);
CHECK(d != nullptr);
if (is_dialog_sponsored(d)) {
sponsored_dialog_count = 1;
}
}
if (list.server_dialog_total_count_ != -1 && list.secret_chat_total_count_ != -1) { if (list.server_dialog_total_count_ != -1 && list.secret_chat_total_count_ != -1) {
return std::max(list.server_dialog_total_count_ + list.secret_chat_total_count_, return std::max(list.server_dialog_total_count_ + list.secret_chat_total_count_,
list.in_memory_dialog_total_count_); list.in_memory_dialog_total_count_) +
sponsored_dialog_count;
} }
if (list.last_dialog_date_ == MAX_DIALOG_DATE) { if (list.last_dialog_date_ == MAX_DIALOG_DATE) {
return list.in_memory_dialog_total_count_; return list.in_memory_dialog_total_count_ + sponsored_dialog_count;
} }
return list.in_memory_dialog_total_count_ + 1; return list.in_memory_dialog_total_count_ + sponsored_dialog_count + 1;
} }
void MessagesManager::repair_server_dialog_total_count(FolderId folder_id) { void MessagesManager::repair_server_dialog_total_count(FolderId folder_id) {
@ -9836,8 +9839,7 @@ void MessagesManager::set_dialog_last_read_inbox_message_id(Dialog *d, MessageId
} }
send_update_unread_chat_count(d->folder_id, d->dialog_id, force_update, source); send_update_unread_chat_count(d->folder_id, d->dialog_id, force_update, source);
} }
if (message_id != MessageId::min() && d->last_read_inbox_message_id.is_valid() && d->order != DEFAULT_ORDER && if (message_id != MessageId::min() && d->last_read_inbox_message_id.is_valid() && d->order != DEFAULT_ORDER) {
d->order != SPONSORED_DIALOG_ORDER) {
VLOG(notifications) << "Remove some notifications in " << d->dialog_id VLOG(notifications) << "Remove some notifications in " << d->dialog_id
<< " after updating last read inbox message to " << message_id << " after updating last read inbox message to " << message_id
<< " and unread message count to " << server_unread_count << " + " << local_unread_count << " and unread message count to " << server_unread_count << " + " << local_unread_count
@ -10344,11 +10346,16 @@ void MessagesManager::init() {
if (r_dialog_id.is_error()) { if (r_dialog_id.is_error()) {
LOG(ERROR) << "Can't parse " << sponsored_dialog_id_string; LOG(ERROR) << "Can't parse " << sponsored_dialog_id_string;
} else { } else {
sponsored_dialog_id_ = DialogId(r_dialog_id.ok()); DialogId dialog_id(r_dialog_id.ok());
Dialog *d = get_dialog_force(sponsored_dialog_id_); Dialog *d = get_dialog_force(dialog_id);
if (d == nullptr) { if (d != nullptr) {
LOG(ERROR) << "Can't load " << sponsored_dialog_id_; sponsored_dialog_id_ = dialog_id;
sponsored_dialog_id_ = DialogId(); if (is_dialog_sponsored(d)) {
send_update_chat_chat_list(d);
send_update_chat_is_sponsored(d);
}
} else {
LOG(ERROR) << "Can't load " << dialog_id;
} }
} }
} }
@ -13077,6 +13084,19 @@ vector<DialogId> MessagesManager::get_dialogs(FolderId folder_id, DialogDate off
limit = MAX_GET_DIALOGS; limit = MAX_GET_DIALOGS;
} }
if (folder_id == FolderId::main() && sponsored_dialog_id_.is_valid()) {
auto d = get_dialog(sponsored_dialog_id_);
CHECK(d != nullptr);
if (is_dialog_sponsored(d)) {
DialogDate date(get_dialog_public_order(d), d->dialog_id);
if (offset < date) {
result.push_back(sponsored_dialog_id_);
offset = date;
limit--;
}
}
}
auto it = list.ordered_dialogs_.upper_bound(offset); auto it = list.ordered_dialogs_.upper_bound(offset);
auto end = list.ordered_dialogs_.end(); auto end = list.ordered_dialogs_.end();
while (it != end && limit-- > 0) { while (it != end && limit-- > 0) {
@ -13261,9 +13281,7 @@ vector<DialogId> MessagesManager::get_pinned_dialogs(FolderId folder_id) const {
if (dialog_date.get_date() < MIN_PINNED_DIALOG_DATE) { if (dialog_date.get_date() < MIN_PINNED_DIALOG_DATE) {
break; break;
} }
if (dialog_date.get_order() != SPONSORED_DIALOG_ORDER) { result.push_back(dialog_date.get_dialog_id());
result.push_back(dialog_date.get_dialog_id());
}
} }
} }
@ -13366,8 +13384,9 @@ vector<DialogId> MessagesManager::sort_dialogs_by_order(const vector<DialogId> &
auto dialog_dates = transform(dialog_ids, [this, &fake_order](DialogId dialog_id) { auto dialog_dates = transform(dialog_ids, [this, &fake_order](DialogId dialog_id) {
const Dialog *d = get_dialog(dialog_id); const Dialog *d = get_dialog(dialog_id);
CHECK(d != nullptr); CHECK(d != nullptr);
if (is_dialog_inited(d) || d->order != DEFAULT_ORDER) { auto order = get_dialog_public_order(d);
return DialogDate(d->order, dialog_id); if (is_dialog_inited(d) || order != DEFAULT_ORDER) {
return DialogDate(order, dialog_id);
} }
// if the dialog is not inited yet, we need to assume that server knows better and the dialog needs to be returned // if the dialog is not inited yet, we need to assume that server knows better and the dialog needs to be returned
return DialogDate(fake_order--, dialog_id); return DialogDate(fake_order--, dialog_id);
@ -15326,7 +15345,10 @@ td_api::object_ptr<td_api::ChatType> MessagesManager::get_chat_type_object(Dialo
} }
} }
td_api::object_ptr<td_api::ChatList> MessagesManager::get_chat_list_object(const Dialog *d) { td_api::object_ptr<td_api::ChatList> MessagesManager::get_chat_list_object(const Dialog *d) const {
if (is_dialog_sponsored(d)) {
return get_chat_list_object(FolderId::main());
}
if (d->order == DEFAULT_ORDER) { if (d->order == DEFAULT_ORDER) {
return nullptr; return nullptr;
} }
@ -15445,7 +15467,7 @@ td_api::object_ptr<td_api::chat> MessagesManager::get_chat_object(const Dialog *
get_chat_photo_object(td_->file_manager_.get(), get_dialog_photo(d->dialog_id)), get_chat_photo_object(td_->file_manager_.get(), get_dialog_photo(d->dialog_id)),
get_dialog_permissions(d->dialog_id).get_chat_permissions_object(), get_dialog_permissions(d->dialog_id).get_chat_permissions_object(),
get_message_object(d->dialog_id, get_message(d, d->last_message_id)), get_dialog_public_order(d), get_message_object(d->dialog_id, get_message(d, d->last_message_id)), get_dialog_public_order(d),
d->pinned_order != DEFAULT_ORDER, d->is_marked_as_unread, d->order == SPONSORED_DIALOG_ORDER, d->pinned_order != DEFAULT_ORDER, d->is_marked_as_unread, is_dialog_sponsored(d),
get_dialog_has_scheduled_messages(d), can_delete_for_self, can_delete_for_all_users, get_dialog_has_scheduled_messages(d), can_delete_for_self, can_delete_for_all_users,
can_report_dialog(d->dialog_id), d->notification_settings.silent_send_message, can_report_dialog(d->dialog_id), d->notification_settings.silent_send_message,
d->server_unread_count + d->local_unread_count, d->last_read_inbox_message_id.get(), d->server_unread_count + d->local_unread_count, d->last_read_inbox_message_id.get(),
@ -22812,11 +22834,10 @@ void MessagesManager::send_update_chat_is_sponsored(const Dialog *d) const {
CHECK(d != nullptr); CHECK(d != nullptr);
LOG_CHECK(d->is_update_new_chat_sent) << "Wrong " << d->dialog_id << " in send_update_chat_is_sponsored"; LOG_CHECK(d->is_update_new_chat_sent) << "Wrong " << d->dialog_id << " in send_update_chat_is_sponsored";
bool is_sponsored = d->order == SPONSORED_DIALOG_ORDER;
LOG(INFO) << "Update chat is sponsored for " << d->dialog_id; LOG(INFO) << "Update chat is sponsored for " << d->dialog_id;
send_closure( send_closure(G()->td(), &Td::send_update,
G()->td(), &Td::send_update, make_tl_object<td_api::updateChatIsSponsored>(d->dialog_id.get(), is_dialog_sponsored(d),
make_tl_object<td_api::updateChatIsSponsored>(d->dialog_id.get(), is_sponsored, get_dialog_public_order(d))); get_dialog_public_order(d)));
} }
void MessagesManager::send_update_chat_online_member_count(DialogId dialog_id, int32 online_member_count) const { void MessagesManager::send_update_chat_online_member_count(DialogId dialog_id, int32 online_member_count) const {
@ -23646,8 +23667,8 @@ void MessagesManager::on_update_pinned_dialogs(FolderId folder_id) {
send_closure(actor_id, &MessagesManager::reload_pinned_dialogs, folder_id, Promise<Unit>()); send_closure(actor_id, &MessagesManager::reload_pinned_dialogs, folder_id, Promise<Unit>());
}); });
// max ordinary pinned dialogs + max pinned secret chats + sponsored proxy // max ordinary pinned dialogs + max pinned secret chats
size_t needed_dialogs = 2 * get_pinned_dialogs_limit(folder_id) + (folder_id == FolderId::main() ? 1 : 0); size_t needed_dialogs = 2 * get_pinned_dialogs_limit(folder_id);
auto &list = get_dialog_list(folder_id); auto &list = get_dialog_list(folder_id);
if (list.ordered_dialogs_.size() >= needed_dialogs) { if (list.ordered_dialogs_.size() >= needed_dialogs) {
query_promise.set_value(Unit()); query_promise.set_value(Unit());
@ -23854,8 +23875,9 @@ void MessagesManager::set_dialog_folder_id(Dialog *d, FolderId folder_id) {
// first remove the dialog from the old chat list // first remove the dialog from the old chat list
// this will send updateChatChatList if needed // this will send updateChatChatList if needed
if (d->pinned_order != DEFAULT_ORDER) { DialogId sponsored_dialog_id = sponsored_dialog_id_;
set_dialog_is_pinned(d, false); if (d->dialog_id == sponsored_dialog_id_ && d->order != DEFAULT_ORDER) {
sponsored_dialog_id_ = DialogId(); // supress updateChatIsSponsored
} }
set_dialog_order(d, DEFAULT_ORDER, true, false, "set_dialog_folder_id old"); set_dialog_order(d, DEFAULT_ORDER, true, false, "set_dialog_folder_id old");
@ -23863,6 +23885,8 @@ void MessagesManager::set_dialog_folder_id(Dialog *d, FolderId folder_id) {
d->folder_id = folder_id; d->folder_id = folder_id;
d->is_folder_id_inited = true; d->is_folder_id_inited = true;
sponsored_dialog_id_ = sponsored_dialog_id;
// update dialog position in the new folder // update dialog position in the new folder
// this will send updateChatChatList if needed // this will send updateChatChatList if needed
update_dialog_pos(d, "set_dialog_folder_id new"); update_dialog_pos(d, "set_dialog_folder_id new");
@ -28020,7 +28044,7 @@ void MessagesManager::fix_new_dialog(Dialog *d, unique_ptr<Message> &&last_datab
} }
if (need_get_history && !td_->auth_manager_->is_bot() && have_input_peer(dialog_id, AccessRights::Read) && if (need_get_history && !td_->auth_manager_->is_bot() && have_input_peer(dialog_id, AccessRights::Read) &&
d->order != DEFAULT_ORDER) { (d->order != DEFAULT_ORDER || is_dialog_sponsored(d))) {
get_history_from_the_end(dialog_id, true, false, Auto()); get_history_from_the_end(dialog_id, true, false, Auto());
} }
@ -28134,10 +28158,15 @@ int64 MessagesManager::get_dialog_order(MessageId message_id, int32 message_date
message_id.get_prev_server_message_id().get_server_message_id().get(); message_id.get_prev_server_message_id().get_server_message_id().get();
} }
bool MessagesManager::is_dialog_sponsored(const Dialog *d) const {
return d->order == DEFAULT_ORDER && d->dialog_id == sponsored_dialog_id_;
}
int64 MessagesManager::get_dialog_public_order(const Dialog *d) const { int64 MessagesManager::get_dialog_public_order(const Dialog *d) const {
DialogDate dialog_date(d->order, d->dialog_id); auto order = is_dialog_sponsored(d) ? static_cast<int64>(2147483647) << 32 : d->order;
DialogDate dialog_date(order, d->dialog_id);
auto *list = get_dialog_list(d->folder_id); auto *list = get_dialog_list(d->folder_id);
return list != nullptr && dialog_date <= list->last_dialog_date_ ? d->order : 0; return list != nullptr && dialog_date <= list->last_dialog_date_ ? order : 0;
} }
int64 MessagesManager::get_next_pinned_dialog_order() { int64 MessagesManager::get_next_pinned_dialog_order() {
@ -28181,11 +28210,7 @@ void MessagesManager::update_dialog_pos(Dialog *d, const char *source, bool need
LOG(INFO) << "Trying to update " << d->dialog_id << " order from " << source; LOG(INFO) << "Trying to update " << d->dialog_id << " order from " << source;
int64 new_order = DEFAULT_ORDER; int64 new_order = DEFAULT_ORDER;
if (is_removed_from_dialog_list(d)) { if (!is_removed_from_dialog_list(d)) {
if (d->pinned_order != DEFAULT_ORDER) {
return set_dialog_is_pinned(d, false);
}
} else {
if (d->pinned_order != DEFAULT_ORDER) { if (d->pinned_order != DEFAULT_ORDER) {
LOG(INFO) << "Pin at " << d->pinned_order << " found"; LOG(INFO) << "Pin at " << d->pinned_order << " found";
new_order = d->pinned_order; new_order = d->pinned_order;
@ -28247,15 +28272,11 @@ void MessagesManager::update_dialog_pos(Dialog *d, const char *source, bool need
} }
} }
} }
if (d->dialog_id != sponsored_dialog_id_ && new_order == DEFAULT_ORDER && !d->is_empty) { if (new_order == DEFAULT_ORDER && !d->is_empty) {
// if there is no known messages in the dialog, just leave it where it is LOG(INFO) << "There is no known messages in the chat, just leave it where it is";
LOG(INFO) << "There is no known messages in the dialog"; new_order = d->order;
return;
} }
} }
if (new_order == DEFAULT_ORDER && d->dialog_id == sponsored_dialog_id_ && d->folder_id == FolderId::main()) {
new_order = SPONSORED_DIALOG_ORDER;
}
if (set_dialog_order(d, new_order, need_send_update_chat_order, is_loaded_from_database, source)) { if (set_dialog_order(d, new_order, need_send_update_chat_order, is_loaded_from_database, source)) {
on_dialog_updated(d->dialog_id, "update_dialog_pos"); on_dialog_updated(d->dialog_id, "update_dialog_pos");
@ -28272,9 +28293,10 @@ bool MessagesManager::set_dialog_order(Dialog *d, int64 new_order, bool need_sen
auto &list = get_dialog_list(d->folder_id); auto &list = get_dialog_list(d->folder_id);
if (old_date == new_date) { if (old_date == new_date) {
if (new_order == DEFAULT_ORDER) { if (new_order == DEFAULT_ORDER) {
// first addition of a new left dialog
list.ordered_server_dialogs_.insert(new_date); list.ordered_server_dialogs_.insert(new_date);
} }
LOG(INFO) << "Dialog order is not changed: " << new_order << " from " << source; LOG(INFO) << "Chat order is not changed: " << new_order << " from " << source;
return false; return false;
} }
LOG(INFO) << "Update order of " << dialog_id << " from " << d->order << " to " << new_order << " from " << source; LOG(INFO) << "Update order of " << dialog_id << " from " << d->order << " to " << new_order << " from " << source;
@ -28302,8 +28324,7 @@ bool MessagesManager::set_dialog_order(Dialog *d, int64 new_order, bool need_sen
list.ordered_server_dialogs_.insert(new_date); list.ordered_server_dialogs_.insert(new_date);
bool add_to_hints = (d->order == DEFAULT_ORDER); bool add_to_hints = (d->order == DEFAULT_ORDER);
bool was_sponsored = (d->order == SPONSORED_DIALOG_ORDER); bool was_sponsored = is_dialog_sponsored(d);
bool is_sponsored = (new_order == SPONSORED_DIALOG_ORDER);
bool had_unread_counter = need_unread_counter(d->order); bool had_unread_counter = need_unread_counter(d->order);
bool has_unread_counter = need_unread_counter(new_order); bool has_unread_counter = need_unread_counter(new_order);
bool is_removed_from_folder = new_order == DEFAULT_ORDER; bool is_removed_from_folder = new_order == DEFAULT_ORDER;
@ -28325,6 +28346,15 @@ bool MessagesManager::set_dialog_order(Dialog *d, int64 new_order, bool need_sen
} }
} }
} }
if (new_order == DEFAULT_ORDER && d->pinned_order != DEFAULT_ORDER) {
d->pinned_order = DEFAULT_ORDER;
send_closure(G()->td(), &Td::send_update, make_tl_object<td_api::updateChatIsPinned>(d->dialog_id.get(), false, 0));
need_update = false;
}
d->order = new_order;
bool need_update_unread_chat_count = old_dialog_total_count != get_dialog_total_count(list); bool need_update_unread_chat_count = old_dialog_total_count != get_dialog_total_count(list);
CHECK(static_cast<int32>(list.ordered_dialogs_.size()) <= list.in_memory_dialog_total_count_); CHECK(static_cast<int32>(list.ordered_dialogs_.size()) <= list.in_memory_dialog_total_count_);
CHECK(static_cast<size_t>(list.in_memory_dialog_total_count_) <= list.ordered_server_dialogs_.size()); CHECK(static_cast<size_t>(list.in_memory_dialog_total_count_) <= list.ordered_server_dialogs_.size());
@ -28383,19 +28413,36 @@ bool MessagesManager::set_dialog_order(Dialog *d, int64 new_order, bool need_sen
send_update_unread_chat_count(d->folder_id, dialog_id, true, "changed total_count"); send_update_unread_chat_count(d->folder_id, dialog_id, true, "changed total_count");
} }
d->order = new_order;
if (add_to_hints) { if (add_to_hints) {
update_dialogs_hints(d); update_dialogs_hints(d);
} }
update_dialogs_hints_rating(d); update_dialogs_hints_rating(d);
if (is_added_to_folder) { if (is_removed_from_folder && d->folder_id != FolderId::main()) {
if (need_update) {
need_update = false;
send_closure(G()->td(), &Td::send_update, make_tl_object<td_api::updateChatOrder>(dialog_id.get(), updated_to));
}
d->folder_id = FolderId::main();
d->is_folder_id_inited = true;
send_update_chat_chat_list(d);
on_dialog_updated(d->dialog_id, "set_dialog_order 5");
if (is_dialog_sponsored(d)) {
auto &main_list = get_dialog_list(d->folder_id);
if (main_list.is_dialog_unread_count_inited_) {
send_update_unread_chat_count(d->folder_id, DialogId(), true, "set_dialog_order 6");
}
}
}
if (is_added_to_folder && !(dialog_id == sponsored_dialog_id_ && d->folder_id == FolderId::main())) {
send_update_chat_chat_list(d); send_update_chat_chat_list(d);
} }
if (was_sponsored != is_sponsored) { if (was_sponsored != is_dialog_sponsored(d)) {
send_update_chat_is_sponsored(d); send_update_chat_is_sponsored(d);
if (!is_loaded_from_database && is_sponsored) { if (!is_loaded_from_database && !was_sponsored) {
// channel is sponsored only if user isn't a channel member // channel is sponsored only if user isn't a channel member
remove_all_dialog_notifications(d, false, "set_dialog_order 3"); remove_all_dialog_notifications(d, false, "set_dialog_order 3");
remove_all_dialog_notifications(d, true, "set_dialog_order 4"); remove_all_dialog_notifications(d, true, "set_dialog_order 4");
@ -28405,7 +28452,7 @@ bool MessagesManager::set_dialog_order(Dialog *d, int64 new_order, bool need_sen
if (need_update && need_send_update_chat_order) { if (need_update && need_send_update_chat_order) {
send_closure(G()->td(), &Td::send_update, make_tl_object<td_api::updateChatOrder>(dialog_id.get(), updated_to)); send_closure(G()->td(), &Td::send_update, make_tl_object<td_api::updateChatOrder>(dialog_id.get(), updated_to));
} }
if (is_removed_from_folder) { if (is_removed_from_folder && !(dialog_id == sponsored_dialog_id_ && d->folder_id == FolderId::main())) {
send_update_chat_chat_list(d); send_update_chat_chat_list(d);
} }
return true; return true;
@ -30419,11 +30466,16 @@ void MessagesManager::set_sponsored_dialog_id(DialogId dialog_id) {
return; return;
} }
bool need_update_total_chat_count = false;
if (sponsored_dialog_id_.is_valid()) { if (sponsored_dialog_id_.is_valid()) {
Dialog *d = get_dialog(sponsored_dialog_id_); Dialog *d = get_dialog(sponsored_dialog_id_);
CHECK(d != nullptr); CHECK(d != nullptr);
bool is_sponsored = is_dialog_sponsored(d);
sponsored_dialog_id_ = DialogId(); sponsored_dialog_id_ = DialogId();
update_dialog_pos(d, "delete_sponsored_dialog_id"); if (is_sponsored) {
send_update_chat_is_sponsored(d);
need_update_total_chat_count = true;
}
} }
if (dialog_id.is_valid()) { if (dialog_id.is_valid()) {
@ -30432,7 +30484,18 @@ void MessagesManager::set_sponsored_dialog_id(DialogId dialog_id) {
Dialog *d = get_dialog(dialog_id); Dialog *d = get_dialog(dialog_id);
CHECK(d != nullptr); CHECK(d != nullptr);
sponsored_dialog_id_ = dialog_id; sponsored_dialog_id_ = dialog_id;
update_dialog_pos(d, "set_sponsored_dialog_id"); if (is_dialog_sponsored(d)) {
send_update_chat_is_sponsored(d);
need_update_total_chat_count = !need_update_total_chat_count;
}
}
if (need_update_total_chat_count) {
auto folder_id = FolderId::main();
auto &list = get_dialog_list(folder_id);
if (list.is_dialog_unread_count_inited_) {
send_update_unread_chat_count(FolderId::main(), DialogId(), true, "set_sponsored_dialog_id");
}
} }
if (G()->parameters().use_message_db) { if (G()->parameters().use_message_db) {
@ -30451,21 +30514,15 @@ td_api::object_ptr<td_api::updateUnreadMessageCount> MessagesManager::get_update
int32 unread_count = list.unread_message_total_count_; int32 unread_count = list.unread_message_total_count_;
int32 unread_unmuted_count = list.unread_message_total_count_ - list.unread_message_muted_count_; int32 unread_unmuted_count = list.unread_message_total_count_ - list.unread_message_muted_count_;
if (!include_sponsored_dialog_to_unread_count_ && sponsored_dialog_id_.is_valid() && if (include_sponsored_dialog_to_unread_count_ && sponsored_dialog_id_.is_valid() &&
list.folder_id == FolderId::main()) { list.folder_id == FolderId::main()) {
const Dialog *d = get_dialog(sponsored_dialog_id_); const Dialog *d = get_dialog(sponsored_dialog_id_);
CHECK(d != nullptr); CHECK(d != nullptr);
auto sponsored_unread_count = d->server_unread_count + d->local_unread_count; auto sponsored_unread_count = d->server_unread_count + d->local_unread_count;
if (sponsored_unread_count != 0) { if (is_dialog_sponsored(d) && sponsored_unread_count != 0) {
unread_count -= sponsored_unread_count; unread_count += sponsored_unread_count;
if (unread_count < 0) {
unread_count = 0;
}
if (!is_dialog_muted(d)) { if (!is_dialog_muted(d)) {
unread_unmuted_count -= sponsored_unread_count; unread_unmuted_count += sponsored_unread_count;
if (unread_unmuted_count < 0) {
unread_unmuted_count = 0;
}
} }
} }
} }
@ -30488,20 +30545,20 @@ td_api::object_ptr<td_api::updateUnreadChatCount> MessagesManager::get_update_un
CHECK(unread_marked_count >= 0); CHECK(unread_marked_count >= 0);
CHECK(unread_unmuted_marked_count >= 0); CHECK(unread_unmuted_marked_count >= 0);
if (!include_sponsored_dialog_to_unread_count_ && sponsored_dialog_id_.is_valid() && if (include_sponsored_dialog_to_unread_count_ && sponsored_dialog_id_.is_valid() &&
list.folder_id == FolderId::main()) { list.folder_id == FolderId::main()) {
const Dialog *d = get_dialog(sponsored_dialog_id_); const Dialog *d = get_dialog(sponsored_dialog_id_);
CHECK(d != nullptr); CHECK(d != nullptr);
auto sponsored_unread_count = d->server_unread_count + d->local_unread_count; auto sponsored_unread_count = d->server_unread_count + d->local_unread_count;
if (sponsored_unread_count != 0 || d->is_marked_as_unread) { if (is_dialog_sponsored(d) && (sponsored_unread_count != 0 || d->is_marked_as_unread)) {
unread_count = td::max(unread_count - 1, 0); unread_count++;
if (sponsored_unread_count == 0 && d->is_marked_as_unread) { if (sponsored_unread_count == 0 && d->is_marked_as_unread) {
unread_marked_count = td::max(unread_marked_count - 1, 0); unread_marked_count++;
} }
if (!is_dialog_muted(d)) { if (!is_dialog_muted(d)) {
unread_unmuted_count = td::max(unread_unmuted_count - 1, 0); unread_unmuted_count++;
if (sponsored_unread_count == 0 && d->is_marked_as_unread) { if (sponsored_unread_count == 0 && d->is_marked_as_unread) {
unread_unmuted_marked_count = td::max(unread_unmuted_marked_count - 1, 0); unread_unmuted_marked_count++;
} }
} }
} }

7
td/telegram/MessagesManager.h
View File

@ -1433,7 +1433,6 @@ class MessagesManager : public Actor {
static constexpr int32 MAX_RECENT_FOUND_DIALOGS = 20; // some reasonable value static constexpr int32 MAX_RECENT_FOUND_DIALOGS = 20; // some reasonable value
static constexpr size_t MAX_TITLE_LENGTH = 128; // server side limit for chat title static constexpr size_t MAX_TITLE_LENGTH = 128; // server side limit for chat title
static constexpr size_t MAX_DESCRIPTION_LENGTH = 255; // server side limit for chat description static constexpr size_t MAX_DESCRIPTION_LENGTH = 255; // server side limit for chat description
static constexpr int64 SPONSORED_DIALOG_ORDER = static_cast<int64>(2147483647) << 32;
static constexpr int32 MIN_PINNED_DIALOG_DATE = 2147000000; // some big date static constexpr int32 MIN_PINNED_DIALOG_DATE = 2147000000; // some big date
static constexpr int32 MAX_PRIVATE_MESSAGE_TTL = 60; // server side limit static constexpr int32 MAX_PRIVATE_MESSAGE_TTL = 60; // server side limit
@ -1952,7 +1951,7 @@ class MessagesManager : public Actor {
static bool need_unread_counter(int64 dialog_order); static bool need_unread_counter(int64 dialog_order);
static int32 get_dialog_total_count(const DialogList &list); int32 get_dialog_total_count(const DialogList &list) const;
void repair_server_dialog_total_count(FolderId folder_id); void repair_server_dialog_total_count(FolderId folder_id);
@ -2076,7 +2075,7 @@ class MessagesManager : public Actor {
td_api::object_ptr<td_api::ChatType> get_chat_type_object(DialogId dialog_id) const; td_api::object_ptr<td_api::ChatType> get_chat_type_object(DialogId dialog_id) const;
static td_api::object_ptr<td_api::ChatList> get_chat_list_object(const Dialog *d); td_api::object_ptr<td_api::ChatList> get_chat_list_object(const Dialog *d) const;
static td_api::object_ptr<td_api::ChatList> get_chat_list_object(FolderId folder_id); static td_api::object_ptr<td_api::ChatList> get_chat_list_object(FolderId folder_id);
@ -2256,6 +2255,8 @@ class MessagesManager : public Actor {
static int64 get_dialog_order(MessageId message_id, int32 message_date); static int64 get_dialog_order(MessageId message_id, int32 message_date);
bool is_dialog_sponsored(const Dialog *d) const;
int64 get_dialog_public_order(const Dialog *d) const; int64 get_dialog_public_order(const Dialog *d) const;
bool update_dialog_draft_message(Dialog *d, unique_ptr<DraftMessage> &&draft_message, bool from_update, bool update_dialog_draft_message(Dialog *d, unique_ptr<DraftMessage> &&draft_message, bool from_update,