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

Add and use is_editable_message_content.

Browse Source
This commit is contained in:
levlam 2024-02-23 20:19:54 +03:00
parent fe6201556b
commit 899587e4fa
3 changed files with 94 additions and 90 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
td/telegram/MessageContentType.cpp
View File

@ -381,6 +381,82 @@ bool is_service_message_content(MessageContentType content_type) {
}
}
bool is_editable_message_content(MessageContentType content_type) {
switch (content_type) {
case MessageContentType::Animation:
case MessageContentType::Audio:
case MessageContentType::Document:
case MessageContentType::Game:
case MessageContentType::Photo:
case MessageContentType::Text:
case MessageContentType::Video:
case MessageContentType::VoiceNote:
return true;
case MessageContentType::LiveLocation:
case MessageContentType::Poll:
case MessageContentType::Story:
case MessageContentType::Contact:
case MessageContentType::Dice:
case MessageContentType::Giveaway:
case MessageContentType::GiveawayWinners:
case MessageContentType::Location:
case MessageContentType::Sticker:
case MessageContentType::Venue:
case MessageContentType::VideoNote:
case MessageContentType::Invoice:
case MessageContentType::Unsupported:
case MessageContentType::ChatCreate:
case MessageContentType::ChatChangeTitle:
case MessageContentType::ChatChangePhoto:
case MessageContentType::ChatDeletePhoto:
case MessageContentType::ChatDeleteHistory:
case MessageContentType::ChatAddUsers:
case MessageContentType::ChatJoinedByLink:
case MessageContentType::ChatDeleteUser:
case MessageContentType::ChatMigrateTo:
case MessageContentType::ChannelCreate:
case MessageContentType::ChannelMigrateFrom:
case MessageContentType::PinMessage:
case MessageContentType::GameScore:
case MessageContentType::ScreenshotTaken:
case MessageContentType::ChatSetTtl:
case MessageContentType::Call:
case MessageContentType::PaymentSuccessful:
case MessageContentType::ContactRegistered:
case MessageContentType::ExpiredPhoto:
case MessageContentType::ExpiredVideo:
case MessageContentType::CustomServiceAction:
case MessageContentType::WebsiteConnected:
case MessageContentType::PassportDataSent:
case MessageContentType::PassportDataReceived:
case MessageContentType::ProximityAlertTriggered:
case MessageContentType::GroupCall:
case MessageContentType::InviteToGroupCall:
case MessageContentType::ChatSetTheme:
case MessageContentType::WebViewDataSent:
case MessageContentType::WebViewDataReceived:
case MessageContentType::GiftPremium:
case MessageContentType::TopicCreate:
case MessageContentType::TopicEdit:
case MessageContentType::SuggestProfilePhoto:
case MessageContentType::WriteAccessAllowed:
case MessageContentType::RequestedDialog:
case MessageContentType::WebViewWriteAccessAllowed:
case MessageContentType::SetBackground:
case MessageContentType::WriteAccessAllowedByRequest:
case MessageContentType::GiftCode:
case MessageContentType::GiveawayLaunch:
case MessageContentType::GiveawayResults:
case MessageContentType::ExpiredVideoNote:
case MessageContentType::ExpiredVoiceNote:
case MessageContentType::BoostApply:
return false;
default:
UNREACHABLE();
return false;
}
}
bool is_supported_reply_message_content(MessageContentType content_type) {
switch (content_type) {
case MessageContentType::Animation:

2
td/telegram/MessageContentType.h
View File

@ -93,6 +93,8 @@ bool can_be_secret_message_content(MessageContentType content_type);
bool is_service_message_content(MessageContentType content_type);
bool is_editable_message_content(MessageContentType content_type);
bool is_supported_reply_message_content(MessageContentType content_type);
bool is_expired_message_content(MessageContentType content_type);

106
td/telegram/MessagesManager.cpp
View File

@ -24815,97 +24815,23 @@ bool MessagesManager::can_edit_message(DialogId dialog_id, const Message *m, boo
}
}
switch (content_type) {
case MessageContentType::Animation:
case MessageContentType::Audio:
case MessageContentType::Document:
case MessageContentType::Game:
case MessageContentType::Photo:
case MessageContentType::Text:
case MessageContentType::Video:
case MessageContentType::VoiceNote:
return true;
case MessageContentType::LiveLocation: {
if (is_bot && only_reply_markup) {
// there is no caption to edit, but bot can edit inline reply_markup
return true;
}
return G()->unix_time() - m->date < get_message_content_live_location_period(m->content.get());
}
case MessageContentType::Poll: {
if (is_bot && only_reply_markup) {
// there is no caption to edit, but bot can edit inline reply_markup
return true;
}
if (m->message_id.is_scheduled()) {
return false;
}
return !get_message_content_poll_is_closed(td_, m->content.get());
}
case MessageContentType::Story:
case MessageContentType::Contact:
case MessageContentType::Dice:
case MessageContentType::Giveaway:
case MessageContentType::GiveawayWinners:
case MessageContentType::Location:
case MessageContentType::Sticker:
case MessageContentType::Venue:
case MessageContentType::VideoNote:
// there is no caption to edit, but bot can edit inline reply_markup
return is_bot && only_reply_markup;
case MessageContentType::Invoice:
case MessageContentType::Unsupported:
case MessageContentType::ChatCreate:
case MessageContentType::ChatChangeTitle:
case MessageContentType::ChatChangePhoto:
case MessageContentType::ChatDeletePhoto:
case MessageContentType::ChatDeleteHistory:
case MessageContentType::ChatAddUsers:
case MessageContentType::ChatJoinedByLink:
case MessageContentType::ChatDeleteUser:
case MessageContentType::ChatMigrateTo:
case MessageContentType::ChannelCreate:
case MessageContentType::ChannelMigrateFrom:
case MessageContentType::PinMessage:
case MessageContentType::GameScore:
case MessageContentType::ScreenshotTaken:
case MessageContentType::ChatSetTtl:
case MessageContentType::Call:
case MessageContentType::PaymentSuccessful:
case MessageContentType::ContactRegistered:
case MessageContentType::ExpiredPhoto:
case MessageContentType::ExpiredVideo:
case MessageContentType::CustomServiceAction:
case MessageContentType::WebsiteConnected:
case MessageContentType::PassportDataSent:
case MessageContentType::PassportDataReceived:
case MessageContentType::ProximityAlertTriggered:
case MessageContentType::GroupCall:
case MessageContentType::InviteToGroupCall:
case MessageContentType::ChatSetTheme:
case MessageContentType::WebViewDataSent:
case MessageContentType::WebViewDataReceived:
case MessageContentType::GiftPremium:
case MessageContentType::TopicCreate:
case MessageContentType::TopicEdit:
case MessageContentType::SuggestProfilePhoto:
case MessageContentType::WriteAccessAllowed:
case MessageContentType::RequestedDialog:
case MessageContentType::WebViewWriteAccessAllowed:
case MessageContentType::SetBackground:
case MessageContentType::WriteAccessAllowedByRequest:
case MessageContentType::GiftCode:
case MessageContentType::GiveawayLaunch:
case MessageContentType::GiveawayResults:
case MessageContentType::ExpiredVideoNote:
case MessageContentType::ExpiredVoiceNote:
case MessageContentType::BoostApply:
return false;
default:
UNREACHABLE();
if (is_editable_message_content(content_type)) {
return true;
}
if (is_bot && only_reply_markup && !is_service_message_content(content_type) &&
!is_expired_message_content(content_type)) {
return true;
}
switch (content_type) {
case MessageContentType::LiveLocation:
// live location can be edited only during the period
return G()->unix_time() - m->date < get_message_content_live_location_period(m->content.get());
case MessageContentType::Poll:
// non-closed non-scheduled polls can be closed
return !m->message_id.is_scheduled() && !get_message_content_poll_is_closed(td_, m->content.get());
default:
return false;
}
return false;
}
bool MessagesManager::can_resend_message(const Message *m) const {