// // 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/ReactionType.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/StoryStealthMode.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 #include #include 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_; ReactionType chosen_reaction_type_; UserPrivacySettingRules privacy_rules_; unique_ptr content_; vector areas_; FormattedText caption_; int64 global_id_ = 0; template void store(StorerT &storer) const; template void parse(ParserT &parser); }; struct StoryInfo { StoryId story_id_; int32 date_ = 0; int32 expire_date_ = 0; bool is_for_close_friends_ = false; template void store(StorerT &storer) const; template void parse(ParserT &parser); }; struct BeingEditedStory { unique_ptr content_; vector areas_; FormattedText caption_; bool edit_media_areas_ = false; bool edit_caption_ = false; vector> 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_; PendingStory() = default; PendingStory(DialogId dialog_id, StoryId story_id, uint32 send_story_num, int64 random_id, unique_ptr &&story); template void store(StorerT &storer) const; template void parse(ParserT &parser); }; struct ReadyToSendStory { FileId file_id_; unique_ptr pending_story_; telegram_api::object_ptr input_file_; ReadyToSendStory(FileId file_id, unique_ptr &&pending_story, telegram_api::object_ptr &&input_file); }; struct PendingStoryViews { FlatHashSet story_ids_; bool has_query_ = false; }; struct ActiveStories { StoryId max_read_story_id_; vector story_ids_; StoryListId story_list_id_; int64 private_order_ = 0; int64 public_order_ = 0; }; struct SavedActiveStories { StoryId max_read_story_id_; vector story_infos_; template void store(StorerT &storer) const; template void parse(ParserT &parser); }; 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> load_list_from_server_queries_; vector> load_list_from_database_queries_; std::set 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 void store(StorerT &storer) const; template 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> &&promise); void can_send_story(Promise> &&promise); void send_story(td_api::object_ptr &&input_story_content, td_api::object_ptr &&input_areas, td_api::object_ptr &&input_caption, td_api::object_ptr &&settings, int32 active_period, bool is_pinned, bool protect_content, Promise> &&promise); void on_send_story_file_parts_missing(unique_ptr &&pending_story, vector &&bad_parts); void edit_story(StoryId story_id, td_api::object_ptr &&input_story_content, td_api::object_ptr &&input_areas, td_api::object_ptr &&input_caption, Promise &&promise); void set_story_privacy_settings(StoryId story_id, td_api::object_ptr &&settings, Promise &&promise); void toggle_story_is_pinned(StoryId story_id, bool is_pinned, Promise &&promise); void delete_story(StoryId story_id, Promise &&promise); void load_active_stories(StoryListId story_list_id, Promise &&promise); void reload_active_stories(); void reload_all_read_stories(); void toggle_dialog_stories_hidden(DialogId dialog_id, StoryListId story_list_id, Promise &&promise); void get_dialog_pinned_stories(DialogId owner_dialog_id, StoryId from_story_id, int32 limit, Promise> &&promise); void get_story_archive(StoryId from_story_id, int32 limit, Promise> &&promise); void get_dialog_expiring_stories(DialogId owner_dialog_id, Promise> &&promise); void reload_dialog_expiring_stories(DialogId dialog_id); void open_story(DialogId owner_dialog_id, StoryId story_id, Promise &&promise); void close_story(DialogId owner_dialog_id, StoryId story_id, Promise &&promise); void view_story_message(StoryFullId story_full_id); void on_story_replied(StoryFullId story_full_id, UserId replier_user_id); void set_story_reaction(StoryFullId story_full_id, ReactionType reaction_type, bool add_to_recent, Promise &&promise); void get_story_viewers(StoryId story_id, const string &query, bool only_contacts, bool prefer_with_reaction, const string &offset, int32 limit, Promise> &&promise); void report_story(StoryFullId story_full_id, ReportReason &&reason, Promise &&promise); void activate_stealth_mode(Promise &&promise); void remove_story_notifications_by_story_ids(DialogId dialog_id, const vector &story_ids); StoryId on_get_story(DialogId owner_dialog_id, telegram_api::object_ptr &&story_item_ptr); std::pair> on_get_stories(DialogId owner_dialog_id, vector &&expected_story_ids, telegram_api::object_ptr &&stories); DialogId on_get_user_stories(DialogId owner_dialog_id, telegram_api::object_ptr &&user_stories, Promise &&promise); void on_update_story_id(int64 random_id, StoryId new_story_id, const char *source); bool on_update_read_stories(DialogId owner_dialog_id, StoryId max_read_story_id); void on_update_story_stealth_mode(telegram_api::object_ptr &&stealth_mode); void on_update_story_chosen_reaction_type(DialogId owner_dialog_id, StoryId story_id, ReactionType chosen_reaction_type); void on_update_dialog_stories_hidden(DialogId owner_dialog_id, bool stories_hidden); 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, bool ignore_premium) const; void on_get_story_views(const vector &story_ids, telegram_api::object_ptr &&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 get_story_object(StoryFullId story_full_id) const; td_api::object_ptr get_stories_object(int32 total_count, const vector &story_full_ids) const; FileSourceId get_story_file_source_id(StoryFullId story_full_id); telegram_api::object_ptr get_input_media(StoryFullId story_full_id) const; void reload_story(StoryFullId story_full_id, Promise &&promise, const char *source); void try_synchronize_archive_all_stories(); void get_current_state(vector> &updates) const; void on_binlog_events(vector &&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 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 get_story_info_object(StoryFullId story_full_id) const; td_api::object_ptr get_story_object(StoryFullId story_full_id, const Story *story) const; td_api::object_ptr 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 &&story_item); StoryId on_get_skipped_story(DialogId owner_dialog_id, telegram_api::object_ptr &&story_item); StoryId on_get_deleted_story(DialogId owner_dialog_id, telegram_api::object_ptr &&story_item); void on_delete_story(StoryFullId story_full_id); void on_get_dialog_pinned_stories(DialogId owner_dialog_id, telegram_api::object_ptr &&stories, Promise> &&promise); void on_get_story_archive(telegram_api::object_ptr &&stories, Promise> &&promise); void on_get_dialog_expiring_stories(DialogId owner_dialog_id, telegram_api::object_ptr &&stories, Promise> &&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 result); void load_active_stories_from_server(StoryListId story_list_id, StoryList &story_list, bool is_next, Promise &&promise); void on_load_active_stories_from_server( StoryListId story_list_id, bool is_next, string old_state, Result> 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 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 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 &&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 &old_file_ids); void do_get_story(StoryFullId story_full_id, Result &&result, Promise> &&promise); void on_reload_story(StoryFullId story_full_id, Result &&result); int64 save_send_story_log_event(const PendingStory *pending_story); void delete_pending_story(FileId file_id, unique_ptr &&pending_story, Status status); Result get_next_yet_unsent_story_id(DialogId dialog_id); void do_send_story(unique_ptr &&pending_story, vector bad_parts); void on_upload_story(FileId file_id, telegram_api::object_ptr input_file); void on_upload_story_error(FileId file_id, Status status); void try_send_story(DialogId dialog_id); void do_edit_story(FileId file_id, unique_ptr &&pending_story, telegram_api::object_ptr input_file); void on_toggle_story_is_pinned(StoryId story_id, bool is_pinned, Promise &&promise); void on_update_dialog_max_story_ids(DialogId owner_dialog_id, StoryId max_story_id, StoryId max_read_story_id); void on_update_dialog_max_read_story_id(DialogId owner_dialog_id, StoryId max_read_story_id); void on_update_dialog_has_pinned_stories(DialogId owner_dialog_id, bool has_pinned_stories); void on_update_active_stories(DialogId owner_dialog_id, StoryId max_read_story_id, vector &&story_ids, Promise &&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 get_update_chat_active_stories_object( DialogId owner_dialog_id, const ActiveStories *active_stories) const; void send_update_chat_active_stories(DialogId owner_dialog_id, const ActiveStories *active_stories, const char *source); void save_active_stories(DialogId owner_dialog_id, const ActiveStories *active_stories, Promise &&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); bool can_use_story_reaction(const ReactionType &reaction_type) const; 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 result); td_api::object_ptr get_update_story_stealth_mode() const; void send_update_story_stealth_mode() const; void schedule_stealth_mode_update(); static void update_stealth_mode_static(void *story_manager); void update_stealth_mode(); static string get_story_stealth_mode_key(); void set_story_stealth_mode(StoryStealthMode stealth_mode); void on_get_story_viewers(StoryId story_id, bool is_full, bool is_first, Result> r_view_list, Promise> &&promise); void on_set_story_reaction(StoryFullId story_full_id, Result &&result, Promise &&promise); void load_expired_database_stories(); void on_load_expired_database_stories(vector stories); std::shared_ptr upload_media_callback_; WaitFreeHashMap story_full_id_to_file_source_id_; WaitFreeHashMap, StoryFullIdHash> stories_; WaitFreeHashMap stories_by_global_id_; WaitFreeHashMap inaccessible_story_full_ids_; WaitFreeHashSet deleted_story_full_ids_; WaitFreeHashSet failed_to_load_story_full_ids_; WaitFreeHashMap, StoryFullIdHash> story_messages_; WaitFreeHashMap, DialogIdHash> active_stories_; WaitFreeHashSet updated_active_stories_; WaitFreeHashMap max_read_story_ids_; WaitFreeHashSet failed_to_load_active_stories_; FlatHashMap load_expiring_stories_log_event_ids_; FlatHashMap, StoryFullIdHash> being_edited_stories_; FlatHashMap edit_generations_; FlatHashMap pending_story_views_; FlatHashMap opened_owned_stories_; FlatHashMap opened_stories_; FlatHashMap>, StoryFullIdHash> reload_story_queries_; FlatHashMap, FileIdHash> being_uploaded_files_; FlatHashMap, DialogIdHash> yet_unsent_stories_; FlatHashMap, DialogIdHash> yet_unsent_story_ids_; FlatHashMap being_sent_stories_; FlatHashMap being_sent_story_random_ids_; FlatHashMap being_uploaded_file_ids_; FlatHashMap update_story_ids_; FlatHashMap>> delete_yet_unsent_story_queries_; FlatHashMap> ready_to_send_stories_; StoryList story_lists_[2]; StoryStealthMode stealth_mode_; uint32 send_story_count_ = 0; int64 max_story_global_id_ = 0; FlatHashMap current_yet_unsent_story_ids_; bool has_active_synchronize_archive_all_stories_query_ = false; Timeout stealth_mode_update_timeout_; 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