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

Add can_delete_message method.

Browse Source 
GitOrigin-RevId: 87fd876f8155d6e93f1b2f50e03d610626c8ede0
This commit is contained in:
levlam 2020-03-20 03:59:09 +03:00
parent c4023a3f9b
commit 76558f0802
2 changed files with 41 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
td/telegram/MessagesManager.cpp
View File

@ -8494,6 +8494,28 @@ bool MessagesManager::can_delete_channel_message(DialogParticipantStatus status,
return true;
}
bool MessagesManager::can_delete_message(DialogId dialog_id, const Message *m) const {
if (m == nullptr) {
return true;
}
switch (dialog_id.get_type()) {
case DialogType::User:
return true;
case DialogType::Chat:
return true;
case DialogType::Channel: {
auto dialog_status = td_->contacts_manager_->get_channel_permissions(dialog_id.get_channel_id());
return can_delete_channel_message(dialog_status, m, td_->auth_manager_->is_bot());
}
case DialogType::SecretChat:
return true;
case DialogType::None:
default:
UNREACHABLE();
return false;
}
}
bool MessagesManager::can_revoke_message(DialogId dialog_id, const Message *m) const {
if (m == nullptr) {
return true;
@ -8571,48 +8593,30 @@ void MessagesManager::delete_messages(DialogId dialog_id, const vector<MessageId
message_id = get_persistent_message_id(d, message_id);
message_ids.push_back(message_id);
auto message = get_message_force(d, message_id, "delete_messages");
if (message != nullptr) {
if (message->message_id.is_scheduled()) {
if (message->message_id.is_scheduled_server()) {
deleted_scheduled_server_message_ids.push_back(message->message_id);
auto m = get_message_force(d, message_id, "delete_messages");
if (m != nullptr) {
if (m->message_id.is_scheduled()) {
if (m->message_id.is_scheduled_server()) {
deleted_scheduled_server_message_ids.push_back(m->message_id);
}
} else {
if (message->message_id.is_server() || is_secret) {
deleted_server_message_ids.push_back(message->message_id);
if (m->message_id.is_server() || is_secret) {
deleted_server_message_ids.push_back(m->message_id);
}
}
}
}
bool is_bot = td_->auth_manager_->is_bot();
switch (dialog_id.get_type()) {
case DialogType::User:
case DialogType::Chat:
if (is_bot) {
for (auto message_id : message_ids) {
if (!message_id.is_scheduled() && message_id.is_server() &&
!can_revoke_message(dialog_id, get_message(d, message_id))) {
auto m = get_message(d, message_id);
if (!can_delete_message(dialog_id, m)) {
return promise.set_error(Status::Error(6, "Message can't be deleted"));
}
if (is_bot && !message_id.is_scheduled() && message_id.is_server() && !can_revoke_message(dialog_id, m)) {
return promise.set_error(Status::Error(6, "Message can't be deleted for everyone"));
}
}
break;
case DialogType::Channel: {
auto dialog_status = td_->contacts_manager_->get_channel_permissions(dialog_id.get_channel_id());
for (auto message_id : message_ids) {
if (!can_delete_channel_message(dialog_status, get_message(d, message_id), is_bot)) {
return promise.set_error(Status::Error(6, "Message can't be deleted"));
}
}
break;
}
case DialogType::SecretChat:
break;
case DialogType::None:
default:
UNREACHABLE();
}
MultiPromiseActorSafe mpas{"DeleteMessagesFromServerMultiPromiseActor"};
mpas.add_promise(std::move(promise));
@ -17692,20 +17696,13 @@ tl_object_ptr<td_api::message> MessagesManager::get_message_object(DialogId dial
CHECK(sending_state == nullptr);
}
bool can_delete = true;
auto dialog_type = dialog_id.get_type();
auto is_bot = td_->auth_manager_->is_bot();
if (dialog_type == DialogType::Channel) {
auto dialog_status = td_->contacts_manager_->get_channel_permissions(dialog_id.get_channel_id());
can_delete = can_delete_channel_message(dialog_status, m, is_bot);
}
bool can_delete = can_delete_message(dialog_id, m);
bool is_scheduled = m->message_id.is_scheduled();
DialogId my_dialog_id = get_my_dialog_id();
bool can_delete_for_self = false;
bool can_delete_for_all_users = can_delete && can_revoke_message(dialog_id, m);
if (can_delete) {
switch (dialog_type) {
switch (dialog_id.get_type()) {
case DialogType::User:
case DialogType::Chat:
// TODO allow to delete yet unsent message just for self
@ -17750,7 +17747,7 @@ tl_object_ptr<td_api::message> MessagesManager::get_message_object(DialogId dial
ttl = 0;
}
auto scheduling_state = is_scheduled ? get_message_scheduling_state_object(m->date) : nullptr;
bool can_be_edited = for_event_log ? false : can_edit_message(dialog_id, m, false, is_bot);
bool can_be_edited = for_event_log ? false : can_edit_message(dialog_id, m, false, td_->auth_manager_->is_bot());
bool can_be_forwarded = for_event_log ? false : can_forward_message(dialog_id, m);
auto media_album_id = for_event_log ? static_cast<int64>(0) : m->media_album_id;
auto reply_to_message_id = for_event_log ? static_cast<int64>(0) : m->reply_to_message_id.get();

2
td/telegram/MessagesManager.h
View File

@ -1625,6 +1625,8 @@ class MessagesManager : public Actor {
static bool can_delete_channel_message(DialogParticipantStatus status, const Message *m, bool is_bot);
bool can_delete_message(DialogId dialog_id, const Message *m) const;
bool can_revoke_message(DialogId dialog_id, const Message *m) const;
bool can_unload_message(const Dialog *d, const Message *m) const;