andreacavalli/tdlib-fork
Archived
1
0
Fork 0
Code Pull Requests Releases Activity

Update layer #76.

Browse Source 
GitOrigin-RevId: d865abc2c8fb9b3b1350fcd689ef086de38aab3e
This commit is contained in:
levlam 2018-03-04 00:33:26 +03:00
parent b5036975b8
commit d64280c890
11 changed files with 410 additions and 218 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
td/generate/scheme/telegram_api.tl
View File

@ -580,7 +580,7 @@ inputBotInlineMessageMediaVenue#aaafadc8 flags:# geo_point:InputGeoPoint title:s
inputBotInlineMessageMediaContact#2daf01a7 flags:# phone_number:string first_name:string last_name:string reply_markup:flags.2?ReplyMarkup = InputBotInlineMessage; inputBotInlineMessageMediaContact#2daf01a7 flags:# phone_number:string first_name:string last_name:string reply_markup:flags.2?ReplyMarkup = InputBotInlineMessage;
inputBotInlineMessageGame#4b425864 flags:# reply_markup:flags.2?ReplyMarkup = InputBotInlineMessage; inputBotInlineMessageGame#4b425864 flags:# reply_markup:flags.2?ReplyMarkup = InputBotInlineMessage;
inputBotInlineResult#2cbbe15a flags:# id:string type:string title:flags.1?string description:flags.2?string url:flags.3?string thumb_url:flags.4?string content_url:flags.5?string content_type:flags.5?string w:flags.6?int h:flags.6?int duration:flags.7?int send_message:InputBotInlineMessage = InputBotInlineResult; inputBotInlineResult#88bf9319 flags:# id:string type:string title:flags.1?string description:flags.2?string url:flags.3?string thumb:flags.4?InputWebDocument content:flags.5?InputWebDocument send_message:InputBotInlineMessage = InputBotInlineResult;
inputBotInlineResultPhoto#a8d864a7 id:string type:string photo:InputPhoto send_message:InputBotInlineMessage = InputBotInlineResult; inputBotInlineResultPhoto#a8d864a7 id:string type:string photo:InputPhoto send_message:InputBotInlineMessage = InputBotInlineResult;
inputBotInlineResultDocument#fff8fdc4 flags:# id:string type:string title:flags.1?string description:flags.2?string document:InputDocument send_message:InputBotInlineMessage = InputBotInlineResult; inputBotInlineResultDocument#fff8fdc4 flags:# id:string type:string title:flags.1?string description:flags.2?string document:InputDocument send_message:InputBotInlineMessage = InputBotInlineResult;
inputBotInlineResultGame#4fa417f2 id:string short_name:string send_message:InputBotInlineMessage = InputBotInlineResult; inputBotInlineResultGame#4fa417f2 id:string short_name:string send_message:InputBotInlineMessage = InputBotInlineResult;
@ -591,7 +591,7 @@ botInlineMessageMediaGeo#b722de65 flags:# geo:GeoPoint period:int reply_markup:f
botInlineMessageMediaVenue#4366232e flags:# geo:GeoPoint title:string address:string provider:string venue_id:string reply_markup:flags.2?ReplyMarkup = BotInlineMessage; botInlineMessageMediaVenue#4366232e flags:# geo:GeoPoint title:string address:string provider:string venue_id:string reply_markup:flags.2?ReplyMarkup = BotInlineMessage;
botInlineMessageMediaContact#35edb4d4 flags:# phone_number:string first_name:string last_name:string reply_markup:flags.2?ReplyMarkup = BotInlineMessage; botInlineMessageMediaContact#35edb4d4 flags:# phone_number:string first_name:string last_name:string reply_markup:flags.2?ReplyMarkup = BotInlineMessage;
botInlineResult#9bebaeb9 flags:# id:string type:string title:flags.1?string description:flags.2?string url:flags.3?string thumb_url:flags.4?string content_url:flags.5?string content_type:flags.5?string w:flags.6?int h:flags.6?int duration:flags.7?int send_message:BotInlineMessage = BotInlineResult; botInlineResult#11965f3a flags:# id:string type:string title:flags.1?string description:flags.2?string url:flags.3?string thumb:flags.4?WebDocument content:flags.5?WebDocument send_message:BotInlineMessage = BotInlineResult;
botInlineMediaResult#17db940b flags:# id:string type:string photo:flags.0?Photo document:flags.1?Document title:flags.2?string description:flags.3?string send_message:BotInlineMessage = BotInlineResult; botInlineMediaResult#17db940b flags:# id:string type:string photo:flags.0?Photo document:flags.1?Document title:flags.2?string description:flags.3?string send_message:BotInlineMessage = BotInlineResult;
messages.botResults#947ca848 flags:# gallery:flags.0?true query_id:long next_offset:flags.1?string switch_pm:flags.2?InlineBotSwitchPM results:Vector<BotInlineResult> cache_time:int users:Vector<User> = messages.BotResults; messages.botResults#947ca848 flags:# gallery:flags.0?true query_id:long next_offset:flags.1?string switch_pm:flags.2?InlineBotSwitchPM results:Vector<BotInlineResult> cache_time:int users:Vector<User> = messages.BotResults;
@ -722,6 +722,7 @@ paymentRequestedInfo#909c3f94 flags:# name:flags.0?string phone:flags.1?string e
paymentSavedCredentialsCard#cdc27a1f id:string title:string = PaymentSavedCredentials; paymentSavedCredentialsCard#cdc27a1f id:string title:string = PaymentSavedCredentials;
webDocument#c61acbd8 url:string access_hash:long size:int mime_type:string attributes:Vector<DocumentAttribute> dc_id:int = WebDocument; webDocument#c61acbd8 url:string access_hash:long size:int mime_type:string attributes:Vector<DocumentAttribute> dc_id:int = WebDocument;
webDocumentNoProxy#f9c8bcc6 url:string size:int mime_type:string attributes:Vector<DocumentAttribute> = WebDocument;
inputWebDocument#9bed434d url:string size:int mime_type:string attributes:Vector<DocumentAttribute> = InputWebDocument; inputWebDocument#9bed434d url:string size:int mime_type:string attributes:Vector<DocumentAttribute> = InputWebDocument;
@ -866,7 +867,7 @@ auth.resendCode#3ef1a9bf phone_number:string phone_code_hash:string = auth.SentC
auth.cancelCode#1f040578 phone_number:string phone_code_hash:string = Bool; auth.cancelCode#1f040578 phone_number:string phone_code_hash:string = Bool;
auth.dropTempAuthKeys#8e48a188 except_auth_keys:Vector<long> = Bool; auth.dropTempAuthKeys#8e48a188 except_auth_keys:Vector<long> = Bool;
account.registerDevice#1389cc token_type:int token:string app_sandbox:Bool other_uids:Vector<int> = Bool; account.registerDevice#5cbea590 token_type:int token:string app_sandbox:Bool secret:bytes other_uids:Vector<int> = Bool;
account.unregisterDevice#3076c4bf token_type:int token:string other_uids:Vector<int> = Bool; account.unregisterDevice#3076c4bf token_type:int token:string other_uids:Vector<int> = Bool;
account.updateNotifySettings#84be5b93 peer:InputNotifyPeer settings:InputPeerNotifySettings = Bool; account.updateNotifySettings#84be5b93 peer:InputNotifyPeer settings:InputPeerNotifySettings = Bool;
account.getNotifySettings#12b3ad31 peer:InputNotifyPeer = PeerNotifySettings; account.getNotifySettings#12b3ad31 peer:InputNotifyPeer = PeerNotifySettings;

BIN
td/generate/scheme/telegram_api.tlo
View File

Binary file not shown.

2
td/telegram/DeviceTokenManager.cpp
View File

@ -328,7 +328,7 @@ void DeviceTokenManager::loop() {
create_storer(telegram_api::account_unregisterDevice(token_type, info.token, std::move(other_user_ids)))); create_storer(telegram_api::account_unregisterDevice(token_type, info.token, std::move(other_user_ids))));
} else { } else {
net_query = G()->net_query_creator().create(create_storer(telegram_api::account_registerDevice( net_query = G()->net_query_creator().create(create_storer(telegram_api::account_registerDevice(
token_type, info.token, info.is_app_sandbox, std::move(other_user_ids)))); token_type, info.token, info.is_app_sandbox, BufferSlice(), std::move(other_user_ids))));
} }
info.net_query_id = net_query->id(); info.net_query_id = net_query->id();
G()->net_query_dispatcher().dispatch_with_callback(std::move(net_query), actor_shared(this, token_type)); G()->net_query_dispatcher().dispatch_with_callback(std::move(net_query), actor_shared(this, token_type));

125
td/telegram/DocumentsManager.cpp
View File

@ -23,10 +23,12 @@
#include "td/utils/common.h" #include "td/utils/common.h"
#include "td/utils/format.h" #include "td/utils/format.h"
#include "td/utils/HttpUrl.h"
#include "td/utils/logging.h" #include "td/utils/logging.h"
#include "td/utils/MimeType.h" #include "td/utils/MimeType.h"
#include "td/utils/misc.h" #include "td/utils/misc.h"
#include "td/utils/PathView.h" #include "td/utils/PathView.h"
#include "td/utils/Random.h"
#include "td/utils/Slice.h" #include "td/utils/Slice.h"
#include "td/utils/StackAllocator.h" #include "td/utils/StackAllocator.h"
#include "td/utils/Status.h" #include "td/utils/Status.h"
@ -34,6 +36,7 @@
#include "td/utils/utf8.h" #include "td/utils/utf8.h"
#include <cstdlib> #include <cstdlib>
#include <limits>
namespace td { namespace td {
@ -55,7 +58,8 @@ tl_object_ptr<td_api::document> DocumentsManager::get_document_object(FileId fil
} }
std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_document( std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_document(
RemoteDocument remote_document, DialogId owner_dialog_id, MultiPromiseActor *load_data_multipromise_ptr) { RemoteDocument remote_document, DialogId owner_dialog_id, MultiPromiseActor *load_data_multipromise_ptr,
DocumentType default_document_type) {
tl_object_ptr<telegram_api::documentAttributeAnimated> animated; tl_object_ptr<telegram_api::documentAttributeAnimated> animated;
tl_object_ptr<telegram_api::documentAttributeVideo> video; tl_object_ptr<telegram_api::documentAttributeVideo> video;
tl_object_ptr<telegram_api::documentAttributeAudio> audio; tl_object_ptr<telegram_api::documentAttributeAudio> audio;
@ -121,17 +125,22 @@ std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_docum
sticker = nullptr; sticker = nullptr;
} }
auto document_type = DocumentType::General; auto document_type = default_document_type;
FileType file_type = FileType::Document; FileType file_type = FileType::Document;
Slice default_extension; Slice default_extension;
bool supports_streaming = false; bool supports_streaming = false;
if (type_attributes == 1) { // not a general document if (type_attributes == 1 || default_document_type != DocumentType::General) { // not a general document
if (animated != nullptr) { if (animated != nullptr || default_document_type == DocumentType::Animation) {
document_type = DocumentType::Animation; document_type = DocumentType::Animation;
file_type = FileType::Animation; file_type = FileType::Animation;
default_extension = "mp4"; default_extension = "mp4";
} else if (audio != nullptr) { } else if (audio != nullptr || default_document_type == DocumentType::Audio ||
if ((audio->flags_ & telegram_api::documentAttributeAudio::Flags::VOICE_MASK) != 0) { default_document_type == DocumentType::VoiceNote) {
bool is_voice_note = default_document_type == DocumentType::VoiceNote;
if (audio != nullptr) {
is_voice_note = (audio->flags_ & telegram_api::documentAttributeAudio::Flags::VOICE_MASK) != 0;
}
if (is_voice_note) {
document_type = DocumentType::VoiceNote; document_type = DocumentType::VoiceNote;
file_type = FileType::VoiceNote; file_type = FileType::VoiceNote;
default_extension = "oga"; default_extension = "oga";
@ -141,21 +150,28 @@ std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_docum
file_type = FileType::Audio; file_type = FileType::Audio;
default_extension = "mp3"; default_extension = "mp3";
} }
} else if (sticker != nullptr) { } else if (sticker != nullptr || default_document_type == DocumentType::Sticker) {
document_type = DocumentType::Sticker; document_type = DocumentType::Sticker;
file_type = FileType::Sticker; file_type = FileType::Sticker;
default_extension = "webp"; default_extension = "webp";
owner_dialog_id = DialogId(); owner_dialog_id = DialogId();
file_name.clear(); file_name.clear();
} else if (video != nullptr) { } else if (video != nullptr || default_document_type == DocumentType::Video ||
if ((video->flags_ & telegram_api::documentAttributeVideo::ROUND_MESSAGE_MASK) != 0) { default_document_type == DocumentType::VideoNote) {
bool is_video_note = default_document_type == DocumentType::VideoNote;
if (video != nullptr) {
is_video_note = (video->flags_ & telegram_api::documentAttributeVideo::ROUND_MESSAGE_MASK) != 0;
if (!is_video_note) {
supports_streaming = (video->flags_ & telegram_api::documentAttributeVideo::SUPPORTS_STREAMING_MASK) != 0;
}
}
if (is_video_note) {
document_type = DocumentType::VideoNote; document_type = DocumentType::VideoNote;
file_type = FileType::VideoNote; file_type = FileType::VideoNote;
file_name.clear(); file_name.clear();
} else { } else {
document_type = DocumentType::Video; document_type = DocumentType::Video;
file_type = FileType::Video; file_type = FileType::Video;
supports_streaming = (video->flags_ & telegram_api::documentAttributeVideo::SUPPORTS_STREAMING_MASK) != 0;
} }
default_extension = "mp4"; default_extension = "mp4";
} }
@ -173,6 +189,9 @@ std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_docum
string mime_type; string mime_type;
PhotoSize thumbnail; PhotoSize thumbnail;
FileEncryptionKey encryption_key; FileEncryptionKey encryption_key;
bool is_web = false;
bool is_web_no_proxy = false;
string url;
if (remote_document.document != nullptr) { if (remote_document.document != nullptr) {
auto document = std::move(remote_document.document); auto document = std::move(remote_document.document);
@ -186,8 +205,7 @@ std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_docum
thumbnail = get_photo_size(td_->file_manager_.get(), FileType::Thumbnail, 0, 0, owner_dialog_id, thumbnail = get_photo_size(td_->file_manager_.get(), FileType::Thumbnail, 0, 0, owner_dialog_id,
std::move(document->thumb_)); std::move(document->thumb_));
} }
} else { } else if (remote_document.secret_file != nullptr) {
CHECK(remote_document.secret_file != nullptr);
CHECK(remote_document.secret_document != nullptr); CHECK(remote_document.secret_document != nullptr);
auto file = std::move(remote_document.secret_file); auto file = std::move(remote_document.secret_file);
auto document = std::move(remote_document.secret_document); auto document = std::move(remote_document.secret_document);
@ -207,10 +225,54 @@ std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_docum
thumbnail = get_thumbnail_photo_size(td_->file_manager_.get(), std::move(document->thumb_), owner_dialog_id, thumbnail = get_thumbnail_photo_size(td_->file_manager_.get(), std::move(document->thumb_), owner_dialog_id,
document->thumb_w_, document->thumb_h_); document->thumb_w_, document->thumb_h_);
} }
} else {
is_web = true;
id = Random::fast(0, std::numeric_limits<int32>::max());
dc_id = 0;
access_hash = 0;
thumbnail = std::move(remote_document.thumbnail);
auto web_document_ptr = std::move(remote_document.web_document);
switch (web_document_ptr->get_id()) {
case telegram_api::webDocument::ID: {
auto web_document = move_tl_object_as<telegram_api::webDocument>(web_document_ptr);
auto r_http_url = parse_url(web_document->url_);
if (r_http_url.is_error()) {
LOG(ERROR) << "Can't parse URL " << web_document->url_;
return {DocumentType::Unknown, FileId()};
}
auto http_url = r_http_url.move_as_ok();
dc_id = web_document->dc_id_;
access_hash = web_document->access_hash_;
url = http_url.get_url();
file_name = get_url_query_file_name(http_url.query_);
mime_type = std::move(web_document->mime_type_);
size = web_document->size_;
break;
}
case telegram_api::webDocumentNoProxy::ID: {
is_web_no_proxy = true;
auto web_document = move_tl_object_as<telegram_api::webDocumentNoProxy>(web_document_ptr);
if (web_document->url_.find('.') == string::npos) {
LOG(ERROR) << "Receive invalid URL " << web_document->url_;
return {DocumentType::Unknown, FileId()};
}
url = std::move(web_document->url_);
file_name = get_url_file_name(url);
mime_type = std::move(web_document->mime_type_);
size = web_document->size_;
break;
}
default:
UNREACHABLE();
}
} }
LOG(DEBUG) << "Receive document with id = " << id << " of type " << static_cast<int32>(document_type); LOG(DEBUG) << "Receive document with id = " << id << " of type " << static_cast<int32>(document_type);
if (!DcId::is_valid(dc_id)) { if (!is_web_no_proxy && !DcId::is_valid(dc_id)) {
LOG(ERROR) << "Wrong dc_id = " << dc_id; LOG(ERROR) << "Wrong dc_id = " << dc_id;
return {DocumentType::Unknown, FileId()}; return {DocumentType::Unknown, FileId()};
} }
@ -224,11 +286,26 @@ std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_docum
suggested_file_name += extension; suggested_file_name += extension;
} }
} }
FileId file_id = td_->file_manager_->register_remote(
FullRemoteFileLocation(file_type, id, access_hash, DcId::internal(dc_id)), FileLocationSource::FromServer, FileId file_id;
owner_dialog_id, size, 0, suggested_file_name); if (!is_web) {
if (!encryption_key.empty()) { file_id = td_->file_manager_->register_remote(
td_->file_manager_->set_encryption_key(file_id, std::move(encryption_key)); FullRemoteFileLocation(file_type, id, access_hash, DcId::internal(dc_id)), FileLocationSource::FromServer,
owner_dialog_id, size, 0, suggested_file_name);
if (!encryption_key.empty()) {
td_->file_manager_->set_encryption_key(file_id, std::move(encryption_key));
}
} else if (!is_web_no_proxy) {
file_id =
td_->file_manager_->register_remote(FullRemoteFileLocation(file_type, url, access_hash, DcId::internal(dc_id)),
FileLocationSource::FromServer, owner_dialog_id, 0, size, file_name);
} else {
auto r_file_id = td_->file_manager_->from_persistent_id(url, file_type);
if (r_file_id.is_error()) {
LOG(ERROR) << "Can't register URL: " << r_file_id.error();
return {DocumentType::Unknown, FileId()};
}
file_id = r_file_id.move_as_ok();
} }
if (dimensions.width != 0 && thumbnail.dimensions.width != 0) { if (dimensions.width != 0 && thumbnail.dimensions.width != 0) {
@ -243,17 +320,17 @@ std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_docum
case DocumentType::Animation: case DocumentType::Animation:
// TODO use has_stickers // TODO use has_stickers
td_->animations_manager_->create_animation(file_id, std::move(thumbnail), std::move(file_name), td_->animations_manager_->create_animation(file_id, std::move(thumbnail), std::move(file_name),
std::move(mime_type), video_duration, dimensions, true); std::move(mime_type), video_duration, dimensions, !is_web);
break; break;
case DocumentType::Audio: case DocumentType::Audio:
CHECK(audio != nullptr); CHECK(audio != nullptr);
td_->audios_manager_->create_audio(file_id, std::move(thumbnail), std::move(file_name), std::move(mime_type), td_->audios_manager_->create_audio(file_id, std::move(thumbnail), std::move(file_name), std::move(mime_type),
audio->duration_, std::move(audio->title_), std::move(audio->performer_), audio->duration_, std::move(audio->title_), std::move(audio->performer_),
true); !is_web);
break; break;
case DocumentType::General: case DocumentType::General:
td_->documents_manager_->create_document(file_id, std::move(thumbnail), std::move(file_name), td_->documents_manager_->create_document(file_id, std::move(thumbnail), std::move(file_name),
std::move(mime_type), true); std::move(mime_type), !is_web);
break; break;
case DocumentType::Sticker: case DocumentType::Sticker:
CHECK(sticker != nullptr); CHECK(sticker != nullptr);
@ -263,15 +340,15 @@ std::pair<DocumentsManager::DocumentType, FileId> DocumentsManager::on_get_docum
case DocumentType::Video: case DocumentType::Video:
td_->videos_manager_->create_video(file_id, std::move(thumbnail), has_stickers, vector<FileId>(), td_->videos_manager_->create_video(file_id, std::move(thumbnail), has_stickers, vector<FileId>(),
std::move(file_name), std::move(mime_type), video_duration, dimensions, std::move(file_name), std::move(mime_type), video_duration, dimensions,
supports_streaming, true); supports_streaming, !is_web);
break; break;
case DocumentType::VideoNote: case DocumentType::VideoNote:
td_->video_notes_manager_->create_video_note(file_id, std::move(thumbnail), video_duration, dimensions, true); td_->video_notes_manager_->create_video_note(file_id, std::move(thumbnail), video_duration, dimensions, !is_web);
break; break;
case DocumentType::VoiceNote: case DocumentType::VoiceNote:
CHECK(audio != nullptr); CHECK(audio != nullptr);
td_->voice_notes_manager_->create_voice_note(file_id, std::move(mime_type), audio->duration_, td_->voice_notes_manager_->create_voice_note(file_id, std::move(mime_type), audio->duration_,
audio->waveform_.as_slice().str(), true); audio->waveform_.as_slice().str(), !is_web);
break; break;
case DocumentType::Unknown: case DocumentType::Unknown:
default: default:

20
td/telegram/DocumentsManager.h
View File

@ -40,6 +40,9 @@ class DocumentsManager {
// or // or
tl_object_ptr<telegram_api::encryptedFile> secret_file; tl_object_ptr<telegram_api::encryptedFile> secret_file;
tl_object_ptr<secret_api::decryptedMessageMediaDocument> secret_document; tl_object_ptr<secret_api::decryptedMessageMediaDocument> secret_document;
// or
tl_object_ptr<telegram_api::WebDocument> web_document;
PhotoSize thumbnail;
vector<tl_object_ptr<telegram_api::DocumentAttribute>> attributes; vector<tl_object_ptr<telegram_api::DocumentAttribute>> attributes;
@ -47,15 +50,29 @@ class DocumentsManager {
: document(std::move(server_document)) : document(std::move(server_document))
, secret_file(nullptr) , secret_file(nullptr)
, secret_document(nullptr) , secret_document(nullptr)
, web_document(nullptr)
, thumbnail()
, attributes(std::move(document->attributes_)) { , attributes(std::move(document->attributes_)) {
} }
RemoteDocument(tl_object_ptr<telegram_api::WebDocument> &&web_document, PhotoSize thumbnail,
vector<tl_object_ptr<telegram_api::DocumentAttribute>> &&attributes)
: document(nullptr)
, secret_file(nullptr)
, secret_document(nullptr)
, web_document(std::move(web_document))
, thumbnail(std::move(thumbnail))
, attributes(std::move(attributes)) {
}
RemoteDocument(tl_object_ptr<telegram_api::encryptedFile> &&secret_file, RemoteDocument(tl_object_ptr<telegram_api::encryptedFile> &&secret_file,
tl_object_ptr<secret_api::decryptedMessageMediaDocument> &&secret_document, tl_object_ptr<secret_api::decryptedMessageMediaDocument> &&secret_document,
vector<tl_object_ptr<telegram_api::DocumentAttribute>> &&attributes) vector<tl_object_ptr<telegram_api::DocumentAttribute>> &&attributes)
: document(nullptr) : document(nullptr)
, secret_file(std::move(secret_file)) , secret_file(std::move(secret_file))
, secret_document(std::move(secret_document)) , secret_document(std::move(secret_document))
, web_document(nullptr)
, thumbnail()
, attributes(std::move(attributes)) { , attributes(std::move(attributes)) {
} }
}; };
@ -63,7 +80,8 @@ class DocumentsManager {
tl_object_ptr<td_api::document> get_document_object(FileId file_id); tl_object_ptr<td_api::document> get_document_object(FileId file_id);
std::pair<DocumentType, FileId> on_get_document(RemoteDocument remote_document, DialogId owner_dialog_id, std::pair<DocumentType, FileId> on_get_document(RemoteDocument remote_document, DialogId owner_dialog_id,
MultiPromiseActor *load_data_multipromise_ptr = nullptr); MultiPromiseActor *load_data_multipromise_ptr = nullptr,
DocumentType default_document_type = DocumentType::General);
void create_document(FileId file_id, PhotoSize thumbnail, string file_name, string mime_type, bool replace); void create_document(FileId file_id, PhotoSize thumbnail, string file_name, string mime_type, bool replace);

316
td/telegram/InlineQueriesManager.cpp
View File

@ -8,6 +8,7 @@
#include "td/telegram/td_api.h" #include "td/telegram/td_api.h"
#include "td/telegram/td_api.hpp" #include "td/telegram/td_api.hpp"
#include "td/telegram/telegram_api.h" #include "td/telegram/telegram_api.h"
#include "td/telegram/telegram_api.hpp"
#include "td/telegram/AccessRights.h" #include "td/telegram/AccessRights.h"
#include "td/telegram/AnimationsManager.h" #include "td/telegram/AnimationsManager.h"
@ -477,6 +478,8 @@ void InlineQueriesManager::answer_inline_query(int64 inline_query_id, bool is_pe
string thumbnail_url; string thumbnail_url;
string content_url; string content_url;
string content_type; string content_type;
int32 thumbnail_width = 0;
int32 thumbnail_height = 0;
int32 width = 0; int32 width = 0;
int32 height = 0; int32 height = 0;
int32 duration = 0; int32 duration = 0;
@ -492,7 +495,7 @@ void InlineQueriesManager::answer_inline_query(int64 inline_query_id, bool is_pe
thumbnail_url = std::move(animated_gif->thumbnail_url_); thumbnail_url = std::move(animated_gif->thumbnail_url_);
content_url = std::move(animated_gif->gif_url_); content_url = std::move(animated_gif->gif_url_);
content_type = "image/gif"; content_type = "image/gif";
duration = animated_gif->gif_duration_; // duration = animated_gif->gif_duration_;
width = animated_gif->gif_width_; width = animated_gif->gif_width_;
height = animated_gif->gif_height_; height = animated_gif->gif_height_;
is_gallery = true; is_gallery = true;
@ -534,8 +537,8 @@ void InlineQueriesManager::answer_inline_query(int64 inline_query_id, bool is_pe
description = std::move(article->description_); description = std::move(article->description_);
thumbnail_url = std::move(article->thumbnail_url_); thumbnail_url = std::move(article->thumbnail_url_);
if (!thumbnail_url.empty()) { if (!thumbnail_url.empty()) {
width = article->thumbnail_width_; thumbnail_width = article->thumbnail_width_;
height = article->thumbnail_height_; thumbnail_height = article->thumbnail_height_;
} }
force_vertical = true; force_vertical = true;
@ -576,8 +579,8 @@ void InlineQueriesManager::answer_inline_query(int64 inline_query_id, bool is_pe
description = std::move(phone_number); description = std::move(phone_number);
thumbnail_url = std::move(contact->thumbnail_url_); thumbnail_url = std::move(contact->thumbnail_url_);
if (!thumbnail_url.empty()) { if (!thumbnail_url.empty()) {
width = contact->thumbnail_width_; thumbnail_width = contact->thumbnail_width_;
height = contact->thumbnail_height_; thumbnail_height = contact->thumbnail_height_;
} }
force_vertical = true; force_vertical = true;
@ -594,8 +597,8 @@ void InlineQueriesManager::answer_inline_query(int64 inline_query_id, bool is_pe
thumbnail_url = std::move(document->thumbnail_url_); thumbnail_url = std::move(document->thumbnail_url_);
content_url = std::move(document->document_url_); content_url = std::move(document->document_url_);
content_type = std::move(document->mime_type_); content_type = std::move(document->mime_type_);
width = document->thumbnail_width_; thumbnail_width = document->thumbnail_width_;
height = document->thumbnail_height_; thumbnail_height = document->thumbnail_height_;
if (content_url.find('.') != string::npos) { if (content_url.find('.') != string::npos) {
if (begins_with(content_type, "application/pdf")) { if (begins_with(content_type, "application/pdf")) {
@ -637,10 +640,10 @@ void InlineQueriesManager::answer_inline_query(int64 inline_query_id, bool is_pe
title = std::move(location->title_); title = std::move(location->title_);
description = to_string(location->location_->latitude_) + " " + to_string(location->location_->longitude_); description = to_string(location->location_->latitude_) + " " + to_string(location->location_->longitude_);
thumbnail_url = std::move(location->thumbnail_url_); thumbnail_url = std::move(location->thumbnail_url_);
duration = location->live_period_; // duration = location->live_period_;
if (!thumbnail_url.empty()) { if (!thumbnail_url.empty()) {
width = location->thumbnail_width_; thumbnail_width = location->thumbnail_width_;
height = location->thumbnail_height_; thumbnail_height = location->thumbnail_height_;
} }
r_inline_message = r_inline_message =
@ -693,8 +696,8 @@ void InlineQueriesManager::answer_inline_query(int64 inline_query_id, bool is_pe
description = std::move(venue->venue_->address_); description = std::move(venue->venue_->address_);
thumbnail_url = std::move(venue->thumbnail_url_); thumbnail_url = std::move(venue->thumbnail_url_);
if (!thumbnail_url.empty()) { if (!thumbnail_url.empty()) {
width = venue->thumbnail_width_; thumbnail_width = venue->thumbnail_width_;
height = venue->thumbnail_height_; thumbnail_height = venue->thumbnail_height_;
} }
r_inline_message = r_inline_message =
@ -806,35 +809,54 @@ void InlineQueriesManager::answer_inline_query(int64 inline_query_id, bool is_pe
return promise.set_error(Status::Error(400, "Strings must be encoded in UTF-8")); return promise.set_error(Status::Error(400, "Strings must be encoded in UTF-8"));
} }
} }
tl_object_ptr<telegram_api::inputWebDocument> thumbnail;
if (!thumbnail_url.empty()) { if (!thumbnail_url.empty()) {
flags |= telegram_api::inputBotInlineResult::THUMB_URL_MASK; flags |= telegram_api::inputBotInlineResult::THUMB_MASK;
if (!clean_input_string(thumbnail_url)) { if (!clean_input_string(thumbnail_url)) {
return promise.set_error(Status::Error(400, "Strings must be encoded in UTF-8")); return promise.set_error(Status::Error(400, "Strings must be encoded in UTF-8"));
} }
vector<tl_object_ptr<telegram_api::DocumentAttribute>> attributes;
if (thumbnail_width > 0 && thumbnail_height > 0) {
attributes.push_back(
make_tl_object<telegram_api::documentAttributeImageSize>(thumbnail_width, thumbnail_height));
}
thumbnail = make_tl_object<telegram_api::inputWebDocument>(thumbnail_url, 0, "image/jpeg", std::move(attributes));
} }
tl_object_ptr<telegram_api::inputWebDocument> content;
if (!content_url.empty() || !content_type.empty()) { if (!content_url.empty() || !content_type.empty()) {
static_assert( flags |= telegram_api::inputBotInlineResult::CONTENT_MASK;
telegram_api::inputBotInlineResult::CONTENT_URL_MASK == telegram_api::inputBotInlineResult::CONTENT_TYPE_MASK,
"");
flags |= telegram_api::inputBotInlineResult::CONTENT_URL_MASK;
if (!clean_input_string(content_url)) { if (!clean_input_string(content_url)) {
return promise.set_error(Status::Error(400, "Strings must be encoded in UTF-8")); return promise.set_error(Status::Error(400, "Strings must be encoded in UTF-8"));
} }
if (!clean_input_string(content_type)) { if (!clean_input_string(content_type)) {
return promise.set_error(Status::Error(400, "Strings must be encoded in UTF-8")); return promise.set_error(Status::Error(400, "Strings must be encoded in UTF-8"));
} }
}
if (width > 0 && height > 0) { vector<tl_object_ptr<telegram_api::DocumentAttribute>> attributes;
static_assert(telegram_api::inputBotInlineResult::W_MASK == telegram_api::inputBotInlineResult::H_MASK, ""); if (width > 0 && height > 0) {
flags |= telegram_api::inputBotInlineResult::W_MASK; if (duration > 0 && !begins_with(content_type, "image/")) {
} attributes.push_back(make_tl_object<telegram_api::documentAttributeVideo>(
if (duration > 0) { 0, false /*ignored*/, false /*ignored*/, duration, width, height));
flags |= telegram_api::inputBotInlineResult::DURATION_MASK; } else {
attributes.push_back(make_tl_object<telegram_api::documentAttributeImageSize>(width, height));
}
} else if (duration > 0) {
if (type == "audio") {
attributes.push_back(make_tl_object<telegram_api::documentAttributeAudio>(
telegram_api::documentAttributeAudio::TITLE_MASK | telegram_api::documentAttributeAudio::PERFORMER_MASK,
false /*ignored*/, duration, title, description, BufferSlice()));
} else if (type == "voice") {
attributes.push_back(make_tl_object<telegram_api::documentAttributeAudio>(
telegram_api::documentAttributeAudio::VOICE_MASK, false /*ignored*/, duration, "", "", BufferSlice()));
}
}
attributes.push_back(make_tl_object<telegram_api::documentAttributeFilename>(get_url_file_name(content_url)));
content = make_tl_object<telegram_api::inputWebDocument>(content_url, 0, content_type, std::move(attributes));
} }
auto result = make_tl_object<telegram_api::inputBotInlineResult>(flags, id, type, title, description, url, auto result = make_tl_object<telegram_api::inputBotInlineResult>(
thumbnail_url, content_url, content_type, width, flags, id, type, title, description, url, std::move(thumbnail), std::move(content), std::move(inline_message));
height, duration, std::move(inline_message));
results.push_back(std::move(result)); results.push_back(std::move(result));
} }
@ -1155,19 +1177,62 @@ tl_object_ptr<td_api::inlineQueryResults> InlineQueriesManager::decrease_pending
return copy(it->second.results); return copy(it->second.results);
} }
tl_object_ptr<td_api::photoSize> InlineQueriesManager::register_thumbnail(const string &thumbnail_url, tl_object_ptr<td_api::photoSize> InlineQueriesManager::register_thumbnail(
Dimensions dimensions) const { tl_object_ptr<telegram_api::WebDocument> &&web_document_ptr) const {
if (thumbnail_url.find('.') != string::npos) { PhotoSize thumbnail = get_web_document_photo_size(td_->file_manager_.get(), FileType::Thumbnail, DialogId(),
auto r_thumbnail_file_id = td_->file_manager_->from_persistent_id(thumbnail_url, FileType::Temp); std::move(web_document_ptr));
if (r_thumbnail_file_id.is_ok()) { if (!thumbnail.file_id.is_valid()) {
PhotoSize thumbnail; return nullptr;
thumbnail.type = 't';
thumbnail.dimensions = dimensions;
thumbnail.file_id = r_thumbnail_file_id.move_as_ok();
return get_photo_size_object(td_->file_manager_.get(), &thumbnail);
}
} }
return nullptr;
return get_photo_size_object(td_->file_manager_.get(), &thumbnail);
}
string InlineQueriesManager::get_web_document_url(const tl_object_ptr<telegram_api::WebDocument> &web_document_ptr) {
if (web_document_ptr == nullptr) {
return {};
}
string url;
switch (web_document_ptr->get_id()) {
case telegram_api::webDocument::ID: {
auto web_document = static_cast<const telegram_api::webDocument *>(web_document_ptr.get());
url = web_document->url_;
break;
}
case telegram_api::webDocumentNoProxy::ID: {
auto web_document = static_cast<const telegram_api::webDocumentNoProxy *>(web_document_ptr.get());
url = web_document->url_;
break;
}
default:
UNREACHABLE();
}
auto r_http_url = parse_url(url);
if (r_http_url.is_error()) {
LOG(ERROR) << "Can't parse URL " << url;
return {};
}
return r_http_url.ok().get_url();
}
string InlineQueriesManager::get_web_document_content_type(
const tl_object_ptr<telegram_api::WebDocument> &web_document_ptr) {
if (web_document_ptr == nullptr) {
return {};
}
switch (web_document_ptr->get_id()) {
case telegram_api::webDocument::ID:
return static_cast<const telegram_api::webDocument *>(web_document_ptr.get())->mime_type_;
case telegram_api::webDocumentNoProxy::ID:
return static_cast<const telegram_api::webDocumentNoProxy *>(web_document_ptr.get())->mime_type_;
default:
UNREACHABLE();
}
return {};
} }
void InlineQueriesManager::on_get_inline_query_results(UserId bot_user_id, uint64 query_hash, void InlineQueriesManager::on_get_inline_query_results(UserId bot_user_id, uint64 query_hash,
@ -1346,25 +1411,11 @@ void InlineQueriesManager::on_get_inline_query_results(UserId bot_user_id, uint6
} }
case telegram_api::botInlineResult::ID: { case telegram_api::botInlineResult::ID: {
auto result = move_tl_object_as<telegram_api::botInlineResult>(result_ptr); auto result = move_tl_object_as<telegram_api::botInlineResult>(result_ptr);
Dimensions dimensions; auto content_type = get_web_document_content_type(result->content_);
if (result->flags_ & (1 << 6)) {
dimensions = get_dimensions(result->w_, result->h_);
} // TODO zero fetch init
int32 duration;
if (result->flags_ & (1 << 7)) {
duration = result->duration_;
if (duration < 0) {
duration = 0;
}
} else {
duration = 0;
} // TODO zero fetch init
if (result->type_ == "article") { if (result->type_ == "article") {
auto article = make_tl_object<td_api::inlineQueryResultArticle>(); auto article = make_tl_object<td_api::inlineQueryResultArticle>();
article->id_ = std::move(result->id_); article->id_ = std::move(result->id_);
article->url_ = std::move(result->content_url_); article->url_ = get_web_document_url(std::move(result->content_));
if (result->url_.empty()) { if (result->url_.empty()) {
article->hide_url_ = true; article->hide_url_ = true;
} else { } else {
@ -1374,7 +1425,7 @@ void InlineQueriesManager::on_get_inline_query_results(UserId bot_user_id, uint6
} }
article->title_ = std::move(result->title_); article->title_ = std::move(result->title_);
article->description_ = std::move(result->description_); article->description_ = std::move(result->description_);
article->thumbnail_ = register_thumbnail(result->thumb_url_, dimensions); article->thumbnail_ = register_thumbnail(std::move(result->thumb_));
if (!register_inline_message_content(results->query_id_, article->id_, FileId(), if (!register_inline_message_content(results->query_id_, article->id_, FileId(),
std::move(result->send_message_), -1)) { std::move(result->send_message_), -1)) {
@ -1394,7 +1445,7 @@ void InlineQueriesManager::on_get_inline_query_results(UserId bot_user_id, uint6
Contact c(std::move(result->description_), std::move(result->title_), string(), 0); Contact c(std::move(result->description_), std::move(result->title_), string(), 0);
contact->contact_ = c.get_contact_object(); contact->contact_ = c.get_contact_object();
} }
contact->thumbnail_ = register_thumbnail(result->thumb_url_, dimensions); contact->thumbnail_ = register_thumbnail(std::move(result->thumb_));
if (!register_inline_message_content(results->query_id_, contact->id_, FileId(), if (!register_inline_message_content(results->query_id_, contact->id_, FileId(),
std::move(result->send_message_), -1)) { std::move(result->send_message_), -1)) {
@ -1415,7 +1466,7 @@ void InlineQueriesManager::on_get_inline_query_results(UserId bot_user_id, uint6
Location l(to_double(coordinates.first), to_double(coordinates.second)); Location l(to_double(coordinates.first), to_double(coordinates.second));
location->location_ = l.get_location_object(); location->location_ = l.get_location_object();
} }
location->thumbnail_ = register_thumbnail(result->thumb_url_, dimensions); location->thumbnail_ = register_thumbnail(std::move(result->thumb_));
if (!register_inline_message_content(results->query_id_, location->id_, FileId(), if (!register_inline_message_content(results->query_id_, location->id_, FileId(),
std::move(result->send_message_), -1)) { std::move(result->send_message_), -1)) {
@ -1441,144 +1492,141 @@ void InlineQueriesManager::on_get_inline_query_results(UserId bot_user_id, uint6
Venue v(nullptr, std::move(result->title_), std::move(result->description_), string(), string()); Venue v(nullptr, std::move(result->title_), std::move(result->description_), string(), string());
venue->venue_ = v.get_venue_object(); venue->venue_ = v.get_venue_object();
} }
venue->thumbnail_ = register_thumbnail(result->thumb_url_, dimensions); venue->thumbnail_ = register_thumbnail(std::move(result->thumb_));
if (!register_inline_message_content(results->query_id_, venue->id_, FileId(), if (!register_inline_message_content(results->query_id_, venue->id_, FileId(),
std::move(result->send_message_), -1)) { std::move(result->send_message_), -1)) {
continue; continue;
} }
output_result = std::move(venue); output_result = std::move(venue);
} else if (result->type_ == "photo" && content_type == "image/jpeg") {
auto photo = make_tl_object<td_api::inlineQueryResultPhoto>();
photo->id_ = std::move(result->id_);
PhotoSize photo_size = get_web_document_photo_size(td_->file_manager_.get(), FileType::Temp, DialogId(),
std::move(result->content_));
if (!photo_size.file_id.is_valid()) {
LOG(ERROR) << "Receive invalid web document photo";
continue;
}
Photo new_photo;
PhotoSize thumbnail = get_web_document_photo_size(td_->file_manager_.get(), FileType::Thumbnail, DialogId(),
std::move(result->thumb_));
if (thumbnail.file_id.is_valid()) {
new_photo.photos.push_back(std::move(thumbnail));
}
new_photo.photos.push_back(std::move(photo_size));
photo->photo_ = get_photo_object(td_->file_manager_.get(), &new_photo);
photo->title_ = std::move(result->title_);
photo->description_ = std::move(result->description_);
if (!register_inline_message_content(results->query_id_, photo->id_, FileId(),
std::move(result->send_message_), td_api::inputMessagePhoto::ID,
&new_photo)) {
continue;
}
output_result = std::move(photo);
} else { } else {
auto r_http_url = parse_url(result->content_url_); if (result->content_ == nullptr) {
if (r_http_url.is_error()) { LOG(ERROR) << "Unsupported inline query result without content " << to_string(result);
LOG(ERROR) << "Can't parse url " << result->content_url_;
continue; continue;
} }
auto http_url = r_http_url.ok();
auto url = http_url.get_url();
auto r_file_id = td_->file_manager_->from_persistent_id(url, FileType::Temp);
if (r_file_id.is_error()) {
LOG(ERROR) << "Can't register url " << url;
continue;
}
auto file_id = r_file_id.move_as_ok();
auto file_name = get_url_query_file_name(http_url.query_);
PhotoSize thumbnail; vector<tl_object_ptr<telegram_api::DocumentAttribute>> attributes;
if (result->thumb_url_.find('.') != string::npos) { downcast_call(*result->content_,
auto r_thumbnail_file_id = td_->file_manager_->from_persistent_id(result->thumb_url_, FileType::Temp); [&attributes](auto &web_document) { attributes = std::move(web_document.attributes_); });
if (r_thumbnail_file_id.is_error()) {
LOG(ERROR) << "Can't register thumbnail url " << result->thumb_url_; bool is_animation = result->type_ == "gif" && (content_type == "image/gif" || content_type == "video/mp4");
continue; if (is_animation) {
attributes.push_back(make_tl_object<telegram_api::documentAttributeAnimated>());
}
auto default_document_type = [type = result->type_, is_animation] {
if (type == "audio") {
return DocumentsManager::DocumentType::Audio;
} }
thumbnail.type = 't'; if (is_animation) {
thumbnail.file_id = r_thumbnail_file_id.move_as_ok(); return DocumentsManager::DocumentType::Animation;
} }
if (type == "sticker") {
return DocumentsManager::DocumentType::Sticker;
}
if (type == "video") {
return DocumentsManager::DocumentType::Video;
}
if (type == "voice") {
return DocumentsManager::DocumentType::VoiceNote;
}
return DocumentsManager::DocumentType::General;
}();
if (result->type_ == "audio") { auto parsed_document = td_->documents_manager_->on_get_document(
{std::move(result->content_),
get_web_document_photo_size(td_->file_manager_.get(), FileType::Thumbnail, DialogId(),
std::move(result->thumb_)),
std::move(attributes)},
DialogId(), nullptr, default_document_type);
auto file_id = parsed_document.second;
if (!file_id.is_valid()) {
continue;
}
if (result->type_ == "audio" && parsed_document.first == DocumentsManager::DocumentType::Audio) {
auto audio = make_tl_object<td_api::inlineQueryResultAudio>(); auto audio = make_tl_object<td_api::inlineQueryResultAudio>();
audio->id_ = std::move(result->id_); audio->id_ = std::move(result->id_);
td_->audios_manager_->create_audio(file_id, std::move(thumbnail), std::move(file_name),
std::move(result->content_type_), duration, std::move(result->title_),
std::move(result->description_), false);
audio->audio_ = td_->audios_manager_->get_audio_object(file_id); audio->audio_ = td_->audios_manager_->get_audio_object(file_id);
if (!register_inline_message_content(results->query_id_, audio->id_, file_id, if (!register_inline_message_content(results->query_id_, audio->id_, file_id,
std::move(result->send_message_), td_api::inputMessageAudio::ID)) { std::move(result->send_message_), td_api::inputMessageAudio::ID)) {
continue; continue;
} }
output_result = std::move(audio); output_result = std::move(audio);
} else if (result->type_ == "file") { } else if (result->type_ == "file" && parsed_document.first == DocumentsManager::DocumentType::General) {
thumbnail.dimensions = dimensions;
auto document = make_tl_object<td_api::inlineQueryResultDocument>(); auto document = make_tl_object<td_api::inlineQueryResultDocument>();
document->id_ = std::move(result->id_); document->id_ = std::move(result->id_);
td_->documents_manager_->create_document(file_id, std::move(thumbnail), std::move(file_name),
std::move(result->content_type_), false);
document->document_ = td_->documents_manager_->get_document_object(file_id); document->document_ = td_->documents_manager_->get_document_object(file_id);
document->title_ = std::move(result->title_); document->title_ = std::move(result->title_);
document->description_ = std::move(result->description_); document->description_ = std::move(result->description_);
if (!register_inline_message_content(results->query_id_, document->id_, file_id, if (!register_inline_message_content(results->query_id_, document->id_, file_id,
std::move(result->send_message_), td_api::inputMessageDocument::ID)) { std::move(result->send_message_), td_api::inputMessageDocument::ID)) {
continue; continue;
} }
output_result = std::move(document); output_result = std::move(document);
} else if (result->type_ == "gif" && } else if (is_animation && parsed_document.first == DocumentsManager::DocumentType::Animation) {
(result->content_type_ == "image/gif" || result->content_type_ == "video/mp4")) {
auto animation = make_tl_object<td_api::inlineQueryResultAnimation>(); auto animation = make_tl_object<td_api::inlineQueryResultAnimation>();
animation->id_ = std::move(result->id_); animation->id_ = std::move(result->id_);
td_->animations_manager_->create_animation(file_id, std::move(thumbnail), std::move(file_name),
std::move(result->content_type_), duration, dimensions, false);
animation->animation_ = animation->animation_ =
td_->animations_manager_->get_animation_object(file_id, "inlineQueryResultAnimationCached"); td_->animations_manager_->get_animation_object(file_id, "inlineQueryResultAnimationCached");
animation->title_ = std::move(result->title_); animation->title_ = std::move(result->title_);
if (!register_inline_message_content(results->query_id_, animation->id_, file_id, if (!register_inline_message_content(results->query_id_, animation->id_, file_id,
std::move(result->send_message_), td_api::inputMessageAnimation::ID)) { std::move(result->send_message_), td_api::inputMessageAnimation::ID)) {
continue; continue;
} }
output_result = std::move(animation); output_result = std::move(animation);
} else if (result->type_ == "photo" && result->content_type_ == "image/jpeg") { } else if (result->type_ == "sticker" && parsed_document.first == DocumentsManager::DocumentType::Sticker) {
auto photo = make_tl_object<td_api::inlineQueryResultPhoto>();
photo->id_ = std::move(result->id_);
// TODO create_photo
Photo new_photo;
PhotoSize s;
s.type = 'u';
s.dimensions = dimensions;
s.size = 0;
s.file_id = file_id;
if (thumbnail.file_id.is_valid()) {
new_photo.photos.push_back(std::move(thumbnail));
}
new_photo.photos.push_back(std::move(s));
photo->photo_ = get_photo_object(td_->file_manager_.get(), &new_photo);
photo->title_ = std::move(result->title_);
photo->description_ = std::move(result->description_);
if (!register_inline_message_content(results->query_id_, photo->id_, FileId(),
std::move(result->send_message_), td_api::inputMessagePhoto::ID,
&new_photo)) {
continue;
}
output_result = std::move(photo);
} else if (result->type_ == "sticker") {
auto sticker = make_tl_object<td_api::inlineQueryResultSticker>(); auto sticker = make_tl_object<td_api::inlineQueryResultSticker>();
sticker->id_ = std::move(result->id_); sticker->id_ = std::move(result->id_);
td_->stickers_manager_->create_sticker(file_id, std::move(thumbnail), dimensions, true, nullptr, nullptr);
sticker->sticker_ = td_->stickers_manager_->get_sticker_object(file_id); sticker->sticker_ = td_->stickers_manager_->get_sticker_object(file_id);
if (!register_inline_message_content(results->query_id_, sticker->id_, file_id, if (!register_inline_message_content(results->query_id_, sticker->id_, file_id,
std::move(result->send_message_), td_api::inputMessageSticker::ID)) { std::move(result->send_message_), td_api::inputMessageSticker::ID)) {
continue; continue;
} }
output_result = std::move(sticker); output_result = std::move(sticker);
} else if (result->type_ == "video") { } else if (result->type_ == "video" && parsed_document.first == DocumentsManager::DocumentType::Video) {
auto video = make_tl_object<td_api::inlineQueryResultVideo>(); auto video = make_tl_object<td_api::inlineQueryResultVideo>();
video->id_ = std::move(result->id_); video->id_ = std::move(result->id_);
td_->videos_manager_->create_video(file_id, std::move(thumbnail), false, {}, std::move(file_name),
std::move(result->content_type_), duration, dimensions, false, false);
video->video_ = td_->videos_manager_->get_video_object(file_id); video->video_ = td_->videos_manager_->get_video_object(file_id);
video->title_ = std::move(result->title_); video->title_ = std::move(result->title_);
video->description_ = std::move(result->description_); video->description_ = std::move(result->description_);
if (!register_inline_message_content(results->query_id_, video->id_, file_id, if (!register_inline_message_content(results->query_id_, video->id_, file_id,
std::move(result->send_message_), td_api::inputMessageVideo::ID)) { std::move(result->send_message_), td_api::inputMessageVideo::ID)) {
continue; continue;
} }
output_result = std::move(video); output_result = std::move(video);
} else if (result->type_ == "voice") { } else if (result->type_ == "voice" && parsed_document.first == DocumentsManager::DocumentType::VoiceNote) {
auto voice_note = make_tl_object<td_api::inlineQueryResultVoiceNote>(); auto voice_note = make_tl_object<td_api::inlineQueryResultVoiceNote>();
voice_note->id_ = std::move(result->id_); voice_note->id_ = std::move(result->id_);
td_->voice_notes_manager_->create_voice_note(file_id, std::move(result->content_type_), duration, string(),
false);
voice_note->voice_note_ = td_->voice_notes_manager_->get_voice_note_object(file_id); voice_note->voice_note_ = td_->voice_notes_manager_->get_voice_note_object(file_id);
voice_note->title_ = std::move(result->title_); voice_note->title_ = std::move(result->title_);
if (!register_inline_message_content(results->query_id_, voice_note->id_, file_id, if (!register_inline_message_content(results->query_id_, voice_note->id_, file_id,
std::move(result->send_message_), td_api::inputMessageVoiceNote::ID)) { std::move(result->send_message_), td_api::inputMessageVoiceNote::ID)) {
continue; continue;

6
td/telegram/InlineQueriesManager.h
View File

@ -100,7 +100,11 @@ class InlineQueriesManager : public Actor {
tl_object_ptr<telegram_api::BotInlineMessage> &&inline_message, tl_object_ptr<telegram_api::BotInlineMessage> &&inline_message,
int32 allowed_media_content_id, Photo *photo = nullptr, Game *game = nullptr); int32 allowed_media_content_id, Photo *photo = nullptr, Game *game = nullptr);
tl_object_ptr<td_api::photoSize> register_thumbnail(const string &thumbnail_url, Dimensions dimensions) const; tl_object_ptr<td_api::photoSize> register_thumbnail(
tl_object_ptr<telegram_api::WebDocument> &&web_document_ptr) const;
static string get_web_document_url(const tl_object_ptr<telegram_api::WebDocument> &web_document_ptr);
static string get_web_document_content_type(const tl_object_ptr<telegram_api::WebDocument> &web_document_ptr);
bool update_bot_usage(UserId bot_user_id); bool update_bot_usage(UserId bot_user_id);

61
td/telegram/MessagesManager.cpp
View File

@ -20233,62 +20233,17 @@ static tl_object_ptr<ToT> secret_to_telegram(FromT &from) {
return res; return res;
} }
Photo MessagesManager::get_web_document_photo(tl_object_ptr<telegram_api::webDocument> web_document, Photo MessagesManager::get_web_document_photo(tl_object_ptr<telegram_api::WebDocument> web_document,
DialogId owner_dialog_id) const { DialogId owner_dialog_id) const {
PhotoSize s =
get_web_document_photo_size(td_->file_manager_.get(), FileType::Photo, owner_dialog_id, std::move(web_document));
Photo photo; Photo photo;
photo.id = -2; if (!s.file_id.is_valid()) {
photo.id = -2;
if (web_document == nullptr) { } else {
return photo; photo.id = 0;
photo.photos.push_back(s);
} }
if (!DcId::is_valid(web_document->dc_id_)) {
LOG(ERROR) << "Wrong dc_id = " << web_document->dc_id_;
return photo;
}
auto r_http_url = parse_url(web_document->url_);
if (r_http_url.is_error()) {
LOG(ERROR) << "Can't parse url " << web_document->url_;
return photo;
}
auto http_url = r_http_url.move_as_ok();
auto url = http_url.get_url();
// TODO real file name
FileId file_id = td_->file_manager_->register_remote(
FullRemoteFileLocation(FileType::Photo, url, web_document->access_hash_, DcId::internal(web_document->dc_id_)),
FileLocationSource::FromServer, owner_dialog_id, 0, web_document->size_, "");
Dimensions dimensions;
for (auto &attribute : web_document->attributes_) {
switch (attribute->get_id()) {
case telegram_api::documentAttributeImageSize::ID: {
auto image_size = move_tl_object_as<telegram_api::documentAttributeImageSize>(attribute);
dimensions = get_dimensions(image_size->w_, image_size->h_);
break;
}
case telegram_api::documentAttributeAnimated::ID:
case telegram_api::documentAttributeHasStickers::ID:
case telegram_api::documentAttributeSticker::ID:
case telegram_api::documentAttributeVideo::ID:
case telegram_api::documentAttributeAudio::ID:
LOG(ERROR) << "Unexpected web document attribute " << to_string(attribute);
break;
case telegram_api::documentAttributeFilename::ID:
break;
default:
UNREACHABLE();
}
}
PhotoSize s;
s.type = 'u';
s.dimensions = dimensions;
s.size = web_document->size_;
s.file_id = file_id;
photo.id = 0;
photo.photos.push_back(s);
return photo; return photo;
} }

2
td/telegram/MessagesManager.h
View File

@ -2229,7 +2229,7 @@ class MessagesManager : public Actor {
static FormattedText get_secret_media_caption(string &&message_text, string &&message_caption); static FormattedText get_secret_media_caption(string &&message_text, string &&message_caption);
Photo get_web_document_photo(tl_object_ptr<telegram_api::webDocument> web_document, DialogId owner_dialog_id) const; Photo get_web_document_photo(tl_object_ptr<telegram_api::WebDocument> web_document, DialogId owner_dialog_id) const;
unique_ptr<MessageContent> get_secret_message_document( unique_ptr<MessageContent> get_secret_message_document(
tl_object_ptr<telegram_api::encryptedFile> file, tl_object_ptr<telegram_api::encryptedFile> file,

87
td/telegram/Photo.cpp
View File

@ -14,6 +14,7 @@
#include "td/utils/common.h" #include "td/utils/common.h"
#include "td/utils/format.h" #include "td/utils/format.h"
#include "td/utils/HttpUrl.h"
#include "td/utils/logging.h" #include "td/utils/logging.h"
#include "td/utils/misc.h" #include "td/utils/misc.h"
#include "td/utils/Random.h" #include "td/utils/Random.h"
@ -286,6 +287,92 @@ PhotoSize get_photo_size(FileManager *file_manager, FileType file_type, int64 id
return res; return res;
} }
PhotoSize get_web_document_photo_size(FileManager *file_manager, FileType file_type, DialogId owner_dialog_id,
tl_object_ptr<telegram_api::WebDocument> web_document_ptr) {
if (web_document_ptr == nullptr) {
return {};
}
FileId file_id;
vector<tl_object_ptr<telegram_api::DocumentAttribute>> attributes;
int32 size = 0;
switch (web_document_ptr->get_id()) {
case telegram_api::webDocument::ID: {
auto web_document = move_tl_object_as<telegram_api::webDocument>(web_document_ptr);
if (!DcId::is_valid(web_document->dc_id_)) {
LOG(ERROR) << "Wrong dc_id = " << web_document->dc_id_;
return {};
}
auto r_http_url = parse_url(web_document->url_);
if (r_http_url.is_error()) {
LOG(ERROR) << "Can't parse URL " << web_document->url_;
return {};
}
auto http_url = r_http_url.move_as_ok();
auto url = http_url.get_url();
file_id = file_manager->register_remote(
FullRemoteFileLocation(file_type, url, web_document->access_hash_, DcId::internal(web_document->dc_id_)),
FileLocationSource::FromServer, owner_dialog_id, 0, web_document->size_,
get_url_query_file_name(http_url.query_));
size = web_document->size_;
attributes = std::move(web_document->attributes_);
break;
}
case telegram_api::webDocumentNoProxy::ID: {
auto web_document = move_tl_object_as<telegram_api::webDocumentNoProxy>(web_document_ptr);
if (web_document->url_.find('.') == string::npos) {
LOG(ERROR) << "Receive invalid URL " << web_document->url_;
return {};
}
auto r_file_id = file_manager->from_persistent_id(web_document->url_, file_type);
if (r_file_id.is_error()) {
LOG(ERROR) << "Can't register URL: " << r_file_id.error();
return {};
}
file_id = r_file_id.move_as_ok();
size = web_document->size_;
attributes = std::move(web_document->attributes_);
break;
}
default:
UNREACHABLE();
}
CHECK(file_id.is_valid());
Dimensions dimensions;
for (auto &attribute : attributes) {
switch (attribute->get_id()) {
case telegram_api::documentAttributeImageSize::ID: {
auto image_size = move_tl_object_as<telegram_api::documentAttributeImageSize>(attribute);
dimensions = get_dimensions(image_size->w_, image_size->h_);
break;
}
case telegram_api::documentAttributeAnimated::ID:
case telegram_api::documentAttributeHasStickers::ID:
case telegram_api::documentAttributeSticker::ID:
case telegram_api::documentAttributeVideo::ID:
case telegram_api::documentAttributeAudio::ID:
LOG(ERROR) << "Unexpected web document attribute " << to_string(attribute);
break;
case telegram_api::documentAttributeFilename::ID:
break;
default:
UNREACHABLE();
}
}
PhotoSize s;
s.type = file_type == FileType::Thumbnail ? 't' : 'u';
s.dimensions = dimensions;
s.size = size;
s.file_id = file_id;
return s;
}
tl_object_ptr<td_api::photoSize> get_photo_size_object(FileManager *file_manager, const PhotoSize *photo_size) { tl_object_ptr<td_api::photoSize> get_photo_size_object(FileManager *file_manager, const PhotoSize *photo_size) {
if (photo_size == nullptr || !photo_size->file_id.is_valid()) { if (photo_size == nullptr || !photo_size->file_id.is_valid()) {
return nullptr; return nullptr;

2
td/telegram/Photo.h
View File

@ -82,6 +82,8 @@ PhotoSize get_thumbnail_photo_size(FileManager *file_manager, BufferSlice bytes,
int32 height); int32 height);
PhotoSize get_photo_size(FileManager *file_manager, FileType file_type, int64 id, int64 access_hash, PhotoSize get_photo_size(FileManager *file_manager, FileType file_type, int64 id, int64 access_hash,
DialogId owner_dialog_id, tl_object_ptr<telegram_api::PhotoSize> &&size_ptr); DialogId owner_dialog_id, tl_object_ptr<telegram_api::PhotoSize> &&size_ptr);
PhotoSize get_web_document_photo_size(FileManager *file_manager, FileType file_type, DialogId owner_dialog_id,
tl_object_ptr<telegram_api::WebDocument> web_document_ptr);
tl_object_ptr<td_api::photoSize> get_photo_size_object(FileManager *file_manager, const PhotoSize *photo_size); tl_object_ptr<td_api::photoSize> get_photo_size_object(FileManager *file_manager, const PhotoSize *photo_size);
bool operator==(const PhotoSize &lhs, const PhotoSize &rhs); bool operator==(const PhotoSize &lhs, const PhotoSize &rhs);