2020-03-04 09:21:19 +00:00
|
|
|
/*
|
|
|
|
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 "chat_helpers/stickers_dice_pack.h"
|
|
|
|
|
|
|
|
#include "main/main_session.h"
|
|
|
|
#include "data/data_session.h"
|
|
|
|
#include "data/data_document.h"
|
2022-11-10 12:18:42 +00:00
|
|
|
#include "ui/chat/attach/attach_prepare.h"
|
2020-04-16 12:37:26 +00:00
|
|
|
#include "ui/image/image_location_factory.h"
|
2020-03-04 09:21:19 +00:00
|
|
|
#include "storage/localimageloader.h"
|
|
|
|
#include "base/unixtime.h"
|
|
|
|
#include "apiwrap.h"
|
|
|
|
|
|
|
|
#include <QtCore/QFile>
|
|
|
|
#include <QtCore/QFileInfo>
|
|
|
|
|
|
|
|
namespace Stickers {
|
|
|
|
|
2020-10-14 13:01:07 +00:00
|
|
|
const QString DicePacks::kDiceString = QString::fromUtf8("\xF0\x9F\x8E\xB2");
|
|
|
|
const QString DicePacks::kDartString = QString::fromUtf8("\xF0\x9F\x8E\xAF");
|
|
|
|
const QString DicePacks::kSlotString = QString::fromUtf8("\xF0\x9F\x8E\xB0");
|
2020-10-14 15:12:24 +00:00
|
|
|
const QString DicePacks::kFballString = QString::fromUtf8("\xE2\x9A\xBD");
|
|
|
|
const QString DicePacks::kBballString = QString::fromUtf8("\xF0\x9F\x8F\x80");
|
2020-10-14 13:01:07 +00:00
|
|
|
|
2020-04-06 09:07:49 +00:00
|
|
|
DicePack::DicePack(not_null<Main::Session*> session, const QString &emoji)
|
|
|
|
: _session(session)
|
|
|
|
, _emoji(emoji) {
|
2020-03-04 09:21:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
DicePack::~DicePack() = default;
|
|
|
|
|
|
|
|
DocumentData *DicePack::lookup(int value) {
|
|
|
|
if (!_requestId) {
|
|
|
|
load();
|
|
|
|
}
|
2020-04-20 09:51:17 +00:00
|
|
|
tryGenerateLocalZero();
|
2020-03-04 09:21:19 +00:00
|
|
|
const auto i = _map.find(value);
|
|
|
|
return (i != end(_map)) ? i->second.get() : nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
void DicePack::load() {
|
|
|
|
if (_requestId) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
_requestId = _session->api().request(MTPmessages_GetStickerSet(
|
2021-11-19 19:04:40 +00:00
|
|
|
MTP_inputStickerSetDice(MTP_string(_emoji)),
|
|
|
|
MTP_int(0) // hash
|
2020-03-04 09:21:19 +00:00
|
|
|
)).done([=](const MTPmessages_StickerSet &result) {
|
|
|
|
result.match([&](const MTPDmessages_stickerSet &data) {
|
|
|
|
applySet(data);
|
2021-11-19 19:04:40 +00:00
|
|
|
}, [](const MTPDmessages_stickerSetNotModified &) {
|
|
|
|
LOG(("API Error: Unexpected messages.stickerSetNotModified."));
|
2020-03-04 09:21:19 +00:00
|
|
|
});
|
2021-11-26 20:46:53 +00:00
|
|
|
}).fail([=] {
|
2020-03-04 09:21:19 +00:00
|
|
|
_requestId = 0;
|
|
|
|
}).send();
|
|
|
|
}
|
|
|
|
|
|
|
|
void DicePack::applySet(const MTPDmessages_stickerSet &data) {
|
2020-10-14 13:01:07 +00:00
|
|
|
const auto isSlotMachine = DicePacks::IsSlot(_emoji);
|
2020-10-14 15:12:24 +00:00
|
|
|
auto index = 0;
|
|
|
|
auto documents = base::flat_map<DocumentId, not_null<DocumentData*>>();
|
2020-03-04 09:21:19 +00:00
|
|
|
for (const auto &sticker : data.vdocuments().v) {
|
2022-07-13 17:12:40 +00:00
|
|
|
const auto document = _session->data().processDocument(sticker);
|
2020-03-04 09:21:19 +00:00
|
|
|
if (document->sticker()) {
|
2020-10-14 13:01:07 +00:00
|
|
|
if (isSlotMachine) {
|
2020-10-14 15:12:24 +00:00
|
|
|
_map.emplace(index++, document);
|
2020-10-14 13:01:07 +00:00
|
|
|
} else {
|
|
|
|
documents.emplace(document->id, document);
|
|
|
|
}
|
2020-03-04 09:21:19 +00:00
|
|
|
}
|
|
|
|
}
|
2020-10-14 13:01:07 +00:00
|
|
|
if (isSlotMachine) {
|
|
|
|
return;
|
|
|
|
}
|
2021-03-13 08:26:58 +00:00
|
|
|
for (const auto &pack : data.vpacks().v) {
|
2020-04-19 12:17:53 +00:00
|
|
|
pack.match([&](const MTPDstickerPack &data) {
|
|
|
|
const auto emoji = qs(data.vemoticon());
|
|
|
|
if (emoji.isEmpty()) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
const auto ch = int(emoji[0].unicode());
|
|
|
|
const auto index = (ch == '#') ? 0 : (ch + 1 - '1');
|
|
|
|
if (index < 0 || index > 6) {
|
|
|
|
return;
|
|
|
|
}
|
2021-09-08 10:53:54 +00:00
|
|
|
for (const auto &id : data.vdocuments().v) {
|
2020-04-19 12:17:53 +00:00
|
|
|
if (const auto document = documents.take(id.v)) {
|
|
|
|
_map.emplace(index, *document);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
2020-03-04 09:21:19 +00:00
|
|
|
}
|
|
|
|
|
2020-04-20 09:51:17 +00:00
|
|
|
void DicePack::tryGenerateLocalZero() {
|
|
|
|
if (!_map.empty()) {
|
2020-03-04 09:21:19 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2020-10-14 15:12:24 +00:00
|
|
|
if (_emoji == DicePacks::kDiceString) {
|
|
|
|
generateLocal(0, u"dice_idle"_q);
|
|
|
|
} else if (_emoji == DicePacks::kDartString) {
|
|
|
|
generateLocal(0, u"dart_idle"_q);
|
|
|
|
} else if (_emoji == DicePacks::kBballString) {
|
|
|
|
generateLocal(0, u"bball_idle"_q);
|
|
|
|
} else if (_emoji == DicePacks::kFballString) {
|
|
|
|
generateLocal(0, u"fball_idle"_q);
|
|
|
|
} else if (_emoji == DicePacks::kSlotString) {
|
|
|
|
generateLocal(0, u"slot_back"_q);
|
|
|
|
generateLocal(2, u"slot_pull"_q);
|
|
|
|
generateLocal(8, u"slot_0_idle"_q);
|
|
|
|
generateLocal(14, u"slot_1_idle"_q);
|
|
|
|
generateLocal(20, u"slot_2_idle"_q);
|
2020-04-20 09:51:17 +00:00
|
|
|
}
|
2020-10-14 15:12:24 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void DicePack::generateLocal(int index, const QString &name) {
|
|
|
|
const auto path = u":/gui/art/"_q + name + u".tgs"_q;
|
2020-03-04 09:21:19 +00:00
|
|
|
auto task = FileLoadTask(
|
2020-06-12 14:09:04 +00:00
|
|
|
_session,
|
2020-03-04 09:21:19 +00:00
|
|
|
path,
|
|
|
|
QByteArray(),
|
|
|
|
nullptr,
|
|
|
|
SendMediaType::File,
|
2022-10-17 16:29:48 +00:00
|
|
|
FileLoadTo(0, {}, 0, 0, 0),
|
2020-03-04 09:21:19 +00:00
|
|
|
{});
|
2020-10-14 15:12:24 +00:00
|
|
|
task.process({ .generateGoodThumbnail = false });
|
2020-03-04 09:21:19 +00:00
|
|
|
const auto result = task.peekResult();
|
|
|
|
Assert(result != nullptr);
|
2020-04-20 09:51:17 +00:00
|
|
|
const auto document = _session->data().processDocument(
|
2020-03-04 09:21:19 +00:00
|
|
|
result->document,
|
2020-10-14 15:12:24 +00:00
|
|
|
Images::FromImageInMemory(result->thumb, "WEBP", result->thumbbytes));
|
2020-10-13 16:43:18 +00:00
|
|
|
document->setLocation(Core::FileLocation(path));
|
2020-04-20 09:51:17 +00:00
|
|
|
|
2020-10-14 15:12:24 +00:00
|
|
|
_map.emplace(index, document);
|
2020-03-04 09:21:19 +00:00
|
|
|
|
2020-04-20 09:51:17 +00:00
|
|
|
Ensures(document->sticker());
|
2022-01-19 14:45:51 +00:00
|
|
|
Ensures(document->sticker()->isLottie());
|
2020-03-04 09:21:19 +00:00
|
|
|
}
|
|
|
|
|
2020-04-06 09:07:49 +00:00
|
|
|
DicePacks::DicePacks(not_null<Main::Session*> session) : _session(session) {
|
|
|
|
}
|
|
|
|
|
|
|
|
DocumentData *DicePacks::lookup(const QString &emoji, int value) {
|
2020-10-14 15:12:24 +00:00
|
|
|
const auto key = emoji.endsWith(QChar(0xFE0F))
|
|
|
|
? emoji.mid(0, emoji.size() - 1)
|
|
|
|
: emoji;
|
|
|
|
const auto i = _packs.find(key);
|
2020-04-06 09:07:49 +00:00
|
|
|
if (i != end(_packs)) {
|
|
|
|
return i->second->lookup(value);
|
|
|
|
}
|
|
|
|
return _packs.emplace(
|
2020-10-14 15:12:24 +00:00
|
|
|
key,
|
|
|
|
std::make_unique<DicePack>(_session, key)
|
2020-04-06 09:07:49 +00:00
|
|
|
).first->second->lookup(value);
|
|
|
|
}
|
|
|
|
|
2020-03-04 09:21:19 +00:00
|
|
|
} // namespace Stickers
|