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

Add DialogFilter::set_dialog_is_pinned/include_dialog.

Browse Source
This commit is contained in:
levlam 2023-03-21 23:21:19 +03:00
parent 739bc08a31
commit 18b7ee5e62
3 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
td/telegram/DialogFilter.cpp
View File

@ -104,6 +104,24 @@ Result<unique_ptr<DialogFilter>> DialogFilter::create_dialog_filter(Td *td, Dial
return std::move(dialog_filter);
}
void DialogFilter::set_dialog_is_pinned(InputDialogId input_dialog_id, bool is_pinned) {
auto dialog_id = input_dialog_id.get_dialog_id();
if (is_pinned) {
pinned_dialog_ids.insert(pinned_dialog_ids.begin(), input_dialog_id);
InputDialogId::remove(included_dialog_ids, dialog_id);
InputDialogId::remove(excluded_dialog_ids, dialog_id);
} else {
bool is_removed = InputDialogId::remove(pinned_dialog_ids, dialog_id);
CHECK(is_removed);
included_dialog_ids.push_back(input_dialog_id);
}
}
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());
}
void DialogFilter::remove_secret_chat_dialog_ids() {
auto remove_secret_chats = [](vector<InputDialogId> &input_dialog_ids) {
td::remove_if(input_dialog_ids, [](InputDialogId input_dialog_id) {

4
td/telegram/DialogFilter.h
View File

@ -54,6 +54,10 @@ class DialogFilter {
static Result<unique_ptr<DialogFilter>> create_dialog_filter(Td *td, DialogFilterId dialog_filter_id,
td_api::object_ptr<td_api::chatFilter> filter);
void set_dialog_is_pinned(InputDialogId input_dialog_id, bool is_pinned);
void include_dialog(InputDialogId input_dialog_id);
void remove_secret_chat_dialog_ids();
void remove_dialog_id(DialogId dialog_id);

19
td/telegram/MessagesManager.cpp
View File

@ -20507,16 +20507,7 @@ Status MessagesManager::toggle_dialog_is_pinned(DialogListId dialog_list_id, Dia
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);
if (is_pinned) {
new_dialog_filter->pinned_dialog_ids.insert(new_dialog_filter->pinned_dialog_ids.begin(),
get_input_dialog_id(dialog_id));
InputDialogId::remove(new_dialog_filter->included_dialog_ids, dialog_id);
InputDialogId::remove(new_dialog_filter->excluded_dialog_ids, dialog_id);
} else {
bool is_removed = InputDialogId::remove(new_dialog_filter->pinned_dialog_ids, dialog_id);
CHECK(is_removed);
new_dialog_filter->included_dialog_ids.push_back(get_input_dialog_id(dialog_id));
}
new_dialog_filter->set_dialog_is_pinned(get_input_dialog_id(dialog_id), is_pinned);
TRY_STATUS(new_dialog_filter->check_limits());
new_dialog_filter->sort_input_dialog_ids(td_, "toggle_dialog_is_pinned");
@ -34499,9 +34490,7 @@ vector<DialogListId> MessagesManager::get_dialog_lists_to_add_dialog(DialogId di
}
auto new_dialog_filter = make_unique<DialogFilter>(*dialog_filter);
new_dialog_filter->included_dialog_ids.push_back(get_input_dialog_id(dialog_id));
InputDialogId::remove(new_dialog_filter->excluded_dialog_ids, dialog_id);
new_dialog_filter->include_dialog(get_input_dialog_id(dialog_id));
if (new_dialog_filter->check_limits().is_ok()) {
result.push_back(DialogListId(dialog_filter_id));
}
@ -34541,9 +34530,7 @@ void MessagesManager::add_dialog_to_list(DialogId dialog_id, DialogListId dialog
}
auto new_dialog_filter = make_unique<DialogFilter>(*old_dialog_filter);
new_dialog_filter->included_dialog_ids.push_back(get_input_dialog_id(dialog_id));
InputDialogId::remove(new_dialog_filter->excluded_dialog_ids, dialog_id);
new_dialog_filter->include_dialog(get_input_dialog_id(dialog_id));
auto status = new_dialog_filter->check_limits();
if (status.is_error()) {
return promise.set_error(std::move(status));