tdesktop/Telegram/SourceFiles/data/data_auto_download.h

128 lines
2.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 <array>
namespace Data {
namespace AutoDownload {
constexpr auto kMaxBytesLimit = 3000 * 512 * 1024;
enum class Source {
User = 0x00,
Group = 0x01,
Channel = 0x02,
};
constexpr auto kSourcesCount = 3;
enum class Type {
Photo = 0x00,
AutoPlayVideo = 0x01,
VoiceMessage = 0x02,
AutoPlayVideoMessage = 0x03,
Music = 0x04,
AutoPlayGIF = 0x05,
File = 0x06,
};
inline constexpr auto kAutoPlayTypes = {
Type::AutoPlayVideo,
Type::AutoPlayVideoMessage,
Type::AutoPlayGIF,
};
inline constexpr auto kStreamedTypes = {
Type::VoiceMessage,
Type::Music,
};
constexpr auto kTypesCount = 7;
class Single {
public:
void setBytesLimit(int bytesLimit);
bool hasValue() const;
bool shouldDownload(int fileSize) const;
int bytesLimit() const;
qint32 serialize() const;
bool setFromSerialized(qint32 serialized);
private:
int _limit = -1;
};
class Set {
public:
void setBytesLimit(Type type, int bytesLimit);
bool hasValue(Type type) const;
bool shouldDownload(Type type, int fileSize) const;
int bytesLimit(Type type) const;
qint32 serialize(Type type) const;
bool setFromSerialized(Type type, qint32 serialized);
private:
const Single &single(Type type) const;
Single &single(Type type);
std::array<Single, kTypesCount> _data;
};
class Full {
public:
void setBytesLimit(Source source, Type type, int bytesLimit);
[[nodiscard]] bool shouldDownload(
Source source,
Type type,
int fileSize) const;
[[nodiscard]] int bytesLimit(Source source, Type type) const;
[[nodiscard]] QByteArray serialize() const;
bool setFromSerialized(const QByteArray &serialized);
[[nodiscard]] static Full FullDisabled();
private:
[[nodiscard]] const Set &set(Source source) const;
[[nodiscard]] Set &set(Source source);
[[nodiscard]] const Set &setOrDefault(Source source, Type type) const;
std::array<Set, kSourcesCount> _data;
};
[[nodiscard]] bool Should(
const Full &data,
not_null<PeerData*> peer,
not_null<DocumentData*> document);
[[nodiscard]] bool Should(
const Full &data,
not_null<DocumentData*> document);
[[nodiscard]] bool Should(
const Full &data,
not_null<PeerData*> peer,
not_null<PhotoData*> photo);
[[nodiscard]] bool ShouldAutoPlay(
const Full &data,
not_null<PeerData*> peer,
not_null<DocumentData*> document);
[[nodiscard]] Full WithDisabledAutoPlay(const Full &data);
} // namespace AutoDownload
} // namespace Data