tdesktop/Telegram/SourceFiles/api/api_transcribes.cpp

119 lines
3.2 KiB
C++

/*
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 "api/api_transcribes.h"
#include "history/history_item.h"
#include "history/history.h"
#include "main/main_session.h"
#include "data/data_document.h"
#include "data/data_session.h"
#include "data/data_peer.h"
#include "apiwrap.h"
namespace Api {
Transcribes::Transcribes(not_null<ApiWrap*> api)
: _session(&api->session())
, _api(&api->instance()) {
}
void Transcribes::toggle(not_null<HistoryItem*> item) {
const auto id = item->fullId();
auto i = _map.find(id);
if (i == _map.end()) {
load(item);
//_session->data().requestItemRepaint(item);
_session->data().requestItemResize(item);
} else if (!i->second.requestId) {
i->second.shown = !i->second.shown;
if (i->second.roundview) {
_session->data().requestItemViewRefresh(item);
}
_session->data().requestItemResize(item);
}
}
const Transcribes::Entry &Transcribes::entry(
not_null<HistoryItem*> item) const {
static auto empty = Entry();
const auto i = _map.find(item->fullId());
return (i != _map.end()) ? i->second : empty;
}
void Transcribes::apply(const MTPDupdateTranscribedAudio &update) {
const auto id = update.vtranscription_id().v;
const auto i = _ids.find(id);
if (i == _ids.end()) {
return;
}
const auto j = _map.find(i->second);
if (j == _map.end()) {
return;
}
const auto text = qs(update.vtext());
j->second.result = text;
j->second.pending = update.is_pending();
if (const auto item = _session->data().message(i->second)) {
if (j->second.roundview) {
_session->data().requestItemViewRefresh(item);
}
_session->data().requestItemResize(item);
}
}
void Transcribes::load(not_null<HistoryItem*> item) {
if (!item->isHistoryEntry() || item->isLocal()) {
return;
}
const auto toggleRound = [](not_null<HistoryItem*> item, Entry &entry) {
if (const auto media = item->media()) {
if (const auto document = media->document()) {
if (document->isVideoMessage()) {
entry.roundview = true;
document->owner().requestItemViewRefresh(item);
}
}
}
};
const auto id = item->fullId();
const auto requestId = _api.request(MTPmessages_TranscribeAudio(
item->history()->peer->input,
MTP_int(item->id)
)).done([=](const MTPmessages_TranscribedAudio &result) {
const auto &data = result.data();
auto &entry = _map[id];
entry.requestId = 0;
entry.pending = data.is_pending();
entry.result = qs(data.vtext());
_ids.emplace(data.vtranscription_id().v, id);
if (const auto item = _session->data().message(id)) {
toggleRound(item, entry);
_session->data().requestItemResize(item);
}
}).fail([=](const MTP::Error &error) {
auto &entry = _map[id];
entry.requestId = 0;
entry.pending = false;
entry.failed = true;
if (error.type() == u"MSG_VOICE_TOO_LONG"_q) {
entry.toolong = true;
}
if (const auto item = _session->data().message(id)) {
toggleRound(item, entry);
_session->data().requestItemResize(item);
}
}).send();
auto &entry = _map.emplace(id).first->second;
entry.requestId = requestId;
entry.shown = true;
entry.failed = false;
entry.pending = false;
}
} // namespace Api