// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024 // // 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/ForumTopicManager.h" #include "td/telegram/AccessRights.h" #include "td/telegram/AuthManager.h" #include "td/telegram/ChannelId.h" #include "td/telegram/ContactsManager.h" #include "td/telegram/CustomEmojiId.h" #include "td/telegram/DialogManager.h" #include "td/telegram/ForumTopic.h" #include "td/telegram/ForumTopic.hpp" #include "td/telegram/ForumTopicIcon.h" #include "td/telegram/ForumTopicInfo.hpp" #include "td/telegram/Global.h" #include "td/telegram/LinkManager.h" #include "td/telegram/logevent/LogEvent.h" #include "td/telegram/MessagesManager.h" #include "td/telegram/MessageThreadDb.h" #include "td/telegram/misc.h" #include "td/telegram/NotificationManager.h" #include "td/telegram/NotificationSettingsManager.h" #include "td/telegram/ServerMessageId.h" #include "td/telegram/Td.h" #include "td/telegram/TdDb.h" #include "td/telegram/telegram_api.h" #include "td/telegram/UpdatesManager.h" #include "td/utils/algorithm.h" #include "td/utils/buffer.h" #include "td/utils/logging.h" #include "td/utils/Random.h" #include "td/utils/Slice.h" #include "td/utils/tl_helpers.h" namespace td { class CreateForumTopicQuery final : public Td::ResultHandler { Promise> promise_; ChannelId channel_id_; DialogId creator_dialog_id_; int64 random_id_; public: explicit CreateForumTopicQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, const string &title, int32 icon_color, CustomEmojiId icon_custom_emoji_id, DialogId as_dialog_id) { channel_id_ = channel_id; creator_dialog_id_ = td_->dialog_manager_->get_my_dialog_id(); int32 flags = 0; if (icon_color != -1) { flags |= telegram_api::channels_createForumTopic::ICON_COLOR_MASK; } if (icon_custom_emoji_id.is_valid()) { flags |= telegram_api::channels_createForumTopic::ICON_EMOJI_ID_MASK; } tl_object_ptr as_input_peer; if (as_dialog_id.is_valid()) { as_input_peer = td_->dialog_manager_->get_input_peer(as_dialog_id, AccessRights::Write); if (as_input_peer != nullptr) { flags |= telegram_api::channels_createForumTopic::SEND_AS_MASK; creator_dialog_id_ = as_dialog_id; } } do { random_id_ = Random::secure_int64(); } while (random_id_ == 0); auto input_channel = td_->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( telegram_api::channels_createForumTopic(flags, std::move(input_channel), title, icon_color, icon_custom_emoji_id.get(), random_id_, std::move(as_input_peer)), {{channel_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()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for CreateForumTopicQuery: " << to_string(ptr); auto message = UpdatesManager::get_message_by_random_id(ptr.get(), DialogId(channel_id_), random_id_); if (message == nullptr || message->get_id() != telegram_api::messageService::ID) { LOG(ERROR) << "Receive invalid result for CreateForumTopicQuery: " << to_string(ptr); return promise_.set_error(Status::Error(400, "Invalid result received")); } auto service_message = static_cast(message); if (service_message->action_->get_id() != telegram_api::messageActionTopicCreate::ID) { LOG(ERROR) << "Receive invalid result for CreateForumTopicQuery: " << to_string(ptr); return promise_.set_error(Status::Error(400, "Invalid result received")); } auto action = static_cast(service_message->action_.get()); auto forum_topic_info = td::make_unique(MessageId(ServerMessageId(service_message->id_)), action->title_, ForumTopicIcon(action->icon_color_, action->icon_emoji_id_), service_message->date_, creator_dialog_id_, true, false, false); td_->updates_manager_->on_get_updates( std::move(ptr), PromiseCreator::lambda([dialog_id = DialogId(channel_id_), forum_topic_info = std::move(forum_topic_info), promise = std::move(promise_)](Unit result) mutable { send_closure(G()->forum_topic_manager(), &ForumTopicManager::on_forum_topic_created, dialog_id, std::move(forum_topic_info), std::move(promise)); })); } void on_error(Status status) final { td_->contacts_manager_->on_get_channel_error(channel_id_, status, "CreateForumTopicQuery"); promise_.set_error(std::move(status)); } }; class EditForumTopicQuery final : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; MessageId top_thread_message_id_; public: explicit EditForumTopicQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, MessageId top_thread_message_id, bool edit_title, const string &title, bool edit_custom_emoji_id, CustomEmojiId icon_custom_emoji_id) { channel_id_ = channel_id; top_thread_message_id_ = top_thread_message_id; auto input_channel = td_->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); int32 flags = 0; if (edit_title) { flags |= telegram_api::channels_editForumTopic::TITLE_MASK; } if (edit_custom_emoji_id) { flags |= telegram_api::channels_editForumTopic::ICON_EMOJI_ID_MASK; } send_query(G()->net_query_creator().create( telegram_api::channels_editForumTopic(flags, std::move(input_channel), top_thread_message_id_.get_server_message_id().get(), title, icon_custom_emoji_id.get(), false, false), {{channel_id}})); } void send(ChannelId channel_id, MessageId top_thread_message_id, bool is_closed) { channel_id_ = channel_id; top_thread_message_id_ = top_thread_message_id; auto input_channel = td_->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); int32 flags = telegram_api::channels_editForumTopic::CLOSED_MASK; send_query(G()->net_query_creator().create( telegram_api::channels_editForumTopic(flags, std::move(input_channel), top_thread_message_id_.get_server_message_id().get(), string(), 0, is_closed, false), {{channel_id}})); } void send(ChannelId channel_id, bool is_hidden) { channel_id_ = channel_id; top_thread_message_id_ = MessageId(ServerMessageId(1)); auto input_channel = td_->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); int32 flags = telegram_api::channels_editForumTopic::HIDDEN_MASK; send_query(G()->net_query_creator().create( telegram_api::channels_editForumTopic(flags, std::move(input_channel), top_thread_message_id_.get_server_message_id().get(), string(), 0, false, is_hidden), {{channel_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()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for EditForumTopicQuery: " << to_string(ptr); td_->updates_manager_->on_get_updates(std::move(ptr), std::move(promise_)); } void on_error(Status status) final { if (status.message() == "TOPIC_NOT_MODIFIED" && !td_->auth_manager_->is_bot()) { return promise_.set_value(Unit()); } td_->contacts_manager_->on_get_channel_error(channel_id_, status, "EditForumTopicQuery"); promise_.set_error(std::move(status)); } }; class UpdatePinnedForumTopicQuery final : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit UpdatePinnedForumTopicQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, MessageId top_thread_message_id, bool is_pinned) { channel_id_ = channel_id; auto input_channel = td_->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( telegram_api::channels_updatePinnedForumTopic(std::move(input_channel), top_thread_message_id.get_server_message_id().get(), is_pinned), {{channel_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()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for UpdatePinnedForumTopicQuery: " << to_string(ptr); td_->updates_manager_->on_get_updates(std::move(ptr), std::move(promise_)); } void on_error(Status status) final { if (status.message() == "PINNED_TOPIC_NOT_MODIFIED" && !td_->auth_manager_->is_bot()) { return promise_.set_value(Unit()); } td_->contacts_manager_->on_get_channel_error(channel_id_, status, "UpdatePinnedForumTopicQuery"); promise_.set_error(std::move(status)); } }; class ReorderPinnedForumTopicsQuery final : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit ReorderPinnedForumTopicsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, const vector &top_thread_message_ids) { channel_id_ = channel_id; auto input_channel = td_->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); int32 flags = telegram_api::channels_reorderPinnedForumTopics::FORCE_MASK; send_query(G()->net_query_creator().create( telegram_api::channels_reorderPinnedForumTopics(flags, true /*ignored*/, std::move(input_channel), MessageId::get_server_message_ids(top_thread_message_ids)), {{channel_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()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for ReorderPinnedForumTopicsQuery: " << to_string(ptr); td_->updates_manager_->on_get_updates(std::move(ptr), std::move(promise_)); } void on_error(Status status) final { if (status.message() == "PINNED_TOPICS_NOT_MODIFIED" && !td_->auth_manager_->is_bot()) { return promise_.set_value(Unit()); } td_->contacts_manager_->on_get_channel_error(channel_id_, status, "ReorderPinnedForumTopicsQuery"); promise_.set_error(std::move(status)); } }; class GetForumTopicQuery final : public Td::ResultHandler { Promise> promise_; ChannelId channel_id_; MessageId top_thread_message_id_; public: explicit GetForumTopicQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, MessageId top_thread_message_id) { channel_id_ = channel_id; top_thread_message_id_ = top_thread_message_id; auto input_channel = td_->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( telegram_api::channels_getForumTopicsByID(std::move(input_channel), {top_thread_message_id_.get_server_message_id().get()}), {{channel_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()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetForumTopicQuery: " << to_string(ptr); td_->contacts_manager_->on_get_users(std::move(ptr->users_), "GetForumTopicQuery"); td_->contacts_manager_->on_get_chats(std::move(ptr->chats_), "GetForumTopicQuery"); if (ptr->topics_.size() != 1u) { return promise_.set_value(nullptr); } MessagesInfo messages_info; messages_info.messages = std::move(ptr->messages_); messages_info.total_count = ptr->count_; messages_info.is_channel_messages = true; td_->messages_manager_->get_channel_difference_if_needed( DialogId(channel_id_), std::move(messages_info), PromiseCreator::lambda([actor_id = td_->forum_topic_manager_actor_.get(), channel_id = channel_id_, top_thread_message_id = top_thread_message_id_, topic = std::move(ptr->topics_[0]), promise = std::move(promise_)](Result &&result) mutable { if (result.is_error()) { promise.set_error(result.move_as_error()); } else { auto info = result.move_as_ok(); send_closure(actor_id, &ForumTopicManager::on_get_forum_topic, channel_id, top_thread_message_id, std::move(info), std::move(topic), std::move(promise)); } }), "GetForumTopicQuery"); } void on_error(Status status) final { td_->contacts_manager_->on_get_channel_error(channel_id_, status, "GetForumTopicQuery"); promise_.set_error(std::move(status)); } }; class GetForumTopicsQuery final : public Td::ResultHandler { Promise> promise_; ChannelId channel_id_; public: explicit GetForumTopicsQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, const string &query, int32 offset_date, MessageId offset_message_id, MessageId offset_top_thread_message_id, int32 limit) { channel_id_ = channel_id; auto input_channel = td_->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); int32 flags = 0; if (!query.empty()) { flags |= telegram_api::channels_getForumTopics::Q_MASK; } send_query(G()->net_query_creator().create( telegram_api::channels_getForumTopics(flags, std::move(input_channel), query, offset_date, offset_message_id.get_server_message_id().get(), offset_top_thread_message_id.get_server_message_id().get(), limit), {{channel_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()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetForumTopicsQuery: " << to_string(ptr); td_->contacts_manager_->on_get_users(std::move(ptr->users_), "GetForumTopicsQuery"); td_->contacts_manager_->on_get_chats(std::move(ptr->chats_), "GetForumTopicsQuery"); MessagesInfo messages_info; messages_info.messages = std::move(ptr->messages_); messages_info.total_count = ptr->count_; messages_info.is_channel_messages = true; // ignore ptr->pts_ td_->messages_manager_->get_channel_difference_if_needed( DialogId(channel_id_), std::move(messages_info), PromiseCreator::lambda([actor_id = td_->forum_topic_manager_actor_.get(), channel_id = channel_id_, order_by_creation_date = ptr->order_by_create_date_, topics = std::move(ptr->topics_), promise = std::move(promise_)](Result &&result) mutable { if (result.is_error()) { promise.set_error(result.move_as_error()); } else { auto info = result.move_as_ok(); send_closure(actor_id, &ForumTopicManager::on_get_forum_topics, channel_id, order_by_creation_date, std::move(info), std::move(topics), std::move(promise)); } }), "GetForumTopicsQuery"); } void on_error(Status status) final { td_->contacts_manager_->on_get_channel_error(channel_id_, status, "GetForumTopicsQuery"); promise_.set_error(std::move(status)); } }; class ReadForumTopicQuery final : public Td::ResultHandler { DialogId dialog_id_; public: void send(DialogId dialog_id, MessageId top_thread_message_id, MessageId max_message_id) { dialog_id_ = dialog_id; auto input_peer = td_->dialog_manager_->get_input_peer(dialog_id, AccessRights::Read); if (input_peer == nullptr) { return on_error(Status::Error(400, "Can't access the chat")); } send_query(G()->net_query_creator().create( telegram_api::messages_readDiscussion(std::move(input_peer), top_thread_message_id.get_server_message_id().get(), max_message_id.get_server_message_id().get()), {{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()); } } void on_error(Status status) final { td_->dialog_manager_->on_get_dialog_error(dialog_id_, status, "ReadForumTopicQuery"); } }; template void ForumTopicManager::Topic::store(StorerT &storer) const { CHECK(info_ != nullptr); using td::store; store(MAGIC, storer); bool has_topic = topic_ != nullptr; BEGIN_STORE_FLAGS(); STORE_FLAG(has_topic); END_STORE_FLAGS(); store(info_, storer); if (has_topic) { store(topic_, storer); } } template void ForumTopicManager::Topic::parse(ParserT &parser) { CHECK(info_ != nullptr); using td::parse; int32 magic; parse(magic, parser); if (magic != MAGIC) { return parser.set_error("Invalid magic"); } bool has_topic; BEGIN_PARSE_FLAGS(); PARSE_FLAG(has_topic); END_PARSE_FLAGS(); parse(info_, parser); if (has_topic) { parse(topic_, parser); } } ForumTopicManager::ForumTopicManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) { } ForumTopicManager::~ForumTopicManager() { Scheduler::instance()->destroy_on_scheduler(G()->get_gc_scheduler_id(), dialog_topics_); } void ForumTopicManager::tear_down() { parent_.reset(); } void ForumTopicManager::create_forum_topic(DialogId dialog_id, string &&title, td_api::object_ptr &&icon, Promise> &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); auto channel_id = dialog_id.get_channel_id(); if (!td_->contacts_manager_->get_channel_permissions(channel_id).can_create_topics()) { return promise.set_error(Status::Error(400, "Not enough rights to create a topic")); } auto new_title = clean_name(std::move(title), MAX_FORUM_TOPIC_TITLE_LENGTH); if (new_title.empty()) { return promise.set_error(Status::Error(400, "Title must be non-empty")); } int32 icon_color = -1; CustomEmojiId icon_custom_emoji_id; if (icon != nullptr) { icon_color = icon->color_; if (icon_color < 0 || icon_color > 0xFFFFFF) { return promise.set_error(Status::Error(400, "Invalid icon color specified")); } icon_custom_emoji_id = CustomEmojiId(icon->custom_emoji_id_); } DialogId as_dialog_id = td_->messages_manager_->get_dialog_default_send_message_as_dialog_id(dialog_id); td_->create_handler(std::move(promise)) ->send(channel_id, new_title, icon_color, icon_custom_emoji_id, as_dialog_id); } void ForumTopicManager::on_forum_topic_created(DialogId dialog_id, unique_ptr &&forum_topic_info, Promise> &&promise) { TRY_STATUS_PROMISE(promise, G()->close_status()); CHECK(forum_topic_info != nullptr); MessageId top_thread_message_id = forum_topic_info->get_top_thread_message_id(); auto topic = add_topic(dialog_id, top_thread_message_id); if (topic == nullptr) { return promise.set_value(forum_topic_info->get_forum_topic_info_object(td_)); } if (topic->info_ == nullptr) { set_topic_info(dialog_id, topic, std::move(forum_topic_info)); } save_topic_to_database(dialog_id, topic); promise.set_value(topic->info_->get_forum_topic_info_object(td_)); } void ForumTopicManager::edit_forum_topic(DialogId dialog_id, MessageId top_thread_message_id, string &&title, bool edit_icon_custom_emoji, CustomEmojiId icon_custom_emoji_id, Promise &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); TRY_STATUS_PROMISE(promise, can_be_message_thread_id(top_thread_message_id)); auto channel_id = dialog_id.get_channel_id(); if (!td_->contacts_manager_->get_channel_permissions(channel_id).can_edit_topics()) { auto topic_info = get_topic_info(dialog_id, top_thread_message_id); if (topic_info != nullptr && !topic_info->is_outgoing()) { return promise.set_error(Status::Error(400, "Not enough rights to edit the topic")); } } bool edit_title = !title.empty(); auto new_title = clean_name(std::move(title), MAX_FORUM_TOPIC_TITLE_LENGTH); if (edit_title && new_title.empty()) { return promise.set_error(Status::Error(400, "Title must be non-empty")); } if (!edit_title && !edit_icon_custom_emoji) { return promise.set_value(Unit()); } td_->create_handler(std::move(promise)) ->send(channel_id, top_thread_message_id, edit_title, new_title, edit_icon_custom_emoji, icon_custom_emoji_id); } void ForumTopicManager::read_forum_topic_messages(DialogId dialog_id, MessageId top_thread_message_id, MessageId last_read_inbox_message_id) { CHECK(!td_->auth_manager_->is_bot()); auto topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr || topic->topic_ == nullptr) { return; } if (topic->topic_->update_last_read_inbox_message_id(last_read_inbox_message_id, -1)) { // TODO send updates auto max_message_id = last_read_inbox_message_id.get_prev_server_message_id(); LOG(INFO) << "Send read topic history request in topic of " << top_thread_message_id << " in " << dialog_id << " up to " << max_message_id; td_->create_handler()->send(dialog_id, top_thread_message_id, max_message_id); } } void ForumTopicManager::on_update_forum_topic_unread(DialogId dialog_id, MessageId top_thread_message_id, MessageId last_message_id, MessageId last_read_inbox_message_id, MessageId last_read_outbox_message_id, int32 unread_count) { if (td_->auth_manager_->is_bot()) { return; } auto topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr || topic->topic_ == nullptr) { return; } if (topic->topic_->update_last_read_outbox_message_id(last_read_outbox_message_id)) { // TODO send updates } if (topic->topic_->update_last_read_inbox_message_id(last_read_inbox_message_id, unread_count)) { // TODO send updates } } DialogNotificationSettings *ForumTopicManager::get_forum_topic_notification_settings(DialogId dialog_id, MessageId top_thread_message_id) { auto topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr || topic->topic_ == nullptr) { return nullptr; } return topic->topic_->get_notification_settings(); } const DialogNotificationSettings *ForumTopicManager::get_forum_topic_notification_settings( DialogId dialog_id, MessageId top_thread_message_id) const { auto topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr || topic->topic_ == nullptr) { return nullptr; } return topic->topic_->get_notification_settings(); } void ForumTopicManager::on_update_forum_topic_notify_settings( DialogId dialog_id, MessageId top_thread_message_id, tl_object_ptr &&peer_notify_settings, const char *source) { if (td_->auth_manager_->is_bot()) { return; } VLOG(notifications) << "Receive notification settings for topic of " << top_thread_message_id << " in " << dialog_id << " from " << source << ": " << to_string(peer_notify_settings); DialogNotificationSettings *current_settings = get_forum_topic_notification_settings(dialog_id, top_thread_message_id); if (current_settings == nullptr) { return; } auto notification_settings = get_dialog_notification_settings(std::move(peer_notify_settings), current_settings); if (!notification_settings.is_synchronized) { return; } update_forum_topic_notification_settings(dialog_id, top_thread_message_id, current_settings, std::move(notification_settings)); } void ForumTopicManager::on_update_forum_topic_is_pinned(DialogId dialog_id, MessageId top_thread_message_id, bool is_pinned) { if (!td_->dialog_manager_->have_dialog_force(dialog_id, "on_update_forum_topic_is_pinned")) { return; } if (!can_be_forum(dialog_id)) { LOG(ERROR) << "Receive pinned topics in " << dialog_id; return; } if (td_->auth_manager_->is_bot()) { return; } auto topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr || topic->topic_ == nullptr) { return; } if (topic->topic_->set_is_pinned(is_pinned)) { topic->need_save_to_database_ = true; save_topic_to_database(dialog_id, topic); } } void ForumTopicManager::on_update_pinned_forum_topics(DialogId dialog_id, vector top_thread_message_ids) { if (!td_->dialog_manager_->have_dialog_force(dialog_id, "on_update_pinned_forum_topics")) { return; } if (!can_be_forum(dialog_id)) { LOG(ERROR) << "Receive pinned topics in " << dialog_id; return; } if (td_->auth_manager_->is_bot()) { return; } auto dialog_topics = get_dialog_topics(dialog_id); if (dialog_topics == nullptr) { return; } dialog_topics->topics_.foreach([&](const MessageId &top_thread_message_id, unique_ptr &topic) { if (topic->topic_ == nullptr) { return; } if (topic->topic_->set_is_pinned(contains(top_thread_message_ids, top_thread_message_id))) { topic->need_save_to_database_ = true; save_topic_to_database(dialog_id, topic.get()); } }); } Status ForumTopicManager::set_forum_topic_notification_settings( DialogId dialog_id, MessageId top_thread_message_id, tl_object_ptr &¬ification_settings) { CHECK(!td_->auth_manager_->is_bot()); TRY_STATUS(is_forum(dialog_id)); TRY_STATUS(can_be_message_thread_id(top_thread_message_id)); auto current_settings = get_forum_topic_notification_settings(dialog_id, top_thread_message_id); if (current_settings == nullptr) { return Status::Error(400, "Unknown forum topic identifier specified"); } TRY_RESULT(new_settings, get_dialog_notification_settings(std::move(notification_settings), current_settings)); if (update_forum_topic_notification_settings(dialog_id, top_thread_message_id, current_settings, std::move(new_settings))) { // TODO log event td_->notification_settings_manager_->update_dialog_notify_settings(dialog_id, top_thread_message_id, *current_settings, Promise()); } return Status::OK(); } bool ForumTopicManager::update_forum_topic_notification_settings(DialogId dialog_id, MessageId top_thread_message_id, DialogNotificationSettings *current_settings, DialogNotificationSettings &&new_settings) { if (td_->auth_manager_->is_bot()) { // just in case return false; } auto need_update = need_update_dialog_notification_settings(current_settings, new_settings); if (need_update.are_changed) { // TODO update unmute timeouts, td_api updates, remove notifications *current_settings = std::move(new_settings); auto topic = get_topic(dialog_id, top_thread_message_id); CHECK(topic != nullptr); topic->need_save_to_database_ = true; save_topic_to_database(dialog_id, topic); } return need_update.need_update_server; } void ForumTopicManager::get_forum_topic(DialogId dialog_id, MessageId top_thread_message_id, Promise> &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); TRY_STATUS_PROMISE(promise, can_be_message_thread_id(top_thread_message_id)); auto channel_id = dialog_id.get_channel_id(); td_->create_handler(std::move(promise))->send(channel_id, top_thread_message_id); } void ForumTopicManager::on_get_forum_topic(ChannelId channel_id, MessageId expected_top_thread_message_id, MessagesInfo &&info, telegram_api::object_ptr &&topic, Promise> &&promise) { DialogId dialog_id(channel_id); TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); td_->messages_manager_->on_get_messages(std::move(info.messages), true, false, Promise(), "on_get_forum_topic"); auto top_thread_message_id = on_get_forum_topic_impl(dialog_id, std::move(topic)); if (!top_thread_message_id.is_valid()) { return promise.set_value(nullptr); } if (top_thread_message_id != expected_top_thread_message_id) { return promise.set_error(Status::Error(500, "Wrong forum topic received")); } promise.set_value(get_forum_topic_object(dialog_id, top_thread_message_id)); } void ForumTopicManager::get_forum_topic_link(DialogId dialog_id, MessageId top_thread_message_id, Promise> &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); TRY_STATUS_PROMISE(promise, can_be_message_thread_id(top_thread_message_id)); auto channel_id = dialog_id.get_channel_id(); SliceBuilder sb; sb << LinkManager::get_t_me_url(); bool is_public = false; auto dialog_username = td_->contacts_manager_->get_channel_first_username(channel_id); if (!dialog_username.empty()) { sb << dialog_username; is_public = true; } else { sb << "c/" << channel_id.get(); } sb << '/' << top_thread_message_id.get_server_message_id().get(); promise.set_value(td_api::make_object(sb.as_cslice().str(), is_public)); } void ForumTopicManager::get_forum_topics(DialogId dialog_id, string query, int32 offset_date, MessageId offset_message_id, MessageId offset_top_thread_message_id, int32 limit, Promise> promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); auto channel_id = dialog_id.get_channel_id(); if (offset_date < 0) { return promise.set_error(Status::Error(400, "Invalid offset date specified")); } if (offset_message_id != MessageId() && !offset_message_id.is_valid() && !offset_message_id.is_server()) { return promise.set_error(Status::Error(400, "Invalid offset message identifier specified")); } if (offset_top_thread_message_id != MessageId()) { TRY_STATUS_PROMISE(promise, can_be_message_thread_id(offset_top_thread_message_id)); } if (limit <= 0) { return promise.set_error(Status::Error(400, "Invalid limit specified")); } td_->create_handler(std::move(promise)) ->send(channel_id, query, offset_date, offset_message_id, offset_top_thread_message_id, limit); } void ForumTopicManager::on_get_forum_topics(ChannelId channel_id, bool order_by_creation_date, MessagesInfo &&info, vector> &&topics, Promise> &&promise) { DialogId dialog_id(channel_id); TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); td_->messages_manager_->on_get_messages(std::move(info.messages), true, false, Promise(), "on_get_forum_topics"); vector> forum_topics; int32 next_offset_date = 0; MessageId next_offset_message_id; MessageId next_offset_top_thread_message_id; for (auto &topic : topics) { auto top_thread_message_id = on_get_forum_topic_impl(dialog_id, std::move(topic)); if (!top_thread_message_id.is_valid()) { continue; } auto forum_topic_object = get_forum_topic_object(dialog_id, top_thread_message_id); CHECK(forum_topic_object != nullptr); if (order_by_creation_date || forum_topic_object->last_message_ == nullptr) { next_offset_date = forum_topic_object->info_->creation_date_; } else { next_offset_date = forum_topic_object->last_message_->date_; } next_offset_message_id = forum_topic_object->last_message_ != nullptr ? MessageId(forum_topic_object->last_message_->id_) : MessageId(); next_offset_top_thread_message_id = top_thread_message_id; forum_topics.push_back(std::move(forum_topic_object)); } promise.set_value(td_api::make_object(info.total_count, std::move(forum_topics), next_offset_date, next_offset_message_id.get(), next_offset_top_thread_message_id.get())); } void ForumTopicManager::toggle_forum_topic_is_closed(DialogId dialog_id, MessageId top_thread_message_id, bool is_closed, Promise &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); TRY_STATUS_PROMISE(promise, can_be_message_thread_id(top_thread_message_id)); auto channel_id = dialog_id.get_channel_id(); if (!td_->contacts_manager_->get_channel_permissions(channel_id).can_edit_topics()) { auto topic_info = get_topic_info(dialog_id, top_thread_message_id); if (topic_info != nullptr && !topic_info->is_outgoing()) { return promise.set_error(Status::Error(400, "Not enough rights to close or open the topic")); } } td_->create_handler(std::move(promise))->send(channel_id, top_thread_message_id, is_closed); } void ForumTopicManager::toggle_forum_topic_is_hidden(DialogId dialog_id, bool is_hidden, Promise &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); auto channel_id = dialog_id.get_channel_id(); if (!td_->contacts_manager_->get_channel_permissions(channel_id).can_edit_topics()) { return promise.set_error(Status::Error(400, "Not enough rights to close or open the topic")); } td_->create_handler(std::move(promise))->send(channel_id, is_hidden); } void ForumTopicManager::toggle_forum_topic_is_pinned(DialogId dialog_id, MessageId top_thread_message_id, bool is_pinned, Promise &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); TRY_STATUS_PROMISE(promise, can_be_message_thread_id(top_thread_message_id)); auto channel_id = dialog_id.get_channel_id(); if (!td_->contacts_manager_->get_channel_permissions(channel_id).can_pin_topics()) { return promise.set_error(Status::Error(400, "Not enough rights to pin or unpin the topic")); } td_->create_handler(std::move(promise)) ->send(channel_id, top_thread_message_id, is_pinned); } void ForumTopicManager::set_pinned_forum_topics(DialogId dialog_id, vector top_thread_message_ids, Promise &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); for (auto top_thread_message_id : top_thread_message_ids) { TRY_STATUS_PROMISE(promise, can_be_message_thread_id(top_thread_message_id)); } auto channel_id = dialog_id.get_channel_id(); if (!td_->contacts_manager_->get_channel_permissions(channel_id).can_pin_topics()) { return promise.set_error(Status::Error(400, "Not enough rights to reorder forum topics")); } td_->create_handler(std::move(promise))->send(channel_id, top_thread_message_ids); } void ForumTopicManager::delete_forum_topic(DialogId dialog_id, MessageId top_thread_message_id, Promise &&promise) { TRY_STATUS_PROMISE(promise, is_forum(dialog_id)); TRY_STATUS_PROMISE(promise, can_be_message_thread_id(top_thread_message_id)); auto channel_id = dialog_id.get_channel_id(); if (!td_->contacts_manager_->get_channel_permissions(channel_id).can_delete_messages()) { auto topic_info = get_topic_info(dialog_id, top_thread_message_id); if (topic_info != nullptr && !topic_info->is_outgoing()) { return promise.set_error(Status::Error(400, "Not enough rights to delete the topic")); } } auto delete_promise = PromiseCreator::lambda([actor_id = actor_id(this), dialog_id, top_thread_message_id, promise = std::move(promise)](Result result) mutable { if (result.is_error()) { return promise.set_error(result.move_as_error()); } send_closure(actor_id, &ForumTopicManager::on_delete_forum_topic, dialog_id, top_thread_message_id, std::move(promise)); }); td_->messages_manager_->delete_topic_history(dialog_id, top_thread_message_id, std::move(delete_promise)); } void ForumTopicManager::on_delete_forum_topic(DialogId dialog_id, MessageId top_thread_message_id, Promise &&promise) { TRY_STATUS_PROMISE(promise, G()->close_status()); auto *dialog_topics = dialog_topics_.get_pointer(dialog_id); if (dialog_topics != nullptr) { dialog_topics->topics_.erase(top_thread_message_id); dialog_topics->deleted_topic_ids_.insert(top_thread_message_id); } delete_topic_from_database(dialog_id, top_thread_message_id, std::move(promise)); } void ForumTopicManager::delete_all_dialog_topics(DialogId dialog_id) { dialog_topics_.erase(dialog_id); auto message_thread_db = G()->td_db()->get_message_thread_db_async(); if (message_thread_db == nullptr) { return; } LOG(INFO) << "Delete all topics in " << dialog_id << " from database"; message_thread_db->delete_all_dialog_message_threads(dialog_id, Auto()); } void ForumTopicManager::on_forum_topic_edited(DialogId dialog_id, MessageId top_thread_message_id, const ForumTopicEditedData &edited_data) { auto topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr || topic->info_ == nullptr) { return; } if (topic->info_->apply_edited_data(edited_data)) { send_update_forum_topic_info(dialog_id, topic->info_.get()); topic->need_save_to_database_ = true; } save_topic_to_database(dialog_id, topic); } void ForumTopicManager::on_get_forum_topic_info(DialogId dialog_id, const ForumTopicInfo &topic_info, const char *source) { if (!can_be_forum(dialog_id)) { LOG(ERROR) << "Receive forum topics in " << dialog_id << " from " << source; return; } auto dialog_topics = add_dialog_topics(dialog_id); CHECK(dialog_topics != nullptr); auto forum_topic_info = td::make_unique(topic_info); MessageId top_thread_message_id = forum_topic_info->get_top_thread_message_id(); CHECK(can_be_message_thread_id(top_thread_message_id).is_ok()); auto topic = add_topic(dialog_topics, top_thread_message_id); if (topic == nullptr) { return; } set_topic_info(dialog_id, topic, std::move(forum_topic_info)); save_topic_to_database(dialog_id, topic); } void ForumTopicManager::on_get_forum_topic_infos(DialogId dialog_id, vector> &&forum_topics, const char *source) { if (forum_topics.empty()) { return; } if (!can_be_forum(dialog_id)) { LOG(ERROR) << "Receive forum topics in " << dialog_id << " from " << source; return; } auto dialog_topics = add_dialog_topics(dialog_id); CHECK(dialog_topics != nullptr); for (auto &forum_topic : forum_topics) { auto forum_topic_info = td::make_unique(td_, forum_topic); MessageId top_thread_message_id = forum_topic_info->get_top_thread_message_id(); if (can_be_message_thread_id(top_thread_message_id).is_error()) { continue; } auto topic = add_topic(dialog_topics, top_thread_message_id); if (topic != nullptr) { set_topic_info(dialog_id, topic, std::move(forum_topic_info)); save_topic_to_database(dialog_id, topic); } } } MessageId ForumTopicManager::on_get_forum_topic_impl(DialogId dialog_id, tl_object_ptr &&forum_topic) { CHECK(forum_topic != nullptr); switch (forum_topic->get_id()) { case telegram_api::forumTopicDeleted::ID: { auto top_thread_message_id = MessageId(ServerMessageId(static_cast(forum_topic.get())->id_)); if (!top_thread_message_id.is_valid()) { LOG(ERROR) << "Receive " << to_string(forum_topic); return MessageId(); } on_delete_forum_topic(dialog_id, top_thread_message_id, Promise()); return MessageId(); } case telegram_api::forumTopic::ID: { auto forum_topic_info = td::make_unique(td_, forum_topic); MessageId top_thread_message_id = forum_topic_info->get_top_thread_message_id(); Topic *topic = add_topic(dialog_id, top_thread_message_id); if (topic == nullptr) { return MessageId(); } auto current_notification_settings = topic->topic_ == nullptr ? nullptr : topic->topic_->get_notification_settings(); auto forum_topic_full = td::make_unique(td_, std::move(forum_topic), current_notification_settings); if (forum_topic_full->is_short()) { LOG(ERROR) << "Receive short " << to_string(forum_topic); return MessageId(); } if (topic->topic_ == nullptr || true) { topic->topic_ = std::move(forum_topic_full); topic->need_save_to_database_ = true; // temporary } set_topic_info(dialog_id, topic, std::move(forum_topic_info)); save_topic_to_database(dialog_id, topic); return top_thread_message_id; } default: UNREACHABLE(); return MessageId(); } } td_api::object_ptr ForumTopicManager::get_forum_topic_object( DialogId dialog_id, MessageId top_thread_message_id) const { auto topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr || topic->topic_ == nullptr) { return nullptr; } CHECK(topic->info_ != nullptr); return topic->topic_->get_forum_topic_object(td_, dialog_id, *topic->info_); } Status ForumTopicManager::is_forum(DialogId dialog_id) { if (!td_->dialog_manager_->have_dialog_force(dialog_id, "ForumTopicManager::is_forum")) { return Status::Error(400, "Chat not found"); } if (dialog_id.get_type() != DialogType::Channel || !td_->contacts_manager_->is_forum_channel(dialog_id.get_channel_id())) { return Status::Error(400, "The chat is not a forum"); } return Status::OK(); } bool ForumTopicManager::can_be_forum(DialogId dialog_id) const { return dialog_id.get_type() == DialogType::Channel && td_->contacts_manager_->is_megagroup_channel(dialog_id.get_channel_id()); } Status ForumTopicManager::can_be_message_thread_id(MessageId top_thread_message_id) { if (!top_thread_message_id.is_valid() || !top_thread_message_id.is_server()) { return Status::Error(400, "Invalid message thread identifier specified"); } return Status::OK(); } ForumTopicManager::DialogTopics *ForumTopicManager::add_dialog_topics(DialogId dialog_id) { auto *dialog_topics = dialog_topics_.get_pointer(dialog_id); if (dialog_topics == nullptr) { auto new_dialog_topics = make_unique(); dialog_topics = new_dialog_topics.get(); dialog_topics_.set(dialog_id, std::move(new_dialog_topics)); } return dialog_topics; } ForumTopicManager::DialogTopics *ForumTopicManager::get_dialog_topics(DialogId dialog_id) { return dialog_topics_.get_pointer(dialog_id); } ForumTopicManager::Topic *ForumTopicManager::add_topic(DialogTopics *dialog_topics, MessageId top_thread_message_id) { auto topic = dialog_topics->topics_.get_pointer(top_thread_message_id); if (topic == nullptr) { if (dialog_topics->deleted_topic_ids_.count(top_thread_message_id) > 0) { return nullptr; } auto new_topic = make_unique(); topic = new_topic.get(); dialog_topics->topics_.set(top_thread_message_id, std::move(new_topic)); } return topic; } ForumTopicManager::Topic *ForumTopicManager::get_topic(DialogTopics *dialog_topics, MessageId top_thread_message_id) { return dialog_topics->topics_.get_pointer(top_thread_message_id); } ForumTopicManager::Topic *ForumTopicManager::add_topic(DialogId dialog_id, MessageId top_thread_message_id) { return add_topic(add_dialog_topics(dialog_id), top_thread_message_id); } ForumTopicManager::Topic *ForumTopicManager::get_topic(DialogId dialog_id, MessageId top_thread_message_id) { auto *dialog_topics = dialog_topics_.get_pointer(dialog_id); if (dialog_topics == nullptr) { return nullptr; } return dialog_topics->topics_.get_pointer(top_thread_message_id); } const ForumTopicManager::Topic *ForumTopicManager::get_topic(DialogId dialog_id, MessageId top_thread_message_id) const { auto *dialog_topics = dialog_topics_.get_pointer(dialog_id); if (dialog_topics == nullptr) { return nullptr; } return dialog_topics->topics_.get_pointer(top_thread_message_id); } ForumTopicInfo *ForumTopicManager::get_topic_info(DialogId dialog_id, MessageId top_thread_message_id) { auto *topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr) { return nullptr; } return topic->info_.get(); } const ForumTopicInfo *ForumTopicManager::get_topic_info(DialogId dialog_id, MessageId top_thread_message_id) const { auto *topic = get_topic(dialog_id, top_thread_message_id); if (topic == nullptr) { return nullptr; } return topic->info_.get(); } void ForumTopicManager::set_topic_info(DialogId dialog_id, Topic *topic, unique_ptr forum_topic_info) { if (topic->info_ == nullptr || *topic->info_ != *forum_topic_info) { topic->info_ = std::move(forum_topic_info); send_update_forum_topic_info(dialog_id, topic->info_.get()); topic->need_save_to_database_ = true; } } td_api::object_ptr ForumTopicManager::get_update_forum_topic_info( DialogId dialog_id, const ForumTopicInfo *topic_info) const { return td_api::make_object( td_->dialog_manager_->get_chat_id_object(dialog_id, "updateForumTopicInfo"), topic_info->get_forum_topic_info_object(td_)); } void ForumTopicManager::send_update_forum_topic_info(DialogId dialog_id, const ForumTopicInfo *topic_info) const { if (td_->auth_manager_->is_bot()) { return; } send_closure(G()->td(), &Td::send_update, get_update_forum_topic_info(dialog_id, topic_info)); } void ForumTopicManager::save_topic_to_database(DialogId dialog_id, const Topic *topic) { CHECK(topic != nullptr); if (topic->info_ == nullptr || !topic->need_save_to_database_) { return; } topic->need_save_to_database_ = false; auto message_thread_db = G()->td_db()->get_message_thread_db_async(); if (message_thread_db == nullptr) { return; } auto top_thread_message_id = topic->info_->get_top_thread_message_id(); LOG(INFO) << "Save topic of " << top_thread_message_id << " in " << dialog_id << " to database"; message_thread_db->add_message_thread(dialog_id, top_thread_message_id, 0, log_event_store(*topic), Auto()); } void ForumTopicManager::delete_topic_from_database(DialogId dialog_id, MessageId top_thread_message_id, Promise &&promise) { auto message_thread_db = G()->td_db()->get_message_thread_db_async(); if (message_thread_db == nullptr) { return promise.set_value(Unit()); } LOG(INFO) << "Delete topic of " << top_thread_message_id << " in " << dialog_id << " from database"; message_thread_db->delete_message_thread(dialog_id, top_thread_message_id, std::move(promise)); } void ForumTopicManager::on_topic_message_count_changed(DialogId dialog_id, MessageId top_thread_message_id, int diff) { if (!can_be_forum(dialog_id) || can_be_message_thread_id(top_thread_message_id).is_error()) { LOG(ERROR) << "Change by " << diff << " number of loaded messages in thread of " << top_thread_message_id << " in " << dialog_id; return; } LOG(INFO) << "Change by " << diff << " number of loaded messages in thread of " << top_thread_message_id << " in " << dialog_id; auto dialog_topics = add_dialog_topics(dialog_id); auto topic = add_topic(dialog_topics, top_thread_message_id); if (topic == nullptr) { return; } topic->message_count_ += diff; CHECK(topic->message_count_ >= 0); if (topic->message_count_ == 0) { // TODO keep topics in the topic list dialog_topics->topics_.erase(top_thread_message_id); } } void ForumTopicManager::memory_stats(vector &output) { output.emplace_back("\"dialog_topics_\":"); output.emplace_back(std::to_string(this->dialog_topics_.calc_size())); } } // namespace td