Store bot_user_ids in ChannelFull.

GitOrigin-RevId: 46b118c7a1e976bffc28fec5a7c2eb369317ee36

td/telegram/ContactsManager.cpp

@@ -3221,6 +3221,7 @@ void ContactsManager::ChannelFull::store(StorerT &storer) const {
   bool has_migrated_from_max_message_id = migrated_from_max_message_id.is_valid();
   bool has_migrated_from_chat_id = migrated_from_chat_id.is_valid();
   bool has_location = !location.empty();
+  bool has_bot_user_ids = !bot_user_ids.empty();
   BEGIN_STORE_FLAGS();
   STORE_FLAG(has_description);
   STORE_FLAG(has_administrator_count);
@@ -3238,6 +3239,7 @@ void ContactsManager::ChannelFull::store(StorerT &storer) const {
   STORE_FLAG(is_all_history_available);
   STORE_FLAG(can_set_location);
   STORE_FLAG(has_location);
+  STORE_FLAG(has_bot_user_ids);
   END_STORE_FLAGS();
   if (has_description) {
     store(description, storer);
@@ -3264,6 +3266,9 @@ void ContactsManager::ChannelFull::store(StorerT &storer) const {
   if (has_location) {
     store(location, storer);
   }
+  if (has_bot_user_ids) {
+    store(bot_user_ids, storer);
+  }
   if (has_migrated_from_max_message_id) {
     store(migrated_from_max_message_id, storer);
   }
@@ -3286,6 +3291,7 @@ void ContactsManager::ChannelFull::parse(ParserT &parser) {
   bool has_migrated_from_max_message_id;
   bool has_migrated_from_chat_id;
   bool has_location;
+  bool has_bot_user_ids;
   BEGIN_PARSE_FLAGS();
   PARSE_FLAG(has_description);
   PARSE_FLAG(has_administrator_count);
@@ -3303,6 +3309,7 @@ void ContactsManager::ChannelFull::parse(ParserT &parser) {
   PARSE_FLAG(is_all_history_available);
   PARSE_FLAG(can_set_location);
   PARSE_FLAG(has_location);
+  PARSE_FLAG(has_bot_user_ids);
   END_PARSE_FLAGS();
   if (has_description) {
     parse(description, parser);
@@ -3329,6 +3336,9 @@ void ContactsManager::ChannelFull::parse(ParserT &parser) {
   if (has_location) {
     parse(location, parser);
   }
+  if (has_bot_user_ids) {
+    parse(bot_user_ids, parser);
+  }
   if (has_migrated_from_max_message_id) {
     parse(migrated_from_max_message_id, parser);
   }
@@ -7247,9 +7257,7 @@ void ContactsManager::on_load_user_full_from_database(UserId user_id, string val
   if (user_full->need_phone_number_privacy_exception && is_user_contact(user_id)) {
     user_full->need_phone_number_privacy_exception = false;
   }
-  if (is_user_bot(user_id)) {
+  get_bot_info_force(user_id, false);
-    get_bot_info_force(user_id, false);
-  }
 
   update_user_full(user_full, user_id, true);
 
@@ -7415,6 +7423,10 @@ void ContactsManager::on_load_chat_full_from_database(ChatId chat_id, string val
   }
   td_->messages_manager_->resolve_dependencies_force(dependencies);
 
+  for (auto &participant : chat_full->participants) {
+    get_bot_info_force(participant.user_id);
+  }
+
   update_chat_full(chat_full, chat_id, true);
 }
 
@@ -7484,8 +7496,13 @@ void ContactsManager::on_load_channel_full_from_database(ChannelId channel_id, s
   dependencies.channel_ids.insert(channel_id);
   td_->messages_manager_->add_dialog_dependencies(dependencies, DialogId(channel_full->linked_channel_id));
   dependencies.chat_ids.insert(channel_full->migrated_from_chat_id);
+  dependencies.user_ids.insert(channel_full->bot_user_ids.begin(), channel_full->bot_user_ids.end());
   td_->messages_manager_->resolve_dependencies_force(dependencies);
 
+  for (auto &user_id : channel_full->bot_user_ids) {
+    get_bot_info_force(user_id);
+  }
+
   update_channel_full(channel_full, channel_id, true);
 }
 
@@ -7871,6 +7888,8 @@ void ContactsManager::update_chat_full(ChatFull *chat_full, ChatId chat_id, bool
 void ContactsManager::update_channel_full(ChannelFull *channel_full, ChannelId channel_id, bool from_database) {
   CHECK(channel_full != nullptr);
   unavailable_channel_fulls_.erase(channel_id);  // don't needed anymore
+
+  channel_full->need_save_to_database |= channel_full->is_changed;
   if (channel_full->is_changed) {
     if (channel_full->linked_channel_id.is_valid()) {
       td_->messages_manager_->force_create_dialog(DialogId(channel_full->linked_channel_id), "update_channel_full");
@@ -7883,10 +7902,10 @@ void ContactsManager::update_channel_full(ChannelFull *channel_full, ChannelId c
         G()->td(), &Td::send_update,
         make_tl_object<td_api::updateSupergroupFullInfo>(get_supergroup_id_object(channel_id, "update_channel_full"),
                                                          get_supergroup_full_info_object(channel_full)));
-
+  }
-    if (!from_database) {
+  if (!from_database && channel_full->need_save_to_database) {
-      save_channel_full(channel_full, channel_id);
+    channel_full->need_save_to_database = false;
-    }
+    save_channel_full(channel_full, channel_id);
   }
 }
 
@@ -8274,9 +8293,17 @@ void ContactsManager::on_get_chat_full(tl_object_ptr<telegram_api::ChatFull> &&c
       td_->messages_manager_->on_update_dialog_online_member_count(DialogId(channel_id), online_member_count, true);
     }
 
+    vector<UserId> bot_user_ids;
     for (auto &bot_info : channel_full->bot_info_) {
+      UserId user_id(bot_info->user_id_);
+      if (!is_user_bot(user_id)) {
+        continue;
+      }
+
+      bot_user_ids.push_back(user_id);
       on_update_bot_info(std::move(bot_info));
     }
+    on_update_channel_full_bot_user_ids(channel, channel_id, std::move(bot_user_ids));
 
     ChannelId linked_channel_id;
     if ((channel_full->flags_ & CHANNEL_FULL_FLAG_HAS_LINKED_CHANNEL_ID) != 0) {
@@ -9115,7 +9142,7 @@ void ContactsManager::on_get_channel_participants_success(
       on_update_dialog_administrators(DialogId(channel_id), std::move(administrator_user_ids), true);
     }
     if (filter.is_bots() || filter.is_recent()) {
-      td_->messages_manager_->on_dialog_bots_updated(DialogId(channel_id), std::move(bot_user_ids));
+      on_update_channel_bot_user_ids(channel_id, std::move(bot_user_ids));
     }
   }
 
@@ -9171,6 +9198,7 @@ bool ContactsManager::speculative_add_count(int32 &count, int32 new_count) {
 void ContactsManager::speculative_add_channel_participants(ChannelId channel_id, const vector<UserId> &added_user_ids,
                                                            UserId inviter_user_id, int32 date, bool by_me) {
   auto it = cached_channel_participants_.find(channel_id);
+  auto channel_full = get_channel_full_force(channel_id);
   bool is_participants_cache_changed = false;
 
   int32 new_participant_count = 0;
@@ -9195,10 +9223,20 @@ void ContactsManager::speculative_add_channel_participants(ChannelId channel_id,
         participants.emplace_back(user_id, inviter_user_id, date, DialogParticipantStatus::Member());
       }
     }
+
+    if (channel_full != nullptr && is_user_bot(user_id) &&
+        std::find(channel_full->bot_user_ids.begin(), channel_full->bot_user_ids.end(), user_id) ==
+            channel_full->bot_user_ids.end()) {
+      channel_full->bot_user_ids.push_back(user_id);
+      channel_full->need_save_to_database = true;
+    }
   }
   if (is_participants_cache_changed) {
     update_channel_online_member_count(channel_id, false);
   }
+  if (channel_full != nullptr) {
+    update_channel_full(channel_full, channel_id);
+  }
   if (new_participant_count == 0) {
     return;
   }
@@ -9223,6 +9261,18 @@ void ContactsManager::speculative_delete_channel_participant(ChannelId channel_i
     }
   }
 
+  if (is_user_bot(deleted_user_id)) {
+    auto channel_full = get_channel_full_force(channel_id);
+    if (channel_full != nullptr) {
+      auto user_it = std::find(channel_full->bot_user_ids.begin(), channel_full->bot_user_ids.end(), deleted_user_id);
+      if (user_it != channel_full->bot_user_ids.end()) {
+        channel_full->bot_user_ids.erase(user_it);
+        channel_full->need_save_to_database = true;
+        update_channel_full(channel_full, channel_id);
+      }
+    }
+  }
+
   speculative_add_channel_participants(channel_id, -1, by_me);
 }
 
@@ -9315,6 +9365,22 @@ void ContactsManager::speculative_add_channel_user(ChannelId channel_id, UserId
   channel_full->is_changed |=
       speculative_add_count(channel_full->banned_count, new_status.is_banned() - old_status.is_banned());
 
+  if (new_status.is_member() != old_status.is_member() && is_user_bot(user_id)) {
+    if (new_status.is_member()) {
+      if (std::find(channel_full->bot_user_ids.begin(), channel_full->bot_user_ids.end(), user_id) ==
+          channel_full->bot_user_ids.end()) {
+        channel_full->bot_user_ids.push_back(user_id);
+        channel_full->need_save_to_database = true;
+      }
+    } else {
+      auto user_it = std::find(channel_full->bot_user_ids.begin(), channel_full->bot_user_ids.end(), user_id);
+      if (user_it != channel_full->bot_user_ids.end()) {
+        channel_full->bot_user_ids.erase(user_it);
+        channel_full->need_save_to_database = true;
+      }
+    }
+  }
+
   update_channel_full(channel_full, channel_id);
 }
 
@@ -10229,6 +10295,32 @@ void ContactsManager::on_update_channel_location(ChannelId channel_id, const Dia
   }
 }
 
+void ContactsManager::on_update_channel_bot_user_ids(ChannelId channel_id, vector<UserId> &&bot_user_ids) {
+  CHECK(channel_id.is_valid());
+  if (!have_channel(channel_id)) {
+    LOG(ERROR) << channel_id << " not found";
+    return;
+  }
+
+  auto channel_full = get_channel_full_force(channel_id);
+  if (channel_full == nullptr) {
+    td_->messages_manager_->on_dialog_bots_updated(DialogId(channel_id), std::move(bot_user_ids));
+    return;
+  }
+  on_update_channel_full_bot_user_ids(channel_full, channel_id, std::move(bot_user_ids));
+  update_channel_full(channel_full, channel_id);
+}
+
+void ContactsManager::on_update_channel_full_bot_user_ids(ChannelFull *channel_full, ChannelId channel_id,
+                                                          vector<UserId> &&bot_user_ids) {
+  CHECK(channel_full != nullptr);
+  if (channel_full->bot_user_ids != bot_user_ids) {
+    td_->messages_manager_->on_dialog_bots_updated(DialogId(channel_id), bot_user_ids);
+    channel_full->bot_user_ids = std::move(bot_user_ids);
+    channel_full->need_save_to_database = true;
+  }
+}
+
 void ContactsManager::on_update_channel_is_all_history_available(ChannelId channel_id, bool is_all_history_available) {
   if (!channel_id.is_valid()) {
     LOG(ERROR) << "Receive invalid " << channel_id;

td/telegram/ContactsManager.h

@@ -737,6 +737,8 @@ class ContactsManager : public Actor {
     MessageId migrated_from_max_message_id;
     ChatId migrated_from_chat_id;
 
+    vector<UserId> bot_user_ids;
+
     bool can_get_participants = false;
     bool can_set_username = false;
     bool can_set_sticker_set = false;
@@ -744,7 +746,8 @@ class ContactsManager : public Actor {
     bool can_view_statistics = false;
     bool is_all_history_available = true;
 
-    bool is_changed = true;
+    bool is_changed = true;             // have new changes that needs to be sent to the client and database
+    bool need_save_to_database = true;  // have new changes that needs only to be saved to database
 
     double expires_at = 0.0;
     bool is_expired() const;
@@ -1044,11 +1047,15 @@ class ContactsManager : public Actor {
   void on_update_channel_status(Channel *c, ChannelId channel_id, DialogParticipantStatus &&status);
   void on_update_channel_default_permissions(Channel *c, ChannelId channel_id, RestrictedRights default_permissions);
 
+  void on_update_channel_bot_user_ids(ChannelId channel_id, vector<UserId> &&bot_user_ids);
+
   void on_update_channel_full_invite_link(ChannelFull *channel_full,
                                           tl_object_ptr<telegram_api::ExportedChatInvite> &&invite_link_ptr);
   void on_update_channel_full_linked_channel_id(ChannelFull *channel_full, ChannelId channel_id,
                                                 ChannelId linked_channel_id);
   void on_update_channel_full_location(ChannelFull *channel_full, ChannelId channel_id, const DialogLocation &location);
+  void on_update_channel_full_bot_user_ids(ChannelFull *channel_full, ChannelId channel_id,
+                                           vector<UserId> &&bot_user_ids);
 
   static bool speculative_add_count(int32 &count, int32 new_count);