tdlight/td/telegram/StoryManager.h
2023-08-01 20:37:37 +03:00

582 lines
21 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once
#include "td/telegram/DialogDate.h"
#include "td/telegram/DialogId.h"
#include "td/telegram/files/FileId.h"
#include "td/telegram/files/FileSourceId.h"
#include "td/telegram/FullMessageId.h"
#include "td/telegram/MediaArea.h"
#include "td/telegram/MessageEntity.h"
#include "td/telegram/MessageViewer.h"
#include "td/telegram/StoryDb.h"
#include "td/telegram/StoryFullId.h"
#include "td/telegram/StoryId.h"
#include "td/telegram/StoryInteractionInfo.h"
#include "td/telegram/StoryListId.h"
#include "td/telegram/td_api.h"
#include "td/telegram/telegram_api.h"
#include "td/telegram/UserId.h"
#include "td/telegram/UserPrivacySettingRule.h"
#include "td/actor/actor.h"
#include "td/actor/MultiTimeout.h"
#include "td/actor/Timeout.h"
#include "td/utils/buffer.h"
#include "td/utils/common.h"
#include "td/utils/FlatHashMap.h"
#include "td/utils/FlatHashSet.h"
#include "td/utils/Promise.h"
#include "td/utils/Status.h"
#include "td/utils/WaitFreeHashMap.h"
#include "td/utils/WaitFreeHashSet.h"
#include <memory>
#include <set>
#include <utility>
namespace td {
struct BinlogEvent;
class Dependencies;
class ReportReason;
class StoryContent;
struct StoryDbStory;
class Td;
class StoryManager final : public Actor {
struct Story {
int32 date_ = 0;
int32 expire_date_ = 0;
int32 receive_date_ = 0;
bool is_edited_ = false;
bool is_pinned_ = false;
bool is_public_ = false;
bool is_for_close_friends_ = false;
bool is_for_contacts_ = false;
bool is_for_selected_contacts_ = false;
bool noforwards_ = false;
mutable bool is_update_sent_ = false; // whether the story is known to the app
StoryInteractionInfo interaction_info_;
UserPrivacySettingRules privacy_rules_;
unique_ptr<StoryContent> content_;
vector<MediaArea> areas_;
FormattedText caption_;
int64 global_id_ = 0;
template <class StorerT>
void store(StorerT &storer) const;
template <class ParserT>
void parse(ParserT &parser);
};
struct StoryInfo {
StoryId story_id_;
int32 date_ = 0;
int32 expire_date_ = 0;
bool is_for_close_friends_ = false;
template <class StorerT>
void store(StorerT &storer) const;
template <class ParserT>
void parse(ParserT &parser);
};
struct BeingEditedStory {
unique_ptr<StoryContent> content_;
vector<MediaArea> areas_;
FormattedText caption_;
bool edit_media_areas_ = false;
bool edit_caption_ = false;
vector<Promise<Unit>> promises_;
int64 log_event_id_ = 0;
};
struct PendingStory {
DialogId dialog_id_;
StoryId story_id_;
uint64 log_event_id_ = 0;
uint32 send_story_num_ = 0;
int64 random_id_ = 0;
bool was_reuploaded_ = false;
unique_ptr<Story> story_;
PendingStory() = default;
PendingStory(DialogId dialog_id, StoryId story_id, uint32 send_story_num, int64 random_id,
unique_ptr<Story> &&story);
template <class StorerT>
void store(StorerT &storer) const;
template <class ParserT>
void parse(ParserT &parser);
};
struct ReadyToSendStory {
FileId file_id_;
unique_ptr<PendingStory> pending_story_;
telegram_api::object_ptr<telegram_api::InputFile> input_file_;
ReadyToSendStory(FileId file_id, unique_ptr<PendingStory> &&pending_story,
telegram_api::object_ptr<telegram_api::InputFile> &&input_file);
};
struct PendingStoryViews {
FlatHashSet<StoryId, StoryIdHash> story_ids_;
bool has_query_ = false;
};
struct ActiveStories {
StoryId max_read_story_id_;
vector<StoryId> story_ids_;
StoryListId story_list_id_;
int64 private_order_ = 0;
int64 public_order_ = 0;
};
struct SavedActiveStories {
StoryId max_read_story_id_;
vector<StoryInfo> story_infos_;
template <class StorerT>
void store(StorerT &storer) const;
template <class ParserT>
void parse(ParserT &parser);
};
struct CachedStoryViewers {
int32 total_count_ = -1;
MessageViewers viewers_;
};
struct StoryList {
int32 server_total_count_ = -1;
int32 sent_total_count_ = -1;
string state_;
bool is_reloaded_server_total_count_ = false;
bool server_has_more_ = true;
bool database_has_more_ = false;
vector<Promise<Unit>> load_list_from_server_queries_;
vector<Promise<Unit>> load_list_from_database_queries_;
std::set<DialogDate> ordered_stories_; // all known active stories from the story list
DialogDate last_loaded_database_dialog_date_ = MIN_DIALOG_DATE; // in memory
DialogDate list_last_story_date_ = MIN_DIALOG_DATE; // in memory
};
struct SavedStoryList {
string state_;
int32 total_count_ = -1;
bool has_more_ = true;
template <class StorerT>
void store(StorerT &storer) const;
template <class ParserT>
void parse(ParserT &parser);
};
public:
StoryManager(Td *td, ActorShared<> parent);
StoryManager(const StoryManager &) = delete;
StoryManager &operator=(const StoryManager &) = delete;
StoryManager(StoryManager &&) = delete;
StoryManager &operator=(StoryManager &&) = delete;
~StoryManager() final;
void get_story(DialogId owner_dialog_id, StoryId story_id, bool only_local,
Promise<td_api::object_ptr<td_api::story>> &&promise);
void send_story(td_api::object_ptr<td_api::InputStoryContent> &&input_story_content,
td_api::object_ptr<td_api::inputStoryAreas> &&input_areas,
td_api::object_ptr<td_api::formattedText> &&input_caption,
td_api::object_ptr<td_api::StoryPrivacySettings> &&settings, int32 active_period, bool is_pinned,
bool protect_content, Promise<td_api::object_ptr<td_api::story>> &&promise);
void on_send_story_file_parts_missing(unique_ptr<PendingStory> &&pending_story, vector<int> &&bad_parts);
void edit_story(StoryId story_id, td_api::object_ptr<td_api::InputStoryContent> &&input_story_content,
td_api::object_ptr<td_api::inputStoryAreas> &&input_areas,
td_api::object_ptr<td_api::formattedText> &&input_caption, Promise<Unit> &&promise);
void set_story_privacy_settings(StoryId story_id, td_api::object_ptr<td_api::StoryPrivacySettings> &&settings,
Promise<Unit> &&promise);
void toggle_story_is_pinned(StoryId story_id, bool is_pinned, Promise<Unit> &&promise);
void delete_story(StoryId story_id, Promise<Unit> &&promise);
void load_active_stories(StoryListId story_list_id, Promise<Unit> &&promise);
void reload_active_stories();
void reload_all_read_stories();
void toggle_dialog_stories_hidden(DialogId dialog_id, StoryListId story_list_id, Promise<Unit> &&promise);
void get_dialog_pinned_stories(DialogId owner_dialog_id, StoryId from_story_id, int32 limit,
Promise<td_api::object_ptr<td_api::stories>> &&promise);
void get_story_archive(StoryId from_story_id, int32 limit, Promise<td_api::object_ptr<td_api::stories>> &&promise);
void get_dialog_expiring_stories(DialogId owner_dialog_id,
Promise<td_api::object_ptr<td_api::chatActiveStories>> &&promise);
void open_story(DialogId owner_dialog_id, StoryId story_id, Promise<Unit> &&promise);
void close_story(DialogId owner_dialog_id, StoryId story_id, Promise<Unit> &&promise);
void view_story_message(StoryFullId story_full_id);
void on_story_replied(StoryFullId story_full_id, UserId replier_user_id);
void get_story_viewers(StoryId story_id, const td_api::messageViewer *offset, int32 limit,
Promise<td_api::object_ptr<td_api::messageViewers>> &&promise);
void report_story(StoryFullId story_full_id, ReportReason &&reason, Promise<Unit> &&promise);
void remove_story_notifications_by_story_ids(DialogId dialog_id, const vector<StoryId> &story_ids);
StoryId on_get_story(DialogId owner_dialog_id, telegram_api::object_ptr<telegram_api::StoryItem> &&story_item_ptr);
std::pair<int32, vector<StoryId>> on_get_stories(DialogId owner_dialog_id, vector<StoryId> &&expected_story_ids,
telegram_api::object_ptr<telegram_api::stories_stories> &&stories);
DialogId on_get_user_stories(DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::userStories> &&user_stories,
Promise<Unit> &&promise);
bool on_update_read_stories(DialogId owner_dialog_id, StoryId max_read_story_id);
void on_dialog_active_stories_order_updated(DialogId owner_dialog_id, const char *source);
Status can_get_story_viewers(StoryFullId story_full_id, const Story *story) const;
void on_get_story_views(const vector<StoryId> &story_ids,
telegram_api::object_ptr<telegram_api::stories_storyViews> &&story_views);
bool have_story(StoryFullId story_full_id) const;
bool have_story_force(StoryFullId story_full_id);
bool is_inaccessible_story(StoryFullId story_full_id) const;
int32 get_story_duration(StoryFullId story_full_id) const;
void register_story(StoryFullId story_full_id, FullMessageId full_message_id, const char *source);
void unregister_story(StoryFullId story_full_id, FullMessageId full_message_id, const char *source);
td_api::object_ptr<td_api::story> get_story_object(StoryFullId story_full_id) const;
td_api::object_ptr<td_api::stories> get_stories_object(int32 total_count,
const vector<StoryFullId> &story_full_ids) const;
FileSourceId get_story_file_source_id(StoryFullId story_full_id);
telegram_api::object_ptr<telegram_api::InputMedia> get_input_media(StoryFullId story_full_id) const;
void reload_story(StoryFullId story_full_id, Promise<Unit> &&promise, const char *source);
void try_synchronize_archive_all_stories();
void get_current_state(vector<td_api::object_ptr<td_api::Update>> &updates) const;
void on_binlog_events(vector<BinlogEvent> &&events);
private:
class UploadMediaCallback;
class SendStoryQuery;
class EditStoryQuery;
class DeleteStoryOnServerLogEvent;
class ReadStoriesOnServerLogEvent;
class LoadDialogExpiringStoriesLogEvent;
class SendStoryLogEvent;
class EditStoryLogEvent;
static constexpr int32 OPENED_STORY_POLL_PERIOD = 60;
static constexpr int32 VIEWED_STORY_POLL_PERIOD = 300;
static constexpr int32 DEFAULT_LOADED_EXPIRED_STORIES = 50;
void start_up() final;
void timeout_expired() final;
void hangup() final;
void tear_down() final;
static void on_story_reload_timeout_callback(void *story_manager_ptr, int64 story_global_id);
void on_story_reload_timeout(int64 story_global_id);
static void on_story_expire_timeout_callback(void *story_manager_ptr, int64 story_global_id);
void on_story_expire_timeout(int64 story_global_id);
static void on_story_can_get_viewers_timeout_callback(void *story_manager_ptr, int64 story_global_id);
void on_story_can_get_viewers_timeout(int64 story_global_id);
bool is_story_owned(DialogId owner_dialog_id) const;
int32 get_story_viewers_expire_date(const Story *story) const;
static bool is_active_story(const Story *story);
DialogId get_changelog_story_dialog_id() const;
bool is_subscribed_to_dialog_stories(DialogId owner_dialog_id) const;
StoryListId get_dialog_story_list_id(DialogId owner_dialog_id) const;
void add_story_dependencies(Dependencies &dependencies, const Story *story);
void add_pending_story_dependencies(Dependencies &dependencies, const PendingStory *pending_story);
const Story *get_story(StoryFullId story_full_id) const;
Story *get_story_editable(StoryFullId story_full_id);
Story *get_story_force(StoryFullId story_full_id, const char *source);
unique_ptr<Story> parse_story(StoryFullId story_full_id, const BufferSlice &value);
Story *on_get_story_from_database(StoryFullId story_full_id, const BufferSlice &value, const char *source);
const ActiveStories *get_active_stories(DialogId owner_dialog_id) const;
ActiveStories *get_active_stories_editable(DialogId owner_dialog_id);
ActiveStories *get_active_stories_force(DialogId owner_dialog_id, const char *source);
ActiveStories *on_get_active_stories_from_database(StoryListId story_list_id, DialogId owner_dialog_id,
const BufferSlice &value, const char *source);
void on_story_changed(StoryFullId story_full_id, const Story *story, bool is_changed, bool need_save_to_database,
bool from_database = false);
void register_story_global_id(StoryFullId story_full_id, Story *story);
void unregister_story_global_id(const Story *story);
StoryId on_get_story_info(DialogId owner_dialog_id, StoryInfo &&story_info);
StoryInfo get_story_info(StoryFullId story_full_id) const;
td_api::object_ptr<td_api::storyInfo> get_story_info_object(StoryFullId story_full_id) const;
td_api::object_ptr<td_api::story> get_story_object(StoryFullId story_full_id, const Story *story) const;
td_api::object_ptr<td_api::chatActiveStories> get_chat_active_stories_object(DialogId owner_dialog_id) const;
td_api::object_ptr<td_api::chatActiveStories> get_chat_active_stories_object(
DialogId owner_dialog_id, const ActiveStories *active_stories) const;
StoryId on_get_new_story(DialogId owner_dialog_id, telegram_api::object_ptr<telegram_api::storyItem> &&story_item);
StoryId on_get_skipped_story(DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::storyItemSkipped> &&story_item);
StoryId on_get_deleted_story(DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::storyItemDeleted> &&story_item);
void on_delete_story(StoryFullId story_full_id);
void on_get_dialog_pinned_stories(DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::stories_stories> &&stories,
Promise<td_api::object_ptr<td_api::stories>> &&promise);
void on_get_story_archive(telegram_api::object_ptr<telegram_api::stories_stories> &&stories,
Promise<td_api::object_ptr<td_api::stories>> &&promise);
void on_get_dialog_expiring_stories(DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::stories_userStories> &&stories,
Promise<td_api::object_ptr<td_api::chatActiveStories>> &&promise);
static uint64 save_load_dialog_expiring_stories_log_event(DialogId owner_dialog_id);
void load_dialog_expiring_stories(DialogId owner_dialog_id, uint64 log_event_id, const char *source);
void on_load_dialog_expiring_stories(DialogId owner_dialog_id);
void on_load_active_stories_from_database(StoryListId story_list_id, Result<StoryDbGetActiveStoryListResult> result);
void load_active_stories_from_server(StoryListId story_list_id, StoryList &story_list, bool is_next,
Promise<Unit> &&promise);
void on_load_active_stories_from_server(
StoryListId story_list_id, bool is_next, string old_state,
Result<telegram_api::object_ptr<telegram_api::stories_AllStories>> r_all_stories);
void save_story_list(StoryListId story_list_id, string state, int32 total_count, bool has_more);
StoryList &get_story_list(StoryListId story_list_id);
const StoryList &get_story_list(StoryListId story_list_id) const;
td_api::object_ptr<td_api::updateStoryListChatCount> get_update_story_list_chat_count_object(
StoryListId story_list_id, const StoryList &story_list) const;
void update_story_list_sent_total_count(StoryListId story_list_id);
void update_story_list_sent_total_count(StoryListId story_list_id, StoryList &story_list);
vector<FileId> get_story_file_ids(const Story *story) const;
static uint64 save_delete_story_on_server_log_event(StoryFullId story_full_id);
void delete_story_on_server(StoryFullId story_full_id, uint64 log_event_id, Promise<Unit> &&promise);
void delete_story_from_database(StoryFullId story_full_id);
void delete_story_files(const Story *story) const;
void change_story_files(StoryFullId story_full_id, const Story *story, const vector<FileId> &old_file_ids);
void do_get_story(StoryFullId story_full_id, Result<Unit> &&result,
Promise<td_api::object_ptr<td_api::story>> &&promise);
void on_reload_story(StoryFullId story_full_id, Result<Unit> &&result);
int64 save_send_story_log_event(const PendingStory *pending_story);
void delete_pending_story(FileId file_id, unique_ptr<PendingStory> &&pending_story, Status status);
void do_send_story(unique_ptr<PendingStory> &&pending_story, vector<int> bad_parts);
void on_upload_story(FileId file_id, telegram_api::object_ptr<telegram_api::InputFile> input_file);
void on_upload_story_error(FileId file_id, Status status);
void try_send_story();
void do_edit_story(FileId file_id, unique_ptr<PendingStory> &&pending_story,
telegram_api::object_ptr<telegram_api::InputFile> input_file);
void on_toggle_story_is_pinned(StoryId story_id, bool is_pinned, Promise<Unit> &&promise);
void on_update_active_stories(DialogId owner_dialog_id, StoryId max_read_story_id, vector<StoryId> &&story_ids,
Promise<Unit> &&promise, const char *source, bool from_database = false);
bool update_active_stories_order(DialogId owner_dialog_id, ActiveStories *active_stories,
bool *need_save_to_database);
void delete_active_stories_from_story_list(DialogId owner_dialog_id, const ActiveStories *active_stories);
void send_update_story(StoryFullId story_full_id, const Story *story);
td_api::object_ptr<td_api::updateChatActiveStories> get_update_chat_active_stories(
DialogId owner_dialog_id, const ActiveStories *active_stories) const;
void send_update_chat_active_stories(DialogId owner_dialog_id, const ActiveStories *active_stories) const;
void save_active_stories(DialogId owner_dialog_id, const ActiveStories *active_stories, Promise<Unit> &&promise,
const char *source) const;
void increment_story_views(DialogId owner_dialog_id, PendingStoryViews &story_views);
void on_increment_story_views(DialogId owner_dialog_id);
static uint64 save_read_stories_on_server_log_event(DialogId dialog_id, StoryId max_story_id);
void read_stories_on_server(DialogId owner_dialog_id, StoryId story_id, uint64 log_event_id);
void schedule_interaction_info_update();
static void update_interaction_info_static(void *story_manager);
void update_interaction_info();
void on_synchronized_archive_all_stories(bool set_archive_all_stories, Result<Unit> result);
void on_get_story_viewers(StoryId story_id, MessageViewer offset,
Result<telegram_api::object_ptr<telegram_api::stories_storyViewsList>> r_view_list,
Promise<td_api::object_ptr<td_api::messageViewers>> &&promise);
void load_expired_database_stories();
void on_load_expired_database_stories(vector<StoryDbStory> stories);
std::shared_ptr<UploadMediaCallback> upload_media_callback_;
WaitFreeHashMap<StoryFullId, FileSourceId, StoryFullIdHash> story_full_id_to_file_source_id_;
WaitFreeHashMap<StoryFullId, unique_ptr<Story>, StoryFullIdHash> stories_;
WaitFreeHashMap<int64, StoryFullId> stories_by_global_id_;
WaitFreeHashMap<StoryFullId, double, StoryFullIdHash> inaccessible_story_full_ids_;
WaitFreeHashSet<StoryFullId, StoryFullIdHash> deleted_story_full_ids_;
WaitFreeHashSet<StoryFullId, StoryFullIdHash> failed_to_load_story_full_ids_;
WaitFreeHashMap<StoryFullId, WaitFreeHashSet<FullMessageId, FullMessageIdHash>, StoryFullIdHash> story_messages_;
WaitFreeHashMap<DialogId, unique_ptr<ActiveStories>, DialogIdHash> active_stories_;
WaitFreeHashMap<DialogId, StoryId, DialogIdHash> max_read_story_ids_;
WaitFreeHashSet<DialogId, DialogIdHash> failed_to_load_active_stories_;
FlatHashMap<DialogId, uint64, DialogIdHash> load_expiring_stories_log_event_ids_;
FlatHashMap<StoryFullId, unique_ptr<BeingEditedStory>, StoryFullIdHash> being_edited_stories_;
FlatHashMap<StoryFullId, int64, StoryFullIdHash> edit_generations_;
FlatHashMap<DialogId, PendingStoryViews, DialogIdHash> pending_story_views_;
FlatHashMap<StoryFullId, uint32, StoryFullIdHash> opened_owned_stories_;
FlatHashMap<StoryFullId, uint32, StoryFullIdHash> opened_stories_;
FlatHashMap<StoryFullId, unique_ptr<CachedStoryViewers>, StoryFullIdHash> cached_story_viewers_;
FlatHashMap<StoryFullId, vector<Promise<Unit>>, StoryFullIdHash> reload_story_queries_;
FlatHashMap<FileId, unique_ptr<PendingStory>, FileIdHash> being_uploaded_files_;
std::set<uint32> yet_unsent_stories_;
FlatHashMap<uint32, unique_ptr<ReadyToSendStory>> ready_to_send_stories_;
StoryList story_lists_[2];
uint32 send_story_count_ = 0;
int64 max_story_global_id_ = 0;
bool has_active_synchronize_archive_all_stories_query_ = false;
Timeout interaction_info_update_timeout_;
int32 load_expired_database_stories_next_limit_ = DEFAULT_LOADED_EXPIRED_STORIES;
MultiTimeout story_reload_timeout_{"StoryReloadTimeout"};
MultiTimeout story_expire_timeout_{"StoryExpireTimeout"};
MultiTimeout story_can_get_viewers_timeout_{"StoryCanGetViewersTimeout"};
Td *td_;
ActorShared<> parent_;
};
} // namespace td