tdesktop/Telegram/SourceFiles/data/data_media_preload.h
2024-10-03 11:54:28 +04:00

84 lines
1.7 KiB
C++

/*
This file is part of Telegram Desktop,
the official desktop application for the Telegram messaging service.
For license and copyright information please follow this link:
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
*/
#pragma once
#include "storage/download_manager_mtproto.h"
namespace Data {
class PhotoMedia;
struct FileOrigin;
class MediaPreload {
public:
explicit MediaPreload(Fn<void()> done);
virtual ~MediaPreload() = default;
protected:
void callDone();
private:
Fn<void()> _done;
};
class PhotoPreload final : public MediaPreload {
public:
[[nodiscard]] static bool Should(
not_null<PhotoData*> photo,
not_null<PeerData*> context);
PhotoPreload(
not_null<PhotoData*> data,
FileOrigin origin,
Fn<void()> done);
~PhotoPreload();
private:
void start(FileOrigin origin);
std::shared_ptr<PhotoMedia> _photo;
rpl::lifetime _lifetime;
};
class VideoPreload final
: public MediaPreload
, private Storage::DownloadMtprotoTask {
public:
[[nodiscard]] static bool Can(not_null<DocumentData*> video);
VideoPreload(
not_null<DocumentData*> video,
FileOrigin origin,
Fn<void()> done);
~VideoPreload();
private:
void check();
void load();
void done(QByteArray result);
bool readyToRequest() const override;
int64 takeNextRequestOffset() override;
bool feedPart(int64 offset, const QByteArray &bytes) override;
void cancelOnFail() override;
bool setWebFileSizeHook(int64 size) override;
const not_null<DocumentData*> _video;
base::flat_map<uint32, QByteArray> _parts;
base::flat_set<int> _requestedOffsets;
int64 _full = 0;
int _nextRequestOffset = 0;
bool _finished = false;
bool _failed = false;
};
} // namespace Data