From a86b7538a77125f88fb668248a5770b5cc495c01 Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Mon, 19 Jun 2023 20:45:58 +0300
Subject: [PATCH] Delete story from active stories or delete completely when it
 expires.

---
 td/telegram/StoryManager.cpp | 44 ++++++++++++++++++++++++++++++++++++
 td/telegram/StoryManager.h   |  7 ++++++
 2 files changed, 51 insertions(+)

diff --git a/td/telegram/StoryManager.cpp b/td/telegram/StoryManager.cpp
index 70bbb4de3..3503c29db 100644
--- a/td/telegram/StoryManager.cpp
+++ b/td/telegram/StoryManager.cpp
@@ -607,6 +607,9 @@ StoryManager::PendingStory::PendingStory(DialogId dialog_id, StoryId story_id, u
 
 StoryManager::StoryManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
   upload_media_callback_ = std::make_shared<UploadMediaCallback>();
+
+  story_expire_timeout_.set_callback(on_story_expire_timeout_callback);
+  story_expire_timeout_.set_callback_data(static_cast<void *>(this));
 }
 
 StoryManager::~StoryManager() {
@@ -623,6 +626,43 @@ void StoryManager::tear_down() {
   parent_.reset();
 }
 
+void StoryManager::on_story_expire_timeout_callback(void *story_manager_ptr, int64 story_global_id) {
+  if (G()->close_flag()) {
+    return;
+  }
+
+  auto story_manager = static_cast<StoryManager *>(story_manager_ptr);
+  send_closure_later(story_manager->actor_id(story_manager), &StoryManager::on_story_expire_timeout, story_global_id);
+}
+
+void StoryManager::on_story_expire_timeout(int64 story_global_id) {
+  if (G()->close_flag()) {
+    return;
+  }
+
+  auto story_full_id = stories_by_global_id_.get(story_global_id);
+  auto story = get_story(story_full_id);
+  if (story == nullptr) {
+    return;
+  }
+  if (is_active_story(story)) {
+    LOG(ERROR) << "Receive timeout for non-expired " << story_full_id;
+    return on_story_changed(story_full_id, story, false, false);
+  }
+  auto owner_dialog_id = story_full_id.get_dialog_id();
+  auto story_id = story_full_id.get_story_id();
+  if (!is_story_owned(owner_dialog_id) && story->content_ != nullptr && !story->is_pinned_) {
+    // non-owned expired non-pinned stories are fully deleted
+    on_delete_story(owner_dialog_id, story_id);
+  }
+
+  auto active_stories = get_active_stories(owner_dialog_id);
+  if (active_stories != nullptr && contains(active_stories->story_ids_, story_id)) {
+    auto story_ids = active_stories->story_ids_;
+    on_update_active_stories(owner_dialog_id, active_stories->max_read_story_id_, std::move(story_ids));
+  }
+}
+
 bool StoryManager::is_story_owned(DialogId owner_dialog_id) const {
   return owner_dialog_id == DialogId(td_->contacts_manager_->get_my_id());
 }
@@ -1452,6 +1492,10 @@ void StoryManager::on_delete_story(DialogId owner_dialog_id, StoryId story_id) {
 
 void StoryManager::on_story_changed(StoryFullId story_full_id, const Story *story, bool is_changed,
                                     bool need_save_to_database) {
+  if (is_active_story(story)) {
+    CHECK(story->global_id_ > 0);
+    story_expire_timeout_.set_timeout_in(story->global_id_, story->expire_date_ - G()->unix_time());
+  }
   if (story->content_ == nullptr) {
     return;
   }
diff --git a/td/telegram/StoryManager.h b/td/telegram/StoryManager.h
index cfe4d0ab0..bdbd2ae98 100644
--- a/td/telegram/StoryManager.h
+++ b/td/telegram/StoryManager.h
@@ -19,6 +19,7 @@
 #include "td/telegram/UserPrivacySettingRule.h"
 
 #include "td/actor/actor.h"
+#include "td/actor/MultiTimeout.h"
 #include "td/actor/Timeout.h"
 
 #include "td/utils/common.h"
@@ -180,6 +181,10 @@ class StoryManager final : public Actor {
 
   void tear_down() final;
 
+  static void on_story_expire_timeout_callback(void *story_manager_ptr, int64 story_global_id);
+
+  void on_story_expire_timeout(int64 story_global_id);
+
   bool is_story_owned(DialogId owner_dialog_id) const;
 
   bool is_active_story(StoryFullId story_full_id) const;
@@ -311,6 +316,8 @@ class StoryManager final : public Actor {
 
   Timeout interaction_info_update_timeout_;
 
+  MultiTimeout story_expire_timeout_{"StoryExpireTimeout"};
+
   Td *td_;
   ActorShared<> parent_;
 };