45e855f89d
I can't maintain anymore this amount of features while keeping the library constantly updated and without bugs. Every merge was taking me multiple hours of revisioning the code. I give up. From this commit onwards TDLight will only have small useful customizations that are easy to maintain. Now the people relying on the OptimizeMemory method can restart the session every N hours to free up the memory. The real way to keep a low memory usage must involve a huge refactoring to allow the unloading of the caches into the sqlite database, similar to what's already happening with messages data. Only Levlam has the ability to implement this without needing to merge the upstream everytime.
158 lines
6.1 KiB
C++
158 lines
6.1 KiB
C++
//
|
|
// 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/DialogId.h"
|
|
#include "td/telegram/files/FileId.h"
|
|
#include "td/telegram/Location.h"
|
|
#include "td/telegram/MessageContent.h"
|
|
#include "td/telegram/MessageEntity.h"
|
|
#include "td/telegram/net/NetQuery.h"
|
|
#include "td/telegram/Photo.h"
|
|
#include "td/telegram/td_api.h"
|
|
#include "td/telegram/telegram_api.h"
|
|
#include "td/telegram/UserId.h"
|
|
|
|
#include "td/actor/actor.h"
|
|
#include "td/actor/MultiPromise.h"
|
|
#include "td/actor/PromiseFuture.h"
|
|
#include "td/actor/Timeout.h"
|
|
|
|
#include "td/utils/common.h"
|
|
#include "td/utils/Status.h"
|
|
|
|
#include <unordered_map>
|
|
#include <utility>
|
|
|
|
namespace td {
|
|
|
|
class Td;
|
|
|
|
class Game;
|
|
|
|
class InlineQueriesManager final : public Actor {
|
|
public:
|
|
InlineQueriesManager(Td *td, ActorShared<> parent);
|
|
|
|
void memory_stats(vector<string> &output);
|
|
|
|
void after_get_difference();
|
|
|
|
void answer_inline_query(int64 inline_query_id, bool is_personal,
|
|
vector<tl_object_ptr<td_api::InputInlineQueryResult>> &&input_results, int32 cache_time,
|
|
const string &next_offset, const string &switch_pm_text, const string &switch_pm_parameter,
|
|
Promise<Unit> &&promise) const;
|
|
|
|
uint64 send_inline_query(UserId bot_user_id, DialogId dialog_id, Location user_location, const string &query,
|
|
const string &offset, Promise<Unit> &&promise);
|
|
|
|
vector<UserId> get_recent_inline_bots(Promise<Unit> &&promise);
|
|
|
|
void remove_recent_inline_bot(UserId bot_user_id, Promise<Unit> &&promise);
|
|
|
|
const InlineMessageContent *get_inline_message_content(int64 query_id, const string &result_id);
|
|
|
|
UserId get_inline_bot_user_id(int64 query_id) const;
|
|
|
|
void on_get_inline_query_results(DialogId dialog_id, UserId bot_user_id, uint64 query_hash,
|
|
tl_object_ptr<telegram_api::messages_botResults> &&results);
|
|
|
|
tl_object_ptr<td_api::inlineQueryResults> get_inline_query_results_object(uint64 query_hash);
|
|
|
|
void on_new_query(int64 query_id, UserId sender_user_id, Location user_location,
|
|
tl_object_ptr<telegram_api::InlineQueryPeerType> peer_type, const string &query,
|
|
const string &offset);
|
|
|
|
void on_chosen_result(UserId user_id, Location user_location, const string &query, const string &result_id,
|
|
tl_object_ptr<telegram_api::InputBotInlineMessageID> &&input_bot_inline_message_id);
|
|
|
|
static int32 get_inline_message_dc_id(const tl_object_ptr<telegram_api::InputBotInlineMessageID> &inline_message_id);
|
|
|
|
static tl_object_ptr<telegram_api::InputBotInlineMessageID> get_input_bot_inline_message_id(
|
|
const string &inline_message_id);
|
|
|
|
static string get_inline_message_id(
|
|
tl_object_ptr<telegram_api::InputBotInlineMessageID> &&input_bot_inline_message_id);
|
|
|
|
private:
|
|
static constexpr int32 MAX_RECENT_INLINE_BOTS = 20; // some reasonable value
|
|
static constexpr int32 INLINE_QUERY_DELAY_MS = 400; // server side limit
|
|
|
|
static constexpr int32 BOT_INLINE_MEDIA_RESULT_FLAG_HAS_PHOTO = 1 << 0;
|
|
static constexpr int32 BOT_INLINE_MEDIA_RESULT_FLAG_HAS_DOCUMENT = 1 << 1;
|
|
static constexpr int32 BOT_INLINE_MEDIA_RESULT_FLAG_HAS_TITLE = 1 << 2;
|
|
static constexpr int32 BOT_INLINE_MEDIA_RESULT_FLAG_HAS_DESCRIPTION = 1 << 3;
|
|
|
|
Result<tl_object_ptr<telegram_api::InputBotInlineMessage>> get_inline_message(
|
|
tl_object_ptr<td_api::InputMessageContent> &&input_message_content,
|
|
tl_object_ptr<td_api::ReplyMarkup> &&reply_markup_ptr,
|
|
int32 allowed_media_content_id) const TD_WARN_UNUSED_RESULT;
|
|
|
|
bool register_inline_message_content(int64 query_id, const string &result_id, FileId file_id,
|
|
tl_object_ptr<telegram_api::BotInlineMessage> &&inline_message,
|
|
int32 allowed_media_content_id, bool allow_invoice, Photo *photo = nullptr,
|
|
Game *game = nullptr);
|
|
|
|
tl_object_ptr<td_api::thumbnail> register_thumbnail(
|
|
tl_object_ptr<telegram_api::WebDocument> &&web_document_ptr) const;
|
|
|
|
static string get_web_document_url(const tl_object_ptr<telegram_api::WebDocument> &web_document_ptr);
|
|
static string get_web_document_content_type(const tl_object_ptr<telegram_api::WebDocument> &web_document_ptr);
|
|
|
|
bool update_bot_usage(UserId bot_user_id);
|
|
|
|
void save_recently_used_bots();
|
|
bool load_recently_used_bots(Promise<Unit> &promise);
|
|
|
|
tl_object_ptr<td_api::inlineQueryResults> decrease_pending_request_count(uint64 query_hash);
|
|
|
|
static void on_drop_inline_query_result_timeout_callback(void *inline_queries_manager_ptr, int64 query_hash);
|
|
|
|
void loop() final;
|
|
|
|
void tear_down() final;
|
|
|
|
int32 recently_used_bots_loaded_ = 0; // 0 - not loaded, 1 - load request was sent, 2 - loaded
|
|
MultiPromiseActor resolve_recent_inline_bots_multipromise_{"ResolveRecentInlineBotsMultiPromiseActor"};
|
|
|
|
vector<UserId> recently_used_bot_user_ids_;
|
|
|
|
struct PendingInlineQuery {
|
|
uint64 query_hash;
|
|
UserId bot_user_id;
|
|
DialogId dialog_id;
|
|
tl_object_ptr<telegram_api::InputPeer> input_peer;
|
|
Location user_location;
|
|
string query;
|
|
string offset;
|
|
Promise<Unit> promise;
|
|
};
|
|
|
|
double next_inline_query_time_ = 0.0;
|
|
unique_ptr<PendingInlineQuery> pending_inline_query_;
|
|
NetQueryRef sent_query_;
|
|
|
|
struct InlineQueryResult {
|
|
tl_object_ptr<td_api::inlineQueryResults> results;
|
|
double cache_expire_time;
|
|
int32 pending_request_count;
|
|
};
|
|
|
|
MultiTimeout drop_inline_query_result_timeout_{"DropInlineQueryResultTimeout"};
|
|
std::unordered_map<uint64, InlineQueryResult> inline_query_results_; // query_hash -> result
|
|
|
|
std::unordered_map<int64, std::unordered_map<string, InlineMessageContent>>
|
|
inline_message_contents_; // query_id -> [result_id -> inline_message_content]
|
|
|
|
std::unordered_map<int64, UserId> query_id_to_bot_user_id_;
|
|
|
|
Td *td_;
|
|
ActorShared<> parent_;
|
|
};
|
|
|
|
} // namespace td
|