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

Add get_message_content_min_user_ids and register story sender.

Browse Source
This commit is contained in:
levlam 2023-06-13 13:46:12 +03:00
parent 55d33a8665
commit 64badee6f2
3 changed files with 142 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

147
td/telegram/MessageContent.cpp
View File

@ -3301,13 +3301,148 @@ std::pair<InputGroupCallId, bool> get_message_content_group_call_info(const Mess
return {m->input_group_call_id, m->duration >= 0};
}
UserId get_message_content_contact_user_id(const MessageContent *content) {
switch (content->get_type()) {
case MessageContentType::Contact:
return static_cast<const MessageContact *>(content)->contact.get_user_id();
default:
return UserId();
vector<UserId> get_message_content_min_user_ids(const MessageContent *message_content) {
switch (message_content->get_type()) {
case MessageContentType::Text: {
// const auto *content = static_cast<const MessageText *>(message_content);
break;
}
case MessageContentType::Animation:
break;
case MessageContentType::Audio:
break;
case MessageContentType::Contact: {
const auto *content = static_cast<const MessageContact *>(message_content);
return {content->contact.get_user_id()};
}
case MessageContentType::Document:
break;
case MessageContentType::Game: {
// not supported
// const auto *content = static_cast<const MessageGame *>(message_content);
// return {content->game.get_bot_user_id())};
}
case MessageContentType::Invoice:
break;
case MessageContentType::LiveLocation:
break;
case MessageContentType::Location:
break;
case MessageContentType::Photo:
break;
case MessageContentType::Sticker:
break;
case MessageContentType::Venue:
break;
case MessageContentType::Video:
break;
case MessageContentType::VideoNote:
break;
case MessageContentType::VoiceNote:
break;
case MessageContentType::ChatCreate: {
const auto *content = static_cast<const MessageChatCreate *>(message_content);
return content->participant_user_ids;
}
case MessageContentType::ChatChangeTitle:
break;
case MessageContentType::ChatChangePhoto:
break;
case MessageContentType::ChatDeletePhoto:
break;
case MessageContentType::ChatDeleteHistory:
break;
case MessageContentType::ChatAddUsers: {
const auto *content = static_cast<const MessageChatAddUsers *>(message_content);
return content->user_ids;
}
case MessageContentType::ChatJoinedByLink:
break;
case MessageContentType::ChatDeleteUser: {
const auto *content = static_cast<const MessageChatDeleteUser *>(message_content);
return {content->user_id};
}
case MessageContentType::ChatMigrateTo:
break;
case MessageContentType::ChannelCreate:
break;
case MessageContentType::ChannelMigrateFrom:
break;
case MessageContentType::PinMessage:
break;
case MessageContentType::GameScore:
break;
case MessageContentType::ScreenshotTaken:
break;
case MessageContentType::ChatSetTtl:
// the content->from_user_id user can't be min
break;
case MessageContentType::Unsupported:
break;
case MessageContentType::Call:
break;
case MessageContentType::PaymentSuccessful:
break;
case MessageContentType::ContactRegistered:
break;
case MessageContentType::ExpiredPhoto:
break;
case MessageContentType::ExpiredVideo:
break;
case MessageContentType::CustomServiceAction:
break;
case MessageContentType::WebsiteConnected:
break;
case MessageContentType::PassportDataSent:
break;
case MessageContentType::PassportDataReceived:
break;
case MessageContentType::Poll:
break;
case MessageContentType::Dice:
break;
case MessageContentType::ProximityAlertTriggered:
break;
case MessageContentType::GroupCall:
break;
case MessageContentType::InviteToGroupCall: {
const auto *content = static_cast<const MessageInviteToGroupCall *>(message_content);
return content->user_ids;
}
case MessageContentType::ChatSetTheme:
break;
case MessageContentType::WebViewDataSent:
break;
case MessageContentType::WebViewDataReceived:
break;
case MessageContentType::GiftPremium:
break;
case MessageContentType::TopicCreate:
break;
case MessageContentType::TopicEdit:
break;
case MessageContentType::SuggestProfilePhoto:
break;
case MessageContentType::WriteAccessAllowed:
break;
case MessageContentType::RequestedDialog:
break;
case MessageContentType::WebViewWriteAccessAllowed:
break;
case MessageContentType::SetBackground:
break;
case MessageContentType::Story: {
const auto *content = static_cast<const MessageStory *>(message_content);
if (content->story_full_id.get_dialog_id().get_type() == DialogType::User) {
return {content->story_full_id.get_dialog_id().get_user_id()};
}
break;
}
default:
UNREACHABLE();
break;
}
return {};
}
vector<UserId> get_message_content_added_user_ids(const MessageContent *content) {

2
td/telegram/MessageContent.h
View File

@ -144,7 +144,7 @@ FullMessageId get_message_content_replied_message_id(DialogId dialog_id, const M
std::pair<InputGroupCallId, bool> get_message_content_group_call_info(const MessageContent *content);
UserId get_message_content_contact_user_id(const MessageContent *content);
vector<UserId> get_message_content_min_user_ids(const MessageContent *message_content);
vector<UserId> get_message_content_added_user_ids(const MessageContent *content);

12
td/telegram/MessagesManager.cpp
View File

@ -13209,17 +13209,7 @@ vector<UserId> MessagesManager::get_message_user_ids(const Message *m) {
if (m->forward_info != nullptr && m->forward_info->sender_user_id.is_valid()) {
user_ids.push_back(m->forward_info->sender_user_id);
}
if (m->content->get_type() == MessageContentType::ChatAddUsers) {
append(user_ids, get_message_content_added_user_ids(m->content.get()));
}
auto deleted_user_id = get_message_content_deleted_user_id(m->content.get());
if (deleted_user_id.is_valid()) {
user_ids.push_back(deleted_user_id);
}
auto contact_user_id = get_message_content_contact_user_id(m->content.get());
if (contact_user_id.is_valid()) {
user_ids.push_back(contact_user_id);
}
append(user_ids, get_message_content_min_user_ids(m->content.get()));
return user_ids;
}