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

Add triggers for file removal from Downloads and search text changes.

Browse Source
This commit is contained in:
levlam 2022-02-28 02:43:53 +03:00
parent a3cef00551
commit 3e4128fc1d
2 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
td/telegram/DownloadManager.cpp
View File

@ -119,6 +119,7 @@ class DownloadManagerImpl final : public DownloadManager {
files_.erase(file_info.download_id);
return Status::OK();
}
Status change_search_text(FileId file_id, FileSourceId file_source_id, string search_by) final {
TRY_STATUS(check_is_active());
TRY_RESULT(file_info_ptr, get_file_info(file_id, file_source_id));

27
td/telegram/MessagesManager.cpp
View File

@ -22880,6 +22880,7 @@ void MessagesManager::remove_message_file_sources(DialogId dialog_id, const Mess
auto file_source_id = get_message_file_source_id(FullMessageId(dialog_id, m->message_id));
if (file_source_id.is_valid()) {
for (auto file_id : file_ids) {
send_closure(td_->download_manager_actor_, &DownloadManager::remove_file, file_id, file_source_id, false);
td_->file_manager_->remove_file_source(file_id, file_source_id);
}
}
@ -22896,15 +22897,19 @@ void MessagesManager::change_message_files(DialogId dialog_id, const Message *m,
}
FullMessageId full_message_id{dialog_id, m->message_id};
if (need_delete_message_files(dialog_id, m)) {
bool need_delete_files = need_delete_message_files(dialog_id, m);
auto file_source_id = get_message_file_source_id(full_message_id);
for (auto file_id : old_file_ids) {
if (!td::contains(new_file_ids, file_id) && need_delete_file(full_message_id, file_id)) {
if (!td::contains(new_file_ids, file_id)) {
if (need_delete_files && need_delete_file(full_message_id, file_id)) {
send_closure(G()->file_manager(), &FileManager::delete_file, file_id, Promise<>(), "change_message_files");
}
if (file_source_id.is_valid()) {
send_closure(td_->download_manager_actor_, &DownloadManager::remove_file, file_id, file_source_id, false);
}
}
}
auto file_source_id = get_message_file_source_id(full_message_id);
if (file_source_id.is_valid()) {
td_->file_manager_->change_files_source(file_source_id, old_file_ids, new_file_ids);
}
@ -35789,9 +35794,23 @@ bool MessagesManager::update_message_content(DialogId dialog_id, Message *old_me
LOG(INFO) << "Content of " << old_message->message_id << " in " << dialog_id << " has changed";
}
if (need_update && need_send_update_message_content) {
if (need_update) {
if (need_send_update_message_content) {
send_update_message_content(dialog_id, old_message, is_message_in_dialog, "update_message_content");
}
auto file_ids = get_message_content_file_ids(old_content.get(), td_);
if (!file_ids.empty()) {
auto file_source_id = get_message_file_source_id(FullMessageId(dialog_id, old_message->message_id));
if (file_source_id.is_valid()) {
auto search_text = get_message_search_text(old_message);
for (auto file_id : file_ids) {
send_closure(td_->download_manager_actor_, &DownloadManager::change_search_text, file_id, file_source_id,
search_text);
}
}
}
}
return need_update;
}