// // 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/actor/actor.h" #include "td/actor/MultiPromise.h" #include "td/actor/Timeout.h" #include "td/telegram/DialogId.h" #include "td/telegram/files/FileId.h" #include "td/telegram/Location.h" #include "td/telegram/MessageEntity.h" #include "td/telegram/net/NetQuery.h" #include "td/telegram/Photo.h" #include "td/telegram/ReplyMarkup.h" #include "td/telegram/UserId.h" #include "td/utils/common.h" #include "td/utils/Status.h" #include <tuple> #include <unordered_map> #include <utility> namespace td { class Td; class MessageContent; class Game; class InlineQueriesManager : public Actor { public: InlineQueriesManager(Td *td, ActorShared<> parent); 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); std::tuple<const MessageContent *, const ReplyMarkup *, bool> 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(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, 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 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<FormattedText> process_input_caption(td_api::object_ptr<td_api::formattedText> &&caption) const; tl_object_ptr<telegram_api::inputBotInlineMessageMediaAuto> get_input_bot_inline_message_media_auto( const FormattedText &caption, tl_object_ptr<telegram_api::ReplyMarkup> &&input_reply_markup) const; 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; // TODO make static 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, Photo *photo = nullptr, Game *game = nullptr); tl_object_ptr<td_api::photoSize> 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() override; void tear_down() override; int32 recently_used_bots_loaded_ = 0; // 0 - not loaded, 1 - load request was sent, 2 - loaded MultiPromiseActor resolve_recent_inline_bots_multipromise_; vector<UserId> recently_used_bot_user_ids_; struct PendingInlineQuery { uint64 query_hash; UserId bot_user_id; DialogId dialog_id; Location user_location; string query; string offset; Promise<Unit> promise; }; double next_inline_query_time_ = -1.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_; std::unordered_map<uint64, InlineQueryResult> inline_query_results_; // query_hash -> result struct InlineMessageContent { unique_ptr<MessageContent> message_content; unique_ptr<ReplyMarkup> message_reply_markup; bool disable_web_page_preview; }; 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