243 lines
9.4 KiB
C++
243 lines
9.4 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
|
|
//
|
|
// 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/td_api.h"
|
|
#include "td/telegram/telegram_api.h"
|
|
|
|
#include "td/telegram/DialogId.h"
|
|
#include "td/telegram/files/FileId.h"
|
|
#include "td/telegram/MessageId.h"
|
|
#include "td/telegram/Photo.h"
|
|
#include "td/telegram/WebPageId.h"
|
|
|
|
#include "td/actor/actor.h"
|
|
#include "td/actor/PromiseFuture.h"
|
|
#include "td/actor/Timeout.h"
|
|
|
|
#include "td/db/binlog/BinlogEvent.h"
|
|
|
|
#include "td/utils/common.h"
|
|
#include "td/utils/Status.h"
|
|
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
|
|
namespace td {
|
|
|
|
class Td;
|
|
|
|
class WebPagesManager : public Actor {
|
|
public:
|
|
WebPagesManager(Td *td, ActorShared<> parent);
|
|
|
|
WebPagesManager(const WebPagesManager &) = delete;
|
|
WebPagesManager &operator=(const WebPagesManager &) = delete;
|
|
WebPagesManager(WebPagesManager &&) = delete;
|
|
WebPagesManager &operator=(WebPagesManager &&) = delete;
|
|
~WebPagesManager() override;
|
|
|
|
WebPageId on_get_web_page(tl_object_ptr<telegram_api::WebPage> &&web_page_ptr, DialogId owner_dialog_id);
|
|
|
|
void on_get_web_page_by_url(const string &url, WebPageId web_page_id, bool from_database);
|
|
|
|
void wait_for_pending_web_page(DialogId dialog_id, MessageId message_id, WebPageId web_page_id);
|
|
|
|
bool have_web_page(WebPageId web_page_id) const;
|
|
|
|
bool have_web_page_force(WebPageId web_page_id);
|
|
|
|
tl_object_ptr<td_api::webPage> get_web_page_object(WebPageId web_page_id) const;
|
|
|
|
tl_object_ptr<td_api::webPageInstantView> get_web_page_instant_view_object(WebPageId web_page_id) const;
|
|
|
|
WebPageId get_web_page_preview(const string &message_text, Promise<Unit> &&promise);
|
|
|
|
WebPageId get_web_page_instant_view(const string &url, bool force_full, Promise<Unit> &&promise);
|
|
|
|
WebPageId get_web_page_by_url(const string &url) const;
|
|
|
|
WebPageId get_web_page_by_url(const string &url, Promise<Unit> &&promise);
|
|
|
|
void on_get_web_page_preview_success(const string &message_text,
|
|
tl_object_ptr<telegram_api::MessageMedia> &&message_media_ptr,
|
|
Promise<Unit> &&promise);
|
|
|
|
void on_get_web_page_preview_fail(const string &message_text, Status error, Promise<Unit> &&promise);
|
|
|
|
SecretInputMedia get_secret_input_media(WebPageId web_page_id) const;
|
|
|
|
void on_binlog_web_page_event(BinlogEvent &&event);
|
|
|
|
string get_web_page_search_text(WebPageId web_page_id) const;
|
|
|
|
private:
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_TYPE = 1;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_SITE_NAME = 2;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_TITLE = 4;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_DESCRIPTION = 8;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_PHOTO = 16;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_EMBEDDED_PREVIEW = 32;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_EMBEDDED_PREVIEW_SIZE = 64;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_DURATION = 128;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_AUTHOR = 256;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_DOCUMENT = 512;
|
|
static constexpr int32 WEBPAGE_FLAG_HAS_INSTANT_VIEW = 1024;
|
|
|
|
class WebPage;
|
|
|
|
class RichText;
|
|
|
|
class PageBlock;
|
|
class PageBlockTitle;
|
|
class PageBlockSubtitle;
|
|
class PageBlockAuthorDate;
|
|
class PageBlockHeader;
|
|
class PageBlockSubheader;
|
|
class PageBlockParagraph;
|
|
class PageBlockPreformatted;
|
|
class PageBlockFooter;
|
|
class PageBlockDivider;
|
|
class PageBlockAnchor;
|
|
class PageBlockList;
|
|
class PageBlockBlockQuote;
|
|
class PageBlockPullQuote;
|
|
class PageBlockAnimation;
|
|
class PageBlockPhoto;
|
|
class PageBlockVideo;
|
|
class PageBlockCover;
|
|
class PageBlockEmbedded;
|
|
class PageBlockEmbeddedPost;
|
|
class PageBlockCollage;
|
|
class PageBlockSlideshow;
|
|
class PageBlockChatLink;
|
|
class PageBlockAudio;
|
|
|
|
class WebPageInstantView;
|
|
|
|
class WebPageLogEvent;
|
|
|
|
template <class T>
|
|
friend void store(const unique_ptr<PageBlock> &block, T &storer);
|
|
|
|
template <class T>
|
|
friend void parse(unique_ptr<PageBlock> &block, T &parser);
|
|
|
|
void update_web_page(unique_ptr<WebPage> web_page, WebPageId web_page_id, bool from_binlog, bool from_database);
|
|
|
|
void update_web_page_instant_view(WebPageId web_page_id, WebPageInstantView &new_instant_view,
|
|
WebPageInstantView &&old_instant_view);
|
|
|
|
static bool need_use_old_instant_view(const WebPageInstantView &new_instant_view,
|
|
const WebPageInstantView &old_instant_view);
|
|
|
|
void update_messages_content(WebPageId web_page_id, bool have_web_page);
|
|
|
|
WebPage *get_web_page(WebPageId web_page_id);
|
|
const WebPage *get_web_page(WebPageId web_page_id) const;
|
|
|
|
WebPageInstantView *get_web_page_instant_view(WebPageId web_page_id);
|
|
|
|
const WebPageInstantView *get_web_page_instant_view(WebPageId web_page_id) const;
|
|
|
|
WebPageId get_web_page_instant_view(WebPageId web_page_id, bool force_full, Promise<Unit> &&promise);
|
|
|
|
tl_object_ptr<td_api::webPageInstantView> get_web_page_instant_view_object(
|
|
const WebPageInstantView *web_page_instant_view) const;
|
|
|
|
static void on_pending_web_page_timeout_callback(void *web_pages_manager_ptr, int64 web_page_id);
|
|
void on_pending_web_page_timeout(WebPageId web_page_id);
|
|
|
|
void on_get_web_page_preview_success(const string &message_text, WebPageId web_page_id, Promise<Unit> &&promise);
|
|
|
|
static RichText get_rich_text(tl_object_ptr<telegram_api::RichText> &&rich_text_ptr);
|
|
|
|
static vector<RichText> get_rich_texts(vector<tl_object_ptr<telegram_api::RichText>> &&rich_text_ptrs);
|
|
|
|
static tl_object_ptr<td_api::RichText> get_rich_text_object(const RichText &rich_text);
|
|
|
|
static vector<tl_object_ptr<td_api::RichText>> get_rich_text_objects(const vector<RichText> &rich_texts);
|
|
|
|
static vector<tl_object_ptr<td_api::PageBlock>> get_page_block_objects(
|
|
const vector<unique_ptr<PageBlock>> &page_blocks);
|
|
|
|
unique_ptr<PageBlock> get_page_block(tl_object_ptr<telegram_api::PageBlock> page_block_ptr,
|
|
const std::unordered_map<int64, FileId> &animations,
|
|
const std::unordered_map<int64, FileId> &audios,
|
|
const std::unordered_map<int64, Photo> &photos,
|
|
const std::unordered_map<int64, FileId> &videos) const;
|
|
|
|
vector<unique_ptr<PageBlock>> get_page_blocks(vector<tl_object_ptr<telegram_api::PageBlock>> page_block_ptrs,
|
|
const std::unordered_map<int64, FileId> &animations,
|
|
const std::unordered_map<int64, FileId> &audios,
|
|
const std::unordered_map<int64, Photo> &photos,
|
|
const std::unordered_map<int64, FileId> &videos) const;
|
|
|
|
void on_get_web_page_instant_view(WebPage *web_page, tl_object_ptr<telegram_api::Page> &&page_ptr, int32 hash,
|
|
DialogId owner_dialog_id);
|
|
|
|
void save_web_page(WebPage *web_page, WebPageId web_page_id, bool from_binlog);
|
|
|
|
static string get_web_page_database_key(WebPageId web_page_id);
|
|
|
|
void on_save_web_page_to_database(WebPageId web_page_id, bool success);
|
|
|
|
void load_web_page_from_database(WebPageId web_page_id, Promise<Unit> promise);
|
|
|
|
void on_load_web_page_from_database(WebPageId web_page_id, string value);
|
|
|
|
WebPage *get_web_page_force(WebPageId web_page_id);
|
|
|
|
static string get_web_page_instant_view_database_key(WebPageId web_page_id);
|
|
|
|
void load_web_page_instant_view(WebPageId web_page_id, bool force_full, Promise<Unit> &&promise);
|
|
|
|
void on_load_web_page_instant_view_from_database(WebPageId web_page_id, string value);
|
|
|
|
void reload_web_page_instant_view(WebPageId web_page_id);
|
|
|
|
void update_web_page_instant_view_load_requests(WebPageId web_page_id, bool force_update, Result<> result);
|
|
|
|
static string get_web_page_url_database_key(const string &url);
|
|
|
|
void load_web_page_by_url(const string &url, Promise<Unit> &&promise);
|
|
|
|
void reload_web_page_by_url(const string &url, Promise<Unit> &&promise);
|
|
|
|
void on_load_web_page_id_by_url_from_database(const string &url, string value, Promise<Unit> &&promise);
|
|
|
|
void on_load_web_page_by_url_from_database(WebPageId web_page_id, const string &url, Promise<Unit> &&promise,
|
|
Result<> result);
|
|
|
|
void tear_down() override;
|
|
|
|
Td *td_;
|
|
ActorShared<> parent_;
|
|
std::unordered_map<WebPageId, unique_ptr<WebPage>, WebPageIdHash> web_pages_;
|
|
|
|
std::unordered_map<WebPageId, vector<Promise<Unit>>, WebPageIdHash> load_web_page_from_database_queries_;
|
|
std::unordered_set<WebPageId, WebPageIdHash> loaded_from_database_web_pages_;
|
|
|
|
struct PendingWebPageInstantViewQueries {
|
|
vector<Promise<Unit>> partial;
|
|
vector<Promise<Unit>> full;
|
|
};
|
|
std::unordered_map<WebPageId, PendingWebPageInstantViewQueries, WebPageIdHash> load_web_page_instant_view_queries_;
|
|
|
|
std::unordered_map<WebPageId, std::unordered_set<FullMessageId, FullMessageIdHash>, WebPageIdHash> pending_web_pages_;
|
|
std::unordered_map<WebPageId, std::unordered_map<string, Promise<Unit>>, WebPageIdHash> pending_get_web_pages_;
|
|
|
|
std::unordered_map<string, WebPageId> got_web_page_previews_;
|
|
std::unordered_set<string> running_get_web_page_previews_;
|
|
|
|
std::unordered_map<string, WebPageId> url_to_web_page_id_;
|
|
|
|
MultiTimeout pending_web_pages_timeout_;
|
|
};
|
|
|
|
} // namespace td
|