void/AyuGramDesktop
2
0
Fork 0
mirror of https://github.com/AyuGram/AyuGramDesktop.git synced 2025-06-05 06:33:57 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow downloading files from webpage collages.

Browse source
This commit is contained in:
John Preston 2023-08-18 19:39:22 +02:00
parent b610de30f4
commit ccc61028b8
1 changed files with 42 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
Telegram/SourceFiles/data/data_download_manager.cpp
View file

@ -86,6 +86,45 @@ constexpr auto ByDocument = [](const auto &entry) {
return nullptr; return nullptr;
} }
[[nodiscard]] bool ItemContainsMedia(const DownloadObject &object) {
if (const auto photo = object.photo) {
if (const auto media = object.item->media()) {
if (const auto page = media->webpage()) {
if (page->photo == photo) {
return true;
}
for (const auto &item : page->collage.items) {
if (const auto v = std::get_if<PhotoData*>(&item)) {
if ((*v) == photo) {
return true;
}
}
}
} else {
return (media->photo() == photo);
}
}
} else if (const auto document = object.document) {
if (const auto media = object.item->media()) {
if (const auto page = media->webpage()) {
if (page->document == document) {
return true;
}
for (const auto &item : page->collage.items) {
if (const auto v = std::get_if<DocumentData*>(&item)) {
if ((*v) == document) {
return true;
}
}
}
} else {
return (media->document() == document);
}
}
}
return false;
}
struct DocumentDescriptor { struct DocumentDescriptor {
uint64 sessionUniqueId = 0; uint64 sessionUniqueId = 0;
DocumentId documentId = 0; DocumentId documentId = 0;
@ -242,12 +281,12 @@ void DownloadManager::check(
std::vector<DownloadingId>::iterator i) { std::vector<DownloadingId>::iterator i) {
auto &entry = *i; auto &entry = *i;
const auto photo = ItemPhoto(entry.object.item); if (!ItemContainsMedia(entry.object)) {
const auto document = ItemDocument(entry.object.item);
if (entry.object.photo != photo || entry.object.document != document) {
cancel(data, i); cancel(data, i);
return; return;
} }
const auto document = entry.object.document;
// Load with progress only documents for now. // Load with progress only documents for now.
Assert(document != nullptr); Assert(document != nullptr);