Add reload_saved_animations.

GitOrigin-RevId: e69e2d60bc5464b880ed478dedc084656a1e878d

td/telegram/AnimationsManager.cpp

@@ -36,8 +36,11 @@
 namespace td {
 
 class GetSavedGifsQuery : public Td::ResultHandler {
+  bool is_reload_ = false;
+
  public:
-  void send(int32 hash) {
+  void send(bool is_reload, int32 hash) {
+    is_reload_ = is_reload;
     LOG(INFO) << "Send get saved animations request with hash = " << hash;
     send_query(G()->net_query_creator().create(create_storer(telegram_api::messages_getSavedGifs(hash))));
   }
@@ -49,12 +52,12 @@ class GetSavedGifsQuery : public Td::ResultHandler {
     }
 
     auto ptr = result_ptr.move_as_ok();
-    td->animations_manager_->on_get_saved_animations(std::move(ptr));
+    td->animations_manager_->on_get_saved_animations(is_reload_, std::move(ptr));
   }
 
   void on_error(uint64 id, Status status) override {
     LOG(ERROR) << "Receive error for get saved animations: " << status;
-    td->animations_manager_->on_get_saved_animations_failed(std::move(status));
+    td->animations_manager_->on_get_saved_animations_failed(is_reload_, std::move(status));
   }
 };
 
@@ -402,7 +405,18 @@ void AnimationsManager::reload_saved_animations(bool force) {
       (next_saved_animations_load_time_ < Time::now() || force)) {
     LOG_IF(INFO, force) << "Reload saved animations";
     next_saved_animations_load_time_ = -1;
-    td_->create_handler<GetSavedGifsQuery>()->send(get_saved_animations_hash("reload_saved_animations"));
+    td_->create_handler<GetSavedGifsQuery>()->send(false, get_saved_animations_hash("reload_saved_animations"));
+  }
+}
+
+void AnimationsManager::reload_saved_animations(Promise<Unit> &&promise) {
+  if (td_->auth_manager_->is_bot()) {
+    return promise.set_error(Status::Error(400, "Bots has no saved animations"));
+  }
+
+  reload_saved_animations_queries_.push_back(std::move(promise));
+  if (reload_saved_animations_queries_.size() == 1u) {
+    td_->create_handler<GetSavedGifsQuery>()->send(true, 0);
   }
 }
 
@@ -471,13 +485,18 @@ void AnimationsManager::on_load_saved_animations_finished(vector<FileId> &&saved
 }
 
 void AnimationsManager::on_get_saved_animations(
-    tl_object_ptr<telegram_api::messages_SavedGifs> &&saved_animations_ptr) {
+    bool is_reload, tl_object_ptr<telegram_api::messages_SavedGifs> &&saved_animations_ptr) {
   CHECK(!td_->auth_manager_->is_bot());
+  if (!is_reload) {
     next_saved_animations_load_time_ = Time::now_cached() + Random::fast(30 * 60, 50 * 60);
+  }
 
   CHECK(saved_animations_ptr != nullptr);
   int32 constructor_id = saved_animations_ptr->get_id();
   if (constructor_id == telegram_api::messages_savedGifsNotModified::ID) {
+    if (is_reload) {
+      return on_get_saved_animations_failed(true, Status::Error(500, "Failed to reload saved animations"));
+    }
     LOG(INFO) << "Saved animations are not modified";
     return;
   }
@@ -500,21 +519,34 @@ void AnimationsManager::on_get_saved_animations(
       LOG(ERROR) << "Receive " << static_cast<int>(document.first) << " instead of animation as saved animation";
       continue;
     }
+    if (!is_reload) {
       saved_animation_ids.push_back(document.second);
     }
+  }
 
+  if (is_reload) {
+    auto promises = std::move(reload_saved_animations_queries_);
+    reload_saved_animations_queries_.clear();
+    for (auto &promise : promises) {
+      promise.set_value(Unit());
+    }
+  } else {
     on_load_saved_animations_finished(std::move(saved_animation_ids));
 
     LOG_IF(ERROR, get_saved_animations_hash("on_get_saved_animations") != saved_animations->hash_)
         << "Saved animations hash mismatch: " << saved_animations->hash_ << " vs "
         << get_saved_animations_hash("on_get_saved_animations 2");
+  }
 }
 
-void AnimationsManager::on_get_saved_animations_failed(Status error) {
+void AnimationsManager::on_get_saved_animations_failed(bool is_reload, Status error) {
   CHECK(error.is_error());
+  if (!is_reload) {
     next_saved_animations_load_time_ = Time::now_cached() + Random::fast(5, 10);
-  auto promises = std::move(load_saved_animations_queries_);
+  }
-  load_saved_animations_queries_.clear();
+  auto &queries = is_reload ? reload_saved_animations_queries_ : load_saved_animations_queries_;
+  auto promises = std::move(queries);
+  queries.clear();
   for (auto &promise : promises) {
     promise.set_error(error.clone());
   }

td/telegram/AnimationsManager.h

@@ -59,9 +59,11 @@ class AnimationsManager : public Actor {
 
   void reload_saved_animations(bool force);
 
-  void on_get_saved_animations(tl_object_ptr<telegram_api::messages_SavedGifs> &&saved_animations_ptr);
+  void reload_saved_animations(Promise<Unit> &&promise);
 
-  void on_get_saved_animations_failed(Status error);
+  void on_get_saved_animations(bool is_reload, tl_object_ptr<telegram_api::messages_SavedGifs> &&saved_animations_ptr);
+
+  void on_get_saved_animations_failed(bool is_reload, Status error);
 
   vector<FileId> get_saved_animations(Promise<Unit> &&promise);
 
@@ -133,6 +135,7 @@ class AnimationsManager : public Actor {
   double next_saved_animations_load_time_ = 0;
   bool are_saved_animations_loaded_ = false;
   vector<Promise<Unit>> load_saved_animations_queries_;
+  vector<Promise<Unit>> reload_saved_animations_queries_;
 };
 
 }  // namespace td