// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2020 // // Distributed under the Boost Software License, Version 1.0. (See accompanying // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) // #include "td/telegram/MessageContent.h" #include "td/telegram/AnimationsManager.h" #include "td/telegram/AnimationsManager.hpp" #include "td/telegram/AudiosManager.h" #include "td/telegram/AudiosManager.hpp" #include "td/telegram/AuthManager.h" #include "td/telegram/CallDiscardReason.h" #include "td/telegram/ChannelId.h" #include "td/telegram/ChatId.h" #include "td/telegram/Contact.h" #include "td/telegram/ContactsManager.h" #include "td/telegram/Document.h" #include "td/telegram/DocumentsManager.h" #include "td/telegram/DocumentsManager.hpp" #include "td/telegram/files/FileId.h" #include "td/telegram/files/FileId.hpp" #include "td/telegram/files/FileLocation.h" #include "td/telegram/files/FileManager.h" #include "td/telegram/files/FileType.h" #include "td/telegram/Game.h" #include "td/telegram/Game.hpp" #include "td/telegram/Global.h" #include "td/telegram/HashtagHints.h" #include "td/telegram/InputMessageText.h" #include "td/telegram/Location.h" #include "td/telegram/MessageEntity.h" #include "td/telegram/MessageEntity.hpp" #include "td/telegram/MessageId.h" #include "td/telegram/MessagesDb.h" #include "td/telegram/misc.h" #include "td/telegram/net/DcId.h" #include "td/telegram/Payments.h" #include "td/telegram/Payments.hpp" #include "td/telegram/Photo.h" #include "td/telegram/Photo.hpp" #include "td/telegram/PhotoSizeSource.h" #include "td/telegram/PollId.h" #include "td/telegram/PollId.hpp" #include "td/telegram/PollManager.h" #include "td/telegram/secret_api.hpp" #include "td/telegram/SecureValue.h" #include "td/telegram/SecureValue.hpp" #include "td/telegram/ServerMessageId.h" #include "td/telegram/StickersManager.h" #include "td/telegram/StickersManager.hpp" #include "td/telegram/Td.h" #include "td/telegram/TopDialogManager.h" #include "td/telegram/UserId.h" #include "td/telegram/Venue.h" #include "td/telegram/Version.h" #include "td/telegram/VideoNotesManager.h" #include "td/telegram/VideoNotesManager.hpp" #include "td/telegram/VideosManager.h" #include "td/telegram/VideosManager.hpp" #include "td/telegram/VoiceNotesManager.h" #include "td/telegram/VoiceNotesManager.hpp" #include "td/telegram/WebPageId.h" #include "td/telegram/WebPagesManager.h" #include "td/actor/actor.h" #include "td/actor/MultiPromise.h" #include "td/actor/PromiseFuture.h" #include "td/utils/format.h" #include "td/utils/HttpUrl.h" #include "td/utils/logging.h" #include "td/utils/MimeType.h" #include "td/utils/misc.h" #include "td/utils/PathView.h" #include "td/utils/Slice.h" #include "td/utils/tl_helpers.h" #include "td/utils/utf8.h" #include namespace td { class MessageText : public MessageContent { public: FormattedText text; WebPageId web_page_id; MessageText() = default; MessageText(FormattedText text, WebPageId web_page_id) : text(std::move(text)), web_page_id(web_page_id) { } MessageContentType get_type() const override { return MessageContentType::Text; } }; class MessageAnimation : public MessageContent { public: FileId file_id; FormattedText caption; MessageAnimation() = default; MessageAnimation(FileId file_id, FormattedText &&caption) : file_id(file_id), caption(std::move(caption)) { } MessageContentType get_type() const override { return MessageContentType::Animation; } }; class MessageAudio : public MessageContent { public: FileId file_id; FormattedText caption; MessageAudio() = default; MessageAudio(FileId file_id, FormattedText &&caption) : file_id(file_id), caption(std::move(caption)) { } MessageContentType get_type() const override { return MessageContentType::Audio; } }; class MessageDocument : public MessageContent { public: FileId file_id; FormattedText caption; MessageDocument() = default; MessageDocument(FileId file_id, FormattedText &&caption) : file_id(file_id), caption(std::move(caption)) { } MessageContentType get_type() const override { return MessageContentType::Document; } }; class MessagePhoto : public MessageContent { public: Photo photo; FormattedText caption; MessagePhoto() = default; MessagePhoto(Photo &&photo, FormattedText &&caption) : photo(std::move(photo)), caption(std::move(caption)) { } MessageContentType get_type() const override { return MessageContentType::Photo; } }; class MessageSticker : public MessageContent { public: FileId file_id; MessageSticker() = default; explicit MessageSticker(FileId file_id) : file_id(file_id) { } MessageContentType get_type() const override { return MessageContentType::Sticker; } }; class MessageVideo : public MessageContent { public: FileId file_id; FormattedText caption; MessageVideo() = default; MessageVideo(FileId file_id, FormattedText &&caption) : file_id(file_id), caption(std::move(caption)) { } MessageContentType get_type() const override { return MessageContentType::Video; } }; class MessageVoiceNote : public MessageContent { public: FileId file_id; FormattedText caption; bool is_listened; MessageVoiceNote() = default; MessageVoiceNote(FileId file_id, FormattedText &&caption, bool is_listened) : file_id(file_id), caption(std::move(caption)), is_listened(is_listened) { } MessageContentType get_type() const override { return MessageContentType::VoiceNote; } }; class MessageContact : public MessageContent { public: Contact contact; MessageContact() = default; explicit MessageContact(Contact &&contact) : contact(std::move(contact)) { } MessageContentType get_type() const override { return MessageContentType::Contact; } }; class MessageLocation : public MessageContent { public: Location location; MessageLocation() = default; explicit MessageLocation(Location &&location) : location(std::move(location)) { } MessageContentType get_type() const override { return MessageContentType::Location; } }; class MessageVenue : public MessageContent { public: Venue venue; MessageVenue() = default; explicit MessageVenue(Venue &&venue) : venue(std::move(venue)) { } MessageContentType get_type() const override { return MessageContentType::Venue; } }; class MessageChatCreate : public MessageContent { public: string title; vector participant_user_ids; MessageChatCreate() = default; MessageChatCreate(string &&title, vector &&participant_user_ids) : title(std::move(title)), participant_user_ids(std::move(participant_user_ids)) { } MessageContentType get_type() const override { return MessageContentType::ChatCreate; } }; class MessageChatChangeTitle : public MessageContent { public: string title; MessageChatChangeTitle() = default; explicit MessageChatChangeTitle(string &&title) : title(std::move(title)) { } MessageContentType get_type() const override { return MessageContentType::ChatChangeTitle; } }; class MessageChatChangePhoto : public MessageContent { public: Photo photo; MessageChatChangePhoto() = default; explicit MessageChatChangePhoto(Photo &&photo) : photo(std::move(photo)) { } MessageContentType get_type() const override { return MessageContentType::ChatChangePhoto; } }; class MessageChatDeletePhoto : public MessageContent { public: MessageContentType get_type() const override { return MessageContentType::ChatDeletePhoto; } }; class MessageChatDeleteHistory : public MessageContent { public: MessageContentType get_type() const override { return MessageContentType::ChatDeleteHistory; } }; class MessageChatAddUsers : public MessageContent { public: vector user_ids; MessageChatAddUsers() = default; explicit MessageChatAddUsers(vector &&user_ids) : user_ids(std::move(user_ids)) { } MessageContentType get_type() const override { return MessageContentType::ChatAddUsers; } }; class MessageChatJoinedByLink : public MessageContent { public: MessageContentType get_type() const override { return MessageContentType::ChatJoinedByLink; } }; class MessageChatDeleteUser : public MessageContent { public: UserId user_id; MessageChatDeleteUser() = default; explicit MessageChatDeleteUser(UserId user_id) : user_id(user_id) { } MessageContentType get_type() const override { return MessageContentType::ChatDeleteUser; } }; class MessageChatMigrateTo : public MessageContent { public: ChannelId migrated_to_channel_id; MessageChatMigrateTo() = default; explicit MessageChatMigrateTo(ChannelId migrated_to_channel_id) : migrated_to_channel_id(migrated_to_channel_id) { } MessageContentType get_type() const override { return MessageContentType::ChatMigrateTo; } }; class MessageChannelCreate : public MessageContent { public: string title; MessageChannelCreate() = default; explicit MessageChannelCreate(string &&title) : title(std::move(title)) { } MessageContentType get_type() const override { return MessageContentType::ChannelCreate; } }; class MessageChannelMigrateFrom : public MessageContent { public: string title; ChatId migrated_from_chat_id; MessageChannelMigrateFrom() = default; MessageChannelMigrateFrom(string &&title, ChatId migrated_from_chat_id) : title(std::move(title)), migrated_from_chat_id(migrated_from_chat_id) { } MessageContentType get_type() const override { return MessageContentType::ChannelMigrateFrom; } }; class MessagePinMessage : public MessageContent { public: MessageId message_id; MessagePinMessage() = default; explicit MessagePinMessage(MessageId message_id) : message_id(message_id) { } MessageContentType get_type() const override { return MessageContentType::PinMessage; } }; class MessageGame : public MessageContent { public: Game game; MessageGame() = default; explicit MessageGame(Game &&game) : game(std::move(game)) { } MessageContentType get_type() const override { return MessageContentType::Game; } }; class MessageGameScore : public MessageContent { public: MessageId game_message_id; int64 game_id; int32 score; MessageGameScore() = default; MessageGameScore(MessageId game_message_id, int64 game_id, int32 score) : game_message_id(game_message_id), game_id(game_id), score(score) { } MessageContentType get_type() const override { return MessageContentType::GameScore; } }; class MessageScreenshotTaken : public MessageContent { public: MessageContentType get_type() const override { return MessageContentType::ScreenshotTaken; } }; class MessageChatSetTtl : public MessageContent { public: int32 ttl; MessageChatSetTtl() = default; explicit MessageChatSetTtl(int32 ttl) : ttl(ttl) { } MessageContentType get_type() const override { return MessageContentType::ChatSetTtl; } }; class MessageUnsupported : public MessageContent { public: static constexpr int32 CURRENT_VERSION = 5; int32 version = CURRENT_VERSION; MessageUnsupported() = default; explicit MessageUnsupported(int32 version) : version(version) { } MessageContentType get_type() const override { return MessageContentType::Unsupported; } }; class MessageCall : public MessageContent { public: int64 call_id; int32 duration; CallDiscardReason discard_reason; bool is_video; MessageCall() = default; MessageCall(int64 call_id, int32 duration, CallDiscardReason discard_reason, bool is_video) : call_id(call_id), duration(duration), discard_reason(discard_reason), is_video(is_video) { } MessageContentType get_type() const override { return MessageContentType::Call; } }; class MessageInvoice : public MessageContent { public: string title; string description; Photo photo; string start_parameter; // InputMessageInvoice Invoice invoice; string payload; string provider_token; string provider_data; // MessageInvoice int64 total_amount = 0; MessageId receipt_message_id; MessageInvoice() = default; MessageInvoice(string &&title, string &&description, Photo &&photo, string &&start_parameter, int64 total_amount, string &¤cy, bool is_test, bool need_shipping_address, MessageId receipt_message_id) : title(std::move(title)) , description(std::move(description)) , photo(std::move(photo)) , start_parameter(std::move(start_parameter)) , invoice(std::move(currency), is_test, need_shipping_address) , payload() , provider_token() , provider_data() , total_amount(total_amount) , receipt_message_id(receipt_message_id) { } MessageContentType get_type() const override { return MessageContentType::Invoice; } }; class MessagePaymentSuccessful : public MessageContent { public: MessageId invoice_message_id; string currency; int64 total_amount = 0; // bots only part string invoice_payload; string shipping_option_id; unique_ptr order_info; string telegram_payment_charge_id; string provider_payment_charge_id; MessagePaymentSuccessful() = default; MessagePaymentSuccessful(MessageId invoice_message_id, string &¤cy, int64 total_amount) : invoice_message_id(invoice_message_id), currency(std::move(currency)), total_amount(total_amount) { } MessageContentType get_type() const override { return MessageContentType::PaymentSuccessful; } }; class MessageVideoNote : public MessageContent { public: FileId file_id; bool is_viewed = false; MessageVideoNote() = default; MessageVideoNote(FileId file_id, bool is_viewed) : file_id(file_id), is_viewed(is_viewed) { } MessageContentType get_type() const override { return MessageContentType::VideoNote; } }; class MessageContactRegistered : public MessageContent { public: MessageContentType get_type() const override { return MessageContentType::ContactRegistered; } }; class MessageExpiredPhoto : public MessageContent { public: MessageExpiredPhoto() = default; MessageContentType get_type() const override { return MessageContentType::ExpiredPhoto; } }; class MessageExpiredVideo : public MessageContent { public: MessageExpiredVideo() = default; MessageContentType get_type() const override { return MessageContentType::ExpiredVideo; } }; class MessageLiveLocation : public MessageContent { public: Location location; int32 period; MessageLiveLocation() = default; MessageLiveLocation(Location &&location, int32 period) : location(std::move(location)), period(period) { } MessageContentType get_type() const override { return MessageContentType::LiveLocation; } }; class MessageCustomServiceAction : public MessageContent { public: string message; MessageCustomServiceAction() = default; explicit MessageCustomServiceAction(string &&message) : message(std::move(message)) { } MessageContentType get_type() const override { return MessageContentType::CustomServiceAction; } }; class MessageWebsiteConnected : public MessageContent { public: string domain_name; MessageWebsiteConnected() = default; explicit MessageWebsiteConnected(string &&domain_name) : domain_name(std::move(domain_name)) { } MessageContentType get_type() const override { return MessageContentType::WebsiteConnected; } }; class MessagePassportDataSent : public MessageContent { public: vector types; MessagePassportDataSent() = default; explicit MessagePassportDataSent(vector &&types) : types(std::move(types)) { } MessageContentType get_type() const override { return MessageContentType::PassportDataSent; } }; class MessagePassportDataReceived : public MessageContent { public: vector values; EncryptedSecureCredentials credentials; MessagePassportDataReceived() = default; MessagePassportDataReceived(vector &&values, EncryptedSecureCredentials &&credentials) : values(std::move(values)), credentials(std::move(credentials)) { } MessageContentType get_type() const override { return MessageContentType::PassportDataReceived; } }; class MessagePoll : public MessageContent { public: PollId poll_id; MessagePoll() = default; explicit MessagePoll(PollId poll_id) : poll_id(poll_id) { } MessageContentType get_type() const override { return MessageContentType::Poll; } }; class MessageDice : public MessageContent { public: string emoji; int32 dice_value = 0; static constexpr const char *DEFAULT_EMOJI = "🎲"; MessageDice() = default; MessageDice(string emoji, int32 dice_value) : emoji(emoji.empty() ? string(DEFAULT_EMOJI) : std::move(emoji)), dice_value(dice_value) { } MessageContentType get_type() const override { return MessageContentType::Dice; } }; constexpr const char *MessageDice::DEFAULT_EMOJI; template static void store(const MessageContent *content, StorerT &storer) { CHECK(content != nullptr); Td *td = storer.context()->td().get_actor_unsafe(); CHECK(td != nullptr); auto content_type = content->get_type(); store(content_type, storer); switch (content_type) { case MessageContentType::Animation: { auto m = static_cast(content); td->animations_manager_->store_animation(m->file_id, storer); store(m->caption, storer); break; } case MessageContentType::Audio: { auto m = static_cast(content); td->audios_manager_->store_audio(m->file_id, storer); store(m->caption, storer); store(true, storer); break; } case MessageContentType::Contact: { auto m = static_cast(content); store(m->contact, storer); break; } case MessageContentType::Document: { auto m = static_cast(content); td->documents_manager_->store_document(m->file_id, storer); store(m->caption, storer); break; } case MessageContentType::Game: { auto m = static_cast(content); store(m->game, storer); break; } case MessageContentType::Invoice: { auto m = static_cast(content); store(m->title, storer); store(m->description, storer); store(m->photo, storer); store(m->start_parameter, storer); store(m->invoice, storer); store(m->payload, storer); store(m->provider_token, storer); store(m->provider_data, storer); store(m->total_amount, storer); store(m->receipt_message_id, storer); break; } case MessageContentType::LiveLocation: { auto m = static_cast(content); store(m->location, storer); store(m->period, storer); break; } case MessageContentType::Location: { auto m = static_cast(content); store(m->location, storer); break; } case MessageContentType::Photo: { auto m = static_cast(content); store(m->photo, storer); store(m->caption, storer); break; } case MessageContentType::Sticker: { auto m = static_cast(content); td->stickers_manager_->store_sticker(m->file_id, false, storer); break; } case MessageContentType::Text: { auto m = static_cast(content); store(m->text, storer); store(m->web_page_id, storer); break; } case MessageContentType::Unsupported: { auto m = static_cast(content); store(m->version, storer); break; } case MessageContentType::Venue: { auto m = static_cast(content); store(m->venue, storer); break; } case MessageContentType::Video: { auto m = static_cast(content); td->videos_manager_->store_video(m->file_id, storer); store(m->caption, storer); break; } case MessageContentType::VideoNote: { auto m = static_cast(content); td->video_notes_manager_->store_video_note(m->file_id, storer); store(m->is_viewed, storer); break; } case MessageContentType::VoiceNote: { auto m = static_cast(content); td->voice_notes_manager_->store_voice_note(m->file_id, storer); store(m->caption, storer); store(m->is_listened, storer); break; } case MessageContentType::ChatCreate: { auto m = static_cast(content); store(m->title, storer); store(m->participant_user_ids, storer); break; } case MessageContentType::ChatChangeTitle: { auto m = static_cast(content); store(m->title, storer); break; } case MessageContentType::ChatChangePhoto: { auto m = static_cast(content); store(m->photo, storer); break; } case MessageContentType::ChatDeletePhoto: case MessageContentType::ChatDeleteHistory: break; case MessageContentType::ChatAddUsers: { auto m = static_cast(content); store(m->user_ids, storer); break; } case MessageContentType::ChatJoinedByLink: break; case MessageContentType::ChatDeleteUser: { auto m = static_cast(content); store(m->user_id, storer); break; } case MessageContentType::ChatMigrateTo: { auto m = static_cast(content); store(m->migrated_to_channel_id, storer); break; } case MessageContentType::ChannelCreate: { auto m = static_cast(content); store(m->title, storer); break; } case MessageContentType::ChannelMigrateFrom: { auto m = static_cast(content); store(m->title, storer); store(m->migrated_from_chat_id, storer); break; } case MessageContentType::PinMessage: { auto m = static_cast(content); store(m->message_id, storer); break; } case MessageContentType::GameScore: { auto m = static_cast(content); store(m->game_message_id, storer); store(m->game_id, storer); store(m->score, storer); break; } case MessageContentType::ScreenshotTaken: break; case MessageContentType::ChatSetTtl: { auto m = static_cast(content); store(m->ttl, storer); break; } case MessageContentType::Call: { auto m = static_cast(content); BEGIN_STORE_FLAGS(); STORE_FLAG(m->is_video); END_STORE_FLAGS(); store(m->call_id, storer); store(m->duration, storer); store(m->discard_reason, storer); break; } case MessageContentType::PaymentSuccessful: { auto m = static_cast(content); bool has_payload = !m->invoice_payload.empty(); bool has_shipping_option_id = !m->shipping_option_id.empty(); bool has_order_info = m->order_info != nullptr; bool has_telegram_payment_charge_id = !m->telegram_payment_charge_id.empty(); bool has_provider_payment_charge_id = !m->provider_payment_charge_id.empty(); bool has_invoice_message_id = m->invoice_message_id.is_valid(); BEGIN_STORE_FLAGS(); STORE_FLAG(has_payload); STORE_FLAG(has_shipping_option_id); STORE_FLAG(has_order_info); STORE_FLAG(has_telegram_payment_charge_id); STORE_FLAG(has_provider_payment_charge_id); STORE_FLAG(has_invoice_message_id); END_STORE_FLAGS(); store(m->currency, storer); store(m->total_amount, storer); if (has_payload) { store(m->total_amount, storer); } if (has_shipping_option_id) { store(m->invoice_payload, storer); } if (has_order_info) { store(m->order_info, storer); } if (has_telegram_payment_charge_id) { store(m->telegram_payment_charge_id, storer); } if (has_provider_payment_charge_id) { store(m->provider_payment_charge_id, storer); } if (has_invoice_message_id) { store(m->invoice_message_id, storer); } break; } case MessageContentType::ContactRegistered: break; case MessageContentType::ExpiredPhoto: break; case MessageContentType::ExpiredVideo: break; case MessageContentType::CustomServiceAction: { auto m = static_cast(content); store(m->message, storer); break; } case MessageContentType::WebsiteConnected: { auto m = static_cast(content); store(m->domain_name, storer); break; } case MessageContentType::PassportDataSent: { auto m = static_cast(content); store(m->types, storer); break; } case MessageContentType::PassportDataReceived: { auto m = static_cast(content); store(m->values, storer); store(m->credentials, storer); break; } case MessageContentType::Poll: { auto m = static_cast(content); store(m->poll_id, storer); break; } case MessageContentType::Dice: { auto m = static_cast(content); store(m->emoji, storer); store(m->dice_value, storer); break; } default: UNREACHABLE(); } } template static void parse_caption(FormattedText &caption, ParserT &parser) { parse(caption.text, parser); if (parser.version() >= static_cast(Version::AddCaptionEntities)) { parse(caption.entities, parser); } else { if (!check_utf8(caption.text)) { caption.text.clear(); } caption.entities = find_entities(caption.text, false); } } template static void parse(unique_ptr &content, ParserT &parser) { Td *td = parser.context()->td().get_actor_unsafe(); CHECK(td != nullptr); MessageContentType content_type; parse(content_type, parser); bool is_bad = false; switch (content_type) { case MessageContentType::Animation: { auto m = make_unique(); m->file_id = td->animations_manager_->parse_animation(parser); parse_caption(m->caption, parser); is_bad = !m->file_id.is_valid(); content = std::move(m); break; } case MessageContentType::Audio: { auto m = make_unique(); m->file_id = td->audios_manager_->parse_audio(parser); parse_caption(m->caption, parser); bool legacy_is_listened; parse(legacy_is_listened, parser); is_bad = !m->file_id.is_valid(); content = std::move(m); break; } case MessageContentType::Contact: { auto m = make_unique(); parse(m->contact, parser); content = std::move(m); break; } case MessageContentType::Document: { auto m = make_unique(); m->file_id = td->documents_manager_->parse_document(parser); parse_caption(m->caption, parser); is_bad = !m->file_id.is_valid(); content = std::move(m); break; } case MessageContentType::Game: { auto m = make_unique(); parse(m->game, parser); content = std::move(m); break; } case MessageContentType::Invoice: { auto m = make_unique(); parse(m->title, parser); parse(m->description, parser); parse(m->photo, parser); parse(m->start_parameter, parser); parse(m->invoice, parser); parse(m->payload, parser); parse(m->provider_token, parser); if (parser.version() >= static_cast(Version::AddMessageInvoiceProviderData)) { parse(m->provider_data, parser); } else { m->provider_data.clear(); } parse(m->total_amount, parser); parse(m->receipt_message_id, parser); content = std::move(m); break; } case MessageContentType::LiveLocation: { auto m = make_unique(); parse(m->location, parser); parse(m->period, parser); content = std::move(m); break; } case MessageContentType::Location: { auto m = make_unique(); parse(m->location, parser); content = std::move(m); break; } case MessageContentType::Photo: { auto m = make_unique(); parse(m->photo, parser); for (auto &photo_size : m->photo.photos) { if (!photo_size.file_id.is_valid()) { is_bad = true; } } parse_caption(m->caption, parser); content = std::move(m); break; } case MessageContentType::Sticker: { auto m = make_unique(); m->file_id = td->stickers_manager_->parse_sticker(false, parser); is_bad = !m->file_id.is_valid(); content = std::move(m); break; } case MessageContentType::Text: { auto m = make_unique(); parse(m->text, parser); parse(m->web_page_id, parser); content = std::move(m); break; } case MessageContentType::Unsupported: { auto m = make_unique(); if (parser.version() >= static_cast(Version::AddMessageUnsupportedVersion)) { parse(m->version, parser); } else { m->version = 0; } content = std::move(m); break; } case MessageContentType::Venue: { auto m = make_unique(); parse(m->venue, parser); content = std::move(m); break; } case MessageContentType::Video: { auto m = make_unique(); m->file_id = td->videos_manager_->parse_video(parser); parse_caption(m->caption, parser); is_bad = !m->file_id.is_valid(); content = std::move(m); break; } case MessageContentType::VideoNote: { auto m = make_unique(); m->file_id = td->video_notes_manager_->parse_video_note(parser); parse(m->is_viewed, parser); is_bad = !m->file_id.is_valid(); content = std::move(m); break; } case MessageContentType::VoiceNote: { auto m = make_unique(); m->file_id = td->voice_notes_manager_->parse_voice_note(parser); parse_caption(m->caption, parser); parse(m->is_listened, parser); is_bad = !m->file_id.is_valid(); content = std::move(m); break; } case MessageContentType::ChatCreate: { auto m = make_unique(); parse(m->title, parser); parse(m->participant_user_ids, parser); content = std::move(m); break; } case MessageContentType::ChatChangeTitle: { auto m = make_unique(); parse(m->title, parser); content = std::move(m); break; } case MessageContentType::ChatChangePhoto: { auto m = make_unique(); parse(m->photo, parser); content = std::move(m); break; } case MessageContentType::ChatDeletePhoto: content = make_unique(); break; case MessageContentType::ChatDeleteHistory: content = make_unique(); break; case MessageContentType::ChatAddUsers: { auto m = make_unique(); parse(m->user_ids, parser); content = std::move(m); break; } case MessageContentType::ChatJoinedByLink: content = make_unique(); break; case MessageContentType::ChatDeleteUser: { auto m = make_unique(); parse(m->user_id, parser); content = std::move(m); break; } case MessageContentType::ChatMigrateTo: { auto m = make_unique(); parse(m->migrated_to_channel_id, parser); content = std::move(m); break; } case MessageContentType::ChannelCreate: { auto m = make_unique(); parse(m->title, parser); content = std::move(m); break; } case MessageContentType::ChannelMigrateFrom: { auto m = make_unique(); parse(m->title, parser); parse(m->migrated_from_chat_id, parser); content = std::move(m); break; } case MessageContentType::PinMessage: { auto m = make_unique(); parse(m->message_id, parser); content = std::move(m); break; } case MessageContentType::GameScore: { auto m = make_unique(); parse(m->game_message_id, parser); parse(m->game_id, parser); parse(m->score, parser); content = std::move(m); break; } case MessageContentType::ScreenshotTaken: content = make_unique(); break; case MessageContentType::ChatSetTtl: { auto m = make_unique(); parse(m->ttl, parser); content = std::move(m); break; } case MessageContentType::Call: { auto m = make_unique(); if (parser.version() >= static_cast(Version::AddVideoCallsSupport)) { BEGIN_PARSE_FLAGS(); PARSE_FLAG(m->is_video); END_PARSE_FLAGS(); } else { m->is_video = false; } parse(m->call_id, parser); parse(m->duration, parser); parse(m->discard_reason, parser); content = std::move(m); break; } case MessageContentType::PaymentSuccessful: { auto m = make_unique(); bool has_payload; bool has_shipping_option_id; bool has_order_info; bool has_telegram_payment_charge_id; bool has_provider_payment_charge_id; bool has_invoice_message_id; BEGIN_PARSE_FLAGS(); PARSE_FLAG(has_payload); PARSE_FLAG(has_shipping_option_id); PARSE_FLAG(has_order_info); PARSE_FLAG(has_telegram_payment_charge_id); PARSE_FLAG(has_provider_payment_charge_id); PARSE_FLAG(has_invoice_message_id); END_PARSE_FLAGS(); parse(m->currency, parser); parse(m->total_amount, parser); if (has_payload) { parse(m->total_amount, parser); } if (has_shipping_option_id) { parse(m->invoice_payload, parser); } if (has_order_info) { parse(m->order_info, parser); } if (has_telegram_payment_charge_id) { parse(m->telegram_payment_charge_id, parser); } if (has_provider_payment_charge_id) { parse(m->provider_payment_charge_id, parser); } if (has_invoice_message_id) { parse(m->invoice_message_id, parser); } content = std::move(m); break; } case MessageContentType::ContactRegistered: content = make_unique(); break; case MessageContentType::ExpiredPhoto: content = make_unique(); break; case MessageContentType::ExpiredVideo: content = make_unique(); break; case MessageContentType::CustomServiceAction: { auto m = make_unique(); parse(m->message, parser); content = std::move(m); break; } case MessageContentType::WebsiteConnected: { auto m = make_unique(); parse(m->domain_name, parser); content = std::move(m); break; } case MessageContentType::PassportDataSent: { auto m = make_unique(); parse(m->types, parser); content = std::move(m); break; } case MessageContentType::PassportDataReceived: { auto m = make_unique(); parse(m->values, parser); parse(m->credentials, parser); content = std::move(m); break; } case MessageContentType::Poll: { auto m = make_unique(); parse(m->poll_id, parser); is_bad = !m->poll_id.is_valid(); content = std::move(m); break; } case MessageContentType::Dice: { auto m = make_unique(); if (parser.version() >= static_cast(Version::AddDiceEmoji)) { parse(m->emoji, parser); } else { m->emoji = MessageDice::DEFAULT_EMOJI; } parse(m->dice_value, parser); is_bad = m->dice_value < 0 || m->dice_value > 6; content = std::move(m); break; } default: LOG(FATAL) << "Have unknown message content type " << static_cast(content_type); } if (is_bad) { LOG(ERROR) << "Load a message with an invalid content of type " << content_type; content = make_unique(0); } } void store_message_content(const MessageContent *content, LogEventStorerCalcLength &storer) { store(content, storer); } void store_message_content(const MessageContent *content, LogEventStorerUnsafe &storer) { store(content, storer); } void parse_message_content(unique_ptr &content, LogEventParser &parser) { parse(content, parser); } InlineMessageContent create_inline_message_content(Td *td, FileId file_id, tl_object_ptr &&inline_message, int32 allowed_media_content_id, Photo *photo, Game *game) { CHECK(inline_message != nullptr); CHECK((allowed_media_content_id == td_api::inputMessagePhoto::ID) == (photo != nullptr)); CHECK((allowed_media_content_id == td_api::inputMessageGame::ID) == (game != nullptr)); CHECK((allowed_media_content_id != td_api::inputMessagePhoto::ID && allowed_media_content_id != td_api::inputMessageGame::ID && allowed_media_content_id != -1) == file_id.is_valid()); InlineMessageContent result; tl_object_ptr reply_markup; result.disable_web_page_preview = false; switch (inline_message->get_id()) { case telegram_api::botInlineMessageText::ID: { auto inline_message_text = move_tl_object_as(inline_message); auto entities = get_message_entities(td->contacts_manager_.get(), std::move(inline_message_text->entities_), "botInlineMessageText"); auto status = fix_formatted_text(inline_message_text->message_, entities, false, true, true, false); if (status.is_error()) { LOG(ERROR) << "Receive error " << status << " while parsing botInlineMessageText " << inline_message_text->message_; break; } result.disable_web_page_preview = (inline_message_text->flags_ & telegram_api::botInlineMessageText::NO_WEBPAGE_MASK) != 0; WebPageId web_page_id; if (!result.disable_web_page_preview) { web_page_id = td->web_pages_manager_->get_web_page_by_url(get_first_url(inline_message_text->message_, entities)); } result.message_content = make_unique( FormattedText{std::move(inline_message_text->message_), std::move(entities)}, web_page_id); reply_markup = std::move(inline_message_text->reply_markup_); break; } case telegram_api::botInlineMessageMediaGeo::ID: { auto inline_message_geo = move_tl_object_as(inline_message); if (inline_message_geo->period_ > 0) { result.message_content = make_unique(Location(inline_message_geo->geo_), inline_message_geo->period_); } else { result.message_content = make_unique(Location(inline_message_geo->geo_)); } reply_markup = std::move(inline_message_geo->reply_markup_); break; } case telegram_api::botInlineMessageMediaVenue::ID: { auto inline_message_venue = move_tl_object_as(inline_message); result.message_content = make_unique( Venue(inline_message_venue->geo_, std::move(inline_message_venue->title_), std::move(inline_message_venue->address_), std::move(inline_message_venue->provider_), std::move(inline_message_venue->venue_id_), std::move(inline_message_venue->venue_type_))); reply_markup = std::move(inline_message_venue->reply_markup_); break; } case telegram_api::botInlineMessageMediaContact::ID: { auto inline_message_contact = move_tl_object_as(inline_message); result.message_content = make_unique( Contact(std::move(inline_message_contact->phone_number_), std::move(inline_message_contact->first_name_), std::move(inline_message_contact->last_name_), std::move(inline_message_contact->vcard_), 0)); reply_markup = std::move(inline_message_contact->reply_markup_); break; } case telegram_api::botInlineMessageMediaAuto::ID: { auto input_message_media_auto = move_tl_object_as(inline_message); auto caption = get_message_text(td->contacts_manager_.get(), input_message_media_auto->message_, std::move(input_message_media_auto->entities_), true, 0, false, "register_inline_message_content"); if (allowed_media_content_id == td_api::inputMessageAnimation::ID) { result.message_content = make_unique(file_id, std::move(caption)); } else if (allowed_media_content_id == td_api::inputMessageAudio::ID) { result.message_content = make_unique(file_id, std::move(caption)); } else if (allowed_media_content_id == td_api::inputMessageDocument::ID) { result.message_content = make_unique(file_id, std::move(caption)); } else if (allowed_media_content_id == td_api::inputMessageGame::ID) { CHECK(game != nullptr); // TODO game->set_short_name(std::move(caption)); result.message_content = make_unique(std::move(*game)); } else if (allowed_media_content_id == td_api::inputMessagePhoto::ID) { result.message_content = make_unique(std::move(*photo), std::move(caption)); } else if (allowed_media_content_id == td_api::inputMessageSticker::ID) { result.message_content = make_unique(file_id); } else if (allowed_media_content_id == td_api::inputMessageVideo::ID) { result.message_content = make_unique(file_id, std::move(caption)); } else if (allowed_media_content_id == td_api::inputMessageVoiceNote::ID) { result.message_content = make_unique(file_id, std::move(caption), true); } else { LOG(WARNING) << "Unallowed bot inline message " << to_string(input_message_media_auto); } reply_markup = std::move(input_message_media_auto->reply_markup_); break; } default: UNREACHABLE(); } result.message_reply_markup = get_reply_markup(std::move(reply_markup), td->auth_manager_->is_bot(), true, false); return result; } unique_ptr create_text_message_content(string text, vector entities, WebPageId web_page_id) { return make_unique(FormattedText{std::move(text), std::move(entities)}, web_page_id); } unique_ptr create_contact_registered_message_content() { return make_unique(); } unique_ptr create_screenshot_taken_message_content() { return make_unique(); } unique_ptr create_chat_set_ttl_message_content(int32 ttl) { return make_unique(ttl); } static Result create_input_message_content( DialogId dialog_id, tl_object_ptr &&input_message_content, Td *td, FormattedText caption, FileId file_id, PhotoSize thumbnail, vector sticker_file_ids) { CHECK(input_message_content != nullptr); LOG(INFO) << "Create InputMessageContent with file " << file_id << " and thumbnail " << thumbnail.file_id; FileView file_view; string file_name; string mime_type; if (file_id.is_valid()) { file_view = td->file_manager_->get_file_view(file_id); auto suggested_name = file_view.suggested_name(); const PathView path_view(suggested_name); file_name = path_view.file_name().str(); mime_type = MimeType::from_extension(path_view.extension()); } bool disable_web_page_preview = false; bool clear_draft = false; unique_ptr content; UserId via_bot_user_id; int32 ttl = 0; bool is_bot = td->auth_manager_->is_bot(); switch (input_message_content->get_id()) { case td_api::inputMessageText::ID: { TRY_RESULT(input_message_text, process_input_message_text(td->contacts_manager_.get(), dialog_id, std::move(input_message_content), is_bot)); disable_web_page_preview = input_message_text.disable_web_page_preview; clear_draft = input_message_text.clear_draft; WebPageId web_page_id; bool can_add_web_page_previews = dialog_id.get_type() != DialogType::Channel || td->contacts_manager_->get_channel_permissions(dialog_id.get_channel_id()).can_add_web_page_previews(); if (!is_bot && !disable_web_page_preview && can_add_web_page_previews) { web_page_id = td->web_pages_manager_->get_web_page_by_url( get_first_url(input_message_text.text.text, input_message_text.text.entities)); } content = make_unique(std::move(input_message_text.text), web_page_id); break; } case td_api::inputMessageAnimation::ID: { auto input_animation = static_cast(input_message_content.get()); td->animations_manager_->create_animation( file_id, string(), thumbnail, std::move(file_name), std::move(mime_type), input_animation->duration_, get_dimensions(input_animation->width_, input_animation->height_), false); content = make_unique(file_id, std::move(caption)); break; } case td_api::inputMessageAudio::ID: { auto input_audio = static_cast(input_message_content.get()); if (!clean_input_string(input_audio->title_)) { return Status::Error(400, "Audio title must be encoded in UTF-8"); } if (!clean_input_string(input_audio->performer_)) { return Status::Error(400, "Audio performer must be encoded in UTF-8"); } td->audios_manager_->create_audio(file_id, string(), thumbnail, std::move(file_name), std::move(mime_type), input_audio->duration_, std::move(input_audio->title_), std::move(input_audio->performer_), false); content = make_unique(file_id, std::move(caption)); break; } case td_api::inputMessageDice::ID: { auto input_dice = static_cast(input_message_content.get()); if (!clean_input_string(input_dice->emoji_)) { return Status::Error(400, "Dice emoji must be encoded in UTF-8"); } content = td::make_unique(std::move(input_dice->emoji_), 0); clear_draft = input_dice->clear_draft_; break; } case td_api::inputMessageDocument::ID: td->documents_manager_->create_document(file_id, string(), thumbnail, std::move(file_name), std::move(mime_type), false); content = make_unique(file_id, std::move(caption)); break; case td_api::inputMessagePhoto::ID: { auto input_photo = static_cast(input_message_content.get()); if (input_photo->width_ < 0 || input_photo->width_ > 10000) { return Status::Error(400, "Wrong photo width"); } if (input_photo->height_ < 0 || input_photo->height_ > 10000) { return Status::Error(400, "Wrong photo height"); } ttl = input_photo->ttl_; auto message_photo = make_unique(); if (file_view.has_remote_location() && !file_view.remote_location().is_web()) { message_photo->photo.id = file_view.remote_location().get_id(); } message_photo->photo.date = G()->unix_time(); int32 type = 'i'; if (file_view.has_remote_location() && !file_view.remote_location().is_web()) { auto photo_size_source = file_view.remote_location().get_source(); if (photo_size_source.get_type() == PhotoSizeSource::Type::Thumbnail) { auto old_type = photo_size_source.thumbnail().thumbnail_type; if (old_type != 't') { type = old_type; } } } PhotoSize s; s.type = type; s.dimensions = get_dimensions(input_photo->width_, input_photo->height_); s.size = static_cast(file_view.size()); s.file_id = file_id; if (thumbnail.file_id.is_valid()) { message_photo->photo.photos.push_back(std::move(thumbnail)); } message_photo->photo.photos.push_back(s); message_photo->photo.has_stickers = !sticker_file_ids.empty(); message_photo->photo.sticker_file_ids = std::move(sticker_file_ids); message_photo->caption = std::move(caption); content = std::move(message_photo); break; } case td_api::inputMessageSticker::ID: { auto input_sticker = static_cast(input_message_content.get()); td->stickers_manager_->create_sticker(file_id, thumbnail, get_dimensions(input_sticker->width_, input_sticker->height_), nullptr, mime_type == "application/x-tgsticker", nullptr); content = make_unique(file_id); break; } case td_api::inputMessageVideo::ID: { auto input_video = static_cast(input_message_content.get()); ttl = input_video->ttl_; bool has_stickers = !sticker_file_ids.empty(); td->videos_manager_->create_video(file_id, string(), thumbnail, has_stickers, std::move(sticker_file_ids), std::move(file_name), std::move(mime_type), input_video->duration_, get_dimensions(input_video->width_, input_video->height_), input_video->supports_streaming_, false); content = make_unique(file_id, std::move(caption)); break; } case td_api::inputMessageVideoNote::ID: { auto input_video_note = static_cast(input_message_content.get()); auto length = input_video_note->length_; if (length < 0 || length >= 640) { return Status::Error(400, "Wrong video note length"); } td->video_notes_manager_->create_video_note(file_id, string(), thumbnail, input_video_note->duration_, get_dimensions(length, length), false); content = make_unique(file_id, false); break; } case td_api::inputMessageVoiceNote::ID: { auto input_voice_note = static_cast(input_message_content.get()); td->voice_notes_manager_->create_voice_note(file_id, std::move(mime_type), input_voice_note->duration_, std::move(input_voice_note->waveform_), false); content = make_unique(file_id, std::move(caption), false); break; } case td_api::inputMessageLocation::ID: { TRY_RESULT(location, process_input_message_location(std::move(input_message_content))); if (location.second == 0) { content = make_unique(std::move(location.first)); } else { content = make_unique(std::move(location.first), location.second); } break; } case td_api::inputMessageVenue::ID: { TRY_RESULT(venue, process_input_message_venue(std::move(input_message_content))); content = make_unique(std::move(venue)); break; } case td_api::inputMessageContact::ID: { TRY_RESULT(contact, process_input_message_contact(std::move(input_message_content))); content = make_unique(std::move(contact)); break; } case td_api::inputMessageGame::ID: { TRY_RESULT(game, process_input_message_game(td->contacts_manager_.get(), std::move(input_message_content))); via_bot_user_id = game.get_bot_user_id(); if (via_bot_user_id == td->contacts_manager_->get_my_id()) { via_bot_user_id = UserId(); } content = make_unique(std::move(game)); break; } case td_api::inputMessageInvoice::ID: { if (!is_bot) { return Status::Error(400, "Invoices can be sent only by bots"); } auto input_invoice = move_tl_object_as(input_message_content); if (!clean_input_string(input_invoice->title_)) { return Status::Error(400, "Invoice title must be encoded in UTF-8"); } if (!clean_input_string(input_invoice->description_)) { return Status::Error(400, "Invoice description must be encoded in UTF-8"); } if (!clean_input_string(input_invoice->photo_url_)) { return Status::Error(400, "Invoice photo URL must be encoded in UTF-8"); } if (!clean_input_string(input_invoice->start_parameter_)) { return Status::Error(400, "Invoice bot start parameter must be encoded in UTF-8"); } if (!clean_input_string(input_invoice->provider_token_)) { return Status::Error(400, "Invoice provider token must be encoded in UTF-8"); } if (!clean_input_string(input_invoice->provider_data_)) { return Status::Error(400, "Invoice provider data must be encoded in UTF-8"); } if (!clean_input_string(input_invoice->invoice_->currency_)) { return Status::Error(400, "Invoice currency must be encoded in UTF-8"); } auto message_invoice = make_unique(); message_invoice->title = std::move(input_invoice->title_); message_invoice->description = std::move(input_invoice->description_); auto r_http_url = parse_url(input_invoice->photo_url_); if (r_http_url.is_error()) { if (!input_invoice->photo_url_.empty()) { LOG(INFO) << "Can't register url " << input_invoice->photo_url_; } message_invoice->photo.id = -2; } else { auto url = r_http_url.ok().get_url(); auto r_invoice_file_id = td->file_manager_->from_persistent_id(url, FileType::Temp); if (r_invoice_file_id.is_error()) { LOG(INFO) << "Can't register url " << url; message_invoice->photo.id = -2; } else { auto invoice_file_id = r_invoice_file_id.move_as_ok(); PhotoSize s; s.type = 'u'; s.dimensions = get_dimensions(input_invoice->photo_width_, input_invoice->photo_height_); s.size = input_invoice->photo_size_; // TODO use invoice_file_id size s.file_id = invoice_file_id; message_invoice->photo.photos.push_back(s); } } message_invoice->start_parameter = std::move(input_invoice->start_parameter_); message_invoice->invoice.currency = std::move(input_invoice->invoice_->currency_); message_invoice->invoice.price_parts.reserve(input_invoice->invoice_->price_parts_.size()); int64 total_amount = 0; const int64 MAX_AMOUNT = 9999'9999'9999; for (auto &price : input_invoice->invoice_->price_parts_) { if (!clean_input_string(price->label_)) { return Status::Error(400, "Invoice price label must be encoded in UTF-8"); } message_invoice->invoice.price_parts.emplace_back(std::move(price->label_), price->amount_); if (price->amount_ < -MAX_AMOUNT || price->amount_ > MAX_AMOUNT) { return Status::Error(400, "Too big amount of currency specified"); } total_amount += price->amount_; } if (total_amount <= 0) { return Status::Error(400, "Total price must be positive"); } if (total_amount > MAX_AMOUNT) { return Status::Error(400, "Total price is too big"); } message_invoice->total_amount = total_amount; message_invoice->invoice.is_test = input_invoice->invoice_->is_test_; message_invoice->invoice.need_name = input_invoice->invoice_->need_name_; message_invoice->invoice.need_phone_number = input_invoice->invoice_->need_phone_number_; message_invoice->invoice.need_email_address = input_invoice->invoice_->need_email_address_; message_invoice->invoice.need_shipping_address = input_invoice->invoice_->need_shipping_address_; message_invoice->invoice.send_phone_number_to_provider = input_invoice->invoice_->send_phone_number_to_provider_; message_invoice->invoice.send_email_address_to_provider = input_invoice->invoice_->send_email_address_to_provider_; message_invoice->invoice.is_flexible = input_invoice->invoice_->is_flexible_; if (message_invoice->invoice.send_phone_number_to_provider) { message_invoice->invoice.need_phone_number = true; } if (message_invoice->invoice.send_email_address_to_provider) { message_invoice->invoice.need_email_address = true; } if (message_invoice->invoice.is_flexible) { message_invoice->invoice.need_shipping_address = true; } message_invoice->payload = std::move(input_invoice->payload_); message_invoice->provider_token = std::move(input_invoice->provider_token_); message_invoice->provider_data = std::move(input_invoice->provider_data_); content = std::move(message_invoice); break; } case td_api::inputMessagePoll::ID: { constexpr size_t MAX_POLL_QUESTION_LENGTH = 255; // server-side limit constexpr size_t MAX_POLL_OPTION_LENGTH = 100; // server-side limit constexpr size_t MAX_POLL_OPTIONS = 10; // server-side limit auto input_poll = static_cast(input_message_content.get()); if (!clean_input_string(input_poll->question_)) { return Status::Error(400, "Poll question must be encoded in UTF-8"); } if (input_poll->question_.empty()) { return Status::Error(400, "Poll question must be non-empty"); } if (utf8_length(input_poll->question_) > MAX_POLL_QUESTION_LENGTH) { return Status::Error(400, PSLICE() << "Poll question length must not exceed " << MAX_POLL_QUESTION_LENGTH); } if (input_poll->options_.size() <= 1) { return Status::Error(400, "Poll must have at least 2 option"); } if (input_poll->options_.size() > MAX_POLL_OPTIONS) { return Status::Error(400, PSLICE() << "Poll can't have more than " << MAX_POLL_OPTIONS << " options"); } for (auto &option : input_poll->options_) { if (!clean_input_string(option)) { return Status::Error(400, "Poll options must be encoded in UTF-8"); } if (option.empty()) { return Status::Error(400, "Poll options must be non-empty"); } if (utf8_length(option) > MAX_POLL_OPTION_LENGTH) { return Status::Error(400, PSLICE() << "Poll options length must not exceed " << MAX_POLL_OPTION_LENGTH); } } bool allow_multiple_answers = false; bool is_quiz = false; int32 correct_option_id = -1; FormattedText explanation; if (input_poll->type_ == nullptr) { return Status::Error(400, "Poll type must not be empty"); } switch (input_poll->type_->get_id()) { case td_api::pollTypeRegular::ID: { auto type = td_api::move_object_as(input_poll->type_); allow_multiple_answers = type->allow_multiple_answers_; break; } case td_api::pollTypeQuiz::ID: { auto type = td_api::move_object_as(input_poll->type_); is_quiz = true; correct_option_id = type->correct_option_id_; if (correct_option_id < 0 || correct_option_id >= static_cast(input_poll->options_.size())) { return Status::Error(400, "Wrong correct option ID specified"); } auto r_explanation = process_input_caption(td->contacts_manager_.get(), dialog_id, std::move(type->explanation_), is_bot); if (r_explanation.is_error()) { return r_explanation.move_as_error(); } explanation = r_explanation.move_as_ok(); break; } default: UNREACHABLE(); } int32 open_period = is_bot ? input_poll->open_period_ : 0; int32 close_date = is_bot ? input_poll->close_date_ : 0; if (open_period != 0) { close_date = 0; } bool is_closed = is_bot ? input_poll->is_closed_ : false; content = make_unique( td->poll_manager_->create_poll(std::move(input_poll->question_), std::move(input_poll->options_), input_poll->is_anonymous_, allow_multiple_answers, is_quiz, correct_option_id, std::move(explanation), open_period, close_date, is_closed)); break; } default: UNREACHABLE(); } return InputMessageContent{std::move(content), disable_web_page_preview, clear_draft, ttl, via_bot_user_id}; } Result get_input_message_content( DialogId dialog_id, tl_object_ptr &&input_message_content, Td *td) { bool is_secret = dialog_id.get_type() == DialogType::SecretChat; LOG(INFO) << "Get input message content from " << to_string(input_message_content); bool have_file = true; // TODO: send from secret chat to common Result r_file_id = Status::Error(500, "Have no file"); tl_object_ptr input_thumbnail; vector sticker_file_ids; switch (input_message_content->get_id()) { case td_api::inputMessageAnimation::ID: { auto input_message = static_cast(input_message_content.get()); r_file_id = td->file_manager_->get_input_file_id(FileType::Animation, input_message->animation_, dialog_id, false, is_secret, true); input_thumbnail = std::move(input_message->thumbnail_); break; } case td_api::inputMessageAudio::ID: { auto input_message = static_cast(input_message_content.get()); r_file_id = td->file_manager_->get_input_file_id(FileType::Audio, input_message->audio_, dialog_id, false, is_secret); input_thumbnail = std::move(input_message->album_cover_thumbnail_); break; } case td_api::inputMessageDocument::ID: { auto input_message = static_cast(input_message_content.get()); r_file_id = td->file_manager_->get_input_file_id(FileType::Document, input_message->document_, dialog_id, false, is_secret, true); input_thumbnail = std::move(input_message->thumbnail_); break; } case td_api::inputMessagePhoto::ID: { auto input_message = static_cast(input_message_content.get()); r_file_id = td->file_manager_->get_input_file_id(FileType::Photo, input_message->photo_, dialog_id, false, is_secret); input_thumbnail = std::move(input_message->thumbnail_); if (!input_message->added_sticker_file_ids_.empty()) { sticker_file_ids = td->stickers_manager_->get_attached_sticker_file_ids(input_message->added_sticker_file_ids_); } break; } case td_api::inputMessageSticker::ID: { auto input_message = static_cast(input_message_content.get()); r_file_id = td->file_manager_->get_input_file_id(FileType::Sticker, input_message->sticker_, dialog_id, false, is_secret); input_thumbnail = std::move(input_message->thumbnail_); break; } case td_api::inputMessageVideo::ID: { auto input_message = static_cast(input_message_content.get()); r_file_id = td->file_manager_->get_input_file_id(FileType::Video, input_message->video_, dialog_id, false, is_secret); input_thumbnail = std::move(input_message->thumbnail_); if (!input_message->added_sticker_file_ids_.empty()) { sticker_file_ids = td->stickers_manager_->get_attached_sticker_file_ids(input_message->added_sticker_file_ids_); } break; } case td_api::inputMessageVideoNote::ID: { auto input_message = static_cast(input_message_content.get()); r_file_id = td->file_manager_->get_input_file_id(FileType::VideoNote, input_message->video_note_, dialog_id, false, is_secret); input_thumbnail = std::move(input_message->thumbnail_); break; } case td_api::inputMessageVoiceNote::ID: { auto input_message = static_cast(input_message_content.get()); r_file_id = td->file_manager_->get_input_file_id(FileType::VoiceNote, input_message->voice_note_, dialog_id, false, is_secret); break; } default: have_file = false; break; } // TODO is path of files must be stored in bytes instead of UTF-8 string? FileId file_id; if (have_file) { if (r_file_id.is_error()) { return Status::Error(7, r_file_id.error().message()); } file_id = r_file_id.ok(); CHECK(file_id.is_valid()); } PhotoSize thumbnail; if (input_thumbnail != nullptr) { auto r_thumbnail_file_id = td->file_manager_->get_input_thumbnail_file_id(input_thumbnail->thumbnail_, dialog_id, is_secret); if (r_thumbnail_file_id.is_error()) { LOG(WARNING) << "Ignore thumbnail file: " << r_thumbnail_file_id.error().message(); } else { thumbnail.type = 't'; thumbnail.dimensions = get_dimensions(input_thumbnail->width_, input_thumbnail->height_); thumbnail.file_id = r_thumbnail_file_id.ok(); CHECK(thumbnail.file_id.is_valid()); FileView thumbnail_file_view = td->file_manager_->get_file_view(thumbnail.file_id); if (thumbnail_file_view.has_remote_location()) { // TODO td->file_manager_->delete_remote_location(thumbnail.file_id); } } } TRY_RESULT(caption, process_input_caption(td->contacts_manager_.get(), dialog_id, extract_input_caption(input_message_content), td->auth_manager_->is_bot())); return create_input_message_content(dialog_id, std::move(input_message_content), td, std::move(caption), file_id, std::move(thumbnail), std::move(sticker_file_ids)); } bool can_have_input_media(const Td *td, const MessageContent *content) { switch (content->get_type()) { case MessageContentType::Game: return static_cast(content)->game.has_input_media(); case MessageContentType::Poll: return td->poll_manager_->has_input_media(static_cast(content)->poll_id); 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: return false; case MessageContentType::Animation: case MessageContentType::Audio: case MessageContentType::Contact: case MessageContentType::Dice: case MessageContentType::Document: case MessageContentType::Invoice: case MessageContentType::LiveLocation: case MessageContentType::Location: case MessageContentType::Photo: case MessageContentType::Sticker: case MessageContentType::Text: case MessageContentType::Venue: case MessageContentType::Video: case MessageContentType::VideoNote: case MessageContentType::VoiceNote: return true; default: UNREACHABLE(); } } SecretInputMedia get_secret_input_media(const MessageContent *content, Td *td, tl_object_ptr input_file, BufferSlice thumbnail, int32 layer) { switch (content->get_type()) { case MessageContentType::Animation: { auto m = static_cast(content); return td->animations_manager_->get_secret_input_media(m->file_id, std::move(input_file), m->caption.text, std::move(thumbnail), layer); } case MessageContentType::Audio: { auto m = static_cast(content); return td->audios_manager_->get_secret_input_media(m->file_id, std::move(input_file), m->caption.text, std::move(thumbnail)); } case MessageContentType::Contact: { auto m = static_cast(content); return m->contact.get_secret_input_media_contact(); } case MessageContentType::Document: { auto m = static_cast(content); return td->documents_manager_->get_secret_input_media(m->file_id, std::move(input_file), m->caption.text, std::move(thumbnail)); } case MessageContentType::Location: { auto m = static_cast(content); return m->location.get_secret_input_media_geo_point(); } case MessageContentType::Photo: { auto m = static_cast(content); return photo_get_secret_input_media(td->file_manager_.get(), m->photo, std::move(input_file), m->caption.text, std::move(thumbnail)); } case MessageContentType::Sticker: { auto m = static_cast(content); return td->stickers_manager_->get_secret_input_media(m->file_id, std::move(input_file), std::move(thumbnail)); } case MessageContentType::Text: { CHECK(input_file == nullptr); CHECK(thumbnail.empty()); auto m = static_cast(content); return td->web_pages_manager_->get_secret_input_media(m->web_page_id); } case MessageContentType::Venue: { auto m = static_cast(content); return m->venue.get_secret_input_media_venue(); } case MessageContentType::Video: { auto m = static_cast(content); return td->videos_manager_->get_secret_input_media(m->file_id, std::move(input_file), m->caption.text, std::move(thumbnail)); } case MessageContentType::VideoNote: { auto m = static_cast(content); return td->video_notes_manager_->get_secret_input_media(m->file_id, std::move(input_file), std::move(thumbnail), layer); } case MessageContentType::VoiceNote: { auto m = static_cast(content); return td->voice_notes_manager_->get_secret_input_media(m->file_id, std::move(input_file), m->caption.text); } case MessageContentType::Call: case MessageContentType::Dice: case MessageContentType::Game: case MessageContentType::Invoice: case MessageContentType::LiveLocation: case MessageContentType::Poll: 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::PaymentSuccessful: case MessageContentType::ContactRegistered: case MessageContentType::ExpiredPhoto: case MessageContentType::ExpiredVideo: case MessageContentType::CustomServiceAction: case MessageContentType::WebsiteConnected: case MessageContentType::PassportDataSent: case MessageContentType::PassportDataReceived: break; default: UNREACHABLE(); } return SecretInputMedia{}; } static tl_object_ptr get_input_invoice(const Invoice &invoice) { int32 flags = 0; if (invoice.is_test) { flags |= telegram_api::invoice::TEST_MASK; } if (invoice.need_name) { flags |= telegram_api::invoice::NAME_REQUESTED_MASK; } if (invoice.need_phone_number) { flags |= telegram_api::invoice::PHONE_REQUESTED_MASK; } if (invoice.need_email_address) { flags |= telegram_api::invoice::EMAIL_REQUESTED_MASK; } if (invoice.need_shipping_address) { flags |= telegram_api::invoice::SHIPPING_ADDRESS_REQUESTED_MASK; } if (invoice.send_phone_number_to_provider) { flags |= telegram_api::invoice::PHONE_TO_PROVIDER_MASK; } if (invoice.send_email_address_to_provider) { flags |= telegram_api::invoice::EMAIL_TO_PROVIDER_MASK; } if (invoice.is_flexible) { flags |= telegram_api::invoice::FLEXIBLE_MASK; } auto prices = transform(invoice.price_parts, [](const LabeledPricePart &price) { return telegram_api::make_object(price.label, price.amount); }); return make_tl_object( flags, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, invoice.currency, std::move(prices)); } static tl_object_ptr get_input_web_document(const FileManager *file_manager, const Photo &photo) { if (photo.id == -2) { return nullptr; } CHECK(photo.photos.size() == 1); const PhotoSize &size = photo.photos[0]; CHECK(size.file_id.is_valid()); vector> attributes; if (size.dimensions.width != 0 && size.dimensions.height != 0) { attributes.push_back( make_tl_object(size.dimensions.width, size.dimensions.height)); } auto file_view = file_manager->get_file_view(size.file_id); CHECK(file_view.has_url()); auto file_name = get_url_file_name(file_view.url()); return make_tl_object( file_view.url(), size.size, MimeType::from_extension(PathView(file_name).extension(), "image/jpeg"), std::move(attributes)); } static tl_object_ptr get_input_media_invoice(const FileManager *file_manager, const MessageInvoice *message_invoice) { CHECK(message_invoice != nullptr); int32 flags = 0; auto input_web_document = get_input_web_document(file_manager, message_invoice->photo); if (input_web_document != nullptr) { flags |= telegram_api::inputMediaInvoice::PHOTO_MASK; } return make_tl_object( flags, message_invoice->title, message_invoice->description, std::move(input_web_document), get_input_invoice(message_invoice->invoice), BufferSlice(message_invoice->payload), message_invoice->provider_token, telegram_api::make_object( message_invoice->provider_data.empty() ? "null" : message_invoice->provider_data), message_invoice->start_parameter); } static tl_object_ptr get_input_media_impl( const MessageContent *content, Td *td, tl_object_ptr input_file, tl_object_ptr input_thumbnail, int32 ttl) { if (!can_have_input_media(td, content)) { return nullptr; } switch (content->get_type()) { case MessageContentType::Animation: { auto m = static_cast(content); return td->animations_manager_->get_input_media(m->file_id, std::move(input_file), std::move(input_thumbnail)); } case MessageContentType::Audio: { auto m = static_cast(content); return td->audios_manager_->get_input_media(m->file_id, std::move(input_file), std::move(input_thumbnail)); } case MessageContentType::Contact: { auto m = static_cast(content); return m->contact.get_input_media_contact(); } case MessageContentType::Dice: { auto m = static_cast(content); return make_tl_object(m->emoji); } case MessageContentType::Document: { auto m = static_cast(content); return td->documents_manager_->get_input_media(m->file_id, std::move(input_file), std::move(input_thumbnail)); } case MessageContentType::Game: { auto m = static_cast(content); return m->game.get_input_media_game(td); } case MessageContentType::Invoice: { auto m = static_cast(content); return get_input_media_invoice(td->file_manager_.get(), m); } case MessageContentType::LiveLocation: { auto m = static_cast(content); int32 flags = telegram_api::inputMediaGeoLive::PERIOD_MASK; return make_tl_object(flags, false /*ignored*/, m->location.get_input_geo_point(), m->period); } case MessageContentType::Location: { auto m = static_cast(content); return m->location.get_input_media_geo_point(); } case MessageContentType::Photo: { auto m = static_cast(content); return photo_get_input_media(td->file_manager_.get(), m->photo, std::move(input_file), ttl); } case MessageContentType::Poll: { auto m = static_cast(content); return td->poll_manager_->get_input_media(m->poll_id); } case MessageContentType::Sticker: { auto m = static_cast(content); return td->stickers_manager_->get_input_media(m->file_id, std::move(input_file), std::move(input_thumbnail)); } case MessageContentType::Venue: { auto m = static_cast(content); return m->venue.get_input_media_venue(); } case MessageContentType::Video: { auto m = static_cast(content); return td->videos_manager_->get_input_media(m->file_id, std::move(input_file), std::move(input_thumbnail), ttl); } case MessageContentType::VideoNote: { auto m = static_cast(content); return td->video_notes_manager_->get_input_media(m->file_id, std::move(input_file), std::move(input_thumbnail)); } case MessageContentType::VoiceNote: { auto m = static_cast(content); return td->voice_notes_manager_->get_input_media(m->file_id, std::move(input_file)); } case MessageContentType::Text: 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: break; default: UNREACHABLE(); } return nullptr; } tl_object_ptr get_input_media(const MessageContent *content, Td *td, tl_object_ptr input_file, tl_object_ptr input_thumbnail, FileId file_id, FileId thumbnail_file_id, int32 ttl, bool force) { bool had_input_file = input_file != nullptr; bool had_input_thumbnail = input_thumbnail != nullptr; auto input_media = get_input_media_impl(content, td, std::move(input_file), std::move(input_thumbnail), ttl); auto was_uploaded = FileManager::extract_was_uploaded(input_media); if (had_input_file) { if (!was_uploaded) { // if we had InputFile, but has failed to use it, then we need to immediately cancel file upload // so the next upload with the same file can succeed CHECK(file_id.is_valid()); td->file_manager_->cancel_upload(file_id); if (had_input_thumbnail) { CHECK(thumbnail_file_id.is_valid()); td->file_manager_->cancel_upload(thumbnail_file_id); } } } else { CHECK(!had_input_thumbnail); } if (!was_uploaded) { auto file_reference = FileManager::extract_file_reference(input_media); if (file_reference == FileReferenceView::invalid_file_reference() && !force) { return nullptr; } } return input_media; } tl_object_ptr get_input_media(const MessageContent *content, Td *td, int32 ttl, bool force) { auto input_media = get_input_media_impl(content, td, nullptr, nullptr, ttl); auto file_reference = FileManager::extract_file_reference(input_media); if (file_reference == FileReferenceView::invalid_file_reference() && !force) { return nullptr; } return input_media; } void delete_message_content_thumbnail(MessageContent *content, Td *td) { switch (content->get_type()) { case MessageContentType::Animation: { auto m = static_cast(content); return td->animations_manager_->delete_animation_thumbnail(m->file_id); } case MessageContentType::Audio: { auto m = static_cast(content); return td->audios_manager_->delete_audio_thumbnail(m->file_id); } case MessageContentType::Document: { auto m = static_cast(content); return td->documents_manager_->delete_document_thumbnail(m->file_id); } case MessageContentType::Photo: { auto m = static_cast(content); return photo_delete_thumbnail(m->photo); } case MessageContentType::Sticker: { auto m = static_cast(content); return td->stickers_manager_->delete_sticker_thumbnail(m->file_id); } case MessageContentType::Video: { auto m = static_cast(content); return td->videos_manager_->delete_video_thumbnail(m->file_id); } case MessageContentType::VideoNote: { auto m = static_cast(content); return td->video_notes_manager_->delete_video_note_thumbnail(m->file_id); } case MessageContentType::Contact: case MessageContentType::Dice: case MessageContentType::Game: case MessageContentType::Invoice: case MessageContentType::LiveLocation: case MessageContentType::Location: case MessageContentType::Venue: case MessageContentType::VoiceNote: case MessageContentType::Text: 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::Poll: break; default: UNREACHABLE(); } } bool can_forward_message_content(const MessageContent *content) { auto content_type = content->get_type(); if (content_type == MessageContentType::Text) { auto *text = static_cast(content); return !is_empty_string(text->text.text); // text can't be empty in the new message } if (content_type == MessageContentType::Poll) { auto *poll = static_cast(content); return !PollManager::is_local_poll_id(poll->poll_id); } return !is_service_message_content(content_type) && content_type != MessageContentType::Unsupported && content_type != MessageContentType::ExpiredPhoto && content_type != MessageContentType::ExpiredVideo; } bool update_opened_message_content(MessageContent *content) { switch (content->get_type()) { case MessageContentType::VideoNote: { auto video_note_content = static_cast(content); if (video_note_content->is_viewed) { return false; } video_note_content->is_viewed = true; return true; } case MessageContentType::VoiceNote: { auto voice_note_content = static_cast(content); if (voice_note_content->is_listened) { return false; } voice_note_content->is_listened = true; return true; } default: return false; } } static int32 get_message_content_text_index_mask(const MessageContent *content) { const FormattedText *text = get_message_content_text(content); if (text == nullptr || content->get_type() == MessageContentType::Game) { return 0; } for (auto &entity : text->entities) { if (entity.type == MessageEntity::Type::Url || entity.type == MessageEntity::Type::EmailAddress || entity.type == MessageEntity::Type::TextUrl) { return search_messages_filter_index_mask(SearchMessagesFilter::Url); } } return 0; } static int32 get_message_content_media_index_mask(const MessageContent *content, const Td *td, bool is_secret, bool is_outgoing) { switch (content->get_type()) { case MessageContentType::Animation: return search_messages_filter_index_mask(SearchMessagesFilter::Animation); case MessageContentType::Audio: { auto message_audio = static_cast(content); auto duration = td->audios_manager_->get_audio_duration(message_audio->file_id); return is_secret || duration > 0 ? search_messages_filter_index_mask(SearchMessagesFilter::Audio) : search_messages_filter_index_mask(SearchMessagesFilter::Document); } case MessageContentType::Document: return search_messages_filter_index_mask(SearchMessagesFilter::Document); case MessageContentType::Photo: return search_messages_filter_index_mask(SearchMessagesFilter::Photo) | search_messages_filter_index_mask(SearchMessagesFilter::PhotoAndVideo); case MessageContentType::Video: { auto message_video = static_cast(content); auto duration = td->videos_manager_->get_video_duration(message_video->file_id); return is_secret || duration > 0 ? search_messages_filter_index_mask(SearchMessagesFilter::Video) | search_messages_filter_index_mask(SearchMessagesFilter::PhotoAndVideo) : search_messages_filter_index_mask(SearchMessagesFilter::Document); } case MessageContentType::VideoNote: { auto message_video_note = static_cast(content); auto duration = td->video_notes_manager_->get_video_note_duration(message_video_note->file_id); return is_secret || duration > 0 ? search_messages_filter_index_mask(SearchMessagesFilter::VideoNote) | search_messages_filter_index_mask(SearchMessagesFilter::VoiceAndVideoNote) : search_messages_filter_index_mask(SearchMessagesFilter::Document); } case MessageContentType::VoiceNote: return search_messages_filter_index_mask(SearchMessagesFilter::VoiceNote) | search_messages_filter_index_mask(SearchMessagesFilter::VoiceAndVideoNote); case MessageContentType::ChatChangePhoto: return search_messages_filter_index_mask(SearchMessagesFilter::ChatPhoto); case MessageContentType::Call: { int32 index_mask = search_messages_filter_index_mask(SearchMessagesFilter::Call); auto message_call = static_cast(content); if (!is_outgoing && (message_call->discard_reason == CallDiscardReason::Declined || message_call->discard_reason == CallDiscardReason::Missed)) { index_mask |= search_messages_filter_index_mask(SearchMessagesFilter::MissedCall); } return index_mask; } case MessageContentType::Text: case MessageContentType::Contact: case MessageContentType::Game: case MessageContentType::Invoice: case MessageContentType::LiveLocation: case MessageContentType::Location: case MessageContentType::Sticker: case MessageContentType::Unsupported: case MessageContentType::Venue: case MessageContentType::ChatCreate: case MessageContentType::ChatChangeTitle: 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::PaymentSuccessful: case MessageContentType::ContactRegistered: case MessageContentType::ExpiredPhoto: case MessageContentType::ExpiredVideo: case MessageContentType::CustomServiceAction: case MessageContentType::WebsiteConnected: case MessageContentType::PassportDataSent: case MessageContentType::PassportDataReceived: case MessageContentType::Poll: case MessageContentType::Dice: return 0; default: UNREACHABLE(); return 0; } return 0; } int32 get_message_content_index_mask(const MessageContent *content, const Td *td, bool is_secret, bool is_outgoing) { return get_message_content_text_index_mask(content) | get_message_content_media_index_mask(content, td, is_secret, is_outgoing); } MessageId get_message_content_pinned_message_id(const MessageContent *content) { switch (content->get_type()) { case MessageContentType::PinMessage: return static_cast(content)->message_id; default: return MessageId(); } } MessageId get_message_content_replied_message_id(const MessageContent *content) { switch (content->get_type()) { case MessageContentType::PinMessage: return static_cast(content)->message_id; case MessageContentType::GameScore: return static_cast(content)->game_message_id; case MessageContentType::PaymentSuccessful: return static_cast(content)->invoice_message_id; default: return MessageId(); } } vector get_message_content_added_user_ids(const MessageContent *content) { CHECK(content->get_type() == MessageContentType::ChatAddUsers); return static_cast(content)->user_ids; } UserId get_message_content_deleted_user_id(const MessageContent *content) { switch (content->get_type()) { case MessageContentType::ChatDeleteUser: return static_cast(content)->user_id; default: return UserId(); } } int32 get_message_content_live_location_period(const MessageContent *content) { switch (content->get_type()) { case MessageContentType::LiveLocation: return static_cast(content)->period; default: return 0; } } bool get_message_content_poll_is_closed(const Td *td, const MessageContent *content) { switch (content->get_type()) { case MessageContentType::Poll: return td->poll_manager_->get_poll_is_closed(static_cast(content)->poll_id); default: return true; } } bool get_message_content_poll_is_anonymous(const Td *td, const MessageContent *content) { switch (content->get_type()) { case MessageContentType::Poll: return td->poll_manager_->get_poll_is_anonymous(static_cast(content)->poll_id); default: return true; } } bool has_message_content_web_page(const MessageContent *content) { if (content->get_type() == MessageContentType::Text) { return static_cast(content)->web_page_id.is_valid(); } return false; } void remove_message_content_web_page(MessageContent *content) { CHECK(content->get_type() == MessageContentType::Text); static_cast(content)->web_page_id = WebPageId(); } void set_message_content_poll_answer(Td *td, const MessageContent *content, FullMessageId full_message_id, vector &&option_ids, Promise &&promise) { CHECK(content->get_type() == MessageContentType::Poll); td->poll_manager_->set_poll_answer(static_cast(content)->poll_id, full_message_id, std::move(option_ids), std::move(promise)); } void get_message_content_poll_voters(Td *td, const MessageContent *content, FullMessageId full_message_id, int32 option_id, int32 offset, int32 limit, Promise>> &&promise) { CHECK(content->get_type() == MessageContentType::Poll); td->poll_manager_->get_poll_voters(static_cast(content)->poll_id, full_message_id, option_id, offset, limit, std::move(promise)); } void stop_message_content_poll(Td *td, const MessageContent *content, FullMessageId full_message_id, unique_ptr &&reply_markup, Promise &&promise) { CHECK(content->get_type() == MessageContentType::Poll); td->poll_manager_->stop_poll(static_cast(content)->poll_id, full_message_id, std::move(reply_markup), std::move(promise)); } static void merge_location_access_hash(const Location &first, const Location &second) { if (second.get_access_hash() != 0) { first.set_access_hash(second.get_access_hash()); } else { second.set_access_hash(first.get_access_hash()); } } static bool need_message_text_changed_warning(const MessageText *old_content, const MessageText *new_content) { if (new_content->text.text == "Unsupported characters" || new_content->text.text == "This channel is blocked because it was used to spread pornographic content.") { // message contained unsupported characters, text is replaced return false; } if (/* old_message->message_id.is_yet_unsent() && */ !old_content->text.entities.empty() && old_content->text.entities[0].offset == 0 && (new_content->text.entities.empty() || new_content->text.entities[0].offset != 0) && old_content->text.text != new_content->text.text && ends_with(old_content->text.text, new_content->text.text)) { // server has deleted first entity and ltrim the message return false; } return true; } static bool need_message_entities_changed_warning(const vector &old_entities, const vector &new_entities) { size_t old_pos = 0; size_t new_pos = 0; // compare entities, skipping some known to be different while (old_pos < old_entities.size() || new_pos < new_entities.size()) { // TODO remove after find_phone_numbers is implemented while (new_pos < new_entities.size() && new_entities[new_pos].type == MessageEntity::Type::PhoneNumber) { new_pos++; } if (old_pos < old_entities.size() && new_pos < new_entities.size() && old_entities[old_pos] == new_entities[new_pos]) { old_pos++; new_pos++; continue; } if (old_pos < old_entities.size() && old_entities[old_pos].type == MessageEntity::Type::MentionName) { // server could delete sime MentionName entities old_pos++; continue; } if (old_pos < old_entities.size() || new_pos < new_entities.size()) { return true; } } return false; } void merge_message_contents(Td *td, const MessageContent *old_content, MessageContent *new_content, bool need_message_changed_warning, DialogId dialog_id, bool need_merge_files, bool &is_content_changed, bool &need_update) { MessageContentType content_type = new_content->get_type(); CHECK(old_content->get_type() == content_type); switch (content_type) { case MessageContentType::Text: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->text.text != new_->text.text) { if (need_message_changed_warning && need_message_text_changed_warning(old_, new_)) { LOG(ERROR) << "Message text has changed from " << to_string(get_message_content_object(old_content, td, -1, false)) << ". New content is " << to_string(get_message_content_object(new_content, td, -1, false)); } need_update = true; } if (old_->text.entities != new_->text.entities) { const int32 MAX_CUSTOM_ENTITIES_COUNT = 100; // server-side limit if (need_message_changed_warning && need_message_text_changed_warning(old_, new_) && old_->text.entities.size() <= MAX_CUSTOM_ENTITIES_COUNT && need_message_entities_changed_warning(old_->text.entities, new_->text.entities)) { LOG(WARNING) << "Entities has changed from " << to_string(get_message_content_object(old_content, td, -1, false)) << ". New content is " << to_string(get_message_content_object(new_content, td, -1, false)); } need_update = true; } if (old_->web_page_id != new_->web_page_id) { LOG(INFO) << "Old: " << old_->web_page_id << ", new: " << new_->web_page_id; is_content_changed = true; need_update |= td->web_pages_manager_->have_web_page(old_->web_page_id) || td->web_pages_manager_->have_web_page(new_->web_page_id); } break; } case MessageContentType::Animation: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (new_->file_id != old_->file_id && (!need_merge_files || td->animations_manager_->merge_animations(new_->file_id, old_->file_id, false))) { need_update = true; } if (old_->caption != new_->caption) { need_update = true; } break; } case MessageContentType::Audio: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (new_->file_id != old_->file_id && (!need_merge_files || td->audios_manager_->merge_audios(new_->file_id, old_->file_id, false))) { need_update = true; } if (old_->caption != new_->caption) { need_update = true; } break; } case MessageContentType::Contact: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->contact != new_->contact) { need_update = true; } break; } case MessageContentType::Document: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (new_->file_id != old_->file_id && (!need_merge_files || td->documents_manager_->merge_documents(new_->file_id, old_->file_id, false))) { need_update = true; } if (old_->caption != new_->caption) { need_update = true; } break; } case MessageContentType::Game: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->game != new_->game) { need_update = true; } break; } case MessageContentType::Invoice: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->title != new_->title || old_->description != new_->description || old_->photo != new_->photo || old_->start_parameter != new_->start_parameter || old_->invoice != new_->invoice || old_->total_amount != new_->total_amount || old_->receipt_message_id != new_->receipt_message_id) { need_update = true; } if (old_->payload != new_->payload || old_->provider_token != new_->provider_token || old_->provider_data != new_->provider_data) { is_content_changed = true; } break; } case MessageContentType::LiveLocation: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->location != new_->location) { need_update = true; } if (old_->period != new_->period) { need_update = true; } if (old_->location.get_access_hash() != new_->location.get_access_hash()) { is_content_changed = true; merge_location_access_hash(old_->location, new_->location); } break; } case MessageContentType::Location: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->location != new_->location) { need_update = true; } if (old_->location.get_access_hash() != new_->location.get_access_hash()) { is_content_changed = true; merge_location_access_hash(old_->location, new_->location); } break; } case MessageContentType::Photo: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); const Photo *old_photo = &old_->photo; Photo *new_photo = &new_->photo; if (old_photo->date != new_photo->date) { is_content_changed = true; } if (old_photo->id != new_photo->id || old_->caption != new_->caption) { need_update = true; } if (old_photo->minithumbnail != new_photo->minithumbnail) { need_update = true; } if (old_photo->photos != new_photo->photos) { LOG(DEBUG) << "Merge photos " << old_photo->photos << " and " << new_photo->photos << ", need_merge_files = " << need_merge_files; auto new_photos_size = new_photo->photos.size(); auto old_photos_size = old_photo->photos.size(); bool need_merge = false; if (need_merge_files && (old_photos_size == 1 || (old_photos_size == 2 && old_photo->photos[0].type == 't')) && old_photo->photos.back().type == 'i') { // first time get info about sent photo if (old_photos_size == 2) { new_photo->photos.push_back(old_photo->photos[0]); } new_photo->photos.push_back(old_photo->photos.back()); need_merge = true; } else { // get sent photo again if (old_photos_size == 2 + new_photos_size && old_photo->photos[new_photos_size].type == 't') { new_photo->photos.push_back(old_photo->photos[new_photos_size]); } if (old_photos_size == 1 + new_photo->photos.size() && old_photo->photos.back().type == 'i') { new_photo->photos.push_back(old_photo->photos.back()); need_merge = true; } if (old_photo->photos != new_photo->photos) { new_photo->photos.resize( new_photos_size); // return previous size, because we shouldn't add local photo sizes need_merge = false; need_update = true; } } LOG(DEBUG) << "Merge photos " << old_photo->photos << " and " << new_photo->photos << " with new photos size = " << new_photos_size << ", need_merge = " << need_merge << ", need_update = " << need_update; if (need_merge && new_photos_size != 0) { FileId old_file_id = get_message_content_upload_file_id(old_content); FileView old_file_view = td->file_manager_->get_file_view(old_file_id); FileId new_file_id = new_photo->photos[0].file_id; FileView new_file_view = td->file_manager_->get_file_view(new_file_id); CHECK(new_file_view.has_remote_location()); LOG(DEBUG) << "Trying to merge old file " << old_file_id << " and new file " << new_file_id; if (new_file_view.remote_location().is_web()) { LOG(ERROR) << "Have remote web photo location"; } else if (!old_file_view.has_remote_location() || old_file_view.main_remote_location().get_file_reference() != new_file_view.remote_location().get_file_reference() || old_file_view.main_remote_location().get_access_hash() != new_file_view.remote_location().get_access_hash()) { FileId file_id = td->file_manager_->register_remote( FullRemoteFileLocation({FileType::Photo, 'i'}, new_file_view.remote_location().get_id(), new_file_view.remote_location().get_access_hash(), 0, 0, DcId::invalid(), new_file_view.remote_location().get_file_reference().str()), FileLocationSource::FromServer, dialog_id, old_photo->photos.back().size, 0, ""); LOG_STATUS(td->file_manager_->merge(file_id, old_file_id)); } } } break; } case MessageContentType::Sticker: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (new_->file_id != old_->file_id && (!need_merge_files || td->stickers_manager_->merge_stickers(new_->file_id, old_->file_id, false))) { need_update = true; } break; } case MessageContentType::Venue: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->venue != new_->venue) { need_update = true; } if (old_->venue.location().get_access_hash() != new_->venue.location().get_access_hash()) { is_content_changed = true; merge_location_access_hash(old_->venue.location(), new_->venue.location()); } break; } case MessageContentType::Video: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (new_->file_id != old_->file_id && (!need_merge_files || td->videos_manager_->merge_videos(new_->file_id, old_->file_id, false))) { need_update = true; } if (old_->caption != new_->caption) { need_update = true; } break; } case MessageContentType::VideoNote: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (new_->file_id != old_->file_id && (!need_merge_files || td->video_notes_manager_->merge_video_notes(new_->file_id, old_->file_id, false))) { need_update = true; } if (old_->is_viewed != new_->is_viewed) { need_update = true; } break; } case MessageContentType::VoiceNote: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (new_->file_id != old_->file_id && (!need_merge_files || td->voice_notes_manager_->merge_voice_notes(new_->file_id, old_->file_id, false))) { need_update = true; } if (old_->caption != new_->caption) { need_update = true; } if (old_->is_listened != new_->is_listened) { need_update = true; } break; } case MessageContentType::ChatCreate: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->title != new_->title || old_->participant_user_ids != new_->participant_user_ids) { need_update = true; } break; } case MessageContentType::ChatChangeTitle: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->title != new_->title) { need_update = true; } break; } case MessageContentType::ChatChangePhoto: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->photo != new_->photo) { need_update = true; } break; } case MessageContentType::ChatDeletePhoto: break; case MessageContentType::ChatDeleteHistory: break; case MessageContentType::ChatAddUsers: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->user_ids != new_->user_ids) { need_update = true; } break; } case MessageContentType::ChatJoinedByLink: break; case MessageContentType::ChatDeleteUser: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->user_id != new_->user_id) { need_update = true; } break; } case MessageContentType::ChatMigrateTo: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->migrated_to_channel_id != new_->migrated_to_channel_id) { need_update = true; } break; } case MessageContentType::ChannelCreate: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->title != new_->title) { need_update = true; } break; } case MessageContentType::ChannelMigrateFrom: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->title != new_->title || old_->migrated_from_chat_id != new_->migrated_from_chat_id) { need_update = true; } break; } case MessageContentType::PinMessage: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->message_id != new_->message_id) { need_update = true; } break; } case MessageContentType::GameScore: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->game_message_id != new_->game_message_id || old_->game_id != new_->game_id || old_->score != new_->score) { need_update = true; } break; } case MessageContentType::ScreenshotTaken: break; case MessageContentType::ChatSetTtl: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->ttl != new_->ttl) { LOG(ERROR) << "Ttl has changed from " << old_->ttl << " to " << new_->ttl; need_update = true; } break; } case MessageContentType::Call: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->call_id != new_->call_id || old_->is_video != new_->is_video) { is_content_changed = true; } if (old_->duration != new_->duration || old_->discard_reason != new_->discard_reason) { need_update = true; } break; } case MessageContentType::PaymentSuccessful: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->invoice_message_id != new_->invoice_message_id || old_->currency != new_->currency || old_->total_amount != new_->total_amount || old_->invoice_payload != new_->invoice_payload || old_->shipping_option_id != new_->shipping_option_id || old_->telegram_payment_charge_id != new_->telegram_payment_charge_id || old_->provider_payment_charge_id != new_->provider_payment_charge_id || ((old_->order_info != nullptr || new_->order_info != nullptr) && (old_->order_info == nullptr || new_->order_info == nullptr || *old_->order_info != *new_->order_info))) { need_update = true; } break; } case MessageContentType::ContactRegistered: break; case MessageContentType::ExpiredPhoto: break; case MessageContentType::ExpiredVideo: break; case MessageContentType::CustomServiceAction: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->message != new_->message) { need_update = true; } break; } case MessageContentType::WebsiteConnected: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->domain_name != new_->domain_name) { need_update = true; } break; } case MessageContentType::PassportDataSent: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->types != new_->types) { need_update = true; } break; } case MessageContentType::PassportDataReceived: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->values != new_->values) { need_update = true; } if (old_->credentials != new_->credentials) { need_update = true; } break; } case MessageContentType::Poll: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->poll_id != new_->poll_id) { need_update = true; } break; } case MessageContentType::Dice: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->emoji != new_->emoji || old_->dice_value != new_->dice_value) { need_update = true; } break; } case MessageContentType::Unsupported: { auto old_ = static_cast(old_content); auto new_ = static_cast(new_content); if (old_->version != new_->version) { is_content_changed = true; } break; } default: UNREACHABLE(); break; } } bool merge_message_content_file_id(Td *td, MessageContent *message_content, FileId new_file_id) { if (!new_file_id.is_valid()) { return false; } LOG(INFO) << "Merge message content of a message with file " << new_file_id; MessageContentType content_type = message_content->get_type(); switch (content_type) { case MessageContentType::Animation: { auto content = static_cast(message_content); if (new_file_id != content->file_id) { td->animations_manager_->merge_animations(new_file_id, content->file_id, false); content->file_id = new_file_id; return true; } break; } case MessageContentType::Audio: { auto content = static_cast(message_content); if (new_file_id != content->file_id) { td->audios_manager_->merge_audios(new_file_id, content->file_id, false); content->file_id = new_file_id; return true; } break; } case MessageContentType::Document: { auto content = static_cast(message_content); if (new_file_id != content->file_id) { td->documents_manager_->merge_documents(new_file_id, content->file_id, false); content->file_id = new_file_id; return true; } break; } case MessageContentType::Photo: { auto content = static_cast(message_content); Photo *photo = &content->photo; if (!photo->photos.empty() && photo->photos.back().type == 'i') { FileId &old_file_id = photo->photos.back().file_id; if (old_file_id != new_file_id) { LOG_STATUS(td->file_manager_->merge(new_file_id, old_file_id)); old_file_id = new_file_id; return true; } } break; } case MessageContentType::Sticker: { auto content = static_cast(message_content); if (new_file_id != content->file_id) { td->stickers_manager_->merge_stickers(new_file_id, content->file_id, false); content->file_id = new_file_id; return true; } break; } case MessageContentType::Video: { auto content = static_cast(message_content); if (new_file_id != content->file_id) { td->videos_manager_->merge_videos(new_file_id, content->file_id, false); content->file_id = new_file_id; return true; } break; } case MessageContentType::VideoNote: { auto content = static_cast(message_content); if (new_file_id != content->file_id) { td->video_notes_manager_->merge_video_notes(new_file_id, content->file_id, false); content->file_id = new_file_id; return true; } break; } case MessageContentType::VoiceNote: { auto content = static_cast(message_content); if (new_file_id != content->file_id) { td->voice_notes_manager_->merge_voice_notes(new_file_id, content->file_id, false); content->file_id = new_file_id; return true; } break; } case MessageContentType::Contact: case MessageContentType::Game: case MessageContentType::Invoice: case MessageContentType::LiveLocation: case MessageContentType::Location: case MessageContentType::Text: case MessageContentType::Venue: 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::Unsupported: 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::Poll: case MessageContentType::Dice: LOG(ERROR) << "Receive new file " << new_file_id << " in a sent message of the type " << content_type; break; default: UNREACHABLE(); break; } return false; } void register_message_content(Td *td, const MessageContent *content, FullMessageId full_message_id, const char *source) { switch (content->get_type()) { case MessageContentType::Text: return td->web_pages_manager_->register_web_page(static_cast(content)->web_page_id, full_message_id, source); case MessageContentType::Poll: return td->poll_manager_->register_poll(static_cast(content)->poll_id, full_message_id, source); default: return; } } void reregister_message_content(Td *td, const MessageContent *old_content, const MessageContent *new_content, FullMessageId full_message_id, const char *source) { auto old_content_type = old_content->get_type(); auto new_content_type = new_content->get_type(); if (old_content_type == new_content_type) { switch (old_content_type) { case MessageContentType::Text: if (static_cast(old_content)->web_page_id == static_cast(new_content)->web_page_id) { return; } break; case MessageContentType::Poll: if (static_cast(old_content)->poll_id == static_cast(new_content)->poll_id) { return; } break; default: return; } } unregister_message_content(td, old_content, full_message_id, source); register_message_content(td, new_content, full_message_id, source); } void unregister_message_content(Td *td, const MessageContent *content, FullMessageId full_message_id, const char *source) { switch (content->get_type()) { case MessageContentType::Text: return td->web_pages_manager_->unregister_web_page(static_cast(content)->web_page_id, full_message_id, source); case MessageContentType::Poll: return td->poll_manager_->unregister_poll(static_cast(content)->poll_id, full_message_id, source); default: return; } } template static tl_object_ptr secret_to_telegram(FromT &from); // fileLocationUnavailable#7c596b46 volume_id:long local_id:int secret:long = FileLocation; static auto secret_to_telegram(secret_api::fileLocationUnavailable &file_location) { return make_tl_object(file_location.volume_id_, file_location.local_id_); } // fileLocation#53d69076 dc_id:int volume_id:long local_id:int secret:long = FileLocation; static auto secret_to_telegram(secret_api::fileLocation &file_location) { return make_tl_object(file_location.volume_id_, file_location.local_id_); } // photoSizeEmpty#e17e23c type:string = PhotoSize; static auto secret_to_telegram(secret_api::photoSizeEmpty &empty) { if (!clean_input_string(empty.type_)) { empty.type_.clear(); } return make_tl_object(empty.type_); } // photoSize#77bfb61b type:string location:FileLocation w:int h:int size:int = PhotoSize; static auto secret_to_telegram(secret_api::photoSize &photo_size) { if (!clean_input_string(photo_size.type_)) { photo_size.type_.clear(); } return make_tl_object( photo_size.type_, secret_to_telegram(*photo_size.location_), photo_size.w_, photo_size.h_, photo_size.size_); } // photoCachedSize#e9a734fa type:string location:FileLocation w:int h:int bytes:bytes = PhotoSize; static auto secret_to_telegram(secret_api::photoCachedSize &photo_size) { if (!clean_input_string(photo_size.type_)) { photo_size.type_.clear(); } return make_tl_object( photo_size.type_, secret_to_telegram(*photo_size.location_), photo_size.w_, photo_size.h_, photo_size.bytes_.clone()); } // documentAttributeImageSize #6c37c15c w:int h:int = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeImageSize &image_size) { return make_tl_object(image_size.w_, image_size.h_); } // documentAttributeAnimated #11b58939 = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeAnimated &animated) { return make_tl_object(); } // documentAttributeSticker23 #fb0a5727 = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeSticker23 &sticker) { return make_tl_object( 0, false /*ignored*/, "", make_tl_object(), nullptr); } static auto secret_to_telegram(secret_api::inputStickerSetEmpty &sticker_set) { return make_tl_object(); } static auto secret_to_telegram(secret_api::inputStickerSetShortName &sticker_set) { if (!clean_input_string(sticker_set.short_name_)) { sticker_set.short_name_.clear(); } return make_tl_object(sticker_set.short_name_); } // documentAttributeSticker #3a556302 alt:string stickerset:InputStickerSet = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeSticker &sticker) { if (!clean_input_string(sticker.alt_)) { sticker.alt_.clear(); } return make_tl_object( 0, false /*ignored*/, sticker.alt_, secret_to_telegram(*sticker.stickerset_), nullptr); } // documentAttributeVideo #5910cccb duration:int w:int h:int = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeVideo &video) { return make_tl_object(0, false /*ignored*/, false /*ignored*/, video.duration_, video.w_, video.h_); } // documentAttributeFilename #15590068 file_name:string = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeFilename &filename) { if (!clean_input_string(filename.file_name_)) { filename.file_name_.clear(); } return make_tl_object(filename.file_name_); } // documentAttributeVideo66#ef02ce6 flags:# round_message:flags.0?true duration:int w:int h:int = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeVideo66 &video) { return make_tl_object( (video.flags_ & secret_api::documentAttributeVideo66::ROUND_MESSAGE_MASK) != 0 ? telegram_api::documentAttributeVideo::ROUND_MESSAGE_MASK : 0, video.round_message_, false, video.duration_, video.w_, video.h_); } static auto telegram_documentAttributeAudio(bool is_voice_note, int duration, string title, string performer, BufferSlice waveform) { if (!clean_input_string(title)) { title.clear(); } if (!clean_input_string(performer)) { performer.clear(); } int32 flags = 0; if (is_voice_note) { flags |= telegram_api::documentAttributeAudio::VOICE_MASK; } if (!title.empty()) { flags |= telegram_api::documentAttributeAudio::TITLE_MASK; } if (!performer.empty()) { flags |= telegram_api::documentAttributeAudio::PERFORMER_MASK; } if (waveform.size()) { flags |= telegram_api::documentAttributeAudio::WAVEFORM_MASK; } return make_tl_object(flags, is_voice_note, duration, std::move(title), std::move(performer), std::move(waveform)); } // documentAttributeAudio23 #51448e5 duration:int = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeAudio23 &audio) { return telegram_documentAttributeAudio(false, audio.duration_, "", "", Auto()); } // documentAttributeAudio45 #ded218e0 duration:int title:string performer:string = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeAudio45 &audio) { return telegram_documentAttributeAudio(false, audio.duration_, audio.title_, audio.performer_, Auto()); } // documentAttributeAudio#9852f9c6 flags:# voice:flags.10?true duration:int title:flags.0?string // performer:flags.1?string waveform:flags.2?bytes = DocumentAttribute; static auto secret_to_telegram(secret_api::documentAttributeAudio &audio) { return telegram_documentAttributeAudio((audio.flags_ & secret_api::documentAttributeAudio::VOICE_MASK) != 0, audio.duration_, audio.title_, audio.performer_, audio.waveform_.clone()); } static auto secret_to_telegram(std::vector> &attributes) { std::vector> res; for (auto &attribute : attributes) { auto telegram_attribute = secret_to_telegram(*attribute); if (telegram_attribute) { res.push_back(std::move(telegram_attribute)); } } return res; } // decryptedMessageMediaExternalDocument#fa95b0dd id:long access_hash:long date:int mime_type:string size:int // thumb:PhotoSize dc_id:int attributes:Vector = DecryptedMessageMedia; static auto secret_to_telegram_document(secret_api::decryptedMessageMediaExternalDocument &from) { if (!clean_input_string(from.mime_type_)) { from.mime_type_.clear(); } vector> thumbnails; thumbnails.push_back(secret_to_telegram(*from.thumb_)); return make_tl_object( telegram_api::document::THUMBS_MASK, from.id_, from.access_hash_, BufferSlice(), from.date_, from.mime_type_, from.size_, std::move(thumbnails), from.dc_id_, secret_to_telegram(from.attributes_)); } template static tl_object_ptr secret_to_telegram(FromT &from) { tl_object_ptr res; downcast_call(from, [&](auto &p) { res = secret_to_telegram(p); }); return res; } static unique_ptr get_document_message_content(Document &&parsed_document, FormattedText &&caption, bool is_opened) { auto file_id = parsed_document.file_id; if (!parsed_document.empty()) { CHECK(file_id.is_valid()); } switch (parsed_document.type) { case Document::Type::Animation: return make_unique(file_id, std::move(caption)); case Document::Type::Audio: return make_unique(file_id, std::move(caption)); case Document::Type::General: return make_unique(file_id, std::move(caption)); case Document::Type::Sticker: return make_unique(file_id); case Document::Type::Unknown: return make_unique(); case Document::Type::Video: return make_unique(file_id, std::move(caption)); case Document::Type::VideoNote: return make_unique(file_id, is_opened); case Document::Type::VoiceNote: return make_unique(file_id, std::move(caption), is_opened); default: UNREACHABLE(); return nullptr; } } static unique_ptr get_document_message_content(Td *td, tl_object_ptr &&document, DialogId owner_dialog_id, FormattedText &&caption, bool is_opened, MultiPromiseActor *load_data_multipromise_ptr) { return get_document_message_content( td->documents_manager_->on_get_document(std::move(document), owner_dialog_id, load_data_multipromise_ptr), std::move(caption), is_opened); } unique_ptr get_secret_message_content( Td *td, string message_text, tl_object_ptr file, tl_object_ptr &&media, vector> &&secret_entities, DialogId owner_dialog_id, MultiPromiseActor &load_data_multipromise) { int32 constructor_id = media == nullptr ? secret_api::decryptedMessageMediaEmpty::ID : media->get_id(); auto caption = [&] { switch (constructor_id) { case secret_api::decryptedMessageMediaVideo::ID: { auto video = static_cast(media.get()); return std::move(video->caption_); } case secret_api::decryptedMessageMediaPhoto::ID: { auto photo = static_cast(media.get()); return std::move(photo->caption_); } case secret_api::decryptedMessageMediaDocument::ID: { auto document = static_cast(media.get()); return std::move(document->caption_); } default: return string(); } }(); if (!clean_input_string(caption)) { caption.clear(); } if (message_text.empty()) { message_text = std::move(caption); } else if (!caption.empty()) { message_text = message_text + "\n\n" + caption; } auto entities = get_message_entities(std::move(secret_entities)); auto status = fix_formatted_text(message_text, entities, true, false, true, false); if (status.is_error()) { LOG(WARNING) << "Receive error " << status << " while parsing secret message \"" << message_text << "\" with entities " << format::as_array(entities); if (!clean_input_string(message_text)) { message_text.clear(); } entities = find_entities(message_text, true); } // support of old layer and old constructions switch (constructor_id) { case secret_api::decryptedMessageMediaVideo::ID: { auto video = move_tl_object_as(media); std::vector> attributes; attributes.emplace_back( make_tl_object(video->duration_, video->w_, video->h_)); media = make_tl_object( std::move(video->thumb_), video->thumb_w_, video->thumb_h_, video->mime_type_, video->size_, std::move(video->key_), std::move(video->iv_), std::move(attributes), string()); constructor_id = secret_api::decryptedMessageMediaDocument::ID; break; } } bool is_media_empty = false; switch (constructor_id) { case secret_api::decryptedMessageMediaEmpty::ID: if (message_text.empty()) { LOG(ERROR) << "Receive empty message text and media"; } is_media_empty = true; break; case secret_api::decryptedMessageMediaGeoPoint::ID: { auto message_geo_point = move_tl_object_as(media); auto m = make_unique(Location(std::move(message_geo_point))); if (m->location.empty()) { is_media_empty = true; break; } return std::move(m); } case secret_api::decryptedMessageMediaVenue::ID: { auto message_venue = move_tl_object_as(media); if (!clean_input_string(message_venue->title_)) { message_venue->title_.clear(); } if (!clean_input_string(message_venue->address_)) { message_venue->address_.clear(); } if (!clean_input_string(message_venue->provider_)) { message_venue->provider_.clear(); } if (!clean_input_string(message_venue->venue_id_)) { message_venue->venue_id_.clear(); } auto m = make_unique(Venue(Location(message_venue->lat_, message_venue->long_, 0), std::move(message_venue->title_), std::move(message_venue->address_), std::move(message_venue->provider_), std::move(message_venue->venue_id_), string())); if (m->venue.empty()) { is_media_empty = true; break; } return std::move(m); } case secret_api::decryptedMessageMediaContact::ID: { auto message_contact = move_tl_object_as(media); if (!clean_input_string(message_contact->phone_number_)) { message_contact->phone_number_.clear(); } if (!clean_input_string(message_contact->first_name_)) { message_contact->first_name_.clear(); } if (!clean_input_string(message_contact->last_name_)) { message_contact->last_name_.clear(); } return make_unique( Contact(std::move(message_contact->phone_number_), std::move(message_contact->first_name_), std::move(message_contact->last_name_), string(), message_contact->user_id_)); } case secret_api::decryptedMessageMediaWebPage::ID: { auto media_web_page = move_tl_object_as(media); if (!clean_input_string(media_web_page->url_)) { media_web_page->url_.clear(); } auto r_http_url = parse_url(media_web_page->url_); if (r_http_url.is_error()) { is_media_empty = true; break; } auto url = r_http_url.ok().get_url(); auto web_page_id = td->web_pages_manager_->get_web_page_by_url(url, load_data_multipromise.get_promise()); auto result = make_unique(FormattedText{std::move(message_text), std::move(entities)}, web_page_id); if (!result->web_page_id.is_valid()) { load_data_multipromise.add_promise( PromiseCreator::lambda([td, url, &web_page_id = result->web_page_id](Result result) { if (result.is_ok()) { web_page_id = td->web_pages_manager_->get_web_page_by_url(url); } })); } return std::move(result); } case secret_api::decryptedMessageMediaExternalDocument::ID: { auto external_document = move_tl_object_as(media); auto document = secret_to_telegram_document(*external_document); return get_document_message_content(td, std::move(document), owner_dialog_id, FormattedText{std::move(message_text), std::move(entities)}, false, &load_data_multipromise); } } if (file == nullptr && !is_media_empty) { LOG(ERROR) << "Received secret message with media, but without a file"; is_media_empty = true; } if (is_media_empty) { return create_text_message_content(std::move(message_text), std::move(entities), WebPageId()); } switch (constructor_id) { case secret_api::decryptedMessageMediaPhoto::ID: { auto message_photo = move_tl_object_as(media); return make_unique( get_encrypted_file_photo(td->file_manager_.get(), std::move(file), std::move(message_photo), owner_dialog_id), FormattedText{std::move(message_text), std::move(entities)}); } case secret_api::decryptedMessageMediaDocument::ID: { auto message_document = move_tl_object_as(media); if (!clean_input_string(message_document->mime_type_)) { message_document->mime_type_.clear(); } auto attributes = secret_to_telegram(message_document->attributes_); message_document->attributes_.clear(); auto document = td->documents_manager_->on_get_document( {std::move(file), std::move(message_document), std::move(attributes)}, owner_dialog_id); return get_document_message_content(std::move(document), {std::move(message_text), std::move(entities)}, false); } default: LOG(ERROR) << "Unsupported: " << to_string(media); return make_unique(); } } unique_ptr get_message_content(Td *td, FormattedText message, tl_object_ptr &&media, DialogId owner_dialog_id, bool is_content_read, UserId via_bot_user_id, int32 *ttl) { if (!td->auth_manager_->is_authorized() && !G()->close_flag() && media != nullptr) { LOG(ERROR) << "Receive without authorization " << to_string(media); media = nullptr; } int32 constructor_id = media == nullptr ? telegram_api::messageMediaEmpty::ID : media->get_id(); switch (constructor_id) { case telegram_api::messageMediaEmpty::ID: if (message.text.empty()) { LOG(ERROR) << "Receive empty message text and media for message from " << owner_dialog_id; } return make_unique(std::move(message), WebPageId()); case telegram_api::messageMediaPhoto::ID: { auto message_photo = move_tl_object_as(media); if ((message_photo->flags_ & telegram_api::messageMediaPhoto::PHOTO_MASK) == 0) { if ((message_photo->flags_ & telegram_api::messageMediaPhoto::TTL_SECONDS_MASK) == 0) { LOG(ERROR) << "Receive messageMediaPhoto without photo and TTL: " << oneline(to_string(message_photo)); break; } return make_unique(); } auto photo = get_photo(td->file_manager_.get(), std::move(message_photo->photo_), owner_dialog_id); if (photo.id == -2) { return make_unique(); } if (ttl != nullptr && (message_photo->flags_ & telegram_api::messageMediaPhoto::TTL_SECONDS_MASK) != 0) { *ttl = message_photo->ttl_seconds_; } return make_unique(std::move(photo), std::move(message)); } case telegram_api::messageMediaDice::ID: { auto message_dice = move_tl_object_as(media); auto m = td::make_unique(message_dice->emoticon_, message_dice->value_); if (m->dice_value < 0 || m->dice_value > 6) { break; } return std::move(m); } case telegram_api::messageMediaGeo::ID: { auto message_geo_point = move_tl_object_as(media); auto m = make_unique(Location(std::move(message_geo_point->geo_))); if (m->location.empty()) { break; } return std::move(m); } case telegram_api::messageMediaGeoLive::ID: { auto message_geo_point_live = move_tl_object_as(media); int32 period = message_geo_point_live->period_; auto location = Location(std::move(message_geo_point_live->geo_)); if (location.empty()) { break; } if (period <= 0) { LOG(ERROR) << "Receive wrong live location period = " << period; return make_unique(std::move(location)); } return make_unique(std::move(location), period); } case telegram_api::messageMediaVenue::ID: { auto message_venue = move_tl_object_as(media); auto m = make_unique(Venue(message_venue->geo_, std::move(message_venue->title_), std::move(message_venue->address_), std::move(message_venue->provider_), std::move(message_venue->venue_id_), std::move(message_venue->venue_type_))); if (m->venue.empty()) { break; } return std::move(m); } case telegram_api::messageMediaContact::ID: { auto message_contact = move_tl_object_as(media); if (message_contact->user_id_ != 0) { td->contacts_manager_->get_user_id_object(UserId(message_contact->user_id_), "messageMediaContact"); // to ensure updateUser } return make_unique(Contact( std::move(message_contact->phone_number_), std::move(message_contact->first_name_), std::move(message_contact->last_name_), std::move(message_contact->vcard_), message_contact->user_id_)); } case telegram_api::messageMediaDocument::ID: { auto message_document = move_tl_object_as(media); if ((message_document->flags_ & telegram_api::messageMediaDocument::DOCUMENT_MASK) == 0) { if ((message_document->flags_ & telegram_api::messageMediaDocument::TTL_SECONDS_MASK) == 0) { LOG(ERROR) << "Receive messageMediaDocument without document and TTL: " << oneline(to_string(message_document)); break; } return make_unique(); } auto document_ptr = std::move(message_document->document_); int32 document_id = document_ptr->get_id(); if (document_id == telegram_api::documentEmpty::ID) { break; } CHECK(document_id == telegram_api::document::ID); if (ttl != nullptr && (message_document->flags_ & telegram_api::messageMediaDocument::TTL_SECONDS_MASK) != 0) { *ttl = message_document->ttl_seconds_; } return get_document_message_content(td, move_tl_object_as(document_ptr), owner_dialog_id, std::move(message), is_content_read, nullptr); } case telegram_api::messageMediaGame::ID: { auto message_game = move_tl_object_as(media); auto m = make_unique(Game(td, std::move(message_game->game_), owner_dialog_id)); if (m->game.empty()) { break; } m->game.set_bot_user_id(via_bot_user_id); m->game.set_text(std::move(message)); return std::move(m); } case telegram_api::messageMediaInvoice::ID: { auto message_invoice = move_tl_object_as(media); MessageId receipt_message_id; if ((message_invoice->flags_ & telegram_api::messageMediaInvoice::RECEIPT_MSG_ID_MASK) != 0) { receipt_message_id = MessageId(ServerMessageId(message_invoice->receipt_msg_id_)); if (!receipt_message_id.is_valid()) { LOG(ERROR) << "Receive as receipt message " << receipt_message_id << " in " << owner_dialog_id; receipt_message_id = MessageId(); } } bool need_shipping_address = (message_invoice->flags_ & telegram_api::messageMediaInvoice::SHIPPING_ADDRESS_REQUESTED_MASK) != 0; bool is_test = (message_invoice->flags_ & telegram_api::messageMediaInvoice::TEST_MASK) != 0; return td::make_unique( std::move(message_invoice->title_), std::move(message_invoice->description_), get_web_document_photo(td->file_manager_.get(), std::move(message_invoice->photo_), owner_dialog_id), std::move(message_invoice->start_param_), message_invoice->total_amount_, std::move(message_invoice->currency_), is_test, need_shipping_address, receipt_message_id); } case telegram_api::messageMediaWebPage::ID: { auto media_web_page = move_tl_object_as(media); auto web_page_id = td->web_pages_manager_->on_get_web_page(std::move(media_web_page->webpage_), owner_dialog_id); return make_unique(std::move(message), web_page_id); } case telegram_api::messageMediaPoll::ID: { auto media_poll = move_tl_object_as(media); auto poll_id = td->poll_manager_->on_get_poll(PollId(), std::move(media_poll->poll_), std::move(media_poll->results_)); if (!poll_id.is_valid()) { break; } return make_unique(poll_id); } case telegram_api::messageMediaUnsupported::ID: return make_unique(); default: UNREACHABLE(); } // explicit empty media message return make_unique(std::move(message), WebPageId()); } unique_ptr dup_message_content(Td *td, DialogId dialog_id, const MessageContent *content, MessageContentDupType type) { CHECK(content != nullptr); if (type != MessageContentDupType::Forward && type != MessageContentDupType::SendViaBot && !can_have_input_media(td, content)) { return nullptr; } bool to_secret = dialog_id.get_type() == DialogType::SecretChat; auto fix_file_id = [dialog_id, to_secret, file_manager = td->file_manager_.get()](FileId file_id) { auto file_view = file_manager->get_file_view(file_id); if (to_secret && !file_view.is_encrypted_secret()) { auto download_file_id = file_manager->dup_file_id(file_id); file_id = file_manager ->register_generate(FileType::Encrypted, FileLocationSource::FromServer, file_view.suggested_name(), PSTRING() << "#file_id#" << download_file_id.get(), dialog_id, file_view.size()) .ok(); } return file_manager->dup_file_id(file_id); }; FileId thumbnail_file_id; if (to_secret) { thumbnail_file_id = get_message_content_thumbnail_file_id(content, td); } auto remove_caption = type == MessageContentDupType::CopyWithoutCaption; switch (content->get_type()) { case MessageContentType::Animation: { auto result = make_unique(*static_cast(content)); if (remove_caption) { result->caption = FormattedText(); } if (td->documents_manager_->has_input_media(result->file_id, thumbnail_file_id, to_secret)) { return std::move(result); } result->file_id = td->animations_manager_->dup_animation(fix_file_id(result->file_id), result->file_id); CHECK(result->file_id.is_valid()); return std::move(result); } case MessageContentType::Audio: { auto result = make_unique(*static_cast(content)); if (remove_caption) { result->caption = FormattedText(); } if (td->documents_manager_->has_input_media(result->file_id, thumbnail_file_id, to_secret)) { return std::move(result); } result->file_id = td->audios_manager_->dup_audio(fix_file_id(result->file_id), result->file_id); CHECK(result->file_id.is_valid()); return std::move(result); } case MessageContentType::Contact: return make_unique(*static_cast(content)); case MessageContentType::Dice: { auto result = td::make_unique(*static_cast(content)); if (type != MessageContentDupType::Forward) { result->dice_value = 0; } return std::move(result); } case MessageContentType::Document: { auto result = make_unique(*static_cast(content)); if (remove_caption) { result->caption = FormattedText(); } if (td->documents_manager_->has_input_media(result->file_id, thumbnail_file_id, to_secret)) { return std::move(result); } result->file_id = td->documents_manager_->dup_document(fix_file_id(result->file_id), result->file_id); CHECK(result->file_id.is_valid()); return std::move(result); } case MessageContentType::Game: return make_unique(*static_cast(content)); case MessageContentType::Invoice: return make_unique(*static_cast(content)); case MessageContentType::LiveLocation: if (!to_secret && (type == MessageContentDupType::Send || type == MessageContentDupType::SendViaBot)) { return make_unique(*static_cast(content)); } else { return make_unique(Location(static_cast(content)->location)); } case MessageContentType::Location: return make_unique(*static_cast(content)); case MessageContentType::Photo: { auto result = make_unique(*static_cast(content)); if (remove_caption) { result->caption = FormattedText(); } CHECK(!result->photo.photos.empty()); if ((result->photo.photos.size() > 2 || result->photo.photos.back().type != 'i') && !to_secret) { // already sent photo return std::move(result); } // Find 'i' or largest PhotoSize photo; for (const auto &size : result->photo.photos) { if (size.type == 'i') { photo = size; } } if (photo.type == 0) { for (const auto &size : result->photo.photos) { if (photo.type == 0 || photo < size) { photo = size; } } } // Find 't' or smallest PhotoSize thumbnail; for (const auto &size : result->photo.photos) { if (size.type == 't') { thumbnail = size; } } if (thumbnail.type == 0) { for (const auto &size : result->photo.photos) { if (size.type != photo.type && (thumbnail.type == 0 || size < thumbnail)) { thumbnail = size; } } } result->photo.photos.clear(); if (thumbnail.type != 0) { thumbnail.type = 't'; result->photo.photos.push_back(std::move(thumbnail)); } photo.type = 'i'; result->photo.photos.push_back(std::move(photo)); if (photo_has_input_media(td->file_manager_.get(), result->photo, to_secret)) { return std::move(result); } result->photo.photos.back().file_id = fix_file_id(result->photo.photos.back().file_id); if (thumbnail.type != 0) { result->photo.photos[0].file_id = td->file_manager_->dup_file_id(result->photo.photos[0].file_id); } return std::move(result); } case MessageContentType::Poll: return make_unique(*static_cast(content)); case MessageContentType::Sticker: { auto result = make_unique(*static_cast(content)); if (td->stickers_manager_->has_input_media(result->file_id, to_secret)) { return std::move(result); } result->file_id = td->stickers_manager_->dup_sticker(fix_file_id(result->file_id), result->file_id); CHECK(result->file_id.is_valid()); return std::move(result); } case MessageContentType::Text: return make_unique(*static_cast(content)); case MessageContentType::Venue: return make_unique(*static_cast(content)); case MessageContentType::Video: { auto result = make_unique(*static_cast(content)); if (remove_caption) { result->caption = FormattedText(); } if (td->documents_manager_->has_input_media(result->file_id, thumbnail_file_id, to_secret)) { return std::move(result); } result->file_id = td->videos_manager_->dup_video(fix_file_id(result->file_id), result->file_id); CHECK(result->file_id.is_valid()); return std::move(result); } case MessageContentType::VideoNote: { auto result = make_unique(*static_cast(content)); result->is_viewed = false; if (td->documents_manager_->has_input_media(result->file_id, thumbnail_file_id, to_secret)) { return std::move(result); } result->file_id = td->video_notes_manager_->dup_video_note(fix_file_id(result->file_id), result->file_id); CHECK(result->file_id.is_valid()); return std::move(result); } case MessageContentType::VoiceNote: { auto result = make_unique(*static_cast(content)); if (remove_caption) { result->caption = FormattedText(); } result->is_listened = false; if (td->documents_manager_->has_input_media(result->file_id, thumbnail_file_id, to_secret)) { return std::move(result); } result->file_id = td->voice_notes_manager_->dup_voice_note(fix_file_id(result->file_id), result->file_id); CHECK(result->file_id.is_valid()); return std::move(result); } 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: return nullptr; default: UNREACHABLE(); } UNREACHABLE(); return nullptr; } unique_ptr get_action_message_content(Td *td, tl_object_ptr &&action, DialogId owner_dialog_id, MessageId reply_to_message_id) { CHECK(action != nullptr); switch (action->get_id()) { case telegram_api::messageActionEmpty::ID: LOG(ERROR) << "Receive empty message action in " << owner_dialog_id; break; case telegram_api::messageActionChatCreate::ID: { auto chat_create = move_tl_object_as(action); vector participant_user_ids; participant_user_ids.reserve(chat_create->users_.size()); for (auto &user : chat_create->users_) { UserId user_id(user); if (user_id.is_valid()) { participant_user_ids.push_back(user_id); } else { LOG(ERROR) << "Receive messageActionChatCreate with invalid " << user_id << " in " << owner_dialog_id; } } return td::make_unique(std::move(chat_create->title_), std::move(participant_user_ids)); } case telegram_api::messageActionChatEditTitle::ID: { auto chat_edit_title = move_tl_object_as(action); return td::make_unique(std::move(chat_edit_title->title_)); } case telegram_api::messageActionChatEditPhoto::ID: { auto chat_edit_photo = move_tl_object_as(action); auto photo = get_photo(td->file_manager_.get(), std::move(chat_edit_photo->photo_), owner_dialog_id); if (photo.id == -2) { break; } return make_unique(std::move(photo)); } case telegram_api::messageActionChatDeletePhoto::ID: { return make_unique(); } case telegram_api::messageActionHistoryClear::ID: { return make_unique(); } case telegram_api::messageActionChatAddUser::ID: { auto chat_add_user = move_tl_object_as(action); vector user_ids; user_ids.reserve(chat_add_user->users_.size()); for (auto &user : chat_add_user->users_) { UserId user_id(user); if (user_id.is_valid()) { user_ids.push_back(user_id); } else { LOG(ERROR) << "Receive messageActionChatAddUser with invalid " << user_id << " in " << owner_dialog_id; } } return td::make_unique(std::move(user_ids)); } case telegram_api::messageActionChatJoinedByLink::ID: return make_unique(); case telegram_api::messageActionChatDeleteUser::ID: { auto chat_delete_user = move_tl_object_as(action); UserId user_id(chat_delete_user->user_id_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive messageActionChatDeleteUser with invalid " << user_id << " in " << owner_dialog_id; break; } return make_unique(user_id); } case telegram_api::messageActionChatMigrateTo::ID: { auto chat_migrate_to = move_tl_object_as(action); ChannelId migrated_to_channel_id(chat_migrate_to->channel_id_); if (!migrated_to_channel_id.is_valid()) { LOG(ERROR) << "Receive messageActionChatMigrateTo with invalid " << migrated_to_channel_id << " in " << owner_dialog_id; break; } return make_unique(migrated_to_channel_id); } case telegram_api::messageActionChannelCreate::ID: { auto channel_create = move_tl_object_as(action); return td::make_unique(std::move(channel_create->title_)); } case telegram_api::messageActionChannelMigrateFrom::ID: { auto channel_migrate_from = move_tl_object_as(action); ChatId chat_id(channel_migrate_from->chat_id_); LOG_IF(ERROR, !chat_id.is_valid()) << "Receive messageActionChannelMigrateFrom with invalid " << chat_id << " in " << owner_dialog_id; return td::make_unique(std::move(channel_migrate_from->title_), chat_id); } case telegram_api::messageActionPinMessage::ID: { if (!reply_to_message_id.is_valid()) { // possible in basic groups LOG(INFO) << "Receive pinned message with " << reply_to_message_id << " in " << owner_dialog_id; reply_to_message_id = MessageId(); } return make_unique(reply_to_message_id); } case telegram_api::messageActionGameScore::ID: { if (!reply_to_message_id.is_valid()) { // possible in basic groups LOG(INFO) << "Receive game score with " << reply_to_message_id << " in " << owner_dialog_id; reply_to_message_id = MessageId(); } auto game_score = move_tl_object_as(action); return make_unique(reply_to_message_id, game_score->game_id_, game_score->score_); } case telegram_api::messageActionPhoneCall::ID: { auto phone_call = move_tl_object_as(action); auto duration = (phone_call->flags_ & telegram_api::messageActionPhoneCall::DURATION_MASK) != 0 ? phone_call->duration_ : 0; auto is_video = (phone_call->flags_ & telegram_api::messageActionPhoneCall::VIDEO_MASK) != 0; return make_unique(phone_call->call_id_, duration, get_call_discard_reason(phone_call->reason_), is_video); } case telegram_api::messageActionPaymentSent::ID: { LOG_IF(ERROR, td->auth_manager_->is_bot()) << "Receive MessageActionPaymentSent in " << owner_dialog_id; if (!reply_to_message_id.is_valid()) { LOG(ERROR) << "Receive succesful payment message with " << reply_to_message_id << " in " << owner_dialog_id; reply_to_message_id = MessageId(); } auto payment_sent = move_tl_object_as(action); return td::make_unique(reply_to_message_id, std::move(payment_sent->currency_), payment_sent->total_amount_); } case telegram_api::messageActionPaymentSentMe::ID: { LOG_IF(ERROR, !td->auth_manager_->is_bot()) << "Receive MessageActionPaymentSentMe in " << owner_dialog_id; if (!reply_to_message_id.is_valid()) { LOG(ERROR) << "Receive succesful payment message with " << reply_to_message_id << " in " << owner_dialog_id; reply_to_message_id = MessageId(); } auto payment_sent = move_tl_object_as(action); auto result = td::make_unique(reply_to_message_id, std::move(payment_sent->currency_), payment_sent->total_amount_); result->invoice_payload = payment_sent->payload_.as_slice().str(); result->shipping_option_id = std::move(payment_sent->shipping_option_id_); result->order_info = get_order_info(std::move(payment_sent->info_)); result->telegram_payment_charge_id = std::move(payment_sent->charge_->id_); result->provider_payment_charge_id = std::move(payment_sent->charge_->provider_charge_id_); return std::move(result); } case telegram_api::messageActionScreenshotTaken::ID: { return make_unique(); } case telegram_api::messageActionCustomAction::ID: { auto custom_action = move_tl_object_as(action); return td::make_unique(std::move(custom_action->message_)); } case telegram_api::messageActionBotAllowed::ID: { auto bot_allowed = move_tl_object_as(action); return td::make_unique(std::move(bot_allowed->domain_)); } case telegram_api::messageActionSecureValuesSent::ID: { LOG_IF(ERROR, td->auth_manager_->is_bot()) << "Receive MessageActionSecureValuesSent in " << owner_dialog_id; auto secure_values = move_tl_object_as(action); return td::make_unique(get_secure_value_types(secure_values->types_)); } case telegram_api::messageActionSecureValuesSentMe::ID: { LOG_IF(ERROR, !td->auth_manager_->is_bot()) << "Receive MessageActionSecureValuesSentMe in " << owner_dialog_id; auto secure_values = move_tl_object_as(action); return td::make_unique( get_encrypted_secure_values(td->file_manager_.get(), std::move(secure_values->values_)), get_encrypted_secure_credentials(std::move(secure_values->credentials_))); } case telegram_api::messageActionContactSignUp::ID: { LOG_IF(ERROR, td->auth_manager_->is_bot()) << "Receive ContactRegistered in " << owner_dialog_id; return td::make_unique(); } default: UNREACHABLE(); } // explicit empty or wrong action return make_unique(FormattedText(), WebPageId()); } tl_object_ptr get_message_content_object(const MessageContent *content, Td *td, int32 message_date, bool is_content_secret) { CHECK(content != nullptr); switch (content->get_type()) { case MessageContentType::Animation: { const MessageAnimation *m = static_cast(content); return make_tl_object( td->animations_manager_->get_animation_object(m->file_id, "get_message_content_object"), get_formatted_text_object(m->caption), is_content_secret); } case MessageContentType::Audio: { const MessageAudio *m = static_cast(content); return make_tl_object(td->audios_manager_->get_audio_object(m->file_id), get_formatted_text_object(m->caption)); } case MessageContentType::Contact: { const MessageContact *m = static_cast(content); return make_tl_object(m->contact.get_contact_object()); } case MessageContentType::Document: { const MessageDocument *m = static_cast(content); return make_tl_object(td->documents_manager_->get_document_object(m->file_id), get_formatted_text_object(m->caption)); } case MessageContentType::Game: { const MessageGame *m = static_cast(content); return make_tl_object(m->game.get_game_object(td)); } case MessageContentType::Invoice: { const MessageInvoice *m = static_cast(content); return make_tl_object( m->title, m->description, get_photo_object(td->file_manager_.get(), &m->photo), m->invoice.currency, m->total_amount, m->start_parameter, m->invoice.is_test, m->invoice.need_shipping_address, m->receipt_message_id.get()); } case MessageContentType::LiveLocation: { const MessageLiveLocation *m = static_cast(content); auto passed = max(G()->unix_time_cached() - message_date, 0); return make_tl_object(m->location.get_location_object(), m->period, max(0, m->period - passed)); } case MessageContentType::Location: { const MessageLocation *m = static_cast(content); return make_tl_object(m->location.get_location_object(), 0, 0); } case MessageContentType::Photo: { const MessagePhoto *m = static_cast(content); return make_tl_object(get_photo_object(td->file_manager_.get(), &m->photo), get_formatted_text_object(m->caption), is_content_secret); } case MessageContentType::Sticker: { const MessageSticker *m = static_cast(content); return make_tl_object(td->stickers_manager_->get_sticker_object(m->file_id)); } case MessageContentType::Text: { const MessageText *m = static_cast(content); return make_tl_object(get_formatted_text_object(m->text), td->web_pages_manager_->get_web_page_object(m->web_page_id)); } case MessageContentType::Unsupported: return make_tl_object(); case MessageContentType::Venue: { const MessageVenue *m = static_cast(content); return make_tl_object(m->venue.get_venue_object()); } case MessageContentType::Video: { const MessageVideo *m = static_cast(content); return make_tl_object(td->videos_manager_->get_video_object(m->file_id), get_formatted_text_object(m->caption), is_content_secret); } case MessageContentType::VideoNote: { const MessageVideoNote *m = static_cast(content); return make_tl_object(td->video_notes_manager_->get_video_note_object(m->file_id), m->is_viewed, is_content_secret); } case MessageContentType::VoiceNote: { const MessageVoiceNote *m = static_cast(content); return make_tl_object(td->voice_notes_manager_->get_voice_note_object(m->file_id), get_formatted_text_object(m->caption), m->is_listened); } case MessageContentType::ChatCreate: { const MessageChatCreate *m = static_cast(content); return make_tl_object( m->title, td->contacts_manager_->get_user_ids_object(m->participant_user_ids, "MessageChatCreate")); } case MessageContentType::ChatChangeTitle: { const MessageChatChangeTitle *m = static_cast(content); return make_tl_object(m->title); } case MessageContentType::ChatChangePhoto: { const MessageChatChangePhoto *m = static_cast(content); return make_tl_object(get_photo_object(td->file_manager_.get(), &m->photo)); } case MessageContentType::ChatDeletePhoto: return make_tl_object(); case MessageContentType::ChatDeleteHistory: return make_tl_object(); case MessageContentType::ChatAddUsers: { const MessageChatAddUsers *m = static_cast(content); return make_tl_object( td->contacts_manager_->get_user_ids_object(m->user_ids, "MessageChatAddUsers")); } case MessageContentType::ChatJoinedByLink: return make_tl_object(); case MessageContentType::ChatDeleteUser: { const MessageChatDeleteUser *m = static_cast(content); return make_tl_object( td->contacts_manager_->get_user_id_object(m->user_id, "messageChatDeleteMember")); } case MessageContentType::ChatMigrateTo: { const MessageChatMigrateTo *m = static_cast(content); return make_tl_object( td->contacts_manager_->get_supergroup_id_object(m->migrated_to_channel_id, "messageChatUpgradeTo")); } case MessageContentType::ChannelCreate: { const MessageChannelCreate *m = static_cast(content); return make_tl_object(m->title); } case MessageContentType::ChannelMigrateFrom: { const MessageChannelMigrateFrom *m = static_cast(content); return make_tl_object( m->title, td->contacts_manager_->get_basic_group_id_object(m->migrated_from_chat_id, "messageChatUpgradeFrom")); } case MessageContentType::PinMessage: { const MessagePinMessage *m = static_cast(content); return make_tl_object(m->message_id.get()); } case MessageContentType::GameScore: { const MessageGameScore *m = static_cast(content); return make_tl_object(m->game_message_id.get(), m->game_id, m->score); } case MessageContentType::ScreenshotTaken: return make_tl_object(); case MessageContentType::ChatSetTtl: { const MessageChatSetTtl *m = static_cast(content); return make_tl_object(m->ttl); } case MessageContentType::Call: { const MessageCall *m = static_cast(content); return make_tl_object(get_call_discard_reason_object(m->discard_reason), m->duration); } case MessageContentType::PaymentSuccessful: { const MessagePaymentSuccessful *m = static_cast(content); if (td->auth_manager_->is_bot()) { return make_tl_object( m->invoice_message_id.get(), m->currency, m->total_amount, m->invoice_payload, m->shipping_option_id, get_order_info_object(m->order_info), m->telegram_payment_charge_id, m->provider_payment_charge_id); } else { return make_tl_object(m->invoice_message_id.get(), m->currency, m->total_amount); } } case MessageContentType::ContactRegistered: return make_tl_object(); case MessageContentType::ExpiredPhoto: return make_tl_object(); case MessageContentType::ExpiredVideo: return make_tl_object(); case MessageContentType::CustomServiceAction: { const MessageCustomServiceAction *m = static_cast(content); return make_tl_object(m->message); } case MessageContentType::WebsiteConnected: { const MessageWebsiteConnected *m = static_cast(content); return make_tl_object(m->domain_name); } case MessageContentType::PassportDataSent: { const MessagePassportDataSent *m = static_cast(content); return make_tl_object(get_passport_element_types_object(m->types)); } case MessageContentType::PassportDataReceived: { const MessagePassportDataReceived *m = static_cast(content); return make_tl_object( get_encrypted_passport_element_object(td->file_manager_.get(), m->values), get_encrypted_credentials_object(m->credentials)); } case MessageContentType::Poll: { const MessagePoll *m = static_cast(content); return make_tl_object(td->poll_manager_->get_poll_object(m->poll_id)); } case MessageContentType::Dice: { const MessageDice *m = static_cast(content); return make_tl_object(m->emoji, m->dice_value); } default: UNREACHABLE(); return nullptr; } UNREACHABLE(); return nullptr; } const FormattedText *get_message_content_text(const MessageContent *content) { switch (content->get_type()) { case MessageContentType::Text: return &static_cast(content)->text; case MessageContentType::Game: return &static_cast(content)->game.get_text(); default: return get_message_content_caption(content); } } const FormattedText *get_message_content_caption(const MessageContent *content) { switch (content->get_type()) { case MessageContentType::Animation: return &static_cast(content)->caption; case MessageContentType::Audio: return &static_cast(content)->caption; case MessageContentType::Document: return &static_cast(content)->caption; case MessageContentType::Photo: return &static_cast(content)->caption; case MessageContentType::Video: return &static_cast(content)->caption; case MessageContentType::VoiceNote: return &static_cast(content)->caption; default: return nullptr; } } int32 get_message_content_duration(const MessageContent *content, const Td *td) { CHECK(content != nullptr); switch (content->get_type()) { case MessageContentType::Animation: { auto animation_file_id = static_cast(content)->file_id; return td->animations_manager_->get_animation_duration(animation_file_id); } case MessageContentType::Audio: { auto audio_file_id = static_cast(content)->file_id; return td->audios_manager_->get_audio_duration(audio_file_id); } case MessageContentType::Video: { auto video_file_id = static_cast(content)->file_id; return td->videos_manager_->get_video_duration(video_file_id); } case MessageContentType::VideoNote: { auto video_note_file_id = static_cast(content)->file_id; return td->video_notes_manager_->get_video_note_duration(video_note_file_id); } case MessageContentType::VoiceNote: { auto voice_file_id = static_cast(content)->file_id; return td->voice_notes_manager_->get_voice_note_duration(voice_file_id); } default: return 0; } } FileId get_message_content_upload_file_id(const MessageContent *content) { switch (content->get_type()) { case MessageContentType::Animation: return static_cast(content)->file_id; case MessageContentType::Audio: return static_cast(content)->file_id; case MessageContentType::Document: return static_cast(content)->file_id; case MessageContentType::Photo: for (auto &size : static_cast(content)->photo.photos) { if (size.type == 'i') { return size.file_id; } } break; case MessageContentType::Sticker: return static_cast(content)->file_id; case MessageContentType::Video: return static_cast(content)->file_id; case MessageContentType::VideoNote: return static_cast(content)->file_id; case MessageContentType::VoiceNote: return static_cast(content)->file_id; default: break; } return FileId(); } FileId get_message_content_any_file_id(const MessageContent *content) { FileId result = get_message_content_upload_file_id(content); if (!result.is_valid() && content->get_type() == MessageContentType::Photo) { const auto &sizes = static_cast(content)->photo.photos; if (!sizes.empty()) { result = sizes.back().file_id; } } return result; } void update_message_content_file_id_remote(MessageContent *content, FileId file_id) { if (file_id.get_remote() == 0) { return; } FileId *old_file_id = [&] { switch (content->get_type()) { case MessageContentType::Animation: return &static_cast(content)->file_id; case MessageContentType::Audio: return &static_cast(content)->file_id; case MessageContentType::Document: return &static_cast(content)->file_id; case MessageContentType::Sticker: return &static_cast(content)->file_id; case MessageContentType::Video: return &static_cast(content)->file_id; case MessageContentType::VideoNote: return &static_cast(content)->file_id; case MessageContentType::VoiceNote: return &static_cast(content)->file_id; default: return static_cast(nullptr); } }(); if (old_file_id != nullptr && *old_file_id == file_id && old_file_id->get_remote() == 0) { *old_file_id = file_id; } } FileId get_message_content_thumbnail_file_id(const MessageContent *content, const Td *td) { switch (content->get_type()) { case MessageContentType::Animation: return td->animations_manager_->get_animation_thumbnail_file_id( static_cast(content)->file_id); case MessageContentType::Audio: return td->audios_manager_->get_audio_thumbnail_file_id(static_cast(content)->file_id); case MessageContentType::Document: return td->documents_manager_->get_document_thumbnail_file_id( static_cast(content)->file_id); case MessageContentType::Photo: for (auto &size : static_cast(content)->photo.photos) { if (size.type == 't') { return size.file_id; } } break; case MessageContentType::Sticker: return td->stickers_manager_->get_sticker_thumbnail_file_id( static_cast(content)->file_id); case MessageContentType::Video: return td->videos_manager_->get_video_thumbnail_file_id(static_cast(content)->file_id); case MessageContentType::VideoNote: return td->video_notes_manager_->get_video_note_thumbnail_file_id( static_cast(content)->file_id); case MessageContentType::VoiceNote: return FileId(); default: break; } return FileId(); } vector get_message_content_file_ids(const MessageContent *content, const Td *td) { switch (content->get_type()) { case MessageContentType::Photo: return photo_get_file_ids(static_cast(content)->photo); case MessageContentType::Animation: case MessageContentType::Audio: case MessageContentType::Document: case MessageContentType::Video: case MessageContentType::VideoNote: case MessageContentType::VoiceNote: { vector result; result.reserve(2); FileId file_id = get_message_content_upload_file_id(content); if (file_id.is_valid()) { result.push_back(file_id); } FileId thumbnail_file_id = get_message_content_thumbnail_file_id(content, td); if (thumbnail_file_id.is_valid()) { result.push_back(thumbnail_file_id); } return result; } case MessageContentType::Sticker: return td->stickers_manager_->get_sticker_file_ids(static_cast(content)->file_id); case MessageContentType::Game: return static_cast(content)->game.get_file_ids(td); case MessageContentType::Invoice: return photo_get_file_ids(static_cast(content)->photo); case MessageContentType::ChatChangePhoto: return photo_get_file_ids(static_cast(content)->photo); case MessageContentType::PassportDataReceived: { vector result; for (auto &value : static_cast(content)->values) { auto process_encrypted_secure_file = [&result](const EncryptedSecureFile &file) { if (file.file.file_id.is_valid()) { result.push_back(file.file.file_id); } }; for (auto &file : value.files) { process_encrypted_secure_file(file); } process_encrypted_secure_file(value.front_side); process_encrypted_secure_file(value.reverse_side); process_encrypted_secure_file(value.selfie); for (auto &file : value.translations) { process_encrypted_secure_file(file); } } return result; } default: return {}; } } string get_message_content_search_text(const Td *td, const MessageContent *content) { switch (content->get_type()) { case MessageContentType::Text: { auto *text = static_cast(content); if (!text->web_page_id.is_valid()) { return text->text.text; } return PSTRING() << text->text.text << " " << td->web_pages_manager_->get_web_page_search_text(text->web_page_id); } case MessageContentType::Animation: { auto animation = static_cast(content); return PSTRING() << td->animations_manager_->get_animation_search_text(animation->file_id) << " " << animation->caption.text; } case MessageContentType::Audio: { auto audio = static_cast(content); return PSTRING() << td->audios_manager_->get_audio_search_text(audio->file_id) << " " << audio->caption.text; } case MessageContentType::Document: { auto document = static_cast(content); return PSTRING() << td->documents_manager_->get_document_search_text(document->file_id) << " " << document->caption.text; } case MessageContentType::Photo: { auto photo = static_cast(content); return photo->caption.text; } case MessageContentType::Video: { auto video = static_cast(content); return PSTRING() << td->videos_manager_->get_video_search_text(video->file_id) << " " << video->caption.text; } case MessageContentType::Poll: { auto poll = static_cast(content); return td->poll_manager_->get_poll_search_text(poll->poll_id); } case MessageContentType::Contact: case MessageContentType::Game: case MessageContentType::Invoice: case MessageContentType::LiveLocation: case MessageContentType::Location: case MessageContentType::Sticker: case MessageContentType::Unsupported: case MessageContentType::Venue: case MessageContentType::VideoNote: case MessageContentType::VoiceNote: 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::Dice: return string(); default: UNREACHABLE(); return string(); } } bool need_reget_message_content(const MessageContent *content) { CHECK(content != nullptr); switch (content->get_type()) { case MessageContentType::Unsupported: { auto message_unsupported = static_cast(content); return message_unsupported->version != MessageUnsupported::CURRENT_VERSION; } default: return false; } } bool need_delay_message_content_notification(const MessageContent *content, UserId my_user_id) { switch (content->get_type()) { case MessageContentType::ChatChangeTitle: case MessageContentType::ChatChangePhoto: case MessageContentType::ChatDeletePhoto: case MessageContentType::ChatJoinedByLink: return true; case MessageContentType::ChatAddUsers: { auto &added_user_ids = static_cast(content)->user_ids; return !td::contains(added_user_ids, my_user_id); } case MessageContentType::ChatDeleteUser: return static_cast(content)->user_id != my_user_id; default: return false; } } void update_expired_message_content(unique_ptr &content) { switch (content->get_type()) { case MessageContentType::Photo: content = make_unique(); break; case MessageContentType::Video: content = make_unique(); break; case MessageContentType::Unsupported: // can happen if message content file id is broken break; case MessageContentType::ExpiredPhoto: case MessageContentType::ExpiredVideo: // can happen if message content has been reget from somewhere break; case MessageContentType::Animation: case MessageContentType::Audio: case MessageContentType::Document: case MessageContentType::Sticker: case MessageContentType::VideoNote: case MessageContentType::VoiceNote: // can happen if server will send a document with a wrong content content = make_unique(); break; default: UNREACHABLE(); } } void update_failed_to_send_message_content(Td *td, unique_ptr &content) { // do not forget about failed to send message forwards switch (content->get_type()) { case MessageContentType::Poll: { const MessagePoll *message_poll = static_cast(content.get()); if (PollManager::is_local_poll_id(message_poll->poll_id)) { td->poll_manager_->stop_local_poll(message_poll->poll_id); } break; } default: // nothing to do break; } } void add_message_content_dependencies(Dependencies &dependencies, const MessageContent *message_content) { switch (message_content->get_type()) { case MessageContentType::Text: { auto content = static_cast(message_content); dependencies.web_page_ids.insert(content->web_page_id); break; } case MessageContentType::Animation: break; case MessageContentType::Audio: break; case MessageContentType::Contact: { auto content = static_cast(message_content); dependencies.user_ids.insert(content->contact.get_user_id()); break; } case MessageContentType::Document: break; case MessageContentType::Game: { auto content = static_cast(message_content); dependencies.user_ids.insert(content->game.get_bot_user_id()); break; } 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: { auto content = static_cast(message_content); dependencies.user_ids.insert(content->participant_user_ids.begin(), content->participant_user_ids.end()); break; } case MessageContentType::ChatChangeTitle: break; case MessageContentType::ChatChangePhoto: break; case MessageContentType::ChatDeletePhoto: break; case MessageContentType::ChatDeleteHistory: break; case MessageContentType::ChatAddUsers: { auto content = static_cast(message_content); dependencies.user_ids.insert(content->user_ids.begin(), content->user_ids.end()); break; } case MessageContentType::ChatJoinedByLink: break; case MessageContentType::ChatDeleteUser: { auto content = static_cast(message_content); dependencies.user_ids.insert(content->user_id); break; } case MessageContentType::ChatMigrateTo: { auto content = static_cast(message_content); dependencies.channel_ids.insert(content->migrated_to_channel_id); break; } case MessageContentType::ChannelCreate: break; case MessageContentType::ChannelMigrateFrom: { auto content = static_cast(message_content); dependencies.chat_ids.insert(content->migrated_from_chat_id); break; } case MessageContentType::PinMessage: break; case MessageContentType::GameScore: break; case MessageContentType::ScreenshotTaken: break; case MessageContentType::ChatSetTtl: 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: // no need to add poll dependencies, because they are forcely loaded with the poll break; case MessageContentType::Dice: break; default: UNREACHABLE(); break; } add_formatted_text_dependencies(dependencies, get_message_content_text(message_content)); } void on_sent_message_content(Td *td, const MessageContent *content) { switch (content->get_type()) { case MessageContentType::Animation: return td->animations_manager_->add_saved_animation_by_id(get_message_content_any_file_id(content)); case MessageContentType::Sticker: return td->stickers_manager_->add_recent_sticker_by_id(false, get_message_content_any_file_id(content)); default: // nothing to do return; } } StickerSetId add_sticker_set(Td *td, tl_object_ptr &&input_sticker_set) { return td->stickers_manager_->add_sticker_set(std::move(input_sticker_set)); } void on_dialog_used(TopDialogCategory category, DialogId dialog_id, int32 date) { send_closure(G()->top_dialog_manager(), &TopDialogManager::on_dialog_used, category, dialog_id, date); } void update_used_hashtags(Td *td, const MessageContent *content) { const FormattedText *text = get_message_content_text(content); if (text == nullptr || text->text.empty()) { return; } const unsigned char *ptr = Slice(text->text).ubegin(); const unsigned char *end = Slice(text->text).uend(); int32 utf16_pos = 0; for (auto &entity : text->entities) { if (entity.type != MessageEntity::Type::Hashtag) { continue; } while (utf16_pos < entity.offset && ptr < end) { utf16_pos += 1 + (ptr[0] >= 0xf0); ptr = next_utf8_unsafe(ptr, nullptr, "update_used_hashtags"); } CHECK(utf16_pos == entity.offset); auto from = ptr; while (utf16_pos < entity.offset + entity.length && ptr < end) { utf16_pos += 1 + (ptr[0] >= 0xf0); ptr = next_utf8_unsafe(ptr, nullptr, "update_used_hashtags 2"); } CHECK(utf16_pos == entity.offset + entity.length); auto to = ptr; send_closure(td->hashtag_hints_, &HashtagHints::hashtag_used, Slice(from + 1, to).str()); } } } // namespace td