695 lines
24 KiB
C++
695 lines
24 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023
|
|
//
|
|
// 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/AccountManager.h"
|
|
|
|
#include "td/telegram/ContactsManager.h"
|
|
#include "td/telegram/DeviceTokenManager.h"
|
|
#include "td/telegram/Global.h"
|
|
#include "td/telegram/net/NetQueryCreator.h"
|
|
#include "td/telegram/Td.h"
|
|
#include "td/telegram/telegram_api.h"
|
|
#include "td/telegram/UserId.h"
|
|
|
|
#include "td/utils/algorithm.h"
|
|
#include "td/utils/base64.h"
|
|
#include "td/utils/buffer.h"
|
|
#include "td/utils/logging.h"
|
|
#include "td/utils/misc.h"
|
|
#include "td/utils/Slice.h"
|
|
#include "td/utils/Status.h"
|
|
|
|
#include <algorithm>
|
|
|
|
namespace td {
|
|
|
|
static td_api::object_ptr<td_api::SessionType> get_session_type_object(
|
|
const tl_object_ptr<telegram_api::authorization> &authorization) {
|
|
auto contains = [](const string &str, const char *substr) {
|
|
return str.find(substr) != string::npos;
|
|
};
|
|
|
|
const string &app_name = authorization->app_name_;
|
|
auto device_model = to_lower(authorization->device_model_);
|
|
auto platform = to_lower(authorization->platform_);
|
|
auto system_version = to_lower(authorization->system_version_);
|
|
|
|
if (device_model.find("xbox") != string::npos) {
|
|
return td_api::make_object<td_api::sessionTypeXbox>();
|
|
}
|
|
|
|
bool is_web = [&] {
|
|
CSlice web_name("Web");
|
|
auto pos = app_name.find(web_name.c_str());
|
|
if (pos == string::npos) {
|
|
return false;
|
|
}
|
|
|
|
auto next_character = app_name[pos + web_name.size()];
|
|
return !('a' <= next_character && next_character <= 'z');
|
|
}();
|
|
|
|
if (is_web) {
|
|
if (contains(device_model, "brave")) {
|
|
return td_api::make_object<td_api::sessionTypeBrave>();
|
|
} else if (contains(device_model, "vivaldi")) {
|
|
return td_api::make_object<td_api::sessionTypeVivaldi>();
|
|
} else if (contains(device_model, "opera") || contains(device_model, "opr")) {
|
|
return td_api::make_object<td_api::sessionTypeOpera>();
|
|
} else if (contains(device_model, "edg")) {
|
|
return td_api::make_object<td_api::sessionTypeEdge>();
|
|
} else if (contains(device_model, "chrome")) {
|
|
return td_api::make_object<td_api::sessionTypeChrome>();
|
|
} else if (contains(device_model, "firefox") || contains(device_model, "fxios")) {
|
|
return td_api::make_object<td_api::sessionTypeFirefox>();
|
|
} else if (contains(device_model, "safari")) {
|
|
return td_api::make_object<td_api::sessionTypeSafari>();
|
|
}
|
|
}
|
|
|
|
if (begins_with(platform, "android") || contains(system_version, "android")) {
|
|
return td_api::make_object<td_api::sessionTypeAndroid>();
|
|
} else if (begins_with(platform, "windows") || contains(system_version, "windows")) {
|
|
return td_api::make_object<td_api::sessionTypeWindows>();
|
|
} else if (begins_with(platform, "ubuntu") || contains(system_version, "ubuntu")) {
|
|
return td_api::make_object<td_api::sessionTypeUbuntu>();
|
|
} else if (begins_with(platform, "linux") || contains(system_version, "linux")) {
|
|
return td_api::make_object<td_api::sessionTypeLinux>();
|
|
}
|
|
|
|
auto is_ios = begins_with(platform, "ios") || contains(system_version, "ios");
|
|
auto is_macos = begins_with(platform, "macos") || contains(system_version, "macos");
|
|
if (is_ios && contains(device_model, "iphone")) {
|
|
return td_api::make_object<td_api::sessionTypeIphone>();
|
|
} else if (is_ios && contains(device_model, "ipad")) {
|
|
return td_api::make_object<td_api::sessionTypeIpad>();
|
|
} else if (is_macos && contains(device_model, "mac")) {
|
|
return td_api::make_object<td_api::sessionTypeMac>();
|
|
} else if (is_ios || is_macos) {
|
|
return td_api::make_object<td_api::sessionTypeApple>();
|
|
}
|
|
|
|
return td_api::make_object<td_api::sessionTypeUnknown>();
|
|
}
|
|
|
|
static td_api::object_ptr<td_api::session> convert_authorization_object(
|
|
tl_object_ptr<telegram_api::authorization> &&authorization) {
|
|
CHECK(authorization != nullptr);
|
|
return td_api::make_object<td_api::session>(
|
|
authorization->hash_, authorization->current_, authorization->password_pending_, authorization->unconfirmed_,
|
|
!authorization->encrypted_requests_disabled_, !authorization->call_requests_disabled_,
|
|
get_session_type_object(authorization), authorization->api_id_, authorization->app_name_,
|
|
authorization->app_version_, authorization->official_app_, authorization->device_model_, authorization->platform_,
|
|
authorization->system_version_, authorization->date_created_, authorization->date_active_, authorization->ip_,
|
|
authorization->country_, authorization->region_);
|
|
}
|
|
|
|
class SetDefaultHistoryTtlQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit SetDefaultHistoryTtlQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(int32 account_ttl) {
|
|
send_query(G()->net_query_creator().create(telegram_api::messages_setDefaultHistoryTTL(account_ttl), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::messages_setDefaultHistoryTTL>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.move_as_ok();
|
|
if (!result) {
|
|
return on_error(Status::Error(500, "Internal Server Error: failed to set default message TTL"));
|
|
}
|
|
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetDefaultHistoryTtlQuery final : public Td::ResultHandler {
|
|
Promise<int32> promise_;
|
|
|
|
public:
|
|
explicit GetDefaultHistoryTtlQuery(Promise<int32> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::messages_getDefaultHistoryTTL()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::messages_getDefaultHistoryTTL>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto ptr = result_ptr.move_as_ok();
|
|
LOG(INFO) << "Receive result for GetDefaultHistoryTtlQuery: " << to_string(ptr);
|
|
|
|
promise_.set_value(std::move(ptr->period_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class SetAccountTtlQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit SetAccountTtlQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(int32 account_ttl) {
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::account_setAccountTTL(make_tl_object<telegram_api::accountDaysTTL>(account_ttl)), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_setAccountTTL>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.move_as_ok();
|
|
if (!result) {
|
|
return on_error(Status::Error(500, "Internal Server Error: failed to set account TTL"));
|
|
}
|
|
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetAccountTtlQuery final : public Td::ResultHandler {
|
|
Promise<int32> promise_;
|
|
|
|
public:
|
|
explicit GetAccountTtlQuery(Promise<int32> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::account_getAccountTTL()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_getAccountTTL>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto ptr = result_ptr.move_as_ok();
|
|
LOG(INFO) << "Receive result for GetAccountTtlQuery: " << to_string(ptr);
|
|
|
|
promise_.set_value(std::move(ptr->days_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class AcceptLoginTokenQuery final : public Td::ResultHandler {
|
|
Promise<td_api::object_ptr<td_api::session>> promise_;
|
|
|
|
public:
|
|
explicit AcceptLoginTokenQuery(Promise<td_api::object_ptr<td_api::session>> &&promise)
|
|
: promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(const string &login_token) {
|
|
send_query(G()->net_query_creator().create(telegram_api::auth_acceptLoginToken(BufferSlice(login_token))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::auth_acceptLoginToken>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for AcceptLoginTokenQuery: " << to_string(result_ptr.ok());
|
|
promise_.set_value(convert_authorization_object(result_ptr.move_as_ok()));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetAuthorizationsQuery final : public Td::ResultHandler {
|
|
Promise<td_api::object_ptr<td_api::sessions>> promise_;
|
|
|
|
public:
|
|
explicit GetAuthorizationsQuery(Promise<td_api::object_ptr<td_api::sessions>> &&promise)
|
|
: promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::account_getAuthorizations()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_getAuthorizations>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto ptr = result_ptr.move_as_ok();
|
|
LOG(INFO) << "Receive result for GetAuthorizationsQuery: " << to_string(ptr);
|
|
|
|
auto ttl_days = ptr->authorization_ttl_days_;
|
|
if (ttl_days <= 0 || ttl_days > 366) {
|
|
LOG(ERROR) << "Receive invalid inactive sessions TTL " << ttl_days;
|
|
ttl_days = 180;
|
|
}
|
|
|
|
auto results = td_api::make_object<td_api::sessions>(
|
|
transform(std::move(ptr->authorizations_), convert_authorization_object), ttl_days);
|
|
std::sort(results->sessions_.begin(), results->sessions_.end(),
|
|
[](const td_api::object_ptr<td_api::session> &lhs, const td_api::object_ptr<td_api::session> &rhs) {
|
|
if (lhs->is_current_ != rhs->is_current_) {
|
|
return lhs->is_current_;
|
|
}
|
|
if (lhs->is_password_pending_ != rhs->is_password_pending_) {
|
|
return lhs->is_password_pending_;
|
|
}
|
|
return lhs->last_active_date_ > rhs->last_active_date_;
|
|
});
|
|
|
|
promise_.set_value(std::move(results));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ResetAuthorizationQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit ResetAuthorizationQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(int64 authorization_id) {
|
|
send_query(G()->net_query_creator().create(telegram_api::account_resetAuthorization(authorization_id)));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_resetAuthorization>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.move_as_ok();
|
|
LOG_IF(WARNING, !result) << "Failed to terminate session";
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ResetAuthorizationsQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit ResetAuthorizationsQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::auth_resetAuthorizations()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::auth_resetAuthorizations>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.move_as_ok();
|
|
LOG_IF(WARNING, !result) << "Failed to terminate all sessions";
|
|
send_closure(td_->device_token_manager_, &DeviceTokenManager::reregister_device);
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ChangeAuthorizationSettingsQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit ChangeAuthorizationSettingsQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(int64 hash, bool set_encrypted_requests_disabled, bool encrypted_requests_disabled,
|
|
bool set_call_requests_disabled, bool call_requests_disabled) {
|
|
int32 flags = 0;
|
|
if (set_encrypted_requests_disabled) {
|
|
flags |= telegram_api::account_changeAuthorizationSettings::ENCRYPTED_REQUESTS_DISABLED_MASK;
|
|
}
|
|
if (set_call_requests_disabled) {
|
|
flags |= telegram_api::account_changeAuthorizationSettings::CALL_REQUESTS_DISABLED_MASK;
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::account_changeAuthorizationSettings(flags, false /*ignored*/, hash, encrypted_requests_disabled,
|
|
call_requests_disabled),
|
|
{{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_changeAuthorizationSettings>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.move_as_ok();
|
|
LOG_IF(WARNING, !result) << "Failed to change session settings";
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class SetAuthorizationTtlQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit SetAuthorizationTtlQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(int32 authorization_ttl_days) {
|
|
send_query(
|
|
G()->net_query_creator().create(telegram_api::account_setAuthorizationTTL(authorization_ttl_days), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_setAuthorizationTTL>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.move_as_ok();
|
|
LOG_IF(WARNING, !result) << "Failed to set inactive session TTL";
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetWebAuthorizationsQuery final : public Td::ResultHandler {
|
|
Promise<td_api::object_ptr<td_api::connectedWebsites>> promise_;
|
|
|
|
public:
|
|
explicit GetWebAuthorizationsQuery(Promise<td_api::object_ptr<td_api::connectedWebsites>> &&promise)
|
|
: promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::account_getWebAuthorizations()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_getWebAuthorizations>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto ptr = result_ptr.move_as_ok();
|
|
LOG(INFO) << "Receive result for GetWebAuthorizationsQuery: " << to_string(ptr);
|
|
|
|
td_->contacts_manager_->on_get_users(std::move(ptr->users_), "GetWebAuthorizationsQuery");
|
|
|
|
auto results = td_api::make_object<td_api::connectedWebsites>();
|
|
results->websites_.reserve(ptr->authorizations_.size());
|
|
for (auto &authorization : ptr->authorizations_) {
|
|
CHECK(authorization != nullptr);
|
|
UserId bot_user_id(authorization->bot_id_);
|
|
if (!bot_user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid bot " << bot_user_id;
|
|
bot_user_id = UserId();
|
|
}
|
|
|
|
results->websites_.push_back(td_api::make_object<td_api::connectedWebsite>(
|
|
authorization->hash_, authorization->domain_,
|
|
td_->contacts_manager_->get_user_id_object(bot_user_id, "GetWebAuthorizationsQuery"), authorization->browser_,
|
|
authorization->platform_, authorization->date_created_, authorization->date_active_, authorization->ip_,
|
|
authorization->region_));
|
|
}
|
|
|
|
promise_.set_value(std::move(results));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ResetWebAuthorizationQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit ResetWebAuthorizationQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(int64 hash) {
|
|
send_query(G()->net_query_creator().create(telegram_api::account_resetWebAuthorization(hash)));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_resetWebAuthorization>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.move_as_ok();
|
|
LOG_IF(WARNING, !result) << "Failed to disconnect website";
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ResetWebAuthorizationsQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit ResetWebAuthorizationsQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::account_resetWebAuthorizations()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_resetWebAuthorizations>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.move_as_ok();
|
|
LOG_IF(WARNING, !result) << "Failed to disconnect all websites";
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ExportContactTokenQuery final : public Td::ResultHandler {
|
|
Promise<td_api::object_ptr<td_api::userLink>> promise_;
|
|
|
|
public:
|
|
explicit ExportContactTokenQuery(Promise<td_api::object_ptr<td_api::userLink>> &&promise)
|
|
: promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_exportContactToken()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_exportContactToken>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto ptr = result_ptr.move_as_ok();
|
|
LOG(INFO) << "Receive result for ExportContactTokenQuery: " << to_string(ptr);
|
|
promise_.set_value(td_api::make_object<td_api::userLink>(
|
|
ptr->url_, td::max(static_cast<int32>(ptr->expires_ - G()->unix_time()), static_cast<int32>(1))));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ImportContactTokenQuery final : public Td::ResultHandler {
|
|
Promise<td_api::object_ptr<td_api::user>> promise_;
|
|
|
|
public:
|
|
explicit ImportContactTokenQuery(Promise<td_api::object_ptr<td_api::user>> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(const string &token) {
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_importContactToken(token)));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_importContactToken>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto user = result_ptr.move_as_ok();
|
|
LOG(DEBUG) << "Receive result for ImportContactTokenQuery: " << to_string(user);
|
|
|
|
auto user_id = ContactsManager::get_user_id(user);
|
|
td_->contacts_manager_->on_get_user(std::move(user), "ImportContactTokenQuery");
|
|
promise_.set_value(td_->contacts_manager_->get_user_object(user_id));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class InvalidateSignInCodesQuery final : public Td::ResultHandler {
|
|
public:
|
|
void send(vector<string> &&codes) {
|
|
send_query(G()->net_query_creator().create(telegram_api::account_invalidateSignInCodes(std::move(codes))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_invalidateSignInCodes>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for InvalidateSignInCodesQuery: " << result_ptr.ok();
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
LOG(DEBUG) << "Receive error for InvalidateSignInCodesQuery: " << status;
|
|
}
|
|
};
|
|
|
|
AccountManager::AccountManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
|
|
}
|
|
|
|
void AccountManager::tear_down() {
|
|
parent_.reset();
|
|
}
|
|
|
|
void AccountManager::set_default_message_ttl(int32 message_ttl, Promise<Unit> &&promise) {
|
|
td_->create_handler<SetDefaultHistoryTtlQuery>(std::move(promise))->send(message_ttl);
|
|
}
|
|
|
|
void AccountManager::get_default_message_ttl(Promise<int32> &&promise) {
|
|
td_->create_handler<GetDefaultHistoryTtlQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
void AccountManager::set_account_ttl(int32 account_ttl, Promise<Unit> &&promise) {
|
|
td_->create_handler<SetAccountTtlQuery>(std::move(promise))->send(account_ttl);
|
|
}
|
|
|
|
void AccountManager::get_account_ttl(Promise<int32> &&promise) {
|
|
td_->create_handler<GetAccountTtlQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
void AccountManager::confirm_qr_code_authentication(const string &link,
|
|
Promise<td_api::object_ptr<td_api::session>> &&promise) {
|
|
Slice prefix("tg://login?token=");
|
|
if (!begins_with(to_lower(link), prefix)) {
|
|
return promise.set_error(Status::Error(400, "AUTH_TOKEN_INVALID"));
|
|
}
|
|
auto r_token = base64url_decode(Slice(link).substr(prefix.size()));
|
|
if (r_token.is_error()) {
|
|
return promise.set_error(Status::Error(400, "AUTH_TOKEN_INVALID"));
|
|
}
|
|
td_->create_handler<AcceptLoginTokenQuery>(std::move(promise))->send(r_token.ok());
|
|
}
|
|
|
|
void AccountManager::get_active_sessions(Promise<td_api::object_ptr<td_api::sessions>> &&promise) {
|
|
td_->create_handler<GetAuthorizationsQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
void AccountManager::terminate_session(int64 session_id, Promise<Unit> &&promise) {
|
|
td_->create_handler<ResetAuthorizationQuery>(std::move(promise))->send(session_id);
|
|
}
|
|
|
|
void AccountManager::terminate_all_other_sessions(Promise<Unit> &&promise) {
|
|
td_->create_handler<ResetAuthorizationsQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
void AccountManager::toggle_session_can_accept_calls(int64 session_id, bool can_accept_calls, Promise<Unit> &&promise) {
|
|
td_->create_handler<ChangeAuthorizationSettingsQuery>(std::move(promise))
|
|
->send(session_id, false, false, true, !can_accept_calls);
|
|
}
|
|
|
|
void AccountManager::toggle_session_can_accept_secret_chats(int64 session_id, bool can_accept_secret_chats,
|
|
Promise<Unit> &&promise) {
|
|
td_->create_handler<ChangeAuthorizationSettingsQuery>(std::move(promise))
|
|
->send(session_id, true, !can_accept_secret_chats, false, false);
|
|
}
|
|
|
|
void AccountManager::set_inactive_session_ttl_days(int32 authorization_ttl_days, Promise<Unit> &&promise) {
|
|
td_->create_handler<SetAuthorizationTtlQuery>(std::move(promise))->send(authorization_ttl_days);
|
|
}
|
|
|
|
void AccountManager::get_connected_websites(Promise<td_api::object_ptr<td_api::connectedWebsites>> &&promise) {
|
|
td_->create_handler<GetWebAuthorizationsQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
void AccountManager::disconnect_website(int64 website_id, Promise<Unit> &&promise) {
|
|
td_->create_handler<ResetWebAuthorizationQuery>(std::move(promise))->send(website_id);
|
|
}
|
|
|
|
void AccountManager::disconnect_all_websites(Promise<Unit> &&promise) {
|
|
td_->create_handler<ResetWebAuthorizationsQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
void AccountManager::export_contact_token(Promise<td_api::object_ptr<td_api::userLink>> &&promise) {
|
|
td_->create_handler<ExportContactTokenQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
void AccountManager::import_contact_token(const string &token, Promise<td_api::object_ptr<td_api::user>> &&promise) {
|
|
td_->create_handler<ImportContactTokenQuery>(std::move(promise))->send(token);
|
|
}
|
|
|
|
void AccountManager::invalidate_authentication_codes(vector<string> &&authentication_codes) {
|
|
td_->create_handler<InvalidateSignInCodesQuery>()->send(std::move(authentication_codes));
|
|
}
|
|
|
|
void AccountManager::get_current_state(vector<td_api::object_ptr<td_api::Update>> &updates) const {
|
|
}
|
|
|
|
} // namespace td
|