tdlight/td/telegram/DocumentsManager.hpp

77 lines
2.2 KiB
C++
Raw Normal View History

//
2021-01-01 13:57:46 +01:00
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2021
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once
#include "td/telegram/DocumentsManager.h"
#include "td/telegram/files/FileId.hpp"
#include "td/telegram/Photo.hpp"
#include "td/telegram/Version.h"
#include "td/utils/logging.h"
#include "td/utils/tl_helpers.h"
2020-05-23 21:27:24 +02:00
#include "td/telegram/ConfigShared.h"
namespace td {
template <class StorerT>
void DocumentsManager::store_document(FileId file_id, StorerT &storer) const {
2020-05-23 21:27:24 +02:00
//LOG(DEBUG) << "Store document " << file_id;
auto it = documents_.find(file_id);
2020-05-23 21:27:24 +02:00
if (it == documents_.end() || it->second == nullptr) {
return;
}
const GeneralDocument *document = it->second.get();
2020-05-23 21:27:24 +02:00
if (document == nullptr) {
return;
}
store(document->file_name, storer);
store(document->mime_type, storer);
store(document->minithumbnail, storer);
store(document->thumbnail, storer);
store(file_id, storer);
}
template <class ParserT>
FileId DocumentsManager::parse_document(ParserT &parser) {
auto document = make_unique<GeneralDocument>();
2020-05-23 21:27:24 +02:00
string tmp_filename;
parse(tmp_filename, parser);
parse(document->mime_type, parser);
2020-05-23 21:27:24 +02:00
if ( G()->shared_config().get_option_boolean("disable_document_filenames") && (
document->mime_type.rfind("image/") == 0 ||
document->mime_type.rfind("video/") == 0 ||
document->mime_type.rfind("audio/") == 0)) {
document->file_name = "0";
} else {
document->file_name = tmp_filename;
}
if (parser.version() >= static_cast<int32>(Version::SupportMinithumbnails)) {
2020-05-23 21:27:24 +02:00
string tmp_minithumbnail;
parse(tmp_minithumbnail, parser);
if (!G()->shared_config().get_option_boolean("disable_minithumbnails")) {
document->minithumbnail = tmp_minithumbnail;
}
}
2020-05-23 21:27:24 +02:00
parse(document->thumbnail, parser);
parse(document->file_id, parser);
2020-05-23 21:27:24 +02:00
LOG(DEBUG) << "Parsed document " << document->file_id;
if (parser.get_error() != nullptr || !document->file_id.is_valid()) {
return FileId();
}
return on_get_document(std::move(document), false);
}
} // namespace td