Add document flags.

This commit is contained in:
levlam 2023-02-21 14:58:52 +03:00
parent 466853496b
commit ba5643f775
2 changed files with 50 additions and 8 deletions

View File

@ -20,22 +20,63 @@ template <class StorerT>
void DocumentsManager::store_document(FileId file_id, StorerT &storer) const {
const GeneralDocument *document = get_document(file_id);
CHECK(document != nullptr);
store(document->file_name, storer);
store(document->mime_type, storer);
store(document->minithumbnail, storer);
store(document->thumbnail, storer);
bool has_file_name = !document->file_name.empty();
bool has_mime_type = !document->mime_type.empty();
bool has_minithumbnail = !document->minithumbnail.empty();
bool has_thumbnail = document->thumbnail.file_id.is_valid();
BEGIN_STORE_FLAGS();
STORE_FLAG(has_file_name);
STORE_FLAG(has_mime_type);
STORE_FLAG(has_minithumbnail);
STORE_FLAG(has_thumbnail);
END_STORE_FLAGS();
if (has_file_name) {
store(document->file_name, storer);
}
if (has_mime_type) {
store(document->mime_type, storer);
}
if (has_minithumbnail) {
store(document->minithumbnail, storer);
}
if (has_thumbnail) {
store(document->thumbnail, storer);
}
store(file_id, storer);
}
template <class ParserT>
FileId DocumentsManager::parse_document(ParserT &parser) {
auto document = make_unique<GeneralDocument>();
parse(document->file_name, parser);
parse(document->mime_type, parser);
if (parser.version() >= static_cast<int32>(Version::SupportMinithumbnails)) {
bool has_file_name;
bool has_mime_type;
bool has_minithumbnail;
bool has_thumbnail;
if (parser.version() >= static_cast<int32>(Version::AddDocumentFlags)) {
BEGIN_PARSE_FLAGS();
PARSE_FLAG(has_file_name);
PARSE_FLAG(has_mime_type);
PARSE_FLAG(has_minithumbnail);
PARSE_FLAG(has_thumbnail);
END_PARSE_FLAGS();
} else {
has_file_name = true;
has_mime_type = true;
has_minithumbnail = parser.version() >= static_cast<int32>(Version::SupportMinithumbnails);
has_thumbnail = true;
}
if (has_file_name) {
parse(document->file_name, parser);
}
if (has_mime_type) {
parse(document->mime_type, parser);
}
if (has_minithumbnail) {
parse(document->minithumbnail, parser);
}
parse(document->thumbnail, parser);
if (has_thumbnail) {
parse(document->thumbnail, parser);
}
parse(document->file_id, parser);
if (parser.get_error() != nullptr || !document->file_id.is_valid()) {
return FileId();

View File

@ -60,6 +60,7 @@ enum class Version : int32 {
AddMessageChatSetTtlFlags,
AddMessageMediaSpoiler, // 45
MakeParticipantFlags64Bit,
AddDocumentFlags,
Next
};