tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
0115a73d6e
tdlight/td/telegram/LinkManager.cpp

3129 lines
121 KiB
C++
Raw Blame History

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/telegram/LinkManager.h"
#include "td/telegram/AccessRights.h"
#include "td/telegram/BackgroundType.h"
#include "td/telegram/ChannelId.h"
#include "td/telegram/ChannelType.h"
#include "td/telegram/ConfigManager.h"
#include "td/telegram/DialogId.h"
#include "td/telegram/DialogManager.h"
#include "td/telegram/DialogParticipant.h"
#include "td/telegram/Global.h"
#include "td/telegram/MessageEntity.h"
#include "td/telegram/MessageId.h"
#include "td/telegram/MessagesManager.h"
#include "td/telegram/misc.h"
#include "td/telegram/net/Proxy.h"
#include "td/telegram/ServerMessageId.h"
#include "td/telegram/StoryId.h"
#include "td/telegram/Td.h"
#include "td/telegram/TdDb.h"
#include "td/telegram/telegram_api.h"
#include "td/telegram/UserId.h"
#include "td/telegram/UserManager.h"
#include "td/mtproto/ProxySecret.h"
#include "td/utils/algorithm.h"
#include "td/utils/base64.h"
#include "td/utils/buffer.h"
#include "td/utils/FlatHashSet.h"
#include "td/utils/HttpUrl.h"
#include "td/utils/logging.h"
#include "td/utils/misc.h"
#include "td/utils/SliceBuilder.h"
#include "td/utils/StringBuilder.h"
#include "td/utils/Time.h"
#include "td/utils/utf8.h"
#include <tuple>
namespace td {
static bool is_valid_start_parameter(Slice start_parameter) {
return is_base64url_characters(start_parameter);
}
static bool is_valid_phone_number(Slice phone_number) {
if (phone_number.empty() || phone_number.size() > 32) {
return false;
}
for (auto c : phone_number) {
if (!is_digit(c)) {
return false;
}
}
return true;
}
static bool is_valid_game_name(Slice name) {
return name.size() >= 3 && is_valid_username(name);
}
static bool is_valid_web_app_name(Slice name) {
return name.size() >= 3 && is_valid_username(name);
}
static bool is_valid_story_id(Slice story_id) {
auto r_story_id = to_integer_safe<int32>(story_id);
return r_story_id.is_ok() && StoryId(r_story_id.ok()).is_server();
}
static string get_url_query_hash(bool is_tg, const HttpUrlQuery &url_query) {
const auto &path = url_query.path_;
if (is_tg) {
if (path.size() == 1 && path[0] == "join") {
// join?invite=<hash>
return url_query.get_arg("invite").str();
}
} else {
if (path.size() >= 2 && path[0] == "joinchat") {
// /joinchat/<hash>
return path[1];
}
if (!path.empty() && path[0].size() >= 2 && (path[0][0] == ' ' || path[0][0] == '+')) {
// /+<link>
return path[0].substr(1);
}
}
return string();
}
static string get_url_query_slug(bool is_tg, const HttpUrlQuery &url_query) {
const auto &path = url_query.path_;
if (is_tg) {
if (path.size() == 1 && path[0] == "addlist") {
// addlist?slug=<hash>
return url_query.get_arg("slug").str();
}
} else {
if (path.size() >= 2 && path[0] == "addlist") {
// /addlist/<hash>
return path[1];
}
}
return string();
}
static string get_url_query_draft_text(const HttpUrlQuery &url_query) {
auto text_slice = url_query.get_arg("text");
if (text_slice.empty()) {
return string();
}
auto text = text_slice.str();
if (!check_utf8(text)) {
return string();
}
text = utf8_truncate(std::move(text), 4096u);
if (text[0] == '@') {
return ' ' + text;
}
return text;
}
static AdministratorRights get_administrator_rights(Slice rights, bool for_channel) {
bool can_manage_dialog = false;
bool can_change_info = false;
bool can_post_messages = false;
bool can_edit_messages = false;
bool can_delete_messages = false;
bool can_invite_users = false;
bool can_restrict_members = false;
bool can_pin_messages = false;
bool can_manage_topics = false;
bool can_promote_members = false;
bool can_manage_calls = false;
bool can_post_stories = false;
bool can_edit_stories = false;
bool can_delete_stories = false;
bool is_anonymous = false;
for (auto right : full_split(rights, ' ')) {
if (right == "change_info") {
can_change_info = true;
} else if (right == "post_messages") {
can_post_messages = true;
} else if (right == "edit_messages") {
can_edit_messages = true;
} else if (right == "delete_messages") {
can_delete_messages = true;
} else if (right == "restrict_members") {
can_restrict_members = true;
} else if (right == "invite_users") {
can_invite_users = true;
} else if (right == "pin_messages") {
can_pin_messages = true;
} else if (right == "manage_topics") {
can_manage_topics = true;
} else if (right == "promote_members") {
can_promote_members = true;
} else if (right == "manage_video_chats") {
can_manage_calls = true;
} else if (right == "post_stories") {
can_post_stories = true;
} else if (right == "edit_stories") {
can_edit_stories = true;
} else if (right == "delete_stories") {
can_delete_stories = true;
} else if (right == "anonymous") {
is_anonymous = true;
} else if (right == "manage_chat") {
can_manage_dialog = true;
}
}
return AdministratorRights(is_anonymous, can_manage_dialog, can_change_info, can_post_messages, can_edit_messages,
can_delete_messages, can_invite_users, can_restrict_members, can_pin_messages,
can_manage_topics, can_promote_members, can_manage_calls, can_post_stories,
can_edit_stories, can_delete_stories,
for_channel ? ChannelType::Broadcast : ChannelType::Megagroup);
}
static string get_admin_string(AdministratorRights rights) {
vector<string> admin_rights;
if (rights.can_change_info_and_settings()) {
admin_rights.emplace_back("change_info");
}
if (rights.can_post_messages()) {
admin_rights.emplace_back("post_messages");
}
if (rights.can_edit_messages()) {
admin_rights.emplace_back("edit_messages");
}
if (rights.can_delete_messages()) {
admin_rights.emplace_back("delete_messages");
}
if (rights.can_restrict_members()) {
admin_rights.emplace_back("restrict_members");
}
if (rights.can_invite_users()) {
admin_rights.emplace_back("invite_users");
}
if (rights.can_pin_messages()) {
admin_rights.emplace_back("pin_messages");
}
if (rights.can_manage_topics()) {
admin_rights.emplace_back("manage_topics");
}
if (rights.can_promote_members()) {
admin_rights.emplace_back("promote_members");
}
if (rights.can_manage_calls()) {
admin_rights.emplace_back("manage_video_chats");
}
if (rights.can_post_stories()) {
admin_rights.emplace_back("post_stories");
}
if (rights.can_edit_stories()) {
admin_rights.emplace_back("edit_stories");
}
if (rights.can_delete_stories()) {
admin_rights.emplace_back("delete_stories");
}
if (rights.is_anonymous()) {
admin_rights.emplace_back("anonymous");
}
if (rights.can_manage_dialog()) {
admin_rights.emplace_back("manage_chat");
}
if (admin_rights.empty()) {
return string();
}
return "&admin=" + implode(admin_rights, '+');
}
td_api::object_ptr<td_api::targetChatChosen> get_target_chat_chosen(Slice chat_types) {
bool allow_users = false;
bool allow_bots = false;
bool allow_groups = false;
bool allow_channels = false;
for (auto chat_type : full_split(chat_types, ' ')) {
if (chat_type == "users") {
allow_users = true;
} else if (chat_type == "bots") {
allow_bots = true;
} else if (chat_type == "groups") {
allow_groups = true;
} else if (chat_type == "channels") {
allow_channels = true;
}
}
if (!allow_users && !allow_bots && !allow_groups && !allow_channels) {
return nullptr;
}
return td_api::make_object<td_api::targetChatChosen>(allow_users, allow_bots, allow_groups, allow_channels);
}
class LinkManager::InternalLinkActiveSessions final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeActiveSessions>();
}
};
class LinkManager::InternalLinkAttachMenuBot final : public InternalLink {
td_api::object_ptr<td_api::targetChatChosen> allowed_chat_types_;
unique_ptr<InternalLink> dialog_link_;
string bot_username_;
string url_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
td_api::object_ptr<td_api::TargetChat> target_chat;
if (dialog_link_ != nullptr) {
target_chat = td_api::make_object<td_api::targetChatInternalLink>(dialog_link_->get_internal_link_type_object());
} else if (allowed_chat_types_ != nullptr) {
target_chat = td_api::make_object<td_api::targetChatChosen>(
allowed_chat_types_->allow_user_chats_, allowed_chat_types_->allow_bot_chats_,
allowed_chat_types_->allow_group_chats_, allowed_chat_types_->allow_channel_chats_);
} else {
target_chat = td_api::make_object<td_api::targetChatCurrent>();
}
return td_api::make_object<td_api::internalLinkTypeAttachmentMenuBot>(std::move(target_chat), bot_username_, url_);
}
public:
InternalLinkAttachMenuBot(td_api::object_ptr<td_api::targetChatChosen> allowed_chat_types,
unique_ptr<InternalLink> dialog_link, string bot_username, Slice start_parameter)
: allowed_chat_types_(std::move(allowed_chat_types))
, dialog_link_(std::move(dialog_link))
, bot_username_(std::move(bot_username)) {
if (!start_parameter.empty()) {
url_ = PSTRING() << "start://" << start_parameter;
}
}
};
class LinkManager::InternalLinkAuthenticationCode final : public InternalLink {
string code_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeAuthenticationCode>(code_);
}
public:
explicit InternalLinkAuthenticationCode(string code) : code_(std::move(code)) {
}
};
class LinkManager::InternalLinkBackground final : public InternalLink {
string background_name_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeBackground>(background_name_);
}
public:
explicit InternalLinkBackground(string background_name) : background_name_(std::move(background_name)) {
}
};
class LinkManager::InternalLinkBotAddToChannel final : public InternalLink {
string bot_username_;
AdministratorRights administrator_rights_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeBotAddToChannel>(
bot_username_, administrator_rights_.get_chat_administrator_rights_object());
}
public:
InternalLinkBotAddToChannel(string bot_username, AdministratorRights &&administrator_rights)
: bot_username_(std::move(bot_username)), administrator_rights_(std::move(administrator_rights)) {
}
};
class LinkManager::InternalLinkBotStart final : public InternalLink {
string bot_username_;
string start_parameter_;
bool autostart_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
bool autostart = autostart_;
if (Scheduler::context() != nullptr && !autostart) {
if (bot_username_ == G()->get_option_string("premium_bot_username")) {
autostart = true;
} else {
const Td *td = G()->td().get_actor_unsafe();
auto dialog_id = td->dialog_manager_->get_resolved_dialog_by_username(bot_username_);
if (dialog_id.is_valid() && dialog_id.get_type() == DialogType::User &&
td->messages_manager_->get_dialog_has_last_message(dialog_id) &&
!td->messages_manager_->is_dialog_blocked(dialog_id)) {
autostart = true;
}
}
}
return td_api::make_object<td_api::internalLinkTypeBotStart>(bot_username_, start_parameter_, autostart);
}
public:
InternalLinkBotStart(string bot_username, string start_parameter, bool autostart)
: bot_username_(std::move(bot_username)), start_parameter_(std::move(start_parameter)), autostart_(autostart) {
}
};
class LinkManager::InternalLinkBotStartInGroup final : public InternalLink {
string bot_username_;
string start_parameter_;
AdministratorRights administrator_rights_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeBotStartInGroup>(
bot_username_, start_parameter_,
administrator_rights_ == AdministratorRights() ? nullptr
: administrator_rights_.get_chat_administrator_rights_object());
}
public:
InternalLinkBotStartInGroup(string bot_username, string start_parameter, AdministratorRights &&administrator_rights)
: bot_username_(std::move(bot_username))
, start_parameter_(std::move(start_parameter))
, administrator_rights_(std::move(administrator_rights)) {
}
};
class LinkManager::InternalLinkBusinessChat final : public InternalLink {
string link_name_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeBusinessChat>(link_name_);
}
public:
explicit InternalLinkBusinessChat(string link_name) : link_name_(std::move(link_name)) {
}
};
class LinkManager::InternalLinkChangePhoneNumber final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeChangePhoneNumber>();
}
};
class LinkManager::InternalLinkConfirmPhone final : public InternalLink {
string hash_;
string phone_number_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypePhoneNumberConfirmation>(hash_, phone_number_);
}
public:
InternalLinkConfirmPhone(string hash, string phone_number)
: hash_(std::move(hash)), phone_number_(std::move(phone_number)) {
}
};
class LinkManager::InternalLinkDefaultMessageAutoDeleteTimerSettings final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeDefaultMessageAutoDeleteTimerSettings>();
}
};
class LinkManager::InternalLinkDialogBoost final : public InternalLink {
string url_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeChatBoost>(url_);
}
public:
explicit InternalLinkDialogBoost(string url) : url_(std::move(url)) {
}
};
class LinkManager::InternalLinkDialogFolderInvite final : public InternalLink {
string url_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeChatFolderInvite>(url_);
}
public:
explicit InternalLinkDialogFolderInvite(string url) : url_(std::move(url)) {
}
};
class LinkManager::InternalLinkDialogInvite final : public InternalLink {
string url_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeChatInvite>(url_);
}
public:
explicit InternalLinkDialogInvite(string url) : url_(std::move(url)) {
}
};
class LinkManager::InternalLinkEditProfileSettings final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeEditProfileSettings>();
}
};
class LinkManager::InternalLinkDialogFolderSettings final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeChatFolderSettings>();
}
};
class LinkManager::InternalLinkGame final : public InternalLink {
string bot_username_;
string game_short_name_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeGame>(bot_username_, game_short_name_);
}
public:
InternalLinkGame(string bot_username, string game_short_name)
: bot_username_(std::move(bot_username)), game_short_name_(std::move(game_short_name)) {
}
};
class LinkManager::InternalLinkInstantView final : public InternalLink {
string url_;
string fallback_url_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeInstantView>(url_, fallback_url_);
}
public:
InternalLinkInstantView(string url, string fallback_url)
: url_(std::move(url)), fallback_url_(std::move(fallback_url)) {
}
};
class LinkManager::InternalLinkInvoice final : public InternalLink {
string invoice_name_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeInvoice>(invoice_name_);
}
public:
explicit InternalLinkInvoice(string invoice_name) : invoice_name_(std::move(invoice_name)) {
}
};
class LinkManager::InternalLinkLanguage final : public InternalLink {
string language_pack_id_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeLanguagePack>(language_pack_id_);
}
public:
explicit InternalLinkLanguage(string language_pack_id) : language_pack_id_(std::move(language_pack_id)) {
}
};
class LinkManager::InternalLinkLanguageSettings final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeLanguageSettings>();
}
};
class LinkManager::InternalLinkMessage final : public InternalLink {
string url_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeMessage>(url_);
}
public:
explicit InternalLinkMessage(string url) : url_(std::move(url)) {
}
};
class LinkManager::InternalLinkMessageDraft final : public InternalLink {
FormattedText text_;
bool contains_link_ = false;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeMessageDraft>(get_formatted_text_object(text_, true, -1),
contains_link_);
}
public:
InternalLinkMessageDraft(FormattedText &&text, bool contains_link)
: text_(std::move(text)), contains_link_(contains_link) {
}
};
class LinkManager::InternalLinkPassportDataRequest final : public InternalLink {
UserId bot_user_id_;
string scope_;
string public_key_;
string nonce_;
string callback_url_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypePassportDataRequest>(bot_user_id_.get(), scope_, public_key_,
nonce_, callback_url_);
}
public:
InternalLinkPassportDataRequest(UserId bot_user_id, string scope, string public_key, string nonce,
string callback_url)
: bot_user_id_(bot_user_id)
, scope_(std::move(scope))
, public_key_(std::move(public_key))
, nonce_(std::move(nonce))
, callback_url_(std::move(callback_url)) {
}
};
class LinkManager::InternalLinkPremiumFeatures final : public InternalLink {
string referrer_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypePremiumFeatures>(referrer_);
}
public:
explicit InternalLinkPremiumFeatures(string referrer) : referrer_(std::move(referrer)) {
}
};
class LinkManager::InternalLinkPremiumGift final : public InternalLink {
string referrer_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypePremiumGift>(referrer_);
}
public:
explicit InternalLinkPremiumGift(string referrer) : referrer_(std::move(referrer)) {
}
};
class LinkManager::InternalLinkPremiumGiftCode final : public InternalLink {
string code_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypePremiumGiftCode>(code_);
}
public:
explicit InternalLinkPremiumGiftCode(string code) : code_(std::move(code)) {
}
};
class LinkManager::InternalLinkPrivacyAndSecuritySettings final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypePrivacyAndSecuritySettings>();
}
};
class LinkManager::InternalLinkProxy final : public InternalLink {
string server_;
int32 port_;
td_api::object_ptr<td_api::ProxyType> type_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
CHECK(type_ != nullptr);
auto type = type_.get();
auto proxy_type = [type]() -> td_api::object_ptr<td_api::ProxyType> {
switch (type->get_id()) {
case td_api::proxyTypeSocks5::ID: {
auto type_socks = static_cast<const td_api::proxyTypeSocks5 *>(type);
return td_api::make_object<td_api::proxyTypeSocks5>(type_socks->username_, type_socks->password_);
}
case td_api::proxyTypeMtproto::ID: {
auto type_mtproto = static_cast<const td_api::proxyTypeMtproto *>(type);
return td_api::make_object<td_api::proxyTypeMtproto>(type_mtproto->secret_);
}
default:
UNREACHABLE();
return nullptr;
}
}();
return td_api::make_object<td_api::internalLinkTypeProxy>(server_, port_, std::move(proxy_type));
}
public:
InternalLinkProxy(string server, int32 port, td_api::object_ptr<td_api::ProxyType> type)
: server_(std::move(server)), port_(port), type_(std::move(type)) {
}
};
class LinkManager::InternalLinkPublicDialog final : public InternalLink {
string dialog_username_;
string draft_text_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypePublicChat>(dialog_username_, draft_text_);
}
public:
InternalLinkPublicDialog(string dialog_username, string draft_text)
: dialog_username_(std::move(dialog_username)), draft_text_(std::move(draft_text)) {
}
};
class LinkManager::InternalLinkQrCodeAuthentication final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeQrCodeAuthentication>();
}
};
class LinkManager::InternalLinkRestorePurchases final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeRestorePurchases>();
}
};
class LinkManager::InternalLinkSettings final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeSettings>();
}
};
class LinkManager::InternalLinkSideMenuBot final : public InternalLink {
string bot_username_;
string url_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeSideMenuBot>(bot_username_, url_);
}
public:
InternalLinkSideMenuBot(string bot_username, string start_parameter) : bot_username_(std::move(bot_username)) {
if (!start_parameter.empty()) {
url_ = PSTRING() << "start://" << start_parameter;
}
}
};
class LinkManager::InternalLinkStickerSet final : public InternalLink {
string sticker_set_name_;
bool expect_custom_emoji_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeStickerSet>(sticker_set_name_, expect_custom_emoji_);
}
public:
InternalLinkStickerSet(string sticker_set_name, bool expect_custom_emoji)
: sticker_set_name_(std::move(sticker_set_name)), expect_custom_emoji_(expect_custom_emoji) {
}
};
class LinkManager::InternalLinkStory final : public InternalLink {
string story_sender_username_;
StoryId story_id_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeStory>(story_sender_username_, story_id_.get());
}
public:
InternalLinkStory(string story_sender_username, StoryId story_id)
: story_sender_username_(std::move(story_sender_username)), story_id_(story_id) {
}
};
class LinkManager::InternalLinkTheme final : public InternalLink {
string theme_name_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeTheme>(theme_name_);
}
public:
explicit InternalLinkTheme(string theme_name) : theme_name_(std::move(theme_name)) {
}
};
class LinkManager::InternalLinkThemeSettings final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeThemeSettings>();
}
};
class LinkManager::InternalLinkUnknownDeepLink final : public InternalLink {
string link_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeUnknownDeepLink>(link_);
}
public:
explicit InternalLinkUnknownDeepLink(string link) : link_(std::move(link)) {
}
};
class LinkManager::InternalLinkUnsupportedProxy final : public InternalLink {
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeUnsupportedProxy>();
}
};
class LinkManager::InternalLinkUserPhoneNumber final : public InternalLink {
string phone_number_;
string draft_text_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeUserPhoneNumber>(phone_number_, draft_text_);
}
public:
InternalLinkUserPhoneNumber(Slice phone_number, string draft_text)
: phone_number_(PSTRING() << '+' << phone_number), draft_text_(std::move(draft_text)) {
}
};
class LinkManager::InternalLinkUserToken final : public InternalLink {
string token_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeUserToken>(token_);
}
public:
explicit InternalLinkUserToken(string token) : token_(std::move(token)) {
}
};
class LinkManager::InternalLinkVoiceChat final : public InternalLink {
string dialog_username_;
string invite_hash_;
bool is_live_stream_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeVideoChat>(dialog_username_, invite_hash_, is_live_stream_);
}
public:
InternalLinkVoiceChat(string dialog_username, string invite_hash, bool is_live_stream)
: dialog_username_(std::move(dialog_username))
, invite_hash_(std::move(invite_hash))
, is_live_stream_(is_live_stream) {
}
};
class LinkManager::InternalLinkWebApp final : public InternalLink {
string bot_username_;
string web_app_short_name_;
string start_parameter_;
td_api::object_ptr<td_api::InternalLinkType> get_internal_link_type_object() const final {
return td_api::make_object<td_api::internalLinkTypeWebApp>(bot_username_, web_app_short_name_, start_parameter_);
}
public:
InternalLinkWebApp(string bot_username, string web_app_short_name, string start_parameter)
: bot_username_(std::move(bot_username))
, web_app_short_name_(std::move(web_app_short_name))
, start_parameter_(std::move(start_parameter)) {
}
};
class GetDeepLinkInfoQuery final : public Td::ResultHandler {
Promise<td_api::object_ptr<td_api::deepLinkInfo>> promise_;
public:
explicit GetDeepLinkInfoQuery(Promise<td_api::object_ptr<td_api::deepLinkInfo>> &&promise)
: promise_(std::move(promise)) {
}
void send(Slice link) {
send_query(G()->net_query_creator().create_unauth(telegram_api::help_getDeepLinkInfo(link.str())));
}
void on_result(BufferSlice packet) final {
auto result_ptr = fetch_result<telegram_api::help_getDeepLinkInfo>(packet);
if (result_ptr.is_error()) {
return on_error(result_ptr.move_as_error());
}
auto result = result_ptr.move_as_ok();
switch (result->get_id()) {
case telegram_api::help_deepLinkInfoEmpty::ID:
return promise_.set_value(nullptr);
case telegram_api::help_deepLinkInfo::ID: {
auto info = telegram_api::move_object_as<telegram_api::help_deepLinkInfo>(result);
auto text = get_formatted_text(nullptr, std::move(info->message_), std::move(info->entities_), true, true,
"GetDeepLinkInfoQuery");
return promise_.set_value(
td_api::make_object<td_api::deepLinkInfo>(get_formatted_text_object(text, true, -1), info->update_app_));
}
default:
UNREACHABLE();
}
}
void on_error(Status status) final {
promise_.set_error(std::move(status));
}
};
class RequestUrlAuthQuery final : public Td::ResultHandler {
Promise<td_api::object_ptr<td_api::LoginUrlInfo>> promise_;
string url_;
DialogId dialog_id_;
public:
explicit RequestUrlAuthQuery(Promise<td_api::object_ptr<td_api::LoginUrlInfo>> &&promise)
: promise_(std::move(promise)) {
}
void send(string url, MessageFullId message_full_id, int32 button_id) {
url_ = std::move(url);
int32 flags = 0;
tl_object_ptr<telegram_api::InputPeer> input_peer;
if (message_full_id.get_dialog_id().is_valid()) {
dialog_id_ = message_full_id.get_dialog_id();
input_peer = td_->dialog_manager_->get_input_peer(dialog_id_, AccessRights::Read);
CHECK(input_peer != nullptr);
flags |= telegram_api::messages_requestUrlAuth::PEER_MASK;
} else {
flags |= telegram_api::messages_requestUrlAuth::URL_MASK;
}
send_query(G()->net_query_creator().create(telegram_api::messages_requestUrlAuth(
flags, std::move(input_peer), message_full_id.get_message_id().get_server_message_id().get(), button_id,
url_)));
}
void on_result(BufferSlice packet) final {
auto result_ptr = fetch_result<telegram_api::messages_requestUrlAuth>(packet);
if (result_ptr.is_error()) {
return on_error(result_ptr.move_as_error());
}
auto result = result_ptr.move_as_ok();
LOG(INFO) << "Receive result for RequestUrlAuthQuery: " << to_string(result);
switch (result->get_id()) {
case telegram_api::urlAuthResultRequest::ID: {
auto request = telegram_api::move_object_as<telegram_api::urlAuthResultRequest>(result);
UserId bot_user_id = UserManager::get_user_id(request->bot_);
if (!bot_user_id.is_valid()) {
return on_error(Status::Error(500, "Receive invalid bot_user_id"));
}
td_->user_manager_->on_get_user(std::move(request->bot_), "RequestUrlAuthQuery");
promise_.set_value(td_api::make_object<td_api::loginUrlInfoRequestConfirmation>(
url_, request->domain_, td_->user_manager_->get_user_id_object(bot_user_id, "RequestUrlAuthQuery"),
request->request_write_access_));
break;
}
case telegram_api::urlAuthResultAccepted::ID: {
auto accepted = telegram_api::move_object_as<telegram_api::urlAuthResultAccepted>(result);
promise_.set_value(td_api::make_object<td_api::loginUrlInfoOpen>(accepted->url_, true));
break;
}
case telegram_api::urlAuthResultDefault::ID:
promise_.set_value(td_api::make_object<td_api::loginUrlInfoOpen>(url_, false));
break;
}
}
void on_error(Status status) final {
if (!dialog_id_.is_valid() ||
!td_->dialog_manager_->on_get_dialog_error(dialog_id_, status, "RequestUrlAuthQuery")) {
LOG(INFO) << "Receive error for RequestUrlAuthQuery: " << status;
}
promise_.set_value(td_api::make_object<td_api::loginUrlInfoOpen>(url_, false));
}
};
class AcceptUrlAuthQuery final : public Td::ResultHandler {
Promise<td_api::object_ptr<td_api::httpUrl>> promise_;
string url_;
DialogId dialog_id_;
public:
explicit AcceptUrlAuthQuery(Promise<td_api::object_ptr<td_api::httpUrl>> &&promise) : promise_(std::move(promise)) {
}
void send(string url, MessageFullId message_full_id, int32 button_id, bool allow_write_access) {
url_ = std::move(url);
int32 flags = 0;
tl_object_ptr<telegram_api::InputPeer> input_peer;
if (message_full_id.get_dialog_id().is_valid()) {
dialog_id_ = message_full_id.get_dialog_id();
input_peer = td_->dialog_manager_->get_input_peer(dialog_id_, AccessRights::Read);
CHECK(input_peer != nullptr);
flags |= telegram_api::messages_acceptUrlAuth::PEER_MASK;
} else {
flags |= telegram_api::messages_acceptUrlAuth::URL_MASK;
}
if (allow_write_access) {
flags |= telegram_api::messages_acceptUrlAuth::WRITE_ALLOWED_MASK;
}
send_query(G()->net_query_creator().create(telegram_api::messages_acceptUrlAuth(
flags, false /*ignored*/, std::move(input_peer), message_full_id.get_message_id().get_server_message_id().get(),
button_id, url_)));
}
void on_result(BufferSlice packet) final {
auto result_ptr = fetch_result<telegram_api::messages_acceptUrlAuth>(packet);
if (result_ptr.is_error()) {
return on_error(result_ptr.move_as_error());
}
auto result = result_ptr.move_as_ok();
LOG(INFO) << "Receive " << to_string(result);
switch (result->get_id()) {
case telegram_api::urlAuthResultRequest::ID:
LOG(ERROR) << "Receive unexpected " << to_string(result);
return on_error(Status::Error(500, "Receive unexpected urlAuthResultRequest"));
case telegram_api::urlAuthResultAccepted::ID: {
auto accepted = telegram_api::move_object_as<telegram_api::urlAuthResultAccepted>(result);
promise_.set_value(td_api::make_object<td_api::httpUrl>(accepted->url_));
break;
}
case telegram_api::urlAuthResultDefault::ID:
promise_.set_value(td_api::make_object<td_api::httpUrl>(url_));
break;
}
}
void on_error(Status status) final {
if (!dialog_id_.is_valid() ||
!td_->dialog_manager_->on_get_dialog_error(dialog_id_, status, "AcceptUrlAuthQuery")) {
LOG(INFO) << "Receive error for AcceptUrlAuthQuery: " << status;
}
promise_.set_error(std::move(status));
}
};
LinkManager::LinkManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
}
LinkManager::~LinkManager() = default;
void LinkManager::start_up() {
autologin_update_time_ = Time::now() - 365 * 86400;
autologin_domains_ = full_split(G()->td_db()->get_binlog_pmc()->get("autologin_domains"), '\xFF');
url_auth_domains_ = full_split(G()->td_db()->get_binlog_pmc()->get("url_auth_domains"), '\xFF');
whitelisted_domains_ = full_split(G()->td_db()->get_binlog_pmc()->get("whitelisted_domains"), '\xFF');
}
void LinkManager::tear_down() {
parent_.reset();
}
static bool tolower_begins_with(Slice str, Slice prefix) {
if (prefix.size() > str.size()) {
return false;
}
for (size_t i = 0; i < prefix.size(); i++) {
if (to_lower(str[i]) != prefix[i]) {
return false;
}
}
return true;
}
Result<string> LinkManager::check_link(CSlice link, bool http_only, bool https_only) {
auto result = check_link_impl(link, http_only, https_only);
if (result.is_ok()) {
return result;
}
auto error = result.move_as_error();
if (check_utf8(link)) {
return Status::Error(400, PSLICE() << "URL '" << link << "' is invalid: " << error.message());
} else {
return Status::Error(400, PSLICE() << "URL is invalid: " << error.message());
}
}
string LinkManager::get_checked_link(Slice link, bool http_only, bool https_only) {
auto result = check_link_impl(link, http_only, https_only);
if (result.is_ok()) {
return result.move_as_ok();
}
return string();
}
Result<string> LinkManager::check_link_impl(Slice link, bool http_only, bool https_only) {
bool is_tg = false;
bool is_ton = false;
if (tolower_begins_with(link, "tg:")) {
link.remove_prefix(3);
is_tg = true;
} else if (tolower_begins_with(link, "ton:")) {
link.remove_prefix(4);
is_ton = true;
}
if ((is_tg || is_ton) && begins_with(link, "//")) {
link.remove_prefix(2);
}
TRY_RESULT(http_url, parse_url(link));
if (https_only && (http_url.protocol_ != HttpUrl::Protocol::Https || is_tg || is_ton)) {
return Status::Error("Only HTTPS links are allowed");
}
if (is_tg || is_ton) {
if (http_only) {
return Status::Error("Only HTTP links are allowed");
}
if (tolower_begins_with(link, "http://") || http_url.protocol_ == HttpUrl::Protocol::Https ||
!http_url.userinfo_.empty() || http_url.specified_port_ != 0 || http_url.is_ipv6_) {
return Status::Error(is_tg ? Slice("Wrong tg URL") : Slice("Wrong ton URL"));
}
Slice query(http_url.query_);
CHECK(query[0] == '/');
if (query.size() > 1 && query[1] == '?') {
query.remove_prefix(1);
}
for (auto c : http_url.host_) {
if (!is_alnum(c) && c != '-' && c != '_') {
return Status::Error("Unallowed characters in URL host");
}
}
return PSTRING() << (is_tg ? "tg" : "ton") << "://" << http_url.host_ << query;
}
if (http_url.host_.find('.') == string::npos && !http_url.is_ipv6_) {
return Status::Error("Wrong HTTP URL");
}
return http_url.get_url();
}
LinkManager::LinkInfo LinkManager::get_link_info(Slice link) {
LinkInfo result;
if (link.empty()) {
return result;
}
link.truncate(link.find('#'));
bool is_tg = false;
if (tolower_begins_with(link, "tg:")) {
link.remove_prefix(3);
if (begins_with(link, "//")) {
link.remove_prefix(2);
}
is_tg = true;
}
auto r_http_url = parse_url(link);
if (r_http_url.is_error()) {
return result;
}
auto http_url = r_http_url.move_as_ok();
if (!http_url.userinfo_.empty() || http_url.is_ipv6_) {
return result;
}
if (is_tg) {
if (tolower_begins_with(link, "http://") || http_url.protocol_ == HttpUrl::Protocol::Https ||
http_url.specified_port_ != 0) {
return result;
}
result.type_ = LinkType::Tg;
result.query_ = link.str();
return result;
} else {
if (http_url.port_ != 80 && http_url.port_ != 443) {
return result;
}
auto host = url_decode(http_url.host_, false);
to_lower_inplace(host);
if (ends_with(host, ".t.me") && host.size() >= 9 && host.find('.') == host.size() - 5) {
Slice subdomain(&host[0], host.size() - 5);
static const FlatHashSet<Slice, SliceHash> disallowed_subdomains(
{"addemoji", "addlist", "addstickers", "addtheme", "auth", "boost", "confirmphone",
"contact", "giftcode", "invoice", "joinchat", "login", "m", "proxy",
"setlanguage", "share", "socks", "web", "a", "k", "z"});
if (is_valid_username(subdomain) && disallowed_subdomains.count(subdomain) == 0) {
result.type_ = LinkType::TMe;
result.query_ = PSTRING() << '/' << subdomain << http_url.query_;
return result;
}
}
if (begins_with(host, "www.")) {
host = host.substr(4);
}
string cur_t_me_url;
vector<Slice> t_me_urls{Slice("t.me"), Slice("telegram.me"), Slice("telegram.dog")};
#if TD_EMSCRIPTEN
t_me_urls.push_back(Slice("web.t.me"));
t_me_urls.push_back(Slice("a.t.me"));
t_me_urls.push_back(Slice("k.t.me"));
t_me_urls.push_back(Slice("z.t.me"));
#endif
if (Scheduler::context() != nullptr) { // for tests only
cur_t_me_url = G()->get_option_string("t_me_url");
if (tolower_begins_with(cur_t_me_url, "http://") || tolower_begins_with(cur_t_me_url, "https://")) {
Slice t_me_url = cur_t_me_url;
t_me_url = t_me_url.substr(t_me_url[4] == 's' ? 8 : 7);
if (!td::contains(t_me_urls, t_me_url)) {
t_me_urls.push_back(t_me_url);
}
}
}
for (auto t_me_url : t_me_urls) {
if (host == t_me_url) {
result.type_ = LinkType::TMe;
Slice query = http_url.query_;
while (true) {
if (begins_with(query, "/s/")) {
query.remove_prefix(2);
continue;
}
if (begins_with(query, "/%73/")) {
query.remove_prefix(4);
continue;
}
break;
}
result.query_ = query.str();
return result;
}
}
if (http_url.query_.size() > 1) {
for (auto telegraph_url : {Slice("telegra.ph"), Slice("te.legra.ph"), Slice("graph.org")}) {
if (host == telegraph_url) {
result.type_ = LinkType::Telegraph;
result.query_ = std::move(http_url.query_);
return result;
}
}
}
}
return result;
}
bool LinkManager::is_internal_link(Slice link) {
auto info = get_link_info(link);
return info.type_ != LinkType::External;
}
unique_ptr<LinkManager::InternalLink> LinkManager::parse_internal_link(Slice link, bool is_trusted) {
auto info = get_link_info(link);
switch (info.type_) {
case LinkType::External:
return nullptr;
case LinkType::Tg:
return parse_tg_link_query(info.query_, is_trusted);
case LinkType::TMe:
return parse_t_me_link_query(info.query_, is_trusted);
case LinkType::Telegraph:
return td::make_unique<InternalLinkInstantView>(PSTRING() << "https://telegra.ph" << info.query_, link.str());
default:
UNREACHABLE();
return nullptr;
}
}
namespace {
struct CopyArg {
Slice name_;
const HttpUrlQuery *url_query_;
bool *is_first_;
CopyArg(Slice name, const HttpUrlQuery *url_query, bool *is_first)
: name_(name), url_query_(url_query), is_first_(is_first) {
}
};
StringBuilder &operator<<(StringBuilder &string_builder, const CopyArg &copy_arg) {
auto arg = copy_arg.url_query_->get_arg(copy_arg.name_);
if (arg.empty()) {
for (const auto &query_arg : copy_arg.url_query_->args_) {
if (query_arg.first == copy_arg.name_) {
char c = *copy_arg.is_first_ ? '?' : '&';
*copy_arg.is_first_ = false;
return string_builder << c << copy_arg.name_;
}
}
return string_builder;
}
char c = *copy_arg.is_first_ ? '?' : '&';
*copy_arg.is_first_ = false;
return string_builder << c << copy_arg.name_ << '=' << url_encode(arg);
}
} // namespace
unique_ptr<LinkManager::InternalLink> LinkManager::parse_tg_link_query(Slice query, bool is_trusted) {
const auto url_query = parse_url_query(query);
const auto &path = url_query.path_;
bool is_first_arg = true;
auto copy_arg = [&](Slice name) {
return CopyArg(name, &url_query, &is_first_arg);
};
auto pass_arg = [&](Slice name) {
return url_encode(url_query.get_arg(name));
};
auto get_arg = [&](Slice name) {
return url_query.get_arg(name).str();
};
auto has_arg = [&](Slice name) {
return !url_query.get_arg(name).empty();
};
if (path.size() == 1 && path[0] == "resolve") {
auto username = get_arg("domain");
if (is_valid_username(username)) {
if (has_arg("post")) {
// resolve?domain=<username>&post=12345&single&thread=<thread_id>&comment=<message_id>&t=<media_timestamp>
return td::make_unique<InternalLinkMessage>(
PSTRING() << "tg://resolve" << copy_arg("domain") << copy_arg("post") << copy_arg("single")
<< copy_arg("thread") << copy_arg("comment") << copy_arg("t"));
}
for (auto &arg : url_query.args_) {
if (arg.first == "voicechat" || arg.first == "videochat" || arg.first == "livestream") {
// resolve?domain=<username>&videochat
// resolve?domain=<username>&videochat=<invite_hash>
if (Scheduler::context() != nullptr) {
send_closure(G()->dialog_manager(), &DialogManager::reload_voice_chat_on_search, username);
}
return td::make_unique<InternalLinkVoiceChat>(std::move(username), arg.second, arg.first == "livestream");
}
if (arg.first == "start" && is_valid_start_parameter(arg.second)) {
// resolve?domain=<bot_username>&start=<parameter>
return td::make_unique<InternalLinkBotStart>(std::move(username), arg.second, is_trusted);
}
if (arg.first == "startgroup" && is_valid_start_parameter(arg.second)) {
// resolve?domain=<bot_username>&startgroup=<parameter>
// resolve?domain=<bot_username>&startgroup=>parameter>&admin=change_info+delete_messages+restrict_members
// resolve?domain=<bot_username>&startgroup&admin=change_info+delete_messages+restrict_members
auto administrator_rights = get_administrator_rights(url_query.get_arg("admin"), false);
return td::make_unique<InternalLinkBotStartInGroup>(std::move(username), arg.second,
std::move(administrator_rights));
}
if (arg.first == "startchannel") {
// resolve?domain=<bot_username>&startchannel&admin=change_info+post_messages+promote_members
auto administrator_rights = get_administrator_rights(url_query.get_arg("admin"), true);
if (administrator_rights != AdministratorRights()) {
return td::make_unique<InternalLinkBotAddToChannel>(std::move(username), std::move(administrator_rights));
}
}
if (arg.first == "game" && is_valid_game_name(arg.second)) {
// resolve?domain=<bot_username>&game=<short_name>
return td::make_unique<InternalLinkGame>(std::move(username), arg.second);
}
if (arg.first == "appname" && is_valid_web_app_name(arg.second)) {
// resolve?domain=<bot_username>&appname=<app_name>
// resolve?domain=<bot_username>&appname=<app_name>&startapp=<start_parameter>
return td::make_unique<InternalLinkWebApp>(std::move(username), arg.second,
url_query.get_arg("startapp").str());
}
if (arg.first == "story" && is_valid_story_id(arg.second)) {
// resolve?domain=<username>&story=<story_id>
return td::make_unique<InternalLinkStory>(std::move(username), StoryId(to_integer<int32>(arg.second)));
}
}
if (url_query.has_arg("startapp") && !url_query.has_arg("appname")) {
// resolve?domain=<bot_username>&startapp=
// resolve?domain=<bot_username>&startapp=<start_parameter>
return td::make_unique<InternalLinkSideMenuBot>(std::move(username), url_query.get_arg("startapp").str());
}
if (!url_query.get_arg("attach").empty()) {
// resolve?domain=<username>&attach=<bot_username>
// resolve?domain=<username>&attach=<bot_username>&startattach=<start_parameter>
return td::make_unique<InternalLinkAttachMenuBot>(
nullptr, td::make_unique<InternalLinkPublicDialog>(std::move(username), string()),
url_query.get_arg("attach").str(), url_query.get_arg("startattach"));
} else if (url_query.has_arg("startattach")) {
// resolve?domain=<bot_username>&startattach&choose=users+bots+groups+channels
// resolve?domain=<bot_username>&startattach=<start_parameter>&choose=users+bots+groups+channels
return td::make_unique<InternalLinkAttachMenuBot>(get_target_chat_chosen(url_query.get_arg("choose")), nullptr,
std::move(username), url_query.get_arg("startattach"));
}
if (username == "telegrampassport") {
// resolve?domain=telegrampassport&bot_id=...&scope=...&public_key=...&nonce=...&callback_url=...
auto passport_link = get_internal_link_passport(query, url_query.args_, false);
if (passport_link != nullptr) {
return passport_link;
}
}
// resolve?domain=<username>
return td::make_unique<InternalLinkPublicDialog>(std::move(username), get_url_query_draft_text(url_query));
} else {
string phone_number_str = get_arg("phone");
auto phone_number = phone_number_str[0] == ' ' ? Slice(phone_number_str).substr(1) : Slice(phone_number_str);
if (is_valid_phone_number(phone_number)) {
if (!url_query.get_arg("attach").empty()) {
// resolve?phone=<phone_number>&attach=<bot_username>
// resolve?phone=<phone_number>&attach=<bot_username>&startattach=<start_parameter>
return td::make_unique<InternalLinkAttachMenuBot>(
nullptr, td::make_unique<InternalLinkUserPhoneNumber>(phone_number, string()),
url_query.get_arg("attach").str(), url_query.get_arg("startattach"));
}
// resolve?phone=12345
return td::make_unique<InternalLinkUserPhoneNumber>(phone_number, get_url_query_draft_text(url_query));
}
}
} else if (path.size() == 1 && path[0] == "contact") {
// contact?token=<token>
if (has_arg("token")) {
return td::make_unique<InternalLinkUserToken>(get_arg("token"));
}
} else if (path.size() == 1 && path[0] == "login") {
// login?code=123456
if (has_arg("code")) {
return td::make_unique<InternalLinkAuthenticationCode>(get_arg("code"));
}
// login?token=<token>
if (has_arg("token")) {
return td::make_unique<InternalLinkQrCodeAuthentication>();
}
} else if (path.size() == 1 && path[0] == "restore_purchases") {
// restore_purchases
return td::make_unique<InternalLinkRestorePurchases>();
} else if (path.size() == 1 && path[0] == "passport") {
// passport?bot_id=...&scope=...&public_key=...&nonce=...&callback_url=...
return get_internal_link_passport(query, url_query.args_, true);
} else if (path.size() == 1 && path[0] == "premium_offer") {
// premium_offer?ref=<referrer>
return td::make_unique<InternalLinkPremiumFeatures>(get_arg("ref"));
} else if (path.size() == 1 && path[0] == "premium_multigift") {
// premium_multigift?ref=<referrer>
return td::make_unique<InternalLinkPremiumGift>(get_arg("ref"));
} else if (!path.empty() && path[0] == "settings") {
if (path.size() == 2 && path[1] == "auto_delete") {
// settings/auto_delete
return td::make_unique<InternalLinkDefaultMessageAutoDeleteTimerSettings>();
}
if (path.size() == 2 && path[1] == "change_number") {
// settings/change_number
return td::make_unique<InternalLinkChangePhoneNumber>();
}
if (path.size() == 2 && path[1] == "devices") {
// settings/devices
return td::make_unique<InternalLinkActiveSessions>();
}
if (path.size() == 2 && path[1] == "edit_profile") {
// settings/edit_profile
return td::make_unique<InternalLinkEditProfileSettings>();
}
if (path.size() == 2 && path[1] == "folders") {
// settings/folders
return td::make_unique<InternalLinkDialogFolderSettings>();
}
if (path.size() == 2 && path[1] == "language") {
// settings/language
return td::make_unique<InternalLinkLanguageSettings>();
}
if (path.size() == 2 && path[1] == "privacy") {
// settings/privacy
return td::make_unique<InternalLinkPrivacyAndSecuritySettings>();
}
if (path.size() == 2 && path[1] == "themes") {
// settings/themes
return td::make_unique<InternalLinkThemeSettings>();
}
// settings
return td::make_unique<InternalLinkSettings>();
} else if (path.size() == 1 && path[0] == "addlist") {
auto slug = get_url_query_slug(true, url_query);
if (!slug.empty() && is_base64url_characters(slug)) {
// addlist?slug=<slug>
return td::make_unique<InternalLinkDialogFolderInvite>(get_dialog_filter_invite_link(slug, true));
}
} else if (path.size() == 1 && path[0] == "join") {
auto invite_hash = get_url_query_hash(true, url_query);
if (!invite_hash.empty() && !is_valid_phone_number(invite_hash) && is_base64url_characters(invite_hash)) {
// join?invite=<hash>
return td::make_unique<InternalLinkDialogInvite>(get_dialog_invite_link(invite_hash, true));
}
} else if (path.size() == 1 && (path[0] == "addstickers" || path[0] == "addemoji")) {
// addstickers?set=<name>
// addemoji?set=<name>
if (has_arg("set")) {
return td::make_unique<InternalLinkStickerSet>(get_arg("set"), path[0] == "addemoji");
}
} else if (path.size() == 1 && path[0] == "setlanguage") {
// setlanguage?lang=<name>
if (has_arg("lang")) {
return td::make_unique<InternalLinkLanguage>(get_arg("lang"));
}
} else if (path.size() == 1 && path[0] == "addtheme") {
// addtheme?slug=<name>
if (has_arg("slug")) {
return td::make_unique<InternalLinkTheme>(get_arg("slug"));
}
} else if (path.size() == 1 && path[0] == "confirmphone") {
if (has_arg("hash") && has_arg("phone")) {
// confirmphone?phone=<phone>&hash=<hash>
return td::make_unique<InternalLinkConfirmPhone>(get_arg("hash"), get_arg("phone"));
}
} else if (path.size() == 1 && path[0] == "socks") {
if (has_arg("server") && has_arg("port")) {
// socks?server=<server>&port=<port>&user=<user>&pass=<pass>
auto port = to_integer<int32>(get_arg("port"));
if (0 < port && port < 65536) {
return td::make_unique<InternalLinkProxy>(
get_arg("server"), port, td_api::make_object<td_api::proxyTypeSocks5>(get_arg("user"), get_arg("pass")));
} else {
return td::make_unique<InternalLinkUnsupportedProxy>();
}
}
} else if (path.size() == 1 && path[0] == "proxy") {
if (has_arg("server") && has_arg("port")) {
// proxy?server=<server>&port=<port>&secret=<secret>
auto port = to_integer<int32>(get_arg("port"));
auto r_secret = mtproto::ProxySecret::from_link(get_arg("secret"));
if (0 < port && port < 65536 && r_secret.is_ok()) {
return td::make_unique<InternalLinkProxy>(
get_arg("server"), port, td_api::make_object<td_api::proxyTypeMtproto>(r_secret.ok().get_encoded_secret()));
} else {
return td::make_unique<InternalLinkUnsupportedProxy>();
}
}
} else if (path.size() == 1 && path[0] == "privatepost") {
// privatepost?channel=123456789&post=12345&single&thread=<thread_id>&comment=<message_id>&t=<media_timestamp>
if (has_arg("channel") && has_arg("post")) {
return td::make_unique<InternalLinkMessage>(
PSTRING() << "tg://privatepost" << copy_arg("channel") << copy_arg("post") << copy_arg("single")
<< copy_arg("thread") << copy_arg("comment") << copy_arg("t"));
}
} else if (path.size() == 1 && path[0] == "boost") {
// boost?domain=channel_username
// boost?channel=123456
if (has_arg("domain")) {
return td::make_unique<InternalLinkDialogBoost>(PSTRING() << "tg://boost" << copy_arg("domain"));
}
if (has_arg("channel")) {
return td::make_unique<InternalLinkDialogBoost>(PSTRING() << "tg://boost" << copy_arg("channel"));
}
} else if (path.size() == 1 && path[0] == "bg") {
// bg?color=<color>
// bg?gradient=<hex_color>-<hex_color>&rotation=...
// bg?gradient=<hex_color>~<hex_color>~<hex_color>~<hex_color>
// bg?slug=<background_name>&mode=blur+motion
// bg?slug=<pattern_name>&intensity=...&bg_color=...&mode=blur+motion
if (has_arg("color")) {
return td::make_unique<InternalLinkBackground>(pass_arg("color"));
}
if (has_arg("gradient")) {
return td::make_unique<InternalLinkBackground>(PSTRING() << pass_arg("gradient") << copy_arg("rotation"));
}
if (has_arg("slug")) {
return td::make_unique<InternalLinkBackground>(PSTRING()
<< pass_arg("slug") << copy_arg("mode") << copy_arg("intensity")
<< copy_arg("bg_color") << copy_arg("rotation"));
}
} else if (path.size() == 1 && path[0] == "invoice") {
// invoice?slug=<invoice_name>
if (has_arg("slug")) {
return td::make_unique<InternalLinkInvoice>(url_query.get_arg("slug").str());
}
} else if (path.size() == 1 && path[0] == "giftcode") {
// giftcode?slug=<code>
if (has_arg("slug")) {
return td::make_unique<InternalLinkPremiumGiftCode>(url_query.get_arg("slug").str());
}
} else if (path.size() == 1 && path[0] == "message") {
// message?slug=<name>
if (has_arg("slug")) {
return td::make_unique<InternalLinkBusinessChat>(url_query.get_arg("slug").str());
}
} else if (path.size() == 1 && (path[0] == "share" || path[0] == "msg" || path[0] == "msg_url")) {
// msg_url?url=<url>
// msg_url?url=<url>&text=<text>
return get_internal_link_message_draft(get_arg("url"), get_arg("text"));
}
if (!path.empty() && !path[0].empty()) {
return td::make_unique<InternalLinkUnknownDeepLink>(PSTRING() << "tg://" << query);
}
return nullptr;
}
unique_ptr<LinkManager::InternalLink> LinkManager::parse_t_me_link_query(Slice query, bool is_trusted) {
CHECK(query[0] == '/');
const auto url_query = parse_url_query(query);
const auto &path = url_query.path_;
if (path.empty() || path[0].empty()) {
return nullptr;
}
bool is_first_arg = true;
auto copy_arg = [&](Slice name) {
return CopyArg(name, &url_query, &is_first_arg);
};
auto get_arg = [&](Slice name) {
return url_query.get_arg(name).str();
};
auto has_arg = [&](Slice name) {
return !url_query.get_arg(name).empty();
};
if (path[0] == "c") {
if (path.size() >= 3 && to_integer<int64>(path[1]) > 0 && to_integer<int64>(path[2]) > 0) {
// /c/123456789/12345?single&thread=<thread_id>&comment=<message_id>&t=<media_timestamp>
// /c/123456789/1234/12345?single&comment=<message_id>&t=<media_timestamp>
is_first_arg = false;
auto post = to_integer<int64>(path[2]);
auto thread = PSTRING() << copy_arg("thread");
if (path.size() >= 4 && to_integer<int64>(path[3]) > 0) {
thread = PSTRING() << "&thread=" << post;
post = to_integer<int64>(path[3]);
}
return td::make_unique<InternalLinkMessage>(PSTRING() << "tg://privatepost?channel=" << to_integer<int64>(path[1])
<< "&post=" << post << copy_arg("single") << thread
<< copy_arg("comment") << copy_arg("t"));
} else if (path.size() >= 2 && to_integer<int64>(path[1]) > 0 && url_query.has_arg("boost")) {
// /c/123456789?boost
return td::make_unique<InternalLinkDialogBoost>(PSTRING() << "tg://boost?channel=" << to_integer<int64>(path[1]));
}
} else if (path[0] == "login") {
if (path.size() >= 2 && !path[1].empty()) {
// /login/<code>
return td::make_unique<InternalLinkAuthenticationCode>(path[1]);
}
} else if (path[0] == "addlist") {
auto slug = get_url_query_slug(false, url_query);
if (!slug.empty() && is_base64url_characters(slug)) {
// /addlist/<slug>
return td::make_unique<InternalLinkDialogFolderInvite>(get_dialog_filter_invite_link(slug, true));
}
} else if (path[0] == "joinchat") {
auto invite_hash = get_url_query_hash(false, url_query);
if (!invite_hash.empty() && !is_valid_phone_number(invite_hash) && is_base64url_characters(invite_hash)) {
// /joinchat/<hash>
return td::make_unique<InternalLinkDialogInvite>(get_dialog_invite_link(invite_hash, true));
}
} else if (path[0][0] == ' ' || path[0][0] == '+') {
auto invite_hash = get_url_query_hash(false, url_query);
if (is_valid_phone_number(invite_hash)) {
if (!url_query.get_arg("attach").empty()) {
// /+<phone_number>?attach=<bot_username>
// /+<phone_number>?attach=<bot_username>&startattach=<start_parameter>
return td::make_unique<InternalLinkAttachMenuBot>(
nullptr, td::make_unique<InternalLinkUserPhoneNumber>(invite_hash, string()),
url_query.get_arg("attach").str(), url_query.get_arg("startattach"));
}
// /+<phone_number>
return td::make_unique<InternalLinkUserPhoneNumber>(invite_hash, get_url_query_draft_text(url_query));
} else if (!invite_hash.empty() && is_base64url_characters(invite_hash)) {
// /+<link>
return td::make_unique<InternalLinkDialogInvite>(get_dialog_invite_link(invite_hash, true));
}
} else if (path[0] == "contact") {
if (path.size() >= 2 && !path[1].empty()) {
// /contact/<token>
return td::make_unique<InternalLinkUserToken>(path[1]);
}
} else if (path[0] == "addstickers" || path[0] == "addemoji") {
if (path.size() >= 2 && !path[1].empty()) {
// /addstickers/<name>
// /addemoji/<name>
return td::make_unique<InternalLinkStickerSet>(path[1], path[0] == "addemoji");
}
} else if (path[0] == "setlanguage") {
if (path.size() >= 2 && !path[1].empty()) {
// /setlanguage/<name>
return td::make_unique<InternalLinkLanguage>(path[1]);
}
} else if (path[0] == "addtheme") {
if (path.size() >= 2 && !path[1].empty()) {
// /addtheme/<name>
return td::make_unique<InternalLinkTheme>(path[1]);
}
} else if (path[0] == "confirmphone") {
if (has_arg("hash") && has_arg("phone")) {
// /confirmphone?phone=<phone>&hash=<hash>
return td::make_unique<InternalLinkConfirmPhone>(get_arg("hash"), get_arg("phone"));
}
} else if (path[0] == "socks") {
if (has_arg("server") && has_arg("port")) {
// /socks?server=<server>&port=<port>&user=<user>&pass=<pass>
auto port = to_integer<int32>(get_arg("port"));
if (0 < port && port < 65536) {
return td::make_unique<InternalLinkProxy>(
get_arg("server"), port, td_api::make_object<td_api::proxyTypeSocks5>(get_arg("user"), get_arg("pass")));
} else {
return td::make_unique<InternalLinkUnsupportedProxy>();
}
}
} else if (path[0] == "proxy") {
if (has_arg("server") && has_arg("port")) {
// /proxy?server=<server>&port=<port>&secret=<secret>
auto port = to_integer<int32>(get_arg("port"));
auto r_secret = mtproto::ProxySecret::from_link(get_arg("secret"));
if (0 < port && port < 65536 && r_secret.is_ok()) {
return td::make_unique<InternalLinkProxy>(
get_arg("server"), port, td_api::make_object<td_api::proxyTypeMtproto>(r_secret.ok().get_encoded_secret()));
} else {
return td::make_unique<InternalLinkUnsupportedProxy>();
}
}
} else if (path[0] == "bg") {
if (path.size() >= 2 && !path[1].empty()) {
// /bg/<hex_color>
// /bg/<hex_color>-<hex_color>?rotation=...
// /bg/<hex_color>~<hex_color>~<hex_color>~<hex_color>
// /bg/<background_name>?mode=blur+motion
// /bg/<pattern_name>?intensity=...&bg_color=...&mode=blur+motion
if (BackgroundType::is_background_name_local(path[1])) {
return td::make_unique<InternalLinkBackground>(PSTRING() << url_encode(path[1]) << copy_arg("rotation"));
}
return td::make_unique<InternalLinkBackground>(PSTRING()
<< url_encode(path[1]) << copy_arg("mode") << copy_arg("intensity")
<< copy_arg("bg_color") << copy_arg("rotation"));
}
} else if (path[0] == "invoice") {
if (path.size() >= 2 && !path[1].empty()) {
// /invoice/<name>
return td::make_unique<InternalLinkInvoice>(path[1]);
}
} else if (path[0] == "giftcode") {
if (path.size() >= 2 && !path[1].empty()) {
// /giftcode/<code>
return td::make_unique<InternalLinkPremiumGiftCode>(path[1]);
}
} else if (path[0] == "m") {
if (path.size() >= 2 && !path[1].empty()) {
// /m/<link_name>
return td::make_unique<InternalLinkBusinessChat>(path[1]);
}
} else if (path[0][0] == '$') {
if (path[0].size() >= 2) {
// /$<invoice_name>
return td::make_unique<InternalLinkInvoice>(path[0].substr(1));
}
} else if (path[0] == "share" || path[0] == "msg") {
if (!(path.size() > 1 && (path[1] == "bookmarklet" || path[1] == "embed"))) {
// /share?url=<url>
// /share?url=<url>&text=<text>
return get_internal_link_message_draft(get_arg("url"), get_arg("text"));
}
} else if (path[0] == "iv") {
if (path.size() == 1 && has_arg("url")) {
// /iv?url=<url>&rhash=<rhash>
return td::make_unique<InternalLinkInstantView>(
PSTRING() << get_t_me_url() << "iv" << copy_arg("url") << copy_arg("rhash"), get_arg("url"));
}
} else if (is_valid_username(path[0])) {
if (path.size() >= 2 && to_integer<int64>(path[1]) > 0) {
// /<username>/12345?single&thread=<thread_id>&comment=<message_id>&t=<media_timestamp>
// /<username>/1234/12345?single&comment=<message_id>&t=<media_timestamp>
is_first_arg = false;
auto post = to_integer<int64>(path[1]);
auto thread = PSTRING() << copy_arg("thread");
if (path.size() >= 3 && to_integer<int64>(path[2]) > 0) {
thread = PSTRING() << "&thread=" << post;
post = to_integer<int64>(path[2]);
}
return td::make_unique<InternalLinkMessage>(PSTRING() << "tg://resolve?domain=" << url_encode(path[0])
<< "&post=" << post << copy_arg("single") << thread
<< copy_arg("comment") << copy_arg("t"));
}
auto username = path[0];
if (to_lower(username) == "boost") {
if (path.size() == 2 && is_valid_username(path[1])) {
// /boost/<username>
return td::make_unique<InternalLinkDialogBoost>(PSTRING() << "tg://boost?domain=" << url_encode(path[1]));
}
auto channel_id = url_query.get_arg("c");
if (path.size() == 1 && to_integer<int64>(channel_id) > 0) {
// /boost?c=<channel_id>
return td::make_unique<InternalLinkDialogBoost>(PSTRING()
<< "tg://boost?channel=" << to_integer<int64>(channel_id));
}
}
if (path.size() == 3 && path[1] == "s" && is_valid_story_id(path[2])) {
// /<username>/s/<story_id>
return td::make_unique<InternalLinkStory>(std::move(username), StoryId(to_integer<int32>(path[2])));
}
if (path.size() == 2 && is_valid_web_app_name(path[1])) {
// /<username>/<web_app_name>
// /<username>/<web_app_name>?startapp=<start_parameter>
return td::make_unique<InternalLinkWebApp>(std::move(username), path[1], url_query.get_arg("startapp").str());
}
for (auto &arg : url_query.args_) {
if (arg.first == "voicechat" || arg.first == "videochat" || arg.first == "livestream") {
// /<username>?videochat
// /<username>?videochat=<invite_hash>
if (Scheduler::context() != nullptr) {
send_closure(G()->dialog_manager(), &DialogManager::reload_voice_chat_on_search, username);
}
return td::make_unique<InternalLinkVoiceChat>(std::move(username), arg.second, arg.first == "livestream");
}
if (arg.first == "boost") {
// /<username>?boost
return td::make_unique<InternalLinkDialogBoost>(PSTRING() << "tg://boost?domain=" << url_encode(username));
}
if (arg.first == "start" && is_valid_start_parameter(arg.second)) {
// /<bot_username>?start=<parameter>
return td::make_unique<InternalLinkBotStart>(std::move(username), arg.second, is_trusted);
}
if (arg.first == "startgroup" && is_valid_start_parameter(arg.second)) {
// /<bot_username>?startgroup=<parameter>
// /<bot_username>?startgroup=<parameter>&admin=change_info+delete_messages+restrict_members
// /<bot_username>?startgroup&admin=change_info+delete_messages+restrict_members
auto administrator_rights = get_administrator_rights(url_query.get_arg("admin"), false);
return td::make_unique<InternalLinkBotStartInGroup>(std::move(username), arg.second,
std::move(administrator_rights));
}
if (arg.first == "startchannel") {
// /<bot_username>?startchannel&admin=change_info+post_messages+promote_members
auto administrator_rights = get_administrator_rights(url_query.get_arg("admin"), true);
if (administrator_rights != AdministratorRights()) {
return td::make_unique<InternalLinkBotAddToChannel>(std::move(username), std::move(administrator_rights));
}
}
if (arg.first == "startapp" && is_valid_start_parameter(arg.second)) {
// /<bot_username>?startapp
// /<bot_username>?startapp=<parameter>
return td::make_unique<InternalLinkSideMenuBot>(std::move(username), arg.second);
}
if (arg.first == "game" && is_valid_game_name(arg.second)) {
// /<bot_username>?game=<short_name>
return td::make_unique<InternalLinkGame>(std::move(username), arg.second);
}
}
if (!url_query.get_arg("attach").empty()) {
// /<username>?attach=<bot_username>
// /<username>?attach=<bot_username>&startattach=<start_parameter>
return td::make_unique<InternalLinkAttachMenuBot>(
nullptr, td::make_unique<InternalLinkPublicDialog>(std::move(username), string()),
url_query.get_arg("attach").str(), url_query.get_arg("startattach"));
} else if (url_query.has_arg("startattach")) {
// /<bot_username>?startattach&choose=users+bots+groups+channels
// /<bot_username>?startattach=<start_parameter>&choose=users+bots+groups+channels
return td::make_unique<InternalLinkAttachMenuBot>(get_target_chat_chosen(url_query.get_arg("choose")), nullptr,
std::move(username), url_query.get_arg("startattach"));
}
// /<username>
return td::make_unique<InternalLinkPublicDialog>(std::move(username), get_url_query_draft_text(url_query));
}
return nullptr;
}
unique_ptr<LinkManager::InternalLink> LinkManager::get_internal_link_message_draft(Slice url, Slice text) {
if (url.empty() && text.empty()) {
return nullptr;
}
while (!text.empty() && text.back() == '\n') {
text.remove_suffix(1);
}
url = trim(url);
if (url.empty()) {
url = text;
text = Slice();
}
FormattedText full_text;
bool contains_url = false;
if (!text.empty()) {
contains_url = true;
full_text.text = PSTRING() << url << '\n' << text;
} else {
full_text.text = url.str();
}
if (fix_formatted_text(full_text.text, full_text.entities, false, false, false, true, true).is_error()) {
return nullptr;
}
if (full_text.text[0] == '@') {
full_text.text = ' ' + full_text.text;
for (auto &entity : full_text.entities) {
entity.offset++;
}
}
return td::make_unique<InternalLinkMessageDraft>(std::move(full_text), contains_url);
}
unique_ptr<LinkManager::InternalLink> LinkManager::get_internal_link_passport(
Slice query, const vector<std::pair<string, string>> &args, bool allow_unknown) {
auto get_arg = [&args](Slice key) {
for (auto &arg : args) {
if (arg.first == key) {
return Slice(arg.second);
}
}
return Slice();
};
UserId bot_user_id(to_integer<int64>(get_arg("bot_id")));
auto scope = get_arg("scope");
auto public_key = get_arg("public_key");
auto nonce = get_arg("nonce");
if (nonce.empty()) {
nonce = get_arg("payload");
}
auto callback_url = get_arg("callback_url");
if (!bot_user_id.is_valid() || scope.empty() || public_key.empty() || nonce.empty()) {
if (!allow_unknown) {
return nullptr;
}
return td::make_unique<InternalLinkUnknownDeepLink>(PSTRING() << "tg://" << query);
}
return td::make_unique<InternalLinkPassportDataRequest>(bot_user_id, scope.str(), public_key.str(), nonce.str(),
callback_url.str());
}
Result<string> LinkManager::get_internal_link(const td_api::object_ptr<td_api::InternalLinkType> &type,
bool is_internal) {
if (type == nullptr) {
return Status::Error(400, "Link type must be non-empty");
}
return get_internal_link_impl(type.get(), is_internal);
}
Result<string> LinkManager::get_internal_link_impl(const td_api::InternalLinkType *type_ptr, bool is_internal) {
switch (type_ptr->get_id()) {
case td_api::internalLinkTypeActiveSessions::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings/devices";
case td_api::internalLinkTypeAttachmentMenuBot::ID: {
auto link = static_cast<const td_api::internalLinkTypeAttachmentMenuBot *>(type_ptr);
if (!is_valid_username(link->bot_username_)) {
return Status::Error(400, "Invalid bot username specified");
}
string start_parameter;
if (!link->url_.empty()) {
if (!begins_with(link->url_, "start://")) {
return Status::Error(400, "Unsupported link URL specified");
}
auto start_parameter_slice = Slice(link->url_).substr(8);
if (start_parameter_slice.empty() || !is_valid_start_parameter(start_parameter_slice)) {
return Status::Error(400, "Invalid start parameter specified");
}
start_parameter = PSTRING() << '=' << start_parameter_slice;
}
if (link->target_chat_ == nullptr) {
return Status::Error(400, "Target chat must be non-empty");
}
switch (link->target_chat_->get_id()) {
case td_api::targetChatChosen::ID: {
auto target = static_cast<const td_api::targetChatChosen *>(link->target_chat_.get());
if (!target->allow_user_chats_ && !target->allow_bot_chats_ && !target->allow_group_chats_ &&
!target->allow_channel_chats_) {
return Status::Error(400, "At least one target chat type must be allowed");
}
vector<string> types;
if (target->allow_user_chats_) {
types.push_back("users");
}
if (target->allow_bot_chats_) {
types.push_back("bots");
}
if (target->allow_group_chats_) {
types.push_back("groups");
}
if (target->allow_channel_chats_) {
types.push_back("channels");
}
auto choose = implode(types, '+');
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->bot_username_ << "&startattach" << start_parameter
<< "&choose=" << choose;
} else {
return PSTRING() << get_t_me_url() << link->bot_username_ << "?startattach" << start_parameter
<< "&choose=" << choose;
}
}
case td_api::targetChatCurrent::ID:
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->bot_username_ << "&startattach" << start_parameter;
} else {
return PSTRING() << get_t_me_url() << link->bot_username_ << "?startattach" << start_parameter;
}
case td_api::targetChatInternalLink::ID: {
auto target = static_cast<const td_api::targetChatInternalLink *>(link->target_chat_.get());
if (!start_parameter.empty()) {
start_parameter = "&startattach" + start_parameter;
}
if (target->link_ == nullptr || target->link_->get_id() != td_api::internalLinkTypePublicChat::ID) {
if (target->link_->get_id() == td_api::internalLinkTypeUserPhoneNumber::ID) {
auto user_phone_number_link =
static_cast<const td_api::internalLinkTypeUserPhoneNumber *>(target->link_.get());
string phone_number;
if (user_phone_number_link->phone_number_[0] == '+') {
phone_number = user_phone_number_link->phone_number_.substr(1);
} else {
phone_number = user_phone_number_link->phone_number_;
}
if (!is_valid_phone_number(phone_number)) {
return Status::Error(400, "Invalid target phone number specified");
}
if (is_internal) {
return PSTRING() << "tg://resolve?phone=+" << phone_number << "&attach=" << link->bot_username_
<< start_parameter;
} else {
return PSTRING() << get_t_me_url() << '+' << phone_number << "?attach=" << link->bot_username_
<< start_parameter;
}
}
return Status::Error(400, "Unsupported target link specified");
}
auto public_chat_link = static_cast<const td_api::internalLinkTypePublicChat *>(target->link_.get());
if (!is_valid_username(public_chat_link->chat_username_)) {
return Status::Error(400, "Invalid target public chat username specified");
}
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << public_chat_link->chat_username_
<< "&attach=" << link->bot_username_ << start_parameter;
} else {
return PSTRING() << get_t_me_url() << public_chat_link->chat_username_ << "?attach=" << link->bot_username_
<< start_parameter;
}
}
default:
UNREACHABLE();
}
break;
}
case td_api::internalLinkTypeAuthenticationCode::ID: {
auto link = static_cast<const td_api::internalLinkTypeAuthenticationCode *>(type_ptr);
if (is_internal) {
return PSTRING() << "tg://login?code=" << url_encode(link->code_);
} else {
return PSTRING() << get_t_me_url() << "login/" << url_encode(link->code_);
}
}
case td_api::internalLinkTypeBackground::ID: {
auto link = static_cast<const td_api::internalLinkTypeBackground *>(type_ptr);
auto params_pos = link->background_name_.find('?');
string slug = params_pos >= link->background_name_.size() ? link->background_name_
: link->background_name_.substr(0, params_pos);
if (slug.empty()) {
return Status::Error(400, "Background name must be non-empty");
}
if (BackgroundType::is_background_name_local(slug)) {
TRY_RESULT(background_type, BackgroundType::get_local_background_type(link->background_name_));
auto background_link = background_type.get_link(!is_internal);
CHECK(!background_type.has_file());
if (is_internal) {
Slice field_name = background_type.has_gradient_fill() ? Slice("gradient") : Slice("color");
return PSTRING() << "tg://bg?" << field_name << '=' << background_link;
} else {
return PSTRING() << get_t_me_url() << "bg/" << background_link;
}
}
auto prefix = is_internal ? string("tg://bg?slug=") : get_t_me_url() + "bg/";
const auto url_query = parse_url_query(link->background_name_);
bool is_first_arg = !is_internal;
auto copy_arg = [&](Slice name) {
return CopyArg(name, &url_query, &is_first_arg);
};
return PSTRING() << prefix << url_encode(slug) << copy_arg("mode") << copy_arg("intensity")
<< copy_arg("bg_color") << copy_arg("rotation");
}
case td_api::internalLinkTypeBotAddToChannel::ID: {
auto link = static_cast<const td_api::internalLinkTypeBotAddToChannel *>(type_ptr);
if (!is_valid_username(link->bot_username_)) {
return Status::Error(400, "Invalid bot username specified");
}
auto admin = get_admin_string(AdministratorRights(link->administrator_rights_, ChannelType::Broadcast));
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->bot_username_ << "&startchannel" << admin;
} else {
return PSTRING() << get_t_me_url() << link->bot_username_ << "?startchannel" << admin;
}
}
case td_api::internalLinkTypeBotStart::ID: {
auto link = static_cast<const td_api::internalLinkTypeBotStart *>(type_ptr);
if (link->autostart_) {
return Status::Error(400, "Can't create an autostart bot link");
}
if (!is_valid_username(link->bot_username_)) {
return Status::Error(400, "Invalid bot username specified");
}
if (!is_valid_start_parameter(link->start_parameter_)) {
return Status::Error(400, "Invalid start parameter specified");
}
auto start_parameter = link->start_parameter_.empty() ? string() : "=" + link->start_parameter_;
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->bot_username_ << "&start" << start_parameter;
} else {
return PSTRING() << get_t_me_url() << link->bot_username_ << "?start" << start_parameter;
}
}
case td_api::internalLinkTypeBotStartInGroup::ID: {
auto link = static_cast<const td_api::internalLinkTypeBotStartInGroup *>(type_ptr);
if (!is_valid_username(link->bot_username_)) {
return Status::Error(400, "Invalid bot username specified");
}
if (!is_valid_start_parameter(link->start_parameter_)) {
return Status::Error(400, "Invalid start parameter specified");
}
auto admin = get_admin_string(AdministratorRights(link->administrator_rights_, ChannelType::Megagroup));
auto start_parameter = link->start_parameter_.empty() ? string() : "=" + link->start_parameter_;
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->bot_username_ << "&startgroup" << start_parameter << admin;
} else {
return PSTRING() << get_t_me_url() << link->bot_username_ << "?startgroup" << start_parameter << admin;
}
}
case td_api::internalLinkTypeBusinessChat::ID: {
auto link = static_cast<const td_api::internalLinkTypeBusinessChat *>(type_ptr);
if (is_internal) {
return PSTRING() << "tg://message?slug=" << url_encode(link->link_name_);
} else {
return PSTRING() << get_t_me_url() << "m/" << url_encode(link->link_name_);
}
}
case td_api::internalLinkTypeChangePhoneNumber::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings/change_number";
case td_api::internalLinkTypeChatBoost::ID: {
auto link = static_cast<const td_api::internalLinkTypeChatBoost *>(type_ptr);
auto parsed_link = parse_internal_link(link->url_);
if (parsed_link == nullptr) {
return Status::Error(400, "Invalid chat boost URL specified");
}
auto parsed_object = parsed_link->get_internal_link_type_object();
if (parsed_object->get_id() != td_api::internalLinkTypeChatBoost::ID) {
return Status::Error(400, "Invalid chat boost URL specified");
}
if (!is_internal) {
return Status::Error(400, "Use getChatBoostLink to get an HTTPS link to boost a chat");
}
return std::move(static_cast<td_api::internalLinkTypeChatBoost &>(*parsed_object).url_);
}
case td_api::internalLinkTypeChatFolderInvite::ID: {
auto link = static_cast<const td_api::internalLinkTypeChatFolderInvite *>(type_ptr);
auto slug = get_dialog_filter_invite_link_slug(link->invite_link_);
if (slug.empty()) {
return Status::Error(400, "Invalid invite link specified");
}
return get_dialog_filter_invite_link(slug, is_internal);
}
case td_api::internalLinkTypeChatFolderSettings::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings/folders";
case td_api::internalLinkTypeChatInvite::ID: {
auto link = static_cast<const td_api::internalLinkTypeChatInvite *>(type_ptr);
auto invite_hash = get_dialog_invite_link_hash(link->invite_link_);
if (invite_hash.empty()) {
return Status::Error(400, "Invalid invite link specified");
}
return get_dialog_invite_link(invite_hash, is_internal);
}
case td_api::internalLinkTypeDefaultMessageAutoDeleteTimerSettings::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings/auto_delete";
case td_api::internalLinkTypeEditProfileSettings::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings/edit_profile";
case td_api::internalLinkTypeGame::ID: {
auto link = static_cast<const td_api::internalLinkTypeGame *>(type_ptr);
if (!is_valid_username(link->bot_username_)) {
return Status::Error(400, "Invalid bot username specified");
}
if (!is_valid_game_name(link->game_short_name_)) {
return Status::Error(400, "Invalid game name specified");
}
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->bot_username_ << "&game=" << link->game_short_name_;
} else {
return PSTRING() << get_t_me_url() << link->bot_username_ << "?game=" << link->game_short_name_;
}
}
case td_api::internalLinkTypeInstantView::ID: {
auto link = static_cast<const td_api::internalLinkTypeInstantView *>(type_ptr);
if (is_internal) {
return Status::Error("Deep link is unavailable for the link type");
}
auto info = get_link_info(link->url_);
auto fallback_info = get_link_info(link->fallback_url_);
switch (info.type_) {
case LinkType::External:
case LinkType::Tg:
return Status::Error("Invalid instant view URL provided");
case LinkType::Telegraph:
if (fallback_info.type_ != LinkType::Telegraph ||
link->url_ != (PSLICE() << "https://telegra.ph" << fallback_info.query_)) {
return Status::Error("Unrelated fallback URL provided");
}
return link->fallback_url_;
case LinkType::TMe:
// skip URL and fallback_url consistency checks
return link->url_;
default:
UNREACHABLE();
break;
}
}
case td_api::internalLinkTypeInvoice::ID: {
auto link = static_cast<const td_api::internalLinkTypeInvoice *>(type_ptr);
if (is_internal) {
return PSTRING() << "tg://invoice?slug=" << url_encode(link->invoice_name_);
} else {
return PSTRING() << get_t_me_url() << '$' << url_encode(link->invoice_name_);
}
}
case td_api::internalLinkTypeLanguagePack::ID: {
auto link = static_cast<const td_api::internalLinkTypeLanguagePack *>(type_ptr);
if (is_internal) {
return PSTRING() << "tg://setlanguage?lang=" << url_encode(link->language_pack_id_);
} else {
return PSTRING() << get_t_me_url() << "setlanguage/" << url_encode(link->language_pack_id_);
}
}
case td_api::internalLinkTypeLanguageSettings::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings/language";
case td_api::internalLinkTypeMessage::ID: {
auto link = static_cast<const td_api::internalLinkTypeMessage *>(type_ptr);
auto parsed_link = parse_internal_link(link->url_);
if (parsed_link == nullptr) {
return Status::Error(400, "Invalid message URL specified");
}
auto parsed_object = parsed_link->get_internal_link_type_object();
if (parsed_object->get_id() != td_api::internalLinkTypeMessage::ID) {
return Status::Error(400, "Invalid message URL specified");
}
if (!is_internal) {
return Status::Error(400, "Use getMessageLink to get an HTTPS link to a message");
}
return std::move(static_cast<td_api::internalLinkTypeMessage &>(*parsed_object).url_);
}
case td_api::internalLinkTypeMessageDraft::ID: {
auto link = static_cast<const td_api::internalLinkTypeMessageDraft *>(type_ptr);
string text;
if (link->text_ != nullptr) {
text = std::move(link->text_->text_);
}
string url;
if (link->contains_link_) {
std::tie(url, text) = split(text, '\n');
} else {
url = std::move(text);
text.clear();
}
if (!text.empty()) {
text = "&text=" + url_encode(text);
}
if (is_internal) {
return PSTRING() << "tg://msg_url?url=" << url_encode(url) << text;
} else {
return PSTRING() << get_t_me_url() << "share?url=" << url_encode(url) << text;
}
}
case td_api::internalLinkTypePassportDataRequest::ID: {
auto link = static_cast<const td_api::internalLinkTypePassportDataRequest *>(type_ptr);
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
if (!UserId(link->bot_user_id_).is_valid()) {
return Status::Error("Invalid bot user identifier specified");
}
return PSTRING() << "tg://resolve?domain=telegrampassport&bot_id=" << link->bot_user_id_
<< "&scope=" << url_encode(link->scope_) << "&public_key=" << url_encode(link->public_key_)
<< "&nonce=" << url_encode(link->nonce_) << "&callback_url=" << url_encode(link->callback_url_);
}
case td_api::internalLinkTypePhoneNumberConfirmation::ID: {
auto link = static_cast<const td_api::internalLinkTypePhoneNumberConfirmation *>(type_ptr);
if (is_internal) {
return PSTRING() << "tg://confirmphone?phone=" << url_encode(link->phone_number_)
<< "&hash=" << url_encode(link->hash_);
} else {
return PSTRING() << get_t_me_url() << "confirmphone?phone=" << url_encode(link->phone_number_)
<< "&hash=" << url_encode(link->hash_);
}
}
case td_api::internalLinkTypePremiumFeatures::ID: {
auto link = static_cast<const td_api::internalLinkTypePremiumFeatures *>(type_ptr);
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return PSTRING() << "tg://premium_offer?ref=" << url_encode(link->referrer_);
}
case td_api::internalLinkTypePremiumGift::ID: {
auto link = static_cast<const td_api::internalLinkTypePremiumGift *>(type_ptr);
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return PSTRING() << "tg://premium_multigift?ref=" << url_encode(link->referrer_);
}
case td_api::internalLinkTypePremiumGiftCode::ID: {
auto link = static_cast<const td_api::internalLinkTypePremiumGiftCode *>(type_ptr);
if (is_internal) {
return PSTRING() << "tg://giftcode?slug=" << url_encode(link->code_);
} else {
return PSTRING() << get_t_me_url() << "giftcode/" << url_encode(link->code_);
}
}
case td_api::internalLinkTypePrivacyAndSecuritySettings::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings/privacy";
case td_api::internalLinkTypeProxy::ID: {
auto link = static_cast<const td_api::internalLinkTypeProxy *>(type_ptr);
TRY_RESULT(proxy, Proxy::create_proxy(link->server_, link->port_, link->type_.get()));
return get_proxy_link(proxy, is_internal);
}
case td_api::internalLinkTypePublicChat::ID: {
auto link = static_cast<const td_api::internalLinkTypePublicChat *>(type_ptr);
if (!is_valid_username(link->chat_username_)) {
return Status::Error(400, "Invalid chat username specified");
}
if (!check_utf8(link->draft_text_)) {
return Status::Error(400, "Draft text must be encoded in UTF-8");
}
return get_public_dialog_link(link->chat_username_, link->draft_text_, is_internal);
}
case td_api::internalLinkTypeQrCodeAuthentication::ID:
return Status::Error("The link must never be generated client-side");
case td_api::internalLinkTypeRestorePurchases::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://restore_purchases";
case td_api::internalLinkTypeSettings::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings";
case td_api::internalLinkTypeSideMenuBot::ID: {
auto link = static_cast<const td_api::internalLinkTypeSideMenuBot *>(type_ptr);
if (!is_valid_username(link->bot_username_)) {
return Status::Error(400, "Invalid bot username specified");
}
string start_parameter;
if (!link->url_.empty()) {
if (!begins_with(link->url_, "start://")) {
return Status::Error(400, "Unsupported link URL specified");
}
auto start_parameter_slice = Slice(link->url_).substr(8);
if (start_parameter_slice.empty() || !is_valid_start_parameter(start_parameter_slice)) {
return Status::Error(400, "Invalid start parameter specified");
}
start_parameter = PSTRING() << '=' << start_parameter_slice;
}
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->bot_username_ << "&startapp" << start_parameter;
} else {
return PSTRING() << get_t_me_url() << link->bot_username_ << "?startapp" << start_parameter;
}
}
case td_api::internalLinkTypeStickerSet::ID: {
auto link = static_cast<const td_api::internalLinkTypeStickerSet *>(type_ptr);
if (link->sticker_set_name_.empty()) {
return Status::Error(400, "Invalid sticker set name specified");
}
if (is_internal) {
return PSTRING() << "tg://add" << (link->expect_custom_emoji_ ? "emoji" : "stickers")
<< "?set=" << url_encode(link->sticker_set_name_);
} else {
return PSTRING() << get_t_me_url() << "add" << (link->expect_custom_emoji_ ? "emoji" : "stickers") << '/'
<< url_encode(link->sticker_set_name_);
}
}
case td_api::internalLinkTypeStory::ID: {
auto link = static_cast<const td_api::internalLinkTypeStory *>(type_ptr);
if (!is_valid_username(link->story_sender_username_)) {
return Status::Error(400, "Invalid story sender username specified");
}
if (!StoryId(link->story_id_).is_server()) {
return Status::Error(400, "Invalid story identifier specified");
}
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->story_sender_username_ << "&story=" << link->story_id_;
} else {
return PSTRING() << get_t_me_url() << link->story_sender_username_ << "/s/" << link->story_id_;
}
}
case td_api::internalLinkTypeTheme::ID: {
auto link = static_cast<const td_api::internalLinkTypeTheme *>(type_ptr);
if (link->theme_name_.empty()) {
return Status::Error(400, "Invalid theme name specified");
}
if (is_internal) {
return PSTRING() << "tg://addtheme?slug=" << url_encode(link->theme_name_);
} else {
return PSTRING() << get_t_me_url() << "addtheme/" << url_encode(link->theme_name_);
}
}
case td_api::internalLinkTypeThemeSettings::ID:
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
return "tg://settings/themes";
case td_api::internalLinkTypeUnknownDeepLink::ID: {
auto link = static_cast<const td_api::internalLinkTypeUnknownDeepLink *>(type_ptr);
if (!is_internal) {
return Status::Error("HTTP link is unavailable for the link type");
}
auto parsed_link = parse_internal_link(link->link_);
if (parsed_link == nullptr) {
return Status::Error(400, "Invalid deep link URL specified");
}
auto parsed_object = parsed_link->get_internal_link_type_object();
if (parsed_object->get_id() != td_api::internalLinkTypeUnknownDeepLink::ID) {
return Status::Error(400, "Invalid deep link URL specified");
}
return std::move(static_cast<td_api::internalLinkTypeUnknownDeepLink &>(*parsed_object).link_);
}
case td_api::internalLinkTypeUnsupportedProxy::ID:
if (is_internal) {
return "tg://proxy?port=-1&server=0.0.0.0";
} else {
return PSTRING() << get_t_me_url() << "proxy?port=-1&server=0.0.0.0";
}
case td_api::internalLinkTypeUserPhoneNumber::ID: {
auto link = static_cast<const td_api::internalLinkTypeUserPhoneNumber *>(type_ptr);
string phone_number;
if (link->phone_number_[0] == '+') {
phone_number = link->phone_number_.substr(1);
} else {
phone_number = link->phone_number_;
}
if (!is_valid_phone_number(phone_number)) {
return Status::Error(400, "Invalid phone number specified");
}
if (!check_utf8(link->draft_text_)) {
return Status::Error(400, "Draft text must be encoded in UTF-8");
}
if (is_internal) {
return PSTRING() << "tg://resolve?phone=+" << phone_number << (link->draft_text_.empty() ? "" : "&text=")
<< url_encode(link->draft_text_);
} else {
return PSTRING() << get_t_me_url() << '+' << phone_number << (link->draft_text_.empty() ? "" : "?text=")
<< url_encode(link->draft_text_);
}
}
case td_api::internalLinkTypeUserToken::ID: {
auto link = static_cast<const td_api::internalLinkTypeUserToken *>(type_ptr);
if (link->token_.empty()) {
return Status::Error(400, "Invalid user token specified");
}
if (is_internal) {
return PSTRING() << "tg://contact?token=" << link->token_;
} else {
return PSTRING() << get_t_me_url() << "contact/" << link->token_;
}
}
case td_api::internalLinkTypeVideoChat::ID: {
auto link = static_cast<const td_api::internalLinkTypeVideoChat *>(type_ptr);
if (!is_valid_username(link->chat_username_)) {
return Status::Error(400, "Invalid chat username specified");
}
string invite_hash;
if (!link->invite_hash_.empty()) {
invite_hash = '=' + url_encode(link->invite_hash_);
}
auto name = link->is_live_stream_ ? Slice("livestream") : Slice("videochat");
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->chat_username_ << '&' << name << invite_hash;
} else {
return PSTRING() << get_t_me_url() << link->chat_username_ << '?' << name << invite_hash;
}
}
case td_api::internalLinkTypeWebApp::ID: {
auto link = static_cast<const td_api::internalLinkTypeWebApp *>(type_ptr);
if (!is_valid_username(link->bot_username_)) {
return Status::Error(400, "Invalid bot username specified");
}
if (!is_valid_web_app_name(link->web_app_short_name_)) {
return Status::Error(400, "Invalid Web App name specified");
}
if (!is_valid_start_parameter(link->start_parameter_)) {
return Status::Error(400, "Invalid start parameter specified");
}
string start_parameter;
if (!link->start_parameter_.empty()) {
start_parameter = PSTRING() << (is_internal ? '&' : '?') << "startapp=" << link->start_parameter_;
}
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << link->bot_username_ << "&appname=" << link->web_app_short_name_
<< start_parameter;
} else {
return PSTRING() << get_t_me_url() << link->bot_username_ << '/' << link->web_app_short_name_
<< start_parameter;
}
}
default:
break;
}
UNREACHABLE();
return Status::Error(500, "Unsupported");
}
void LinkManager::update_autologin_token(string autologin_token) {
autologin_update_time_ = Time::now();
autologin_token_ = std::move(autologin_token);
}
void LinkManager::update_autologin_domains(vector<string> autologin_domains, vector<string> url_auth_domains,
vector<string> whitelisted_domains) {
if (autologin_domains_ != autologin_domains) {
autologin_domains_ = std::move(autologin_domains);
G()->td_db()->get_binlog_pmc()->set("autologin_domains", implode(autologin_domains_, '\xFF'));
}
if (url_auth_domains_ != url_auth_domains) {
url_auth_domains_ = std::move(url_auth_domains);
G()->td_db()->get_binlog_pmc()->set("url_auth_domains", implode(url_auth_domains_, '\xFF'));
}
if (whitelisted_domains_ != whitelisted_domains) {
whitelisted_domains_ = std::move(whitelisted_domains);
G()->td_db()->get_binlog_pmc()->set("whitelisted_domains", implode(whitelisted_domains_, '\xFF'));
}
}
void LinkManager::get_deep_link_info(Slice link, Promise<td_api::object_ptr<td_api::deepLinkInfo>> &&promise) {
Slice link_scheme("tg:");
if (begins_with(link, link_scheme)) {
link.remove_prefix(link_scheme.size());
if (begins_with(link, "//")) {
link.remove_prefix(2);
}
}
size_t pos = 0;
while (pos < link.size() && link[pos] != '/' && link[pos] != '?' && link[pos] != '#') {
pos++;
}
link.truncate(pos);
td_->create_handler<GetDeepLinkInfoQuery>(std::move(promise))->send(link);
}
void LinkManager::get_external_link_info(string &&link, Promise<td_api::object_ptr<td_api::LoginUrlInfo>> &&promise) {
auto default_result = td_api::make_object<td_api::loginUrlInfoOpen>(link, false);
if (G()->close_flag()) {
return promise.set_value(std::move(default_result));
}
auto r_url = parse_url(link);
if (r_url.is_error()) {
return promise.set_value(std::move(default_result));
}
auto url = r_url.move_as_ok();
if (!url.userinfo_.empty() || url.is_ipv6_) {
return promise.set_value(std::move(default_result));
}
bool skip_confirmation = td::contains(whitelisted_domains_, url.host_);
default_result->skip_confirmation_ = skip_confirmation;
if (!td::contains(autologin_domains_, url.host_)) {
if (td::contains(url_auth_domains_, url.host_)) {
td_->create_handler<RequestUrlAuthQuery>(std::move(promise))->send(link, MessageFullId(), 0);
return;
}
return promise.set_value(std::move(default_result));
}
if (autologin_update_time_ < Time::now() - 10000) {
auto query_promise = PromiseCreator::lambda([link = std::move(link), default_result = std::move(default_result),
promise = std::move(promise)](Result<Unit> &&result) mutable {
if (result.is_error()) {
return promise.set_value(std::move(default_result));
}
send_closure(G()->link_manager(), &LinkManager::get_external_link_info, std::move(link), std::move(promise));
});
return send_closure(G()->config_manager(), &ConfigManager::reget_config, std::move(query_promise));
}
if (autologin_token_.empty()) {
return promise.set_value(std::move(default_result));
}
url.protocol_ = HttpUrl::Protocol::Https;
Slice path = url.query_;
path.truncate(url.query_.find_first_of("?#"));
Slice parameters_hash = Slice(url.query_).substr(path.size());
Slice parameters = parameters_hash;
parameters.truncate(parameters.find('#'));
Slice hash = parameters_hash.substr(parameters.size());
string added_parameter;
if (parameters.empty()) {
added_parameter = '?';
} else if (parameters.size() == 1) {
CHECK(parameters == "?");
} else {
added_parameter = '&';
}
added_parameter += "autologin_token=";
added_parameter += autologin_token_;
url.query_ = PSTRING() << path << parameters << added_parameter << hash;
promise.set_value(td_api::make_object<td_api::loginUrlInfoOpen>(url.get_url(), skip_confirmation));
}
void LinkManager::get_login_url_info(MessageFullId message_full_id, int64 button_id,
Promise<td_api::object_ptr<td_api::LoginUrlInfo>> &&promise) {
TRY_RESULT_PROMISE(promise, url, td_->messages_manager_->get_login_button_url(message_full_id, button_id));
td_->create_handler<RequestUrlAuthQuery>(std::move(promise))
->send(std::move(url), message_full_id, narrow_cast<int32>(button_id));
}
void LinkManager::get_login_url(MessageFullId message_full_id, int64 button_id, bool allow_write_access,
Promise<td_api::object_ptr<td_api::httpUrl>> &&promise) {
TRY_RESULT_PROMISE(promise, url, td_->messages_manager_->get_login_button_url(message_full_id, button_id));
td_->create_handler<AcceptUrlAuthQuery>(std::move(promise))
->send(std::move(url), message_full_id, narrow_cast<int32>(button_id), allow_write_access);
}
void LinkManager::get_link_login_url(const string &url, bool allow_write_access,
Promise<td_api::object_ptr<td_api::httpUrl>> &&promise) {
td_->create_handler<AcceptUrlAuthQuery>(std::move(promise))->send(url, MessageFullId(), 0, allow_write_access);
}
Result<string> LinkManager::get_background_url(const string &name,
td_api::object_ptr<td_api::BackgroundType> background_type) {
if (background_type == nullptr) {
return Status::Error(400, "Type must be non-empty");
}
if (background_type->get_id() == td_api::backgroundTypeChatTheme::ID) {
return Status::Error(400, "Background has no link");
}
TRY_RESULT(type, BackgroundType::get_background_type(background_type.get(), 0));
auto url = PSTRING() << get_t_me_url() << "bg/";
auto link = type.get_link();
if (type.has_file()) {
url += name;
if (!link.empty()) {
url += '?';
url += link;
}
} else {
url += link;
}
return url;
}
string LinkManager::get_dialog_filter_invite_link_slug(Slice invite_link) {
auto link_info = get_link_info(invite_link);
if (link_info.type_ != LinkType::Tg && link_info.type_ != LinkType::TMe) {
return string();
}
const auto url_query = parse_url_query(link_info.query_);
auto slug = get_url_query_slug(link_info.type_ == LinkType::Tg, url_query);
if (!is_base64url_characters(slug)) {
return string();
}
return slug;
}
string LinkManager::get_dialog_filter_invite_link(Slice slug, bool is_internal) {
if (!is_base64url_characters(slug)) {
return string();
}
if (is_internal) {
return PSTRING() << "tg:addlist?slug=" << slug;
} else {
return PSTRING() << get_t_me_url() << "addlist/" << slug;
}
}
string LinkManager::get_dialog_invite_link_hash(Slice invite_link) {
auto link_info = get_link_info(invite_link);
if (link_info.type_ != LinkType::Tg && link_info.type_ != LinkType::TMe) {
return string();
}
const auto url_query = parse_url_query(link_info.query_);
auto invite_hash = get_url_query_hash(link_info.type_ == LinkType::Tg, url_query);
if (is_valid_phone_number(invite_hash)) {
return string();
}
if (!is_base64url_characters(invite_hash)) {
return string();
}
return invite_hash;
}
string LinkManager::get_dialog_invite_link(Slice invite_hash, bool is_internal) {
if (!is_base64url_characters(invite_hash)) {
return string();
}
if (is_internal) {
return PSTRING() << "tg:join?invite=" << invite_hash;
} else {
return PSTRING() << get_t_me_url() << '+' << invite_hash;
}
}
string LinkManager::get_instant_view_link_url(Slice link) {
auto link_info = get_link_info(link);
if (link_info.type_ != LinkType::TMe) {
return string();
}
const auto url_query = parse_url_query(link_info.query_);
const auto &path = url_query.path_;
if (path.size() == 1 && path[0] == "iv") {
return url_query.get_arg("url").str();
}
return string();
}
string LinkManager::get_instant_view_link_rhash(Slice link) {
auto link_info = get_link_info(link);
if (link_info.type_ != LinkType::TMe) {
return string();
}
const auto url_query = parse_url_query(link_info.query_);
const auto &path = url_query.path_;
if (path.size() == 1 && path[0] == "iv" && !url_query.get_arg("url").empty()) {
return url_query.get_arg("rhash").str();
}
return string();
}
string LinkManager::get_instant_view_link(Slice url, Slice rhash) {
return PSTRING() << get_t_me_url() << "iv?url=" << url_encode(url) << "&rhash=" << url_encode(rhash);
}
string LinkManager::get_public_dialog_link(Slice username, Slice draft_text, bool is_internal) {
if (is_internal) {
return PSTRING() << "tg://resolve?domain=" << url_encode(username) << (draft_text.empty() ? "" : "&text=")
<< url_encode(draft_text);
} else {
return PSTRING() << get_t_me_url() << url_encode(username) << (draft_text.empty() ? "" : "?text=")
<< url_encode(draft_text);
}
}
Result<string> LinkManager::get_proxy_link(const Proxy &proxy, bool is_internal) {
string url = is_internal ? "tg://" : get_t_me_url();
bool is_socks = false;
switch (proxy.type()) {
case Proxy::Type::Socks5:
url += "socks";
is_socks = true;
break;
case Proxy::Type::HttpTcp:
case Proxy::Type::HttpCaching:
return Status::Error(400, "HTTP proxies have no public links");
case Proxy::Type::Mtproto:
url += "proxy";
break;
default:
UNREACHABLE();
}
url += "?server=";
url += url_encode(proxy.server());
url += "&port=";
url += to_string(proxy.port());
if (is_socks) {
if (!proxy.user().empty() || !proxy.password().empty()) {
url += "&user=";
url += url_encode(proxy.user());
url += "&pass=";
url += url_encode(proxy.password());
}
} else {
url += "&secret=";
url += proxy.secret().get_encoded_secret();
}
return std::move(url);
}
UserId LinkManager::get_link_user_id(Slice url) {
string lower_cased_url = to_lower(url);
url = lower_cased_url;
Slice link_scheme("tg:");
if (!begins_with(url, link_scheme)) {
return UserId();
}
url.remove_prefix(link_scheme.size());
if (begins_with(url, "//")) {
url.remove_prefix(2);
}
Slice host("user");
if (!begins_with(url, host) || (url.size() > host.size() && Slice("/?#").find(url[host.size()]) == Slice::npos)) {
return UserId();
}
url.remove_prefix(host.size());
if (begins_with(url, "/")) {
url.remove_prefix(1);
}
if (!begins_with(url, "?")) {
return UserId();
}
url.remove_prefix(1);
url.truncate(url.find('#'));
for (auto parameter : full_split(url, '&')) {
Slice key;
Slice value;
std::tie(key, value) = split(parameter, '=');
if (key == Slice("id")) {
auto r_user_id = to_integer_safe<int64>(value);
if (r_user_id.is_error()) {
return UserId();
}
return UserId(r_user_id.ok());
}
}
return UserId();
}
string LinkManager::get_t_me_url() {
if (Scheduler::context() != nullptr) {
return G()->get_option_string("t_me_url", "https://t.me/");
} else {
return "https://t.me/";
}
}
Result<CustomEmojiId> LinkManager::get_link_custom_emoji_id(Slice url) {
string lower_cased_url = to_lower(url);
url = lower_cased_url;
Slice link_scheme("tg:");
if (!begins_with(url, link_scheme)) {
return Status::Error(400, "Custom emoji URL must have scheme tg");
}
url.remove_prefix(link_scheme.size());
if (begins_with(url, "//")) {
url.remove_prefix(2);
}
Slice host("emoji");
if (!begins_with(url, host) || (url.size() > host.size() && Slice("/?#").find(url[host.size()]) == Slice::npos)) {
return Status::Error(400, PSLICE() << "Custom emoji URL must have host \"" << host << '"');
}
url.remove_prefix(host.size());
if (begins_with(url, "/")) {
url.remove_prefix(1);
}
if (!begins_with(url, "?")) {
return Status::Error(400, "Custom emoji URL must have an emoji identifier");
}
url.remove_prefix(1);
url.truncate(url.find('#'));
for (auto parameter : full_split(url, '&')) {
Slice key;
Slice value;
std::tie(key, value) = split(parameter, '=');
if (key == Slice("id")) {
auto r_document_id = to_integer_safe<int64>(value);
if (r_document_id.is_error() || r_document_id.ok() == 0) {
return Status::Error(400, "Invalid custom emoji identifier specified");
}
return CustomEmojiId(r_document_id.ok());
}
}
return Status::Error(400, "Custom emoji URL must have an emoji identifier");
}
Result<DialogBoostLinkInfo> LinkManager::get_dialog_boost_link_info(Slice url) {
if (url.empty()) {
return Status::Error("URL must be non-empty");
}
auto link_info = get_link_info(url);
if (link_info.type_ != LinkType::Tg && link_info.type_ != LinkType::TMe) {
return Status::Error("Invalid chat boost link URL");
}
url = link_info.query_;
Slice username;
Slice channel_id_slice;
if (link_info.type_ == LinkType::Tg) {
// boost?domain=username
// boost?channel=123456789
if (!begins_with(url, "boost")) {
return Status::Error("Wrong chat boost link URL");
}
url = url.substr(5);
if (begins_with(url, "/")) {
url = url.substr(1);
}
if (!begins_with(url, "?")) {
return Status::Error("Wrong chat boost link URL");
}
url = url.substr(1);
auto args = full_split(url, '&');
for (auto arg : args) {
auto key_value = split(arg, '=');
if (key_value.first == "domain") {
username = key_value.second;
} else if (key_value.first == "channel") {
channel_id_slice = key_value.second;
}
}
} else {
// /username?boost
// /c/123456789?boost
CHECK(!url.empty() && url[0] == '/');
url.remove_prefix(1);
size_t username_end_pos = 0;
while (username_end_pos < url.size() && url[username_end_pos] != '/' && url[username_end_pos] != '?' &&
url[username_end_pos] != '#') {
username_end_pos++;
}
username = url.substr(0, username_end_pos);
url = url.substr(username_end_pos);
if (!url.empty() && url[0] == '/') {
url = url.substr(1);
}
if (username == "c") {
username = Slice();
size_t channel_id_end_pos = 0;
while (channel_id_end_pos < url.size() && url[channel_id_end_pos] != '/' && url[channel_id_end_pos] != '?' &&
url[channel_id_end_pos] != '#') {
channel_id_end_pos++;
}
channel_id_slice = url.substr(0, channel_id_end_pos);
url = url.substr(channel_id_end_pos);
}
bool is_boost = false;
auto query_pos = url.find('?');
if (query_pos != Slice::npos) {
auto args = full_split(url.substr(query_pos + 1), '&');
for (auto arg : args) {
auto key_value = split(arg, '=');
if (key_value.first == "boost") {
is_boost = true;
}
}
}
if (!is_boost) {
return Status::Error("Wrong chat boost link URL");
}
}
ChannelId channel_id;
if (username.empty()) {
auto r_channel_id = to_integer_safe<int64>(channel_id_slice);
if (r_channel_id.is_error() || !ChannelId(r_channel_id.ok()).is_valid()) {
return Status::Error("Wrong channel ID");
}
channel_id = ChannelId(r_channel_id.ok());
}
DialogBoostLinkInfo info;
info.username = username.str();
info.channel_id = channel_id;
LOG(INFO) << "Have link to boost chat @" << info.username << '/' << channel_id.get();
return std::move(info);
}
Result<MessageLinkInfo> LinkManager::get_message_link_info(Slice url) {
if (url.empty()) {
return Status::Error("URL must be non-empty");
}
auto link_info = get_link_info(url);
if (link_info.type_ != LinkType::Tg && link_info.type_ != LinkType::TMe) {
return Status::Error("Invalid message link URL");
}
url = link_info.query_;
Slice username;
Slice channel_id_slice;
Slice message_id_slice;
Slice comment_message_id_slice = "0";
Slice top_thread_message_id_slice;
Slice media_timestamp_slice;
bool is_single = false;
bool for_comment = false;
if (link_info.type_ == LinkType::Tg) {
// resolve?domain=username&post=12345&single&t=123&comment=12&thread=21
// privatepost?channel=123456789&post=12345&single&t=123&comment=12&thread=21
bool is_resolve = false;
if (begins_with(url, "resolve")) {
url = url.substr(7);
is_resolve = true;
} else if (begins_with(url, "privatepost")) {
url = url.substr(11);
} else {
return Status::Error("Wrong message link URL");
}
if (begins_with(url, "/")) {
url = url.substr(1);
}
if (!begins_with(url, "?")) {
return Status::Error("Wrong message link URL");
}
url = url.substr(1);
auto args = full_split(url, '&');
for (auto arg : args) {
auto key_value = split(arg, '=');
if (is_resolve) {
if (key_value.first == "domain") {
username = key_value.second;
}
} else {
if (key_value.first == "channel") {
channel_id_slice = key_value.second;
}
}
if (key_value.first == "post") {
message_id_slice = key_value.second;
}
if (key_value.first == "t") {
media_timestamp_slice = key_value.second;
}
if (key_value.first == "single") {
is_single = true;
}
if (key_value.first == "comment") {
comment_message_id_slice = key_value.second;
}
if (key_value.first == "thread") {
for_comment = true;
top_thread_message_id_slice = key_value.second;
}
}
} else {
// /c/123456789/12345
// /c/123456789/1234/12345
// /username/12345?single
CHECK(!url.empty() && url[0] == '/');
url.remove_prefix(1);
auto username_end_pos = url.find('/');
if (username_end_pos == Slice::npos) {
return Status::Error("Wrong message link URL");
}
username = url.substr(0, username_end_pos);
url = url.substr(username_end_pos + 1);
if (username == "c") {
username = Slice();
auto channel_id_end_pos = url.find('/');
if (channel_id_end_pos == Slice::npos) {
return Status::Error("Wrong message link URL");
}
channel_id_slice = url.substr(0, channel_id_end_pos);
url = url.substr(channel_id_end_pos + 1);
}
auto query_pos = url.find('?');
message_id_slice = url.substr(0, query_pos);
if (query_pos != Slice::npos) {
auto args = full_split(url.substr(query_pos + 1), '&');
for (auto arg : args) {
auto key_value = split(arg, '=');
if (key_value.first == "t") {
media_timestamp_slice = key_value.second;
}
if (key_value.first == "single") {
is_single = true;
}
if (key_value.first == "comment") {
comment_message_id_slice = key_value.second;
}
if (key_value.first == "thread") {
for_comment = true;
top_thread_message_id_slice = key_value.second;
}
}
}
auto slash_pos = message_id_slice.find('/');
if (slash_pos != Slice::npos) {
top_thread_message_id_slice = message_id_slice.substr(0, slash_pos);
message_id_slice.remove_prefix(slash_pos + 1);
}
}
ChannelId channel_id;
if (username.empty()) {
auto r_channel_id = to_integer_safe<int64>(channel_id_slice);
if (r_channel_id.is_error() || !ChannelId(r_channel_id.ok()).is_valid()) {
return Status::Error("Wrong channel ID");
}
channel_id = ChannelId(r_channel_id.ok());
}
auto r_message_id = to_integer_safe<int32>(message_id_slice);
if (r_message_id.is_error() || !ServerMessageId(r_message_id.ok()).is_valid()) {
return Status::Error("Wrong message ID");
}
int32 top_thread_message_id = 0;
if (!top_thread_message_id_slice.empty()) {
auto r_top_thread_message_id = to_integer_safe<int32>(top_thread_message_id_slice);
if (r_top_thread_message_id.is_error()) {
return Status::Error("Wrong message thread ID");
}
top_thread_message_id = r_top_thread_message_id.ok();
if (!ServerMessageId(top_thread_message_id).is_valid()) {
return Status::Error("Invalid message thread ID");
}
}
auto r_comment_message_id = to_integer_safe<int32>(comment_message_id_slice);
if (r_comment_message_id.is_error() ||
!(r_comment_message_id.ok() == 0 || ServerMessageId(r_comment_message_id.ok()).is_valid())) {
return Status::Error("Wrong comment message ID");
}
bool is_media_timestamp_invalid = false;
int32 media_timestamp = 0;
const int32 MAX_MEDIA_TIMESTAMP = 10000000;
if (!media_timestamp_slice.empty()) {
int32 current_value = 0;
for (size_t i = 0; i <= media_timestamp_slice.size(); i++) {
auto c = i < media_timestamp_slice.size() ? media_timestamp_slice[i] : 's';
if ('0' <= c && c <= '9') {
current_value = current_value * 10 + c - '0';
if (current_value > MAX_MEDIA_TIMESTAMP) {
is_media_timestamp_invalid = true;
break;
}
} else {
auto mul = 0;
switch (to_lower(c)) {
case 'h':
mul = 3600;
break;
case 'm':
mul = 60;
break;
case 's':
mul = 1;
break;
}
if (mul == 0 || current_value > MAX_MEDIA_TIMESTAMP / mul ||
media_timestamp + current_value * mul > MAX_MEDIA_TIMESTAMP) {
is_media_timestamp_invalid = true;
break;
}
media_timestamp += current_value * mul;
current_value = 0;
}
}
}
MessageLinkInfo info;
info.username = username.str();
info.channel_id = channel_id;
info.message_id = MessageId(ServerMessageId(r_message_id.ok()));
info.comment_message_id = MessageId(ServerMessageId(r_comment_message_id.ok()));
info.top_thread_message_id = MessageId(ServerMessageId(top_thread_message_id));
info.media_timestamp = is_media_timestamp_invalid ? 0 : media_timestamp;
info.is_single = is_single;
info.for_comment = for_comment;
LOG(INFO) << "Have link to " << info.message_id << " in chat @" << info.username << '/' << channel_id.get();
return std::move(info);
}
} // namespace td
Reference in New Issue View Git Blame Copy Permalink