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

Move GetAllDraftsQuery to DraftMessage.cpp.

Browse Source
This commit is contained in:
levlam 2023-05-31 16:17:20 +03:00
parent 1aeaaaa733
commit 394ce38278
3 changed files with 33 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
td/telegram/DraftMessage.cpp
View File

@ -12,6 +12,7 @@
#include "td/telegram/misc.h"
#include "td/telegram/ServerMessageId.h"
#include "td/telegram/Td.h"
#include "td/telegram/UpdatesManager.h"
#include "td/utils/buffer.h"
#include "td/utils/logging.h"
@ -82,6 +83,31 @@ class SaveDraftMessageQuery final : public Td::ResultHandler {
}
};
class GetAllDraftsQuery final : public Td::ResultHandler {
public:
void send() {
send_query(G()->net_query_creator().create(telegram_api::messages_getAllDrafts()));
}
void on_result(BufferSlice packet) final {
auto result_ptr = fetch_result<telegram_api::messages_getAllDrafts>(packet);
if (result_ptr.is_error()) {
return on_error(result_ptr.move_as_error());
}
auto ptr = result_ptr.move_as_ok();
LOG(INFO) << "Receive result for GetAllDraftsQuery: " << to_string(ptr);
td_->updates_manager_->on_get_updates(std::move(ptr), Promise<Unit>());
}
void on_error(Status status) final {
if (!G()->is_expected_error(status)) {
LOG(ERROR) << "Receive error for GetAllDraftsQuery: " << status;
}
status.ignore();
}
};
bool need_update_draft_message(const unique_ptr<DraftMessage> &old_draft_message,
const unique_ptr<DraftMessage> &new_draft_message, bool from_update) {
if (new_draft_message == nullptr) {
@ -181,4 +207,8 @@ void save_draft_message(Td *td, DialogId dialog_id, const unique_ptr<DraftMessag
td->create_handler<SaveDraftMessageQuery>(std::move(promise))->send(dialog_id, draft_message);
}
void load_all_draft_messages(Td *td) {
td->create_handler<GetAllDraftsQuery>()->send();
}
} // namespace td

2
td/telegram/DraftMessage.h
View File

@ -42,4 +42,6 @@ Result<unique_ptr<DraftMessage>> get_draft_message(Td *td, DialogId dialog_id,
void save_draft_message(Td *td, DialogId dialog_id, const unique_ptr<DraftMessage> &draft_message,
Promise<Unit> &&promise);
void load_all_draft_messages(Td *td);
} // namespace td

27
td/telegram/MessagesManager.cpp
View File

@ -127,31 +127,6 @@ class GetOnlinesQuery final : public Td::ResultHandler {
}
};
class GetAllDraftsQuery final : public Td::ResultHandler {
public:
void send() {
send_query(G()->net_query_creator().create(telegram_api::messages_getAllDrafts()));
}
void on_result(BufferSlice packet) final {
auto result_ptr = fetch_result<telegram_api::messages_getAllDrafts>(packet);
if (result_ptr.is_error()) {
return on_error(result_ptr.move_as_error());
}
auto ptr = result_ptr.move_as_ok();
LOG(INFO) << "Receive result for GetAllDraftsQuery: " << to_string(ptr);
td_->updates_manager_->on_get_updates(std::move(ptr), Promise<Unit>());
}
void on_error(Status status) final {
if (!G()->is_expected_error(status)) {
LOG(ERROR) << "Receive error for GetAllDraftsQuery: " << status;
}
status.ignore();
}
};
class GetDialogQuery final : public Td::ResultHandler {
DialogId dialog_id_;
@ -16943,7 +16918,7 @@ void MessagesManager::load_folder_dialog_list(FolderId folder_id, int32 limit, b
}
if (folder_id == FolderId::main() && folder.last_server_dialog_date_ == MIN_DIALOG_DATE) {
// do not pass promise to not wait for drafts before showing chat list
td_->create_handler<GetAllDraftsQuery>()->send();
load_all_draft_messages(td_);
}
lock.set_value(Unit());
}