/* 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 "data/data_document.h" #include "data/data_session.h" #include "data/data_user.h" #include "ui/boxes/confirm_box.h" #include "lang/lang_keys.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 "core/application.h" #include "core/core_settings.h" #include "main/main_session.h" #include "mtproto/mtproto_config.h" #include "ui/toast/toast.h" #include "ui/image/image_location_factory.h" #include "base/unixtime.h" #include "boxes/abstract_box.h" // Ui::show(). #include "styles/style_chat_helpers.h" namespace Data { namespace { using SetFlag = StickersSetFlag; 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->indexOf(document); if (index >= 0) { i->removeAt(index); if (i->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() { _updated.fire({}); } rpl::producer<> Stickers::updated() const { return _updated.events(); } void Stickers::notifyRecentUpdated(Recent recent) { _recentUpdated.fire(std::move(recent)); } rpl::producer Stickers::recentUpdated() const { return _recentUpdated.events(); } 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::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->indexOf(document); index >= 0) { removedFromEmoji.emplace_back(i.key()); i->removeAt(index); if (i->isEmpty()) { 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(); } void Stickers::addSavedGif(not_null document) { const auto index = _savedGifs.indexOf(document); if (!index) { return; } if (index > 0) { _savedGifs.remove(index); } _savedGifs.push_front(document); if (_savedGifs.size() > session().serverConfig().savedGifsLimit) { _savedGifs.pop_back(); } session().local().writeSavedGifs(); notifySavedGifsUpdated(); setLastSavedGifsUpdate(0); session().api().updateStickers(); } 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(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 &stickerSet : v) { const MTPDstickerSet *setData = nullptr; switch (stickerSet.type()) { case mtpc_stickerSetCovered: { auto &d = stickerSet.c_stickerSetCovered(); if (d.vset().type() == mtpc_stickerSet) { setData = &d.vset().c_stickerSet(); } } break; case mtpc_stickerSetMultiCovered: { auto &d = stickerSet.c_stickerSetMultiCovered(); if (d.vset().type() == mtpc_stickerSet) { setData = &d.vset().c_stickerSet(); } } break; } if (setData) { auto set = feedSet(*setData); if (set->stickers.isEmpty()) { setsToRequest.insert(set->id, set->accessHash); } const auto masks = !!(set->flags & SetFlag::Masks); (masks ? masksCount : stickersCount)++; auto &order = masks ? 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(); } 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); notifyUpdated(); } void Stickers::installLocally(uint64 setId) { auto &sets = setsRef(); auto it = sets.find(setId); if (it == sets.end()) { return; } const auto set = it->second.get(); 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 masks = !!(flags & SetFlag::Masks); auto &order = masks ? 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 (changedFlags & SetFlag::Unread) { session().local().writeFeaturedStickers(); } if (changedFlags & SetFlag::Archived) { auto &archivedOrder = masks ? archivedMaskSetsOrderRef() : archivedSetsOrderRef(); const auto index = archivedOrder.indexOf(setId); if (index >= 0) { archivedOrder.removeAt(index); if (masks) { session().local().writeArchivedMasks(); } else { session().local().writeArchivedStickers(); } } } notifyUpdated(); } 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(); 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) { if (set.stickers.size() <= session().serverConfig().stickersFavedLimit) { return; } auto removing = set.stickers.back(); set.stickers.pop_back(); for (auto i = set.emoji.begin(); i != set.emoji.end();) { auto index = i->indexOf(removing); if (index >= 0) { i->removeAt(index); if (i->empty()) { i = set.emoji.erase(i); continue; } } ++i; } } void Stickers::pushFavedToFront( StickersSet &set, not_null document, const std::vector> &emojiList) { set.stickers.push_front(document); for (auto emoji : emojiList) { set.emoji[emoji].push_front(document); } checkFavedLimit(set); } 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 &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( 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, document, *emojiList); } else if (auto list = getEmojiListFromSet(document)) { pushFavedToFront(*set, document, *list); } else { requestSetToPushFaved(document); return; } session().local().writeFavedStickers(); notifyUpdated(); notifyStickerSetInstalled(FavedSetId); } void Stickers::requestSetToPushFaved(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(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(); } void Stickers::setIsNotFaved(not_null document) { RemoveFromSet(setsRef(), document, FavedSetId); session().local().writeFavedStickers(); notifyUpdated(); } void Stickers::setFaved(not_null document, bool faved) { if (faved) { setIsFaved(document); } else { setIsNotFaved(document); } } void Stickers::setsReceived( const QVector &data, uint64 hash) { setsOrMasksReceived(data, hash, false); } void Stickers::masksReceived( const QVector &data, uint64 hash) { setsOrMasksReceived(data, hash, true); } void Stickers::setsOrMasksReceived( const QVector &data, uint64 hash, bool masks) { auto &setsOrder = masks ? maskSetsOrderRef() : setsOrderRef(); setsOrder.clear(); auto &sets = setsRef(); QMap setsToRequest; for (auto &[id, set] : sets) { const auto archived = !!(set->flags & SetFlag::Archived); const auto maskset = !!(set->flags & SetFlag::Masks); if (!archived && (masks == maskset)) { // Mark for removing. set->flags &= ~SetFlag::Installed; set->installDate = 0; } } for (const auto &setData : data) { if (setData.type() != mtpc_stickerSet) { continue; } const auto set = feedSet(setData.c_stickerSet()); 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); 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) { ++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 (masks) { session().local().writeInstalledMasks(); } else { session().local().writeInstalledStickers(); } if (writeRecent) { session().saveSettings(); } const auto counted = masks ? Api::CountMasksHash(&session()) : Api::CountStickersHash(&session()); if (counted != hash) { LOG(("API Error: received %1 hash %2 while counted hash is %3" ).arg(masks ? "masks" : "stickers" ).arg(hash ).arg(counted)); } notifyUpdated(); } 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.insert(emoji, 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, 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(); } void Stickers::featuredSetsReceived( const QVector &list, const QVector &unread, uint64 hash) { auto &&unreadIds = ranges::views::all( unread ) | ranges::views::transform([](const MTPlong &id) { return id.v; }); const auto unreadMap = base::flat_set{ unreadIds.begin(), unreadIds.end() }; auto &setsOrder = featuredSetsOrderRef(); setsOrder.clear(); auto &sets = setsRef(); auto setsToRequest = base::flat_map(); for (auto &[id, set] : sets) { // Mark for removing. 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 flags = SetFlag::Featured | (unreadMap.contains(data->vid().v) ? SetFlag::Unread : SetFlag()) | 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, installDate)).first; it->second->setThumbnail(thumbnail); } 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; set->setThumbnail(thumbnail); if (set->count != data->vcount().v || set->hash != data->vhash().v || set->emoji.isEmpty()) { set->count = data->vcount().v; set->hash = data->vhash().v; set->flags |= SetFlag::NotLoaded; // need to request this set } } setsOrder.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(); bool installed = (set->flags & SetFlag::Installed); bool featured = (set->flags & SetFlag::Featured); bool special = (set->flags & SetFlag::Special); bool archived = (set->flags & SetFlag::Archived); if (installed || featured || special || archived) { if (featured && (set->flags & SetFlag::Unread)) { ++unreadCount; } ++it; } else { it = sets.erase(it); } } setFeaturedSetsUnreadCount(unreadCount); const auto counted = 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(); } session().local().writeFeaturedStickers(); notifyUpdated(); } 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( not_null emoji, uint64 seed) { const auto original = emoji->original(); 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 InstallDate = [&](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(); auto i = recent->emoji.constFind(original); if (i != recent->emoji.cend()) { result.reserve(i->size()); for (const auto document : *i) { const auto usageDate = [&] { if (recent->dates.empty()) { return TimeId(0); } const auto index = recent->stickers.indexOf(document); if (index < 0) { return TimeId(0); } Assert(index < recent->dates.size()); return recent->dates[index]; }(); const auto date = usageDate ? usageDate : InstallDate(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.isEmpty()) { setsToRequest.emplace(set->id, set->accessHash); set->flags |= SetFlag::NotLoaded; continue; } auto i = set->emoji.constFind(original); if (i == set->emoji.cend()) { continue; } const auto my = (set->flags & SetFlag::Installed); result.reserve(result.size() + i->size()); for (const auto document : *i) { 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 (Core::App().settings().suggestStickersByEmoji()) { const auto others = session().api().stickersByEmoji(original); if (!others) { return {}; } result.reserve(result.size() + others->size()); for (const auto document : *others) { add(document, CreateOtherSortKey(document)); } } ranges::actions::sort( result, std::greater<>(), &StickerWithDate::date); return ranges::views::all( result ) | ranges::views::transform([](const StickerWithDate &data) { return data.document; }) | ranges::to_vector; } 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->contains(document)) { result.emplace_back(i.key()); } } if (result.empty()) { return std::nullopt; } return result; } return std::nullopt; } StickersSet *Stickers::feedSet(const MTPDstickerSet &data) { auto &sets = setsRef(); 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; it->second->setThumbnail(thumbnail); } 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); it->second->setThumbnail(thumbnail); if (set->count != data.vcount().v || set->hash != data.vhash().v || set->emoji.isEmpty()) { // 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(); auto changedFlags = (oldFlags ^ set->flags); if (changedFlags & SetFlag::Archived) { const auto masks = !!(set->flags & SetFlag::Masks); auto &archivedOrder = masks ? 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(); } StickersSet *Stickers::feedSetFull(const MTPDmessages_stickerSet &d) { Expects(d.vset().type() == mtpc_stickerSet); const auto &s = d.vset().c_stickerSet(); auto &sets = setsRef(); const auto wasArchived = [&] { auto it = sets.find(s.vid().v); return (it != sets.end()) && (it->second->flags & SetFlag::Archived); }(); auto set = feedSet(s); set->flags &= ~SetFlag::NotLoaded; const auto &d_docs = d.vdocuments().v; auto customIt = sets.find(Stickers::CustomSetId); const auto inputSet = set->identifier(); auto pack = StickersPack(); pack.reserve(d_docs.size()); for (const auto &item : d_docs) { 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 isMasks = !!(set->flags & SetFlag::Masks); if (pack.isEmpty()) { const auto removeIndex = (isMasks ? maskSetsOrder() : setsOrder()).indexOf(set->id); if (removeIndex >= 0) { (isMasks ? maskSetsOrderRef() : setsOrderRef()).removeAt(removeIndex); } sets.remove(set->id); set = nullptr; } else { set->stickers = pack; set->emoji.clear(); auto &v = d.vpacks().v; for (auto i = 0, l = int(v.size()); i != l; ++i) { if (v[i].type() != mtpc_stickerPack) continue; auto &pack = v[i].c_stickerPack(); if (auto emoji = Ui::Emoji::Find(qs(pack.vemoticon()))) { emoji = emoji->original(); auto &stickers = pack.vdocuments().v; StickersPack p; p.reserve(stickers.size()); for (auto j = 0, c = int(stickers.size()); j != c; ++j) { auto doc = owner().document(stickers[j].v); if (!doc || !doc->sticker()) continue; p.push_back(doc); } set->emoji.insert(emoji, p); } } } if (writeRecent) { session().saveSettings(); } if (set) { const auto isArchived = !!(set->flags & SetFlag::Archived); if (isMasks) { session().local().writeInstalledMasks(); } else if (set->flags & SetFlag::Installed) { if (!isArchived) { session().local().writeInstalledStickers(); } } if (set->flags & SetFlag::Featured) { session().local().writeFeaturedStickers(); } if (wasArchived != isArchived) { if (isMasks) { session().local().writeArchivedMasks(); } else { session().local().writeArchivedStickers(); } } } notifyUpdated(); return set; } 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_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(qstr("Great Minds"), 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