// // 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) // #include "td/telegram/BusinessManager.h" #include "td/telegram/AccessRights.h" #include "td/telegram/BusinessAwayMessage.h" #include "td/telegram/BusinessChatLink.h" #include "td/telegram/BusinessConnectedBot.h" #include "td/telegram/BusinessGreetingMessage.h" #include "td/telegram/BusinessIntro.h" #include "td/telegram/BusinessRecipients.h" #include "td/telegram/BusinessWorkHours.h" #include "td/telegram/ChatManager.h" #include "td/telegram/DialogLocation.h" #include "td/telegram/DialogManager.h" #include "td/telegram/Global.h" #include "td/telegram/InputBusinessChatLink.h" #include "td/telegram/MessageEntity.h" #include "td/telegram/MessagesManager.h" #include "td/telegram/Td.h" #include "td/telegram/telegram_api.h" #include "td/telegram/UpdatesManager.h" #include "td/telegram/UserManager.h" #include "td/utils/buffer.h" #include "td/utils/logging.h" #include "td/utils/Status.h" namespace td { class GetConnectedBotsQuery final : public Td::ResultHandler { Promise> promise_; public: explicit GetConnectedBotsQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(telegram_api::account_getConnectedBots(), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } auto result = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetConnectedBotsQuery: " << to_string(result); td_->user_manager_->on_get_users(std::move(result->users_), "GetConnectedBotsQuery"); if (result->connected_bots_.size() > 1u) { return on_error(Status::Error(500, "Receive invalid response")); } if (result->connected_bots_.empty()) { return promise_.set_value(nullptr); } auto bot = BusinessConnectedBot(std::move(result->connected_bots_[0])); if (!bot.is_valid()) { return on_error(Status::Error(500, "Receive invalid bot")); } promise_.set_value(bot.get_business_connected_bot_object(td_)); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class UpdateConnectedBotQuery final : public Td::ResultHandler { Promise promise_; public: explicit UpdateConnectedBotQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(const BusinessConnectedBot &bot, telegram_api::object_ptr &&input_user) { int32 flags = 0; if (bot.get_can_reply()) { flags |= telegram_api::account_updateConnectedBot::CAN_REPLY_MASK; } send_query(G()->net_query_creator().create( telegram_api::account_updateConnectedBot(flags, false /*ignored*/, false /*ignored*/, std::move(input_user), bot.get_recipients().get_input_business_bot_recipients(td_)), {{"me"}})); } void send(telegram_api::object_ptr &&input_user) { int32 flags = telegram_api::account_updateConnectedBot::DELETED_MASK; send_query(G()->net_query_creator().create( telegram_api::account_updateConnectedBot(flags, false /*ignored*/, false /*ignored*/, std::move(input_user), BusinessRecipients().get_input_business_bot_recipients(td_)), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for UpdateConnectedBotQuery: " << to_string(ptr); td_->messages_manager_->hide_all_business_bot_manager_bars(); td_->updates_manager_->on_get_updates(std::move(ptr), std::move(promise_)); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class ToggleConnectedBotPausedQuery final : public Td::ResultHandler { Promise promise_; DialogId dialog_id_; public: explicit ToggleConnectedBotPausedQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(DialogId dialog_id, bool is_paused) { dialog_id_ = dialog_id; auto input_peer = td_->dialog_manager_->get_input_peer(dialog_id, AccessRights::Write); CHECK(input_peer != nullptr); send_query(G()->net_query_creator().create( telegram_api::account_toggleConnectedBotPaused(std::move(input_peer), is_paused), {{"me"}, {dialog_id}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } if (!result_ptr.ok()) { LOG(INFO) << "Failed to toggle business bot is paused"; } promise_.set_value(Unit()); } void on_error(Status status) final { td_->dialog_manager_->on_get_dialog_error(dialog_id_, status, "ToggleConnectedBotPausedQuery"); promise_.set_error(std::move(status)); } }; class DisablePeerConnectedBotQuery final : public Td::ResultHandler { Promise promise_; DialogId dialog_id_; public: explicit DisablePeerConnectedBotQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(DialogId dialog_id) { dialog_id_ = dialog_id; auto input_peer = td_->dialog_manager_->get_input_peer(dialog_id, AccessRights::Write); CHECK(input_peer != nullptr); send_query(G()->net_query_creator().create(telegram_api::account_disablePeerConnectedBot(std::move(input_peer)), {{"me"}, {dialog_id}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } if (!result_ptr.ok()) { LOG(INFO) << "Failed to remove business bot"; } else { td_->messages_manager_->on_update_dialog_business_bot_removed(dialog_id_); } promise_.set_value(Unit()); } void on_error(Status status) final { td_->dialog_manager_->on_get_dialog_error(dialog_id_, status, "DisablePeerConnectedBotQuery"); promise_.set_error(std::move(status)); } }; class GetBusinessChatLinksQuery final : public Td::ResultHandler { Promise> promise_; public: explicit GetBusinessChatLinksQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(telegram_api::account_getBusinessChatLinks(), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetBusinessChatLinksQuery: " << to_string(ptr); td_->user_manager_->on_get_users(std::move(ptr->users_), "GetBusinessChatLinksQuery"); td_->chat_manager_->on_get_chats(std::move(ptr->chats_), "GetBusinessChatLinksQuery"); promise_.set_value( BusinessChatLinks(td_->user_manager_.get(), std::move(ptr->links_)).get_business_chat_links_object()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class CreateBusinessChatLinkQuery final : public Td::ResultHandler { Promise> promise_; public: explicit CreateBusinessChatLinkQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send(InputBusinessChatLink &&link) { send_query(G()->net_query_creator().create( telegram_api::account_createBusinessChatLink(link.get_input_business_chat_link(td_->user_manager_.get())), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for CreateBusinessChatLinkQuery: " << to_string(ptr); promise_.set_value(BusinessChatLink(td_->user_manager_.get(), std::move(ptr)).get_business_chat_link_object()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class EditBusinessChatLinkQuery final : public Td::ResultHandler { Promise> promise_; public: explicit EditBusinessChatLinkQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send(const string &link, InputBusinessChatLink &&input_link) { send_query( G()->net_query_creator().create(telegram_api::account_editBusinessChatLink( link, input_link.get_input_business_chat_link(td_->user_manager_.get())), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for EditBusinessChatLinkQuery: " << to_string(ptr); promise_.set_value(BusinessChatLink(td_->user_manager_.get(), std::move(ptr)).get_business_chat_link_object()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class DeleteBusinessChatLinkQuery final : public Td::ResultHandler { Promise promise_; public: explicit DeleteBusinessChatLinkQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(const string &link) { send_query(G()->net_query_creator().create(telegram_api::account_deleteBusinessChatLink(link), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } promise_.set_value(Unit()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class ResolveBusinessChatLinkQuery final : public Td::ResultHandler { Promise> promise_; public: explicit ResolveBusinessChatLinkQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send(const string &link) { send_query(G()->net_query_creator().create(telegram_api::account_resolveBusinessChatLink(link), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for ResolveBusinessChatLinkQuery: " << to_string(ptr); td_->user_manager_->on_get_users(std::move(ptr->users_), "ResolveBusinessChatLinkQuery"); td_->chat_manager_->on_get_chats(std::move(ptr->chats_), "ResolveBusinessChatLinkQuery"); auto text = get_message_text(td_->user_manager_.get(), std::move(ptr->message_), std::move(ptr->entities_), true, true, 0, false, "ResolveBusinessChatLinkQuery"); if (text.text[0] == '@') { text.text = ' ' + text.text; for (auto &entity : text.entities) { entity.offset++; } } DialogId dialog_id(ptr->peer_); if (dialog_id.get_type() != DialogType::User) { LOG(ERROR) << "Receive " << dialog_id; return on_error(Status::Error(500, "Receive invalid business chat")); } remove_unallowed_entities(td_, text, dialog_id); td_->dialog_manager_->force_create_dialog(dialog_id, "ResolveBusinessChatLinkQuery"); promise_.set_value(td_api::make_object( td_->dialog_manager_->get_chat_id_object(dialog_id, "businessChatLinkInfo"), get_formatted_text_object(text, true, -1))); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class UpdateBusinessLocationQuery final : public Td::ResultHandler { Promise promise_; DialogLocation location_; public: explicit UpdateBusinessLocationQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(DialogLocation &&location) { location_ = std::move(location); int32 flags = 0; if (!location_.empty()) { flags |= telegram_api::account_updateBusinessLocation::GEO_POINT_MASK; } if (!location_.get_address().empty()) { flags |= telegram_api::account_updateBusinessLocation::ADDRESS_MASK; } send_query(G()->net_query_creator().create( telegram_api::account_updateBusinessLocation(flags, location_.get_input_geo_point(), location_.get_address()), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } td_->user_manager_->on_update_user_location(td_->user_manager_->get_my_id(), std::move(location_)); promise_.set_value(Unit()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class UpdateBusinessWorkHoursQuery final : public Td::ResultHandler { Promise promise_; BusinessWorkHours work_hours_; public: explicit UpdateBusinessWorkHoursQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(BusinessWorkHours &&work_hours) { work_hours_ = std::move(work_hours); int32 flags = 0; if (!work_hours_.is_empty()) { flags |= telegram_api::account_updateBusinessWorkHours::BUSINESS_WORK_HOURS_MASK; } send_query(G()->net_query_creator().create( telegram_api::account_updateBusinessWorkHours(flags, work_hours_.get_input_business_work_hours()), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } td_->user_manager_->on_update_user_work_hours(td_->user_manager_->get_my_id(), std::move(work_hours_)); promise_.set_value(Unit()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class UpdateBusinessGreetingMessageQuery final : public Td::ResultHandler { Promise promise_; BusinessGreetingMessage greeting_message_; public: explicit UpdateBusinessGreetingMessageQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(BusinessGreetingMessage &&greeting_message) { greeting_message_ = std::move(greeting_message); int32 flags = 0; if (!greeting_message_.is_empty()) { flags |= telegram_api::account_updateBusinessGreetingMessage::MESSAGE_MASK; } send_query(G()->net_query_creator().create(telegram_api::account_updateBusinessGreetingMessage( flags, greeting_message_.get_input_business_greeting_message(td_)), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } td_->user_manager_->on_update_user_greeting_message(td_->user_manager_->get_my_id(), std::move(greeting_message_)); promise_.set_value(Unit()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class UpdateBusinessAwayMessageQuery final : public Td::ResultHandler { Promise promise_; BusinessAwayMessage away_message_; public: explicit UpdateBusinessAwayMessageQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(BusinessAwayMessage &&away_message) { away_message_ = std::move(away_message); int32 flags = 0; if (!away_message_.is_empty()) { flags |= telegram_api::account_updateBusinessAwayMessage::MESSAGE_MASK; } send_query(G()->net_query_creator().create( telegram_api::account_updateBusinessAwayMessage(flags, away_message_.get_input_business_away_message(td_)), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } td_->user_manager_->on_update_user_away_message(td_->user_manager_->get_my_id(), std::move(away_message_)); promise_.set_value(Unit()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class UpdateBusinessIntroQuery final : public Td::ResultHandler { Promise promise_; BusinessIntro intro_; public: explicit UpdateBusinessIntroQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(BusinessIntro &&intro) { intro_ = std::move(intro); int32 flags = 0; if (!intro_.is_empty()) { flags |= telegram_api::account_updateBusinessIntro::INTRO_MASK; } send_query(G()->net_query_creator().create( telegram_api::account_updateBusinessIntro(flags, intro_.get_input_business_intro(td_)), {{"me"}})); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } td_->user_manager_->on_update_user_intro(td_->user_manager_->get_my_id(), std::move(intro_)); promise_.set_value(Unit()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; BusinessManager::BusinessManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) { } void BusinessManager::tear_down() { parent_.reset(); } void BusinessManager::get_business_connected_bot(Promise> &&promise) { td_->create_handler(std::move(promise))->send(); } void BusinessManager::set_business_connected_bot(td_api::object_ptr &&bot, Promise &&promise) { if (bot == nullptr) { return promise.set_error(Status::Error(400, "Bot must be non-empty")); } BusinessConnectedBot connected_bot(std::move(bot)); TRY_RESULT_PROMISE(promise, input_user, td_->user_manager_->get_input_user(connected_bot.get_user_id())); td_->create_handler(std::move(promise))->send(connected_bot, std::move(input_user)); } void BusinessManager::delete_business_connected_bot(UserId bot_user_id, Promise &&promise) { TRY_RESULT_PROMISE(promise, input_user, td_->user_manager_->get_input_user(bot_user_id)); td_->create_handler(std::move(promise))->send(std::move(input_user)); } void BusinessManager::toggle_business_connected_bot_dialog_is_paused(DialogId dialog_id, bool is_paused, Promise &&promise) { TRY_STATUS_PROMISE(promise, td_->dialog_manager_->check_dialog_access(dialog_id, false, AccessRights::Write, "toggle_business_connected_bot_dialog_is_paused")); if (dialog_id.get_type() != DialogType::User) { return promise.set_error(Status::Error(400, "The chat has no connected bot")); } td_->messages_manager_->on_update_dialog_business_bot_is_paused(dialog_id, is_paused); td_->create_handler(std::move(promise))->send(dialog_id, is_paused); } void BusinessManager::remove_business_connected_bot_from_dialog(DialogId dialog_id, Promise &&promise) { TRY_STATUS_PROMISE(promise, td_->dialog_manager_->check_dialog_access(dialog_id, false, AccessRights::Write, "remove_business_connected_bot_from_dialog")); if (dialog_id.get_type() != DialogType::User) { return promise.set_error(Status::Error(400, "The chat has no connected bot")); } td_->messages_manager_->on_update_dialog_business_bot_removed(dialog_id); td_->create_handler(std::move(promise))->send(dialog_id); } void BusinessManager::get_business_chat_links(Promise> &&promise) { td_->create_handler(std::move(promise))->send(); } void BusinessManager::create_business_chat_link(td_api::object_ptr &&link_info, Promise> &&promise) { td_->create_handler(std::move(promise)) ->send(InputBusinessChatLink(td_, std::move(link_info))); } void BusinessManager::edit_business_chat_link(const string &link, td_api::object_ptr &&link_info, Promise> &&promise) { td_->create_handler(std::move(promise)) ->send(link, InputBusinessChatLink(td_, std::move(link_info))); } void BusinessManager::delete_business_chat_link(const string &link, Promise &&promise) { td_->create_handler(std::move(promise))->send(link); } void BusinessManager::get_business_chat_link_info(const string &link, Promise> &&promise) { td_->create_handler(std::move(promise))->send(link); } void BusinessManager::set_business_location(DialogLocation &&location, Promise &&promise) { td_->create_handler(std::move(promise))->send(std::move(location)); } void BusinessManager::set_business_work_hours(BusinessWorkHours &&work_hours, Promise &&promise) { td_->create_handler(std::move(promise))->send(std::move(work_hours)); } void BusinessManager::set_business_greeting_message(BusinessGreetingMessage &&greeting_message, Promise &&promise) { td_->create_handler(std::move(promise))->send(std::move(greeting_message)); } void BusinessManager::set_business_away_message(BusinessAwayMessage &&away_message, Promise &&promise) { td_->create_handler(std::move(promise))->send(std::move(away_message)); } void BusinessManager::set_business_intro(BusinessIntro &&intro, Promise &&promise) { td_->create_handler(std::move(promise))->send(std::move(intro)); } } // namespace td