tdesktop/Telegram/SourceFiles/chat_helpers/stickers_dice_pack.cpp

139 lines
3.4 KiB
C++
Raw Normal View History

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"
#include "storage/localimageloader.h"
#include "base/unixtime.h"
#include "apiwrap.h"
#include <QtCore/QFile>
#include <QtCore/QFileInfo>
namespace Stickers {
namespace {
constexpr auto kZeroDiceDocumentId = 0xa3b83c9f84fa9e83ULL;
} // namespace
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();
}
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(
MTP_inputStickerSetDice(MTP_string(_emoji))
2020-03-04 09:21:19 +00:00
)).done([=](const MTPmessages_StickerSet &result) {
result.match([&](const MTPDmessages_stickerSet &data) {
applySet(data);
});
}).fail([=](const RPCError &error) {
_requestId = 0;
}).send();
}
void DicePack::applySet(const MTPDmessages_stickerSet &data) {
2020-04-19 12:17:53 +00:00
_map.clear();
auto documents = base::flat_map<DocumentId, not_null<DocumentData*>>();
2020-03-04 09:21:19 +00:00
for (const auto &sticker : data.vdocuments().v) {
const auto document = _session->data().processDocument(
sticker);
if (document->sticker()) {
2020-04-19 12:17:53 +00:00
documents.emplace(document->id, document);
2020-03-04 09:21:19 +00:00
}
}
2020-04-19 12:17:53 +00:00
for (const auto pack : data.vpacks().v) {
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;
}
for (const auto id : data.vdocuments().v) {
if (const auto document = documents.take(id.v)) {
_map.emplace(index, *document);
}
}
});
}
2020-03-04 09:21:19 +00:00
}
void DicePack::tryGenerateLocalZero() {
if (!_map.empty()) {
2020-03-04 09:21:19 +00:00
return;
}
static const auto kDiceString = QString::fromUtf8("\xF0\x9F\x8E\xB2");
static const auto kDartString = QString::fromUtf8("\xF0\x9F\x8E\xAF");
const auto path = (_emoji == kDiceString)
? qsl(":/gui/art/dice_idle.tgs")
: (_emoji == kDartString)
? qsl(":/gui/art/dart_idle.tgs")
: QString();
if (path.isEmpty()) {
return;
}
2020-03-04 09:21:19 +00:00
auto task = FileLoadTask(
path,
QByteArray(),
nullptr,
SendMediaType::File,
FileLoadTo(0, {}, 0),
{});
task.process();
const auto result = task.peekResult();
Assert(result != nullptr);
const auto document = _session->data().processDocument(
2020-03-04 09:21:19 +00:00
result->document,
std::move(result->thumb));
document->setLocation(FileLocation(path));
_map.emplace(0, document);
2020-03-04 09:21:19 +00:00
Ensures(document->sticker());
Ensures(document->sticker()->animated);
2020-03-04 09:21:19 +00:00
}
DicePacks::DicePacks(not_null<Main::Session*> session) : _session(session) {
}
DocumentData *DicePacks::lookup(const QString &emoji, int value) {
const auto i = _packs.find(emoji);
if (i != end(_packs)) {
return i->second->lookup(value);
}
return _packs.emplace(
emoji,
std::make_unique<DicePack>(_session, emoji)
).first->second->lookup(value);
}
2020-03-04 09:21:19 +00:00
} // namespace Stickers