Save autosave settings to database.

td/generate/scheme/td_api.tl

@@ -7849,10 +7849,10 @@ setAutoDownloadSettings settings:autoDownloadSettings type:NetworkType = Ok;
 //@description Returns autosave settings for the current user
 getAutosaveSettings = AutosaveSettings;
 
-//@description Sets autosave settings for the given scope @scope Autosave settings scope @settings New autosave settings for the scope; pass null to set autosave settings to default
+//@description Sets autosave settings for the given scope. The method is guaranteed to work only after at least one call to getAutosaveSettings @scope Autosave settings scope @settings New autosave settings for the scope; pass null to set autosave settings to default
 setAutosaveSettings scope:AutosaveSettingsScope settings:scopeAutosaveSettings = Ok;
 
-//@description Clears the list of all autosave settings exceptions
+//@description Clears the list of all autosave settings exceptions. The method is guaranteed to work only after at least one call to getAutosaveSettings
 clearAutosaveSettingsExceptions = Ok;
 
 

td/telegram/AutosaveManager.cpp

@@ -7,9 +7,12 @@
 #include "td/telegram/AutosaveManager.h"
 
 #include "td/telegram/ContactsManager.h"
+#include "td/telegram/Dependencies.h"
 #include "td/telegram/MessagesManager.h"
 #include "td/telegram/Td.h"
 
+#include "td/db/SqliteKeyValueAsync.h"
+
 #include "td/utils/algorithm.h"
 #include "td/utils/buffer.h"
 #include "td/utils/misc.h"
@@ -88,7 +91,7 @@ class SaveAutoSaveSettingsQuery final : public Td::ResultHandler {
 
   void on_error(Status status) final {
     promise_.set_error(std::move(status));
-    td_->autosave_manager_->reload_autosave_settings(Auto());
+    td_->autosave_manager_->reload_autosave_settings();
   }
 };
 
@@ -114,7 +117,7 @@ class DeleteAutoSaveExceptionsQuery final : public Td::ResultHandler {
 
   void on_error(Status status) final {
     promise_.set_error(std::move(status));
-    td_->autosave_manager_->reload_autosave_settings(Auto());
+    td_->autosave_manager_->reload_autosave_settings();
   }
 };
 
@@ -181,6 +184,26 @@ bool AutosaveManager::DialogAutosaveSettings::operator!=(const DialogAutosaveSet
   return !operator==(other);
 }
 
+template <class StorerT>
+void AutosaveManager::DialogAutosaveSettings::store(StorerT &storer) const {
+  CHECK(are_inited_);
+  BEGIN_STORE_FLAGS();
+  STORE_FLAG(autosave_photos_);
+  STORE_FLAG(autosave_videos_);
+  END_STORE_FLAGS();
+  td::store(max_video_file_size_, storer);
+}
+
+template <class ParserT>
+void AutosaveManager::DialogAutosaveSettings::parse(ParserT &parser) {
+  are_inited_ = true;
+  BEGIN_PARSE_FLAGS();
+  PARSE_FLAG(autosave_photos_);
+  PARSE_FLAG(autosave_videos_);
+  END_PARSE_FLAGS();
+  td::parse(max_video_file_size_, parser);
+}
+
 td_api::object_ptr<td_api::autosaveSettings> AutosaveManager::AutosaveSettings::get_autosave_settings_object() const {
   CHECK(are_inited_);
   auto exceptions = transform(exceptions_, [](const auto &exception) {
@@ -191,20 +214,140 @@ td_api::object_ptr<td_api::autosaveSettings> AutosaveManager::AutosaveSettings::
       broadcast_settings_.get_scope_autosave_settings_object(), std::move(exceptions));
 }
 
+template <class StorerT>
+void AutosaveManager::AutosaveSettings::store(StorerT &storer) const {
+  CHECK(are_inited_);
+  bool has_exceptions = !exceptions_.empty();
+  BEGIN_STORE_FLAGS();
+  STORE_FLAG(has_exceptions);
+  END_STORE_FLAGS();
+  td::store(user_settings_, storer);
+  td::store(chat_settings_, storer);
+  td::store(broadcast_settings_, storer);
+  if (has_exceptions) {
+    td::store(narrow_cast<uint32>(exceptions_.size()), storer);
+    for (auto &exception : exceptions_) {
+      td::store(exception.first, storer);
+      td::store(exception.second, storer);
+    }
+  }
+}
+
+template <class ParserT>
+void AutosaveManager::AutosaveSettings::parse(ParserT &parser) {
+  are_inited_ = true;
+  bool has_exceptions;
+  BEGIN_PARSE_FLAGS();
+  PARSE_FLAG(has_exceptions);
+  END_PARSE_FLAGS();
+  td::parse(user_settings_, parser);
+  td::parse(chat_settings_, parser);
+  td::parse(broadcast_settings_, parser);
+  if (has_exceptions) {
+    uint32 size;
+    td::parse(size, parser);
+    for (size_t i = 0; i < size; i++) {
+      DialogId dialog_id;
+      DialogAutosaveSettings settings;
+      td::parse(dialog_id, parser);
+      td::parse(settings, parser);
+      CHECK(dialog_id.is_valid());
+      exceptions_.emplace(dialog_id, std::move(settings));
+    }
+  }
+}
+
 void AutosaveManager::get_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise) {
   if (settings_.are_inited_) {
     return promise.set_value(settings_.get_autosave_settings_object());
   }
 
-  reload_autosave_settings(std::move(promise));
+  load_autosave_settings(std::move(promise));
 }
 
-void AutosaveManager::reload_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise) {
+string AutosaveManager::get_autosave_settings_database_key() {
+  return "autosave_settings";
+}
+
+void AutosaveManager::load_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise) {
   load_settings_queries_.push_back(std::move(promise));
   if (load_settings_queries_.size() != 1) {
     return;
   }
 
+  if (G()->parameters().use_message_db) {
+    G()->td_db()->get_sqlite_pmc()->get(
+        get_autosave_settings_database_key(),
+        PromiseCreator::lambda(
+            [actor_id = actor_id(this)](string value) mutable {
+              send_closure(actor_id, &AutosaveManager::on_load_autosave_settings_from_database, std::move(value));
+            }));
+    return;
+  }
+
+  reload_autosave_settings();
+}
+
+void AutosaveManager::on_load_autosave_settings_from_database(string value) {
+  if (settings_.are_inited_) {
+    CHECK(load_settings_queries_.empty());
+    return;
+  }
+  if (G()->close_flag()) {
+    return fail_promises(load_settings_queries_, Global::request_aborted_error());
+  }
+  if (value.empty()) {
+    LOG(INFO) << "Autosave settings aren't found in database";
+    return reload_autosave_settings();
+  }
+
+  LOG(INFO) << "Successfully loaded autosave settings from database";
+
+  auto status = log_event_parse(settings_, value);
+  if (status.is_error()) {
+    LOG(ERROR) << "Can't load autosave settings: " << status;
+    settings_ = {};
+    return reload_autosave_settings();
+  }
+
+  Dependencies dependencies;
+  for (auto &exception : settings_.exceptions_) {
+    dependencies.add_dialog_and_dependencies(exception.first);
+  }
+  if (!dependencies.resolve_force(td_, "on_load_autosave_settings_from_database")) {
+    G()->td_db()->get_binlog_pmc()->erase(get_autosave_settings_database_key());
+    settings_ = {};
+    return reload_autosave_settings();
+  }
+
+  settings_.are_inited_ = true;
+  send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopePrivateChats>(),
+                                settings_.user_settings_);
+  send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopeGroupChats>(),
+                                settings_.chat_settings_);
+  send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopeChannelChats>(),
+                                settings_.broadcast_settings_);
+  for (auto &exception : settings_.exceptions_) {
+    send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopeChat>(exception.first.get()),
+                                  exception.second);
+  }
+
+  auto promises = std::move(load_settings_queries_);
+  for (auto &promise : promises) {
+    promise.set_value(settings_.get_autosave_settings_object());
+  }
+}
+
+void AutosaveManager::reload_autosave_settings() {
+  if (G()->close_flag()) {
+    return fail_promises(load_settings_queries_, Global::request_aborted_error());
+  }
+  if (settings_.are_being_reloaded_) {
+    settings_.need_reload_ = true;
+    return;
+  }
+  settings_.are_being_reloaded_ = true;
+
   auto query_promise = PromiseCreator::lambda(
       [actor_id = actor_id(this)](Result<telegram_api::object_ptr<telegram_api::account_autoSaveSettings>> r_settings) {
         send_closure(actor_id, &AutosaveManager::on_get_autosave_settings, std::move(r_settings));
@@ -214,6 +357,14 @@ void AutosaveManager::reload_autosave_settings(Promise<td_api::object_ptr<td_api
 
 void AutosaveManager::on_get_autosave_settings(
     Result<telegram_api::object_ptr<telegram_api::account_autoSaveSettings>> r_settings) {
+  CHECK(settings_.are_being_reloaded_);
+  settings_.are_being_reloaded_ = false;
+  SCOPE_EXIT {
+    if (settings_.need_reload_) {
+      settings_.need_reload_ = false;
+      reload_autosave_settings();
+    }
+  };
   if (G()->close_flag() && r_settings.is_ok()) {
     r_settings = Global::request_aborted_error();
   }
@@ -270,11 +421,19 @@ void AutosaveManager::on_get_autosave_settings(
                                   DialogAutosaveSettings());
   }
 
+  save_autosave_settings();
+
   auto promises = std::move(load_settings_queries_);
   for (auto &promise : promises) {
-    if (promise) {
     promise.set_value(settings_.get_autosave_settings_object());
   }
+}
+
+void AutosaveManager::save_autosave_settings() {
+  if (G()->parameters().use_message_db) {
+    LOG(INFO) << "Save autosave settings to database";
+    G()->td_db()->get_sqlite_pmc()->set(get_autosave_settings_database_key(),
+                                        log_event_store(settings_).as_slice().str(), Auto());
   }
 }
 
@@ -284,6 +443,9 @@ void AutosaveManager::set_autosave_settings(td_api::object_ptr<td_api::AutosaveS
   if (scope == nullptr) {
     return promise.set_error(Status::Error(400, "Scope must be non-empty"));
   }
+  if (!settings_.are_inited_) {
+    return promise.set_error(Status::Error(400, "Autosave settings must be loaded first"));
+  }
   auto new_settings = DialogAutosaveSettings(settings.get());
   DialogAutosaveSettings *old_settings = nullptr;
   bool users = false;
@@ -320,7 +482,6 @@ void AutosaveManager::set_autosave_settings(td_api::object_ptr<td_api::AutosaveS
   if (*old_settings == new_settings) {
     return promise.set_value(Unit());
   }
-  if (settings_.are_inited_) {
   if (new_settings.are_inited_) {
     *old_settings = std::move(new_settings);
     send_update_autosave_settings(std::move(scope), *old_settings);
@@ -329,17 +490,23 @@ void AutosaveManager::set_autosave_settings(td_api::object_ptr<td_api::AutosaveS
     settings_.exceptions_.erase(dialog_id);
     send_update_autosave_settings(std::move(scope), DialogAutosaveSettings());
   }
-  }
+
+  save_autosave_settings();
+
   td_->create_handler<SaveAutoSaveSettingsQuery>(std::move(promise))
       ->send(users, chats, broadcasts, dialog_id, new_settings.get_input_auto_save_settings());
 }
 
 void AutosaveManager::clear_autosave_settings_excpetions(Promise<Unit> &&promise) {
+  if (!settings_.are_inited_) {
+    return promise.set_error(Status::Error(400, "Autosave settings must be loaded first"));
+  }
   for (const auto &exception : settings_.exceptions_) {
     send_update_autosave_settings(td_api::make_object<td_api::autosaveSettingsScopeChat>(exception.first.get()),
                                   DialogAutosaveSettings());
   }
   settings_.exceptions_.clear();
+  save_autosave_settings();
   td_->create_handler<DeleteAutoSaveExceptionsQuery>(std::move(promise))->send();
 }
 

td/telegram/AutosaveManager.h

@@ -25,7 +25,7 @@ class AutosaveManager final : public Actor {
  public:
   AutosaveManager(Td *td, ActorShared<> parent);
 
-  void reload_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise);
+  void reload_autosave_settings();
 
   void get_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise);
 
@@ -63,22 +63,44 @@ class AutosaveManager final : public Actor {
     bool operator==(const DialogAutosaveSettings &other) const;
 
     bool operator!=(const DialogAutosaveSettings &other) const;
+
+    template <class StorerT>
+    void store(StorerT &storer) const;
+
+    template <class ParserT>
+    void parse(ParserT &parser);
   };
 
   struct AutosaveSettings {
     bool are_inited_ = false;
+    bool are_being_reloaded_ = false;
+    bool need_reload_ = false;
     DialogAutosaveSettings user_settings_;
     DialogAutosaveSettings chat_settings_;
     DialogAutosaveSettings broadcast_settings_;
     FlatHashMap<DialogId, DialogAutosaveSettings, DialogIdHash> exceptions_;
 
     td_api::object_ptr<td_api::autosaveSettings> get_autosave_settings_object() const;
+
+    template <class StorerT>
+    void store(StorerT &storer) const;
+
+    template <class ParserT>
+    void parse(ParserT &parser);
   };
 
   void tear_down() final;
 
+  string get_autosave_settings_database_key();
+
+  void load_autosave_settings(Promise<td_api::object_ptr<td_api::autosaveSettings>> &&promise);
+
+  void on_load_autosave_settings_from_database(string value);
+
   void on_get_autosave_settings(Result<telegram_api::object_ptr<telegram_api::account_autoSaveSettings>> r_settings);
 
+  void save_autosave_settings();
+
   static td_api::object_ptr<td_api::updateAutosaveSettings> get_update_autosave_settings(
       td_api::object_ptr<td_api::AutosaveSettingsScope> &&scope, const DialogAutosaveSettings &settings);
 

td/telegram/UpdatesManager.cpp

@@ -3917,7 +3917,7 @@ void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateTranscribedAudi
 }
 
 void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateAutoSaveSettings> update, Promise<Unit> &&promise) {
-  td_->autosave_manager_->reload_autosave_settings(Auto());
+  td_->autosave_manager_->reload_autosave_settings();
   promise.set_value(Unit());
 }