tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Load default tag reactions.

Browse Source
This commit is contained in:
levlam 2024-01-18 20:08:37 +03:00
parent 20c4815a12
commit 74aab0a7b2
3 changed files with 103 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

93
td/telegram/ReactionManager.cpp
View File

@ -77,7 +77,7 @@ class GetRecentReactionsQuery final : public Td::ResultHandler {
class GetTopReactionsQuery final : public Td::ResultHandler {
public:
void send(int64 hash) {
send_query(G()->net_query_creator().create(telegram_api::messages_getTopReactions(50, hash)));
send_query(G()->net_query_creator().create(telegram_api::messages_getTopReactions(200, hash)));
}
void on_result(BufferSlice packet) final {
@ -97,6 +97,29 @@ class GetTopReactionsQuery final : public Td::ResultHandler {
}
};
class GetDefaultTagReactionsQuery final : public Td::ResultHandler {
public:
void send(int64 hash) {
send_query(G()->net_query_creator().create(telegram_api::messages_getDefaultTagReactions(hash)));
}
void on_result(BufferSlice packet) final {
auto result_ptr = fetch_result<telegram_api::messages_getDefaultTagReactions>(packet);
if (result_ptr.is_error()) {
return on_error(result_ptr.move_as_error());
}
auto ptr = result_ptr.move_as_ok();
LOG(INFO) << "Receive result for GetDefaultTagReactionsQuery: " << to_string(ptr);
td_->reaction_manager_->on_get_default_tag_reactions(std::move(ptr));
}
void on_error(Status status) final {
LOG(INFO) << "Receive error for GetDefaultTagReactionsQuery: " << status;
td_->reaction_manager_->on_get_default_tag_reactions(nullptr);
}
};
class ClearRecentReactionsQuery final : public Td::ResultHandler {
Promise<Unit> promise_;
@ -387,6 +410,16 @@ void ReactionManager::reload_top_reactions() {
td_->create_handler<GetTopReactionsQuery>()->send(top_reactions_.hash_);
}
void ReactionManager::reload_default_tag_reactions() {
if (G()->close_flag() || default_tag_reactions_.is_being_reloaded_) {
return;
}
CHECK(!td_->auth_manager_->is_bot());
default_tag_reactions_.is_being_reloaded_ = true;
load_default_tag_reactions(); // must be after is_being_reloaded_ is set to true to avoid recursion
td_->create_handler<GetDefaultTagReactionsQuery>()->send(default_tag_reactions_.hash_);
}
td_api::object_ptr<td_api::updateActiveEmojiReactions> ReactionManager::get_update_active_emoji_reactions_object()
const {
return td_api::make_object<td_api::updateActiveEmojiReactions>(
@ -416,6 +449,13 @@ void ReactionManager::save_top_reactions() {
G()->td_db()->get_binlog_pmc()->set("top_reactions", log_event_store(top_reactions_).as_slice().str());
}
void ReactionManager::save_default_tag_reactions() {
LOG(INFO) << "Save " << default_tag_reactions_.reaction_types_.size() << " default tag reactions";
are_default_tag_reactions_loaded_from_database_ = true;
G()->td_db()->get_binlog_pmc()->set("default_tag_reactions",
log_event_store(default_tag_reactions_).as_slice().str());
}
void ReactionManager::load_active_reactions() {
LOG(INFO) << "Loading active reactions";
string active_reaction_types = G()->td_db()->get_binlog_pmc()->get("active_reactions");
@ -512,6 +552,28 @@ void ReactionManager::load_top_reactions() {
LOG(INFO) << "Successfully loaded " << top_reactions_.reaction_types_.size() << " top reactions";
}
void ReactionManager::load_default_tag_reactions() {
if (are_default_tag_reactions_loaded_from_database_) {
return;
}
are_default_tag_reactions_loaded_from_database_ = true;
LOG(INFO) << "Loading default tag reactions";
string default_tag_reactions = G()->td_db()->get_binlog_pmc()->get("default_tag_reactions");
if (default_tag_reactions.empty()) {
return reload_default_tag_reactions();
}
auto status = log_event_parse(default_tag_reactions_, default_tag_reactions);
if (status.is_error()) {
LOG(ERROR) << "Can't load default tag reactions: " << status;
default_tag_reactions_ = {};
return reload_default_tag_reactions();
}
LOG(INFO) << "Successfully loaded " << default_tag_reactions_.reaction_types_.size() << " default tag reactions";
}
void ReactionManager::update_active_reactions() {
vector<ReactionType> active_reaction_types;
for (auto &reaction : reactions_.reactions_) {
@ -673,6 +735,35 @@ void ReactionManager::on_get_top_reactions(tl_object_ptr<telegram_api::messages_
save_top_reactions();
}
void ReactionManager::on_get_default_tag_reactions(tl_object_ptr<telegram_api::messages_Reactions> &&reactions_ptr) {
CHECK(default_tag_reactions_.is_being_reloaded_);
default_tag_reactions_.is_being_reloaded_ = false;
if (reactions_ptr == nullptr) {
// failed to get default tag reactions
return;
}
int32 constructor_id = reactions_ptr->get_id();
if (constructor_id == telegram_api::messages_reactionsNotModified::ID) {
LOG(INFO) << "Top reactions are not modified";
return;
}
CHECK(constructor_id == telegram_api::messages_reactions::ID);
auto reactions = move_tl_object_as<telegram_api::messages_reactions>(reactions_ptr);
auto new_reaction_types = ReactionType::get_reaction_types(reactions->reactions_);
if (new_reaction_types == default_tag_reactions_.reaction_types_ &&
default_tag_reactions_.hash_ == reactions->hash_) {
LOG(INFO) << "Top reactions are not modified";
return;
}
default_tag_reactions_.reaction_types_ = std::move(new_reaction_types);
default_tag_reactions_.hash_ = reactions->hash_;
save_default_tag_reactions();
}
bool ReactionManager::is_active_reaction(const ReactionType &reaction_type) const {
return td::contains(active_reaction_types_, reaction_type);
}

10
td/telegram/ReactionManager.h
View File

@ -54,12 +54,16 @@ class ReactionManager final : public Actor {
void reload_top_reactions();
void reload_default_tag_reactions();
void on_get_available_reactions(tl_object_ptr<telegram_api::messages_AvailableReactions> &&available_reactions_ptr);
void on_get_recent_reactions(tl_object_ptr<telegram_api::messages_Reactions> &&reactions_ptr);
void on_get_top_reactions(tl_object_ptr<telegram_api::messages_Reactions> &&reactions_ptr);
void on_get_default_tag_reactions(tl_object_ptr<telegram_api::messages_Reactions> &&reactions_ptr);
void set_default_reaction(ReactionType reaction_type, Promise<Unit> &&promise);
void send_set_default_reaction_query();
@ -132,6 +136,8 @@ class ReactionManager final : public Actor {
void save_top_reactions();
void save_default_tag_reactions();
void load_active_reactions();
void load_reactions();
@ -140,6 +146,8 @@ class ReactionManager final : public Actor {
void load_top_reactions();
void load_default_tag_reactions();
void update_active_reactions();
td_api::object_ptr<td_api::updateActiveEmojiReactions> get_update_active_emoji_reactions_object() const;
@ -156,10 +164,12 @@ class ReactionManager final : public Actor {
ReactionList recent_reactions_;
ReactionList top_reactions_;
ReactionList default_tag_reactions_;
bool are_reactions_loaded_from_database_ = false;
bool are_recent_reactions_loaded_from_database_ = false;
bool are_top_reactions_loaded_from_database_ = false;
bool are_default_tag_reactions_loaded_from_database_ = false;
};
} // namespace td

1
td/telegram/UpdatesManager.cpp
View File

@ -2267,6 +2267,7 @@ void UpdatesManager::try_reload_data() {
td_->reaction_manager_->reload_reactions();
td_->reaction_manager_->reload_recent_reactions();
td_->reaction_manager_->reload_top_reactions();
td_->reaction_manager_->reload_default_tag_reactions();
for (int32 type = 0; type < MAX_STICKER_TYPE; type++) {
auto sticker_type = static_cast<StickerType>(type);
td_->stickers_manager_->get_installed_sticker_sets(sticker_type, Auto());