/* This file is part of Telegram Desktop, the official desktop application for the Telegram messaging service. For license and copyright information please follow this link: https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL */ #include "data/stickers/data_stickers.h" #include "api/api_hash.h" #include "chat_helpers/compose/compose_show.h" #include "data/data_document.h" #include "data/data_session.h" #include "data/data_user.h" #include "ui/boxes/confirm_box.h" #include "ui/text/text_utilities.h" #include "lang/lang_keys.h" #include "data/data_premium_limits.h" #include "boxes/premium_limits_box.h" #include "history/history.h" #include "history/history_item.h" #include "history/history_item_components.h" #include "apiwrap.h" #include "storage/storage_account.h" #include "settings/settings_premium.h" #include "core/application.h" #include "core/core_settings.h" #include "main/main_session.h" #include "main/main_account.h" #include "main/main_app_config.h" #include "mtproto/mtproto_config.h" #include "ui/toast/toast.h" #include "ui/image/image_location_factory.h" #include "window/window_controller.h" #include "window/window_session_controller.h" #include "mainwindow.h" #include "base/unixtime.h" #include "boxes/abstract_box.h" // Ui::show(). #include "styles/style_chat_helpers.h" namespace Data { namespace { constexpr auto kPremiumToastDuration = 5 * crl::time(1000); using SetFlag = StickersSetFlag; [[nodiscard]] TextWithEntities SavedGifsToast( const Data::PremiumLimits &limits) { const auto defaultLimit = limits.gifsDefault(); const auto premiumLimit = limits.gifsPremium(); return Ui::Text::Bold( tr::lng_saved_gif_limit_title(tr::now, lt_count, defaultLimit) ).append('\n').append( tr::lng_saved_gif_limit_more( tr::now, lt_count, premiumLimit, lt_link, Ui::Text::Link(tr::lng_saved_gif_limit_link(tr::now)), Ui::Text::WithEntities)); } [[nodiscard]] TextWithEntities FaveStickersToast( const Data::PremiumLimits &limits) { const auto defaultLimit = limits.stickersFavedDefault(); const auto premiumLimit = limits.stickersFavedPremium(); return Ui::Text::Bold( tr::lng_fave_sticker_limit_title(tr::now, lt_count, defaultLimit) ).append('\n').append( tr::lng_fave_sticker_limit_more( tr::now, lt_count, premiumLimit, lt_link, Ui::Text::Link(tr::lng_fave_sticker_limit_link(tr::now)), Ui::Text::WithEntities)); } void MaybeShowPremiumToast( std::shared_ptr show, TextWithEntities text, const QString &ref) { if (!show) { return; } const auto session = &show->session(); if (session->user()->isPremium()) { return; } const auto filter = [=](const auto ...) { const auto usage = ChatHelpers::WindowUsage::PremiumPromo; if (const auto controller = show->resolveWindow(usage)) { Settings::ShowPremium(controller, ref); } return false; }; show->showToast({ .text = std::move(text), .duration = kPremiumToastDuration, .filter = filter, }); } void RemoveFromSet( StickersSets &sets, not_null document, uint64 setId) { const auto it = sets.find(setId); if (it == sets.end()) { return; } const auto set = it->second.get(); const auto index = set->stickers.indexOf(document); if (index < 0) { return; } set->stickers.removeAt(index); if (!set->dates.empty()) { set->dates.erase(set->dates.begin() + index); } for (auto i = set->emoji.begin(); i != set->emoji.end();) { const auto index = i->second.indexOf(document); if (index >= 0) { i->second.removeAt(index); if (i->second.empty()) { i = set->emoji.erase(i); continue; } } ++i; } if (set->stickers.empty()) { sets.erase(it); } } } // namespace Stickers::Stickers(not_null owner) : _owner(owner) { } Session &Stickers::owner() const { return *_owner; } Main::Session &Stickers::session() const { return _owner->session(); } void Stickers::notifyUpdated(StickersType type) { _updated.fire_copy(type); } rpl::producer Stickers::updated() const { return _updated.events(); } rpl::producer<> Stickers::updated(StickersType type) const { using namespace rpl::mappers; return updated() | rpl::filter(_1 == type) | rpl::to_empty; } void Stickers::notifyRecentUpdated(StickersType type) { _recentUpdated.fire_copy(type); } rpl::producer Stickers::recentUpdated() const { return _recentUpdated.events(); } rpl::producer<> Stickers::recentUpdated(StickersType type) const { using namespace rpl::mappers; return recentUpdated() | rpl::filter(_1 == type) | rpl::to_empty; } void Stickers::notifySavedGifsUpdated() { _savedGifsUpdated.fire({}); } rpl::producer<> Stickers::savedGifsUpdated() const { return _savedGifsUpdated.events(); } void Stickers::notifyStickerSetInstalled(uint64 setId) { _stickerSetInstalled.fire(std::move(setId)); } rpl::producer Stickers::stickerSetInstalled() const { return _stickerSetInstalled.events(); } void Stickers::notifyEmojiSetInstalled(uint64 setId) { _emojiSetInstalled.fire(std::move(setId)); } rpl::producer Stickers::emojiSetInstalled() const { return _emojiSetInstalled.events(); } void Stickers::incrementSticker(not_null document) { if (!document->sticker() || !document->sticker()->set) { return; } bool writeRecentStickers = false; auto &sets = setsRef(); auto it = sets.find(Data::Stickers::CloudRecentSetId); if (it == sets.cend()) { if (it == sets.cend()) { it = sets.emplace( Data::Stickers::CloudRecentSetId, std::make_unique( &session().data(), Data::Stickers::CloudRecentSetId, uint64(0), // accessHash uint64(0), // hash tr::lng_recent_stickers(tr::now), QString(), 0, // count SetFlag::Special, TimeId(0))).first; } else { it->second->title = tr::lng_recent_stickers(tr::now); } } const auto set = it->second.get(); auto removedFromEmoji = std::vector>(); auto index = set->stickers.indexOf(document); if (index > 0) { if (set->dates.empty()) { session().api().requestRecentStickersForce(); } else { Assert(set->dates.size() == set->stickers.size()); set->dates.erase(set->dates.begin() + index); } set->stickers.removeAt(index); for (auto i = set->emoji.begin(); i != set->emoji.end();) { if (const auto index = i->second.indexOf(document); index >= 0) { removedFromEmoji.emplace_back(i->first); i->second.removeAt(index); if (i->second.empty()) { i = set->emoji.erase(i); continue; } } ++i; } } if (index) { if (set->dates.size() == set->stickers.size()) { set->dates.insert(set->dates.begin(), base::unixtime::now()); } set->stickers.push_front(document); if (const auto emojiList = getEmojiListFromSet(document)) { for (const auto &emoji : *emojiList) { set->emoji[emoji].push_front(document); } } else if (!removedFromEmoji.empty()) { for (const auto emoji : removedFromEmoji) { set->emoji[emoji].push_front(document); } } else { session().api().requestRecentStickersForce(); } writeRecentStickers = true; } // Remove that sticker from old recent, now it is in cloud recent stickers. bool writeOldRecent = false; auto &recent = getRecentPack(); for (auto i = recent.begin(), e = recent.end(); i != e; ++i) { if (i->first == document) { writeOldRecent = true; recent.erase(i); break; } } while (!recent.isEmpty() && (set->stickers.size() + recent.size() > session().serverConfig().stickersRecentLimit)) { writeOldRecent = true; recent.pop_back(); } if (writeOldRecent) { session().saveSettings(); } // Remove that sticker from custom stickers, now it is in cloud recent stickers. bool writeInstalledStickers = false; auto customIt = sets.find(Data::Stickers::CustomSetId); if (customIt != sets.cend()) { const auto custom = customIt->second.get(); int removeIndex = custom->stickers.indexOf(document); if (removeIndex >= 0) { custom->stickers.removeAt(removeIndex); if (custom->stickers.isEmpty()) { sets.erase(customIt); } writeInstalledStickers = true; } } if (writeInstalledStickers) { session().local().writeInstalledStickers(); } if (writeRecentStickers) { session().local().writeRecentStickers(); } notifyRecentUpdated(StickersType::Stickers); } void Stickers::addSavedGif( std::shared_ptr show, not_null document) { const auto index = _savedGifs.indexOf(document); if (!index) { return; } if (index > 0) { _savedGifs.remove(index); } _savedGifs.push_front(document); const auto session = &document->session(); const auto limits = Data::PremiumLimits(session); if (_savedGifs.size() > limits.gifsCurrent()) { _savedGifs.pop_back(); MaybeShowPremiumToast( show, SavedGifsToast(limits), LimitsPremiumRef("saved_gifs")); } session->local().writeSavedGifs(); notifySavedGifsUpdated(); setLastSavedGifsUpdate(0); session->api().updateSavedGifs(); } void Stickers::checkSavedGif(not_null item) { if (item->Has() || (!item->out() && item->history()->peer != session().user())) { return; } if (const auto media = item->media()) { if (const auto document = media->document()) { if (document->isGifv()) { addSavedGif(nullptr, document); } } } } void Stickers::applyArchivedResult( const MTPDmessages_stickerSetInstallResultArchive &d) { auto &v = d.vsets().v; StickersSetsOrder archived; archived.reserve(v.size()); QMap setsToRequest; auto masksCount = 0; auto stickersCount = 0; for (const auto &data : v) { const auto set = feedSet(data); if (set->flags & SetFlag::NotLoaded) { setsToRequest.insert(set->id, set->accessHash); } if (set->type() == StickersType::Emoji) { continue; } const auto isMasks = (set->type() == StickersType::Masks); (isMasks ? masksCount : stickersCount)++; auto &order = isMasks ? maskSetsOrderRef() : setsOrderRef(); const auto index = order.indexOf(set->id); if (index >= 0) { order.removeAt(index); } archived.push_back(set->id); } if (!setsToRequest.isEmpty()) { for (auto i = setsToRequest.cbegin(), e = setsToRequest.cend(); i != e; ++i) { session().api().scheduleStickerSetRequest(i.key(), i.value()); } session().api().requestStickerSets(); } if (stickersCount) { session().local().writeInstalledStickers(); session().local().writeArchivedStickers(); } if (masksCount) { session().local().writeInstalledMasks(); session().local().writeArchivedMasks(); } // TODO async toast. Ui::Toast::Show(Ui::Toast::Config{ .text = { tr::lng_stickers_packs_archived(tr::now) }, .st = &st::stickersToast, .multiline = true, }); //Ui::show( // Box(archived, &session()), // Ui::LayerOption::KeepOther); if (stickersCount) { notifyUpdated(StickersType::Stickers); } if (masksCount) { notifyUpdated(StickersType::Masks); } } void Stickers::installLocally(uint64 setId) { auto &sets = setsRef(); auto it = sets.find(setId); if (it == sets.end()) { return; } const auto set = it->second.get(); const auto flags = set->flags; set->flags &= ~(SetFlag::Archived | SetFlag::Unread); set->flags |= SetFlag::Installed; set->installDate = base::unixtime::now(); auto changedFlags = flags ^ set->flags; const auto isMasks = (set->type() == StickersType::Masks); const auto isEmoji = (set->type() == StickersType::Emoji); auto &order = isEmoji ? emojiSetsOrderRef() : isMasks ? maskSetsOrderRef() : setsOrderRef(); int insertAtIndex = 0, currentIndex = order.indexOf(setId); if (currentIndex != insertAtIndex) { if (currentIndex > 0) { order.removeAt(currentIndex); } order.insert(insertAtIndex, setId); } auto customIt = sets.find(CustomSetId); if (customIt != sets.cend()) { const auto custom = customIt->second.get(); for (const auto sticker : std::as_const(set->stickers)) { int removeIndex = custom->stickers.indexOf(sticker); if (removeIndex >= 0) custom->stickers.removeAt(removeIndex); } if (custom->stickers.isEmpty()) { sets.erase(customIt); } } session().local().writeInstalledStickers(); if (!isMasks && (changedFlags & SetFlag::Unread)) { if (isEmoji) { session().local().writeFeaturedCustomEmoji(); } else { session().local().writeFeaturedStickers(); } } if (!isEmoji && (changedFlags & SetFlag::Archived)) { auto &archivedOrder = isMasks ? archivedMaskSetsOrderRef() : archivedSetsOrderRef(); const auto index = archivedOrder.indexOf(setId); if (index >= 0) { archivedOrder.removeAt(index); if (isMasks) { session().local().writeArchivedMasks(); } else { session().local().writeArchivedStickers(); } } } notifyUpdated(set->type()); } void Stickers::undoInstallLocally(uint64 setId) { const auto &sets = this->sets(); const auto it = sets.find(setId); if (it == sets.end()) { return; } const auto set = it->second.get(); set->flags &= ~SetFlag::Installed; set->installDate = TimeId(0); auto &order = setsOrderRef(); int currentIndex = order.indexOf(setId); if (currentIndex >= 0) { order.removeAt(currentIndex); } session().local().writeInstalledStickers(); notifyUpdated(set->type()); Ui::show( Ui::MakeInformBox(tr::lng_stickers_not_found()), Ui::LayerOption::KeepOther); } bool Stickers::isFaved(not_null document) { const auto &sets = this->sets(); const auto it = sets.find(FavedSetId); if (it == sets.cend()) { return false; } for (const auto sticker : std::as_const(it->second->stickers)) { if (sticker == document) { return true; } } return false; } void Stickers::checkFavedLimit( StickersSet &set, std::shared_ptr show) { const auto session = &_owner->session(); const auto limits = Data::PremiumLimits(session); if (set.stickers.size() <= limits.stickersFavedCurrent()) { return; } auto removing = set.stickers.back(); set.stickers.pop_back(); for (auto i = set.emoji.begin(); i != set.emoji.end();) { auto index = i->second.indexOf(removing); if (index >= 0) { i->second.removeAt(index); if (i->second.empty()) { i = set.emoji.erase(i); continue; } } ++i; } MaybeShowPremiumToast( std::move(show), FaveStickersToast(limits), LimitsPremiumRef("stickers_faved")); } void Stickers::pushFavedToFront( StickersSet &set, std::shared_ptr show, not_null document, const std::vector> &emojiList) { set.stickers.push_front(document); for (auto emoji : emojiList) { set.emoji[emoji].push_front(document); } checkFavedLimit(set, std::move(show)); } void Stickers::moveFavedToFront(StickersSet &set, int index) { Expects(index > 0 && index < set.stickers.size()); auto document = set.stickers[index]; while (index-- != 0) { set.stickers[index + 1] = set.stickers[index]; } set.stickers[0] = document; for (auto &[emoji, list] : set.emoji) { auto index = list.indexOf(document); if (index > 0) { while (index-- != 0) { list[index + 1] = list[index]; } list[0] = document; } } } void Stickers::setIsFaved( std::shared_ptr show, not_null document, std::optional>> emojiList) { auto &sets = setsRef(); auto it = sets.find(FavedSetId); if (it == sets.end()) { it = sets.emplace(FavedSetId, std::make_unique( &document->owner(), FavedSetId, uint64(0), // accessHash uint64(0), // hash Lang::Hard::FavedSetTitle(), QString(), 0, // count SetFlag::Special, TimeId(0))).first; } const auto set = it->second.get(); auto index = set->stickers.indexOf(document); if (index == 0) { return; } if (index > 0) { moveFavedToFront(*set, index); } else if (emojiList) { pushFavedToFront(*set, show, document, *emojiList); } else if (auto list = getEmojiListFromSet(document)) { pushFavedToFront(*set, show, document, *list); } else { requestSetToPushFaved(show, document); return; } session().local().writeFavedStickers(); notifyUpdated(StickersType::Stickers); notifyStickerSetInstalled(FavedSetId); } void Stickers::requestSetToPushFaved( std::shared_ptr show, not_null document) { auto addAnyway = [=](std::vector> list) { if (list.empty()) { if (auto sticker = document->sticker()) { if (auto emoji = Ui::Emoji::Find(sticker->alt)) { list.push_back(emoji); } } } setIsFaved(nullptr, document, std::move(list)); }; session().api().request(MTPmessages_GetStickerSet( Data::InputStickerSet(document->sticker()->set), MTP_int(0) // hash )).done([=](const MTPmessages_StickerSet &result) { result.match([&](const MTPDmessages_stickerSet &data) { auto list = std::vector>(); list.reserve(data.vpacks().v.size()); for (const auto &mtpPack : data.vpacks().v) { auto &pack = mtpPack.c_stickerPack(); for (const auto &documentId : pack.vdocuments().v) { if (documentId.v == document->id) { if (const auto emoji = Ui::Emoji::Find(qs(mtpPack.c_stickerPack().vemoticon()))) { list.emplace_back(emoji); } break; } } } addAnyway(std::move(list)); }, [](const MTPDmessages_stickerSetNotModified &) { LOG(("API Error: Unexpected messages.stickerSetNotModified.")); }); }).fail([=] { // Perhaps this is a deleted sticker pack. Add anyway. addAnyway({}); }).send(); } void Stickers::removeFromRecentSet(not_null document) { RemoveFromSet(setsRef(), document, CloudRecentSetId); session().local().writeRecentStickers(); notifyRecentUpdated(StickersType::Stickers); } void Stickers::setIsNotFaved(not_null document) { RemoveFromSet(setsRef(), document, FavedSetId); session().local().writeFavedStickers(); notifyUpdated(StickersType::Stickers); } void Stickers::setFaved( std::shared_ptr show, not_null document, bool faved) { if (faved) { setIsFaved(std::move(show), document); } else { setIsNotFaved(document); } } void Stickers::setsReceived( const QVector &data, uint64 hash) { somethingReceived(data, hash, StickersType::Stickers); } void Stickers::masksReceived( const QVector &data, uint64 hash) { somethingReceived(data, hash, StickersType::Masks); } void Stickers::emojiReceived( const QVector &data, uint64 hash) { somethingReceived(data, hash, StickersType::Emoji); } void Stickers::somethingReceived( const QVector &list, uint64 hash, StickersType type) { auto &setsOrder = (type == StickersType::Emoji) ? emojiSetsOrderRef() : (type == StickersType::Masks) ? maskSetsOrderRef() : setsOrderRef(); setsOrder.clear(); auto &sets = setsRef(); QMap setsToRequest; for (auto &[id, set] : sets) { const auto archived = !!(set->flags & SetFlag::Archived); if (!archived && (type == set->type())) { // Mark for removing. set->flags &= ~SetFlag::Installed; set->installDate = 0; } } for (const auto &info : list) { const auto set = feedSet(info); if (!(set->flags & SetFlag::Archived) || (set->flags & SetFlag::Official)) { setsOrder.push_back(set->id); if (set->stickers.isEmpty() || (set->flags & SetFlag::NotLoaded)) { setsToRequest.insert(set->id, set->accessHash); } } } auto writeRecent = false; auto &recent = getRecentPack(); for (auto it = sets.begin(); it != sets.end();) { const auto set = it->second.get(); const auto installed = !!(set->flags & SetFlag::Installed); const auto featured = !!(set->flags & SetFlag::Featured); const auto special = !!(set->flags & SetFlag::Special); const auto archived = !!(set->flags & SetFlag::Archived); const auto emoji = !!(set->flags & SetFlag::Emoji); const auto locked = (set->locked > 0); if (!installed) { // remove not mine sets from recent stickers for (auto i = recent.begin(); i != recent.cend();) { if (set->stickers.indexOf(i->first) >= 0) { i = recent.erase(i); writeRecent = true; } else { ++i; } } } if (installed || featured || special || archived || emoji || locked) { ++it; } else { it = sets.erase(it); } } if (!setsToRequest.isEmpty()) { auto &api = session().api(); for (auto i = setsToRequest.cbegin(), e = setsToRequest.cend(); i != e; ++i) { api.scheduleStickerSetRequest(i.key(), i.value()); } api.requestStickerSets(); } if (type == StickersType::Emoji) { session().local().writeInstalledCustomEmoji(); } else if (type == StickersType::Masks) { session().local().writeInstalledMasks(); } else { session().local().writeInstalledStickers(); } if (writeRecent) { session().saveSettings(); } const auto counted = (type == StickersType::Emoji) ? Api::CountCustomEmojiHash(&session()) : (type == StickersType::Masks) ? Api::CountMasksHash(&session()) : Api::CountStickersHash(&session()); if (counted != hash) { LOG(("API Error: received %1 hash %2 while counted hash is %3" ).arg((type == StickersType::Emoji) ? "custom-emoji" : (type == StickersType::Masks) ? "masks" : "stickers" ).arg(hash ).arg(counted)); } notifyUpdated(type); } void Stickers::setPackAndEmoji( StickersSet &set, StickersPack &&pack, const std::vector &&dates, const QVector &packs) { set.stickers = std::move(pack); set.dates = std::move(dates); set.emoji.clear(); for (const auto &mtpPack : packs) { Assert(mtpPack.type() == mtpc_stickerPack); auto &pack = mtpPack.c_stickerPack(); if (auto emoji = Ui::Emoji::Find(qs(pack.vemoticon()))) { emoji = emoji->original(); auto &stickers = pack.vdocuments().v; auto p = StickersPack(); p.reserve(stickers.size()); for (auto j = 0, c = int(stickers.size()); j != c; ++j) { auto document = owner().document(stickers[j].v); if (!document || !document->sticker()) continue; p.push_back(document); } set.emoji[emoji] = std::move(p); } } } void Stickers::specialSetReceived( uint64 setId, const QString &setTitle, const QVector &items, uint64 hash, const QVector &packs, const QVector &usageDates) { auto &sets = setsRef(); auto it = sets.find(setId); if (items.isEmpty()) { if (it != sets.cend()) { sets.erase(it); } } else { if (it == sets.cend()) { it = sets.emplace(setId, std::make_unique( &owner(), setId, uint64(0), // accessHash uint64(0), // hash setTitle, QString(), 0, // count SetFlag::Special, TimeId(0))).first; } else { it->second->title = setTitle; } const auto set = it->second.get(); set->hash = hash; auto dates = std::vector(); auto dateIndex = 0; auto datesAvailable = (items.size() == usageDates.size()) && ((setId == CloudRecentSetId) || (setId == CloudRecentAttachedSetId)); auto customIt = sets.find(CustomSetId); auto pack = StickersPack(); pack.reserve(items.size()); for (const auto &item : items) { ++dateIndex; const auto document = owner().processDocument(item); if (!document->sticker()) { continue; } pack.push_back(document); if (datesAvailable) { dates.push_back(TimeId(usageDates[dateIndex - 1].v)); } if (customIt != sets.cend()) { const auto custom = customIt->second.get(); auto index = custom->stickers.indexOf(document); if (index >= 0) { custom->stickers.removeAt(index); } } } if (customIt != sets.cend() && customIt->second->stickers.isEmpty()) { sets.erase(customIt); customIt = sets.end(); } auto writeRecent = false; auto &recent = getRecentPack(); for (auto i = recent.begin(); i != recent.cend();) { if (set->stickers.indexOf(i->first) >= 0 && pack.indexOf(i->first) < 0) { i = recent.erase(i); writeRecent = true; } else { ++i; } } if (pack.isEmpty()) { sets.erase(it); } else { setPackAndEmoji(*set, std::move(pack), std::move(dates), packs); } if (writeRecent) { session().saveSettings(); } } switch (setId) { case CloudRecentSetId: { const auto counted = Api::CountRecentStickersHash(&session()); if (counted != hash) { LOG(("API Error: " "received recent stickers hash %1 while counted hash is %2" ).arg(hash ).arg(counted)); } session().local().writeRecentStickers(); } break; case CloudRecentAttachedSetId: { const auto counted = Api::CountRecentStickersHash(&session(), true); if (counted != hash) { LOG(("API Error: " "received recent attached stickers hash %1 " "while counted hash is %2" ).arg(hash ).arg(counted)); } session().local().writeRecentMasks(); } break; case FavedSetId: { const auto counted = Api::CountFavedStickersHash(&session()); if (counted != hash) { LOG(("API Error: " "received faved stickers hash %1 while counted hash is %2" ).arg(hash ).arg(counted)); } session().local().writeFavedStickers(); } break; default: Unexpected("setId in SpecialSetReceived()"); } notifyUpdated((setId == CloudRecentAttachedSetId) ? StickersType::Masks : StickersType::Stickers); } void Stickers::featuredSetsReceived( const MTPmessages_FeaturedStickers &result) { setLastFeaturedUpdate(crl::now()); result.match([](const MTPDmessages_featuredStickersNotModified &) { }, [&](const MTPDmessages_featuredStickers &data) { featuredReceived(data, StickersType::Stickers); }); } void Stickers::featuredEmojiSetsReceived( const MTPmessages_FeaturedStickers &result) { setLastFeaturedEmojiUpdate(crl::now()); result.match([](const MTPDmessages_featuredStickersNotModified &) { }, [&](const MTPDmessages_featuredStickers &data) { featuredReceived(data, StickersType::Emoji); }); } void Stickers::featuredReceived( const MTPDmessages_featuredStickers &data, StickersType type) { const auto &list = data.vsets().v; const auto &unread = data.vunread().v; const auto hash = data.vhash().v; auto &&unreadIds = ranges::views::all( unread ) | ranges::views::transform(&MTPlong::v); const auto unreadMap = base::flat_set{ unreadIds.begin(), unreadIds.end() }; const auto isEmoji = (type == StickersType::Emoji); auto &featuredOrder = isEmoji ? featuredEmojiSetsOrderRef() : featuredSetsOrderRef(); featuredOrder.clear(); auto &sets = setsRef(); auto setsToRequest = base::flat_map(); for (auto &[id, set] : sets) { // Mark for removing. if (set->type() == type) { set->flags &= ~SetFlag::Featured; } } for (const auto &entry : list) { const auto data = entry.match([&](const auto &data) { return data.vset().match([&](const MTPDstickerSet &data) { return &data; }); }); auto it = sets.find(data->vid().v); const auto title = getSetTitle(*data); const auto installDate = data->vinstalled_date().value_or_empty(); const auto thumbnail = [&] { if (const auto thumbs = data->vthumbs()) { for (const auto &thumb : thumbs->v) { const auto result = Images::FromPhotoSize( &session(), *data, thumb); if (result.location.valid()) { return result; } } } return ImageWithLocation(); }(); const auto setId = data->vid().v; const auto flags = SetFlag::Featured | (unreadMap.contains(setId) ? SetFlag::Unread : SetFlag()) | ParseStickersSetFlags(*data); if (it == sets.cend()) { it = sets.emplace(data->vid().v, std::make_unique( &owner(), setId, data->vaccess_hash().v, data->vhash().v, title, qs(data->vshort_name()), data->vcount().v, flags | SetFlag::NotLoaded, installDate)).first; } else { const auto set = it->second.get(); set->accessHash = data->vaccess_hash().v; set->title = title; set->shortName = qs(data->vshort_name()); set->flags = flags | (set->flags & (SetFlag::NotLoaded | SetFlag::Special)); set->installDate = installDate; if (set->count != data->vcount().v || set->hash != data->vhash().v || set->emoji.empty()) { set->count = data->vcount().v; set->hash = data->vhash().v; set->flags |= SetFlag::NotLoaded; // need to request this set } } it->second->setThumbnail(thumbnail); it->second->thumbnailDocumentId = data->vthumb_document_id().value_or_empty(); featuredOrder.push_back(data->vid().v); if (it->second->stickers.isEmpty() || (it->second->flags & SetFlag::NotLoaded)) { setsToRequest.emplace(data->vid().v, data->vaccess_hash().v); } } auto unreadCount = 0; for (auto it = sets.begin(); it != sets.end();) { const auto set = it->second.get(); const auto installed = (set->flags & SetFlag::Installed); const auto featured = (set->flags & SetFlag::Featured); const auto special = (set->flags & SetFlag::Special); const auto archived = (set->flags & SetFlag::Archived); const auto emoji = !!(set->flags & SetFlag::Emoji); const auto locked = (set->locked > 0); if (installed || featured || special || archived || emoji || locked) { if (featured && (set->flags & SetFlag::Unread)) { if (!(set->flags & SetFlag::Emoji)) { ++unreadCount; } } ++it; } else { it = sets.erase(it); } } setFeaturedSetsUnreadCount(unreadCount); const auto counted = isEmoji ? Api::CountFeaturedEmojiHash(&session()) : Api::CountFeaturedStickersHash(&session()); if (counted != hash) { LOG(("API Error: " "received featured stickers hash %1 while counted hash is %2" ).arg(hash ).arg(counted)); } if (!setsToRequest.empty()) { auto &api = session().api(); for (const auto &[setId, accessHash] : setsToRequest) { api.scheduleStickerSetRequest(setId, accessHash); } api.requestStickerSets(); } if (isEmoji) { session().local().writeFeaturedCustomEmoji(); } else { session().local().writeFeaturedStickers(); } notifyUpdated(type); } void Stickers::gifsReceived(const QVector &items, uint64 hash) { auto &saved = savedGifsRef(); saved.clear(); saved.reserve(items.size()); for (const auto &item : items) { const auto document = owner().processDocument(item); if (!document->isGifv()) { LOG(("API Error: " "bad document returned in Stickers::gifsReceived!")); continue; } saved.push_back(document); } const auto counted = Api::CountSavedGifsHash(&session()); if (counted != hash) { LOG(("API Error: " "received saved gifs hash %1 while counted hash is %2" ).arg(hash ).arg(counted)); } session().local().writeSavedGifs(); notifySavedGifsUpdated(); } std::vector> Stickers::getListByEmoji( std::vector emoji, uint64 seed, bool forceAllResults) { auto all = base::flat_set(); for (const auto &one : emoji) { all.emplace(one->original()); } const auto single = (all.size() == 1) ? all.front() : nullptr; struct StickerWithDate { not_null document; TimeId date = 0; }; auto result = std::vector(); auto &sets = setsRef(); auto setsToRequest = base::flat_map(); const auto add = [&](not_null document, TimeId date) { if (ranges::find(result, document, [](const StickerWithDate &data) { return data.document; }) == result.end()) { result.push_back({ document, date }); } }; constexpr auto kSlice = 65536; const auto CreateSortKey = [&]( not_null document, int base) { if (document->sticker() && document->sticker()->isAnimated()) { base += kSlice; } return TimeId(base + int((document->id ^ seed) % kSlice)); }; const auto CreateRecentSortKey = [&](not_null document) { return CreateSortKey(document, kSlice * 6); }; auto myCounter = 0; const auto CreateMySortKey = [&](not_null document) { auto base = kSlice * 6; if (!document->sticker() || !document->sticker()->isAnimated()) { base -= kSlice; } return (base - (++myCounter)); }; const auto CreateFeaturedSortKey = [&](not_null document) { return CreateSortKey(document, kSlice * 2); }; const auto CreateOtherSortKey = [&](not_null document) { return CreateSortKey(document, 0); }; const auto InstallDateAdjusted = [&]( TimeId date, not_null document) { return (document->sticker() && document->sticker()->isAnimated()) ? date : date / 2; }; const auto RecentInstallDate = [&](not_null document) { Expects(document->sticker() != nullptr); const auto sticker = document->sticker(); if (sticker->set.id) { const auto setIt = sets.find(sticker->set.id); if (setIt != sets.end()) { return InstallDateAdjusted(setIt->second->installDate, document); } } return TimeId(0); }; auto recentIt = sets.find(Stickers::CloudRecentSetId); if (recentIt != sets.cend()) { const auto recent = recentIt->second.get(); const auto i = single ? recent->emoji.find(single) : recent->emoji.end(); const auto list = (i != recent->emoji.end()) ? &i->second : !single ? &recent->stickers : nullptr; if (list) { const auto count = int(list->size()); result.reserve(count); for (auto i = 0; i != count; ++i) { const auto document = (*list)[i]; const auto sticker = document->sticker(); auto index = i; if (!sticker) { continue; } else if (!single) { const auto main = Ui::Emoji::Find(sticker->alt); if (!main || !all.contains(main)) { continue; } } else { index = recent->stickers.indexOf(document); } const auto usageDate = (recent->dates.empty() || index < 0) ? 0 : recent->dates[index]; const auto date = usageDate ? usageDate : RecentInstallDate(document); result.push_back({ document, date ? date : CreateRecentSortKey(document) }); } } } const auto addList = [&]( const StickersSetsOrder &order, SetFlag skip) { for (const auto setId : order) { auto it = sets.find(setId); if (it == sets.cend() || (it->second->flags & skip)) { continue; } const auto set = it->second.get(); if (set->emoji.empty()) { setsToRequest.emplace(set->id, set->accessHash); set->flags |= SetFlag::NotLoaded; continue; } const auto my = (set->flags & SetFlag::Installed); const auto i = single ? set->emoji.find(single) : set->emoji.end(); const auto list = (i != set->emoji.end()) ? &i->second : !single ? &set->stickers : nullptr; if (list) { result.reserve(result.size() + list->size()); for (const auto document : *list) { const auto sticker = document->sticker(); if (!sticker) { continue; } else if (!single) { const auto main = Ui::Emoji::Find(sticker->alt); if (!main || !all.contains(main)) { continue; } } const auto installDate = my ? set->installDate : TimeId(0); const auto date = (installDate > 1) ? InstallDateAdjusted(installDate, document) : my ? CreateMySortKey(document) : CreateFeaturedSortKey(document); add(document, date); } } } }; addList(setsOrder(), SetFlag::Archived); //addList(featuredSetsOrder(), SetFlag::Installed); if (!setsToRequest.empty()) { for (const auto &[setId, accessHash] : setsToRequest) { session().api().scheduleStickerSetRequest(setId, accessHash); } session().api().requestStickerSets(); } if (forceAllResults || Core::App().settings().suggestStickersByEmoji()) { const auto key = ranges::accumulate( all, QString(), ranges::plus(), &Ui::Emoji::One::text); const auto others = session().api().stickersByEmoji(key); if (others) { result.reserve(result.size() + others->size()); for (const auto document : *others) { add(document, CreateOtherSortKey(document)); } } else if (!forceAllResults) { return {}; } } ranges::sort(result, std::greater<>(), &StickerWithDate::date); const auto appConfig = &session().account().appConfig(); auto mixed = std::vector>(); mixed.reserve(result.size()); auto premiumIndex = 0, nonPremiumIndex = 0; const auto skipToNext = [&](bool premium) { auto &index = premium ? premiumIndex : nonPremiumIndex; while (index < result.size() && result[index].document->isPremiumSticker() != premium) { ++index; } }; const auto done = [&](bool premium) { skipToNext(premium); const auto &index = premium ? premiumIndex : nonPremiumIndex; return (index == result.size()); }; const auto take = [&](bool premium) { if (done(premium)) { return false; } auto &index = premium ? premiumIndex : nonPremiumIndex; mixed.push_back(result[index++].document); return true; }; if (session().premium()) { const auto normalsPerPremium = appConfig->get( u"stickers_normal_by_emoji_per_premium_num"_q, 2); do { // Add "stickers_normal_by_emoji_per_premium_num" non-premium. for (auto i = 0; i < normalsPerPremium; ++i) { if (!take(false)) { break; } } // Then one premium. } while (!done(false) && take(true)); // Add what's left. while (take(false)) { } while (take(true)) { } } else { // All non-premium. while (take(false)) { } // In the end add "stickers_premium_by_emoji_num" premium. const auto premiumsToEnd = appConfig->get( u"stickers_premium_by_emoji_num"_q, 0); for (auto i = 0; i < premiumsToEnd; ++i) { if (!take(true)) { break; } } } return mixed; } std::optional>> Stickers::getEmojiListFromSet( not_null document) { if (auto sticker = document->sticker()) { auto &inputSet = sticker->set; if (!inputSet.id) { return std::nullopt; } const auto &sets = this->sets(); auto it = sets.find(inputSet.id); if (it == sets.cend()) { return std::nullopt; } const auto set = it->second.get(); auto result = std::vector>(); for (auto i = set->emoji.cbegin(), e = set->emoji.cend(); i != e; ++i) { if (i->second.contains(document)) { result.emplace_back(i->first); } } if (result.empty()) { return std::nullopt; } return result; } return std::nullopt; } not_null Stickers::feedSet(const MTPStickerSet &info) { auto &sets = setsRef(); const auto &data = info.data(); auto it = sets.find(data.vid().v); auto title = getSetTitle(data); auto oldFlags = StickersSetFlags(0); const auto thumbnail = [&] { if (const auto thumbs = data.vthumbs()) { for (const auto &thumb : thumbs->v) { const auto result = Images::FromPhotoSize( &session(), data, thumb); if (result.location.valid()) { return result; } } } return ImageWithLocation(); }(); const auto flags = ParseStickersSetFlags(data); if (it == sets.cend()) { it = sets.emplace(data.vid().v, std::make_unique( &owner(), data.vid().v, data.vaccess_hash().v, data.vhash().v, title, qs(data.vshort_name()), data.vcount().v, flags | SetFlag::NotLoaded, data.vinstalled_date().value_or_empty())).first; } else { const auto set = it->second.get(); set->accessHash = data.vaccess_hash().v; set->title = title; set->shortName = qs(data.vshort_name()); oldFlags = set->flags; const auto clientFlags = set->flags & (SetFlag::Featured | SetFlag::Unread | SetFlag::NotLoaded | SetFlag::Special); set->flags = flags | clientFlags; const auto installDate = data.vinstalled_date(); set->installDate = installDate ? (installDate->v ? installDate->v : base::unixtime::now()) : TimeId(0); if (set->count != data.vcount().v || set->hash != data.vhash().v || set->emoji.empty()) { // Need to request this data. set->count = data.vcount().v; set->hash = data.vhash().v; set->flags |= SetFlag::NotLoaded; } } const auto set = it->second.get(); set->setThumbnail(thumbnail); set->thumbnailDocumentId = data.vthumb_document_id().value_or_empty(); auto changedFlags = (oldFlags ^ set->flags); if (changedFlags & SetFlag::Archived) { const auto isMasks = (set->type() == StickersType::Masks); auto &archivedOrder = isMasks ? archivedMaskSetsOrderRef() : archivedSetsOrderRef(); const auto index = archivedOrder.indexOf(set->id); if (set->flags & SetFlag::Archived) { if (index < 0) { archivedOrder.push_front(set->id); } } else if (index >= 0) { archivedOrder.removeAt(index); } } return it->second.get(); } not_null Stickers::feedSetFull( const MTPDmessages_stickerSet &data) { const auto set = feedSet(data.vset()); feedSetStickers(set, data.vdocuments().v, data.vpacks().v); return set; } not_null Stickers::feedSet( const MTPStickerSetCovered &data) { const auto set = data.match([&](const auto &data) { return feedSet(data.vset()); }); data.match([](const MTPDstickerSetCovered &data) { }, [&](const MTPDstickerSetNoCovered &data) { }, [&](const MTPDstickerSetMultiCovered &data) { feedSetCovers(set, data.vcovers().v); }, [&](const MTPDstickerSetFullCovered &data) { feedSetStickers(set, data.vdocuments().v, data.vpacks().v); }); return set; } void Stickers::feedSetStickers( not_null set, const QVector &documents, const QVector &packs) { set->flags &= ~SetFlag::NotLoaded; auto &sets = setsRef(); const auto wasArchived = [&] { const auto it = sets.find(set->id); return (it != sets.end()) && (it->second->flags & SetFlag::Archived); }(); auto customIt = sets.find(Stickers::CustomSetId); const auto inputSet = set->identifier(); auto pack = StickersPack(); pack.reserve(documents.size()); for (const auto &item : documents) { const auto document = owner().processDocument(item); if (!document->sticker()) { continue; } pack.push_back(document); if (!document->sticker()->set.id) { document->sticker()->set = inputSet; } if (customIt != sets.cend()) { const auto custom = customIt->second.get(); const auto index = custom->stickers.indexOf(document); if (index >= 0) { custom->stickers.removeAt(index); } } } if (customIt != sets.cend() && customIt->second->stickers.isEmpty()) { sets.erase(customIt); customIt = sets.end(); } auto writeRecent = false; auto &recent = getRecentPack(); for (auto i = recent.begin(); i != recent.cend();) { if (set->stickers.indexOf(i->first) >= 0 && pack.indexOf(i->first) < 0) { i = recent.erase(i); writeRecent = true; } else { ++i; } } const auto isEmoji = (set->type() == StickersType::Emoji); const auto isMasks = (set->type() == StickersType::Masks); set->stickers = pack; set->emoji.clear(); for (auto i = 0, l = int(packs.size()); i != l; ++i) { const auto &pack = packs[i].data(); if (auto emoji = Ui::Emoji::Find(qs(pack.vemoticon()))) { emoji = emoji->original(); auto &stickers = pack.vdocuments().v; auto p = StickersPack(); p.reserve(stickers.size()); for (auto j = 0, c = int(stickers.size()); j != c; ++j) { const auto document = owner().document(stickers[j].v); if (!document->sticker()) { continue; } p.push_back(document); } set->emoji[emoji] = std::move(p); } } if (writeRecent) { session().saveSettings(); } const auto isArchived = !!(set->flags & SetFlag::Archived); if ((set->flags & SetFlag::Installed) && !isArchived) { if (isEmoji) { session().local().writeInstalledCustomEmoji(); } else if (isMasks) { session().local().writeInstalledMasks(); } else { session().local().writeInstalledStickers(); } } if (set->flags & SetFlag::Featured) { if (isEmoji) { session().local().writeFeaturedCustomEmoji(); } else if (isMasks) { } else { session().local().writeFeaturedStickers(); } } if (wasArchived != isArchived) { if (isEmoji) { } else if (isMasks) { session().local().writeArchivedMasks(); } else { session().local().writeArchivedStickers(); } } notifyUpdated(set->type()); } void Stickers::feedSetCovers( not_null set, const QVector &documents) { set->covers = StickersPack(); for (const auto &cover : documents) { const auto document = session().data().processDocument(cover); if (document->sticker()) { set->covers.push_back(document); } } } void Stickers::newSetReceived(const MTPDmessages_stickerSet &set) { const auto &s = set.vset().c_stickerSet(); if (!s.vinstalled_date()) { LOG(("API Error: " "updateNewStickerSet without install_date flag.")); return; } else if (s.is_archived()) { LOG(("API Error: " "updateNewStickerSet with archived flag.")); return; } auto &order = s.is_emojis() ? emojiSetsOrderRef() : s.is_masks() ? maskSetsOrderRef() : setsOrderRef(); int32 insertAtIndex = 0, currentIndex = order.indexOf(s.vid().v); if (currentIndex != insertAtIndex) { if (currentIndex > 0) { order.removeAt(currentIndex); } order.insert(insertAtIndex, s.vid().v); } feedSetFull(set); } QString Stickers::getSetTitle(const MTPDstickerSet &s) { auto title = qs(s.vtitle()); if ((s.vflags().v & MTPDstickerSet::Flag::f_official) && !title.compare(u"Great Minds"_q, Qt::CaseInsensitive)) { return tr::lng_stickers_default_set(tr::now); } return title; } RecentStickerPack &Stickers::getRecentPack() const { if (cRecentStickers().isEmpty() && !cRecentStickersPreload().isEmpty()) { const auto p = cRecentStickersPreload(); cSetRecentStickersPreload(RecentStickerPreload()); auto &recent = cRefRecentStickers(); recent.reserve(p.size()); for (const auto &preloaded : p) { const auto document = owner().document(preloaded.first); if (!document || !document->sticker()) continue; recent.push_back(qMakePair(document, preloaded.second)); } } return cRefRecentStickers(); } } // namespace Stickers