Add class Story.

td/telegram/StoryManager.cpp

@@ -6,6 +6,13 @@
 //
 #include "td/telegram/StoryManager.h"
 
+#include "td/telegram/AuthManager.h"
+#include "td/telegram/ContactsManager.h"
+#include "td/telegram/MessageContent.h"
+#include "td/telegram/MessageContentType.h"
+#include "td/telegram/MessageEntity.h"
+#include "td/telegram/Td.h"
+
 namespace td {
 
 StoryManager::StoryManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
@@ -15,4 +22,95 @@ void StoryManager::tear_down() {
   parent_.reset();
 }
 
+bool StoryManager::is_local_story_id(StoryId story_id) {
+  return story_id.get() < 0;
+}
+
+const StoryManager::Story *StoryManager::get_story(StoryId story_id) const {
+  return stories_.get_pointer(story_id);
+}
+
+StoryManager::Story *StoryManager::get_story_editable(StoryId story_id) {
+  return stories_.get_pointer(story_id);
+}
+
+StoryId StoryManager::on_get_story(DialogId owner_dialog_id,
+                                   telegram_api::object_ptr<telegram_api::storyItem> &&story_item) {
+  CHECK(story_item != nullptr);
+  StoryId story_id(story_item->id_);
+  if (!story_id.is_valid() || is_local_story_id(story_id)) {
+    return StoryId();
+  }
+
+  auto story = get_story_editable(story_id);
+  bool is_changed = false;
+  bool need_save_to_database = false;
+  if (story == nullptr) {
+    auto s = make_unique<Story>();
+    story = s.get();
+    stories_.set(story_id, std::move(s));
+  }
+  CHECK(story != nullptr);
+
+  bool is_bot = td_->auth_manager_->is_bot();
+  auto message_text =
+      get_message_text(td_->contacts_manager_.get(), std::move(story_item->caption_), std::move(story_item->entities_),
+                       true, is_bot, story_item->date_, false, "on_get_story");
+  int32 ttl = 0;
+  auto content = get_message_content(td_, std::move(message_text), std::move(story_item->media_), owner_dialog_id,
+                                     false, UserId(), &ttl, nullptr, "on_get_story");
+  auto content_type = content->get_type();
+  if (content_type != MessageContentType::Photo && content_type != MessageContentType::Video &&
+      content_type != MessageContentType::Unsupported) {
+    LOG(ERROR) << "Receive " << story_id << " of type " << content_type;
+    return StoryId();
+  }
+
+  auto privacy_rules = UserPrivacySettingRules::get_user_privacy_setting_rules(td_, std::move(story_item->privacy_));
+
+  vector<UserId> recent_viewer_user_ids;
+  int32 view_count = 0;
+  if (story_item->views_ != nullptr) {
+    for (auto &viewer_id : story_item->views_->recent_viewers_) {
+      UserId user_id(viewer_id);
+      if (user_id.is_valid() && td_->contacts_manager_->have_min_user(user_id)) {
+        recent_viewer_user_ids.push_back(user_id);
+      } else {
+        LOG(ERROR) << "Receive " << user_id << " as recent viewer in " << story_id;
+      }
+    }
+    view_count = story_item->views_->views_count_;
+  }
+
+  if (story->is_pinned_ != story_item->pinned_ || story->is_public_ != story_item->public_ ||
+      story->is_for_close_friends_ != story_item->close_friends_ || story->date_ != story_item->date_ ||
+      story->expire_date_ != story_item->expire_date_ || !(story->privacy_rules_ == privacy_rules) ||
+      story->recent_viewer_user_ids_ != recent_viewer_user_ids || story->view_count_ != view_count) {
+    story->is_pinned_ = story_item->pinned_;
+    story->is_public_ = story_item->public_;
+    story->is_for_close_friends_ = story_item->close_friends_;
+    story->date_ = story_item->date_;
+    story->expire_date_ = story_item->expire_date_;
+    story->privacy_rules_ = std::move(privacy_rules);
+    story->recent_viewer_user_ids_ = std::move(recent_viewer_user_ids);
+    story->view_count_ = view_count;
+    is_changed = true;
+  }
+  if (story->content_ == nullptr || story->content_->get_type() != content_type) {
+    story->content_ = std::move(content);
+    is_changed = true;
+  } else {
+    merge_message_contents(td_, story->content_.get(), content.get(), false, owner_dialog_id, false,
+                           need_save_to_database, is_changed);
+  }
+
+  if (is_changed || need_save_to_database) {
+    // save_story(story, story_id);
+  }
+  if (is_changed) {
+    // send_closure(G()->td(), &Td::send_update, td_api::make_object<td_api::updateStory>(get_story_object(story_id, story)));
+  }
+  return story_id;
+}
+
 }  // namespace td

td/telegram/StoryManager.h

@@ -6,21 +6,50 @@
 //
 #pragma once
 
+#include "td/telegram/DialogId.h"
+#include "td/telegram/StoryId.h"
+#include "td/telegram/UserId.h"
+#include "td/telegram/UserPrivacySettingRule.h"
+
 #include "td/actor/actor.h"
 
 #include "td/utils/common.h"
+#include "td/utils/WaitFreeHashMap.h"
 
 namespace td {
 
+class MessageContent;
 class Td;
 
 class StoryManager final : public Actor {
  public:
   StoryManager(Td *td, ActorShared<> parent);
 
+  StoryId on_get_story(DialogId owner_dialog_id, telegram_api::object_ptr<telegram_api::storyItem> &&story_item);
+
  private:
+  struct Story {
+    int32 date_ = 0;
+    int32 expire_date_ = 0;
+    bool is_pinned_ = false;
+    bool is_public_ = false;
+    bool is_for_close_friends_ = false;
+    vector<UserId> recent_viewer_user_ids_;
+    int32 view_count_ = 0;
+    UserPrivacySettingRules privacy_rules_;
+    unique_ptr<MessageContent> content_;
+  };
+
   void tear_down() final;
 
+  const Story *get_story(StoryId story_id) const;
+
+  Story *get_story_editable(StoryId story_id);
+
+  static bool is_local_story_id(StoryId story_id);
+
+  WaitFreeHashMap<StoryId, unique_ptr<Story>, StoryIdHash> stories_;
+
   Td *td_;
   ActorShared<> parent_;
 };