1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2026
//
// 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/MessageCover.h"
#include "td/telegram/Document.h"
#include "td/telegram/DocumentsManager.h"
#include "td/telegram/files/FileType.h"
#include "td/telegram/Td.h"
#include "td/telegram/VideosManager.h"
namespace td {
FileId MessageCover::get_any_file_id() const {
switch (type_) {
case Type::Empty:
return FileId();
case Type::Photo:
return get_photo_any_file_id(photo_);
case Type::Video:
return video_file_id_;
default:
UNREACHABLE();
return FileId();
}
}
telegram_api::object_ptr<telegram_api::InputMedia> MessageCover::get_cover_input_media(Td *td, bool force,
bool allow_external) const {
switch (type_) {
case Type::Photo:
return photo_get_cover_input_media(td->file_manager_.get(), photo_, force, allow_external);
case Type::Video:
return td->videos_manager_->get_video_cover_input_media(video_file_id_, force, allow_external);
case Type::Empty:
default:
UNREACHABLE();
return nullptr;
}
}
telegram_api::object_ptr<telegram_api::InputMedia> MessageCover::get_input_media(
Td *td, telegram_api::object_ptr<telegram_api::InputFile> &&input_file) {
switch (type_) {
case Type::Photo:
return photo_get_input_media(td->file_manager_.get(), photo_, std::move(input_file), 0, false, FileId());
case Type::Video:
return td->videos_manager_->get_input_media(video_file_id_, std::move(input_file), nullptr, Photo(), 0, 0, false);
case Type::Empty:
default:
UNREACHABLE();
return nullptr;
}
}
Status MessageCover::merge_with_media(Td *td, DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::MessageMedia> &&media_ptr) {
switch (type_) {
case Type::Photo: {
if (media_ptr->get_id() != telegram_api::messageMediaPhoto::ID) {
return Status::Error(500, "Receive invalid response");
}
auto media = telegram_api::move_object_as<telegram_api::messageMediaPhoto>(media_ptr);
if (media->photo_ == nullptr || media->ttl_seconds_ != 0 || media->live_photo_) {
return Status::Error(500, "Receive invalid response without photo");
}
auto new_photo = get_photo(td, std::move(media->photo_), owner_dialog_id, FileType::Photo);
if (new_photo.is_empty()) {
return Status::Error(500, "Receive invalid photo in response");
}
bool is_content_changed = false;
bool need_update = false;
merge_photos(td, &photo_, &new_photo, owner_dialog_id, true, is_content_changed, need_update);
break;
}
case Type::Video: {
if (media_ptr->get_id() != telegram_api::messageMediaDocument::ID) {
return Status::Error(500, "Receive invalid response");
}
auto media = telegram_api::move_object_as<telegram_api::messageMediaDocument>(media_ptr);
if (media->document_ == nullptr || media->ttl_seconds_ != 0 ||
media->document_->get_id() != telegram_api::document::ID) {
return Status::Error(500, "Receive invalid response without photo");
}
auto document = telegram_api::move_object_as<telegram_api::document>(media->document_);
auto parsed_document = td->documents_manager_->on_get_document(std::move(document), owner_dialog_id, false, true,
nullptr, Document::Type::Video);
if (parsed_document.is_empty() || parsed_document.type != Document::Type::Video) {
return Status::Error(500, "Receive invalid live photo video");
}
if (video_file_id_ != parsed_document.file_id) {
td->videos_manager_->merge_videos(parsed_document.file_id, video_file_id_);
}
break;
}
case Type::Empty:
default:
UNREACHABLE();
break;
}
return Status::OK();
}
} // namespace td
|
