tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add DialogFilter::set_pinned_dialog_ids.

Browse Source
This commit is contained in:
levlam 2023-03-22 14:14:21 +03:00
parent d3ef57e961
commit 4c5651027e
3 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
td/telegram/DialogFilter.cpp
View File

@ -119,6 +119,23 @@ void DialogFilter::set_dialog_is_pinned(InputDialogId input_dialog_id, bool is_p
}
}
void DialogFilter::set_pinned_dialog_ids(vector<InputDialogId> &&input_dialog_ids) {
FlatHashSet<DialogId, DialogIdHash> new_pinned_dialog_ids;
for (auto input_dialog_id : input_dialog_ids) {
new_pinned_dialog_ids.insert(input_dialog_id.get_dialog_id());
}
auto old_pinned_dialog_ids = std::move(pinned_dialog_ids);
pinned_dialog_ids = std::move(input_dialog_ids);
auto is_new_pinned = [&new_pinned_dialog_ids](InputDialogId input_dialog_id) {
return new_pinned_dialog_ids.count(input_dialog_id.get_dialog_id()) > 0;
};
td::remove_if(old_pinned_dialog_ids, is_new_pinned);
td::remove_if(included_dialog_ids, is_new_pinned);
td::remove_if(excluded_dialog_ids, is_new_pinned);
append(included_dialog_ids, old_pinned_dialog_ids);
}
void DialogFilter::include_dialog(InputDialogId input_dialog_id) {
included_dialog_ids.push_back(input_dialog_id);
InputDialogId::remove(excluded_dialog_ids, input_dialog_id.get_dialog_id());

2
td/telegram/DialogFilter.h
View File

@ -56,6 +56,8 @@ class DialogFilter {
void set_dialog_is_pinned(InputDialogId input_dialog_id, bool is_pinned);
void set_pinned_dialog_ids(vector<InputDialogId> &&input_dialog_ids);
void include_dialog(InputDialogId input_dialog_id);
void remove_secret_chat_dialog_ids();

13
td/telegram/MessagesManager.cpp
View File

@ -20637,17 +20637,8 @@ Status MessagesManager::set_pinned_dialogs(DialogListId dialog_list_id, vector<D
auto old_dialog_filter = get_dialog_filter(dialog_filter_id);
CHECK(old_dialog_filter != nullptr);
auto new_dialog_filter = make_unique<DialogFilter>(*old_dialog_filter);
auto old_pinned_dialog_ids = std::move(new_dialog_filter->pinned_dialog_ids);
new_dialog_filter->pinned_dialog_ids =
transform(dialog_ids, [this](DialogId dialog_id) { return get_input_dialog_id(dialog_id); });
auto is_new_pinned = [&new_pinned_dialog_ids](InputDialogId input_dialog_id) {
return new_pinned_dialog_ids.count(input_dialog_id.get_dialog_id()) > 0;
};
td::remove_if(old_pinned_dialog_ids, is_new_pinned);
td::remove_if(new_dialog_filter->included_dialog_ids, is_new_pinned);
td::remove_if(new_dialog_filter->excluded_dialog_ids, is_new_pinned);
append(new_dialog_filter->included_dialog_ids, old_pinned_dialog_ids);
new_dialog_filter->set_pinned_dialog_ids(
transform(dialog_ids, [this](DialogId dialog_id) { return get_input_dialog_id(dialog_id); }));
TRY_STATUS(new_dialog_filter->check_limits());
new_dialog_filter->sort_input_dialog_ids(td_, "set_pinned_dialogs");