MrBesen
/
tdesktop
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Allowed to save not loaded photos from bulk download menu item.

This commit is contained in:
23rd 2023-09-06 13:31:14 +03:00
parent ed345e0823
commit 1b5b9f46d2
1 changed files with 69 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
Telegram/SourceFiles/menu/menu_item_download_files.cpp
View File

@ -35,7 +35,7 @@ namespace Menu {
namespace {
using Documents = std::vector<std::pair<not_null<DocumentData*>, FullMsgId>>;
using Photos = std::vector<std::shared_ptr<Data::PhotoMedia>>;
using Photos = std::vector<std::pair<not_null<PhotoData*>, FullMsgId>>;
[[nodiscard]] bool Added(
HistoryItem *item,
@ -44,12 +44,8 @@ using Photos = std::vector<std::shared_ptr<Data::PhotoMedia>>;
if (item && !item->forbidsForward()) {
if (const auto media = item->media()) {
if (const auto photo = media->photo()) {
if (const auto view = photo->activeMediaView()) {
if (view->loaded()) {
photos.push_back(view);
return true;
}
}
photos.emplace_back(photo, item->fullId());
return true;
} else if (const auto document = media->document()) {
documents.emplace_back(document, item->fullId());
return true;
@ -71,7 +67,7 @@ void AddAction(
const auto icon = documents.empty()
? &st::menuIconSaveImage
: &st::menuIconDownload;
const auto showToast = documents.empty();
const auto shouldShowToast = documents.empty();
const auto weak = base::make_weak(controller);
const auto saveImages = [=](const QString &folderPath) {
@ -91,38 +87,75 @@ void AddAction(
if (path.isEmpty()) {
return;
}
QDir().mkpath(path);
const auto fullPath = [&](int i) {
return filedialogDefaultName(
u"photo_"_q + QString::number(i),
u".jpg"_q,
path);
};
auto lastPath = QString();
for (auto i = 0; i < photos.size(); i++) {
lastPath = fullPath(i + 1);
photos[i]->saveToFile(lastPath);
const auto showToast = !shouldShowToast
? Fn<void(const QString &)>(nullptr)
: [=](const QString &lastPath) {
const auto filter = [lastPath](const auto ...) {
File::ShowInFolder(lastPath);
return false;
};
controller->showToast({
.text = (photos.size() > 1
? tr::lng_mediaview_saved_images_to
: tr::lng_mediaview_saved_to)(
tr::now,
lt_downloads,
Ui::Text::Link(
tr::lng_mediaview_downloads(tr::now),
"internal:show_saved_message"),
Ui::Text::WithEntities),
.st = &st::defaultToast,
.filter = filter,
});
};
auto views = std::vector<std::shared_ptr<Data::PhotoMedia>>();
for (const auto &[photo, fullId] : photos) {
if (const auto view = photo->createMediaView()) {
view->wanted(Data::PhotoSize::Large, fullId);
views.push_back(view);
}
}
if (showToast) {
const auto filter = [lastPath](const auto ...) {
File::ShowInFolder(lastPath);
return false;
const auto finalCheck = [=] {
for (const auto &[photo, _] : photos) {
if (photo->loading()) {
return false;
}
}
return true;
};
const auto saveToFiles = [=] {
const auto fullPath = [&](int i) {
return filedialogDefaultName(
u"photo_"_q + QString::number(i),
u".jpg"_q,
path);
};
controller->showToast({
.text = (photos.size() > 1
? tr::lng_mediaview_saved_images_to
: tr::lng_mediaview_saved_to)(
tr::now,
lt_downloads,
Ui::Text::Link(
tr::lng_mediaview_downloads(tr::now),
"internal:show_saved_message"),
Ui::Text::WithEntities),
.st = &st::defaultToast,
.filter = filter,
});
auto lastPath = QString();
for (auto i = 0; i < views.size(); i++) {
lastPath = fullPath(i + 1);
views[i]->saveToFile(lastPath);
}
if (showToast) {
showToast(lastPath);
}
};
if (finalCheck()) {
saveToFiles();
} else {
auto lifetime = std::make_shared<rpl::lifetime>();
session->downloaderTaskFinished(
) | rpl::start_with_next([=]() mutable {
if (finalCheck()) {
saveToFiles();
base::take(lifetime)->destroy();
}
}, *lifetime);
}
};
const auto saveDocuments = [=](const QString &folderPath) {