// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023 // // 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/BusinessConnectionId.h" #include "td/telegram/DialogId.h" #include "td/telegram/files/FileId.h" #include "td/telegram/MessageInputReplyTo.h" #include "td/telegram/net/DcId.h" #include "td/telegram/td_api.h" #include "td/telegram/telegram_api.h" #include "td/actor/actor.h" #include "td/utils/common.h" #include "td/utils/FlatHashMap.h" #include "td/utils/Promise.h" #include "td/utils/Status.h" #include "td/utils/WaitFreeHashMap.h" #include <memory> namespace td { struct InputMessageContent; struct ReplyMarkup; class Td; class BusinessConnectionManager final : public Actor { public: BusinessConnectionManager(Td *td, ActorShared<> parent); BusinessConnectionManager(const BusinessConnectionManager &) = delete; BusinessConnectionManager &operator=(const BusinessConnectionManager &) = delete; BusinessConnectionManager(BusinessConnectionManager &&) = delete; BusinessConnectionManager &operator=(BusinessConnectionManager &&) = delete; ~BusinessConnectionManager() final; Status check_business_connection(const BusinessConnectionId &connection_id, DialogId dialog_id) const; DcId get_business_connection_dc_id(const BusinessConnectionId &connection_id) const; void on_update_bot_business_connect(telegram_api::object_ptr<telegram_api::botBusinessConnection> &&connection); void on_update_bot_new_business_message(const BusinessConnectionId &connection_id, telegram_api::object_ptr<telegram_api::Message> &&message, telegram_api::object_ptr<telegram_api::Message> &&reply_to_message); void on_update_bot_edit_business_message(const BusinessConnectionId &connection_id, telegram_api::object_ptr<telegram_api::Message> &&message, telegram_api::object_ptr<telegram_api::Message> &&reply_to_message); void on_update_bot_delete_business_messages(const BusinessConnectionId &connection_id, DialogId dialog_id, vector<int32> &&messages); void get_business_connection(const BusinessConnectionId &connection_id, Promise<td_api::object_ptr<td_api::businessConnection>> &&promise); void send_message(BusinessConnectionId business_connection_id, DialogId dialog_id, td_api::object_ptr<td_api::InputMessageReplyTo> &&reply_to, bool disable_notification, bool protect_content, td_api::object_ptr<td_api::ReplyMarkup> &&reply_markup, td_api::object_ptr<td_api::InputMessageContent> &&input_message_content, Promise<td_api::object_ptr<td_api::businessMessage>> &&promise); void send_message_album(BusinessConnectionId business_connection_id, DialogId dialog_id, td_api::object_ptr<td_api::InputMessageReplyTo> &&reply_to, bool disable_notification, bool protect_content, vector<td_api::object_ptr<td_api::InputMessageContent>> &&input_message_contents, Promise<td_api::object_ptr<td_api::businessMessages>> &&promise); void get_current_state(vector<td_api::object_ptr<td_api::Update>> &updates) const; private: static constexpr size_t MAX_GROUPED_MESSAGES = 10; // server side limit struct BusinessConnection; struct PendingMessage; class SendBusinessMessageQuery; class SendBusinessMediaQuery; class SendBusinessMultiMediaQuery; class UploadBusinessMediaQuery; class UploadMediaCallback; class UploadThumbnailCallback; struct UploadMediaResult { unique_ptr<PendingMessage> message_; telegram_api::object_ptr<telegram_api::InputMedia> input_media_; }; struct BeingUploadedMedia { unique_ptr<PendingMessage> message_; telegram_api::object_ptr<telegram_api::InputFile> input_file_; Promise<UploadMediaResult> promise_; }; struct MediaGroupSendRequest { size_t finished_count_ = 0; vector<Result<UploadMediaResult>> upload_results_; Promise<td_api::object_ptr<td_api::businessMessages>> promise_; }; void tear_down() final; void on_get_business_connection(const BusinessConnectionId &connection_id, Result<telegram_api::object_ptr<telegram_api::Updates>> r_updates); MessageInputReplyTo create_business_message_input_reply_to( td_api::object_ptr<td_api::InputMessageReplyTo> &&reply_to); Result<InputMessageContent> process_input_message_content( td_api::object_ptr<td_api::InputMessageContent> &&input_message_content); unique_ptr<PendingMessage> create_business_message_to_send(BusinessConnectionId business_connection_id, DialogId dialog_id, MessageInputReplyTo &&input_reply_to, bool disable_notification, bool protect_content, unique_ptr<ReplyMarkup> &&reply_markup, InputMessageContent &&input_content) const; void do_send_message(unique_ptr<PendingMessage> &&message, Promise<td_api::object_ptr<td_api::businessMessage>> &&promise); void process_sent_business_message(telegram_api::object_ptr<telegram_api::Updates> &&updates_ptr, Promise<td_api::object_ptr<td_api::businessMessage>> &&promise); static FileId get_message_file_id(const unique_ptr<PendingMessage> &message); FileId get_message_thumbnail_file_id(const unique_ptr<PendingMessage> &message, FileId file_id) const; void upload_media(unique_ptr<PendingMessage> &&message, Promise<UploadMediaResult> &&promise, vector<int> bad_parts = {}); void complete_send_media(unique_ptr<PendingMessage> &&message, telegram_api::object_ptr<telegram_api::InputMedia> &&input_media, Promise<td_api::object_ptr<td_api::businessMessage>> &&promise); void on_upload_media(FileId file_id, telegram_api::object_ptr<telegram_api::InputFile> input_file); void on_upload_media_error(FileId file_id, Status status); void on_upload_thumbnail(FileId thumbnail_file_id, telegram_api::object_ptr<telegram_api::InputFile> thumbnail_input_file); void do_upload_media(BeingUploadedMedia &&being_uploaded_media, telegram_api::object_ptr<telegram_api::InputFile> input_thumbnail); void complete_upload_media(unique_ptr<PendingMessage> &&message, telegram_api::object_ptr<telegram_api::MessageMedia> &&media, Promise<UploadMediaResult> &&promise); int64 generate_new_media_album_id(); void on_upload_message_album_media(int64 request_id, size_t media_pos, Result<UploadMediaResult> &&result); void process_sent_business_message_album(telegram_api::object_ptr<telegram_api::Updates> &&updates_ptr, Promise<td_api::object_ptr<td_api::businessMessages>> &&promise); td_api::object_ptr<td_api::updateBusinessConnection> get_update_business_connection( const BusinessConnection *connection) const; WaitFreeHashMap<BusinessConnectionId, unique_ptr<BusinessConnection>, BusinessConnectionIdHash> business_connections_; FlatHashMap<BusinessConnectionId, vector<Promise<td_api::object_ptr<td_api::businessConnection>>>, BusinessConnectionIdHash> get_business_connection_queries_; int64 current_media_group_send_request_id_ = 0; FlatHashMap<int64, MediaGroupSendRequest> media_group_send_requests_; std::shared_ptr<UploadMediaCallback> upload_media_callback_; std::shared_ptr<UploadThumbnailCallback> upload_thumbnail_callback_; FlatHashMap<FileId, BeingUploadedMedia, FileIdHash> being_uploaded_files_; FlatHashMap<FileId, BeingUploadedMedia, FileIdHash> being_uploaded_thumbnails_; Td *td_; ActorShared<> parent_; }; } // namespace td