// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023 // // 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/MessageExtendedMedia.h" #include "td/telegram/Document.h" #include "td/telegram/DocumentsManager.h" #include "td/telegram/MessageContent.h" #include "td/telegram/MessageContentType.h" #include "td/telegram/PhotoSize.h" #include "td/telegram/Td.h" #include "td/telegram/VideosManager.h" #include "td/utils/algorithm.h" #include "td/utils/logging.h" namespace td { MessageExtendedMedia::MessageExtendedMedia( Td *td, telegram_api::object_ptr<telegram_api::MessageExtendedMedia> &&extended_media, FormattedText &&caption, DialogId owner_dialog_id) { if (extended_media == nullptr) { return; } caption_ = std::move(caption); switch (extended_media->get_id()) { case telegram_api::messageExtendedMediaPreview::ID: { auto media = move_tl_object_as<telegram_api::messageExtendedMediaPreview>(extended_media); type_ = Type::Preview; duration_ = media->video_duration_; dimensions_ = get_dimensions(media->w_, media->h_, "MessageExtendedMedia"); if (media->thumb_ != nullptr) { if (media->thumb_->get_id() == telegram_api::photoStrippedSize::ID) { auto thumbnail = move_tl_object_as<telegram_api::photoStrippedSize>(media->thumb_); minithumbnail_ = thumbnail->bytes_.as_slice().str(); } else { LOG(ERROR) << "Receive " << to_string(media->thumb_); } } break; } case telegram_api::messageExtendedMedia::ID: { auto media = move_tl_object_as<telegram_api::messageExtendedMedia>(extended_media); type_ = Type::Unsupported; switch (media->media_->get_id()) { case telegram_api::messageMediaPhoto::ID: { auto photo = move_tl_object_as<telegram_api::messageMediaPhoto>(media->media_); if (photo->photo_ == nullptr) { break; } photo_ = get_photo(td, std::move(photo->photo_), owner_dialog_id); if (photo_.is_empty()) { break; } type_ = Type::Photo; break; } case telegram_api::messageMediaDocument::ID: { auto document = move_tl_object_as<telegram_api::messageMediaDocument>(media->media_); if (document->document_ == nullptr) { break; } auto document_ptr = std::move(document->document_); int32 document_id = document_ptr->get_id(); if (document_id == telegram_api::documentEmpty::ID) { break; } CHECK(document_id == telegram_api::document::ID); auto parsed_document = td->documents_manager_->on_get_document( move_tl_object_as<telegram_api::document>(document_ptr), owner_dialog_id, nullptr); if (parsed_document.empty() || parsed_document.type != Document::Type::Video) { break; } CHECK(parsed_document.file_id.is_valid()); video_file_id_ = parsed_document.file_id; type_ = Type::Video; break; } default: break; } if (type_ == Type::Unsupported) { unsupported_version_ = CURRENT_VERSION; } break; } default: UNREACHABLE(); } } Result<MessageExtendedMedia> MessageExtendedMedia::get_message_extended_media( Td *td, td_api::object_ptr<td_api::InputMessageContent> &&extended_media_content, DialogId owner_dialog_id, bool is_premium) { if (extended_media_content == nullptr) { return MessageExtendedMedia(); } if (!owner_dialog_id.is_valid()) { return Status::Error(400, "Extended media can't be added to the invoice"); } auto input_content_type = extended_media_content->get_id(); if (input_content_type != td_api::inputMessagePhoto::ID && input_content_type != td_api::inputMessageVideo::ID) { return Status::Error("Invalid extended media content specified"); } TRY_RESULT(input_message_content, get_input_message_content(owner_dialog_id, std::move(extended_media_content), td, is_premium)); if (input_message_content.ttl != 0) { return Status::Error("Can't use self-destructing extended media"); } auto content = input_message_content.content.get(); auto content_type = content->get_type(); MessageExtendedMedia result; CHECK(content_type == MessageContentType::Photo || content_type == MessageContentType::Video); result.caption_ = *get_message_content_caption(content); if (content_type == MessageContentType::Photo) { result.type_ = Type::Photo; result.photo_ = *get_message_content_photo(content); } else { result.type_ = Type::Video; result.video_file_id_ = get_message_content_upload_file_id(content); } return result; } void MessageExtendedMedia::update_from(const MessageExtendedMedia &old_extended_media) { if (!is_media() && old_extended_media.is_media()) { *this = old_extended_media; } } bool MessageExtendedMedia::update_to(Td *td, telegram_api::object_ptr<telegram_api::MessageExtendedMedia> extended_media_ptr, DialogId owner_dialog_id) { MessageExtendedMedia new_extended_media(td, std::move(extended_media_ptr), FormattedText(caption_), owner_dialog_id); if (!new_extended_media.is_media() && is_media()) { return false; } if (*this != new_extended_media || is_equal_but_different(new_extended_media)) { *this = std::move(new_extended_media); return true; } return false; } td_api::object_ptr<td_api::MessageExtendedMedia> MessageExtendedMedia::get_message_extended_media_object( Td *td, bool skip_bot_commands, int32 max_media_timestamp) const { if (type_ == Type::Empty) { return nullptr; } auto caption = get_formatted_text_object(caption_, skip_bot_commands, max_media_timestamp); switch (type_) { case Type::Unsupported: return td_api::make_object<td_api::messageExtendedMediaUnsupported>(std::move(caption)); case Type::Preview: return td_api::make_object<td_api::messageExtendedMediaPreview>(dimensions_.width, dimensions_.height, duration_, get_minithumbnail_object(minithumbnail_), std::move(caption)); case Type::Photo: { auto photo = get_photo_object(td->file_manager_.get(), photo_); CHECK(photo != nullptr); return td_api::make_object<td_api::messageExtendedMediaPhoto>(std::move(photo), std::move(caption)); } case Type::Video: return td_api::make_object<td_api::messageExtendedMediaVideo>( td->videos_manager_->get_video_object(video_file_id_), std::move(caption)); default: UNREACHABLE(); return nullptr; } } void MessageExtendedMedia::append_file_ids(const Td *td, vector<FileId> &file_ids) const { switch (type_) { case Type::Empty: case Type::Unsupported: case Type::Preview: break; case Type::Photo: append(file_ids, photo_get_file_ids(photo_)); break; case Type::Video: Document(Document::Type::Video, video_file_id_).append_file_ids(td, file_ids); break; default: UNREACHABLE(); break; } } void MessageExtendedMedia::delete_thumbnail(Td *td) { switch (type_) { case Type::Empty: case Type::Unsupported: case Type::Preview: break; case Type::Photo: photo_delete_thumbnail(photo_); break; case Type::Video: td->videos_manager_->delete_video_thumbnail(video_file_id_); break; default: UNREACHABLE(); break; } } int32 MessageExtendedMedia::get_duration(const Td *td) const { if (!has_media_timestamp()) { return 0; } return td->videos_manager_->get_video_duration(video_file_id_); } FileId MessageExtendedMedia::get_upload_file_id() const { switch (type_) { case Type::Empty: case Type::Unsupported: case Type::Preview: break; case Type::Photo: return get_photo_upload_file_id(photo_); case Type::Video: return video_file_id_; default: UNREACHABLE(); break; } return FileId(); } FileId MessageExtendedMedia::get_any_file_id() const { switch (type_) { case Type::Empty: case Type::Unsupported: case Type::Preview: break; case Type::Photo: return get_photo_any_file_id(photo_); case Type::Video: return video_file_id_; default: UNREACHABLE(); break; } return FileId(); } FileId MessageExtendedMedia::get_thumbnail_file_id(const Td *td) const { switch (type_) { case Type::Empty: case Type::Unsupported: case Type::Preview: break; case Type::Photo: return get_photo_thumbnail_file_id(photo_); case Type::Video: return td->videos_manager_->get_video_thumbnail_file_id(video_file_id_); default: UNREACHABLE(); break; } return FileId(); } telegram_api::object_ptr<telegram_api::InputMedia> MessageExtendedMedia::get_input_media( Td *td, tl_object_ptr<telegram_api::InputFile> input_file, tl_object_ptr<telegram_api::InputFile> input_thumbnail) const { switch (type_) { case Type::Empty: case Type::Unsupported: case Type::Preview: break; case Type::Photo: return photo_get_input_media(td->file_manager_.get(), photo_, std::move(input_file), 0, false); case Type::Video: return td->videos_manager_->get_input_media(video_file_id_, std::move(input_file), std::move(input_thumbnail), 0, false); default: UNREACHABLE(); break; } return nullptr; } bool MessageExtendedMedia::is_equal_but_different(const MessageExtendedMedia &other) const { return type_ == Type::Unsupported && other.type_ == Type::Unsupported && unsupported_version_ != other.unsupported_version_; } bool operator==(const MessageExtendedMedia &lhs, const MessageExtendedMedia &rhs) { if (lhs.type_ != rhs.type_ || lhs.caption_ != rhs.caption_) { return false; } switch (lhs.type_) { case MessageExtendedMedia::Type::Empty: return true; case MessageExtendedMedia::Type::Unsupported: // don't compare unsupported_version_ return true; case MessageExtendedMedia::Type::Preview: return lhs.duration_ == rhs.duration_ && lhs.dimensions_ == rhs.dimensions_ && lhs.minithumbnail_ == rhs.minithumbnail_; case MessageExtendedMedia::Type::Photo: return lhs.photo_ == rhs.photo_; case MessageExtendedMedia::Type::Video: return lhs.video_file_id_ == rhs.video_file_id_; default: UNREACHABLE(); return true; } } bool operator!=(const MessageExtendedMedia &lhs, const MessageExtendedMedia &rhs) { return !(lhs == rhs); } } // namespace td