2018-12-31 20:04:05 +01:00
|
|
|
//
|
2022-01-01 01:35:39 +01:00
|
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
|
2018-12-31 20:04:05 +01:00
|
|
|
//
|
|
|
|
// 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/DialogId.h"
|
2019-04-09 17:38:57 +02:00
|
|
|
#include "td/telegram/Document.h"
|
2021-08-01 05:17:51 +02:00
|
|
|
#include "td/telegram/EncryptedFile.h"
|
2018-12-31 20:04:05 +01:00
|
|
|
#include "td/telegram/files/FileId.h"
|
2022-04-09 22:21:07 +02:00
|
|
|
#include "td/telegram/PhotoFormat.h"
|
2022-04-10 00:15:49 +02:00
|
|
|
#include "td/telegram/PhotoSize.h"
|
2021-10-27 16:32:09 +02:00
|
|
|
#include "td/telegram/secret_api.h"
|
2018-12-31 20:04:05 +01:00
|
|
|
#include "td/telegram/SecretInputMedia.h"
|
2021-10-27 16:32:09 +02:00
|
|
|
#include "td/telegram/td_api.h"
|
|
|
|
#include "td/telegram/telegram_api.h"
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
#include "td/utils/buffer.h"
|
|
|
|
#include "td/utils/common.h"
|
2022-02-07 20:41:07 +01:00
|
|
|
#include "td/utils/FlatHashMap.h"
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
#include <utility>
|
|
|
|
|
|
|
|
namespace td {
|
2018-09-29 03:41:15 +02:00
|
|
|
|
2018-12-31 20:04:05 +01:00
|
|
|
class MultiPromiseActor;
|
|
|
|
class Td;
|
|
|
|
|
|
|
|
class DocumentsManager {
|
|
|
|
public:
|
|
|
|
explicit DocumentsManager(Td *td);
|
2022-07-20 12:40:14 +02:00
|
|
|
DocumentsManager(const DocumentsManager &) = delete;
|
|
|
|
DocumentsManager &operator=(const DocumentsManager &) = delete;
|
|
|
|
DocumentsManager(DocumentsManager &&) = delete;
|
|
|
|
DocumentsManager &operator=(DocumentsManager &&) = delete;
|
|
|
|
~DocumentsManager();
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
class RemoteDocument {
|
|
|
|
public:
|
|
|
|
tl_object_ptr<telegram_api::document> document;
|
|
|
|
// or
|
2021-08-01 05:17:51 +02:00
|
|
|
unique_ptr<EncryptedFile> secret_file;
|
2018-12-31 20:04:05 +01:00
|
|
|
tl_object_ptr<secret_api::decryptedMessageMediaDocument> secret_document;
|
2018-03-03 22:33:26 +01:00
|
|
|
// or
|
|
|
|
tl_object_ptr<telegram_api::WebDocument> web_document;
|
|
|
|
PhotoSize thumbnail;
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
vector<tl_object_ptr<telegram_api::DocumentAttribute>> attributes;
|
|
|
|
|
|
|
|
RemoteDocument(tl_object_ptr<telegram_api::document> &&server_document)
|
|
|
|
: document(std::move(server_document))
|
|
|
|
, secret_file(nullptr)
|
|
|
|
, secret_document(nullptr)
|
2018-03-03 22:33:26 +01:00
|
|
|
, web_document(nullptr)
|
|
|
|
, thumbnail()
|
2018-12-31 20:04:05 +01:00
|
|
|
, attributes(std::move(document->attributes_)) {
|
|
|
|
}
|
|
|
|
|
2018-03-03 22:33:26 +01:00
|
|
|
RemoteDocument(tl_object_ptr<telegram_api::WebDocument> &&web_document, PhotoSize thumbnail,
|
|
|
|
vector<tl_object_ptr<telegram_api::DocumentAttribute>> &&attributes)
|
|
|
|
: document(nullptr)
|
|
|
|
, secret_file(nullptr)
|
|
|
|
, secret_document(nullptr)
|
|
|
|
, web_document(std::move(web_document))
|
|
|
|
, thumbnail(std::move(thumbnail))
|
|
|
|
, attributes(std::move(attributes)) {
|
|
|
|
}
|
|
|
|
|
2021-08-01 05:17:51 +02:00
|
|
|
RemoteDocument(unique_ptr<EncryptedFile> &&secret_file,
|
2018-12-31 20:04:05 +01:00
|
|
|
tl_object_ptr<secret_api::decryptedMessageMediaDocument> &&secret_document,
|
|
|
|
vector<tl_object_ptr<telegram_api::DocumentAttribute>> &&attributes)
|
|
|
|
: document(nullptr)
|
|
|
|
, secret_file(std::move(secret_file))
|
|
|
|
, secret_document(std::move(secret_document))
|
2018-03-03 22:33:26 +01:00
|
|
|
, web_document(nullptr)
|
|
|
|
, thumbnail()
|
2018-12-31 20:04:05 +01:00
|
|
|
, attributes(std::move(attributes)) {
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2021-08-28 13:16:29 +02:00
|
|
|
tl_object_ptr<td_api::document> get_document_object(FileId file_id, PhotoFormat thumbnail_format) const;
|
2018-12-31 20:04:05 +01:00
|
|
|
|
2020-10-25 02:20:24 +01:00
|
|
|
void memory_stats(vector<string> &output);
|
|
|
|
|
2019-04-09 17:38:57 +02:00
|
|
|
Document on_get_document(RemoteDocument remote_document, DialogId owner_dialog_id,
|
|
|
|
MultiPromiseActor *load_data_multipromise_ptr = nullptr,
|
2019-05-07 04:51:56 +02:00
|
|
|
Document::Type default_document_type = Document::Type::General, bool is_background = false,
|
2022-04-12 21:50:20 +02:00
|
|
|
bool is_pattern = false, bool is_ringtone = false);
|
2018-12-31 20:04:05 +01:00
|
|
|
|
2019-03-01 20:51:33 +01:00
|
|
|
void create_document(FileId file_id, string minithumbnail, PhotoSize thumbnail, string file_name, string mime_type,
|
|
|
|
bool replace);
|
2018-12-31 20:04:05 +01:00
|
|
|
|
2018-01-31 10:18:40 +01:00
|
|
|
bool has_input_media(FileId file_id, FileId thumbnail_file_id, bool is_secret) const;
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
SecretInputMedia get_secret_input_media(FileId document_file_id,
|
|
|
|
tl_object_ptr<telegram_api::InputEncryptedFile> input_file,
|
2022-05-11 06:46:06 +02:00
|
|
|
const string &caption, BufferSlice thumbnail, int32 layer) const;
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
tl_object_ptr<telegram_api::InputMedia> get_input_media(FileId file_id,
|
|
|
|
tl_object_ptr<telegram_api::InputFile> input_file,
|
2018-01-30 18:06:54 +01:00
|
|
|
tl_object_ptr<telegram_api::InputFile> input_thumbnail) const;
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
FileId get_document_thumbnail_file_id(FileId file_id) const;
|
|
|
|
|
|
|
|
void delete_document_thumbnail(FileId file_id);
|
|
|
|
|
|
|
|
FileId dup_document(FileId new_id, FileId old_id);
|
|
|
|
|
2021-08-26 17:50:28 +02:00
|
|
|
void merge_documents(FileId new_id, FileId old_id, bool can_delete_old);
|
2018-12-31 20:04:05 +01:00
|
|
|
|
2019-02-21 18:54:20 +01:00
|
|
|
template <class StorerT>
|
|
|
|
void store_document(FileId file_id, StorerT &storer) const;
|
2018-12-31 20:04:05 +01:00
|
|
|
|
2019-02-21 18:54:20 +01:00
|
|
|
template <class ParserT>
|
|
|
|
FileId parse_document(ParserT &parser);
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
string get_document_search_text(FileId file_id) const;
|
|
|
|
|
|
|
|
private:
|
2019-04-09 17:38:57 +02:00
|
|
|
class GeneralDocument {
|
2018-12-31 20:04:05 +01:00
|
|
|
public:
|
|
|
|
string file_name;
|
|
|
|
string mime_type;
|
2019-03-01 20:51:33 +01:00
|
|
|
string minithumbnail;
|
2018-12-31 20:04:05 +01:00
|
|
|
PhotoSize thumbnail;
|
|
|
|
FileId file_id;
|
|
|
|
};
|
|
|
|
|
2019-04-09 17:38:57 +02:00
|
|
|
const GeneralDocument *get_document(FileId file_id) const;
|
2018-12-31 20:04:05 +01:00
|
|
|
|
2019-04-09 17:38:57 +02:00
|
|
|
FileId on_get_document(unique_ptr<GeneralDocument> new_document, bool replace);
|
2018-12-31 20:04:05 +01:00
|
|
|
|
|
|
|
Td *td_;
|
2022-02-07 20:41:07 +01:00
|
|
|
FlatHashMap<FileId, unique_ptr<GeneralDocument>, FileIdHash> documents_; // file_id -> GeneralDocument
|
2018-12-31 20:04:05 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace td
|