From d17a1734a4ae4e2a710bfa2b5245179f820bff0b Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Fri, 11 Oct 2019 02:32:25 +0300
Subject: [PATCH] Add repair_dialog_action_bar.

GitOrigin-RevId: 4157f18ad20ff09807b59433d86bce04b676fdf2
---
 td/telegram/MessagesManager.cpp | 38 +++++++++++++++++++++++++++------
 td/telegram/MessagesManager.h   |  2 ++
 2 files changed, 34 insertions(+), 6 deletions(-)

diff --git a/td/telegram/MessagesManager.cpp b/td/telegram/MessagesManager.cpp
index b5459fdd5..b44c8b5a2 100644
--- a/td/telegram/MessagesManager.cpp
+++ b/td/telegram/MessagesManager.cpp
@@ -6440,6 +6440,25 @@ bool MessagesManager::update_dialog_silent_send_message(Dialog *d, bool silent_s
   return true;
 }
 
+void MessagesManager::repair_dialog_action_bar(DialogId dialog_id) {
+  switch (dialog_id.get_type()) {
+    case DialogType::User:
+      td_->contacts_manager_->get_user_full(dialog_id.get_user_id(), Auto());
+      return;
+    case DialogType::Chat:
+    case DialogType::Channel:
+      if (!have_input_peer(dialog_id, AccessRights::Read)) {
+        return;
+      }
+
+      return td_->create_handler<GetPeerSettingsQuery>(Promise<Unit>())->send(dialog_id);
+    case DialogType::SecretChat:
+    case DialogType::None:
+    default:
+      UNREACHABLE();
+  }
+}
+
 bool MessagesManager::get_dialog_report_spam_state(DialogId dialog_id, Promise<Unit> &&promise) {
   Dialog *d = get_dialog_force(dialog_id);
   if (d == nullptr) {
@@ -6447,11 +6466,6 @@ bool MessagesManager::get_dialog_report_spam_state(DialogId dialog_id, Promise<U
     return false;
   }
 
-  if (!have_input_peer(dialog_id, AccessRights::Read)) {
-    promise.set_error(Status::Error(3, "Can't access the chat"));
-    return false;
-  }
-
   if (d->know_can_report_spam) {
     promise.set_value(Unit());
     return d->can_report_spam;
@@ -22181,7 +22195,9 @@ void MessagesManager::on_dialog_is_blocked_updated(DialogId dialog_id, bool is_b
         }
       } else {
         d->know_action_bar = false;
-        // TODO repair_dialog_action_bar(d);
+        if (have_input_peer(dialog_id, AccessRights::Read)) {
+          repair_dialog_action_bar(dialog_id);
+        }
         // there is no need to change action bar
         on_dialog_updated(dialog_id, "on_dialog_is_blocked_updated");
       }
@@ -25784,6 +25800,16 @@ void MessagesManager::fix_new_dialog(Dialog *d, unique_ptr<Message> &&last_datab
     // asynchronously get dialog folder id from the server
     get_dialog_info_full(dialog_id, Auto());
   }
+  if (!d->know_action_bar && !td_->auth_manager_->is_bot() && dialog_id != get_my_dialog_id() &&
+      have_input_peer(dialog_id, AccessRights::Read)) {
+    // asynchronously get action bar from the server
+    if (dialog_id.get_type() == DialogType::SecretChat) {
+      auto user_id = td_->contacts_manager_->get_secret_chat_user_id(dialog_id.get_secret_chat_id());
+      force_create_dialog(DialogId(user_id), "add chat with user to load/store action_bar");
+    } else {
+      repair_dialog_action_bar(dialog_id);
+    }
+  }
 
   if (d->notification_settings.is_synchronized && !d->notification_settings.is_use_default_fixed &&
       have_input_peer(dialog_id, AccessRights::Read) && !td_->auth_manager_->is_bot()) {
diff --git a/td/telegram/MessagesManager.h b/td/telegram/MessagesManager.h
index 5c05528f1..2f092f9ed 100644
--- a/td/telegram/MessagesManager.h
+++ b/td/telegram/MessagesManager.h
@@ -1629,6 +1629,8 @@ class MessagesManager : public Actor {
   void load_messages(DialogId dialog_id, MessageId from_message_id, int32 offset, int32 limit, int left_tries,
                      bool only_local, Promise<Unit> &&promise);
 
+  void repair_dialog_action_bar(DialogId dialog_id);
+
   static int32 get_random_y(MessageId message_id);
 
   bool is_allowed_useless_update(const tl_object_ptr<telegram_api::Update> &update) const;