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

Save custom emoji to database.

Browse Source
This commit is contained in:
levlam 2022-07-25 21:18:43 +03:00
parent 8331941754
commit f20f015caa
3 changed files with 75 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
td/telegram/StickersManager.cpp
View File

@ -1695,7 +1695,7 @@ StickerType StickersManager::get_sticker_type(FileId file_id) const {
bool StickersManager::is_premium_custom_emoji(int64 custom_emoji_id) const {
auto it = custom_emoji_to_sticker_id_.find(custom_emoji_id);
if (it == custom_emoji_to_sticker_id_.end()) {
// pretend that unknown custom emoji isn't premium
// pretend that the unknown custom emoji isn't a premium
return false;
}
const Sticker *s = get_sticker(it->second);
@ -2410,6 +2410,36 @@ tl_object_ptr<telegram_api::InputStickerSet> StickersManager::get_input_sticker_
return get_input_sticker_set(sticker_set);
}
class StickersManager::CustomEmojiLogEvent {
public:
FileId sticker_id;
CustomEmojiLogEvent() = default;
explicit CustomEmojiLogEvent(FileId sticker_id) : sticker_id(sticker_id) {
}
template <class StorerT>
void store(StorerT &storer) const {
BEGIN_STORE_FLAGS();
END_STORE_FLAGS();
StickersManager *stickers_manager = storer.context()->td().get_actor_unsafe()->stickers_manager_.get();
stickers_manager->store_sticker(sticker_id, false, storer, "CustomEmoji");
}
template <class ParserT>
void parse(ParserT &parser) {
BEGIN_PARSE_FLAGS();
END_PARSE_FLAGS();
StickersManager *stickers_manager = parser.context()->td().get_actor_unsafe()->stickers_manager_.get();
sticker_id = stickers_manager->parse_sticker(false, parser);
}
};
string StickersManager::get_custom_emoji_database_key(int64 custom_emoji_id) {
return PSTRING() << "emoji" << custom_emoji_id;
}
FileId StickersManager::on_get_sticker(unique_ptr<Sticker> new_sticker, bool replace) {
auto file_id = new_sticker->file_id;
CHECK(file_id.is_valid());
@ -2429,45 +2459,63 @@ FileId StickersManager::on_get_sticker(unique_ptr<Sticker> new_sticker, bool rep
}
}
bool is_changed = false;
if (s->dimensions != new_sticker->dimensions && new_sticker->dimensions.width != 0) {
LOG(DEBUG) << "Sticker " << file_id << " dimensions have changed";
s->dimensions = new_sticker->dimensions;
is_changed = true;
}
if (s->set_id != new_sticker->set_id && new_sticker->set_id.is_valid()) {
LOG_IF(ERROR, s->set_id.is_valid()) << "Sticker " << file_id << " set_id has changed";
s->set_id = new_sticker->set_id;
is_changed = true;
}
if (s->alt != new_sticker->alt && !new_sticker->alt.empty()) {
LOG(DEBUG) << "Sticker " << file_id << " emoji has changed";
s->alt = std::move(new_sticker->alt);
is_changed = true;
}
if (s->minithumbnail != new_sticker->minithumbnail) {
LOG(DEBUG) << "Sticker " << file_id << " minithumbnail has changed";
s->minithumbnail = std::move(new_sticker->minithumbnail);
is_changed = true;
}
if (s->s_thumbnail != new_sticker->s_thumbnail && new_sticker->s_thumbnail.file_id.is_valid()) {
LOG_IF(INFO, s->s_thumbnail.file_id.is_valid()) << "Sticker " << file_id << " s thumbnail has changed from "
<< s->s_thumbnail << " to " << new_sticker->s_thumbnail;
s->s_thumbnail = std::move(new_sticker->s_thumbnail);
is_changed = true;
}
if (s->m_thumbnail != new_sticker->m_thumbnail && new_sticker->m_thumbnail.file_id.is_valid()) {
LOG_IF(INFO, s->m_thumbnail.file_id.is_valid()) << "Sticker " << file_id << " m thumbnail has changed from "
<< s->m_thumbnail << " to " << new_sticker->m_thumbnail;
s->m_thumbnail = std::move(new_sticker->m_thumbnail);
is_changed = true;
}
s->is_premium = new_sticker->is_premium;
s->premium_animation_file_id = new_sticker->premium_animation_file_id;
if (s->format != new_sticker->format && new_sticker->format != StickerFormat::Unknown) {
s->format = new_sticker->format;
is_changed = true;
}
if (s->type != new_sticker->type && new_sticker->type != StickerType::Regular) {
s->type = new_sticker->type;
is_changed = true;
}
if (s->point != new_sticker->point && new_sticker->point != -1) {
s->point = new_sticker->point;
s->x_shift = new_sticker->x_shift;
s->y_shift = new_sticker->y_shift;
s->scale = new_sticker->scale;
is_changed = true;
}
if (s->emoji_receive_date < new_sticker->emoji_receive_date) {
s->emoji_receive_date = new_sticker->emoji_receive_date;
is_changed = true;
}
if (is_changed) {
s->is_from_database = false;
}
}
@ -2475,6 +2523,14 @@ FileId StickersManager::on_get_sticker(unique_ptr<Sticker> new_sticker, bool rep
auto custom_emoji_id = get_custom_emoji_id(file_id);
if (custom_emoji_id != 0) {
custom_emoji_to_sticker_id_[custom_emoji_id] = file_id;
if (!s->is_from_database && G()->parameters().use_file_db && !G()->close_flag()) {
LOG(INFO) << "Save custom emoji " << custom_emoji_id << " to database";
s->is_from_database = true;
CustomEmojiLogEvent log_event(file_id);
G()->td_db()->get_sqlite_pmc()->set(get_custom_emoji_database_key(custom_emoji_id),
log_event_store(log_event).as_slice().str(), Auto());
}
}
}
return file_id;
@ -2977,6 +3033,7 @@ void StickersManager::create_sticker(FileId file_id, FileId premium_animation_fi
s->alt = std::move(custom_emoji->alt_);
s->type = StickerType::CustomEmoji;
s->is_premium = !custom_emoji->free_;
s->emoji_receive_date = G()->unix_time();
}
s->format = format;
on_get_sticker(std::move(s),

5
td/telegram/StickersManager.h
View File

@ -384,10 +384,12 @@ class StickersManager final : public Actor {
StickerFormat format = StickerFormat::Unknown;
StickerType type = StickerType::Regular;
bool is_premium = false;
bool is_from_database = false;
int32 point = -1;
double x_shift = 0;
double y_shift = 0;
double scale = 0;
int32 emoji_receive_date = 0;
};
class StickerSet {
@ -513,6 +515,7 @@ class StickersManager final : public Actor {
void parse(ParserT &parser);
};
class CustomEmojiLogEvent;
class StickerListLogEvent;
class StickerSetListLogEvent;
@ -536,6 +539,8 @@ class StickersManager final : public Actor {
Sticker *get_sticker(FileId file_id);
const Sticker *get_sticker(FileId file_id) const;
string get_custom_emoji_database_key(int64 custom_emoji_id);
FileId on_get_sticker(unique_ptr<Sticker> new_sticker, bool replace);
StickerSet *get_sticker_set(StickerSetId sticker_set_id);

12
td/telegram/StickersManager.hpp
View File

@ -36,6 +36,7 @@ void StickersManager::store_sticker(FileId file_id, bool in_sticker_set, StorerT
bool has_premium_animation = sticker->premium_animation_file_id.is_valid();
bool is_mask = sticker->type == StickerType::Mask;
bool is_emoji = sticker->type == StickerType::CustomEmoji;
bool has_emoji_receive_date = is_emoji && sticker->emoji_receive_date != 0;
BEGIN_STORE_FLAGS();
STORE_FLAG(is_mask);
STORE_FLAG(has_sticker_set_access_hash);
@ -46,6 +47,7 @@ void StickersManager::store_sticker(FileId file_id, bool in_sticker_set, StorerT
STORE_FLAG(has_premium_animation);
STORE_FLAG(is_emoji);
STORE_FLAG(sticker->is_premium);
STORE_FLAG(has_emoji_receive_date);
END_STORE_FLAGS();
if (!in_sticker_set) {
store(sticker->set_id.get(), storer);
@ -72,6 +74,9 @@ void StickersManager::store_sticker(FileId file_id, bool in_sticker_set, StorerT
if (has_premium_animation) {
store(sticker->premium_animation_file_id, storer);
}
if (has_emoji_receive_date) {
store(sticker->emoji_receive_date, storer);
}
}
template <class ParserT>
@ -89,6 +94,7 @@ FileId StickersManager::parse_sticker(bool in_sticker_set, ParserT &parser) {
bool has_premium_animation;
bool is_mask;
bool is_emoji;
bool has_emoji_receive_date;
BEGIN_PARSE_FLAGS();
PARSE_FLAG(is_mask);
PARSE_FLAG(has_sticker_set_access_hash);
@ -99,6 +105,7 @@ FileId StickersManager::parse_sticker(bool in_sticker_set, ParserT &parser) {
PARSE_FLAG(has_premium_animation);
PARSE_FLAG(is_emoji);
PARSE_FLAG(sticker->is_premium);
PARSE_FLAG(has_emoji_receive_date);
END_PARSE_FLAGS();
if (is_webm) {
sticker->format = StickerFormat::Webm;
@ -159,9 +166,14 @@ FileId StickersManager::parse_sticker(bool in_sticker_set, ParserT &parser) {
sticker->is_premium = true;
parse(sticker->premium_animation_file_id, parser);
}
if (has_emoji_receive_date) {
parse(sticker->emoji_receive_date, parser);
}
if (parser.get_error() != nullptr || !sticker->file_id.is_valid()) {
return FileId();
}
sticker->is_from_database = true;
return on_get_sticker(std::move(sticker), false); // data in the database is always outdated
}