8144 lines
303 KiB
C++
8144 lines
303 KiB
C++
//
|
|
// 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/UserManager.h"
|
|
|
|
#include "td/telegram/AnimationsManager.h"
|
|
#include "td/telegram/AuthManager.h"
|
|
#include "td/telegram/Birthdate.hpp"
|
|
#include "td/telegram/BlockListId.h"
|
|
#include "td/telegram/BotMenuButton.h"
|
|
#include "td/telegram/BusinessAwayMessage.h"
|
|
#include "td/telegram/BusinessGreetingMessage.h"
|
|
#include "td/telegram/BusinessInfo.h"
|
|
#include "td/telegram/BusinessInfo.hpp"
|
|
#include "td/telegram/BusinessIntro.h"
|
|
#include "td/telegram/BusinessWorkHours.h"
|
|
#include "td/telegram/ChannelType.h"
|
|
#include "td/telegram/ChatManager.h"
|
|
#include "td/telegram/CommonDialogManager.h"
|
|
#include "td/telegram/ConfigManager.h"
|
|
#include "td/telegram/Dependencies.h"
|
|
#include "td/telegram/DialogLocation.h"
|
|
#include "td/telegram/DialogManager.h"
|
|
#include "td/telegram/DialogParticipantManager.h"
|
|
#include "td/telegram/Document.h"
|
|
#include "td/telegram/DocumentsManager.h"
|
|
#include "td/telegram/FileReferenceManager.h"
|
|
#include "td/telegram/files/FileManager.h"
|
|
#include "td/telegram/files/FileType.h"
|
|
#include "td/telegram/FolderId.h"
|
|
#include "td/telegram/Global.h"
|
|
#include "td/telegram/GroupCallManager.h"
|
|
#include "td/telegram/InlineQueriesManager.h"
|
|
#include "td/telegram/LinkManager.h"
|
|
#include "td/telegram/logevent/LogEvent.h"
|
|
#include "td/telegram/logevent/LogEventHelper.h"
|
|
#include "td/telegram/MessageId.h"
|
|
#include "td/telegram/MessagesManager.h"
|
|
#include "td/telegram/MessageTtl.h"
|
|
#include "td/telegram/misc.h"
|
|
#include "td/telegram/net/NetQuery.h"
|
|
#include "td/telegram/NotificationManager.h"
|
|
#include "td/telegram/OnlineManager.h"
|
|
#include "td/telegram/OptionManager.h"
|
|
#include "td/telegram/PeerColor.h"
|
|
#include "td/telegram/PeopleNearbyManager.h"
|
|
#include "td/telegram/Photo.h"
|
|
#include "td/telegram/Photo.hpp"
|
|
#include "td/telegram/PhotoSize.h"
|
|
#include "td/telegram/PremiumGiftOption.hpp"
|
|
#include "td/telegram/ReactionListType.h"
|
|
#include "td/telegram/ReactionManager.h"
|
|
#include "td/telegram/SecretChatLayer.h"
|
|
#include "td/telegram/SecretChatsManager.h"
|
|
#include "td/telegram/ServerMessageId.h"
|
|
#include "td/telegram/StickerPhotoSize.h"
|
|
#include "td/telegram/StoryManager.h"
|
|
#include "td/telegram/SuggestedAction.h"
|
|
#include "td/telegram/Td.h"
|
|
#include "td/telegram/TdDb.h"
|
|
#include "td/telegram/telegram_api.h"
|
|
#include "td/telegram/ThemeManager.h"
|
|
#include "td/telegram/UpdatesManager.h"
|
|
#include "td/telegram/Version.h"
|
|
|
|
#include "td/db/binlog/BinlogEvent.h"
|
|
#include "td/db/binlog/BinlogHelper.h"
|
|
#include "td/db/SqliteKeyValue.h"
|
|
#include "td/db/SqliteKeyValueAsync.h"
|
|
|
|
#include "td/utils/algorithm.h"
|
|
#include "td/utils/buffer.h"
|
|
#include "td/utils/format.h"
|
|
#include "td/utils/logging.h"
|
|
#include "td/utils/misc.h"
|
|
#include "td/utils/Random.h"
|
|
#include "td/utils/ScopeGuard.h"
|
|
#include "td/utils/Slice.h"
|
|
#include "td/utils/SliceBuilder.h"
|
|
#include "td/utils/StringBuilder.h"
|
|
#include "td/utils/Time.h"
|
|
#include "td/utils/tl_helpers.h"
|
|
#include "td/utils/utf8.h"
|
|
|
|
#include <algorithm>
|
|
#include <limits>
|
|
#include <type_traits>
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
#include <utility>
|
|
|
|
namespace td {
|
|
|
|
class GetContactsQuery final : public Td::ResultHandler {
|
|
public:
|
|
void send(int64 hash) {
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_getContacts(hash)));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_getContacts>(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 GetContactsQuery: " << to_string(ptr);
|
|
td_->user_manager_->on_get_contacts(std::move(ptr));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
td_->user_manager_->on_get_contacts_failed(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetContactsBirthdaysQuery final : public Td::ResultHandler {
|
|
public:
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_getBirthdays()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_getBirthdays>(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 GetContactsBirthdaysQuery: " << to_string(ptr);
|
|
td_->user_manager_->on_get_contact_birthdates(std::move(ptr));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
td_->user_manager_->on_get_contact_birthdates(nullptr);
|
|
}
|
|
};
|
|
|
|
class DismissContactBirthdaysSuggestionQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit DismissContactBirthdaysSuggestionQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::help_dismissSuggestion(
|
|
telegram_api::make_object<telegram_api::inputPeerEmpty>(), "BIRTHDAY_CONTACTS_TODAY")));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::help_dismissSuggestion>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetContactsStatusesQuery final : public Td::ResultHandler {
|
|
public:
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_getStatuses()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_getStatuses>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
td_->user_manager_->on_get_contacts_statuses(result_ptr.move_as_ok());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (!G()->is_expected_error(status)) {
|
|
LOG(ERROR) << "Receive error for GetContactsStatusesQuery: " << status;
|
|
}
|
|
}
|
|
};
|
|
|
|
class AddContactQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
UserId user_id_;
|
|
|
|
public:
|
|
explicit AddContactQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(UserId user_id, telegram_api::object_ptr<telegram_api::InputUser> &&input_user, const Contact &contact,
|
|
bool share_phone_number) {
|
|
user_id_ = user_id;
|
|
int32 flags = 0;
|
|
if (share_phone_number) {
|
|
flags |= telegram_api::contacts_addContact::ADD_PHONE_PRIVACY_EXCEPTION_MASK;
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::contacts_addContact(flags, false /*ignored*/, std::move(input_user), contact.get_first_name(),
|
|
contact.get_last_name(), contact.get_phone_number()),
|
|
{{DialogId(user_id)}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_addContact>(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 AddContactQuery: " << to_string(ptr);
|
|
td_->updates_manager_->on_get_updates(std::move(ptr), std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
td_->user_manager_->reload_contacts(true);
|
|
td_->messages_manager_->reget_dialog_action_bar(DialogId(user_id_), "AddContactQuery");
|
|
}
|
|
};
|
|
|
|
class EditCloseFriendsQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
vector<UserId> user_ids_;
|
|
|
|
public:
|
|
explicit EditCloseFriendsQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(vector<UserId> user_ids) {
|
|
user_ids_ = std::move(user_ids);
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::contacts_editCloseFriends(UserId::get_input_user_ids(user_ids_))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_editCloseFriends>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
td_->user_manager_->on_set_close_friends(user_ids_, std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ResolvePhoneQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
string phone_number_;
|
|
|
|
public:
|
|
explicit ResolvePhoneQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(const string &phone_number) {
|
|
phone_number_ = phone_number;
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_resolvePhone(phone_number)));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_resolvePhone>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto ptr = result_ptr.move_as_ok();
|
|
LOG(DEBUG) << "Receive result for ResolvePhoneQuery: " << to_string(ptr);
|
|
td_->user_manager_->on_get_users(std::move(ptr->users_), "ResolvePhoneQuery");
|
|
// on_get_chats(std::move(ptr->chats_), "ResolvePhoneQuery");
|
|
|
|
DialogId dialog_id(ptr->peer_);
|
|
if (dialog_id.get_type() != DialogType::User) {
|
|
LOG(ERROR) << "Receive " << dialog_id << " by " << phone_number_;
|
|
return on_error(Status::Error(500, "Receive invalid response"));
|
|
}
|
|
|
|
td_->user_manager_->on_resolved_phone_number(phone_number_, dialog_id.get_user_id());
|
|
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (status.message() == Slice("PHONE_NOT_OCCUPIED")) {
|
|
td_->user_manager_->on_resolved_phone_number(phone_number_, UserId());
|
|
return promise_.set_value(Unit());
|
|
}
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class AcceptContactQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
UserId user_id_;
|
|
|
|
public:
|
|
explicit AcceptContactQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(UserId user_id, telegram_api::object_ptr<telegram_api::InputUser> &&input_user) {
|
|
user_id_ = user_id;
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_acceptContact(std::move(input_user))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_acceptContact>(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 AcceptContactQuery: " << to_string(ptr);
|
|
td_->updates_manager_->on_get_updates(std::move(ptr), std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
td_->user_manager_->reload_contacts(true);
|
|
td_->messages_manager_->reget_dialog_action_bar(DialogId(user_id_), "AcceptContactQuery");
|
|
}
|
|
};
|
|
|
|
class ImportContactsQuery final : public Td::ResultHandler {
|
|
int64 random_id_ = 0;
|
|
size_t sent_size_ = 0;
|
|
|
|
public:
|
|
void send(vector<telegram_api::object_ptr<telegram_api::inputPhoneContact>> &&input_phone_contacts, int64 random_id) {
|
|
random_id_ = random_id;
|
|
sent_size_ = input_phone_contacts.size();
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_importContacts(std::move(input_phone_contacts))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_importContacts>(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 ImportContactsQuery: " << to_string(ptr);
|
|
if (sent_size_ == ptr->retry_contacts_.size()) {
|
|
return on_error(Status::Error(429, "Too Many Requests: retry after 3600"));
|
|
}
|
|
td_->user_manager_->on_imported_contacts(random_id_, std::move(ptr));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
td_->user_manager_->on_imported_contacts(random_id_, std::move(status));
|
|
}
|
|
};
|
|
|
|
class DeleteContactsQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit DeleteContactsQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(vector<telegram_api::object_ptr<telegram_api::InputUser>> &&input_users) {
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_deleteContacts(std::move(input_users))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_deleteContacts>(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 DeleteContactsQuery: " << to_string(ptr);
|
|
td_->updates_manager_->on_get_updates(std::move(ptr), std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
td_->user_manager_->reload_contacts(true);
|
|
}
|
|
};
|
|
|
|
class DeleteContactsByPhoneNumberQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
vector<UserId> user_ids_;
|
|
|
|
public:
|
|
explicit DeleteContactsByPhoneNumberQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(vector<string> &&user_phone_numbers, vector<UserId> &&user_ids) {
|
|
if (user_phone_numbers.empty()) {
|
|
return promise_.set_value(Unit());
|
|
}
|
|
user_ids_ = std::move(user_ids);
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_deleteByPhones(std::move(user_phone_numbers))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_deleteByPhones>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.ok();
|
|
if (!result) {
|
|
return on_error(Status::Error(500, "Some contacts can't be deleted"));
|
|
}
|
|
|
|
td_->user_manager_->on_deleted_contacts(user_ids_);
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
td_->user_manager_->reload_contacts(true);
|
|
}
|
|
};
|
|
|
|
class ResetContactsQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit ResetContactsQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::contacts_resetSaved()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::contacts_resetSaved>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.ok();
|
|
if (!result) {
|
|
LOG(ERROR) << "Failed to delete imported contacts";
|
|
td_->user_manager_->reload_contacts(true);
|
|
} else {
|
|
td_->user_manager_->on_update_contacts_reset();
|
|
}
|
|
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
td_->user_manager_->reload_contacts(true);
|
|
}
|
|
};
|
|
|
|
class UploadProfilePhotoQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
UserId user_id_;
|
|
FileId file_id_;
|
|
bool is_fallback_;
|
|
bool only_suggest_;
|
|
|
|
public:
|
|
explicit UploadProfilePhotoQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(UserId user_id, FileId file_id, telegram_api::object_ptr<telegram_api::InputFile> &&input_file,
|
|
bool is_fallback, bool only_suggest, bool is_animation, double main_frame_timestamp) {
|
|
CHECK(input_file != nullptr);
|
|
CHECK(file_id.is_valid());
|
|
|
|
user_id_ = user_id;
|
|
file_id_ = file_id;
|
|
is_fallback_ = is_fallback;
|
|
only_suggest_ = only_suggest;
|
|
|
|
static_assert(static_cast<int32>(telegram_api::photos_uploadProfilePhoto::VIDEO_MASK) ==
|
|
static_cast<int32>(telegram_api::photos_uploadContactProfilePhoto::VIDEO_MASK),
|
|
"");
|
|
static_assert(static_cast<int32>(telegram_api::photos_uploadProfilePhoto::VIDEO_START_TS_MASK) ==
|
|
static_cast<int32>(telegram_api::photos_uploadContactProfilePhoto::VIDEO_START_TS_MASK),
|
|
"");
|
|
static_assert(static_cast<int32>(telegram_api::photos_uploadProfilePhoto::FILE_MASK) ==
|
|
static_cast<int32>(telegram_api::photos_uploadContactProfilePhoto::FILE_MASK),
|
|
"");
|
|
|
|
int32 flags = 0;
|
|
telegram_api::object_ptr<telegram_api::InputFile> photo_input_file;
|
|
telegram_api::object_ptr<telegram_api::InputFile> video_input_file;
|
|
if (is_animation) {
|
|
flags |= telegram_api::photos_uploadProfilePhoto::VIDEO_MASK;
|
|
video_input_file = std::move(input_file);
|
|
|
|
if (main_frame_timestamp != 0.0) {
|
|
flags |= telegram_api::photos_uploadProfilePhoto::VIDEO_START_TS_MASK;
|
|
}
|
|
} else {
|
|
flags |= telegram_api::photos_uploadProfilePhoto::FILE_MASK;
|
|
photo_input_file = std::move(input_file);
|
|
}
|
|
if (td_->user_manager_->is_user_bot(user_id)) {
|
|
auto r_input_user = td_->user_manager_->get_input_user(user_id);
|
|
if (r_input_user.is_error()) {
|
|
return on_error(r_input_user.move_as_error());
|
|
}
|
|
flags |= telegram_api::photos_uploadProfilePhoto::BOT_MASK;
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_uploadProfilePhoto(flags, false /*ignored*/, r_input_user.move_as_ok(),
|
|
std::move(photo_input_file), std::move(video_input_file),
|
|
main_frame_timestamp, nullptr),
|
|
{{user_id}}));
|
|
} else if (user_id == td_->user_manager_->get_my_id()) {
|
|
if (is_fallback) {
|
|
flags |= telegram_api::photos_uploadProfilePhoto::FALLBACK_MASK;
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_uploadProfilePhoto(flags, false /*ignored*/, nullptr, std::move(photo_input_file),
|
|
std::move(video_input_file), main_frame_timestamp, nullptr),
|
|
{{"me"}}));
|
|
} else {
|
|
if (only_suggest) {
|
|
flags |= telegram_api::photos_uploadContactProfilePhoto::SUGGEST_MASK;
|
|
} else {
|
|
flags |= telegram_api::photos_uploadContactProfilePhoto::SAVE_MASK;
|
|
}
|
|
auto r_input_user = td_->user_manager_->get_input_user(user_id);
|
|
if (r_input_user.is_error()) {
|
|
return on_error(r_input_user.move_as_error());
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_uploadContactProfilePhoto(flags, false /*ignored*/, false /*ignored*/,
|
|
r_input_user.move_as_ok(), std::move(photo_input_file),
|
|
std::move(video_input_file), main_frame_timestamp, nullptr),
|
|
{{user_id}}));
|
|
}
|
|
}
|
|
|
|
void send(UserId user_id, unique_ptr<StickerPhotoSize> sticker_photo_size, bool is_fallback, bool only_suggest) {
|
|
CHECK(sticker_photo_size != nullptr);
|
|
user_id_ = user_id;
|
|
file_id_ = FileId();
|
|
is_fallback_ = is_fallback;
|
|
only_suggest_ = only_suggest;
|
|
|
|
if (td_->user_manager_->is_user_bot(user_id)) {
|
|
auto r_input_user = td_->user_manager_->get_input_user(user_id);
|
|
if (r_input_user.is_error()) {
|
|
return on_error(r_input_user.move_as_error());
|
|
}
|
|
int32 flags = telegram_api::photos_uploadProfilePhoto::VIDEO_EMOJI_MARKUP_MASK;
|
|
flags |= telegram_api::photos_uploadProfilePhoto::BOT_MASK;
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_uploadProfilePhoto(flags, false /*ignored*/, r_input_user.move_as_ok(), nullptr, nullptr,
|
|
0.0, sticker_photo_size->get_input_video_size_object(td_)),
|
|
{{user_id}}));
|
|
} else if (user_id == td_->user_manager_->get_my_id()) {
|
|
int32 flags = telegram_api::photos_uploadProfilePhoto::VIDEO_EMOJI_MARKUP_MASK;
|
|
if (is_fallback) {
|
|
flags |= telegram_api::photos_uploadProfilePhoto::FALLBACK_MASK;
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_uploadProfilePhoto(flags, false /*ignored*/, nullptr, nullptr, nullptr, 0.0,
|
|
sticker_photo_size->get_input_video_size_object(td_)),
|
|
{{"me"}}));
|
|
} else {
|
|
int32 flags = telegram_api::photos_uploadContactProfilePhoto::VIDEO_EMOJI_MARKUP_MASK;
|
|
if (only_suggest) {
|
|
flags |= telegram_api::photos_uploadContactProfilePhoto::SUGGEST_MASK;
|
|
} else {
|
|
flags |= telegram_api::photos_uploadContactProfilePhoto::SAVE_MASK;
|
|
}
|
|
auto r_input_user = td_->user_manager_->get_input_user(user_id);
|
|
if (r_input_user.is_error()) {
|
|
return on_error(r_input_user.move_as_error());
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_uploadContactProfilePhoto(flags, false /*ignored*/, false /*ignored*/,
|
|
r_input_user.move_as_ok(), nullptr, nullptr, 0.0,
|
|
sticker_photo_size->get_input_video_size_object(td_)),
|
|
{{user_id}}));
|
|
}
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
static_assert(std::is_same<telegram_api::photos_uploadProfilePhoto::ReturnType,
|
|
telegram_api::photos_uploadContactProfilePhoto::ReturnType>::value,
|
|
"");
|
|
auto result_ptr = fetch_result<telegram_api::photos_uploadProfilePhoto>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
if (!only_suggest_) {
|
|
td_->user_manager_->on_set_profile_photo(user_id_, result_ptr.move_as_ok(), is_fallback_, 0, std::move(promise_));
|
|
} else {
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
if (file_id_.is_valid()) {
|
|
td_->file_manager_->delete_partial_remote_location(file_id_);
|
|
}
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (file_id_.is_valid()) {
|
|
td_->file_manager_->delete_partial_remote_location(file_id_);
|
|
}
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class UpdateProfilePhotoQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
UserId user_id_;
|
|
FileId file_id_;
|
|
int64 old_photo_id_;
|
|
bool is_fallback_;
|
|
string file_reference_;
|
|
|
|
public:
|
|
explicit UpdateProfilePhotoQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(UserId user_id, FileId file_id, int64 old_photo_id, bool is_fallback,
|
|
telegram_api::object_ptr<telegram_api::InputPhoto> &&input_photo) {
|
|
CHECK(input_photo != nullptr);
|
|
user_id_ = user_id;
|
|
file_id_ = file_id;
|
|
old_photo_id_ = old_photo_id;
|
|
is_fallback_ = is_fallback;
|
|
file_reference_ = FileManager::extract_file_reference(input_photo);
|
|
int32 flags = 0;
|
|
if (is_fallback) {
|
|
flags |= telegram_api::photos_updateProfilePhoto::FALLBACK_MASK;
|
|
}
|
|
if (td_->user_manager_->is_user_bot(user_id)) {
|
|
auto r_input_user = td_->user_manager_->get_input_user(user_id);
|
|
if (r_input_user.is_error()) {
|
|
return on_error(r_input_user.move_as_error());
|
|
}
|
|
flags |= telegram_api::photos_updateProfilePhoto::BOT_MASK;
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_updateProfilePhoto(flags, false /*ignored*/, r_input_user.move_as_ok(),
|
|
std::move(input_photo)),
|
|
{{user_id}}));
|
|
} else {
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_updateProfilePhoto(flags, false /*ignored*/, nullptr, std::move(input_photo)),
|
|
{{"me"}}));
|
|
}
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::photos_updateProfilePhoto>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
td_->user_manager_->on_set_profile_photo(user_id_, result_ptr.move_as_ok(), is_fallback_, old_photo_id_,
|
|
std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (!td_->auth_manager_->is_bot() && FileReferenceManager::is_file_reference_error(status)) {
|
|
if (file_id_.is_valid()) {
|
|
VLOG(file_references) << "Receive " << status << " for " << file_id_;
|
|
td_->file_manager_->delete_file_reference(file_id_, file_reference_);
|
|
td_->file_reference_manager_->repair_file_reference(
|
|
file_id_, PromiseCreator::lambda([user_id = user_id_, file_id = file_id_, is_fallback = is_fallback_,
|
|
old_photo_id = old_photo_id_,
|
|
promise = std::move(promise_)](Result<Unit> result) mutable {
|
|
if (result.is_error()) {
|
|
return promise.set_error(Status::Error(400, "Can't find the photo"));
|
|
}
|
|
|
|
send_closure(G()->user_manager(), &UserManager::send_update_profile_photo_query, user_id, file_id,
|
|
old_photo_id, is_fallback, std::move(promise));
|
|
}));
|
|
return;
|
|
} else {
|
|
LOG(ERROR) << "Receive file reference error, but file_id = " << file_id_;
|
|
}
|
|
}
|
|
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class DeleteContactProfilePhotoQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
UserId user_id_;
|
|
|
|
public:
|
|
explicit DeleteContactProfilePhotoQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(UserId user_id, telegram_api::object_ptr<telegram_api::InputUser> &&input_user) {
|
|
CHECK(input_user != nullptr);
|
|
user_id_ = user_id;
|
|
|
|
int32 flags = 0;
|
|
flags |= telegram_api::photos_uploadContactProfilePhoto::SAVE_MASK;
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_uploadContactProfilePhoto(flags, false /*ignored*/, false /*ignored*/,
|
|
std::move(input_user), nullptr, nullptr, 0, nullptr),
|
|
{{user_id}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::photos_uploadContactProfilePhoto>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto ptr = result_ptr.move_as_ok();
|
|
ptr->photo_ = nullptr;
|
|
td_->user_manager_->on_set_profile_photo(user_id_, std::move(ptr), false, 0, std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class DeleteProfilePhotoQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
int64 profile_photo_id_;
|
|
|
|
public:
|
|
explicit DeleteProfilePhotoQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(int64 profile_photo_id) {
|
|
profile_photo_id_ = profile_photo_id;
|
|
vector<telegram_api::object_ptr<telegram_api::InputPhoto>> input_photo_ids;
|
|
input_photo_ids.push_back(telegram_api::make_object<telegram_api::inputPhoto>(profile_photo_id, 0, BufferSlice()));
|
|
send_query(G()->net_query_creator().create(telegram_api::photos_deletePhotos(std::move(input_photo_ids))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::photos_deletePhotos>(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 DeleteProfilePhotoQuery: " << format::as_array(result);
|
|
if (result.size() != 1u) {
|
|
LOG(WARNING) << "Photo can't be deleted";
|
|
return on_error(Status::Error(400, "Photo can't be deleted"));
|
|
}
|
|
|
|
td_->user_manager_->on_delete_profile_photo(profile_photo_id_, std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class UpdateColorQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
bool for_profile_;
|
|
AccentColorId accent_color_id_;
|
|
CustomEmojiId background_custom_emoji_id_;
|
|
|
|
public:
|
|
explicit UpdateColorQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(bool for_profile, AccentColorId accent_color_id, CustomEmojiId background_custom_emoji_id) {
|
|
for_profile_ = for_profile;
|
|
accent_color_id_ = accent_color_id;
|
|
background_custom_emoji_id_ = background_custom_emoji_id;
|
|
int32 flags = 0;
|
|
if (for_profile) {
|
|
flags |= telegram_api::account_updateColor::FOR_PROFILE_MASK;
|
|
}
|
|
if (accent_color_id.is_valid()) {
|
|
flags |= telegram_api::account_updateColor::COLOR_MASK;
|
|
}
|
|
if (background_custom_emoji_id.is_valid()) {
|
|
flags |= telegram_api::account_updateColor::BACKGROUND_EMOJI_ID_MASK;
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::account_updateColor(flags, false /*ignored*/, accent_color_id.get(),
|
|
background_custom_emoji_id.get()),
|
|
{{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_updateColor>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for UpdateColorQuery: " << result_ptr.ok();
|
|
td_->user_manager_->on_update_accent_color_success(for_profile_, accent_color_id_, background_custom_emoji_id_);
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class UpdateProfileQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
int32 flags_;
|
|
string first_name_;
|
|
string last_name_;
|
|
string about_;
|
|
|
|
public:
|
|
explicit UpdateProfileQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(int32 flags, const string &first_name, const string &last_name, const string &about) {
|
|
flags_ = flags;
|
|
first_name_ = first_name;
|
|
last_name_ = last_name;
|
|
about_ = about;
|
|
send_query(G()->net_query_creator().create(telegram_api::account_updateProfile(flags, first_name, last_name, about),
|
|
{{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_updateProfile>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for UpdateProfileQuery: " << to_string(result_ptr.ok());
|
|
td_->user_manager_->on_get_user(result_ptr.move_as_ok(), "UpdateProfileQuery");
|
|
td_->user_manager_->on_update_profile_success(flags_, first_name_, last_name_, about_);
|
|
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class UpdateUsernameQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit UpdateUsernameQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(const string &username) {
|
|
send_query(G()->net_query_creator().create(telegram_api::account_updateUsername(username), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_updateUsername>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for UpdateUsernameQuery: " << to_string(result_ptr.ok());
|
|
td_->user_manager_->on_get_user(result_ptr.move_as_ok(), "UpdateUsernameQuery");
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (status.message() == "USERNAME_NOT_MODIFIED" && !td_->auth_manager_->is_bot()) {
|
|
promise_.set_value(Unit());
|
|
return;
|
|
}
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ToggleUsernameQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
string username_;
|
|
bool is_active_;
|
|
|
|
public:
|
|
explicit ToggleUsernameQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(string &&username, bool is_active) {
|
|
username_ = std::move(username);
|
|
is_active_ = is_active;
|
|
send_query(G()->net_query_creator().create(telegram_api::account_toggleUsername(username_, is_active_), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_toggleUsername>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.ok();
|
|
LOG(DEBUG) << "Receive result for ToggleUsernameQuery: " << result;
|
|
td_->user_manager_->on_update_username_is_active(td_->user_manager_->get_my_id(), std::move(username_), is_active_,
|
|
std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (status.message() == "USERNAME_NOT_MODIFIED") {
|
|
td_->user_manager_->on_update_username_is_active(td_->user_manager_->get_my_id(), std::move(username_),
|
|
is_active_, std::move(promise_));
|
|
return;
|
|
}
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ReorderUsernamesQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
vector<string> usernames_;
|
|
|
|
public:
|
|
explicit ReorderUsernamesQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(vector<string> &&usernames) {
|
|
usernames_ = usernames;
|
|
send_query(G()->net_query_creator().create(telegram_api::account_reorderUsernames(std::move(usernames)), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_reorderUsernames>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.ok();
|
|
LOG(DEBUG) << "Receive result for ReorderUsernamesQuery: " << result;
|
|
if (!result) {
|
|
return on_error(Status::Error(500, "Usernames weren't updated"));
|
|
}
|
|
|
|
td_->user_manager_->on_update_active_usernames_order(td_->user_manager_->get_my_id(), std::move(usernames_),
|
|
std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (status.message() == "USERNAME_NOT_MODIFIED") {
|
|
td_->user_manager_->on_update_active_usernames_order(td_->user_manager_->get_my_id(), std::move(usernames_),
|
|
std::move(promise_));
|
|
return;
|
|
}
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ToggleBotUsernameQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
UserId bot_user_id_;
|
|
string username_;
|
|
bool is_active_;
|
|
|
|
public:
|
|
explicit ToggleBotUsernameQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(UserId bot_user_id, string &&username, bool is_active) {
|
|
bot_user_id_ = bot_user_id;
|
|
username_ = std::move(username);
|
|
is_active_ = is_active;
|
|
auto r_input_user = td_->user_manager_->get_input_user(bot_user_id_);
|
|
if (r_input_user.is_error()) {
|
|
return on_error(r_input_user.move_as_error());
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::bots_toggleUsername(r_input_user.move_as_ok(), username_, is_active_), {{bot_user_id_}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::bots_toggleUsername>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.ok();
|
|
LOG(DEBUG) << "Receive result for ToggleBotUsernameQuery: " << result;
|
|
td_->user_manager_->on_update_username_is_active(bot_user_id_, std::move(username_), is_active_,
|
|
std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (status.message() == "USERNAME_NOT_MODIFIED") {
|
|
td_->user_manager_->on_update_username_is_active(bot_user_id_, std::move(username_), is_active_,
|
|
std::move(promise_));
|
|
return;
|
|
}
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ReorderBotUsernamesQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
UserId bot_user_id_;
|
|
vector<string> usernames_;
|
|
|
|
public:
|
|
explicit ReorderBotUsernamesQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(UserId bot_user_id, vector<string> &&usernames) {
|
|
bot_user_id_ = bot_user_id;
|
|
usernames_ = usernames;
|
|
auto r_input_user = td_->user_manager_->get_input_user(bot_user_id_);
|
|
if (r_input_user.is_error()) {
|
|
return on_error(r_input_user.move_as_error());
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::bots_reorderUsernames(r_input_user.move_as_ok(), std::move(usernames)), {{bot_user_id_}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::bots_reorderUsernames>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
bool result = result_ptr.ok();
|
|
LOG(DEBUG) << "Receive result for ReorderBotUsernamesQuery: " << result;
|
|
if (!result) {
|
|
return on_error(Status::Error(500, "Usernames weren't updated"));
|
|
}
|
|
|
|
td_->user_manager_->on_update_active_usernames_order(bot_user_id_, std::move(usernames_), std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (status.message() == "USERNAME_NOT_MODIFIED") {
|
|
td_->user_manager_->on_update_active_usernames_order(bot_user_id_, std::move(usernames_), std::move(promise_));
|
|
return;
|
|
}
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class UpdateBirthdayQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit UpdateBirthdayQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(const Birthdate &birthdate) {
|
|
int32 flags = 0;
|
|
if (!birthdate.is_empty()) {
|
|
flags |= telegram_api::account_updateBirthday::BIRTHDAY_MASK;
|
|
}
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::account_updateBirthday(flags, birthdate.get_input_birthday()), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_updateBirthday>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for UpdateBirthdayQuery: " << result_ptr.ok();
|
|
if (result_ptr.ok()) {
|
|
promise_.set_value(Unit());
|
|
} else {
|
|
promise_.set_error(Status::Error(400, "Failed to change birthdate"));
|
|
}
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class UpdatePersonalChannelQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
ChannelId channel_id_;
|
|
|
|
public:
|
|
explicit UpdatePersonalChannelQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(ChannelId channel_id) {
|
|
channel_id_ = channel_id;
|
|
telegram_api::object_ptr<telegram_api::InputChannel> input_channel;
|
|
if (channel_id == ChannelId()) {
|
|
input_channel = telegram_api::make_object<telegram_api::inputChannelEmpty>();
|
|
} else {
|
|
input_channel = td_->chat_manager_->get_input_channel(channel_id);
|
|
CHECK(input_channel != nullptr);
|
|
}
|
|
send_query(G()->net_query_creator().create(telegram_api::account_updatePersonalChannel(std::move(input_channel)),
|
|
{{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_updatePersonalChannel>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for UpdatePersonalChannelQuery: " << result_ptr.ok();
|
|
if (result_ptr.ok()) {
|
|
promise_.set_value(Unit());
|
|
} else {
|
|
promise_.set_error(Status::Error(400, "Failed to change personal chat"));
|
|
}
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
if (channel_id_.is_valid()) {
|
|
td_->chat_manager_->on_get_channel_error(channel_id_, status, "UpdatePersonalChannelQuery");
|
|
}
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class UpdateEmojiStatusQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit UpdateEmojiStatusQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(const EmojiStatus &emoji_status) {
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::account_updateEmojiStatus(emoji_status.get_input_emoji_status()), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_updateEmojiStatus>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for UpdateEmojiStatusQuery: " << result_ptr.ok();
|
|
if (result_ptr.ok()) {
|
|
promise_.set_value(Unit());
|
|
} else {
|
|
promise_.set_error(Status::Error(400, "Failed to change Premium badge"));
|
|
}
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
get_recent_emoji_statuses(td_, Auto());
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class ToggleSponsoredMessagesQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit ToggleSponsoredMessagesQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(bool sponsored_enabled) {
|
|
send_query(
|
|
G()->net_query_creator().create(telegram_api::account_toggleSponsoredMessages(sponsored_enabled), {{"me"}}));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::account_toggleSponsoredMessages>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
LOG(DEBUG) << "Receive result for ToggleSponsoredMessagesQuery: " << result_ptr.ok();
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetUsersQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit GetUsersQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(vector<telegram_api::object_ptr<telegram_api::InputUser>> &&input_users) {
|
|
send_query(G()->net_query_creator().create(telegram_api::users_getUsers(std::move(input_users))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::users_getUsers>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
td_->user_manager_->on_get_users(result_ptr.move_as_ok(), "GetUsersQuery");
|
|
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetFullUserQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
|
|
public:
|
|
explicit GetFullUserQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(telegram_api::object_ptr<telegram_api::InputUser> &&input_user) {
|
|
send_query(G()->net_query_creator().create(telegram_api::users_getFullUser(std::move(input_user))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::users_getFullUser>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
auto ptr = result_ptr.move_as_ok();
|
|
LOG(DEBUG) << "Receive result for GetFullUserQuery: " << to_string(ptr);
|
|
td_->user_manager_->on_get_users(std::move(ptr->users_), "GetFullUserQuery");
|
|
td_->chat_manager_->on_get_chats(std::move(ptr->chats_), "GetFullUserQuery");
|
|
td_->user_manager_->on_get_user_full(std::move(ptr->full_user_));
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetUserPhotosQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
UserId user_id_;
|
|
int32 offset_;
|
|
int32 limit_;
|
|
|
|
public:
|
|
explicit GetUserPhotosQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(UserId user_id, telegram_api::object_ptr<telegram_api::InputUser> &&input_user, int32 offset, int32 limit,
|
|
int64 photo_id) {
|
|
user_id_ = user_id;
|
|
offset_ = offset;
|
|
limit_ = limit;
|
|
send_query(G()->net_query_creator().create(
|
|
telegram_api::photos_getUserPhotos(std::move(input_user), offset, photo_id, limit)));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::photos_getUserPhotos>(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 GetUserPhotosQuery: " << to_string(ptr);
|
|
int32 constructor_id = ptr->get_id();
|
|
if (constructor_id == telegram_api::photos_photos::ID) {
|
|
auto photos = move_tl_object_as<telegram_api::photos_photos>(ptr);
|
|
|
|
td_->user_manager_->on_get_users(std::move(photos->users_), "GetUserPhotosQuery");
|
|
auto photos_size = narrow_cast<int32>(photos->photos_.size());
|
|
td_->user_manager_->on_get_user_photos(user_id_, offset_, limit_, photos_size, std::move(photos->photos_));
|
|
} else {
|
|
CHECK(constructor_id == telegram_api::photos_photosSlice::ID);
|
|
auto photos = move_tl_object_as<telegram_api::photos_photosSlice>(ptr);
|
|
|
|
td_->user_manager_->on_get_users(std::move(photos->users_), "GetUserPhotosQuery slice");
|
|
td_->user_manager_->on_get_user_photos(user_id_, offset_, limit_, photos->count_, std::move(photos->photos_));
|
|
}
|
|
|
|
promise_.set_value(Unit());
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetSupportUserQuery final : public Td::ResultHandler {
|
|
Promise<UserId> promise_;
|
|
|
|
public:
|
|
explicit GetSupportUserQuery(Promise<UserId> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send() {
|
|
send_query(G()->net_query_creator().create(telegram_api::help_getSupport()));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::help_getSupport>(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 GetSupportUserQuery: " << to_string(ptr);
|
|
|
|
auto user_id = UserManager::get_user_id(ptr->user_);
|
|
td_->user_manager_->on_get_user(std::move(ptr->user_), "GetSupportUserQuery");
|
|
|
|
promise_.set_value(std::move(user_id));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
class GetIsPremiumRequiredToContactQuery final : public Td::ResultHandler {
|
|
Promise<Unit> promise_;
|
|
vector<UserId> user_ids_;
|
|
|
|
public:
|
|
explicit GetIsPremiumRequiredToContactQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
|
}
|
|
|
|
void send(vector<UserId> &&user_ids, vector<telegram_api::object_ptr<telegram_api::InputUser>> &&input_users) {
|
|
user_ids_ = std::move(user_ids);
|
|
send_query(
|
|
G()->net_query_creator().create(telegram_api::users_getIsPremiumRequiredToContact(std::move(input_users))));
|
|
}
|
|
|
|
void on_result(BufferSlice packet) final {
|
|
auto result_ptr = fetch_result<telegram_api::users_getIsPremiumRequiredToContact>(packet);
|
|
if (result_ptr.is_error()) {
|
|
return on_error(result_ptr.move_as_error());
|
|
}
|
|
|
|
td_->user_manager_->on_get_is_premium_required_to_contact_users(std::move(user_ids_), result_ptr.move_as_ok(),
|
|
std::move(promise_));
|
|
}
|
|
|
|
void on_error(Status status) final {
|
|
promise_.set_error(std::move(status));
|
|
}
|
|
};
|
|
|
|
template <class StorerT>
|
|
void UserManager::User::store(StorerT &storer) const {
|
|
using td::store;
|
|
bool has_last_name = !last_name.empty();
|
|
bool legacy_has_username = false;
|
|
bool has_photo = photo.small_file_id.is_valid();
|
|
bool has_language_code = !language_code.empty();
|
|
bool have_access_hash = access_hash != -1;
|
|
bool has_cache_version = cache_version != 0;
|
|
bool has_is_contact = true;
|
|
bool has_restriction_reasons = !restriction_reasons.empty();
|
|
bool has_emoji_status = !emoji_status.is_empty();
|
|
bool has_usernames = !usernames.is_empty();
|
|
bool has_flags2 = true;
|
|
bool has_max_active_story_id = max_active_story_id.is_valid();
|
|
bool has_max_read_story_id = max_read_story_id.is_valid();
|
|
bool has_max_active_story_id_next_reload_time = max_active_story_id_next_reload_time > Time::now();
|
|
bool has_accent_color_id = accent_color_id.is_valid();
|
|
bool has_background_custom_emoji_id = background_custom_emoji_id.is_valid();
|
|
bool has_profile_accent_color_id = profile_accent_color_id.is_valid();
|
|
bool has_profile_background_custom_emoji_id = profile_background_custom_emoji_id.is_valid();
|
|
bool has_bot_active_users = bot_active_users != 0;
|
|
BEGIN_STORE_FLAGS();
|
|
STORE_FLAG(is_received);
|
|
STORE_FLAG(is_verified);
|
|
STORE_FLAG(is_deleted);
|
|
STORE_FLAG(is_bot);
|
|
STORE_FLAG(can_join_groups);
|
|
STORE_FLAG(can_read_all_group_messages);
|
|
STORE_FLAG(is_inline_bot);
|
|
STORE_FLAG(need_location_bot);
|
|
STORE_FLAG(has_last_name);
|
|
STORE_FLAG(legacy_has_username);
|
|
STORE_FLAG(has_photo);
|
|
STORE_FLAG(false); // legacy is_restricted
|
|
STORE_FLAG(has_language_code);
|
|
STORE_FLAG(have_access_hash);
|
|
STORE_FLAG(is_support);
|
|
STORE_FLAG(is_min_access_hash);
|
|
STORE_FLAG(is_scam);
|
|
STORE_FLAG(has_cache_version);
|
|
STORE_FLAG(has_is_contact);
|
|
STORE_FLAG(is_contact);
|
|
STORE_FLAG(is_mutual_contact);
|
|
STORE_FLAG(has_restriction_reasons);
|
|
STORE_FLAG(need_apply_min_photo);
|
|
STORE_FLAG(is_fake);
|
|
STORE_FLAG(can_be_added_to_attach_menu);
|
|
STORE_FLAG(is_premium);
|
|
STORE_FLAG(attach_menu_enabled);
|
|
STORE_FLAG(has_emoji_status);
|
|
STORE_FLAG(has_usernames);
|
|
STORE_FLAG(can_be_edited_bot);
|
|
END_STORE_FLAGS();
|
|
if (has_flags2) {
|
|
BEGIN_STORE_FLAGS();
|
|
STORE_FLAG(is_close_friend);
|
|
STORE_FLAG(stories_hidden);
|
|
STORE_FLAG(false);
|
|
STORE_FLAG(has_max_active_story_id);
|
|
STORE_FLAG(has_max_read_story_id);
|
|
STORE_FLAG(has_max_active_story_id_next_reload_time);
|
|
STORE_FLAG(has_accent_color_id);
|
|
STORE_FLAG(has_background_custom_emoji_id);
|
|
STORE_FLAG(has_profile_accent_color_id);
|
|
STORE_FLAG(has_profile_background_custom_emoji_id);
|
|
STORE_FLAG(contact_require_premium);
|
|
STORE_FLAG(is_business_bot);
|
|
STORE_FLAG(has_bot_active_users);
|
|
STORE_FLAG(has_main_app);
|
|
END_STORE_FLAGS();
|
|
}
|
|
store(first_name, storer);
|
|
if (has_last_name) {
|
|
store(last_name, storer);
|
|
}
|
|
store(phone_number, storer);
|
|
if (have_access_hash) {
|
|
store(access_hash, storer);
|
|
}
|
|
if (has_photo) {
|
|
store(photo, storer);
|
|
}
|
|
store(was_online, storer);
|
|
if (has_restriction_reasons) {
|
|
store(restriction_reasons, storer);
|
|
}
|
|
if (is_inline_bot) {
|
|
store(inline_query_placeholder, storer);
|
|
}
|
|
if (is_bot) {
|
|
store(bot_info_version, storer);
|
|
}
|
|
if (has_language_code) {
|
|
store(language_code, storer);
|
|
}
|
|
if (has_cache_version) {
|
|
store(cache_version, storer);
|
|
}
|
|
if (has_emoji_status) {
|
|
store(emoji_status, storer);
|
|
}
|
|
if (has_usernames) {
|
|
store(usernames, storer);
|
|
}
|
|
if (has_max_active_story_id) {
|
|
store(max_active_story_id, storer);
|
|
}
|
|
if (has_max_read_story_id) {
|
|
store(max_read_story_id, storer);
|
|
}
|
|
if (has_max_active_story_id_next_reload_time) {
|
|
store_time(max_active_story_id_next_reload_time, storer);
|
|
}
|
|
if (has_accent_color_id) {
|
|
store(accent_color_id, storer);
|
|
}
|
|
if (has_background_custom_emoji_id) {
|
|
store(background_custom_emoji_id, storer);
|
|
}
|
|
if (has_profile_accent_color_id) {
|
|
store(profile_accent_color_id, storer);
|
|
}
|
|
if (has_profile_background_custom_emoji_id) {
|
|
store(profile_background_custom_emoji_id, storer);
|
|
}
|
|
if (has_bot_active_users) {
|
|
store(bot_active_users, storer);
|
|
}
|
|
}
|
|
|
|
template <class ParserT>
|
|
void UserManager::User::parse(ParserT &parser) {
|
|
using td::parse;
|
|
bool has_last_name;
|
|
bool legacy_has_username;
|
|
bool has_photo;
|
|
bool legacy_is_restricted;
|
|
bool has_language_code;
|
|
bool have_access_hash;
|
|
bool has_cache_version;
|
|
bool has_is_contact;
|
|
bool has_restriction_reasons;
|
|
bool has_emoji_status;
|
|
bool has_usernames;
|
|
bool has_flags2 = parser.version() >= static_cast<int32>(Version::AddUserFlags2);
|
|
bool legacy_has_stories = false;
|
|
bool has_max_active_story_id = false;
|
|
bool has_max_read_story_id = false;
|
|
bool has_max_active_story_id_next_reload_time = false;
|
|
bool has_accent_color_id = false;
|
|
bool has_background_custom_emoji_id = false;
|
|
bool has_profile_accent_color_id = false;
|
|
bool has_profile_background_custom_emoji_id = false;
|
|
bool has_bot_active_users = false;
|
|
BEGIN_PARSE_FLAGS();
|
|
PARSE_FLAG(is_received);
|
|
PARSE_FLAG(is_verified);
|
|
PARSE_FLAG(is_deleted);
|
|
PARSE_FLAG(is_bot);
|
|
PARSE_FLAG(can_join_groups);
|
|
PARSE_FLAG(can_read_all_group_messages);
|
|
PARSE_FLAG(is_inline_bot);
|
|
PARSE_FLAG(need_location_bot);
|
|
PARSE_FLAG(has_last_name);
|
|
PARSE_FLAG(legacy_has_username);
|
|
PARSE_FLAG(has_photo);
|
|
PARSE_FLAG(legacy_is_restricted);
|
|
PARSE_FLAG(has_language_code);
|
|
PARSE_FLAG(have_access_hash);
|
|
PARSE_FLAG(is_support);
|
|
PARSE_FLAG(is_min_access_hash);
|
|
PARSE_FLAG(is_scam);
|
|
PARSE_FLAG(has_cache_version);
|
|
PARSE_FLAG(has_is_contact);
|
|
PARSE_FLAG(is_contact);
|
|
PARSE_FLAG(is_mutual_contact);
|
|
PARSE_FLAG(has_restriction_reasons);
|
|
PARSE_FLAG(need_apply_min_photo);
|
|
PARSE_FLAG(is_fake);
|
|
PARSE_FLAG(can_be_added_to_attach_menu);
|
|
PARSE_FLAG(is_premium);
|
|
PARSE_FLAG(attach_menu_enabled);
|
|
PARSE_FLAG(has_emoji_status);
|
|
PARSE_FLAG(has_usernames);
|
|
PARSE_FLAG(can_be_edited_bot);
|
|
END_PARSE_FLAGS();
|
|
if (has_flags2) {
|
|
BEGIN_PARSE_FLAGS();
|
|
PARSE_FLAG(is_close_friend);
|
|
PARSE_FLAG(stories_hidden);
|
|
PARSE_FLAG(legacy_has_stories);
|
|
PARSE_FLAG(has_max_active_story_id);
|
|
PARSE_FLAG(has_max_read_story_id);
|
|
PARSE_FLAG(has_max_active_story_id_next_reload_time);
|
|
PARSE_FLAG(has_accent_color_id);
|
|
PARSE_FLAG(has_background_custom_emoji_id);
|
|
PARSE_FLAG(has_profile_accent_color_id);
|
|
PARSE_FLAG(has_profile_background_custom_emoji_id);
|
|
PARSE_FLAG(contact_require_premium);
|
|
PARSE_FLAG(is_business_bot);
|
|
PARSE_FLAG(has_bot_active_users);
|
|
PARSE_FLAG(has_main_app);
|
|
END_PARSE_FLAGS();
|
|
}
|
|
parse(first_name, parser);
|
|
if (has_last_name) {
|
|
parse(last_name, parser);
|
|
}
|
|
if (legacy_has_username) {
|
|
CHECK(!has_usernames);
|
|
string username;
|
|
parse(username, parser);
|
|
usernames = Usernames(std::move(username), vector<telegram_api::object_ptr<telegram_api::username>>());
|
|
}
|
|
parse(phone_number, parser);
|
|
if (parser.version() < static_cast<int32>(Version::FixMinUsers)) {
|
|
have_access_hash = is_received;
|
|
}
|
|
if (have_access_hash) {
|
|
parse(access_hash, parser);
|
|
} else {
|
|
is_min_access_hash = true;
|
|
}
|
|
if (has_photo) {
|
|
parse(photo, parser);
|
|
}
|
|
if (!has_is_contact) {
|
|
// enum class LinkState : uint8 { Unknown, None, KnowsPhoneNumber, Contact };
|
|
|
|
uint32 link_state_inbound;
|
|
uint32 link_state_outbound;
|
|
parse(link_state_inbound, parser);
|
|
parse(link_state_outbound, parser);
|
|
|
|
is_contact = link_state_outbound == 3;
|
|
is_mutual_contact = is_contact && link_state_inbound == 3;
|
|
is_close_friend = false;
|
|
}
|
|
parse(was_online, parser);
|
|
if (legacy_is_restricted) {
|
|
string restriction_reason;
|
|
parse(restriction_reason, parser);
|
|
restriction_reasons = get_restriction_reasons(restriction_reason);
|
|
} else if (has_restriction_reasons) {
|
|
parse(restriction_reasons, parser);
|
|
}
|
|
if (is_inline_bot) {
|
|
parse(inline_query_placeholder, parser);
|
|
}
|
|
if (is_bot) {
|
|
parse(bot_info_version, parser);
|
|
}
|
|
if (has_language_code) {
|
|
parse(language_code, parser);
|
|
}
|
|
if (has_cache_version) {
|
|
parse(cache_version, parser);
|
|
}
|
|
if (has_emoji_status) {
|
|
parse(emoji_status, parser);
|
|
}
|
|
if (has_usernames) {
|
|
CHECK(!legacy_has_username);
|
|
parse(usernames, parser);
|
|
}
|
|
if (has_max_active_story_id) {
|
|
parse(max_active_story_id, parser);
|
|
}
|
|
if (has_max_read_story_id) {
|
|
parse(max_read_story_id, parser);
|
|
}
|
|
if (has_max_active_story_id_next_reload_time) {
|
|
parse_time(max_active_story_id_next_reload_time, parser);
|
|
}
|
|
if (has_accent_color_id) {
|
|
parse(accent_color_id, parser);
|
|
}
|
|
if (has_background_custom_emoji_id) {
|
|
parse(background_custom_emoji_id, parser);
|
|
}
|
|
if (has_profile_accent_color_id) {
|
|
parse(profile_accent_color_id, parser);
|
|
}
|
|
if (has_profile_background_custom_emoji_id) {
|
|
parse(profile_background_custom_emoji_id, parser);
|
|
}
|
|
if (has_bot_active_users) {
|
|
parse(bot_active_users, parser);
|
|
}
|
|
|
|
if (!check_utf8(first_name)) {
|
|
LOG(ERROR) << "Have invalid first name \"" << first_name << '"';
|
|
first_name.clear();
|
|
cache_version = 0;
|
|
}
|
|
if (!check_utf8(last_name)) {
|
|
LOG(ERROR) << "Have invalid last name \"" << last_name << '"';
|
|
last_name.clear();
|
|
cache_version = 0;
|
|
}
|
|
|
|
clean_phone_number(phone_number);
|
|
if (first_name.empty() && last_name.empty()) {
|
|
first_name = phone_number;
|
|
}
|
|
if (!is_contact && is_mutual_contact) {
|
|
LOG(ERROR) << "Have invalid flag is_mutual_contact";
|
|
is_mutual_contact = false;
|
|
cache_version = 0;
|
|
}
|
|
if (!is_contact && is_close_friend) {
|
|
LOG(ERROR) << "Have invalid flag is_close_friend";
|
|
is_close_friend = false;
|
|
cache_version = 0;
|
|
}
|
|
}
|
|
|
|
template <class StorerT>
|
|
void UserManager::UserFull::store(StorerT &storer) const {
|
|
using td::store;
|
|
bool has_about = !about.empty();
|
|
bool has_photo = !photo.is_empty();
|
|
bool has_description = !description.empty();
|
|
bool has_commands = !commands.empty();
|
|
bool has_private_forward_name = !private_forward_name.empty();
|
|
bool has_group_administrator_rights = group_administrator_rights != AdministratorRights();
|
|
bool has_broadcast_administrator_rights = broadcast_administrator_rights != AdministratorRights();
|
|
bool has_menu_button = menu_button != nullptr;
|
|
bool has_description_photo = !description_photo.is_empty();
|
|
bool has_description_animation = description_animation_file_id.is_valid();
|
|
bool has_premium_gift_options = !premium_gift_options.empty();
|
|
bool has_personal_photo = !personal_photo.is_empty();
|
|
bool has_fallback_photo = !fallback_photo.is_empty();
|
|
bool has_business_info = business_info != nullptr && !business_info->is_empty();
|
|
bool has_birthdate = !birthdate.is_empty();
|
|
bool has_personal_channel_id = personal_channel_id.is_valid();
|
|
bool has_flags2 = true;
|
|
BEGIN_STORE_FLAGS();
|
|
STORE_FLAG(has_about);
|
|
STORE_FLAG(is_blocked);
|
|
STORE_FLAG(can_be_called);
|
|
STORE_FLAG(has_private_calls);
|
|
STORE_FLAG(can_pin_messages);
|
|
STORE_FLAG(need_phone_number_privacy_exception);
|
|
STORE_FLAG(has_photo);
|
|
STORE_FLAG(supports_video_calls);
|
|
STORE_FLAG(has_description);
|
|
STORE_FLAG(has_commands);
|
|
STORE_FLAG(has_private_forward_name);
|
|
STORE_FLAG(has_group_administrator_rights);
|
|
STORE_FLAG(has_broadcast_administrator_rights);
|
|
STORE_FLAG(has_menu_button);
|
|
STORE_FLAG(has_description_photo);
|
|
STORE_FLAG(has_description_animation);
|
|
STORE_FLAG(has_premium_gift_options);
|
|
STORE_FLAG(voice_messages_forbidden);
|
|
STORE_FLAG(has_personal_photo);
|
|
STORE_FLAG(has_fallback_photo);
|
|
STORE_FLAG(has_pinned_stories);
|
|
STORE_FLAG(is_blocked_for_stories);
|
|
STORE_FLAG(wallpaper_overridden);
|
|
STORE_FLAG(read_dates_private);
|
|
STORE_FLAG(contact_require_premium);
|
|
STORE_FLAG(has_business_info);
|
|
STORE_FLAG(has_birthdate);
|
|
STORE_FLAG(has_personal_channel_id);
|
|
STORE_FLAG(sponsored_enabled);
|
|
STORE_FLAG(has_flags2);
|
|
END_STORE_FLAGS();
|
|
if (has_flags2) {
|
|
BEGIN_STORE_FLAGS();
|
|
STORE_FLAG(has_preview_medias);
|
|
END_STORE_FLAGS();
|
|
}
|
|
if (has_about) {
|
|
store(about, storer);
|
|
}
|
|
store(common_chat_count, storer);
|
|
store_time(expires_at, storer);
|
|
if (has_photo) {
|
|
store(photo, storer);
|
|
}
|
|
if (has_description) {
|
|
store(description, storer);
|
|
}
|
|
if (has_commands) {
|
|
store(commands, storer);
|
|
}
|
|
if (has_private_forward_name) {
|
|
store(private_forward_name, storer);
|
|
}
|
|
if (has_group_administrator_rights) {
|
|
store(group_administrator_rights, storer);
|
|
}
|
|
if (has_broadcast_administrator_rights) {
|
|
store(broadcast_administrator_rights, storer);
|
|
}
|
|
if (has_menu_button) {
|
|
store(menu_button, storer);
|
|
}
|
|
if (has_description_photo) {
|
|
store(description_photo, storer);
|
|
}
|
|
if (has_description_animation) {
|
|
storer.context()->td().get_actor_unsafe()->animations_manager_->store_animation(description_animation_file_id,
|
|
storer);
|
|
}
|
|
if (has_premium_gift_options) {
|
|
store(premium_gift_options, storer);
|
|
}
|
|
if (has_personal_photo) {
|
|
store(personal_photo, storer);
|
|
}
|
|
if (has_fallback_photo) {
|
|
store(fallback_photo, storer);
|
|
}
|
|
if (has_business_info) {
|
|
store(business_info, storer);
|
|
}
|
|
if (has_birthdate) {
|
|
store(birthdate, storer);
|
|
}
|
|
if (has_personal_channel_id) {
|
|
store(personal_channel_id, storer);
|
|
}
|
|
}
|
|
|
|
template <class ParserT>
|
|
void UserManager::UserFull::parse(ParserT &parser) {
|
|
using td::parse;
|
|
bool has_about;
|
|
bool has_photo;
|
|
bool has_description;
|
|
bool has_commands;
|
|
bool has_private_forward_name;
|
|
bool has_group_administrator_rights;
|
|
bool has_broadcast_administrator_rights;
|
|
bool has_menu_button;
|
|
bool has_description_photo;
|
|
bool has_description_animation;
|
|
bool has_premium_gift_options;
|
|
bool has_personal_photo;
|
|
bool has_fallback_photo;
|
|
bool has_business_info;
|
|
bool has_birthdate;
|
|
bool has_personal_channel_id;
|
|
bool has_flags2;
|
|
BEGIN_PARSE_FLAGS();
|
|
PARSE_FLAG(has_about);
|
|
PARSE_FLAG(is_blocked);
|
|
PARSE_FLAG(can_be_called);
|
|
PARSE_FLAG(has_private_calls);
|
|
PARSE_FLAG(can_pin_messages);
|
|
PARSE_FLAG(need_phone_number_privacy_exception);
|
|
PARSE_FLAG(has_photo);
|
|
PARSE_FLAG(supports_video_calls);
|
|
PARSE_FLAG(has_description);
|
|
PARSE_FLAG(has_commands);
|
|
PARSE_FLAG(has_private_forward_name);
|
|
PARSE_FLAG(has_group_administrator_rights);
|
|
PARSE_FLAG(has_broadcast_administrator_rights);
|
|
PARSE_FLAG(has_menu_button);
|
|
PARSE_FLAG(has_description_photo);
|
|
PARSE_FLAG(has_description_animation);
|
|
PARSE_FLAG(has_premium_gift_options);
|
|
PARSE_FLAG(voice_messages_forbidden);
|
|
PARSE_FLAG(has_personal_photo);
|
|
PARSE_FLAG(has_fallback_photo);
|
|
PARSE_FLAG(has_pinned_stories);
|
|
PARSE_FLAG(is_blocked_for_stories);
|
|
PARSE_FLAG(wallpaper_overridden);
|
|
PARSE_FLAG(read_dates_private);
|
|
PARSE_FLAG(contact_require_premium);
|
|
PARSE_FLAG(has_business_info);
|
|
PARSE_FLAG(has_birthdate);
|
|
PARSE_FLAG(has_personal_channel_id);
|
|
PARSE_FLAG(sponsored_enabled);
|
|
PARSE_FLAG(has_flags2);
|
|
END_PARSE_FLAGS();
|
|
if (has_flags2) {
|
|
BEGIN_PARSE_FLAGS();
|
|
PARSE_FLAG(has_preview_medias);
|
|
END_PARSE_FLAGS();
|
|
}
|
|
if (has_about) {
|
|
parse(about, parser);
|
|
}
|
|
parse(common_chat_count, parser);
|
|
parse_time(expires_at, parser);
|
|
if (has_photo) {
|
|
parse(photo, parser);
|
|
}
|
|
if (has_description) {
|
|
parse(description, parser);
|
|
}
|
|
if (has_commands) {
|
|
parse(commands, parser);
|
|
}
|
|
if (has_private_forward_name) {
|
|
parse(private_forward_name, parser);
|
|
}
|
|
if (has_group_administrator_rights) {
|
|
parse(group_administrator_rights, parser);
|
|
}
|
|
if (has_broadcast_administrator_rights) {
|
|
parse(broadcast_administrator_rights, parser);
|
|
}
|
|
if (has_menu_button) {
|
|
parse(menu_button, parser);
|
|
}
|
|
if (has_description_photo) {
|
|
parse(description_photo, parser);
|
|
}
|
|
if (has_description_animation) {
|
|
description_animation_file_id =
|
|
parser.context()->td().get_actor_unsafe()->animations_manager_->parse_animation(parser);
|
|
}
|
|
if (has_premium_gift_options) {
|
|
parse(premium_gift_options, parser);
|
|
}
|
|
if (has_personal_photo) {
|
|
parse(personal_photo, parser);
|
|
}
|
|
if (has_fallback_photo) {
|
|
parse(fallback_photo, parser);
|
|
}
|
|
if (has_business_info) {
|
|
parse(business_info, parser);
|
|
}
|
|
if (has_birthdate) {
|
|
parse(birthdate, parser);
|
|
}
|
|
if (has_personal_channel_id) {
|
|
parse(personal_channel_id, parser);
|
|
}
|
|
}
|
|
|
|
template <class StorerT>
|
|
void UserManager::SecretChat::store(StorerT &storer) const {
|
|
using td::store;
|
|
bool has_layer = layer > static_cast<int32>(SecretChatLayer::Default);
|
|
bool has_initial_folder_id = initial_folder_id != FolderId();
|
|
BEGIN_STORE_FLAGS();
|
|
STORE_FLAG(is_outbound);
|
|
STORE_FLAG(has_layer);
|
|
STORE_FLAG(has_initial_folder_id);
|
|
END_STORE_FLAGS();
|
|
|
|
store(access_hash, storer);
|
|
store(user_id, storer);
|
|
store(state, storer);
|
|
store(ttl, storer);
|
|
store(date, storer);
|
|
store(key_hash, storer);
|
|
if (has_layer) {
|
|
store(layer, storer);
|
|
}
|
|
if (has_initial_folder_id) {
|
|
store(initial_folder_id, storer);
|
|
}
|
|
}
|
|
|
|
template <class ParserT>
|
|
void UserManager::SecretChat::parse(ParserT &parser) {
|
|
using td::parse;
|
|
bool has_layer;
|
|
bool has_initial_folder_id;
|
|
BEGIN_PARSE_FLAGS();
|
|
PARSE_FLAG(is_outbound);
|
|
PARSE_FLAG(has_layer);
|
|
PARSE_FLAG(has_initial_folder_id);
|
|
END_PARSE_FLAGS();
|
|
|
|
if (parser.version() >= static_cast<int32>(Version::AddAccessHashToSecretChat)) {
|
|
parse(access_hash, parser);
|
|
}
|
|
parse(user_id, parser);
|
|
parse(state, parser);
|
|
parse(ttl, parser);
|
|
parse(date, parser);
|
|
if (parser.version() >= static_cast<int32>(Version::AddKeyHashToSecretChat)) {
|
|
parse(key_hash, parser);
|
|
}
|
|
if (has_layer) {
|
|
parse(layer, parser);
|
|
} else {
|
|
layer = static_cast<int32>(SecretChatLayer::Default);
|
|
}
|
|
if (has_initial_folder_id) {
|
|
parse(initial_folder_id, parser);
|
|
}
|
|
}
|
|
|
|
class UserManager::UserLogEvent {
|
|
public:
|
|
UserId user_id;
|
|
const User *u_in = nullptr;
|
|
unique_ptr<User> u_out;
|
|
|
|
UserLogEvent() = default;
|
|
|
|
UserLogEvent(UserId user_id, const User *u) : user_id(user_id), u_in(u) {
|
|
}
|
|
|
|
template <class StorerT>
|
|
void store(StorerT &storer) const {
|
|
td::store(user_id, storer);
|
|
td::store(*u_in, storer);
|
|
}
|
|
|
|
template <class ParserT>
|
|
void parse(ParserT &parser) {
|
|
td::parse(user_id, parser);
|
|
td::parse(u_out, parser);
|
|
}
|
|
};
|
|
|
|
class UserManager::SecretChatLogEvent {
|
|
public:
|
|
SecretChatId secret_chat_id;
|
|
const SecretChat *c_in = nullptr;
|
|
unique_ptr<SecretChat> c_out;
|
|
|
|
SecretChatLogEvent() = default;
|
|
|
|
SecretChatLogEvent(SecretChatId secret_chat_id, const SecretChat *c) : secret_chat_id(secret_chat_id), c_in(c) {
|
|
}
|
|
|
|
template <class StorerT>
|
|
void store(StorerT &storer) const {
|
|
td::store(secret_chat_id, storer);
|
|
td::store(*c_in, storer);
|
|
}
|
|
|
|
template <class ParserT>
|
|
void parse(ParserT &parser) {
|
|
td::parse(secret_chat_id, parser);
|
|
td::parse(c_out, parser);
|
|
}
|
|
};
|
|
|
|
class UserManager::UploadProfilePhotoCallback final : public FileManager::UploadCallback {
|
|
public:
|
|
void on_upload_ok(FileId file_id, telegram_api::object_ptr<telegram_api::InputFile> input_file) final {
|
|
send_closure_later(G()->user_manager(), &UserManager::on_upload_profile_photo, file_id, std::move(input_file));
|
|
}
|
|
void on_upload_encrypted_ok(FileId file_id,
|
|
telegram_api::object_ptr<telegram_api::InputEncryptedFile> input_file) final {
|
|
UNREACHABLE();
|
|
}
|
|
void on_upload_secure_ok(FileId file_id, telegram_api::object_ptr<telegram_api::InputSecureFile> input_file) final {
|
|
UNREACHABLE();
|
|
}
|
|
void on_upload_error(FileId file_id, Status error) final {
|
|
send_closure_later(G()->user_manager(), &UserManager::on_upload_profile_photo_error, file_id, std::move(error));
|
|
}
|
|
};
|
|
|
|
UserManager::UserManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
|
|
upload_profile_photo_callback_ = std::make_shared<UploadProfilePhotoCallback>();
|
|
|
|
my_id_ = load_my_id();
|
|
|
|
if (G()->use_chat_info_database()) {
|
|
auto next_contacts_sync_date_string = G()->td_db()->get_binlog_pmc()->get("next_contacts_sync_date");
|
|
if (!next_contacts_sync_date_string.empty()) {
|
|
next_contacts_sync_date_ = min(to_integer<int32>(next_contacts_sync_date_string), G()->unix_time() + 100000);
|
|
}
|
|
|
|
auto saved_contact_count_string = G()->td_db()->get_binlog_pmc()->get("saved_contact_count");
|
|
if (!saved_contact_count_string.empty()) {
|
|
saved_contact_count_ = to_integer<int32>(saved_contact_count_string);
|
|
}
|
|
} else if (!td_->auth_manager_->is_bot()) {
|
|
G()->td_db()->get_binlog_pmc()->erase("next_contacts_sync_date");
|
|
G()->td_db()->get_binlog_pmc()->erase("saved_contact_count");
|
|
}
|
|
if (G()->use_sqlite_pmc()) {
|
|
G()->td_db()->get_sqlite_pmc()->erase_by_prefix("us_bot_info", Auto());
|
|
}
|
|
|
|
if (!td_->auth_manager_->is_bot()) {
|
|
was_online_local_ = to_integer<int32>(G()->td_db()->get_binlog_pmc()->get("my_was_online_local"));
|
|
was_online_remote_ = to_integer<int32>(G()->td_db()->get_binlog_pmc()->get("my_was_online_remote"));
|
|
auto unix_time = G()->unix_time();
|
|
if (was_online_local_ >= unix_time && !td_->online_manager_->is_online()) {
|
|
was_online_local_ = unix_time - 1;
|
|
}
|
|
}
|
|
|
|
user_online_timeout_.set_callback(on_user_online_timeout_callback);
|
|
user_online_timeout_.set_callback_data(static_cast<void *>(this));
|
|
|
|
user_emoji_status_timeout_.set_callback(on_user_emoji_status_timeout_callback);
|
|
user_emoji_status_timeout_.set_callback_data(static_cast<void *>(this));
|
|
|
|
get_user_queries_.set_merge_function([this](vector<int64> query_ids, Promise<Unit> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
auto input_users = transform(query_ids, [this](int64 query_id) { return get_input_user_force(UserId(query_id)); });
|
|
td_->create_handler<GetUsersQuery>(std::move(promise))->send(std::move(input_users));
|
|
});
|
|
get_is_premium_required_to_contact_queries_.set_merge_function(
|
|
[this](vector<int64> query_ids, Promise<Unit> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
auto user_ids = UserId::get_user_ids(query_ids);
|
|
auto input_users = transform(user_ids, [this](UserId user_id) { return get_input_user_force(user_id); });
|
|
td_->create_handler<GetIsPremiumRequiredToContactQuery>(std::move(promise))
|
|
->send(std::move(user_ids), std::move(input_users));
|
|
});
|
|
}
|
|
|
|
UserManager::~UserManager() {
|
|
Scheduler::instance()->destroy_on_scheduler(G()->get_gc_scheduler_id(), users_, users_full_, user_photos_,
|
|
unknown_users_, pending_user_photos_, user_profile_photo_file_source_ids_,
|
|
my_photo_file_id_, user_full_file_source_ids_, secret_chats_,
|
|
unknown_secret_chats_, secret_chats_with_user_);
|
|
Scheduler::instance()->destroy_on_scheduler(G()->get_gc_scheduler_id(), loaded_from_database_users_,
|
|
unavailable_user_fulls_, loaded_from_database_secret_chats_,
|
|
resolved_phone_numbers_, all_imported_contacts_, restricted_user_ids_);
|
|
}
|
|
|
|
void UserManager::tear_down() {
|
|
parent_.reset();
|
|
|
|
LOG(DEBUG) << "Have " << users_.calc_size() << " users and " << secret_chats_.calc_size() << " secret chats to free";
|
|
LOG(DEBUG) << "Have " << users_full_.calc_size() << " full users to free";
|
|
}
|
|
|
|
void UserManager::on_user_online_timeout_callback(void *user_manager_ptr, int64 user_id_long) {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
|
|
auto user_manager = static_cast<UserManager *>(user_manager_ptr);
|
|
send_closure_later(user_manager->actor_id(user_manager), &UserManager::on_user_online_timeout, UserId(user_id_long));
|
|
}
|
|
|
|
void UserManager::on_user_online_timeout(UserId user_id) {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
|
|
auto u = get_user(user_id);
|
|
CHECK(u != nullptr);
|
|
CHECK(u->is_update_user_sent);
|
|
|
|
LOG(INFO) << "Update " << user_id << " online status to offline";
|
|
send_closure(G()->td(), &Td::send_update,
|
|
td_api::make_object<td_api::updateUserStatus>(user_id.get(),
|
|
get_user_status_object(user_id, u, G()->unix_time())));
|
|
|
|
td_->dialog_participant_manager_->update_user_online_member_count(user_id);
|
|
}
|
|
|
|
void UserManager::on_user_emoji_status_timeout_callback(void *user_manager_ptr, int64 user_id_long) {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
|
|
auto user_manager = static_cast<UserManager *>(user_manager_ptr);
|
|
send_closure_later(user_manager->actor_id(user_manager), &UserManager::on_user_emoji_status_timeout,
|
|
UserId(user_id_long));
|
|
}
|
|
|
|
void UserManager::on_user_emoji_status_timeout(UserId user_id) {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
|
|
auto u = get_user(user_id);
|
|
CHECK(u != nullptr);
|
|
CHECK(u->is_update_user_sent);
|
|
|
|
update_user(u, user_id);
|
|
}
|
|
|
|
UserId UserManager::get_user_id(const telegram_api::object_ptr<telegram_api::User> &user) {
|
|
CHECK(user != nullptr);
|
|
switch (user->get_id()) {
|
|
case telegram_api::userEmpty::ID:
|
|
return UserId(static_cast<const telegram_api::userEmpty *>(user.get())->id_);
|
|
case telegram_api::user::ID:
|
|
return UserId(static_cast<const telegram_api::user *>(user.get())->id_);
|
|
default:
|
|
UNREACHABLE();
|
|
return UserId();
|
|
}
|
|
}
|
|
|
|
UserId UserManager::load_my_id() {
|
|
auto id_string = G()->td_db()->get_binlog_pmc()->get("my_id");
|
|
if (!id_string.empty()) {
|
|
UserId my_id(to_integer<int64>(id_string));
|
|
if (my_id.is_valid()) {
|
|
return my_id;
|
|
}
|
|
|
|
my_id = UserId(to_integer<int64>(Slice(id_string).substr(5)));
|
|
if (my_id.is_valid()) {
|
|
G()->td_db()->get_binlog_pmc()->set("my_id", to_string(my_id.get()));
|
|
return my_id;
|
|
}
|
|
|
|
LOG(ERROR) << "Wrong my ID = \"" << id_string << "\" stored in database";
|
|
}
|
|
return UserId();
|
|
}
|
|
|
|
UserId UserManager::get_my_id() const {
|
|
LOG_IF(ERROR, !my_id_.is_valid()) << "Wrong or unknown my ID returned";
|
|
return my_id_;
|
|
}
|
|
|
|
void UserManager::set_my_id(UserId my_id) {
|
|
UserId my_old_id = my_id_;
|
|
if (my_old_id.is_valid() && my_old_id != my_id) {
|
|
LOG(ERROR) << "Already know that me is " << my_old_id << " but received userSelf with " << my_id;
|
|
return;
|
|
}
|
|
if (!my_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid my ID " << my_id;
|
|
return;
|
|
}
|
|
if (my_old_id != my_id) {
|
|
my_id_ = my_id;
|
|
G()->td_db()->get_binlog_pmc()->set("my_id", to_string(my_id.get()));
|
|
td_->option_manager_->set_option_integer("my_id", my_id_.get());
|
|
if (!td_->auth_manager_->is_bot()) {
|
|
G()->td_db()->get_binlog_pmc()->force_sync(Promise<Unit>(), "set_my_id");
|
|
}
|
|
}
|
|
}
|
|
|
|
UserId UserManager::get_service_notifications_user_id() {
|
|
return UserId(static_cast<int64>(777000));
|
|
}
|
|
|
|
UserId UserManager::add_service_notifications_user() {
|
|
auto user_id = get_service_notifications_user_id();
|
|
if (!have_user_force(user_id, "add_service_notifications_user")) {
|
|
LOG(FATAL) << "Failed to load service notification user";
|
|
}
|
|
return user_id;
|
|
}
|
|
|
|
UserId UserManager::get_replies_bot_user_id() {
|
|
return UserId(static_cast<int64>(G()->is_test_dc() ? 708513 : 1271266957));
|
|
}
|
|
|
|
UserId UserManager::get_anonymous_bot_user_id() {
|
|
return UserId(static_cast<int64>(G()->is_test_dc() ? 552888 : 1087968824));
|
|
}
|
|
|
|
UserId UserManager::get_channel_bot_user_id() {
|
|
return UserId(static_cast<int64>(G()->is_test_dc() ? 936174 : 136817688));
|
|
}
|
|
|
|
UserId UserManager::get_anti_spam_bot_user_id() {
|
|
return UserId(static_cast<int64>(G()->is_test_dc() ? 2200583762ll : 5434988373ll));
|
|
}
|
|
|
|
UserId UserManager::add_anonymous_bot_user() {
|
|
auto user_id = get_anonymous_bot_user_id();
|
|
if (!have_user_force(user_id, "add_anonymous_bot_user")) {
|
|
LOG(FATAL) << "Failed to load anonymous bot user";
|
|
}
|
|
return user_id;
|
|
}
|
|
|
|
UserId UserManager::add_channel_bot_user() {
|
|
auto user_id = get_channel_bot_user_id();
|
|
if (!have_user_force(user_id, "add_channel_bot_user")) {
|
|
LOG(FATAL) << "Failed to load channel bot user";
|
|
}
|
|
return user_id;
|
|
}
|
|
|
|
UserManager::MyOnlineStatusInfo UserManager::get_my_online_status() const {
|
|
MyOnlineStatusInfo status_info;
|
|
status_info.is_online_local = td_->online_manager_->is_online();
|
|
status_info.is_online_remote = was_online_remote_ > G()->unix_time();
|
|
status_info.was_online_local = was_online_local_;
|
|
status_info.was_online_remote = was_online_remote_;
|
|
|
|
return status_info;
|
|
}
|
|
|
|
void UserManager::set_my_online_status(bool is_online, bool send_update, bool is_local) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return; // just in case
|
|
}
|
|
|
|
auto my_id = get_my_id();
|
|
User *u = get_user_force(my_id, "set_my_online_status");
|
|
if (u != nullptr) {
|
|
int32 new_online;
|
|
int32 unix_time = G()->unix_time();
|
|
if (is_online) {
|
|
new_online = unix_time + 300;
|
|
} else {
|
|
new_online = unix_time - 1;
|
|
}
|
|
|
|
auto old_was_online = get_user_was_online(u, my_id, unix_time);
|
|
if (is_local) {
|
|
LOG(INFO) << "Update my local online from " << my_was_online_local_ << " to " << new_online;
|
|
if (!is_online) {
|
|
new_online = min(new_online, u->was_online);
|
|
}
|
|
if (new_online != my_was_online_local_) {
|
|
my_was_online_local_ = new_online;
|
|
}
|
|
} else {
|
|
if (my_was_online_local_ != 0 || new_online != u->was_online) {
|
|
LOG(INFO) << "Update my online from " << u->was_online << " to " << new_online;
|
|
my_was_online_local_ = 0;
|
|
u->was_online = new_online;
|
|
u->need_save_to_database = true;
|
|
}
|
|
}
|
|
if (old_was_online != get_user_was_online(u, my_id, unix_time)) {
|
|
u->is_status_changed = true;
|
|
u->is_online_status_changed = true;
|
|
}
|
|
|
|
if (was_online_local_ != new_online) {
|
|
was_online_local_ = new_online;
|
|
VLOG(notifications) << "Set was_online_local to " << was_online_local_;
|
|
G()->td_db()->get_binlog_pmc()->set("my_was_online_local", to_string(was_online_local_));
|
|
}
|
|
|
|
if (send_update) {
|
|
update_user(u, my_id);
|
|
}
|
|
}
|
|
}
|
|
|
|
void UserManager::on_get_user(telegram_api::object_ptr<telegram_api::User> &&user_ptr, const char *source) {
|
|
LOG(DEBUG) << "Receive from " << source << ' ' << to_string(user_ptr);
|
|
int32 constructor_id = user_ptr->get_id();
|
|
if (constructor_id == telegram_api::userEmpty::ID) {
|
|
auto user = move_tl_object_as<telegram_api::userEmpty>(user_ptr);
|
|
UserId user_id(user->id_);
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id << " from " << source;
|
|
return;
|
|
}
|
|
LOG(INFO) << "Receive empty " << user_id << " from " << source;
|
|
|
|
User *u = get_user_force(user_id, source);
|
|
if (u == nullptr && Slice(source) != Slice("GetUsersQuery")) {
|
|
// userEmpty should be received only through getUsers for nonexistent users
|
|
LOG(ERROR) << "Have no information about " << user_id << ", but received userEmpty from " << source;
|
|
}
|
|
return;
|
|
}
|
|
|
|
CHECK(constructor_id == telegram_api::user::ID);
|
|
auto user = move_tl_object_as<telegram_api::user>(user_ptr);
|
|
UserId user_id(user->id_);
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
int32 flags = user->flags_;
|
|
int32 flags2 = user->flags2_;
|
|
LOG(INFO) << "Receive " << user_id << " with flags " << flags << ' ' << flags2 << " from " << source;
|
|
|
|
// the True fields aren't set for manually created telegram_api::user objects, therefore the flags must be used
|
|
bool is_bot = (flags & USER_FLAG_IS_BOT) != 0;
|
|
if (flags & USER_FLAG_IS_ME) {
|
|
set_my_id(user_id);
|
|
if (!is_bot) {
|
|
td_->option_manager_->set_option_string("my_phone_number", user->phone_);
|
|
}
|
|
}
|
|
|
|
bool have_access_hash = (flags & USER_FLAG_HAS_ACCESS_HASH) != 0;
|
|
bool is_received = (flags & USER_FLAG_IS_INACCESSIBLE) == 0;
|
|
bool is_contact = (flags & USER_FLAG_IS_CONTACT) != 0;
|
|
|
|
User *u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
if (!is_received) {
|
|
// we must preload received inaccessible users from database in order to not save
|
|
// the min-user to the database and to not override access_hash and other data
|
|
u = get_user_force(user_id, "on_get_user 2");
|
|
if (u == nullptr) {
|
|
LOG(INFO) << "Receive inaccessible " << user_id;
|
|
u = add_user(user_id);
|
|
}
|
|
} else if (is_contact && !are_contacts_loaded_) {
|
|
// preload contact users from database to know that is_contact didn't changed
|
|
// and the list of contacts doesn't need to be saved to the database
|
|
u = get_user_force(user_id, "on_get_user 3");
|
|
if (u == nullptr) {
|
|
LOG(INFO) << "Receive contact " << user_id << " for the first time";
|
|
u = add_user(user_id);
|
|
}
|
|
} else {
|
|
u = add_user(user_id);
|
|
}
|
|
CHECK(u != nullptr);
|
|
}
|
|
|
|
if (have_access_hash) { // access_hash must be updated before photo
|
|
auto access_hash = user->access_hash_;
|
|
bool is_min_access_hash = !is_received && !((flags & USER_FLAG_HAS_PHONE_NUMBER) != 0 && user->phone_.empty());
|
|
if (u->access_hash != access_hash && (!is_min_access_hash || u->is_min_access_hash || u->access_hash == -1)) {
|
|
LOG(DEBUG) << "Access hash has changed for " << user_id << " from " << u->access_hash << "/"
|
|
<< u->is_min_access_hash << " to " << access_hash << "/" << is_min_access_hash;
|
|
u->access_hash = access_hash;
|
|
u->is_min_access_hash = is_min_access_hash;
|
|
u->need_save_to_database = true;
|
|
}
|
|
}
|
|
|
|
bool is_verified = (flags & USER_FLAG_IS_VERIFIED) != 0;
|
|
bool is_premium = (flags & USER_FLAG_IS_PREMIUM) != 0;
|
|
bool is_support = (flags & USER_FLAG_IS_SUPPORT) != 0;
|
|
bool is_deleted = (flags & USER_FLAG_IS_DELETED) != 0;
|
|
bool can_join_groups = (flags & USER_FLAG_IS_PRIVATE_BOT) == 0;
|
|
bool can_read_all_group_messages = (flags & USER_FLAG_IS_BOT_WITH_PRIVACY_DISABLED) != 0;
|
|
bool can_be_added_to_attach_menu = (flags & USER_FLAG_IS_ATTACH_MENU_BOT) != 0;
|
|
bool has_main_app = user->bot_has_main_app_;
|
|
bool attach_menu_enabled = (flags & USER_FLAG_ATTACH_MENU_ENABLED) != 0;
|
|
bool is_scam = (flags & USER_FLAG_IS_SCAM) != 0;
|
|
bool can_be_edited_bot = (flags2 & USER_FLAG_CAN_BE_EDITED_BOT) != 0;
|
|
bool is_inline_bot = (flags & USER_FLAG_IS_INLINE_BOT) != 0;
|
|
bool is_business_bot = user->bot_business_;
|
|
string inline_query_placeholder = std::move(user->bot_inline_placeholder_);
|
|
int32 bot_active_users = user->bot_active_users_;
|
|
bool need_location_bot = (flags & USER_FLAG_NEED_LOCATION_BOT) != 0;
|
|
bool has_bot_info_version = (flags & USER_FLAG_HAS_BOT_INFO_VERSION) != 0;
|
|
bool need_apply_min_photo = (flags & USER_FLAG_NEED_APPLY_MIN_PHOTO) != 0;
|
|
bool is_fake = (flags & USER_FLAG_IS_FAKE) != 0;
|
|
bool stories_available = user->stories_max_id_ > 0;
|
|
bool stories_unavailable = user->stories_unavailable_;
|
|
bool stories_hidden = user->stories_hidden_;
|
|
bool contact_require_premium = user->contact_require_premium_;
|
|
|
|
if (!is_bot && (!can_join_groups || can_read_all_group_messages || can_be_added_to_attach_menu || can_be_edited_bot ||
|
|
has_main_app || is_inline_bot || is_business_bot)) {
|
|
LOG(ERROR) << "Receive not bot " << user_id << " with bot properties from " << source;
|
|
can_join_groups = true;
|
|
can_read_all_group_messages = false;
|
|
can_be_added_to_attach_menu = false;
|
|
can_be_edited_bot = false;
|
|
has_main_app = false;
|
|
is_inline_bot = false;
|
|
is_business_bot = false;
|
|
}
|
|
if (need_location_bot && !is_inline_bot) {
|
|
LOG(ERROR) << "Receive not inline bot " << user_id << " which needs user location from " << source;
|
|
need_location_bot = false;
|
|
}
|
|
|
|
if (is_deleted) {
|
|
// just in case
|
|
is_verified = false;
|
|
is_premium = false;
|
|
is_support = false;
|
|
is_bot = false;
|
|
can_join_groups = false;
|
|
can_read_all_group_messages = false;
|
|
can_be_added_to_attach_menu = false;
|
|
can_be_edited_bot = false;
|
|
has_main_app = false;
|
|
is_inline_bot = false;
|
|
is_business_bot = false;
|
|
inline_query_placeholder = string();
|
|
bot_active_users = 0;
|
|
need_location_bot = false;
|
|
has_bot_info_version = false;
|
|
need_apply_min_photo = false;
|
|
}
|
|
|
|
LOG_IF(ERROR, has_bot_info_version && !is_bot)
|
|
<< "Receive not bot " << user_id << " which has bot info version from " << source;
|
|
|
|
int32 bot_info_version = has_bot_info_version ? user->bot_info_version_ : -1;
|
|
if (is_verified != u->is_verified || is_support != u->is_support || is_bot != u->is_bot ||
|
|
can_join_groups != u->can_join_groups || can_read_all_group_messages != u->can_read_all_group_messages ||
|
|
is_scam != u->is_scam || is_fake != u->is_fake || is_inline_bot != u->is_inline_bot ||
|
|
is_business_bot != u->is_business_bot || inline_query_placeholder != u->inline_query_placeholder ||
|
|
need_location_bot != u->need_location_bot || can_be_added_to_attach_menu != u->can_be_added_to_attach_menu ||
|
|
bot_active_users != u->bot_active_users || has_main_app != u->has_main_app) {
|
|
if (is_bot != u->is_bot) {
|
|
LOG_IF(ERROR, !is_deleted && !u->is_deleted && u->is_received)
|
|
<< "User.is_bot has changed for " << user_id << "/" << u->usernames << " from " << source << " from "
|
|
<< u->is_bot << " to " << is_bot;
|
|
u->is_full_info_changed = true;
|
|
}
|
|
u->is_verified = is_verified;
|
|
u->is_support = is_support;
|
|
u->is_bot = is_bot;
|
|
u->can_join_groups = can_join_groups;
|
|
u->can_read_all_group_messages = can_read_all_group_messages;
|
|
u->is_scam = is_scam;
|
|
u->is_fake = is_fake;
|
|
u->is_inline_bot = is_inline_bot;
|
|
u->is_business_bot = is_business_bot;
|
|
u->inline_query_placeholder = std::move(inline_query_placeholder);
|
|
u->need_location_bot = need_location_bot;
|
|
u->can_be_added_to_attach_menu = can_be_added_to_attach_menu;
|
|
u->bot_active_users = bot_active_users;
|
|
u->has_main_app = has_main_app;
|
|
|
|
LOG(DEBUG) << "Info has changed for " << user_id;
|
|
u->is_changed = true;
|
|
}
|
|
if (u->contact_require_premium != contact_require_premium) {
|
|
u->contact_require_premium = contact_require_premium;
|
|
u->is_changed = true;
|
|
user_full_contact_require_premium_.erase(user_id);
|
|
}
|
|
if (is_received && attach_menu_enabled != u->attach_menu_enabled) {
|
|
u->attach_menu_enabled = attach_menu_enabled;
|
|
u->is_changed = true;
|
|
}
|
|
if (is_premium != u->is_premium) {
|
|
u->is_premium = is_premium;
|
|
u->is_is_premium_changed = true;
|
|
u->is_changed = true;
|
|
u->is_full_info_changed = true;
|
|
}
|
|
if (is_received && can_be_edited_bot != u->can_be_edited_bot) {
|
|
u->can_be_edited_bot = can_be_edited_bot;
|
|
u->is_changed = true;
|
|
u->is_full_info_changed = true;
|
|
}
|
|
|
|
if (u->bot_info_version != bot_info_version) {
|
|
u->bot_info_version = bot_info_version;
|
|
LOG(DEBUG) << "Bot info version has changed for " << user_id;
|
|
u->need_save_to_database = true;
|
|
}
|
|
if (is_received && u->need_apply_min_photo != need_apply_min_photo) {
|
|
LOG(DEBUG) << "Need apply min photo has changed for " << user_id;
|
|
u->need_apply_min_photo = need_apply_min_photo;
|
|
u->need_save_to_database = true;
|
|
}
|
|
|
|
if (is_received && !u->is_received) {
|
|
u->is_received = true;
|
|
|
|
LOG(DEBUG) << "Receive " << user_id;
|
|
u->is_changed = true;
|
|
}
|
|
|
|
if (is_deleted != u->is_deleted) {
|
|
u->is_deleted = is_deleted;
|
|
|
|
LOG(DEBUG) << "User.is_deleted has changed for " << user_id << " to " << u->is_deleted;
|
|
u->is_is_deleted_changed = true;
|
|
u->is_changed = true;
|
|
}
|
|
|
|
bool has_language_code = (flags & USER_FLAG_HAS_LANGUAGE_CODE) != 0;
|
|
LOG_IF(ERROR, has_language_code && !td_->auth_manager_->is_bot())
|
|
<< "Receive language code for " << user_id << " from " << source;
|
|
if (u->language_code != user->lang_code_ && !user->lang_code_.empty()) {
|
|
u->language_code = user->lang_code_;
|
|
|
|
LOG(DEBUG) << "Language code has changed for " << user_id << " to " << u->language_code;
|
|
u->is_changed = true;
|
|
}
|
|
|
|
bool is_me_regular_user = !td_->auth_manager_->is_bot();
|
|
if (is_received || u->need_apply_min_photo || !u->is_received) {
|
|
on_update_user_photo(u, user_id, std::move(user->photo_), source);
|
|
}
|
|
if (is_me_regular_user) {
|
|
if (is_received || !u->is_received) {
|
|
on_update_user_phone_number(u, user_id, std::move(user->phone_));
|
|
}
|
|
if (is_received || !u->is_received || u->was_online == 0) {
|
|
on_update_user_online(u, user_id, std::move(user->status_));
|
|
}
|
|
if (is_received) {
|
|
auto is_mutual_contact = (flags & USER_FLAG_IS_MUTUAL_CONTACT) != 0;
|
|
auto is_close_friend = (flags2 & USER_FLAG_IS_CLOSE_FRIEND) != 0;
|
|
on_update_user_is_contact(u, user_id, is_contact, is_mutual_contact, is_close_friend);
|
|
}
|
|
}
|
|
|
|
if (is_received || !u->is_received) {
|
|
on_update_user_name(u, user_id, std::move(user->first_name_), std::move(user->last_name_));
|
|
on_update_user_usernames(u, user_id, Usernames{std::move(user->username_), std::move(user->usernames_)});
|
|
}
|
|
on_update_user_emoji_status(u, user_id, EmojiStatus(std::move(user->emoji_status_)));
|
|
PeerColor peer_color(user->color_);
|
|
on_update_user_accent_color_id(u, user_id, peer_color.accent_color_id_);
|
|
on_update_user_background_custom_emoji_id(u, user_id, peer_color.background_custom_emoji_id_);
|
|
PeerColor profile_peer_color(user->profile_color_);
|
|
on_update_user_profile_accent_color_id(u, user_id, profile_peer_color.accent_color_id_);
|
|
on_update_user_profile_background_custom_emoji_id(u, user_id, profile_peer_color.background_custom_emoji_id_);
|
|
if (is_me_regular_user) {
|
|
if (is_received) {
|
|
on_update_user_stories_hidden(u, user_id, stories_hidden);
|
|
}
|
|
if (stories_available || stories_unavailable) {
|
|
// update at the end, because it calls need_poll_user_active_stories
|
|
on_update_user_story_ids_impl(u, user_id, StoryId(user->stories_max_id_), StoryId());
|
|
}
|
|
auto restriction_reasons = get_restriction_reasons(std::move(user->restriction_reason_));
|
|
if (restriction_reasons != u->restriction_reasons) {
|
|
u->restriction_reasons = std::move(restriction_reasons);
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
if (u->cache_version != User::CACHE_VERSION && u->is_received) {
|
|
u->cache_version = User::CACHE_VERSION;
|
|
u->need_save_to_database = true;
|
|
}
|
|
u->is_received_from_server = true;
|
|
update_user(u, user_id);
|
|
}
|
|
|
|
void UserManager::on_get_users(vector<telegram_api::object_ptr<telegram_api::User>> &&users, const char *source) {
|
|
for (auto &user : users) {
|
|
on_get_user(std::move(user), source);
|
|
}
|
|
}
|
|
|
|
void UserManager::on_binlog_user_event(BinlogEvent &&event) {
|
|
if (!G()->use_chat_info_database()) {
|
|
binlog_erase(G()->td_db()->get_binlog(), event.id_);
|
|
return;
|
|
}
|
|
|
|
UserLogEvent log_event;
|
|
if (log_event_parse(log_event, event.get_data()).is_error()) {
|
|
LOG(ERROR) << "Failed to load a user from binlog";
|
|
binlog_erase(G()->td_db()->get_binlog(), event.id_);
|
|
return;
|
|
}
|
|
|
|
auto user_id = log_event.user_id;
|
|
if (have_min_user(user_id) || !user_id.is_valid()) {
|
|
LOG(ERROR) << "Skip adding already added " << user_id;
|
|
binlog_erase(G()->td_db()->get_binlog(), event.id_);
|
|
return;
|
|
}
|
|
|
|
LOG(INFO) << "Add " << user_id << " from binlog";
|
|
users_.set(user_id, std::move(log_event.u_out));
|
|
|
|
User *u = get_user(user_id);
|
|
CHECK(u != nullptr);
|
|
u->log_event_id = event.id_;
|
|
|
|
update_user(u, user_id, true, false);
|
|
}
|
|
|
|
void UserManager::on_binlog_secret_chat_event(BinlogEvent &&event) {
|
|
if (!G()->use_chat_info_database()) {
|
|
binlog_erase(G()->td_db()->get_binlog(), event.id_);
|
|
return;
|
|
}
|
|
|
|
SecretChatLogEvent log_event;
|
|
if (log_event_parse(log_event, event.get_data()).is_error()) {
|
|
LOG(ERROR) << "Failed to load a secret chat from binlog";
|
|
binlog_erase(G()->td_db()->get_binlog(), event.id_);
|
|
return;
|
|
}
|
|
|
|
auto secret_chat_id = log_event.secret_chat_id;
|
|
if (have_secret_chat(secret_chat_id) || !secret_chat_id.is_valid()) {
|
|
LOG(ERROR) << "Skip adding already added " << secret_chat_id;
|
|
binlog_erase(G()->td_db()->get_binlog(), event.id_);
|
|
return;
|
|
}
|
|
|
|
LOG(INFO) << "Add " << secret_chat_id << " from binlog";
|
|
secret_chats_.set(secret_chat_id, std::move(log_event.c_out));
|
|
|
|
SecretChat *c = get_secret_chat(secret_chat_id);
|
|
CHECK(c != nullptr);
|
|
c->log_event_id = event.id_;
|
|
|
|
update_secret_chat(c, secret_chat_id, true, false);
|
|
}
|
|
|
|
void UserManager::on_update_user_name(UserId user_id, string &&first_name, string &&last_name, Usernames &&usernames) {
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
User *u = get_user_force(user_id, "on_update_user_name");
|
|
if (u != nullptr) {
|
|
on_update_user_name(u, user_id, std::move(first_name), std::move(last_name));
|
|
on_update_user_usernames(u, user_id, std::move(usernames));
|
|
update_user(u, user_id);
|
|
} else {
|
|
LOG(INFO) << "Ignore update user name about unknown " << user_id;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_name(User *u, UserId user_id, string &&first_name, string &&last_name) {
|
|
if (first_name.empty() && last_name.empty()) {
|
|
first_name = u->phone_number;
|
|
}
|
|
if (u->first_name != first_name || u->last_name != last_name) {
|
|
u->first_name = std::move(first_name);
|
|
u->last_name = std::move(last_name);
|
|
u->is_name_changed = true;
|
|
LOG(DEBUG) << "Name has changed for " << user_id;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_usernames(User *u, UserId user_id, Usernames &&usernames) {
|
|
if (u->usernames != usernames) {
|
|
td_->dialog_manager_->on_dialog_usernames_updated(DialogId(user_id), u->usernames, usernames);
|
|
td_->messages_manager_->on_dialog_usernames_updated(DialogId(user_id), u->usernames, usernames);
|
|
if (u->can_be_edited_bot && u->usernames.get_editable_username() != usernames.get_editable_username()) {
|
|
u->is_full_info_changed = true;
|
|
}
|
|
u->usernames = std::move(usernames);
|
|
u->is_username_changed = true;
|
|
LOG(DEBUG) << "Usernames have changed for " << user_id;
|
|
u->is_changed = true;
|
|
} else if (u->is_bot || !td_->auth_manager_->is_bot()) {
|
|
td_->dialog_manager_->on_dialog_usernames_received(DialogId(user_id), usernames, false);
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_phone_number(UserId user_id, string &&phone_number) {
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
User *u = get_user_force(user_id, "on_update_user_phone_number");
|
|
if (u != nullptr) {
|
|
on_update_user_phone_number(u, user_id, std::move(phone_number));
|
|
update_user(u, user_id);
|
|
} else {
|
|
LOG(INFO) << "Ignore update user phone number about unknown " << user_id;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_phone_number(User *u, UserId user_id, string &&phone_number) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
clean_phone_number(phone_number);
|
|
if (u->phone_number != phone_number) {
|
|
if (!u->phone_number.empty()) {
|
|
auto it = resolved_phone_numbers_.find(u->phone_number);
|
|
if (it != resolved_phone_numbers_.end() && it->second == user_id) {
|
|
resolved_phone_numbers_.erase(it);
|
|
}
|
|
}
|
|
|
|
u->phone_number = std::move(phone_number);
|
|
u->is_phone_number_changed = true;
|
|
LOG(DEBUG) << "Phone number has changed for " << user_id;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_photo(User *u, UserId user_id,
|
|
telegram_api::object_ptr<telegram_api::UserProfilePhoto> &&photo,
|
|
const char *source) {
|
|
if (td_->auth_manager_->is_bot() && !G()->use_chat_info_database()) {
|
|
if (!u->is_photo_inited) {
|
|
auto new_photo_id = get_profile_photo_id(photo);
|
|
auto &old_photo = pending_user_photos_[user_id];
|
|
if (new_photo_id == get_profile_photo_id(old_photo)) {
|
|
return;
|
|
}
|
|
if (photo != nullptr && photo->get_id() == telegram_api::userProfilePhoto::ID) {
|
|
auto *profile_photo = static_cast<telegram_api::userProfilePhoto *>(photo.get());
|
|
if ((profile_photo->flags_ & telegram_api::userProfilePhoto::STRIPPED_THUMB_MASK) != 0) {
|
|
profile_photo->flags_ -= telegram_api::userProfilePhoto::STRIPPED_THUMB_MASK;
|
|
profile_photo->stripped_thumb_ = BufferSlice();
|
|
}
|
|
}
|
|
|
|
old_photo = std::move(photo);
|
|
|
|
drop_user_photos(user_id, new_photo_id == 0, "on_update_user_photo");
|
|
auto user_full = get_user_full(user_id); // must not load UserFull
|
|
if (user_full != nullptr && new_photo_id != get_user_full_profile_photo_id(user_full)) {
|
|
// we didn't sent updateUser yet, so we must not sent updateUserFull with new_photo_id yet
|
|
drop_user_full_photos(user_full, user_id, 0, "on_update_user_photo");
|
|
}
|
|
return;
|
|
}
|
|
if (u->is_received) {
|
|
auto new_photo_id = get_profile_photo_id(photo);
|
|
if (new_photo_id == u->photo.id) {
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
do_update_user_photo(u, user_id, std::move(photo), source);
|
|
}
|
|
|
|
void UserManager::do_update_user_photo(User *u, UserId user_id,
|
|
telegram_api::object_ptr<telegram_api::UserProfilePhoto> &&photo,
|
|
const char *source) {
|
|
ProfilePhoto new_photo = get_profile_photo(td_->file_manager_.get(), user_id, u->access_hash, std::move(photo));
|
|
if (td_->auth_manager_->is_bot()) {
|
|
new_photo.minithumbnail.clear();
|
|
}
|
|
do_update_user_photo(u, user_id, std::move(new_photo), true, source);
|
|
}
|
|
|
|
void UserManager::do_update_user_photo(User *u, UserId user_id, ProfilePhoto &&new_photo, bool invalidate_photo_cache,
|
|
const char *source) {
|
|
u->is_photo_inited = true;
|
|
if (need_update_profile_photo(u->photo, new_photo)) {
|
|
LOG_IF(ERROR, u->access_hash == -1 && new_photo.small_file_id.is_valid())
|
|
<< "Update profile photo of " << user_id << " without access hash from " << source;
|
|
u->photo = new_photo;
|
|
u->is_photo_changed = true;
|
|
LOG(DEBUG) << "Photo has changed for " << user_id << " to " << u->photo
|
|
<< ", invalidate_photo_cache = " << invalidate_photo_cache << " from " << source;
|
|
u->is_changed = true;
|
|
|
|
if (invalidate_photo_cache) {
|
|
drop_user_photos(user_id, u->photo.id == 0, source);
|
|
}
|
|
auto user_full = get_user_full(user_id); // must not load UserFull
|
|
if (user_full != nullptr && u->photo.id != get_user_full_profile_photo_id(user_full)) {
|
|
// we didn't sent updateUser yet, so we must not sent updateUserFull with u->photo.id yet
|
|
drop_user_full_photos(user_full, user_id, 0, "do_update_user_photo");
|
|
}
|
|
} else if (need_update_dialog_photo_minithumbnail(u->photo.minithumbnail, new_photo.minithumbnail)) {
|
|
LOG(DEBUG) << "Photo minithumbnail has changed for " << user_id << " from " << source;
|
|
u->photo.minithumbnail = std::move(new_photo.minithumbnail);
|
|
u->is_photo_changed = true;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::register_suggested_profile_photo(const Photo &photo) {
|
|
auto photo_file_ids = photo_get_file_ids(photo);
|
|
if (photo.is_empty() || photo_file_ids.empty()) {
|
|
return;
|
|
}
|
|
auto first_file_id = photo_file_ids[0];
|
|
auto file_type = td_->file_manager_->get_file_view(first_file_id).get_type();
|
|
if (file_type == FileType::ProfilePhoto) {
|
|
return;
|
|
}
|
|
CHECK(file_type == FileType::Photo);
|
|
auto photo_id = photo.id.get();
|
|
if (photo_id != 0) {
|
|
my_photo_file_id_[photo_id] = first_file_id;
|
|
}
|
|
}
|
|
|
|
void UserManager::register_user_photo(User *u, UserId user_id, const Photo &photo) {
|
|
auto photo_file_ids = photo_get_file_ids(photo);
|
|
if (photo.is_empty() || photo_file_ids.empty()) {
|
|
return;
|
|
}
|
|
auto first_file_id = photo_file_ids[0];
|
|
auto file_type = td_->file_manager_->get_file_view(first_file_id).get_type();
|
|
if (file_type == FileType::ProfilePhoto) {
|
|
return;
|
|
}
|
|
CHECK(file_type == FileType::Photo);
|
|
CHECK(u != nullptr);
|
|
auto photo_id = photo.id.get();
|
|
if (photo_id != 0 && u->photo_ids.emplace(photo_id).second) {
|
|
VLOG(file_references) << "Register photo " << photo_id << " of " << user_id;
|
|
if (user_id == get_my_id()) {
|
|
my_photo_file_id_[photo_id] = first_file_id;
|
|
}
|
|
auto file_source_id = user_profile_photo_file_source_ids_.get(std::make_pair(user_id, photo_id));
|
|
if (file_source_id.is_valid()) {
|
|
VLOG(file_references) << "Move " << file_source_id << " inside of " << user_id;
|
|
user_profile_photo_file_source_ids_.erase(std::make_pair(user_id, photo_id));
|
|
} else {
|
|
VLOG(file_references) << "Need to create new file source for photo " << photo_id << " of " << user_id;
|
|
file_source_id = td_->file_reference_manager_->create_user_photo_file_source(user_id, photo_id);
|
|
}
|
|
for (auto &file_id : photo_file_ids) {
|
|
td_->file_manager_->add_file_source(file_id, file_source_id);
|
|
}
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_accent_color_id(User *u, UserId user_id, AccentColorId accent_color_id) {
|
|
if (accent_color_id == AccentColorId(user_id) || !accent_color_id.is_valid()) {
|
|
accent_color_id = AccentColorId();
|
|
}
|
|
if (u->accent_color_id != accent_color_id) {
|
|
u->accent_color_id = accent_color_id;
|
|
u->is_accent_color_changed = true;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_background_custom_emoji_id(User *u, UserId user_id,
|
|
CustomEmojiId background_custom_emoji_id) {
|
|
if (u->background_custom_emoji_id != background_custom_emoji_id) {
|
|
u->background_custom_emoji_id = background_custom_emoji_id;
|
|
u->is_accent_color_changed = true;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_profile_accent_color_id(User *u, UserId user_id, AccentColorId accent_color_id) {
|
|
if (!accent_color_id.is_valid()) {
|
|
accent_color_id = AccentColorId();
|
|
}
|
|
if (u->profile_accent_color_id != accent_color_id) {
|
|
u->profile_accent_color_id = accent_color_id;
|
|
u->is_accent_color_changed = true;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_profile_background_custom_emoji_id(User *u, UserId user_id,
|
|
CustomEmojiId background_custom_emoji_id) {
|
|
if (u->profile_background_custom_emoji_id != background_custom_emoji_id) {
|
|
u->profile_background_custom_emoji_id = background_custom_emoji_id;
|
|
u->is_accent_color_changed = true;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_emoji_status(UserId user_id,
|
|
telegram_api::object_ptr<telegram_api::EmojiStatus> &&emoji_status) {
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
User *u = get_user_force(user_id, "on_update_user_emoji_status");
|
|
if (u != nullptr) {
|
|
on_update_user_emoji_status(u, user_id, EmojiStatus(std::move(emoji_status)));
|
|
update_user(u, user_id);
|
|
} else {
|
|
LOG(INFO) << "Ignore update user emoji status about unknown " << user_id;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_emoji_status(User *u, UserId user_id, EmojiStatus emoji_status) {
|
|
if (u->emoji_status != emoji_status) {
|
|
LOG(DEBUG) << "Change emoji status of " << user_id << " from " << u->emoji_status << " to " << emoji_status;
|
|
u->emoji_status = std::move(emoji_status);
|
|
u->is_emoji_status_changed = true;
|
|
// effective emoji status might not be changed; checked in update_user
|
|
// u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_story_ids(UserId user_id, StoryId max_active_story_id, StoryId max_read_story_id) {
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
User *u = get_user_force(user_id, "on_update_user_story_ids");
|
|
if (u != nullptr) {
|
|
on_update_user_story_ids_impl(u, user_id, max_active_story_id, max_read_story_id);
|
|
update_user(u, user_id);
|
|
} else {
|
|
LOG(INFO) << "Ignore update user story identifiers about unknown " << user_id;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_story_ids_impl(User *u, UserId user_id, StoryId max_active_story_id,
|
|
StoryId max_read_story_id) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
if (max_active_story_id != StoryId() && !max_active_story_id.is_server()) {
|
|
LOG(ERROR) << "Receive max active " << max_active_story_id << " for " << user_id;
|
|
return;
|
|
}
|
|
if (max_read_story_id != StoryId() && !max_read_story_id.is_server()) {
|
|
LOG(ERROR) << "Receive max read " << max_read_story_id << " for " << user_id;
|
|
return;
|
|
}
|
|
|
|
auto has_unread_stories = get_user_has_unread_stories(u);
|
|
if (u->max_active_story_id != max_active_story_id) {
|
|
LOG(DEBUG) << "Change last active story of " << user_id << " from " << u->max_active_story_id << " to "
|
|
<< max_active_story_id;
|
|
u->max_active_story_id = max_active_story_id;
|
|
u->need_save_to_database = true;
|
|
}
|
|
if (need_poll_user_active_stories(u, user_id)) {
|
|
auto max_active_story_id_next_reload_time = Time::now() + MAX_ACTIVE_STORY_ID_RELOAD_TIME;
|
|
if (max_active_story_id_next_reload_time >
|
|
u->max_active_story_id_next_reload_time + MAX_ACTIVE_STORY_ID_RELOAD_TIME / 5) {
|
|
LOG(DEBUG) << "Change max_active_story_id_next_reload_time of " << user_id;
|
|
u->max_active_story_id_next_reload_time = max_active_story_id_next_reload_time;
|
|
u->need_save_to_database = true;
|
|
}
|
|
}
|
|
if (!max_active_story_id.is_valid()) {
|
|
CHECK(max_read_story_id == StoryId());
|
|
if (u->max_read_story_id != StoryId()) {
|
|
LOG(DEBUG) << "Drop last read " << u->max_read_story_id << " of " << user_id;
|
|
u->max_read_story_id = StoryId();
|
|
u->need_save_to_database = true;
|
|
}
|
|
} else if (max_read_story_id.get() > u->max_read_story_id.get()) {
|
|
LOG(DEBUG) << "Change last read story of " << user_id << " from " << u->max_read_story_id << " to "
|
|
<< max_read_story_id;
|
|
u->max_read_story_id = max_read_story_id;
|
|
u->need_save_to_database = true;
|
|
}
|
|
if (has_unread_stories != get_user_has_unread_stories(u)) {
|
|
LOG(DEBUG) << "Change has_unread_stories of " << user_id << " to " << !has_unread_stories;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_max_read_story_id(UserId user_id, StoryId max_read_story_id) {
|
|
CHECK(user_id.is_valid());
|
|
|
|
User *u = get_user(user_id);
|
|
if (u != nullptr) {
|
|
on_update_user_max_read_story_id(u, user_id, max_read_story_id);
|
|
update_user(u, user_id);
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_max_read_story_id(User *u, UserId user_id, StoryId max_read_story_id) {
|
|
if (td_->auth_manager_->is_bot() || !u->is_received) {
|
|
return;
|
|
}
|
|
|
|
auto has_unread_stories = get_user_has_unread_stories(u);
|
|
if (max_read_story_id.get() > u->max_read_story_id.get()) {
|
|
LOG(DEBUG) << "Change last read story of " << user_id << " from " << u->max_read_story_id << " to "
|
|
<< max_read_story_id;
|
|
u->max_read_story_id = max_read_story_id;
|
|
u->need_save_to_database = true;
|
|
}
|
|
if (has_unread_stories != get_user_has_unread_stories(u)) {
|
|
LOG(DEBUG) << "Change has_unread_stories of " << user_id << " to " << !has_unread_stories;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_stories_hidden(UserId user_id, bool stories_hidden) {
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
User *u = get_user_force(user_id, "on_update_user_stories_hidden");
|
|
if (u != nullptr) {
|
|
on_update_user_stories_hidden(u, user_id, stories_hidden);
|
|
update_user(u, user_id);
|
|
} else {
|
|
LOG(INFO) << "Ignore update user stories are archived about unknown " << user_id;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_stories_hidden(User *u, UserId user_id, bool stories_hidden) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
if (u->stories_hidden != stories_hidden) {
|
|
LOG(DEBUG) << "Change stories are archived of " << user_id << " to " << stories_hidden;
|
|
u->stories_hidden = stories_hidden;
|
|
u->is_stories_hidden_changed = true;
|
|
u->need_save_to_database = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_is_contact(User *u, UserId user_id, bool is_contact, bool is_mutual_contact,
|
|
bool is_close_friend) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
UserId my_id = get_my_id();
|
|
if (user_id == my_id) {
|
|
is_mutual_contact = is_contact;
|
|
is_close_friend = false;
|
|
}
|
|
if (!is_contact && (is_mutual_contact || is_close_friend)) {
|
|
LOG(ERROR) << "Receive is_mutual_contact = " << is_mutual_contact << ", and is_close_friend = " << is_close_friend
|
|
<< " for non-contact " << user_id;
|
|
is_mutual_contact = false;
|
|
is_close_friend = false;
|
|
}
|
|
|
|
if (u->is_contact != is_contact || u->is_mutual_contact != is_mutual_contact ||
|
|
u->is_close_friend != is_close_friend) {
|
|
LOG(DEBUG) << "Update " << user_id << " is_contact from (" << u->is_contact << ", " << u->is_mutual_contact << ", "
|
|
<< u->is_close_friend << ") to (" << is_contact << ", " << is_mutual_contact << ", " << is_close_friend
|
|
<< ")";
|
|
if (u->is_contact != is_contact) {
|
|
u->is_contact = is_contact;
|
|
u->is_is_contact_changed = true;
|
|
}
|
|
if (u->is_mutual_contact != is_mutual_contact) {
|
|
u->is_mutual_contact = is_mutual_contact;
|
|
u->is_is_mutual_contact_changed = true;
|
|
|
|
reload_contact_birthdates(true);
|
|
}
|
|
u->is_close_friend = is_close_friend;
|
|
u->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_online(UserId user_id, telegram_api::object_ptr<telegram_api::UserStatus> &&status) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
User *u = get_user_force(user_id, "on_update_user_online");
|
|
if (u != nullptr) {
|
|
if (u->is_bot) {
|
|
LOG(ERROR) << "Receive updateUserStatus about bot " << user_id;
|
|
return;
|
|
}
|
|
on_update_user_online(u, user_id, std::move(status));
|
|
update_user(u, user_id);
|
|
|
|
if (user_id == get_my_id() &&
|
|
was_online_remote_ != u->was_online) { // only update was_online_remote_ from updateUserStatus
|
|
was_online_remote_ = u->was_online;
|
|
VLOG(notifications) << "Set was_online_remote to " << was_online_remote_;
|
|
G()->td_db()->get_binlog_pmc()->set("my_was_online_remote", to_string(was_online_remote_));
|
|
}
|
|
} else {
|
|
LOG(INFO) << "Ignore update user online about unknown " << user_id;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_online(User *u, UserId user_id,
|
|
telegram_api::object_ptr<telegram_api::UserStatus> &&status) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
int32 id = status == nullptr ? telegram_api::userStatusEmpty::ID : status->get_id();
|
|
int32 new_online;
|
|
bool is_offline = false;
|
|
if (id == telegram_api::userStatusOnline::ID) {
|
|
int32 now = G()->unix_time();
|
|
|
|
auto st = move_tl_object_as<telegram_api::userStatusOnline>(status);
|
|
new_online = st->expires_;
|
|
LOG_IF(ERROR, new_online < now - 86400)
|
|
<< "Receive userStatusOnline expired more than one day in past " << new_online;
|
|
} else if (id == telegram_api::userStatusOffline::ID) {
|
|
int32 now = G()->unix_time();
|
|
|
|
auto st = move_tl_object_as<telegram_api::userStatusOffline>(status);
|
|
new_online = st->was_online_;
|
|
if (new_online >= now) {
|
|
LOG_IF(ERROR, new_online > now + 10)
|
|
<< "Receive userStatusOffline but was online points to future time " << new_online << ", now is " << now;
|
|
new_online = now - 1;
|
|
}
|
|
is_offline = true;
|
|
} else if (id == telegram_api::userStatusRecently::ID) {
|
|
auto st = telegram_api::move_object_as<telegram_api::userStatusRecently>(status);
|
|
new_online = st->by_me_ ? -4 : -1;
|
|
} else if (id == telegram_api::userStatusLastWeek::ID) {
|
|
auto st = telegram_api::move_object_as<telegram_api::userStatusLastWeek>(status);
|
|
new_online = st->by_me_ ? -5 : -2;
|
|
} else if (id == telegram_api::userStatusLastMonth::ID) {
|
|
auto st = telegram_api::move_object_as<telegram_api::userStatusLastMonth>(status);
|
|
new_online = st->by_me_ ? -6 : -3;
|
|
} else {
|
|
CHECK(id == telegram_api::userStatusEmpty::ID);
|
|
new_online = 0;
|
|
}
|
|
|
|
if (new_online != u->was_online && !(new_online < 0 && user_id == get_my_id())) {
|
|
LOG(DEBUG) << "Update " << user_id << " online from " << u->was_online << " to " << new_online;
|
|
auto unix_time = G()->unix_time();
|
|
bool old_is_online = u->was_online > unix_time;
|
|
bool new_is_online = new_online > unix_time;
|
|
u->was_online = new_online;
|
|
u->is_status_changed = true;
|
|
if (u->was_online > 0) {
|
|
u->local_was_online = 0;
|
|
}
|
|
|
|
if (user_id == get_my_id()) {
|
|
if (my_was_online_local_ != 0 || old_is_online != new_is_online) {
|
|
my_was_online_local_ = 0;
|
|
u->is_online_status_changed = true;
|
|
}
|
|
if (is_offline) {
|
|
td_->online_manager_->on_online_updated(false, false);
|
|
}
|
|
} else if (old_is_online != new_is_online) {
|
|
u->is_online_status_changed = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_local_was_online(UserId user_id, int32 local_was_online) {
|
|
CHECK(user_id.is_valid());
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
User *u = get_user_force(user_id, "on_update_user_local_was_online");
|
|
if (u == nullptr) {
|
|
return;
|
|
}
|
|
|
|
on_update_user_local_was_online(u, user_id, local_was_online);
|
|
update_user(u, user_id);
|
|
}
|
|
|
|
void UserManager::on_update_user_local_was_online(User *u, UserId user_id, int32 local_was_online) {
|
|
CHECK(u != nullptr);
|
|
if (u->is_deleted || u->is_bot || u->is_support || user_id == get_my_id()) {
|
|
return;
|
|
}
|
|
int32 unix_time = G()->unix_time();
|
|
if (u->was_online > unix_time) {
|
|
// if user is currently online, ignore local online
|
|
return;
|
|
}
|
|
|
|
// bring users online for 30 seconds
|
|
local_was_online += 30;
|
|
if (local_was_online < unix_time + 2 || local_was_online <= u->local_was_online ||
|
|
local_was_online <= u->was_online) {
|
|
return;
|
|
}
|
|
|
|
LOG(DEBUG) << "Update " << user_id << " local online from " << u->local_was_online << " to " << local_was_online;
|
|
bool old_is_online = u->local_was_online > unix_time;
|
|
u->local_was_online = local_was_online;
|
|
u->is_status_changed = true;
|
|
|
|
if (!old_is_online) {
|
|
u->is_online_status_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_is_blocked(UserId user_id, bool is_blocked, bool is_blocked_for_stories) {
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_is_blocked");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_is_blocked(user_full, user_id, is_blocked, is_blocked_for_stories);
|
|
update_user_full(user_full, user_id, "on_update_user_is_blocked");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_is_blocked(UserFull *user_full, UserId user_id, bool is_blocked,
|
|
bool is_blocked_for_stories) {
|
|
CHECK(user_full != nullptr);
|
|
if (user_full->is_blocked != is_blocked || user_full->is_blocked_for_stories != is_blocked_for_stories) {
|
|
LOG(INFO) << "Receive update user full is blocked with " << user_id << " and is_blocked = " << is_blocked << '/'
|
|
<< is_blocked_for_stories;
|
|
user_full->is_blocked = is_blocked;
|
|
user_full->is_blocked_for_stories = is_blocked_for_stories;
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_has_pinned_stories(UserId user_id, bool has_pinned_stories) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_has_pinned_stories");
|
|
if (user_full == nullptr || user_full->has_pinned_stories == has_pinned_stories) {
|
|
return;
|
|
}
|
|
user_full->has_pinned_stories = has_pinned_stories;
|
|
user_full->is_changed = true;
|
|
update_user_full(user_full, user_id, "on_update_user_has_pinned_stories");
|
|
}
|
|
|
|
void UserManager::on_update_user_common_chat_count(UserId user_id, int32 common_chat_count) {
|
|
LOG(INFO) << "Receive " << common_chat_count << " common chat count with " << user_id;
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_common_chat_count");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_common_chat_count(user_full, user_id, common_chat_count);
|
|
update_user_full(user_full, user_id, "on_update_user_common_chat_count");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_common_chat_count(UserFull *user_full, UserId user_id, int32 common_chat_count) {
|
|
CHECK(user_full != nullptr);
|
|
if (common_chat_count < 0) {
|
|
LOG(ERROR) << "Receive " << common_chat_count << " as common group count with " << user_id;
|
|
common_chat_count = 0;
|
|
}
|
|
if (user_full->common_chat_count != common_chat_count) {
|
|
user_full->common_chat_count = common_chat_count;
|
|
user_full->is_common_chat_count_changed = true;
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_location(UserId user_id, DialogLocation &&location) {
|
|
LOG(INFO) << "Receive " << location << " for " << user_id;
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_location");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_location(user_full, user_id, std::move(location));
|
|
update_user_full(user_full, user_id, "on_update_user_location");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_location(UserFull *user_full, UserId user_id, DialogLocation &&location) {
|
|
CHECK(user_full != nullptr);
|
|
if (BusinessInfo::set_location(user_full->business_info, std::move(location))) {
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_work_hours(UserId user_id, BusinessWorkHours &&work_hours) {
|
|
LOG(INFO) << "Receive " << work_hours << " for " << user_id;
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_work_hours");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_work_hours(user_full, user_id, std::move(work_hours));
|
|
update_user_full(user_full, user_id, "on_update_user_work_hours");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_work_hours(UserFull *user_full, UserId user_id, BusinessWorkHours &&work_hours) {
|
|
CHECK(user_full != nullptr);
|
|
if (BusinessInfo::set_work_hours(user_full->business_info, std::move(work_hours))) {
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_away_message(UserId user_id, BusinessAwayMessage &&away_message) {
|
|
LOG(INFO) << "Receive " << away_message << " for " << user_id;
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_away_message");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_away_message(user_full, user_id, std::move(away_message));
|
|
update_user_full(user_full, user_id, "on_update_user_away_message");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_away_message(UserFull *user_full, UserId user_id,
|
|
BusinessAwayMessage &&away_message) const {
|
|
CHECK(user_full != nullptr);
|
|
if (away_message.is_valid() && user_id != get_my_id()) {
|
|
LOG(ERROR) << "Receive " << away_message << " for " << user_id;
|
|
return;
|
|
}
|
|
if (BusinessInfo::set_away_message(user_full->business_info, std::move(away_message))) {
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_greeting_message(UserId user_id, BusinessGreetingMessage &&greeting_message) {
|
|
LOG(INFO) << "Receive " << greeting_message << " for " << user_id;
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_greeting_message");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_greeting_message(user_full, user_id, std::move(greeting_message));
|
|
update_user_full(user_full, user_id, "on_update_user_greeting_message");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_greeting_message(UserFull *user_full, UserId user_id,
|
|
BusinessGreetingMessage &&greeting_message) const {
|
|
CHECK(user_full != nullptr);
|
|
if (greeting_message.is_valid() && user_id != get_my_id()) {
|
|
LOG(ERROR) << "Receive " << greeting_message << " for " << user_id;
|
|
return;
|
|
}
|
|
if (BusinessInfo::set_greeting_message(user_full->business_info, std::move(greeting_message))) {
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_intro(UserId user_id, BusinessIntro &&intro) {
|
|
LOG(INFO) << "Receive " << intro << " for " << user_id;
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_intro");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_intro(user_full, user_id, std::move(intro));
|
|
update_user_full(user_full, user_id, "on_update_user_intro");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_intro(UserFull *user_full, UserId user_id, BusinessIntro &&intro) {
|
|
CHECK(user_full != nullptr);
|
|
if (BusinessInfo::set_intro(user_full->business_info, std::move(intro))) {
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_commands(UserId user_id,
|
|
vector<telegram_api::object_ptr<telegram_api::botCommand>> &&bot_commands) {
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_commands");
|
|
if (user_full != nullptr) {
|
|
on_update_user_full_commands(user_full, user_id, std::move(bot_commands));
|
|
update_user_full(user_full, user_id, "on_update_user_commands");
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_full_commands(
|
|
UserFull *user_full, UserId user_id, vector<telegram_api::object_ptr<telegram_api::botCommand>> &&bot_commands) {
|
|
CHECK(user_full != nullptr);
|
|
auto commands =
|
|
transform(std::move(bot_commands), [](telegram_api::object_ptr<telegram_api::botCommand> &&bot_command) {
|
|
return BotCommand(std::move(bot_command));
|
|
});
|
|
if (user_full->commands != commands) {
|
|
user_full->commands = std::move(commands);
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_need_phone_number_privacy_exception(UserId user_id,
|
|
bool need_phone_number_privacy_exception) {
|
|
LOG(INFO) << "Receive " << need_phone_number_privacy_exception << " need phone number privacy exception with "
|
|
<< user_id;
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_need_phone_number_privacy_exception");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_need_phone_number_privacy_exception(user_full, user_id, need_phone_number_privacy_exception);
|
|
update_user_full(user_full, user_id, "on_update_user_need_phone_number_privacy_exception");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_need_phone_number_privacy_exception(
|
|
UserFull *user_full, UserId user_id, bool need_phone_number_privacy_exception) const {
|
|
CHECK(user_full != nullptr);
|
|
if (need_phone_number_privacy_exception) {
|
|
const User *u = get_user(user_id);
|
|
if (u == nullptr || u->is_contact || user_id == get_my_id()) {
|
|
need_phone_number_privacy_exception = false;
|
|
}
|
|
}
|
|
if (user_full->need_phone_number_privacy_exception != need_phone_number_privacy_exception) {
|
|
user_full->need_phone_number_privacy_exception = need_phone_number_privacy_exception;
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_wallpaper_overridden(UserId user_id, bool wallpaper_overridden) {
|
|
LOG(INFO) << "Receive " << wallpaper_overridden << " set chat background for " << user_id;
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full_force(user_id, "on_update_user_wallpaper_overridden");
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
on_update_user_full_wallpaper_overridden(user_full, user_id, wallpaper_overridden);
|
|
update_user_full(user_full, user_id, "on_update_user_wallpaper_overridden");
|
|
}
|
|
|
|
void UserManager::on_update_user_full_wallpaper_overridden(UserFull *user_full, UserId user_id,
|
|
bool wallpaper_overridden) const {
|
|
CHECK(user_full != nullptr);
|
|
if (user_full->wallpaper_overridden != wallpaper_overridden) {
|
|
user_full->wallpaper_overridden = wallpaper_overridden;
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_bot_menu_button(UserId bot_user_id,
|
|
telegram_api::object_ptr<telegram_api::BotMenuButton> &&bot_menu_button) {
|
|
if (!bot_user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive updateBotMenuButton about invalid " << bot_user_id;
|
|
return;
|
|
}
|
|
if (!have_user_force(bot_user_id, "on_update_bot_menu_button") || !is_user_bot(bot_user_id)) {
|
|
return;
|
|
}
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
auto user_full = get_user_full_force(bot_user_id, "on_update_bot_menu_button");
|
|
if (user_full != nullptr) {
|
|
on_update_user_full_menu_button(user_full, bot_user_id, std::move(bot_menu_button));
|
|
update_user_full(user_full, bot_user_id, "on_update_bot_menu_button");
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_full_menu_button(
|
|
UserFull *user_full, UserId user_id, telegram_api::object_ptr<telegram_api::BotMenuButton> &&bot_menu_button) {
|
|
CHECK(user_full != nullptr);
|
|
auto new_button = get_bot_menu_button(std::move(bot_menu_button));
|
|
bool is_changed;
|
|
if (user_full->menu_button == nullptr) {
|
|
is_changed = (new_button != nullptr);
|
|
} else {
|
|
is_changed = (new_button == nullptr || *user_full->menu_button != *new_button);
|
|
}
|
|
if (is_changed) {
|
|
user_full->menu_button = std::move(new_button);
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_bot_has_preview_medias(UserId bot_user_id, bool has_preview_medias) {
|
|
if (!bot_user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive updateBotHasPreviewMedias about invalid " << bot_user_id;
|
|
return;
|
|
}
|
|
if (!have_user_force(bot_user_id, "on_update_bot_has_preview_medias") || !is_user_bot(bot_user_id)) {
|
|
return;
|
|
}
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
auto user_full = get_user_full_force(bot_user_id, "on_update_bot_has_preview_medias");
|
|
if (user_full != nullptr) {
|
|
on_update_user_full_has_preview_medias(user_full, bot_user_id, has_preview_medias);
|
|
update_user_full(user_full, bot_user_id, "on_update_bot_has_preview_medias");
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_user_full_has_preview_medias(UserFull *user_full, UserId user_id, bool has_preview_medias) {
|
|
CHECK(user_full != nullptr);
|
|
if (user_full->has_preview_medias != has_preview_medias) {
|
|
user_full->has_preview_medias = has_preview_medias;
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::on_update_secret_chat(SecretChatId secret_chat_id, int64 access_hash, UserId user_id,
|
|
SecretChatState state, bool is_outbound, int32 ttl, int32 date, string key_hash,
|
|
int32 layer, FolderId initial_folder_id) {
|
|
LOG(INFO) << "Update " << secret_chat_id << " with " << user_id << " and access_hash " << access_hash;
|
|
auto *secret_chat = add_secret_chat(secret_chat_id);
|
|
if (access_hash != secret_chat->access_hash) {
|
|
secret_chat->access_hash = access_hash;
|
|
secret_chat->need_save_to_database = true;
|
|
}
|
|
if (user_id.is_valid() && user_id != secret_chat->user_id) {
|
|
if (secret_chat->user_id.is_valid()) {
|
|
LOG(ERROR) << "Secret chat user has changed from " << secret_chat->user_id << " to " << user_id;
|
|
auto &old_secret_chat_ids = secret_chats_with_user_[secret_chat->user_id];
|
|
td::remove(old_secret_chat_ids, secret_chat_id);
|
|
}
|
|
secret_chat->user_id = user_id;
|
|
secret_chats_with_user_[secret_chat->user_id].push_back(secret_chat_id);
|
|
secret_chat->is_changed = true;
|
|
}
|
|
if (state != SecretChatState::Unknown && state != secret_chat->state) {
|
|
secret_chat->state = state;
|
|
secret_chat->is_changed = true;
|
|
secret_chat->is_state_changed = true;
|
|
}
|
|
if (is_outbound != secret_chat->is_outbound) {
|
|
secret_chat->is_outbound = is_outbound;
|
|
secret_chat->is_changed = true;
|
|
}
|
|
|
|
if (ttl != -1 && ttl != secret_chat->ttl) {
|
|
secret_chat->ttl = ttl;
|
|
secret_chat->need_save_to_database = true;
|
|
secret_chat->is_ttl_changed = true;
|
|
}
|
|
if (date != 0 && date != secret_chat->date) {
|
|
secret_chat->date = date;
|
|
secret_chat->need_save_to_database = true;
|
|
}
|
|
if (!key_hash.empty() && key_hash != secret_chat->key_hash) {
|
|
secret_chat->key_hash = std::move(key_hash);
|
|
secret_chat->is_changed = true;
|
|
}
|
|
if (layer != 0 && layer != secret_chat->layer) {
|
|
secret_chat->layer = layer;
|
|
secret_chat->is_changed = true;
|
|
}
|
|
if (initial_folder_id != FolderId() && initial_folder_id != secret_chat->initial_folder_id) {
|
|
secret_chat->initial_folder_id = initial_folder_id;
|
|
secret_chat->is_changed = true;
|
|
}
|
|
|
|
update_secret_chat(secret_chat, secret_chat_id);
|
|
}
|
|
|
|
void UserManager::on_update_online_status_privacy() {
|
|
td_->create_handler<GetContactsStatusesQuery>()->send();
|
|
}
|
|
|
|
void UserManager::on_update_phone_number_privacy() {
|
|
// all UserFull.need_phone_number_privacy_exception can be outdated now,
|
|
// so mark all of them as expired
|
|
users_full_.foreach([&](const UserId &user_id, unique_ptr<UserFull> &user_full) { user_full->expires_at = 0.0; });
|
|
}
|
|
|
|
void UserManager::on_ignored_restriction_reasons_changed() {
|
|
restricted_user_ids_.foreach([&](const UserId &user_id) {
|
|
send_closure(G()->td(), &Td::send_update, get_update_user_object(user_id, get_user(user_id)));
|
|
});
|
|
}
|
|
|
|
void UserManager::invalidate_user_full(UserId user_id) {
|
|
auto user_full = get_user_full_force(user_id, "invalidate_user_full");
|
|
if (user_full != nullptr) {
|
|
td_->dialog_manager_->on_dialog_info_full_invalidated(DialogId(user_id));
|
|
|
|
if (!user_full->is_expired()) {
|
|
user_full->expires_at = 0.0;
|
|
user_full->need_save_to_database = true;
|
|
|
|
update_user_full(user_full, user_id, "invalidate_user_full");
|
|
}
|
|
}
|
|
}
|
|
|
|
bool UserManager::have_user(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
return u != nullptr && u->is_received;
|
|
}
|
|
|
|
bool UserManager::have_min_user(UserId user_id) const {
|
|
return users_.count(user_id) > 0;
|
|
}
|
|
|
|
const UserManager::User *UserManager::get_user(UserId user_id) const {
|
|
return users_.get_pointer(user_id);
|
|
}
|
|
|
|
UserManager::User *UserManager::get_user(UserId user_id) {
|
|
return users_.get_pointer(user_id);
|
|
}
|
|
|
|
UserManager::User *UserManager::add_user(UserId user_id) {
|
|
CHECK(user_id.is_valid());
|
|
auto &user_ptr = users_[user_id];
|
|
if (user_ptr == nullptr) {
|
|
user_ptr = make_unique<User>();
|
|
}
|
|
return user_ptr.get();
|
|
}
|
|
|
|
void UserManager::save_user(User *u, UserId user_id, bool from_binlog) {
|
|
if (!G()->use_chat_info_database()) {
|
|
return;
|
|
}
|
|
CHECK(u != nullptr);
|
|
if (!u->is_saved || !u->is_status_saved) { // TODO more effective handling of !u->is_status_saved
|
|
if (!from_binlog) {
|
|
auto log_event = UserLogEvent(user_id, u);
|
|
auto storer = get_log_event_storer(log_event);
|
|
if (u->log_event_id == 0) {
|
|
u->log_event_id = binlog_add(G()->td_db()->get_binlog(), LogEvent::HandlerType::Users, storer);
|
|
} else {
|
|
binlog_rewrite(G()->td_db()->get_binlog(), u->log_event_id, LogEvent::HandlerType::Users, storer);
|
|
}
|
|
}
|
|
|
|
save_user_to_database(u, user_id);
|
|
}
|
|
}
|
|
|
|
string UserManager::get_user_database_key(UserId user_id) {
|
|
return PSTRING() << "us" << user_id.get();
|
|
}
|
|
|
|
string UserManager::get_user_database_value(const User *u) {
|
|
return log_event_store(*u).as_slice().str();
|
|
}
|
|
|
|
void UserManager::save_user_to_database(User *u, UserId user_id) {
|
|
CHECK(u != nullptr);
|
|
if (u->is_being_saved) {
|
|
return;
|
|
}
|
|
if (loaded_from_database_users_.count(user_id)) {
|
|
save_user_to_database_impl(u, user_id, get_user_database_value(u));
|
|
return;
|
|
}
|
|
if (load_user_from_database_queries_.count(user_id) != 0) {
|
|
return;
|
|
}
|
|
|
|
load_user_from_database_impl(user_id, Auto());
|
|
}
|
|
|
|
void UserManager::save_user_to_database_impl(User *u, UserId user_id, string value) {
|
|
CHECK(u != nullptr);
|
|
CHECK(load_user_from_database_queries_.count(user_id) == 0);
|
|
CHECK(!u->is_being_saved);
|
|
u->is_being_saved = true;
|
|
u->is_saved = true;
|
|
u->is_status_saved = true;
|
|
LOG(INFO) << "Trying to save to database " << user_id;
|
|
G()->td_db()->get_sqlite_pmc()->set(
|
|
get_user_database_key(user_id), std::move(value), PromiseCreator::lambda([user_id](Result<> result) {
|
|
send_closure(G()->user_manager(), &UserManager::on_save_user_to_database, user_id, result.is_ok());
|
|
}));
|
|
}
|
|
|
|
void UserManager::on_save_user_to_database(UserId user_id, bool success) {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
|
|
User *u = get_user(user_id);
|
|
CHECK(u != nullptr);
|
|
LOG_CHECK(u->is_being_saved) << success << ' ' << user_id << ' ' << u->is_saved << ' ' << u->is_status_saved << ' '
|
|
<< load_user_from_database_queries_.count(user_id) << ' ' << u->is_received << ' '
|
|
<< u->is_deleted << ' ' << u->is_bot << ' ' << u->need_save_to_database << ' '
|
|
<< u->is_changed << ' ' << u->is_status_changed << ' ' << u->is_name_changed << ' '
|
|
<< u->is_username_changed << ' ' << u->is_photo_changed << ' '
|
|
<< u->is_is_contact_changed << ' ' << u->is_is_deleted_changed << ' '
|
|
<< u->is_stories_hidden_changed << ' ' << u->log_event_id;
|
|
CHECK(load_user_from_database_queries_.count(user_id) == 0);
|
|
u->is_being_saved = false;
|
|
|
|
if (!success) {
|
|
LOG(ERROR) << "Failed to save " << user_id << " to database";
|
|
u->is_saved = false;
|
|
u->is_status_saved = false;
|
|
} else {
|
|
LOG(INFO) << "Successfully saved " << user_id << " to database";
|
|
}
|
|
if (u->is_saved && u->is_status_saved) {
|
|
if (u->log_event_id != 0) {
|
|
binlog_erase(G()->td_db()->get_binlog(), u->log_event_id);
|
|
u->log_event_id = 0;
|
|
}
|
|
} else {
|
|
save_user(u, user_id, u->log_event_id != 0);
|
|
}
|
|
}
|
|
|
|
void UserManager::load_user_from_database(User *u, UserId user_id, Promise<Unit> promise) {
|
|
if (loaded_from_database_users_.count(user_id)) {
|
|
promise.set_value(Unit());
|
|
return;
|
|
}
|
|
|
|
CHECK(u == nullptr || !u->is_being_saved);
|
|
load_user_from_database_impl(user_id, std::move(promise));
|
|
}
|
|
|
|
void UserManager::load_user_from_database_impl(UserId user_id, Promise<Unit> promise) {
|
|
LOG(INFO) << "Load " << user_id << " from database";
|
|
auto &load_user_queries = load_user_from_database_queries_[user_id];
|
|
load_user_queries.push_back(std::move(promise));
|
|
if (load_user_queries.size() == 1u) {
|
|
G()->td_db()->get_sqlite_pmc()->get(get_user_database_key(user_id), PromiseCreator::lambda([user_id](string value) {
|
|
send_closure(G()->user_manager(), &UserManager::on_load_user_from_database,
|
|
user_id, std::move(value), false);
|
|
}));
|
|
}
|
|
}
|
|
|
|
void UserManager::on_load_user_from_database(UserId user_id, string value, bool force) {
|
|
if (G()->close_flag() && !force) {
|
|
// the user is in Binlog and will be saved after restart
|
|
return;
|
|
}
|
|
|
|
CHECK(user_id.is_valid());
|
|
if (!loaded_from_database_users_.insert(user_id).second) {
|
|
return;
|
|
}
|
|
|
|
auto it = load_user_from_database_queries_.find(user_id);
|
|
vector<Promise<Unit>> promises;
|
|
if (it != load_user_from_database_queries_.end()) {
|
|
promises = std::move(it->second);
|
|
CHECK(!promises.empty());
|
|
load_user_from_database_queries_.erase(it);
|
|
}
|
|
|
|
LOG(INFO) << "Successfully loaded " << user_id << " of size " << value.size() << " from database";
|
|
// G()->td_db()->get_sqlite_pmc()->erase(get_user_database_key(user_id), Auto());
|
|
// return;
|
|
|
|
User *u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
if (!value.empty()) {
|
|
u = add_user(user_id);
|
|
|
|
if (log_event_parse(*u, value).is_error()) {
|
|
LOG(ERROR) << "Failed to load " << user_id << " from database";
|
|
users_.erase(user_id);
|
|
} else {
|
|
u->is_saved = true;
|
|
u->is_status_saved = true;
|
|
update_user(u, user_id, true, true);
|
|
}
|
|
}
|
|
} else {
|
|
CHECK(!u->is_saved); // user can't be saved before load completes
|
|
CHECK(!u->is_being_saved);
|
|
auto new_value = get_user_database_value(u);
|
|
if (value != new_value) {
|
|
save_user_to_database_impl(u, user_id, std::move(new_value));
|
|
} else if (u->log_event_id != 0) {
|
|
binlog_erase(G()->td_db()->get_binlog(), u->log_event_id);
|
|
u->log_event_id = 0;
|
|
}
|
|
}
|
|
|
|
set_promises(promises);
|
|
}
|
|
|
|
bool UserManager::have_user_force(UserId user_id, const char *source) {
|
|
return get_user_force(user_id, source) != nullptr;
|
|
}
|
|
|
|
UserManager::User *UserManager::get_user_force(UserId user_id, const char *source) {
|
|
auto u = get_user_force_impl(user_id, source);
|
|
if ((u == nullptr || !u->is_received) &&
|
|
(user_id == get_service_notifications_user_id() || user_id == get_replies_bot_user_id() ||
|
|
user_id == get_anonymous_bot_user_id() || user_id == get_channel_bot_user_id() ||
|
|
user_id == get_anti_spam_bot_user_id())) {
|
|
int32 flags = USER_FLAG_HAS_ACCESS_HASH | USER_FLAG_HAS_FIRST_NAME | USER_FLAG_NEED_APPLY_MIN_PHOTO;
|
|
int64 profile_photo_id = 0;
|
|
int32 profile_photo_dc_id = 1;
|
|
string first_name;
|
|
string last_name;
|
|
string username;
|
|
string phone_number;
|
|
int32 bot_info_version = 0;
|
|
|
|
if (user_id == get_service_notifications_user_id()) {
|
|
flags |= USER_FLAG_HAS_PHONE_NUMBER | USER_FLAG_IS_VERIFIED | USER_FLAG_IS_SUPPORT;
|
|
first_name = "Telegram";
|
|
if (G()->is_test_dc()) {
|
|
flags |= USER_FLAG_HAS_LAST_NAME;
|
|
last_name = "Notifications";
|
|
}
|
|
phone_number = "42777";
|
|
profile_photo_id = 3337190045231023;
|
|
} else if (user_id == get_replies_bot_user_id()) {
|
|
flags |= USER_FLAG_HAS_USERNAME | USER_FLAG_IS_BOT;
|
|
if (!G()->is_test_dc()) {
|
|
flags |= USER_FLAG_IS_PRIVATE_BOT;
|
|
}
|
|
first_name = "Replies";
|
|
username = "replies";
|
|
bot_info_version = G()->is_test_dc() ? 1 : 3;
|
|
} else if (user_id == get_anonymous_bot_user_id()) {
|
|
flags |= USER_FLAG_HAS_USERNAME | USER_FLAG_IS_BOT;
|
|
if (!G()->is_test_dc()) {
|
|
flags |= USER_FLAG_IS_PRIVATE_BOT;
|
|
}
|
|
first_name = "Group";
|
|
username = G()->is_test_dc() ? "izgroupbot" : "GroupAnonymousBot";
|
|
bot_info_version = G()->is_test_dc() ? 1 : 3;
|
|
profile_photo_id = 5159307831025969322;
|
|
} else if (user_id == get_channel_bot_user_id()) {
|
|
flags |= USER_FLAG_HAS_USERNAME | USER_FLAG_IS_BOT;
|
|
if (!G()->is_test_dc()) {
|
|
flags |= USER_FLAG_IS_PRIVATE_BOT;
|
|
}
|
|
first_name = G()->is_test_dc() ? "Channels" : "Channel";
|
|
username = G()->is_test_dc() ? "channelsbot" : "Channel_Bot";
|
|
bot_info_version = G()->is_test_dc() ? 1 : 4;
|
|
profile_photo_id = 587627495930570665;
|
|
} else if (user_id == get_service_notifications_user_id()) {
|
|
flags |= USER_FLAG_HAS_USERNAME | USER_FLAG_IS_BOT;
|
|
if (G()->is_test_dc()) {
|
|
first_name = "antispambot";
|
|
username = "tantispambot";
|
|
} else {
|
|
flags |= USER_FLAG_IS_VERIFIED;
|
|
first_name = "Telegram Anti-Spam";
|
|
username = "tgsantispambot";
|
|
profile_photo_id = 5170408289966598902;
|
|
}
|
|
}
|
|
|
|
telegram_api::object_ptr<telegram_api::userProfilePhoto> profile_photo;
|
|
if (!G()->is_test_dc() && profile_photo_id != 0) {
|
|
profile_photo = telegram_api::make_object<telegram_api::userProfilePhoto>(0, false, false, profile_photo_id,
|
|
BufferSlice(), profile_photo_dc_id);
|
|
}
|
|
|
|
auto user = telegram_api::make_object<telegram_api::user>(
|
|
flags, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/,
|
|
false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/,
|
|
false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/,
|
|
false /*ignored*/, false /*ignored*/, false /*ignored*/, 0, false /*ignored*/, false /*ignored*/,
|
|
false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, user_id.get(), 1,
|
|
first_name, string(), username, phone_number, std::move(profile_photo), nullptr, bot_info_version, Auto(),
|
|
string(), string(), nullptr, vector<telegram_api::object_ptr<telegram_api::username>>(), 0, nullptr, nullptr,
|
|
0);
|
|
on_get_user(std::move(user), "get_user_force");
|
|
u = get_user(user_id);
|
|
CHECK(u != nullptr && u->is_received);
|
|
|
|
reload_user(user_id, Promise<Unit>(), "get_user_force");
|
|
}
|
|
return u;
|
|
}
|
|
|
|
UserManager::User *UserManager::get_user_force_impl(UserId user_id, const char *source) {
|
|
if (!user_id.is_valid()) {
|
|
return nullptr;
|
|
}
|
|
|
|
User *u = get_user(user_id);
|
|
if (u != nullptr) {
|
|
return u;
|
|
}
|
|
if (!G()->use_chat_info_database()) {
|
|
return nullptr;
|
|
}
|
|
if (loaded_from_database_users_.count(user_id)) {
|
|
return nullptr;
|
|
}
|
|
|
|
LOG(INFO) << "Trying to load " << user_id << " from database from " << source;
|
|
on_load_user_from_database(user_id, G()->td_db()->get_sqlite_sync_pmc()->get(get_user_database_key(user_id)), true);
|
|
return get_user(user_id);
|
|
}
|
|
|
|
void UserManager::send_get_me_query(Td *td, Promise<Unit> &&promise) {
|
|
vector<telegram_api::object_ptr<telegram_api::InputUser>> users;
|
|
users.push_back(telegram_api::make_object<telegram_api::inputUserSelf>());
|
|
td->create_handler<GetUsersQuery>(std::move(promise))->send(std::move(users));
|
|
}
|
|
|
|
UserId UserManager::get_me(Promise<Unit> &&promise) {
|
|
auto my_id = get_my_id();
|
|
if (!have_user_force(my_id, "get_me")) {
|
|
get_user_queries_.add_query(my_id.get(), std::move(promise), "get_me");
|
|
return UserId();
|
|
}
|
|
|
|
promise.set_value(Unit());
|
|
return my_id;
|
|
}
|
|
|
|
bool UserManager::get_user(UserId user_id, int left_tries, Promise<Unit> &&promise) {
|
|
if (!user_id.is_valid()) {
|
|
promise.set_error(Status::Error(400, "Invalid user identifier"));
|
|
return false;
|
|
}
|
|
|
|
if (user_id == get_service_notifications_user_id() || user_id == get_replies_bot_user_id() ||
|
|
user_id == get_anonymous_bot_user_id() || user_id == get_channel_bot_user_id() ||
|
|
user_id == get_anti_spam_bot_user_id()) {
|
|
get_user_force(user_id, "get_user");
|
|
}
|
|
|
|
if (td_->auth_manager_->is_bot() ? !have_user(user_id) : !have_min_user(user_id)) {
|
|
if (left_tries > 2 && G()->use_chat_info_database()) {
|
|
send_closure_later(actor_id(this), &UserManager::load_user_from_database, nullptr, user_id, std::move(promise));
|
|
return false;
|
|
}
|
|
auto r_input_user = get_input_user(user_id);
|
|
if (left_tries == 1 || r_input_user.is_error()) {
|
|
if (r_input_user.is_error()) {
|
|
promise.set_error(r_input_user.move_as_error());
|
|
} else {
|
|
promise.set_error(Status::Error(400, "User not found"));
|
|
}
|
|
return false;
|
|
}
|
|
|
|
get_user_queries_.add_query(user_id.get(), std::move(promise), "get_user");
|
|
return false;
|
|
}
|
|
|
|
promise.set_value(Unit());
|
|
return true;
|
|
}
|
|
|
|
void UserManager::reload_user(UserId user_id, Promise<Unit> &&promise, const char *source) {
|
|
if (!user_id.is_valid()) {
|
|
return promise.set_error(Status::Error(400, "Invalid user identifier"));
|
|
}
|
|
|
|
have_user_force(user_id, source);
|
|
|
|
TRY_STATUS_PROMISE(promise, get_input_user(user_id));
|
|
|
|
get_user_queries_.add_query(user_id.get(), std::move(promise), source);
|
|
}
|
|
|
|
Result<telegram_api::object_ptr<telegram_api::InputUser>> UserManager::get_input_user(UserId user_id) const {
|
|
if (user_id == get_my_id()) {
|
|
return telegram_api::make_object<telegram_api::inputUserSelf>();
|
|
}
|
|
|
|
const User *u = get_user(user_id);
|
|
if (u == nullptr || u->access_hash == -1 || u->is_min_access_hash) {
|
|
if (td_->auth_manager_->is_bot() && user_id.is_valid()) {
|
|
return telegram_api::make_object<telegram_api::inputUser>(user_id.get(), 0);
|
|
}
|
|
auto it = user_messages_.find(user_id);
|
|
if (it != user_messages_.end()) {
|
|
CHECK(!it->second.empty());
|
|
auto message_full_id = *it->second.begin();
|
|
return telegram_api::make_object<telegram_api::inputUserFromMessage>(
|
|
td_->chat_manager_->get_simple_input_peer(message_full_id.get_dialog_id()),
|
|
message_full_id.get_message_id().get_server_message_id().get(), user_id.get());
|
|
}
|
|
if (u == nullptr) {
|
|
return Status::Error(400, "User not found");
|
|
} else {
|
|
return Status::Error(400, "Have no access to the user");
|
|
}
|
|
}
|
|
|
|
return telegram_api::make_object<telegram_api::inputUser>(user_id.get(), u->access_hash);
|
|
}
|
|
|
|
telegram_api::object_ptr<telegram_api::InputUser> UserManager::get_input_user_force(UserId user_id) const {
|
|
auto r_input_user = get_input_user(user_id);
|
|
if (r_input_user.is_error()) {
|
|
CHECK(user_id.is_valid());
|
|
return telegram_api::make_object<telegram_api::inputUser>(user_id.get(), 0);
|
|
}
|
|
return r_input_user.move_as_ok();
|
|
}
|
|
|
|
bool UserManager::have_input_peer_user(UserId user_id, AccessRights access_rights) const {
|
|
if (user_id == get_my_id()) {
|
|
return true;
|
|
}
|
|
return have_input_peer_user(get_user(user_id), user_id, access_rights);
|
|
}
|
|
|
|
bool UserManager::have_input_peer_user(const User *u, UserId user_id, AccessRights access_rights) const {
|
|
if (u == nullptr || u->access_hash == -1 || u->is_min_access_hash) {
|
|
if (u == nullptr) {
|
|
LOG(DEBUG) << "Have no user";
|
|
} else {
|
|
LOG(DEBUG) << "Have user without access hash";
|
|
}
|
|
if (td_->auth_manager_->is_bot() && user_id.is_valid()) {
|
|
return true;
|
|
}
|
|
if (user_messages_.count(user_id) != 0) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
if (access_rights == AccessRights::Know) {
|
|
return true;
|
|
}
|
|
if (access_rights == AccessRights::Read) {
|
|
return true;
|
|
}
|
|
if (u->is_deleted) {
|
|
LOG(DEBUG) << "Have a deleted user";
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
telegram_api::object_ptr<telegram_api::InputPeer> UserManager::get_input_peer_user(UserId user_id,
|
|
AccessRights access_rights) const {
|
|
if (user_id == get_my_id()) {
|
|
return telegram_api::make_object<telegram_api::inputPeerSelf>();
|
|
}
|
|
const User *u = get_user(user_id);
|
|
if (!have_input_peer_user(u, user_id, access_rights)) {
|
|
return nullptr;
|
|
}
|
|
if (u == nullptr || u->access_hash == -1 || u->is_min_access_hash) {
|
|
if (td_->auth_manager_->is_bot() && user_id.is_valid()) {
|
|
return telegram_api::make_object<telegram_api::inputPeerUser>(user_id.get(), 0);
|
|
}
|
|
auto it = user_messages_.find(user_id);
|
|
CHECK(it != user_messages_.end());
|
|
CHECK(!it->second.empty());
|
|
auto message_full_id = *it->second.begin();
|
|
return telegram_api::make_object<telegram_api::inputPeerUserFromMessage>(
|
|
td_->chat_manager_->get_simple_input_peer(message_full_id.get_dialog_id()),
|
|
message_full_id.get_message_id().get_server_message_id().get(), user_id.get());
|
|
}
|
|
|
|
return telegram_api::make_object<telegram_api::inputPeerUser>(user_id.get(), u->access_hash);
|
|
}
|
|
|
|
bool UserManager::have_input_encrypted_peer(SecretChatId secret_chat_id, AccessRights access_rights) const {
|
|
return have_input_encrypted_peer(get_secret_chat(secret_chat_id), access_rights);
|
|
}
|
|
|
|
bool UserManager::have_input_encrypted_peer(const SecretChat *secret_chat, AccessRights access_rights) {
|
|
if (secret_chat == nullptr) {
|
|
LOG(DEBUG) << "Have no secret chat";
|
|
return false;
|
|
}
|
|
if (access_rights == AccessRights::Know) {
|
|
return true;
|
|
}
|
|
if (access_rights == AccessRights::Read) {
|
|
return true;
|
|
}
|
|
return secret_chat->state == SecretChatState::Active;
|
|
}
|
|
|
|
telegram_api::object_ptr<telegram_api::inputEncryptedChat> UserManager::get_input_encrypted_chat(
|
|
SecretChatId secret_chat_id, AccessRights access_rights) const {
|
|
auto sc = get_secret_chat(secret_chat_id);
|
|
if (!have_input_encrypted_peer(sc, access_rights)) {
|
|
return nullptr;
|
|
}
|
|
|
|
return telegram_api::make_object<telegram_api::inputEncryptedChat>(secret_chat_id.get(), sc->access_hash);
|
|
}
|
|
|
|
bool UserManager::is_user_contact(UserId user_id, bool is_mutual) const {
|
|
return is_user_contact(get_user(user_id), user_id, is_mutual);
|
|
}
|
|
|
|
bool UserManager::is_user_contact(const User *u, UserId user_id, bool is_mutual) const {
|
|
return u != nullptr && (is_mutual ? u->is_mutual_contact : u->is_contact) && user_id != get_my_id();
|
|
}
|
|
|
|
bool UserManager::is_user_premium(UserId user_id) const {
|
|
return is_user_premium(get_user(user_id));
|
|
}
|
|
|
|
bool UserManager::is_user_premium(const User *u) {
|
|
return u != nullptr && u->is_premium;
|
|
}
|
|
|
|
bool UserManager::is_user_deleted(UserId user_id) const {
|
|
return is_user_deleted(get_user(user_id));
|
|
}
|
|
|
|
bool UserManager::is_user_deleted(const User *u) {
|
|
return u == nullptr || u->is_deleted;
|
|
}
|
|
|
|
bool UserManager::is_user_support(UserId user_id) const {
|
|
return is_user_support(get_user(user_id));
|
|
}
|
|
|
|
bool UserManager::is_user_support(const User *u) {
|
|
return u != nullptr && !u->is_deleted && u->is_support;
|
|
}
|
|
|
|
bool UserManager::is_user_bot(UserId user_id) const {
|
|
return is_user_bot(get_user(user_id));
|
|
}
|
|
|
|
bool UserManager::is_user_bot(const User *u) {
|
|
return u != nullptr && !u->is_deleted && u->is_bot;
|
|
}
|
|
|
|
Result<UserManager::BotData> UserManager::get_bot_data(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return Status::Error(400, "Bot not found");
|
|
}
|
|
if (!u->is_bot) {
|
|
return Status::Error(400, "User is not a bot");
|
|
}
|
|
if (u->is_deleted) {
|
|
return Status::Error(400, "Bot is deleted");
|
|
}
|
|
if (!u->is_received) {
|
|
return Status::Error(400, "Bot is inaccessible");
|
|
}
|
|
|
|
BotData bot_data;
|
|
bot_data.username = u->usernames.get_first_username();
|
|
bot_data.can_be_edited = u->can_be_edited_bot;
|
|
bot_data.can_join_groups = u->can_join_groups;
|
|
bot_data.can_read_all_group_messages = u->can_read_all_group_messages;
|
|
bot_data.has_main_app = u->has_main_app;
|
|
bot_data.is_inline = u->is_inline_bot;
|
|
bot_data.is_business = u->is_business_bot;
|
|
bot_data.need_location = u->need_location_bot;
|
|
bot_data.can_be_added_to_attach_menu = u->can_be_added_to_attach_menu;
|
|
return bot_data;
|
|
}
|
|
|
|
bool UserManager::is_user_online(UserId user_id, int32 tolerance, int32 unix_time) const {
|
|
if (unix_time <= 0) {
|
|
unix_time = G()->unix_time();
|
|
}
|
|
int32 was_online = get_user_was_online(get_user(user_id), user_id, unix_time);
|
|
return was_online > unix_time - tolerance;
|
|
}
|
|
|
|
int32 UserManager::get_user_was_online(UserId user_id, int32 unix_time) const {
|
|
if (unix_time <= 0) {
|
|
unix_time = G()->unix_time();
|
|
}
|
|
return get_user_was_online(get_user(user_id), user_id, unix_time);
|
|
}
|
|
|
|
int32 UserManager::get_user_was_online(const User *u, UserId user_id, int32 unix_time) const {
|
|
if (u == nullptr || u->is_deleted) {
|
|
return 0;
|
|
}
|
|
|
|
int32 was_online = u->was_online;
|
|
if (user_id == get_my_id()) {
|
|
if (my_was_online_local_ != 0) {
|
|
was_online = my_was_online_local_;
|
|
}
|
|
} else {
|
|
if (u->local_was_online > 0 && u->local_was_online > was_online && u->local_was_online > unix_time) {
|
|
was_online = u->local_was_online;
|
|
}
|
|
}
|
|
return was_online;
|
|
}
|
|
|
|
bool UserManager::is_user_status_exact(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
return u != nullptr && !u->is_deleted && !u->is_bot && u->was_online > 0;
|
|
}
|
|
|
|
bool UserManager::is_user_received_from_server(UserId user_id) const {
|
|
const auto *u = get_user(user_id);
|
|
return u != nullptr && u->is_received_from_server;
|
|
}
|
|
|
|
bool UserManager::can_report_user(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
return u != nullptr && !u->is_deleted && !u->is_support &&
|
|
(u->is_bot || td_->people_nearby_manager_->is_user_nearby(user_id));
|
|
}
|
|
|
|
const DialogPhoto *UserManager::get_user_dialog_photo(UserId user_id) {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return nullptr;
|
|
}
|
|
|
|
apply_pending_user_photo(u, user_id);
|
|
return &u->photo;
|
|
}
|
|
|
|
const DialogPhoto *UserManager::get_secret_chat_dialog_photo(SecretChatId secret_chat_id) {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return nullptr;
|
|
}
|
|
return get_user_dialog_photo(c->user_id);
|
|
}
|
|
|
|
int32 UserManager::get_user_accent_color_id_object(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr || !u->accent_color_id.is_valid()) {
|
|
return td_->theme_manager_->get_accent_color_id_object(AccentColorId(user_id));
|
|
}
|
|
|
|
return td_->theme_manager_->get_accent_color_id_object(u->accent_color_id, AccentColorId(user_id));
|
|
}
|
|
|
|
int32 UserManager::get_secret_chat_accent_color_id_object(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return 5;
|
|
}
|
|
return get_user_accent_color_id_object(c->user_id);
|
|
}
|
|
|
|
CustomEmojiId UserManager::get_user_background_custom_emoji_id(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return CustomEmojiId();
|
|
}
|
|
|
|
return u->background_custom_emoji_id;
|
|
}
|
|
|
|
CustomEmojiId UserManager::get_secret_chat_background_custom_emoji_id(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return CustomEmojiId();
|
|
}
|
|
return get_user_background_custom_emoji_id(c->user_id);
|
|
}
|
|
|
|
int32 UserManager::get_user_profile_accent_color_id_object(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return -1;
|
|
}
|
|
|
|
return td_->theme_manager_->get_profile_accent_color_id_object(u->profile_accent_color_id);
|
|
}
|
|
|
|
int32 UserManager::get_secret_chat_profile_accent_color_id_object(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return -1;
|
|
}
|
|
return get_user_profile_accent_color_id_object(c->user_id);
|
|
}
|
|
|
|
CustomEmojiId UserManager::get_user_profile_background_custom_emoji_id(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return CustomEmojiId();
|
|
}
|
|
|
|
return u->profile_background_custom_emoji_id;
|
|
}
|
|
|
|
CustomEmojiId UserManager::get_secret_chat_profile_background_custom_emoji_id(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return CustomEmojiId();
|
|
}
|
|
return get_user_profile_background_custom_emoji_id(c->user_id);
|
|
}
|
|
|
|
string UserManager::get_user_title(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return string();
|
|
}
|
|
if (u->last_name.empty()) {
|
|
return u->first_name;
|
|
}
|
|
if (u->first_name.empty()) {
|
|
return u->last_name;
|
|
}
|
|
return PSTRING() << u->first_name << ' ' << u->last_name;
|
|
}
|
|
|
|
string UserManager::get_secret_chat_title(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return string();
|
|
}
|
|
return get_user_title(c->user_id);
|
|
}
|
|
|
|
RestrictedRights UserManager::get_user_default_permissions(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr || user_id == get_replies_bot_user_id()) {
|
|
return RestrictedRights(false, false, false, false, false, false, false, false, false, false, false, false, false,
|
|
false, false, u != nullptr, false, ChannelType::Unknown);
|
|
}
|
|
return RestrictedRights(true, true, true, true, true, true, true, true, true, true, true, true, true, false, false,
|
|
true, false, ChannelType::Unknown);
|
|
}
|
|
|
|
RestrictedRights UserManager::get_secret_chat_default_permissions(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return RestrictedRights(false, false, false, false, false, false, false, false, false, false, false, false, false,
|
|
false, false, false, false, ChannelType::Unknown);
|
|
}
|
|
return RestrictedRights(true, true, true, true, true, true, true, true, true, true, true, true, true, false, false,
|
|
false, false, ChannelType::Unknown);
|
|
}
|
|
|
|
td_api::object_ptr<td_api::emojiStatus> UserManager::get_user_emoji_status_object(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return nullptr;
|
|
}
|
|
return u->last_sent_emoji_status.get_emoji_status_object();
|
|
}
|
|
|
|
td_api::object_ptr<td_api::emojiStatus> UserManager::get_secret_chat_emoji_status_object(
|
|
SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return nullptr;
|
|
}
|
|
return get_user_emoji_status_object(c->user_id);
|
|
}
|
|
|
|
bool UserManager::get_user_stories_hidden(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return false;
|
|
}
|
|
return u->stories_hidden;
|
|
}
|
|
|
|
bool UserManager::can_poll_user_active_stories(UserId user_id) const {
|
|
const User *u = get_user(user_id);
|
|
return need_poll_user_active_stories(u, user_id) && Time::now() >= u->max_active_story_id_next_reload_time;
|
|
}
|
|
|
|
bool UserManager::need_poll_user_active_stories(const User *u, UserId user_id) const {
|
|
return u != nullptr && user_id != get_my_id() && !is_user_contact(u, user_id, false) && !is_user_bot(u) &&
|
|
!is_user_support(u) && !is_user_deleted(u) && u->was_online != 0;
|
|
}
|
|
|
|
string UserManager::get_user_about(UserId user_id) {
|
|
auto user_full = get_user_full_force(user_id, "get_user_about");
|
|
if (user_full != nullptr) {
|
|
return user_full->about;
|
|
}
|
|
return string();
|
|
}
|
|
|
|
string UserManager::get_secret_chat_about(SecretChatId secret_chat_id) {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return string();
|
|
}
|
|
return get_user_about(c->user_id);
|
|
}
|
|
|
|
string UserManager::get_user_private_forward_name(UserId user_id) {
|
|
auto user_full = get_user_full_force(user_id, "get_user_private_forward_name");
|
|
if (user_full != nullptr) {
|
|
return user_full->private_forward_name;
|
|
}
|
|
return string();
|
|
}
|
|
|
|
bool UserManager::get_user_voice_messages_forbidden(UserId user_id) const {
|
|
if (!is_user_premium(user_id)) {
|
|
return false;
|
|
}
|
|
auto user_full = get_user_full(user_id);
|
|
if (user_full != nullptr) {
|
|
return user_full->voice_messages_forbidden;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool UserManager::get_user_read_dates_private(UserId user_id) {
|
|
auto user_full = get_user_full_force(user_id, "get_user_read_dates_private");
|
|
if (user_full != nullptr) {
|
|
return user_full->read_dates_private;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
string UserManager::get_user_search_text(UserId user_id) const {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return string();
|
|
}
|
|
return get_user_search_text(u);
|
|
}
|
|
|
|
string UserManager::get_user_search_text(const User *u) {
|
|
CHECK(u != nullptr);
|
|
return PSTRING() << u->first_name << ' ' << u->last_name << ' ' << implode(u->usernames.get_active_usernames());
|
|
}
|
|
|
|
void UserManager::for_each_secret_chat_with_user(UserId user_id, const std::function<void(SecretChatId)> &f) {
|
|
auto it = secret_chats_with_user_.find(user_id);
|
|
if (it != secret_chats_with_user_.end()) {
|
|
for (auto secret_chat_id : it->second) {
|
|
f(secret_chat_id);
|
|
}
|
|
}
|
|
}
|
|
|
|
string UserManager::get_user_first_username(UserId user_id) const {
|
|
if (!user_id.is_valid()) {
|
|
return string();
|
|
}
|
|
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return string();
|
|
}
|
|
return u->usernames.get_first_username();
|
|
}
|
|
|
|
int32 UserManager::get_secret_chat_date(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return 0;
|
|
}
|
|
return c->date;
|
|
}
|
|
|
|
int32 UserManager::get_secret_chat_ttl(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return 0;
|
|
}
|
|
return c->ttl;
|
|
}
|
|
|
|
UserId UserManager::get_secret_chat_user_id(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return UserId();
|
|
}
|
|
return c->user_id;
|
|
}
|
|
|
|
bool UserManager::get_secret_chat_is_outbound(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return false;
|
|
}
|
|
return c->is_outbound;
|
|
}
|
|
|
|
SecretChatState UserManager::get_secret_chat_state(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return SecretChatState::Unknown;
|
|
}
|
|
return c->state;
|
|
}
|
|
|
|
int32 UserManager::get_secret_chat_layer(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return 0;
|
|
}
|
|
return c->layer;
|
|
}
|
|
|
|
FolderId UserManager::get_secret_chat_initial_folder_id(SecretChatId secret_chat_id) const {
|
|
auto c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
return FolderId::main();
|
|
}
|
|
return c->initial_folder_id;
|
|
}
|
|
|
|
vector<BotCommands> UserManager::get_bot_commands(vector<telegram_api::object_ptr<telegram_api::botInfo>> &&bot_infos,
|
|
const vector<DialogParticipant> *participants) {
|
|
vector<BotCommands> result;
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return result;
|
|
}
|
|
for (auto &bot_info : bot_infos) {
|
|
if (bot_info->commands_.empty()) {
|
|
continue;
|
|
}
|
|
|
|
auto user_id = UserId(bot_info->user_id_);
|
|
const User *u = get_user_force(user_id, "get_bot_commands");
|
|
if (u == nullptr) {
|
|
LOG(ERROR) << "Receive unknown " << user_id;
|
|
continue;
|
|
}
|
|
if (!is_user_bot(u)) {
|
|
if (!is_user_deleted(u)) {
|
|
LOG(ERROR) << "Receive non-bot " << user_id;
|
|
}
|
|
continue;
|
|
}
|
|
if (participants != nullptr) {
|
|
bool is_participant = false;
|
|
for (auto &participant : *participants) {
|
|
if (participant.dialog_id_ == DialogId(user_id)) {
|
|
is_participant = true;
|
|
break;
|
|
}
|
|
}
|
|
if (!is_participant) {
|
|
LOG(ERROR) << "Skip commands of non-member bot " << user_id;
|
|
continue;
|
|
}
|
|
}
|
|
result.emplace_back(user_id, std::move(bot_info->commands_));
|
|
}
|
|
return result;
|
|
}
|
|
|
|
void UserManager::set_name(const string &first_name, const string &last_name, Promise<Unit> &&promise) {
|
|
auto new_first_name = clean_name(first_name, MAX_NAME_LENGTH);
|
|
auto new_last_name = clean_name(last_name, MAX_NAME_LENGTH);
|
|
if (new_first_name.empty()) {
|
|
return promise.set_error(Status::Error(400, "First name must be non-empty"));
|
|
}
|
|
|
|
const User *u = get_user(get_my_id());
|
|
int32 flags = 0;
|
|
// TODO we can already send request for changing first_name and last_name and wanting to set initial values
|
|
// TODO need to be rewritten using invoke after and cancelling previous request
|
|
if (u == nullptr || u->first_name != new_first_name) {
|
|
flags |= ACCOUNT_UPDATE_FIRST_NAME;
|
|
}
|
|
if (u == nullptr || u->last_name != new_last_name) {
|
|
flags |= ACCOUNT_UPDATE_LAST_NAME;
|
|
}
|
|
if (flags == 0) {
|
|
return promise.set_value(Unit());
|
|
}
|
|
|
|
td_->create_handler<UpdateProfileQuery>(std::move(promise))->send(flags, new_first_name, new_last_name, "");
|
|
}
|
|
|
|
void UserManager::set_bio(const string &bio, Promise<Unit> &&promise) {
|
|
auto max_bio_length = static_cast<size_t>(td_->option_manager_->get_option_integer("bio_length_max"));
|
|
auto new_bio = strip_empty_characters(bio, max_bio_length);
|
|
for (auto &c : new_bio) {
|
|
if (c == '\n') {
|
|
c = ' ';
|
|
}
|
|
}
|
|
|
|
const UserFull *user_full = get_user_full(get_my_id());
|
|
int32 flags = 0;
|
|
// TODO we can already send request for changing bio and wanting to set initial values
|
|
// TODO need to be rewritten using invoke after and cancelling previous request
|
|
if (user_full == nullptr || user_full->about != new_bio) {
|
|
flags |= ACCOUNT_UPDATE_ABOUT;
|
|
}
|
|
if (flags == 0) {
|
|
return promise.set_value(Unit());
|
|
}
|
|
|
|
td_->create_handler<UpdateProfileQuery>(std::move(promise))->send(flags, "", "", new_bio);
|
|
}
|
|
|
|
void UserManager::on_update_profile_success(int32 flags, const string &first_name, const string &last_name,
|
|
const string &about) {
|
|
CHECK(flags != 0);
|
|
|
|
auto my_user_id = get_my_id();
|
|
const User *u = get_user(my_user_id);
|
|
if (u == nullptr) {
|
|
LOG(ERROR) << "Doesn't receive info about me during update profile";
|
|
return;
|
|
}
|
|
LOG_IF(ERROR, (flags & ACCOUNT_UPDATE_FIRST_NAME) != 0 && u->first_name != first_name)
|
|
<< "Wrong first name \"" << u->first_name << "\", expected \"" << first_name << '"';
|
|
LOG_IF(ERROR, (flags & ACCOUNT_UPDATE_LAST_NAME) != 0 && u->last_name != last_name)
|
|
<< "Wrong last name \"" << u->last_name << "\", expected \"" << last_name << '"';
|
|
|
|
if ((flags & ACCOUNT_UPDATE_ABOUT) != 0) {
|
|
UserFull *user_full = get_user_full_force(my_user_id, "on_update_profile_success");
|
|
if (user_full != nullptr) {
|
|
user_full->about = about;
|
|
user_full->is_changed = true;
|
|
update_user_full(user_full, my_user_id, "on_update_profile_success");
|
|
td_->group_call_manager_->on_update_dialog_about(DialogId(my_user_id), user_full->about, true);
|
|
}
|
|
}
|
|
}
|
|
|
|
FileId UserManager::get_profile_photo_file_id(int64 photo_id) const {
|
|
auto it = my_photo_file_id_.find(photo_id);
|
|
if (it == my_photo_file_id_.end()) {
|
|
return FileId();
|
|
}
|
|
return it->second;
|
|
}
|
|
|
|
void UserManager::set_bot_profile_photo(UserId bot_user_id,
|
|
const td_api::object_ptr<td_api::InputChatPhoto> &input_photo,
|
|
Promise<Unit> &&promise) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
if (bot_user_id != UserId() && bot_user_id != get_my_id()) {
|
|
return promise.set_error(Status::Error(400, "Invalid bot user identifier specified"));
|
|
}
|
|
bot_user_id = get_my_id();
|
|
} else {
|
|
TRY_RESULT_PROMISE(promise, bot_data, get_bot_data(bot_user_id));
|
|
if (!bot_data.can_be_edited) {
|
|
return promise.set_error(Status::Error(400, "The bot can't be edited"));
|
|
}
|
|
}
|
|
if (input_photo == nullptr) {
|
|
td_->create_handler<UpdateProfilePhotoQuery>(std::move(promise))
|
|
->send(bot_user_id, FileId(), 0, false, telegram_api::make_object<telegram_api::inputPhotoEmpty>());
|
|
return;
|
|
}
|
|
set_profile_photo_impl(bot_user_id, input_photo, false, false, std::move(promise));
|
|
}
|
|
|
|
void UserManager::set_profile_photo(const td_api::object_ptr<td_api::InputChatPhoto> &input_photo, bool is_fallback,
|
|
Promise<Unit> &&promise) {
|
|
set_profile_photo_impl(get_my_id(), input_photo, is_fallback, false, std::move(promise));
|
|
}
|
|
|
|
void UserManager::set_profile_photo_impl(UserId user_id, const td_api::object_ptr<td_api::InputChatPhoto> &input_photo,
|
|
bool is_fallback, bool only_suggest, Promise<Unit> &&promise) {
|
|
if (input_photo == nullptr) {
|
|
return promise.set_error(Status::Error(400, "New profile photo must be non-empty"));
|
|
}
|
|
|
|
const td_api::object_ptr<td_api::InputFile> *input_file = nullptr;
|
|
double main_frame_timestamp = 0.0;
|
|
bool is_animation = false;
|
|
switch (input_photo->get_id()) {
|
|
case td_api::inputChatPhotoPrevious::ID: {
|
|
if (user_id != get_my_id() || td_->auth_manager_->is_bot()) {
|
|
return promise.set_error(Status::Error(400, "Can't use inputChatPhotoPrevious"));
|
|
}
|
|
auto photo = static_cast<const td_api::inputChatPhotoPrevious *>(input_photo.get());
|
|
auto photo_id = photo->chat_photo_id_;
|
|
auto *u = get_user(user_id);
|
|
if (u != nullptr && u->photo.id > 0 && photo_id == u->photo.id) {
|
|
// it is possible that u->photo.is_fallback != is_fallback, so we need to set the photo anyway
|
|
// return promise.set_value(Unit());
|
|
}
|
|
|
|
auto file_id = get_profile_photo_file_id(photo_id);
|
|
if (!file_id.is_valid()) {
|
|
return promise.set_error(Status::Error(400, "Unknown profile photo ID specified"));
|
|
}
|
|
return send_update_profile_photo_query(user_id,
|
|
td_->file_manager_->dup_file_id(file_id, "set_profile_photo_impl"),
|
|
photo_id, is_fallback, std::move(promise));
|
|
}
|
|
case td_api::inputChatPhotoStatic::ID: {
|
|
auto photo = static_cast<const td_api::inputChatPhotoStatic *>(input_photo.get());
|
|
input_file = &photo->photo_;
|
|
break;
|
|
}
|
|
case td_api::inputChatPhotoAnimation::ID: {
|
|
auto photo = static_cast<const td_api::inputChatPhotoAnimation *>(input_photo.get());
|
|
input_file = &photo->animation_;
|
|
main_frame_timestamp = photo->main_frame_timestamp_;
|
|
is_animation = true;
|
|
break;
|
|
}
|
|
case td_api::inputChatPhotoSticker::ID: {
|
|
auto photo = static_cast<const td_api::inputChatPhotoSticker *>(input_photo.get());
|
|
TRY_RESULT_PROMISE(promise, sticker_photo_size, StickerPhotoSize::get_sticker_photo_size(td_, photo->sticker_));
|
|
|
|
td_->create_handler<UploadProfilePhotoQuery>(std::move(promise))
|
|
->send(user_id, std::move(sticker_photo_size), is_fallback, only_suggest);
|
|
return;
|
|
}
|
|
default:
|
|
UNREACHABLE();
|
|
break;
|
|
}
|
|
|
|
const double MAX_ANIMATION_DURATION = 10.0;
|
|
if (main_frame_timestamp < 0.0 || main_frame_timestamp > MAX_ANIMATION_DURATION) {
|
|
return promise.set_error(Status::Error(400, "Wrong main frame timestamp specified"));
|
|
}
|
|
|
|
auto file_type = is_animation ? FileType::Animation : FileType::Photo;
|
|
TRY_RESULT_PROMISE(promise, file_id,
|
|
td_->file_manager_->get_input_file_id(file_type, *input_file, DialogId(user_id), false, false));
|
|
CHECK(file_id.is_valid());
|
|
|
|
upload_profile_photo(user_id, td_->file_manager_->dup_file_id(file_id, "set_profile_photo_impl"), is_fallback,
|
|
only_suggest, is_animation, main_frame_timestamp, std::move(promise));
|
|
}
|
|
|
|
void UserManager::set_user_profile_photo(UserId user_id, const td_api::object_ptr<td_api::InputChatPhoto> &input_photo,
|
|
bool only_suggest, Promise<Unit> &&promise) {
|
|
TRY_RESULT_PROMISE(promise, input_user, get_input_user(user_id));
|
|
if (!only_suggest && !is_user_contact(user_id)) {
|
|
return promise.set_error(Status::Error(400, "User isn't a contact"));
|
|
}
|
|
if (user_id == get_my_id()) {
|
|
return promise.set_error(Status::Error(400, "Can't set personal or suggest photo to self"));
|
|
}
|
|
if (is_user_bot(user_id)) {
|
|
return promise.set_error(Status::Error(400, "Can't set personal or suggest photo to bots"));
|
|
}
|
|
if (input_photo == nullptr) {
|
|
td_->create_handler<DeleteContactProfilePhotoQuery>(std::move(promise))->send(user_id, std::move(input_user));
|
|
return;
|
|
}
|
|
|
|
set_profile_photo_impl(user_id, input_photo, false, only_suggest, std::move(promise));
|
|
}
|
|
|
|
void UserManager::send_update_profile_photo_query(UserId user_id, FileId file_id, int64 old_photo_id, bool is_fallback,
|
|
Promise<Unit> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
FileView file_view = td_->file_manager_->get_file_view(file_id);
|
|
td_->create_handler<UpdateProfilePhotoQuery>(std::move(promise))
|
|
->send(user_id, file_id, old_photo_id, is_fallback, file_view.main_remote_location().as_input_photo());
|
|
}
|
|
|
|
void UserManager::upload_profile_photo(UserId user_id, FileId file_id, bool is_fallback, bool only_suggest,
|
|
bool is_animation, double main_frame_timestamp, Promise<Unit> &&promise,
|
|
int reupload_count, vector<int> bad_parts) {
|
|
CHECK(file_id.is_valid());
|
|
bool is_inserted =
|
|
uploaded_profile_photos_
|
|
.emplace(file_id, UploadedProfilePhoto{user_id, is_fallback, only_suggest, main_frame_timestamp, is_animation,
|
|
reupload_count, std::move(promise)})
|
|
.second;
|
|
CHECK(is_inserted);
|
|
LOG(INFO) << "Ask to upload " << (is_animation ? "animated" : "static") << " profile photo " << file_id
|
|
<< " for user " << user_id << " with bad parts " << bad_parts;
|
|
// TODO use force_reupload if reupload_count >= 1, replace reupload_count with is_reupload
|
|
td_->file_manager_->resume_upload(file_id, std::move(bad_parts), upload_profile_photo_callback_, 32, 0);
|
|
}
|
|
|
|
void UserManager::on_upload_profile_photo(FileId file_id,
|
|
telegram_api::object_ptr<telegram_api::InputFile> input_file) {
|
|
auto it = uploaded_profile_photos_.find(file_id);
|
|
CHECK(it != uploaded_profile_photos_.end());
|
|
|
|
UserId user_id = it->second.user_id;
|
|
bool is_fallback = it->second.is_fallback;
|
|
bool only_suggest = it->second.only_suggest;
|
|
double main_frame_timestamp = it->second.main_frame_timestamp;
|
|
bool is_animation = it->second.is_animation;
|
|
int32 reupload_count = it->second.reupload_count;
|
|
auto promise = std::move(it->second.promise);
|
|
|
|
uploaded_profile_photos_.erase(it);
|
|
|
|
LOG(INFO) << "Uploaded " << (is_animation ? "animated" : "static") << " profile photo " << file_id << " for "
|
|
<< user_id << " with reupload_count = " << reupload_count;
|
|
FileView file_view = td_->file_manager_->get_file_view(file_id);
|
|
if (file_view.has_remote_location() && input_file == nullptr) {
|
|
if (file_view.main_remote_location().is_web()) {
|
|
return promise.set_error(Status::Error(400, "Can't use web photo as profile photo"));
|
|
}
|
|
if (reupload_count == 3) { // upload, ForceReupload repair file reference, reupload
|
|
return promise.set_error(Status::Error(400, "Failed to reupload the file"));
|
|
}
|
|
|
|
// delete file reference and forcely reupload the file
|
|
if (is_animation) {
|
|
CHECK(file_view.get_type() == FileType::Animation);
|
|
LOG_CHECK(file_view.main_remote_location().is_common()) << file_view.main_remote_location();
|
|
} else {
|
|
CHECK(file_view.get_type() == FileType::Photo);
|
|
LOG_CHECK(file_view.main_remote_location().is_photo()) << file_view.main_remote_location();
|
|
}
|
|
auto file_reference =
|
|
is_animation ? FileManager::extract_file_reference(file_view.main_remote_location().as_input_document())
|
|
: FileManager::extract_file_reference(file_view.main_remote_location().as_input_photo());
|
|
td_->file_manager_->delete_file_reference(file_id, file_reference);
|
|
upload_profile_photo(user_id, file_id, is_fallback, only_suggest, is_animation, main_frame_timestamp,
|
|
std::move(promise), reupload_count + 1, {-1});
|
|
return;
|
|
}
|
|
CHECK(input_file != nullptr);
|
|
|
|
td_->create_handler<UploadProfilePhotoQuery>(std::move(promise))
|
|
->send(user_id, file_id, std::move(input_file), is_fallback, only_suggest, is_animation, main_frame_timestamp);
|
|
}
|
|
|
|
void UserManager::on_upload_profile_photo_error(FileId file_id, Status status) {
|
|
LOG(INFO) << "File " << file_id << " has upload error " << status;
|
|
CHECK(status.is_error());
|
|
|
|
auto it = uploaded_profile_photos_.find(file_id);
|
|
CHECK(it != uploaded_profile_photos_.end());
|
|
|
|
auto promise = std::move(it->second.promise);
|
|
|
|
uploaded_profile_photos_.erase(it);
|
|
|
|
promise.set_error(std::move(status)); // TODO check that status has valid error code
|
|
}
|
|
|
|
void UserManager::on_set_profile_photo(UserId user_id, telegram_api::object_ptr<telegram_api::photos_photo> &&photo,
|
|
bool is_fallback, int64 old_photo_id, Promise<Unit> &&promise) {
|
|
LOG(INFO) << "Changed profile photo to " << to_string(photo);
|
|
|
|
bool is_bot = is_user_bot(user_id);
|
|
bool is_my = (user_id == get_my_id());
|
|
if (is_my && !is_fallback) {
|
|
delete_my_profile_photo_from_cache(old_photo_id);
|
|
}
|
|
bool have_user = false;
|
|
for (const auto &user : photo->users_) {
|
|
if (get_user_id(user) == user_id) {
|
|
have_user = true;
|
|
}
|
|
}
|
|
on_get_users(std::move(photo->users_), "on_set_profile_photo");
|
|
if (!is_bot) {
|
|
add_set_profile_photo_to_cache(user_id, get_photo(td_, std::move(photo->photo_), DialogId(user_id)), is_fallback);
|
|
}
|
|
if (have_user) {
|
|
promise.set_value(Unit());
|
|
} else {
|
|
reload_user(user_id, std::move(promise), "on_set_profile_photo");
|
|
}
|
|
}
|
|
|
|
void UserManager::add_set_profile_photo_to_cache(UserId user_id, Photo &&photo, bool is_fallback) {
|
|
// we have subsequence of user photos in user_photos_
|
|
// ProfilePhoto in User and Photo in UserFull
|
|
|
|
User *u = get_user_force(user_id, "add_set_profile_photo_to_cache");
|
|
if (u == nullptr) {
|
|
return;
|
|
}
|
|
|
|
LOG(INFO) << "Add profile photo " << photo.id.get() << " to cache";
|
|
|
|
bool is_me = user_id == get_my_id();
|
|
|
|
// update photo list
|
|
auto user_photos = user_photos_.get_pointer(user_id);
|
|
if (is_me && !is_fallback && user_photos != nullptr && user_photos->count != -1 && !photo.is_empty()) {
|
|
if (user_photos->offset == 0) {
|
|
if (user_photos->photos.empty() || user_photos->photos[0].id.get() != photo.id.get()) {
|
|
user_photos->photos.insert(user_photos->photos.begin(), photo);
|
|
user_photos->count++;
|
|
register_user_photo(u, user_id, user_photos->photos[0]);
|
|
}
|
|
} else {
|
|
user_photos->count++;
|
|
user_photos->offset++;
|
|
}
|
|
}
|
|
|
|
// update ProfilePhoto in User
|
|
if ((!is_fallback || u->photo.id == 0) && !photo.is_empty()) {
|
|
do_update_user_photo(u, user_id, as_profile_photo(td_->file_manager_.get(), user_id, u->access_hash, photo, !is_me),
|
|
false, "add_set_profile_photo_to_cache");
|
|
update_user(u, user_id);
|
|
}
|
|
|
|
// update Photo in UserFull
|
|
auto user_full = get_user_full_force(user_id, "add_set_profile_photo_to_cache");
|
|
if (user_full != nullptr) {
|
|
Photo *current_photo = nullptr;
|
|
// don't update the changed photo if other photos aren't known to avoid having only some photos known
|
|
bool need_apply = get_user_full_profile_photo_id(user_full) > 0;
|
|
if (!is_me) {
|
|
current_photo = &user_full->personal_photo;
|
|
if (photo.is_empty()) {
|
|
// always can apply empty personal photo
|
|
need_apply = true;
|
|
}
|
|
} else if (!is_fallback) {
|
|
current_photo = &user_full->photo;
|
|
if (photo.is_empty()) {
|
|
// never can apply empty photo
|
|
need_apply = false;
|
|
}
|
|
} else {
|
|
current_photo = &user_full->fallback_photo;
|
|
if (photo.is_empty()) {
|
|
// always can apply empty fallback photo
|
|
need_apply = true;
|
|
}
|
|
}
|
|
if (*current_photo != photo && need_apply) {
|
|
LOG(INFO) << "Update full photo of " << user_id << " to " << photo;
|
|
*current_photo = photo;
|
|
user_full->is_changed = true;
|
|
if (is_me && !photo.is_empty()) {
|
|
if (!is_fallback) {
|
|
register_user_photo(u, user_id, photo);
|
|
} else {
|
|
register_suggested_profile_photo(photo);
|
|
}
|
|
}
|
|
drop_user_full_photos(user_full, user_id, u->photo.id, "add_set_profile_photo_to_cache"); // just in case
|
|
}
|
|
if (user_full->expires_at > 0.0) {
|
|
user_full->expires_at = 0.0;
|
|
user_full->need_save_to_database = true;
|
|
}
|
|
update_user_full(user_full, user_id, "add_set_profile_photo_to_cache");
|
|
reload_user_full(user_id, Auto(), "add_set_profile_photo_to_cache");
|
|
}
|
|
}
|
|
|
|
bool UserManager::delete_my_profile_photo_from_cache(int64 profile_photo_id) {
|
|
if (profile_photo_id == 0 || profile_photo_id == -2) {
|
|
return false;
|
|
}
|
|
|
|
// we have subsequence of user photos in user_photos_
|
|
// ProfilePhoto in User and Photo in UserFull
|
|
|
|
LOG(INFO) << "Delete profile photo " << profile_photo_id << " from cache";
|
|
|
|
auto user_id = get_my_id();
|
|
User *u = get_user_force(user_id, "delete_my_profile_photo_from_cache");
|
|
bool is_main_photo_deleted = u != nullptr && u->photo.id == profile_photo_id;
|
|
|
|
// update photo list
|
|
auto user_photos = user_photos_.get_pointer(user_id);
|
|
if (user_photos != nullptr && user_photos->count > 0) {
|
|
auto old_size = user_photos->photos.size();
|
|
if (td::remove_if(user_photos->photos,
|
|
[profile_photo_id](const auto &photo) { return photo.id.get() == profile_photo_id; })) {
|
|
auto removed_photos = old_size - user_photos->photos.size();
|
|
CHECK(removed_photos > 0);
|
|
LOG_IF(ERROR, removed_photos != 1) << "Had " << removed_photos << " photos with ID " << profile_photo_id;
|
|
user_photos->count -= narrow_cast<int32>(removed_photos);
|
|
// offset was not changed
|
|
CHECK(user_photos->count >= 0);
|
|
} else {
|
|
// failed to find photo to remove from cache
|
|
// don't know how to adjust user_photos->offset, so drop photos cache
|
|
LOG(INFO) << "Drop photos of " << user_id;
|
|
user_photos->photos.clear();
|
|
user_photos->count = -1;
|
|
user_photos->offset = -1;
|
|
}
|
|
}
|
|
bool have_new_photo =
|
|
user_photos != nullptr && user_photos->count != -1 && user_photos->offset == 0 && !user_photos->photos.empty();
|
|
|
|
auto user_full = get_user_full_force(user_id, "delete_my_profile_photo_from_cache");
|
|
|
|
// update ProfilePhoto in User
|
|
bool need_reget_user = false;
|
|
if (is_main_photo_deleted) {
|
|
if (have_new_photo) {
|
|
do_update_user_photo(
|
|
u, user_id,
|
|
as_profile_photo(td_->file_manager_.get(), user_id, u->access_hash, user_photos->photos[0], false), false,
|
|
"delete_my_profile_photo_from_cache");
|
|
} else {
|
|
do_update_user_photo(u, user_id, ProfilePhoto(), false, "delete_my_profile_photo_from_cache 2");
|
|
need_reget_user = user_photos == nullptr || user_photos->count != 0;
|
|
}
|
|
update_user(u, user_id);
|
|
|
|
// update Photo in UserFull
|
|
if (user_full != nullptr) {
|
|
if (user_full->fallback_photo.id.get() == profile_photo_id) {
|
|
LOG(INFO) << "Drop full public photo of " << user_id;
|
|
user_full->photo = Photo();
|
|
user_full->is_changed = true;
|
|
} else if (have_new_photo) {
|
|
if (user_full->photo.id.get() == profile_photo_id && user_photos->photos[0] != user_full->photo) {
|
|
LOG(INFO) << "Update full photo of " << user_id << " to " << user_photos->photos[0];
|
|
user_full->photo = user_photos->photos[0];
|
|
user_full->is_changed = true;
|
|
}
|
|
} else {
|
|
// repair UserFull photo
|
|
if (!user_full->photo.is_empty()) {
|
|
user_full->photo = Photo();
|
|
user_full->is_changed = true;
|
|
}
|
|
if (!user_full->fallback_photo.is_empty()) {
|
|
user_full->fallback_photo = Photo();
|
|
user_full->is_changed = true;
|
|
}
|
|
}
|
|
if (user_full->expires_at > 0.0) {
|
|
user_full->expires_at = 0.0;
|
|
user_full->need_save_to_database = true;
|
|
}
|
|
reload_user_full(user_id, Auto(), "delete_my_profile_photo_from_cache");
|
|
update_user_full(user_full, user_id, "delete_my_profile_photo_from_cache");
|
|
}
|
|
}
|
|
|
|
return need_reget_user;
|
|
}
|
|
|
|
void UserManager::delete_profile_photo(int64 profile_photo_id, bool is_recursive, Promise<Unit> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
const UserFull *user_full = get_user_full_force(get_my_id(), "delete_profile_photo");
|
|
if (user_full == nullptr) {
|
|
// must load UserFull first, because fallback photo can't be deleted via DeleteProfilePhotoQuery
|
|
if (is_recursive) {
|
|
return promise.set_error(Status::Error(500, "Failed to load UserFullInfo"));
|
|
}
|
|
auto reload_promise = PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), profile_photo_id, promise = std::move(promise)](Result<Unit> result) mutable {
|
|
if (result.is_error()) {
|
|
return promise.set_error(result.move_as_error());
|
|
}
|
|
send_closure(actor_id, &UserManager::delete_profile_photo, profile_photo_id, true, std::move(promise));
|
|
});
|
|
reload_user_full(get_my_id(), std::move(reload_promise), "delete_profile_photo");
|
|
return;
|
|
}
|
|
if (user_full->photo.id.get() == profile_photo_id || user_full->fallback_photo.id.get() == profile_photo_id) {
|
|
td_->create_handler<UpdateProfilePhotoQuery>(std::move(promise))
|
|
->send(get_my_id(), FileId(), profile_photo_id, user_full->fallback_photo.id.get() == profile_photo_id,
|
|
telegram_api::make_object<telegram_api::inputPhotoEmpty>());
|
|
return;
|
|
}
|
|
|
|
td_->create_handler<DeleteProfilePhotoQuery>(std::move(promise))->send(profile_photo_id);
|
|
}
|
|
|
|
void UserManager::on_delete_profile_photo(int64 profile_photo_id, Promise<Unit> promise) {
|
|
bool need_reget_user = delete_my_profile_photo_from_cache(profile_photo_id);
|
|
if (need_reget_user && !G()->close_flag()) {
|
|
return reload_user(get_my_id(), std::move(promise), "on_delete_profile_photo");
|
|
}
|
|
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::set_username(const string &username, Promise<Unit> &&promise) {
|
|
if (!username.empty() && !is_allowed_username(username)) {
|
|
return promise.set_error(Status::Error(400, "Username is invalid"));
|
|
}
|
|
td_->create_handler<UpdateUsernameQuery>(std::move(promise))->send(username);
|
|
}
|
|
|
|
void UserManager::toggle_username_is_active(string &&username, bool is_active, Promise<Unit> &&promise) {
|
|
get_me(PromiseCreator::lambda([actor_id = actor_id(this), username = std::move(username), is_active,
|
|
promise = std::move(promise)](Result<Unit> &&result) mutable {
|
|
if (result.is_error()) {
|
|
promise.set_error(result.move_as_error());
|
|
} else {
|
|
send_closure(actor_id, &UserManager::toggle_username_is_active_impl, std::move(username), is_active,
|
|
std::move(promise));
|
|
}
|
|
}));
|
|
}
|
|
|
|
void UserManager::toggle_username_is_active_impl(string &&username, bool is_active, Promise<Unit> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
const User *u = get_user(get_my_id());
|
|
CHECK(u != nullptr);
|
|
if (!u->usernames.can_toggle(username)) {
|
|
return promise.set_error(Status::Error(400, "Wrong username specified"));
|
|
}
|
|
td_->create_handler<ToggleUsernameQuery>(std::move(promise))->send(std::move(username), is_active);
|
|
}
|
|
|
|
void UserManager::reorder_usernames(vector<string> &&usernames, Promise<Unit> &&promise) {
|
|
get_me(PromiseCreator::lambda([actor_id = actor_id(this), usernames = std::move(usernames),
|
|
promise = std::move(promise)](Result<Unit> &&result) mutable {
|
|
if (result.is_error()) {
|
|
promise.set_error(result.move_as_error());
|
|
} else {
|
|
send_closure(actor_id, &UserManager::reorder_usernames_impl, std::move(usernames), std::move(promise));
|
|
}
|
|
}));
|
|
}
|
|
|
|
void UserManager::reorder_usernames_impl(vector<string> &&usernames, Promise<Unit> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
const User *u = get_user(get_my_id());
|
|
CHECK(u != nullptr);
|
|
if (!u->usernames.can_reorder_to(usernames)) {
|
|
return promise.set_error(Status::Error(400, "Invalid username order specified"));
|
|
}
|
|
if (usernames.size() <= 1) {
|
|
return promise.set_value(Unit());
|
|
}
|
|
td_->create_handler<ReorderUsernamesQuery>(std::move(promise))->send(std::move(usernames));
|
|
}
|
|
|
|
void UserManager::on_update_username_is_active(UserId user_id, string &&username, bool is_active,
|
|
Promise<Unit> &&promise) {
|
|
User *u = get_user(user_id);
|
|
CHECK(u != nullptr);
|
|
if (!u->usernames.can_toggle(username)) {
|
|
return reload_user(user_id, std::move(promise), "on_update_username_is_active");
|
|
}
|
|
on_update_user_usernames(u, user_id, u->usernames.toggle(username, is_active));
|
|
update_user(u, user_id);
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::on_update_active_usernames_order(UserId user_id, vector<string> &&usernames,
|
|
Promise<Unit> &&promise) {
|
|
User *u = get_user(user_id);
|
|
CHECK(u != nullptr);
|
|
if (!u->usernames.can_reorder_to(usernames)) {
|
|
return reload_user(user_id, std::move(promise), "on_update_active_usernames_order");
|
|
}
|
|
on_update_user_usernames(u, user_id, u->usernames.reorder_to(std::move(usernames)));
|
|
update_user(u, user_id);
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::toggle_bot_username_is_active(UserId bot_user_id, string &&username, bool is_active,
|
|
Promise<Unit> &&promise) {
|
|
TRY_RESULT_PROMISE(promise, bot_data, get_bot_data(bot_user_id));
|
|
if (!bot_data.can_be_edited) {
|
|
return promise.set_error(Status::Error(400, "The bot can't be edited"));
|
|
}
|
|
const User *u = get_user(bot_user_id);
|
|
CHECK(u != nullptr);
|
|
if (!u->usernames.can_toggle(username)) {
|
|
return promise.set_error(Status::Error(400, "Wrong username specified"));
|
|
}
|
|
td_->create_handler<ToggleBotUsernameQuery>(std::move(promise))->send(bot_user_id, std::move(username), is_active);
|
|
}
|
|
|
|
void UserManager::reorder_bot_usernames(UserId bot_user_id, vector<string> &&usernames, Promise<Unit> &&promise) {
|
|
TRY_RESULT_PROMISE(promise, bot_data, get_bot_data(bot_user_id));
|
|
if (!bot_data.can_be_edited) {
|
|
return promise.set_error(Status::Error(400, "The bot can't be edited"));
|
|
}
|
|
const User *u = get_user(bot_user_id);
|
|
CHECK(u != nullptr);
|
|
if (!u->usernames.can_reorder_to(usernames)) {
|
|
return promise.set_error(Status::Error(400, "Invalid username order specified"));
|
|
}
|
|
if (usernames.size() <= 1) {
|
|
return promise.set_value(Unit());
|
|
}
|
|
td_->create_handler<ReorderBotUsernamesQuery>(std::move(promise))->send(bot_user_id, std::move(usernames));
|
|
}
|
|
|
|
void UserManager::set_accent_color(AccentColorId accent_color_id, CustomEmojiId background_custom_emoji_id,
|
|
Promise<Unit> &&promise) {
|
|
if (!accent_color_id.is_valid()) {
|
|
return promise.set_error(Status::Error(400, "Invalid accent color identifier specified"));
|
|
}
|
|
if (accent_color_id == AccentColorId(get_my_id())) {
|
|
accent_color_id = AccentColorId();
|
|
}
|
|
|
|
td_->create_handler<UpdateColorQuery>(std::move(promise))->send(false, accent_color_id, background_custom_emoji_id);
|
|
}
|
|
|
|
void UserManager::set_profile_accent_color(AccentColorId accent_color_id, CustomEmojiId background_custom_emoji_id,
|
|
Promise<Unit> &&promise) {
|
|
td_->create_handler<UpdateColorQuery>(std::move(promise))->send(true, accent_color_id, background_custom_emoji_id);
|
|
}
|
|
|
|
void UserManager::on_update_accent_color_success(bool for_profile, AccentColorId accent_color_id,
|
|
CustomEmojiId background_custom_emoji_id) {
|
|
auto user_id = get_my_id();
|
|
User *u = get_user_force(user_id, "on_update_accent_color_success");
|
|
if (u == nullptr) {
|
|
return;
|
|
}
|
|
if (for_profile) {
|
|
on_update_user_profile_accent_color_id(u, user_id, accent_color_id);
|
|
on_update_user_profile_background_custom_emoji_id(u, user_id, background_custom_emoji_id);
|
|
} else {
|
|
on_update_user_accent_color_id(u, user_id, accent_color_id);
|
|
on_update_user_background_custom_emoji_id(u, user_id, background_custom_emoji_id);
|
|
}
|
|
update_user(u, user_id);
|
|
}
|
|
|
|
void UserManager::set_birthdate(Birthdate &&birthdate, Promise<Unit> &&promise) {
|
|
dismiss_suggested_action(SuggestedAction{SuggestedAction::Type::BirthdaySetup}, Promise<Unit>());
|
|
auto query_promise = PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), birthdate, promise = std::move(promise)](Result<Unit> result) mutable {
|
|
if (result.is_ok()) {
|
|
send_closure(actor_id, &UserManager::on_set_birthdate, birthdate, std::move(promise));
|
|
} else {
|
|
promise.set_error(result.move_as_error());
|
|
}
|
|
});
|
|
td_->create_handler<UpdateBirthdayQuery>(std::move(query_promise))->send(birthdate);
|
|
}
|
|
|
|
void UserManager::on_set_birthdate(Birthdate birthdate, Promise<Unit> &&promise) {
|
|
auto my_user_id = get_my_id();
|
|
UserFull *user_full = get_user_full_force(my_user_id, "on_set_birthdate");
|
|
if (user_full != nullptr && user_full->birthdate != birthdate) {
|
|
user_full->birthdate = std::move(birthdate);
|
|
user_full->is_changed = true;
|
|
update_user_full(user_full, my_user_id, "on_set_birthdate");
|
|
}
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::set_personal_channel(DialogId dialog_id, Promise<Unit> &&promise) {
|
|
ChannelId channel_id;
|
|
if (dialog_id != DialogId()) {
|
|
if (!td_->dialog_manager_->have_dialog_force(dialog_id, "set_personal_channel")) {
|
|
return promise.set_error(Status::Error(400, "Chat not found"));
|
|
}
|
|
if (!td_->dialog_manager_->is_broadcast_channel(dialog_id)) {
|
|
return promise.set_error(Status::Error(400, "Chat can't be set as a personal chat"));
|
|
}
|
|
channel_id = dialog_id.get_channel_id();
|
|
}
|
|
auto query_promise = PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), channel_id, promise = std::move(promise)](Result<Unit> result) mutable {
|
|
if (result.is_ok()) {
|
|
send_closure(actor_id, &UserManager::on_set_personal_channel, channel_id, std::move(promise));
|
|
} else {
|
|
promise.set_error(result.move_as_error());
|
|
}
|
|
});
|
|
td_->create_handler<UpdatePersonalChannelQuery>(std::move(query_promise))->send(channel_id);
|
|
}
|
|
|
|
void UserManager::on_set_personal_channel(ChannelId channel_id, Promise<Unit> &&promise) {
|
|
auto my_user_id = get_my_id();
|
|
UserFull *user_full = get_user_full_force(my_user_id, "on_set_personal_channel");
|
|
if (user_full != nullptr && user_full->personal_channel_id != channel_id) {
|
|
user_full->personal_channel_id = channel_id;
|
|
user_full->is_changed = true;
|
|
update_user_full(user_full, my_user_id, "on_set_personal_channel");
|
|
}
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::set_emoji_status(const EmojiStatus &emoji_status, Promise<Unit> &&promise) {
|
|
if (!td_->option_manager_->get_option_boolean("is_premium")) {
|
|
return promise.set_error(Status::Error(400, "The method is available only to Telegram Premium users"));
|
|
}
|
|
add_recent_emoji_status(td_, emoji_status);
|
|
auto query_promise = PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), emoji_status, promise = std::move(promise)](Result<Unit> result) mutable {
|
|
if (result.is_ok()) {
|
|
send_closure(actor_id, &UserManager::on_set_emoji_status, emoji_status, std::move(promise));
|
|
} else {
|
|
promise.set_error(result.move_as_error());
|
|
}
|
|
});
|
|
td_->create_handler<UpdateEmojiStatusQuery>(std::move(query_promise))->send(emoji_status);
|
|
}
|
|
|
|
void UserManager::on_set_emoji_status(EmojiStatus emoji_status, Promise<Unit> &&promise) {
|
|
auto user_id = get_my_id();
|
|
User *u = get_user(user_id);
|
|
if (u != nullptr) {
|
|
on_update_user_emoji_status(u, user_id, emoji_status);
|
|
update_user(u, user_id);
|
|
}
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::toggle_sponsored_messages(bool sponsored_enabled, Promise<Unit> &&promise) {
|
|
auto query_promise = PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), sponsored_enabled, promise = std::move(promise)](Result<Unit> result) mutable {
|
|
if (result.is_ok()) {
|
|
send_closure(actor_id, &UserManager::on_toggle_sponsored_messages, sponsored_enabled, std::move(promise));
|
|
} else {
|
|
promise.set_error(result.move_as_error());
|
|
}
|
|
});
|
|
td_->create_handler<ToggleSponsoredMessagesQuery>(std::move(query_promise))->send(sponsored_enabled);
|
|
}
|
|
|
|
void UserManager::on_toggle_sponsored_messages(bool sponsored_enabled, Promise<Unit> &&promise) {
|
|
auto my_user_id = get_my_id();
|
|
UserFull *user_full = get_user_full_force(my_user_id, "on_toggle_sponsored_messages");
|
|
if (user_full != nullptr && user_full->sponsored_enabled != sponsored_enabled) {
|
|
user_full->sponsored_enabled = sponsored_enabled;
|
|
user_full->is_changed = true;
|
|
update_user_full(user_full, my_user_id, "on_toggle_sponsored_messages");
|
|
}
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::get_support_user(Promise<td_api::object_ptr<td_api::user>> &&promise) {
|
|
if (support_user_id_.is_valid()) {
|
|
return promise.set_value(get_user_object(support_user_id_));
|
|
}
|
|
|
|
auto query_promise = PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), promise = std::move(promise)](Result<UserId> &&result) mutable {
|
|
if (result.is_error()) {
|
|
promise.set_error(result.move_as_error());
|
|
} else {
|
|
send_closure(actor_id, &UserManager::on_get_support_user, result.move_as_ok(), std::move(promise));
|
|
}
|
|
});
|
|
td_->create_handler<GetSupportUserQuery>(std::move(query_promise))->send();
|
|
}
|
|
|
|
void UserManager::on_get_support_user(UserId user_id, Promise<td_api::object_ptr<td_api::user>> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
|
|
const User *u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return promise.set_error(Status::Error(500, "Can't find support user"));
|
|
}
|
|
if (!u->is_support) {
|
|
LOG(ERROR) << "Receive non-support " << user_id << ", but expected a support user";
|
|
}
|
|
|
|
support_user_id_ = user_id;
|
|
promise.set_value(get_user_object(user_id, u));
|
|
}
|
|
|
|
void UserManager::get_user_profile_photos(UserId user_id, int32 offset, int32 limit,
|
|
Promise<td_api::object_ptr<td_api::chatPhotos>> &&promise) {
|
|
if (offset < 0) {
|
|
return promise.set_error(Status::Error(400, "Parameter offset must be non-negative"));
|
|
}
|
|
if (limit <= 0) {
|
|
return promise.set_error(Status::Error(400, "Parameter limit must be positive"));
|
|
}
|
|
if (limit > MAX_GET_PROFILE_PHOTOS) {
|
|
limit = MAX_GET_PROFILE_PHOTOS;
|
|
}
|
|
|
|
TRY_STATUS_PROMISE(promise, get_input_user(user_id));
|
|
|
|
auto *u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return promise.set_error(Status::Error(400, "User not found"));
|
|
}
|
|
|
|
apply_pending_user_photo(u, user_id);
|
|
|
|
auto user_photos = add_user_photos(user_id);
|
|
if (user_photos->count != -1) { // know photo count
|
|
CHECK(user_photos->offset != -1);
|
|
LOG(INFO) << "Have " << user_photos->count << " cached user profile photos at offset " << user_photos->offset;
|
|
vector<td_api::object_ptr<td_api::chatPhoto>> photo_objects;
|
|
|
|
if (offset >= user_photos->count) {
|
|
// offset if too big
|
|
return promise.set_value(td_api::make_object<td_api::chatPhotos>(user_photos->count, std::move(photo_objects)));
|
|
}
|
|
|
|
if (limit > user_photos->count - offset) {
|
|
limit = user_photos->count - offset;
|
|
}
|
|
|
|
int32 cache_begin = user_photos->offset;
|
|
int32 cache_end = cache_begin + narrow_cast<int32>(user_photos->photos.size());
|
|
if (cache_begin <= offset && offset + limit <= cache_end) {
|
|
// answer query from cache
|
|
for (int i = 0; i < limit; i++) {
|
|
photo_objects.push_back(
|
|
get_chat_photo_object(td_->file_manager_.get(), user_photos->photos[i + offset - cache_begin]));
|
|
}
|
|
return promise.set_value(td_api::make_object<td_api::chatPhotos>(user_photos->count, std::move(photo_objects)));
|
|
}
|
|
}
|
|
|
|
PendingGetPhotoRequest pending_request;
|
|
pending_request.offset = offset;
|
|
pending_request.limit = limit;
|
|
pending_request.promise = std::move(promise);
|
|
user_photos->pending_requests.push_back(std::move(pending_request));
|
|
if (user_photos->pending_requests.size() != 1u) {
|
|
return;
|
|
}
|
|
|
|
send_get_user_photos_query(user_id, user_photos);
|
|
}
|
|
|
|
void UserManager::send_get_user_photos_query(UserId user_id, const UserPhotos *user_photos) {
|
|
CHECK(!user_photos->pending_requests.empty());
|
|
auto offset = user_photos->pending_requests[0].offset;
|
|
auto limit = user_photos->pending_requests[0].limit;
|
|
|
|
if (user_photos->count != -1 && offset >= user_photos->offset) {
|
|
int32 cache_end = user_photos->offset + narrow_cast<int32>(user_photos->photos.size());
|
|
if (offset < cache_end) {
|
|
// adjust offset to the end of cache
|
|
CHECK(offset + limit > cache_end); // otherwise the request has already been answered
|
|
limit = offset + limit - cache_end;
|
|
offset = cache_end;
|
|
}
|
|
}
|
|
|
|
auto query_promise = PromiseCreator::lambda([actor_id = actor_id(this), user_id](Result<Unit> &&result) {
|
|
send_closure(actor_id, &UserManager::on_get_user_profile_photos, user_id, std::move(result));
|
|
});
|
|
|
|
td_->create_handler<GetUserPhotosQuery>(std::move(query_promise))
|
|
->send(user_id, get_input_user_force(user_id), offset, max(limit, MAX_GET_PROFILE_PHOTOS / 5), 0);
|
|
}
|
|
|
|
void UserManager::on_get_user_profile_photos(UserId user_id, Result<Unit> &&result) {
|
|
G()->ignore_result_if_closing(result);
|
|
auto user_photos = add_user_photos(user_id);
|
|
auto pending_requests = std::move(user_photos->pending_requests);
|
|
CHECK(!pending_requests.empty());
|
|
if (result.is_error()) {
|
|
for (auto &request : pending_requests) {
|
|
request.promise.set_error(result.error().clone());
|
|
}
|
|
return;
|
|
}
|
|
if (user_photos->count == -1) {
|
|
CHECK(have_user(user_id));
|
|
// received result has just been dropped; resend request
|
|
if (++pending_requests[0].retry_count >= 3) {
|
|
pending_requests[0].promise.set_error(Status::Error(500, "Failed to return profile photos"));
|
|
pending_requests.erase(pending_requests.begin());
|
|
if (pending_requests.empty()) {
|
|
return;
|
|
}
|
|
}
|
|
user_photos->pending_requests = std::move(pending_requests);
|
|
return send_get_user_photos_query(user_id, user_photos);
|
|
}
|
|
|
|
CHECK(user_photos->offset != -1);
|
|
LOG(INFO) << "Have " << user_photos->count << " cached user profile photos at offset " << user_photos->offset;
|
|
vector<PendingGetPhotoRequest> left_requests;
|
|
for (size_t request_index = 0; request_index < pending_requests.size(); request_index++) {
|
|
auto &request = pending_requests[request_index];
|
|
vector<td_api::object_ptr<td_api::chatPhoto>> photo_objects;
|
|
|
|
if (request.offset >= user_photos->count) {
|
|
// offset if too big
|
|
request.promise.set_value(td_api::make_object<td_api::chatPhotos>(user_photos->count, std::move(photo_objects)));
|
|
continue;
|
|
}
|
|
|
|
if (request.limit > user_photos->count - request.offset) {
|
|
request.limit = user_photos->count - request.offset;
|
|
}
|
|
|
|
int32 cache_begin = user_photos->offset;
|
|
int32 cache_end = cache_begin + narrow_cast<int32>(user_photos->photos.size());
|
|
if (cache_begin <= request.offset && request.offset + request.limit <= cache_end) {
|
|
// answer query from cache
|
|
for (int i = 0; i < request.limit; i++) {
|
|
photo_objects.push_back(
|
|
get_chat_photo_object(td_->file_manager_.get(), user_photos->photos[i + request.offset - cache_begin]));
|
|
}
|
|
request.promise.set_value(td_api::make_object<td_api::chatPhotos>(user_photos->count, std::move(photo_objects)));
|
|
continue;
|
|
}
|
|
|
|
if (request_index == 0 && ++request.retry_count >= 3) {
|
|
request.promise.set_error(Status::Error(500, "Failed to get profile photos"));
|
|
continue;
|
|
}
|
|
|
|
left_requests.push_back(std::move(request));
|
|
}
|
|
|
|
if (!left_requests.empty()) {
|
|
bool need_send = user_photos->pending_requests.empty();
|
|
append(user_photos->pending_requests, std::move(left_requests));
|
|
if (need_send) {
|
|
send_get_user_photos_query(user_id, user_photos);
|
|
}
|
|
}
|
|
}
|
|
|
|
void UserManager::reload_user_profile_photo(UserId user_id, int64 photo_id, Promise<Unit> &&promise) {
|
|
get_user_force(user_id, "reload_user_profile_photo");
|
|
TRY_RESULT_PROMISE(promise, input_user, get_input_user(user_id));
|
|
|
|
// this request will be needed only to download the photo,
|
|
// so there is no reason to combine different requests for a photo into one request
|
|
td_->create_handler<GetUserPhotosQuery>(std::move(promise))->send(user_id, std::move(input_user), -1, 1, photo_id);
|
|
}
|
|
|
|
FileSourceId UserManager::get_user_profile_photo_file_source_id(UserId user_id, int64 photo_id) {
|
|
if (!user_id.is_valid()) {
|
|
return FileSourceId();
|
|
}
|
|
|
|
auto u = get_user(user_id);
|
|
if (u != nullptr && u->photo_ids.count(photo_id) != 0) {
|
|
VLOG(file_references) << "Don't need to create file source for photo " << photo_id << " of " << user_id;
|
|
// photo was already added, source ID was registered and shouldn't be needed
|
|
return FileSourceId();
|
|
}
|
|
|
|
auto &source_id = user_profile_photo_file_source_ids_[std::make_pair(user_id, photo_id)];
|
|
if (!source_id.is_valid()) {
|
|
source_id = td_->file_reference_manager_->create_user_photo_file_source(user_id, photo_id);
|
|
}
|
|
VLOG(file_references) << "Return " << source_id << " for photo " << photo_id << " of " << user_id;
|
|
return source_id;
|
|
}
|
|
|
|
UserManager::UserPhotos *UserManager::add_user_photos(UserId user_id) {
|
|
CHECK(user_id.is_valid());
|
|
auto &user_photos_ptr = user_photos_[user_id];
|
|
if (user_photos_ptr == nullptr) {
|
|
user_photos_ptr = make_unique<UserPhotos>();
|
|
}
|
|
return user_photos_ptr.get();
|
|
}
|
|
|
|
void UserManager::on_get_user_photos(UserId user_id, int32 offset, int32 limit, int32 total_count,
|
|
vector<telegram_api::object_ptr<telegram_api::Photo>> photos) {
|
|
auto photo_count = narrow_cast<int32>(photos.size());
|
|
int32 min_total_count = (offset >= 0 && photo_count > 0 ? offset : 0) + photo_count;
|
|
if (total_count < min_total_count) {
|
|
LOG(ERROR) << "Receive wrong photos total_count " << total_count << " for user " << user_id << ": receive "
|
|
<< photo_count << " photos with offset " << offset;
|
|
total_count = min_total_count;
|
|
}
|
|
LOG_IF(ERROR, limit < photo_count) << "Requested not more than " << limit << " photos, but " << photo_count
|
|
<< " received";
|
|
|
|
User *u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
LOG(ERROR) << "Can't find " << user_id;
|
|
return;
|
|
}
|
|
|
|
if (offset == -1) {
|
|
// from reload_user_profile_photo
|
|
CHECK(limit == 1);
|
|
for (auto &photo_ptr : photos) {
|
|
if (photo_ptr->get_id() == telegram_api::photo::ID) {
|
|
auto server_photo = telegram_api::move_object_as<telegram_api::photo>(photo_ptr);
|
|
if (server_photo->id_ == u->photo.id) {
|
|
auto profile_photo = convert_photo_to_profile_photo(server_photo, u->photo.is_personal);
|
|
if (profile_photo) {
|
|
LOG_IF(ERROR, u->access_hash == -1) << "Receive profile photo of " << user_id << " without access hash";
|
|
get_profile_photo(td_->file_manager_.get(), user_id, u->access_hash, std::move(profile_photo));
|
|
} else {
|
|
LOG(ERROR) << "Failed to get profile photo from " << to_string(server_photo);
|
|
}
|
|
}
|
|
|
|
auto photo = get_photo(td_, std::move(server_photo), DialogId(user_id));
|
|
register_user_photo(u, user_id, photo);
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
LOG(INFO) << "Receive " << photo_count << " photos of " << user_id << " out of " << total_count << " with offset "
|
|
<< offset << " and limit " << limit;
|
|
UserPhotos *user_photos = add_user_photos(user_id);
|
|
user_photos->count = total_count;
|
|
CHECK(!user_photos->pending_requests.empty());
|
|
|
|
if (user_photos->offset == -1) {
|
|
user_photos->offset = 0;
|
|
CHECK(user_photos->photos.empty());
|
|
}
|
|
|
|
if (offset != narrow_cast<int32>(user_photos->photos.size()) + user_photos->offset) {
|
|
LOG(INFO) << "Inappropriate offset to append " << user_id << " profile photos to cache: offset = " << offset
|
|
<< ", current_offset = " << user_photos->offset << ", photo_count = " << user_photos->photos.size();
|
|
user_photos->photos.clear();
|
|
user_photos->offset = offset;
|
|
}
|
|
|
|
for (auto &photo : photos) {
|
|
auto user_photo = get_photo(td_, std::move(photo), DialogId(user_id));
|
|
if (user_photo.is_empty()) {
|
|
LOG(ERROR) << "Receive empty profile photo in getUserPhotos request for " << user_id << " with offset " << offset
|
|
<< " and limit " << limit << ". Receive " << photo_count << " photos out of " << total_count
|
|
<< " photos";
|
|
user_photos->count--;
|
|
CHECK(user_photos->count >= 0);
|
|
continue;
|
|
}
|
|
|
|
user_photos->photos.push_back(std::move(user_photo));
|
|
register_user_photo(u, user_id, user_photos->photos.back());
|
|
}
|
|
if (user_photos->offset > user_photos->count) {
|
|
user_photos->offset = user_photos->count;
|
|
user_photos->photos.clear();
|
|
}
|
|
|
|
auto known_photo_count = narrow_cast<int32>(user_photos->photos.size());
|
|
if (user_photos->offset + known_photo_count > user_photos->count) {
|
|
user_photos->photos.resize(user_photos->count - user_photos->offset);
|
|
}
|
|
}
|
|
|
|
void UserManager::apply_pending_user_photo(User *u, UserId user_id) {
|
|
if (u == nullptr || u->is_photo_inited) {
|
|
return;
|
|
}
|
|
|
|
if (pending_user_photos_.count(user_id) > 0) {
|
|
do_update_user_photo(u, user_id, std::move(pending_user_photos_[user_id]), "apply_pending_user_photo");
|
|
pending_user_photos_.erase(user_id);
|
|
update_user(u, user_id);
|
|
}
|
|
}
|
|
|
|
void UserManager::register_message_users(MessageFullId message_full_id, vector<UserId> user_ids) {
|
|
auto dialog_id = message_full_id.get_dialog_id();
|
|
CHECK(dialog_id.get_type() == DialogType::Channel);
|
|
if (!td_->chat_manager_->have_channel(dialog_id.get_channel_id())) {
|
|
return;
|
|
}
|
|
for (auto user_id : user_ids) {
|
|
CHECK(user_id.is_valid());
|
|
const User *u = get_user(user_id);
|
|
if (u == nullptr || u->access_hash == -1 || u->is_min_access_hash) {
|
|
auto &user_messages = user_messages_[user_id];
|
|
auto need_update = user_messages.empty();
|
|
user_messages.insert(message_full_id);
|
|
if (need_update) {
|
|
send_closure(G()->td(), &Td::send_update, get_update_user_object(user_id, u));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void UserManager::unregister_message_users(MessageFullId message_full_id, vector<UserId> user_ids) {
|
|
if (user_messages_.empty()) {
|
|
// fast path
|
|
return;
|
|
}
|
|
for (auto user_id : user_ids) {
|
|
auto it = user_messages_.find(user_id);
|
|
if (it != user_messages_.end()) {
|
|
it->second.erase(message_full_id);
|
|
if (it->second.empty()) {
|
|
user_messages_.erase(it);
|
|
|
|
const User *u = get_user(user_id);
|
|
if (u == nullptr || u->access_hash == -1 || u->is_min_access_hash) {
|
|
send_closure(G()->td(), &Td::send_update, get_update_user_object(user_id, u));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void UserManager::can_send_message_to_user(UserId user_id, bool force,
|
|
Promise<td_api::object_ptr<td_api::CanSendMessageToUserResult>> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
if (user_id == get_my_id()) {
|
|
return promise.set_value(td_api::make_object<td_api::canSendMessageToUserResultOk>());
|
|
}
|
|
const auto *u = get_user(user_id);
|
|
if (!have_input_peer_user(u, user_id, AccessRights::Write)) {
|
|
return promise.set_value(td_api::make_object<td_api::canSendMessageToUserResultUserIsDeleted>());
|
|
}
|
|
CHECK(user_id.is_valid());
|
|
if ((u != nullptr && (!u->contact_require_premium || u->is_mutual_contact)) ||
|
|
td_->option_manager_->get_option_boolean("is_premium")) {
|
|
return promise.set_value(td_api::make_object<td_api::canSendMessageToUserResultOk>());
|
|
}
|
|
|
|
auto user_full = get_user_full_force(user_id, "can_send_message_to_user");
|
|
if (user_full != nullptr) {
|
|
if (!user_full->contact_require_premium) {
|
|
return promise.set_value(td_api::make_object<td_api::canSendMessageToUserResultOk>());
|
|
}
|
|
return promise.set_value(td_api::make_object<td_api::canSendMessageToUserResultUserRestrictsNewChats>());
|
|
}
|
|
|
|
auto it = user_full_contact_require_premium_.find(user_id);
|
|
if (it != user_full_contact_require_premium_.end()) {
|
|
if (!it->second) {
|
|
return promise.set_value(td_api::make_object<td_api::canSendMessageToUserResultOk>());
|
|
}
|
|
return promise.set_value(td_api::make_object<td_api::canSendMessageToUserResultUserRestrictsNewChats>());
|
|
}
|
|
|
|
if (force) {
|
|
return promise.set_value(td_api::make_object<td_api::canSendMessageToUserResultOk>());
|
|
}
|
|
|
|
auto query_promise = PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), user_id, promise = std::move(promise)](Result<Unit> &&result) mutable {
|
|
if (result.is_error()) {
|
|
return promise.set_error(result.move_as_error());
|
|
}
|
|
send_closure(actor_id, &UserManager::can_send_message_to_user, user_id, true, std::move(promise));
|
|
});
|
|
get_is_premium_required_to_contact_queries_.add_query(user_id.get(), std::move(query_promise),
|
|
"can_send_message_to_user");
|
|
}
|
|
|
|
void UserManager::on_get_is_premium_required_to_contact_users(vector<UserId> &&user_ids,
|
|
vector<bool> &&is_premium_required,
|
|
Promise<Unit> &&promise) {
|
|
if (user_ids.size() != is_premium_required.size()) {
|
|
LOG(ERROR) << "Receive " << is_premium_required.size() << " flags instead of " << user_ids.size();
|
|
return promise.set_error(Status::Error(500, "Receive invalid response"));
|
|
}
|
|
for (size_t i = 0; i < user_ids.size(); i++) {
|
|
auto user_id = user_ids[i];
|
|
CHECK(user_id.is_valid());
|
|
if (get_user_full(user_id) == nullptr) {
|
|
user_full_contact_require_premium_[user_id] = is_premium_required[i];
|
|
}
|
|
}
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::allow_send_message_to_user(UserId user_id) {
|
|
if (get_user_full(user_id) == nullptr) {
|
|
CHECK(user_id.is_valid());
|
|
user_full_contact_require_premium_[user_id] = true;
|
|
}
|
|
}
|
|
|
|
void UserManager::share_phone_number(UserId user_id, Promise<Unit> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
|
|
if (!are_contacts_loaded_) {
|
|
load_contacts(PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), user_id, promise = std::move(promise)](Result<Unit> &&) mutable {
|
|
send_closure(actor_id, &UserManager::share_phone_number, user_id, std::move(promise));
|
|
}));
|
|
return;
|
|
}
|
|
|
|
LOG(INFO) << "Share phone number with " << user_id;
|
|
TRY_RESULT_PROMISE(promise, input_user, get_input_user(user_id));
|
|
|
|
td_->messages_manager_->hide_dialog_action_bar(DialogId(user_id));
|
|
|
|
td_->create_handler<AcceptContactQuery>(std::move(promise))->send(user_id, std::move(input_user));
|
|
}
|
|
|
|
void UserManager::load_contacts(Promise<Unit> &&promise) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
are_contacts_loaded_ = true;
|
|
saved_contact_count_ = 0;
|
|
}
|
|
if (are_contacts_loaded_ && saved_contact_count_ != -1) {
|
|
LOG(INFO) << "Contacts are already loaded";
|
|
return promise.set_value(Unit());
|
|
}
|
|
load_contacts_queries_.push_back(std::move(promise));
|
|
if (load_contacts_queries_.size() == 1u) {
|
|
if (G()->use_chat_info_database() && next_contacts_sync_date_ > 0 && saved_contact_count_ != -1) {
|
|
LOG(INFO) << "Load contacts from database";
|
|
G()->td_db()->get_sqlite_pmc()->get(
|
|
"user_contacts", PromiseCreator::lambda([](string value) {
|
|
send_closure(G()->user_manager(), &UserManager::on_load_contacts_from_database, std::move(value));
|
|
}));
|
|
} else {
|
|
LOG(INFO) << "Load contacts from server";
|
|
reload_contacts(true);
|
|
}
|
|
} else {
|
|
LOG(INFO) << "Load contacts request has already been sent";
|
|
}
|
|
}
|
|
|
|
int64 UserManager::get_contacts_hash() {
|
|
if (!are_contacts_loaded_) {
|
|
return 0;
|
|
}
|
|
|
|
vector<int64> user_ids = contacts_hints_.search_empty(100000).second;
|
|
CHECK(std::is_sorted(user_ids.begin(), user_ids.end()));
|
|
auto my_id = get_my_id();
|
|
const User *u = get_user_force(my_id, "get_contacts_hash");
|
|
if (u != nullptr && u->is_contact) {
|
|
user_ids.insert(std::upper_bound(user_ids.begin(), user_ids.end(), my_id.get()), my_id.get());
|
|
}
|
|
|
|
vector<uint64> numbers;
|
|
numbers.reserve(user_ids.size() + 1);
|
|
numbers.push_back(saved_contact_count_);
|
|
for (auto user_id : user_ids) {
|
|
numbers.push_back(user_id);
|
|
}
|
|
return get_vector_hash(numbers);
|
|
}
|
|
|
|
void UserManager::reload_contacts(bool force) {
|
|
if (!G()->close_flag() && !td_->auth_manager_->is_bot() &&
|
|
next_contacts_sync_date_ != std::numeric_limits<int32>::max() &&
|
|
(next_contacts_sync_date_ < G()->unix_time() || force)) {
|
|
next_contacts_sync_date_ = std::numeric_limits<int32>::max();
|
|
td_->create_handler<GetContactsQuery>()->send(get_contacts_hash());
|
|
}
|
|
}
|
|
|
|
void UserManager::save_next_contacts_sync_date() {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
if (!G()->use_chat_info_database()) {
|
|
return;
|
|
}
|
|
G()->td_db()->get_binlog_pmc()->set("next_contacts_sync_date", to_string(next_contacts_sync_date_));
|
|
}
|
|
|
|
void UserManager::save_contacts_to_database() {
|
|
if (!G()->use_chat_info_database() || !are_contacts_loaded_) {
|
|
return;
|
|
}
|
|
|
|
LOG(INFO) << "Schedule save contacts to database";
|
|
vector<UserId> user_ids =
|
|
transform(contacts_hints_.search_empty(100000).second, [](int64 key) { return UserId(key); });
|
|
|
|
G()->td_db()->get_binlog_pmc()->set("saved_contact_count", to_string(saved_contact_count_));
|
|
G()->td_db()->get_binlog()->force_sync(
|
|
PromiseCreator::lambda([user_ids = std::move(user_ids)](Result<> result) {
|
|
if (result.is_ok()) {
|
|
LOG(INFO) << "Saved contacts to database";
|
|
G()->td_db()->get_sqlite_pmc()->set(
|
|
"user_contacts", log_event_store(user_ids).as_slice().str(), PromiseCreator::lambda([](Result<> result) {
|
|
if (result.is_ok()) {
|
|
send_closure(G()->user_manager(), &UserManager::save_next_contacts_sync_date);
|
|
}
|
|
}));
|
|
}
|
|
}),
|
|
"save_contacts_to_database");
|
|
}
|
|
|
|
void UserManager::on_get_contacts(telegram_api::object_ptr<telegram_api::contacts_Contacts> &&new_contacts) {
|
|
next_contacts_sync_date_ = G()->unix_time() + Random::fast(70000, 100000);
|
|
|
|
CHECK(new_contacts != nullptr);
|
|
if (new_contacts->get_id() == telegram_api::contacts_contactsNotModified::ID) {
|
|
if (saved_contact_count_ == -1) {
|
|
saved_contact_count_ = 0;
|
|
}
|
|
on_get_contacts_finished(contacts_hints_.size());
|
|
td_->create_handler<GetContactsStatusesQuery>()->send();
|
|
return;
|
|
}
|
|
|
|
auto contacts = move_tl_object_as<telegram_api::contacts_contacts>(new_contacts);
|
|
FlatHashSet<UserId, UserIdHash> contact_user_ids;
|
|
for (auto &user : contacts->users_) {
|
|
auto user_id = get_user_id(user);
|
|
if (!user_id.is_valid()) {
|
|
LOG(ERROR) << "Receive invalid " << user_id;
|
|
continue;
|
|
}
|
|
contact_user_ids.insert(user_id);
|
|
}
|
|
on_get_users(std::move(contacts->users_), "on_get_contacts");
|
|
|
|
UserId my_id = get_my_id();
|
|
users_.foreach([&](const UserId &user_id, unique_ptr<User> &user) {
|
|
User *u = user.get();
|
|
bool should_be_contact = contact_user_ids.count(user_id) == 1;
|
|
if (u->is_contact != should_be_contact) {
|
|
if (u->is_contact) {
|
|
LOG(INFO) << "Drop contact with " << user_id;
|
|
if (user_id != my_id) {
|
|
LOG_CHECK(contacts_hints_.has_key(user_id.get()))
|
|
<< my_id << " " << user_id << " " << to_string(get_user_object(user_id, u));
|
|
}
|
|
on_update_user_is_contact(u, user_id, false, false, false);
|
|
CHECK(u->is_is_contact_changed);
|
|
u->cache_version = 0;
|
|
u->is_repaired = false;
|
|
update_user(u, user_id);
|
|
CHECK(!u->is_contact);
|
|
if (user_id != my_id) {
|
|
CHECK(!contacts_hints_.has_key(user_id.get()));
|
|
}
|
|
} else {
|
|
LOG(ERROR) << "Receive non-contact " << user_id << " in the list of contacts";
|
|
}
|
|
}
|
|
});
|
|
|
|
saved_contact_count_ = contacts->saved_count_;
|
|
on_get_contacts_finished(std::numeric_limits<size_t>::max());
|
|
}
|
|
|
|
void UserManager::on_get_contacts_failed(Status error) {
|
|
CHECK(error.is_error());
|
|
next_contacts_sync_date_ = G()->unix_time() + Random::fast(5, 10);
|
|
fail_promises(load_contacts_queries_, std::move(error));
|
|
}
|
|
|
|
void UserManager::on_load_contacts_from_database(string value) {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
if (value.empty()) {
|
|
reload_contacts(true);
|
|
return;
|
|
}
|
|
|
|
vector<UserId> user_ids;
|
|
if (log_event_parse(user_ids, value).is_error()) {
|
|
LOG(ERROR) << "Failed to load contacts from database";
|
|
reload_contacts(true);
|
|
return;
|
|
}
|
|
|
|
if (log_event_get_version(value) < static_cast<int32>(Version::AddUserFlags2)) {
|
|
next_contacts_sync_date_ = 0;
|
|
save_next_contacts_sync_date();
|
|
reload_contacts(true);
|
|
}
|
|
|
|
LOG(INFO) << "Successfully loaded " << user_ids.size() << " contacts from database";
|
|
|
|
load_contact_users_multipromise_.add_promise(PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), expected_contact_count = user_ids.size()](Result<Unit> result) {
|
|
if (result.is_ok()) {
|
|
send_closure(actor_id, &UserManager::on_get_contacts_finished, expected_contact_count);
|
|
} else {
|
|
LOG(INFO) << "Failed to load contact users from database: " << result.error();
|
|
send_closure(actor_id, &UserManager::reload_contacts, true);
|
|
}
|
|
}));
|
|
|
|
auto lock_promise = load_contact_users_multipromise_.get_promise();
|
|
|
|
for (auto user_id : user_ids) {
|
|
get_user(user_id, 3, load_contact_users_multipromise_.get_promise());
|
|
}
|
|
|
|
lock_promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::on_get_contacts_finished(size_t expected_contact_count) {
|
|
LOG(INFO) << "Finished to get " << contacts_hints_.size() << " contacts out of expected " << expected_contact_count;
|
|
are_contacts_loaded_ = true;
|
|
set_promises(load_contacts_queries_);
|
|
if (expected_contact_count != contacts_hints_.size()) {
|
|
save_contacts_to_database();
|
|
}
|
|
}
|
|
|
|
void UserManager::on_get_contacts_statuses(vector<telegram_api::object_ptr<telegram_api::contactStatus>> &&statuses) {
|
|
auto my_user_id = get_my_id();
|
|
for (auto &status : statuses) {
|
|
UserId user_id(status->user_id_);
|
|
if (user_id != my_user_id) {
|
|
on_update_user_online(user_id, std::move(status->status_));
|
|
}
|
|
}
|
|
save_next_contacts_sync_date();
|
|
}
|
|
|
|
void UserManager::add_contact(Contact contact, bool share_phone_number, Promise<Unit> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
|
|
if (!are_contacts_loaded_) {
|
|
load_contacts(PromiseCreator::lambda([actor_id = actor_id(this), contact = std::move(contact), share_phone_number,
|
|
promise = std::move(promise)](Result<Unit> &&) mutable {
|
|
send_closure(actor_id, &UserManager::add_contact, std::move(contact), share_phone_number, std::move(promise));
|
|
}));
|
|
return;
|
|
}
|
|
|
|
LOG(INFO) << "Add " << contact << " with share_phone_number = " << share_phone_number;
|
|
|
|
auto user_id = contact.get_user_id();
|
|
TRY_RESULT_PROMISE(promise, input_user, get_input_user(user_id));
|
|
|
|
td_->create_handler<AddContactQuery>(std::move(promise))
|
|
->send(user_id, std::move(input_user), contact, share_phone_number);
|
|
}
|
|
|
|
std::pair<vector<UserId>, vector<int32>> UserManager::import_contacts(const vector<Contact> &contacts, int64 &random_id,
|
|
Promise<Unit> &&promise) {
|
|
if (!are_contacts_loaded_) {
|
|
load_contacts(std::move(promise));
|
|
return {};
|
|
}
|
|
|
|
LOG(INFO) << "Asked to import " << contacts.size() << " contacts with random_id = " << random_id;
|
|
if (random_id != 0) {
|
|
// request has already been sent before
|
|
auto it = imported_contacts_.find(random_id);
|
|
CHECK(it != imported_contacts_.end());
|
|
auto result = std::move(it->second);
|
|
imported_contacts_.erase(it);
|
|
|
|
promise.set_value(Unit());
|
|
return result;
|
|
}
|
|
|
|
do {
|
|
random_id = Random::secure_int64();
|
|
} while (random_id == 0 || random_id == 1 || imported_contacts_.count(random_id) > 0);
|
|
imported_contacts_[random_id]; // reserve place for result
|
|
|
|
do_import_contacts(contacts, random_id, std::move(promise));
|
|
return {};
|
|
}
|
|
|
|
void UserManager::do_import_contacts(vector<Contact> contacts, int64 random_id, Promise<Unit> &&promise) {
|
|
size_t size = contacts.size();
|
|
if (size == 0) {
|
|
on_import_contacts_finished(random_id, {}, {});
|
|
return promise.set_value(Unit());
|
|
}
|
|
|
|
vector<telegram_api::object_ptr<telegram_api::inputPhoneContact>> input_phone_contacts;
|
|
input_phone_contacts.reserve(size);
|
|
for (size_t i = 0; i < size; i++) {
|
|
input_phone_contacts.push_back(contacts[i].get_input_phone_contact(static_cast<int64>(i)));
|
|
}
|
|
|
|
auto task = make_unique<ImportContactsTask>();
|
|
task->promise_ = std::move(promise);
|
|
task->input_contacts_ = std::move(contacts);
|
|
task->imported_user_ids_.resize(size);
|
|
task->unimported_contact_invites_.resize(size);
|
|
|
|
bool is_added = import_contact_tasks_.emplace(random_id, std::move(task)).second;
|
|
CHECK(is_added);
|
|
|
|
td_->create_handler<ImportContactsQuery>()->send(std::move(input_phone_contacts), random_id);
|
|
}
|
|
|
|
void UserManager::on_imported_contacts(
|
|
int64 random_id, Result<telegram_api::object_ptr<telegram_api::contacts_importedContacts>> result) {
|
|
auto it = import_contact_tasks_.find(random_id);
|
|
CHECK(it != import_contact_tasks_.end());
|
|
CHECK(it->second != nullptr);
|
|
|
|
auto task = it->second.get();
|
|
if (result.is_error()) {
|
|
auto promise = std::move(task->promise_);
|
|
import_contact_tasks_.erase(it);
|
|
return promise.set_error(result.move_as_error());
|
|
}
|
|
|
|
auto imported_contacts = result.move_as_ok();
|
|
on_get_users(std::move(imported_contacts->users_), "on_imported_contacts");
|
|
|
|
for (auto &imported_contact : imported_contacts->imported_) {
|
|
int64 client_id = imported_contact->client_id_;
|
|
if (client_id < 0 || client_id >= static_cast<int64>(task->imported_user_ids_.size())) {
|
|
LOG(ERROR) << "Wrong client_id " << client_id << " returned";
|
|
continue;
|
|
}
|
|
|
|
task->imported_user_ids_[static_cast<size_t>(client_id)] = UserId(imported_contact->user_id_);
|
|
}
|
|
for (auto &popular_contact : imported_contacts->popular_invites_) {
|
|
int64 client_id = popular_contact->client_id_;
|
|
if (client_id < 0 || client_id >= static_cast<int64>(task->unimported_contact_invites_.size())) {
|
|
LOG(ERROR) << "Wrong client_id " << client_id << " returned";
|
|
continue;
|
|
}
|
|
if (popular_contact->importers_ < 0) {
|
|
LOG(ERROR) << "Wrong number of importers " << popular_contact->importers_ << " returned";
|
|
continue;
|
|
}
|
|
|
|
task->unimported_contact_invites_[static_cast<size_t>(client_id)] = popular_contact->importers_;
|
|
}
|
|
|
|
if (!imported_contacts->retry_contacts_.empty()) {
|
|
auto total_size = static_cast<int64>(task->input_contacts_.size());
|
|
vector<telegram_api::object_ptr<telegram_api::inputPhoneContact>> input_phone_contacts;
|
|
input_phone_contacts.reserve(imported_contacts->retry_contacts_.size());
|
|
for (auto &client_id : imported_contacts->retry_contacts_) {
|
|
if (client_id < 0 || client_id >= total_size) {
|
|
LOG(ERROR) << "Wrong client_id " << client_id << " returned";
|
|
continue;
|
|
}
|
|
auto i = static_cast<size_t>(client_id);
|
|
input_phone_contacts.push_back(task->input_contacts_[i].get_input_phone_contact(client_id));
|
|
}
|
|
td_->create_handler<ImportContactsQuery>()->send(std::move(input_phone_contacts), random_id);
|
|
return;
|
|
}
|
|
|
|
auto promise = std::move(task->promise_);
|
|
on_import_contacts_finished(random_id, std::move(task->imported_user_ids_),
|
|
std::move(task->unimported_contact_invites_));
|
|
import_contact_tasks_.erase(it);
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::on_import_contacts_finished(int64 random_id, vector<UserId> imported_contact_user_ids,
|
|
vector<int32> unimported_contact_invites) {
|
|
LOG(INFO) << "Contacts import with random_id " << random_id
|
|
<< " has finished: " << format::as_array(imported_contact_user_ids);
|
|
if (random_id == 1) {
|
|
// import from change_imported_contacts
|
|
all_imported_contacts_ = std::move(next_all_imported_contacts_);
|
|
next_all_imported_contacts_.clear();
|
|
|
|
auto result_size = imported_contacts_unique_id_.size();
|
|
auto unique_size = all_imported_contacts_.size();
|
|
auto add_size = imported_contacts_pos_.size();
|
|
|
|
imported_contact_user_ids_.resize(result_size);
|
|
unimported_contact_invites_.resize(result_size);
|
|
|
|
CHECK(imported_contact_user_ids.size() == add_size);
|
|
CHECK(unimported_contact_invites.size() == add_size);
|
|
CHECK(imported_contacts_unique_id_.size() == result_size);
|
|
|
|
std::unordered_map<int64, int32, Hash<int64>> unique_id_to_unimported_contact_invites;
|
|
for (size_t i = 0; i < add_size; i++) {
|
|
auto unique_id = imported_contacts_pos_[i];
|
|
get_user_id_object(imported_contact_user_ids[i], "on_import_contacts_finished"); // to ensure updateUser
|
|
all_imported_contacts_[unique_id].set_user_id(imported_contact_user_ids[i]);
|
|
unique_id_to_unimported_contact_invites[narrow_cast<int64>(unique_id)] = unimported_contact_invites[i];
|
|
}
|
|
|
|
if (G()->use_chat_info_database()) {
|
|
G()->td_db()->get_binlog()->force_sync(
|
|
PromiseCreator::lambda(
|
|
[log_event = log_event_store(all_imported_contacts_).as_slice().str()](Result<> result) mutable {
|
|
if (result.is_ok()) {
|
|
LOG(INFO) << "Save imported contacts to database";
|
|
G()->td_db()->get_sqlite_pmc()->set("user_imported_contacts", std::move(log_event), Auto());
|
|
}
|
|
}),
|
|
"on_import_contacts_finished");
|
|
}
|
|
|
|
for (size_t i = 0; i < result_size; i++) {
|
|
auto unique_id = imported_contacts_unique_id_[i];
|
|
CHECK(unique_id < unique_size);
|
|
imported_contact_user_ids_[i] = all_imported_contacts_[unique_id].get_user_id();
|
|
auto it = unique_id_to_unimported_contact_invites.find(narrow_cast<int64>(unique_id));
|
|
if (it == unique_id_to_unimported_contact_invites.end()) {
|
|
unimported_contact_invites_[i] = 0;
|
|
} else {
|
|
unimported_contact_invites_[i] = it->second;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
auto it = imported_contacts_.find(random_id);
|
|
CHECK(it != imported_contacts_.end());
|
|
CHECK(it->second.first.empty());
|
|
CHECK(it->second.second.empty());
|
|
imported_contacts_[random_id] = {std::move(imported_contact_user_ids), std::move(unimported_contact_invites)};
|
|
}
|
|
|
|
void UserManager::remove_contacts(const vector<UserId> &user_ids, Promise<Unit> &&promise) {
|
|
LOG(INFO) << "Delete contacts: " << format::as_array(user_ids);
|
|
if (!are_contacts_loaded_) {
|
|
load_contacts(std::move(promise));
|
|
return;
|
|
}
|
|
|
|
vector<UserId> to_delete_user_ids;
|
|
vector<telegram_api::object_ptr<telegram_api::InputUser>> input_users;
|
|
for (auto &user_id : user_ids) {
|
|
const User *u = get_user(user_id);
|
|
if (u != nullptr && u->is_contact) {
|
|
auto r_input_user = get_input_user(user_id);
|
|
if (r_input_user.is_ok()) {
|
|
to_delete_user_ids.push_back(user_id);
|
|
input_users.push_back(r_input_user.move_as_ok());
|
|
}
|
|
}
|
|
}
|
|
|
|
if (input_users.empty()) {
|
|
return promise.set_value(Unit());
|
|
}
|
|
|
|
td_->create_handler<DeleteContactsQuery>(std::move(promise))->send(std::move(input_users));
|
|
}
|
|
|
|
void UserManager::remove_contacts_by_phone_number(vector<string> user_phone_numbers, vector<UserId> user_ids,
|
|
Promise<Unit> &&promise) {
|
|
LOG(INFO) << "Delete contacts by phone number: " << format::as_array(user_phone_numbers);
|
|
if (!are_contacts_loaded_) {
|
|
load_contacts(std::move(promise));
|
|
return;
|
|
}
|
|
|
|
td_->create_handler<DeleteContactsByPhoneNumberQuery>(std::move(promise))
|
|
->send(std::move(user_phone_numbers), std::move(user_ids));
|
|
}
|
|
|
|
void UserManager::on_deleted_contacts(const vector<UserId> &deleted_contact_user_ids) {
|
|
LOG(INFO) << "Contacts deletion has finished for " << deleted_contact_user_ids;
|
|
|
|
for (auto user_id : deleted_contact_user_ids) {
|
|
auto u = get_user(user_id);
|
|
CHECK(u != nullptr);
|
|
if (!u->is_contact) {
|
|
continue;
|
|
}
|
|
|
|
LOG(INFO) << "Drop contact with " << user_id;
|
|
on_update_user_is_contact(u, user_id, false, false, false);
|
|
CHECK(u->is_is_contact_changed);
|
|
u->cache_version = 0;
|
|
u->is_repaired = false;
|
|
update_user(u, user_id);
|
|
CHECK(!u->is_contact);
|
|
CHECK(!contacts_hints_.has_key(user_id.get()));
|
|
}
|
|
}
|
|
|
|
int32 UserManager::get_imported_contact_count(Promise<Unit> &&promise) {
|
|
LOG(INFO) << "Get imported contact count";
|
|
|
|
if (!are_contacts_loaded_ || saved_contact_count_ == -1) {
|
|
load_contacts(std::move(promise));
|
|
return 0;
|
|
}
|
|
reload_contacts(false);
|
|
|
|
promise.set_value(Unit());
|
|
return saved_contact_count_;
|
|
}
|
|
|
|
void UserManager::load_imported_contacts(Promise<Unit> &&promise) {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
are_imported_contacts_loaded_ = true;
|
|
}
|
|
if (are_imported_contacts_loaded_) {
|
|
LOG(INFO) << "Imported contacts are already loaded";
|
|
promise.set_value(Unit());
|
|
return;
|
|
}
|
|
load_imported_contacts_queries_.push_back(std::move(promise));
|
|
if (load_imported_contacts_queries_.size() == 1u) {
|
|
if (G()->use_chat_info_database()) {
|
|
LOG(INFO) << "Load imported contacts from database";
|
|
G()->td_db()->get_sqlite_pmc()->get("user_imported_contacts", PromiseCreator::lambda([](string value) {
|
|
send_closure_later(G()->user_manager(),
|
|
&UserManager::on_load_imported_contacts_from_database,
|
|
std::move(value));
|
|
}));
|
|
} else {
|
|
LOG(INFO) << "Have no previously imported contacts";
|
|
send_closure_later(G()->user_manager(), &UserManager::on_load_imported_contacts_from_database, string());
|
|
}
|
|
} else {
|
|
LOG(INFO) << "Load imported contacts request has already been sent";
|
|
}
|
|
}
|
|
|
|
void UserManager::on_load_imported_contacts_from_database(string value) {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
|
|
CHECK(!are_imported_contacts_loaded_);
|
|
if (need_clear_imported_contacts_) {
|
|
need_clear_imported_contacts_ = false;
|
|
value.clear();
|
|
}
|
|
if (value.empty()) {
|
|
CHECK(all_imported_contacts_.empty());
|
|
} else {
|
|
if (log_event_parse(all_imported_contacts_, value).is_error()) {
|
|
LOG(ERROR) << "Failed to load all imported contacts from database";
|
|
all_imported_contacts_.clear();
|
|
} else {
|
|
LOG(INFO) << "Successfully loaded " << all_imported_contacts_.size() << " imported contacts from database";
|
|
}
|
|
}
|
|
|
|
load_imported_contact_users_multipromise_.add_promise(
|
|
PromiseCreator::lambda([actor_id = actor_id(this)](Result<Unit> result) {
|
|
if (result.is_ok()) {
|
|
send_closure_later(actor_id, &UserManager::on_load_imported_contacts_finished);
|
|
}
|
|
}));
|
|
|
|
auto lock_promise = load_imported_contact_users_multipromise_.get_promise();
|
|
|
|
for (const auto &contact : all_imported_contacts_) {
|
|
auto user_id = contact.get_user_id();
|
|
if (user_id.is_valid()) {
|
|
get_user(user_id, 3, load_imported_contact_users_multipromise_.get_promise());
|
|
}
|
|
}
|
|
|
|
lock_promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::on_load_imported_contacts_finished() {
|
|
LOG(INFO) << "Finished to load " << all_imported_contacts_.size() << " imported contacts";
|
|
|
|
for (const auto &contact : all_imported_contacts_) {
|
|
get_user_id_object(contact.get_user_id(), "on_load_imported_contacts_finished"); // to ensure updateUser
|
|
}
|
|
|
|
if (need_clear_imported_contacts_) {
|
|
need_clear_imported_contacts_ = false;
|
|
all_imported_contacts_.clear();
|
|
}
|
|
are_imported_contacts_loaded_ = true;
|
|
set_promises(load_imported_contacts_queries_);
|
|
}
|
|
|
|
std::pair<vector<UserId>, vector<int32>> UserManager::change_imported_contacts(vector<Contact> &contacts,
|
|
int64 &random_id,
|
|
Promise<Unit> &&promise) {
|
|
if (!are_contacts_loaded_) {
|
|
load_contacts(std::move(promise));
|
|
return {};
|
|
}
|
|
if (!are_imported_contacts_loaded_) {
|
|
load_imported_contacts(std::move(promise));
|
|
return {};
|
|
}
|
|
|
|
LOG(INFO) << "Asked to change imported contacts to a list of " << contacts.size()
|
|
<< " contacts with random_id = " << random_id;
|
|
if (random_id != 0) {
|
|
// request has already been sent before
|
|
if (need_clear_imported_contacts_) {
|
|
need_clear_imported_contacts_ = false;
|
|
all_imported_contacts_.clear();
|
|
if (G()->use_chat_info_database()) {
|
|
G()->td_db()->get_sqlite_pmc()->erase("user_imported_contacts", Auto());
|
|
}
|
|
reload_contacts(true);
|
|
}
|
|
|
|
CHECK(are_imported_contacts_changing_);
|
|
are_imported_contacts_changing_ = false;
|
|
|
|
auto unimported_contact_invites = std::move(unimported_contact_invites_);
|
|
unimported_contact_invites_.clear();
|
|
|
|
auto imported_contact_user_ids = std::move(imported_contact_user_ids_);
|
|
imported_contact_user_ids_.clear();
|
|
|
|
promise.set_value(Unit());
|
|
return {std::move(imported_contact_user_ids), std::move(unimported_contact_invites)};
|
|
}
|
|
|
|
if (are_imported_contacts_changing_) {
|
|
promise.set_error(Status::Error(400, "ChangeImportedContacts can be called only once at the same time"));
|
|
return {};
|
|
}
|
|
|
|
vector<size_t> new_contacts_unique_id(contacts.size());
|
|
vector<Contact> unique_new_contacts;
|
|
unique_new_contacts.reserve(contacts.size());
|
|
std::unordered_map<Contact, size_t, ContactHash, ContactEqual> different_new_contacts;
|
|
std::unordered_set<string, Hash<string>> different_new_phone_numbers;
|
|
size_t unique_size = 0;
|
|
for (size_t i = 0; i < contacts.size(); i++) {
|
|
auto it_success = different_new_contacts.emplace(std::move(contacts[i]), unique_size);
|
|
new_contacts_unique_id[i] = it_success.first->second;
|
|
if (it_success.second) {
|
|
unique_new_contacts.push_back(it_success.first->first);
|
|
different_new_phone_numbers.insert(unique_new_contacts.back().get_phone_number());
|
|
unique_size++;
|
|
}
|
|
}
|
|
|
|
vector<string> to_delete;
|
|
vector<UserId> to_delete_user_ids;
|
|
for (auto &old_contact : all_imported_contacts_) {
|
|
auto user_id = old_contact.get_user_id();
|
|
auto it = different_new_contacts.find(old_contact);
|
|
if (it == different_new_contacts.end()) {
|
|
auto phone_number = old_contact.get_phone_number();
|
|
if (different_new_phone_numbers.count(phone_number) == 0) {
|
|
to_delete.push_back(std::move(phone_number));
|
|
if (user_id.is_valid()) {
|
|
to_delete_user_ids.push_back(user_id);
|
|
}
|
|
}
|
|
} else {
|
|
unique_new_contacts[it->second].set_user_id(user_id);
|
|
different_new_contacts.erase(it);
|
|
}
|
|
}
|
|
std::pair<vector<size_t>, vector<Contact>> to_add;
|
|
for (auto &new_contact : different_new_contacts) {
|
|
to_add.first.push_back(new_contact.second);
|
|
to_add.second.push_back(new_contact.first);
|
|
}
|
|
|
|
if (to_add.first.empty() && to_delete.empty()) {
|
|
for (size_t i = 0; i < contacts.size(); i++) {
|
|
auto unique_id = new_contacts_unique_id[i];
|
|
contacts[i].set_user_id(unique_new_contacts[unique_id].get_user_id());
|
|
}
|
|
|
|
promise.set_value(Unit());
|
|
return {transform(contacts, [&](const Contact &contact) { return contact.get_user_id(); }),
|
|
vector<int32>(contacts.size())};
|
|
}
|
|
|
|
are_imported_contacts_changing_ = true;
|
|
random_id = 1;
|
|
|
|
remove_contacts_by_phone_number(
|
|
std::move(to_delete), std::move(to_delete_user_ids),
|
|
PromiseCreator::lambda([new_contacts = std::move(unique_new_contacts),
|
|
new_contacts_unique_id = std::move(new_contacts_unique_id), to_add = std::move(to_add),
|
|
promise = std::move(promise)](Result<> result) mutable {
|
|
if (result.is_ok()) {
|
|
send_closure_later(G()->user_manager(), &UserManager::on_clear_imported_contacts, std::move(new_contacts),
|
|
std::move(new_contacts_unique_id), std::move(to_add), std::move(promise));
|
|
} else {
|
|
promise.set_error(result.move_as_error());
|
|
}
|
|
}));
|
|
return {};
|
|
}
|
|
|
|
void UserManager::clear_imported_contacts(Promise<Unit> &&promise) {
|
|
LOG(INFO) << "Delete imported contacts";
|
|
|
|
if (saved_contact_count_ == 0) {
|
|
promise.set_value(Unit());
|
|
return;
|
|
}
|
|
|
|
td_->create_handler<ResetContactsQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
void UserManager::on_clear_imported_contacts(vector<Contact> &&contacts, vector<size_t> contacts_unique_id,
|
|
std::pair<vector<size_t>, vector<Contact>> &&to_add,
|
|
Promise<Unit> &&promise) {
|
|
LOG(INFO) << "Add " << to_add.first.size() << " contacts";
|
|
next_all_imported_contacts_ = std::move(contacts);
|
|
imported_contacts_unique_id_ = std::move(contacts_unique_id);
|
|
imported_contacts_pos_ = std::move(to_add.first);
|
|
|
|
do_import_contacts(std::move(to_add.second), 1, std::move(promise));
|
|
}
|
|
|
|
void UserManager::on_update_contacts_reset() {
|
|
/*
|
|
UserId my_id = get_my_id();
|
|
users_.foreach([&](const UserId &user_id, unique_ptr<User> &user) {
|
|
User *u = user.get();
|
|
if (u->is_contact) {
|
|
LOG(INFO) << "Drop contact with " << user_id;
|
|
if (user_id != my_id) {
|
|
CHECK(contacts_hints_.has_key(user_id.get()));
|
|
}
|
|
on_update_user_is_contact(u, user_id, false, false, false);
|
|
CHECK(u->is_is_contact_changed);
|
|
u->cache_version = 0;
|
|
u->is_repaired = false;
|
|
update_user(u, user_id);
|
|
CHECK(!u->is_contact);
|
|
if (user_id != my_id) {
|
|
CHECK(!contacts_hints_.has_key(user_id.get()));
|
|
}
|
|
}
|
|
});
|
|
*/
|
|
|
|
saved_contact_count_ = 0;
|
|
if (G()->use_chat_info_database()) {
|
|
G()->td_db()->get_binlog_pmc()->set("saved_contact_count", "0");
|
|
G()->td_db()->get_sqlite_pmc()->erase("user_imported_contacts", Auto());
|
|
}
|
|
if (!are_imported_contacts_loaded_) {
|
|
if (load_imported_contacts_queries_.empty()) {
|
|
CHECK(all_imported_contacts_.empty());
|
|
LOG(INFO) << "Imported contacts were never loaded, just clear them";
|
|
} else {
|
|
LOG(INFO) << "Imported contacts are being loaded, clear them after they will be loaded";
|
|
need_clear_imported_contacts_ = true;
|
|
}
|
|
} else {
|
|
if (!are_imported_contacts_changing_) {
|
|
LOG(INFO) << "Imported contacts were loaded, but aren't changing now, just clear them";
|
|
all_imported_contacts_.clear();
|
|
} else {
|
|
LOG(INFO) << "Imported contacts are changing now, clear them after they will be changed";
|
|
need_clear_imported_contacts_ = true;
|
|
}
|
|
}
|
|
reload_contacts(true);
|
|
}
|
|
|
|
void UserManager::update_contacts_hints(const User *u, UserId user_id, bool from_database) {
|
|
bool is_contact = is_user_contact(u, user_id, false);
|
|
if (td_->auth_manager_->is_bot()) {
|
|
LOG_IF(ERROR, is_contact) << "Bot has " << user_id << " in the contacts list";
|
|
return;
|
|
}
|
|
|
|
int64 key = user_id.get();
|
|
string old_value = contacts_hints_.key_to_string(key);
|
|
string new_value = is_contact ? get_user_search_text(u) : string();
|
|
|
|
if (new_value != old_value) {
|
|
if (is_contact) {
|
|
contacts_hints_.add(key, new_value);
|
|
} else {
|
|
contacts_hints_.remove(key);
|
|
}
|
|
}
|
|
|
|
if (G()->use_chat_info_database()) {
|
|
// update contacts database
|
|
if (!are_contacts_loaded_) {
|
|
if (!from_database && load_contacts_queries_.empty() && is_contact && u->is_is_contact_changed) {
|
|
search_contacts("", std::numeric_limits<int32>::max(), Auto());
|
|
}
|
|
} else {
|
|
if (old_value.empty() == is_contact) {
|
|
save_contacts_to_database();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
std::pair<int32, vector<UserId>> UserManager::search_contacts(const string &query, int32 limit,
|
|
Promise<Unit> &&promise) {
|
|
LOG(INFO) << "Search contacts with query = \"" << query << "\" and limit = " << limit;
|
|
|
|
if (limit < 0) {
|
|
promise.set_error(Status::Error(400, "Limit must be non-negative"));
|
|
return {};
|
|
}
|
|
|
|
if (!are_contacts_loaded_) {
|
|
load_contacts(std::move(promise));
|
|
return {};
|
|
}
|
|
reload_contacts(false);
|
|
|
|
std::pair<size_t, vector<int64>> result;
|
|
if (query.empty()) {
|
|
result = contacts_hints_.search_empty(limit);
|
|
} else {
|
|
result = contacts_hints_.search(query, limit);
|
|
}
|
|
|
|
vector<UserId> user_ids;
|
|
user_ids.reserve(result.second.size());
|
|
for (auto key : result.second) {
|
|
user_ids.emplace_back(key);
|
|
}
|
|
|
|
promise.set_value(Unit());
|
|
return {narrow_cast<int32>(result.first), std::move(user_ids)};
|
|
}
|
|
|
|
void UserManager::reload_contact_birthdates(bool force) {
|
|
if (td_->option_manager_->get_option_boolean("dismiss_birthday_contact_today")) {
|
|
contact_birthdates_.need_drop_ = true;
|
|
if (!contact_birthdates_.is_being_synced_) {
|
|
contact_birthdates_.is_being_synced_ = true;
|
|
on_get_contact_birthdates(nullptr);
|
|
}
|
|
return;
|
|
}
|
|
if (!G()->close_flag() && !td_->auth_manager_->is_bot() && !contact_birthdates_.is_being_synced_ &&
|
|
(contact_birthdates_.next_sync_time_ < Time::now() || force)) {
|
|
contact_birthdates_.is_being_synced_ = true;
|
|
td_->create_handler<GetContactsBirthdaysQuery>()->send();
|
|
}
|
|
}
|
|
|
|
void UserManager::on_get_contact_birthdates(
|
|
telegram_api::object_ptr<telegram_api::contacts_contactBirthdays> &&birthdays) {
|
|
CHECK(contact_birthdates_.is_being_synced_);
|
|
contact_birthdates_.is_being_synced_ = false;
|
|
if (contact_birthdates_.need_drop_) {
|
|
birthdays = telegram_api::make_object<telegram_api::contacts_contactBirthdays>(Auto(), Auto());
|
|
contact_birthdates_.need_drop_ = false;
|
|
}
|
|
if (birthdays == nullptr) {
|
|
contact_birthdates_.next_sync_time_ = Time::now() + Random::fast(120, 180);
|
|
return;
|
|
}
|
|
contact_birthdates_.next_sync_time_ = Time::now() + Random::fast(86400 / 4, 86400 / 3);
|
|
|
|
on_get_users(std::move(birthdays->users_), "on_get_contact_birthdates");
|
|
vector<std::pair<UserId, Birthdate>> users;
|
|
for (auto &contact : birthdays->contacts_) {
|
|
UserId user_id(contact->contact_id_);
|
|
if (is_user_contact(user_id)) {
|
|
Birthdate birthdate(std::move(contact->birthday_));
|
|
UserFull *user_full = get_user_full_force(user_id, "on_get_contact_birthdates");
|
|
if (user_full != nullptr && user_full->birthdate != birthdate) {
|
|
user_full->birthdate = birthdate;
|
|
user_full->is_changed = true;
|
|
update_user_full(user_full, user_id, "on_get_contact_birthdates");
|
|
}
|
|
if (!birthdate.is_empty()) {
|
|
users.emplace_back(user_id, birthdate);
|
|
}
|
|
}
|
|
}
|
|
if (contact_birthdates_.users_ != users) {
|
|
contact_birthdates_.users_ = std::move(users);
|
|
send_closure(G()->td(), &Td::send_update, get_update_contact_close_birthdays());
|
|
}
|
|
// there is no need to save them between restarts
|
|
}
|
|
|
|
void UserManager::hide_contact_birthdays(Promise<Unit> &&promise) {
|
|
td_->create_handler<DismissContactBirthdaysSuggestionQuery>(std::move(promise))->send();
|
|
}
|
|
|
|
vector<UserId> UserManager::get_close_friends(Promise<Unit> &&promise) {
|
|
if (!are_contacts_loaded_) {
|
|
load_contacts(std::move(promise));
|
|
return {};
|
|
}
|
|
reload_contacts(false);
|
|
|
|
auto result = contacts_hints_.search_empty(10000);
|
|
|
|
vector<UserId> user_ids;
|
|
for (auto key : result.second) {
|
|
UserId user_id(key);
|
|
const User *u = get_user(user_id);
|
|
if (u != nullptr && u->is_close_friend) {
|
|
user_ids.push_back(user_id);
|
|
}
|
|
}
|
|
|
|
promise.set_value(Unit());
|
|
return user_ids;
|
|
}
|
|
|
|
void UserManager::set_close_friends(vector<UserId> user_ids, Promise<Unit> &&promise) {
|
|
for (auto &user_id : user_ids) {
|
|
if (!have_user(user_id)) {
|
|
return promise.set_error(Status::Error(400, "User not found"));
|
|
}
|
|
}
|
|
|
|
td_->create_handler<EditCloseFriendsQuery>(std::move(promise))->send(std::move(user_ids));
|
|
}
|
|
|
|
void UserManager::on_set_close_friends(const vector<UserId> &user_ids, Promise<Unit> &&promise) {
|
|
FlatHashSet<UserId, UserIdHash> close_friend_user_ids;
|
|
for (auto &user_id : user_ids) {
|
|
CHECK(user_id.is_valid());
|
|
close_friend_user_ids.insert(user_id);
|
|
}
|
|
users_.foreach([&](const UserId &user_id, unique_ptr<User> &user) {
|
|
User *u = user.get();
|
|
if (u->is_contact && u->is_close_friend != (close_friend_user_ids.count(user_id) > 0)) {
|
|
on_update_user_is_contact(u, user_id, u->is_contact, u->is_mutual_contact, !u->is_close_friend);
|
|
update_user(u, user_id);
|
|
}
|
|
});
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
UserId UserManager::search_user_by_phone_number(string phone_number, bool only_local, Promise<Unit> &&promise) {
|
|
clean_phone_number(phone_number);
|
|
if (phone_number.empty()) {
|
|
promise.set_error(Status::Error(200, "Phone number is invalid"));
|
|
return UserId();
|
|
}
|
|
|
|
auto it = resolved_phone_numbers_.find(phone_number);
|
|
if (it != resolved_phone_numbers_.end()) {
|
|
promise.set_value(Unit());
|
|
return it->second;
|
|
}
|
|
|
|
if (only_local) {
|
|
promise.set_value(Unit());
|
|
} else {
|
|
td_->create_handler<ResolvePhoneQuery>(std::move(promise))->send(phone_number);
|
|
}
|
|
return UserId();
|
|
}
|
|
|
|
void UserManager::on_resolved_phone_number(const string &phone_number, UserId user_id) {
|
|
if (!user_id.is_valid()) {
|
|
resolved_phone_numbers_.emplace(phone_number, UserId()); // negative cache
|
|
return;
|
|
}
|
|
|
|
auto it = resolved_phone_numbers_.find(phone_number);
|
|
if (it != resolved_phone_numbers_.end()) {
|
|
if (it->second != user_id) {
|
|
LOG(WARNING) << "Resolve phone number \"" << phone_number << "\" to " << user_id << ", but have it in "
|
|
<< it->second;
|
|
it->second = user_id;
|
|
}
|
|
return;
|
|
}
|
|
|
|
auto *u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
LOG(ERROR) << "Resolve phone number \"" << phone_number << "\" to unknown " << user_id;
|
|
} else if (!u->phone_number.empty()) {
|
|
LOG(ERROR) << "Resolve phone number \"" << phone_number << "\" to " << user_id << " with phone number "
|
|
<< u->phone_number;
|
|
} else {
|
|
// the user's phone number can be hidden by privacy settings, despite the user can be found by the phone number
|
|
}
|
|
resolved_phone_numbers_[phone_number] = user_id; // always update cached value
|
|
}
|
|
|
|
const UserManager::UserFull *UserManager::get_user_full(UserId user_id) const {
|
|
return users_full_.get_pointer(user_id);
|
|
}
|
|
|
|
UserManager::UserFull *UserManager::get_user_full(UserId user_id) {
|
|
return users_full_.get_pointer(user_id);
|
|
}
|
|
|
|
UserManager::UserFull *UserManager::add_user_full(UserId user_id) {
|
|
CHECK(user_id.is_valid());
|
|
auto &user_full_ptr = users_full_[user_id];
|
|
if (user_full_ptr == nullptr) {
|
|
user_full_ptr = make_unique<UserFull>();
|
|
user_full_contact_require_premium_.erase(user_id);
|
|
}
|
|
return user_full_ptr.get();
|
|
}
|
|
|
|
UserManager::UserFull *UserManager::get_user_full_force(UserId user_id, const char *source) {
|
|
if (!have_user_force(user_id, source)) {
|
|
return nullptr;
|
|
}
|
|
|
|
UserFull *user_full = get_user_full(user_id);
|
|
if (user_full != nullptr) {
|
|
return user_full;
|
|
}
|
|
if (!G()->use_chat_info_database()) {
|
|
return nullptr;
|
|
}
|
|
if (!unavailable_user_fulls_.insert(user_id).second) {
|
|
return nullptr;
|
|
}
|
|
|
|
LOG(INFO) << "Trying to load full " << user_id << " from database from " << source;
|
|
on_load_user_full_from_database(user_id,
|
|
G()->td_db()->get_sqlite_sync_pmc()->get(get_user_full_database_key(user_id)));
|
|
return get_user_full(user_id);
|
|
}
|
|
|
|
void UserManager::load_user_full(UserId user_id, bool force, Promise<Unit> &&promise, const char *source) {
|
|
auto u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
return promise.set_error(Status::Error(400, "User not found"));
|
|
}
|
|
|
|
auto user_full = get_user_full_force(user_id, source);
|
|
if (user_full == nullptr) {
|
|
TRY_RESULT_PROMISE(promise, input_user, get_input_user(user_id));
|
|
return send_get_user_full_query(user_id, std::move(input_user), std::move(promise), source);
|
|
}
|
|
if (user_full->is_expired()) {
|
|
auto input_user = get_input_user_force(user_id);
|
|
if (td_->auth_manager_->is_bot() && !force) {
|
|
return send_get_user_full_query(user_id, std::move(input_user), std::move(promise), "load expired user_full");
|
|
}
|
|
|
|
send_get_user_full_query(user_id, std::move(input_user), Auto(), "load expired user_full");
|
|
}
|
|
|
|
td_->story_manager_->on_view_dialog_active_stories({DialogId(user_id)});
|
|
promise.set_value(Unit());
|
|
}
|
|
|
|
void UserManager::reload_user_full(UserId user_id, Promise<Unit> &&promise, const char *source) {
|
|
TRY_RESULT_PROMISE(promise, input_user, get_input_user(user_id));
|
|
send_get_user_full_query(user_id, std::move(input_user), std::move(promise), source);
|
|
}
|
|
|
|
void UserManager::send_get_user_full_query(UserId user_id,
|
|
telegram_api::object_ptr<telegram_api::InputUser> &&input_user,
|
|
Promise<Unit> &&promise, const char *source) {
|
|
LOG(INFO) << "Get full " << user_id << " from " << source;
|
|
if (!user_id.is_valid()) {
|
|
return promise.set_error(Status::Error(500, "Invalid user_id"));
|
|
}
|
|
auto send_query =
|
|
PromiseCreator::lambda([td = td_, input_user = std::move(input_user)](Result<Promise<Unit>> &&promise) mutable {
|
|
if (promise.is_ok() && !G()->close_flag()) {
|
|
td->create_handler<GetFullUserQuery>(promise.move_as_ok())->send(std::move(input_user));
|
|
}
|
|
});
|
|
get_user_full_queries_.add_query(user_id.get(), std::move(send_query), std::move(promise));
|
|
}
|
|
|
|
void UserManager::on_get_user_full(telegram_api::object_ptr<telegram_api::userFull> &&user) {
|
|
LOG(INFO) << "Receive " << to_string(user);
|
|
|
|
UserId user_id(user->id_);
|
|
User *u = get_user(user_id);
|
|
if (u == nullptr) {
|
|
LOG(ERROR) << "Failed to find " << user_id;
|
|
return;
|
|
}
|
|
|
|
apply_pending_user_photo(u, user_id);
|
|
|
|
td_->messages_manager_->on_update_dialog_notify_settings(DialogId(user_id), std::move(user->notify_settings_),
|
|
"on_get_user_full");
|
|
|
|
td_->messages_manager_->on_update_dialog_background(DialogId(user_id), std::move(user->wallpaper_));
|
|
|
|
td_->messages_manager_->on_update_dialog_theme_name(DialogId(user_id), std::move(user->theme_emoticon_));
|
|
|
|
td_->messages_manager_->on_update_dialog_last_pinned_message_id(DialogId(user_id),
|
|
MessageId(ServerMessageId(user->pinned_msg_id_)));
|
|
|
|
td_->messages_manager_->on_update_dialog_folder_id(DialogId(user_id), FolderId(user->folder_id_));
|
|
|
|
td_->messages_manager_->on_update_dialog_has_scheduled_server_messages(DialogId(user_id), user->has_scheduled_);
|
|
|
|
td_->messages_manager_->on_update_dialog_message_ttl(DialogId(user_id), MessageTtl(user->ttl_period_));
|
|
|
|
td_->messages_manager_->on_update_dialog_is_blocked(DialogId(user_id), user->blocked_,
|
|
user->blocked_my_stories_from_);
|
|
|
|
td_->messages_manager_->on_update_dialog_is_translatable(DialogId(user_id), !user->translations_disabled_);
|
|
|
|
send_closure_later(td_->story_manager_actor_, &StoryManager::on_get_dialog_stories, DialogId(user_id),
|
|
std::move(user->stories_), Promise<Unit>());
|
|
|
|
UserFull *user_full = add_user_full(user_id);
|
|
user_full->expires_at = Time::now() + USER_FULL_EXPIRE_TIME;
|
|
|
|
on_update_user_full_is_blocked(user_full, user_id, user->blocked_, user->blocked_my_stories_from_);
|
|
on_update_user_full_common_chat_count(user_full, user_id, user->common_chats_count_);
|
|
on_update_user_full_location(user_full, user_id, DialogLocation(td_, std::move(user->business_location_)));
|
|
on_update_user_full_work_hours(user_full, user_id, BusinessWorkHours(std::move(user->business_work_hours_)));
|
|
on_update_user_full_away_message(user_full, user_id, BusinessAwayMessage(std::move(user->business_away_message_)));
|
|
on_update_user_full_greeting_message(user_full, user_id,
|
|
BusinessGreetingMessage(std::move(user->business_greeting_message_)));
|
|
on_update_user_full_intro(user_full, user_id, BusinessIntro(td_, std::move(user->business_intro_)));
|
|
on_update_user_full_need_phone_number_privacy_exception(user_full, user_id,
|
|
user->settings_->need_contacts_exception_);
|
|
on_update_user_full_wallpaper_overridden(user_full, user_id, user->wallpaper_overridden_);
|
|
|
|
bool can_pin_messages = user->can_pin_message_;
|
|
bool can_be_called = user->phone_calls_available_ && !user->phone_calls_private_;
|
|
bool supports_video_calls = user->video_calls_available_ && !user->phone_calls_private_;
|
|
bool has_private_calls = user->phone_calls_private_;
|
|
bool voice_messages_forbidden = u->is_premium ? user->voice_messages_forbidden_ : false;
|
|
auto premium_gift_options = get_premium_gift_options(std::move(user->premium_gifts_));
|
|
AdministratorRights group_administrator_rights(user->bot_group_admin_rights_, ChannelType::Megagroup);
|
|
AdministratorRights broadcast_administrator_rights(user->bot_broadcast_admin_rights_, ChannelType::Broadcast);
|
|
bool has_pinned_stories = user->stories_pinned_available_;
|
|
auto birthdate = Birthdate(std::move(user->birthday_));
|
|
auto personal_channel_id = ChannelId(user->personal_channel_id_);
|
|
auto sponsored_enabled = user->sponsored_enabled_;
|
|
if (user_full->can_be_called != can_be_called || user_full->supports_video_calls != supports_video_calls ||
|
|
user_full->has_private_calls != has_private_calls ||
|
|
user_full->group_administrator_rights != group_administrator_rights ||
|
|
user_full->broadcast_administrator_rights != broadcast_administrator_rights ||
|
|
user_full->premium_gift_options != premium_gift_options ||
|
|
user_full->voice_messages_forbidden != voice_messages_forbidden ||
|
|
user_full->can_pin_messages != can_pin_messages || user_full->has_pinned_stories != has_pinned_stories ||
|
|
user_full->sponsored_enabled != sponsored_enabled) {
|
|
user_full->can_be_called = can_be_called;
|
|
user_full->supports_video_calls = supports_video_calls;
|
|
user_full->has_private_calls = has_private_calls;
|
|
user_full->group_administrator_rights = group_administrator_rights;
|
|
user_full->broadcast_administrator_rights = broadcast_administrator_rights;
|
|
user_full->premium_gift_options = std::move(premium_gift_options);
|
|
user_full->voice_messages_forbidden = voice_messages_forbidden;
|
|
user_full->can_pin_messages = can_pin_messages;
|
|
user_full->has_pinned_stories = has_pinned_stories;
|
|
user_full->sponsored_enabled = sponsored_enabled;
|
|
|
|
user_full->is_changed = true;
|
|
}
|
|
if (user_full->birthdate != birthdate) {
|
|
user_full->birthdate = birthdate;
|
|
user_full->is_changed = true;
|
|
|
|
if (u->is_mutual_contact) {
|
|
reload_contact_birthdates(true);
|
|
}
|
|
}
|
|
|
|
if (user_full->private_forward_name != user->private_forward_name_) {
|
|
if (user_full->private_forward_name.empty() != user->private_forward_name_.empty()) {
|
|
user_full->is_changed = true;
|
|
}
|
|
user_full->private_forward_name = std::move(user->private_forward_name_);
|
|
user_full->need_save_to_database = true;
|
|
}
|
|
if (user_full->read_dates_private != user->read_dates_private_ ||
|
|
user_full->contact_require_premium != user->contact_require_premium_) {
|
|
user_full->read_dates_private = user->read_dates_private_;
|
|
user_full->contact_require_premium = user->contact_require_premium_;
|
|
user_full->need_save_to_database = true;
|
|
}
|
|
if (user_full->about != user->about_) {
|
|
user_full->about = std::move(user->about_);
|
|
user_full->is_changed = true;
|
|
td_->group_call_manager_->on_update_dialog_about(DialogId(user_id), user_full->about, true);
|
|
}
|
|
string description;
|
|
Photo description_photo;
|
|
FileId description_animation_file_id;
|
|
if (user->bot_info_ != nullptr && !td_->auth_manager_->is_bot()) {
|
|
description = std::move(user->bot_info_->description_);
|
|
description_photo = get_photo(td_, std::move(user->bot_info_->description_photo_), DialogId(user_id));
|
|
auto document = std::move(user->bot_info_->description_document_);
|
|
if (document != nullptr) {
|
|
int32 document_id = document->get_id();
|
|
if (document_id == telegram_api::document::ID) {
|
|
auto parsed_document = td_->documents_manager_->on_get_document(
|
|
move_tl_object_as<telegram_api::document>(document), DialogId(user_id));
|
|
if (parsed_document.type == Document::Type::Animation) {
|
|
description_animation_file_id = parsed_document.file_id;
|
|
} else {
|
|
LOG(ERROR) << "Receive non-animation document in bot description";
|
|
}
|
|
}
|
|
}
|
|
|
|
on_update_user_full_commands(user_full, user_id, std::move(user->bot_info_->commands_));
|
|
on_update_user_full_menu_button(user_full, user_id, std::move(user->bot_info_->menu_button_));
|
|
on_update_user_full_has_preview_medias(user_full, user_id, std::move(user->bot_info_->has_preview_medias_));
|
|
}
|
|
if (user_full->description != description) {
|
|
user_full->description = std::move(description);
|
|
user_full->is_changed = true;
|
|
}
|
|
if (user_full->description_photo != description_photo ||
|
|
user_full->description_animation_file_id != description_animation_file_id) {
|
|
user_full->description_photo = std::move(description_photo);
|
|
user_full->description_animation_file_id = description_animation_file_id;
|
|
user_full->is_changed = true;
|
|
}
|
|
if (personal_channel_id != ChannelId() &&
|
|
td_->chat_manager_->get_channel_type(personal_channel_id) != ChannelType::Broadcast) {
|
|
LOG(ERROR) << "Receive personal " << personal_channel_id << " of the type "
|
|
<< static_cast<uint8>(td_->chat_manager_->get_channel_type(personal_channel_id));
|
|
personal_channel_id = ChannelId();
|
|
}
|
|
if (user_full->personal_channel_id != personal_channel_id) {
|
|
user_full->personal_channel_id = personal_channel_id;
|
|
user_full->is_changed = true;
|
|
}
|
|
if (user_full->personal_channel_id != ChannelId()) {
|
|
auto personal_message_id = MessageId(ServerMessageId(user->personal_channel_message_));
|
|
td_->messages_manager_->get_channel_difference_if_needed(DialogId(user_full->personal_channel_id),
|
|
personal_message_id, "on_get_user_full personal chat");
|
|
}
|
|
|
|
auto photo = get_photo(td_, std::move(user->profile_photo_), DialogId(user_id));
|
|
auto personal_photo = get_photo(td_, std::move(user->personal_photo_), DialogId(user_id));
|
|
auto fallback_photo = get_photo(td_, std::move(user->fallback_photo_), DialogId(user_id));
|
|
// do_update_user_photo should be a no-op if server sent consistent data
|
|
const Photo *photo_ptr = nullptr;
|
|
bool is_personal = false;
|
|
if (!personal_photo.is_empty()) {
|
|
photo_ptr = &personal_photo;
|
|
is_personal = true;
|
|
} else if (!photo.is_empty()) {
|
|
photo_ptr = &photo;
|
|
} else {
|
|
photo_ptr = &fallback_photo;
|
|
}
|
|
bool is_photo_empty = photo_ptr->is_empty();
|
|
do_update_user_photo(u, user_id,
|
|
as_profile_photo(td_->file_manager_.get(), user_id, u->access_hash, *photo_ptr, is_personal),
|
|
false, "on_get_user_full");
|
|
if (photo != user_full->photo) {
|
|
user_full->photo = std::move(photo);
|
|
user_full->is_changed = true;
|
|
}
|
|
if (personal_photo != user_full->personal_photo) {
|
|
user_full->personal_photo = std::move(personal_photo);
|
|
user_full->is_changed = true;
|
|
}
|
|
if (fallback_photo != user_full->fallback_photo) {
|
|
user_full->fallback_photo = std::move(fallback_photo);
|
|
user_full->is_changed = true;
|
|
}
|
|
if (!user_full->photo.is_empty()) {
|
|
register_user_photo(u, user_id, user_full->photo);
|
|
}
|
|
if (user_id == get_my_id() && !user_full->fallback_photo.is_empty()) {
|
|
register_suggested_profile_photo(user_full->fallback_photo);
|
|
}
|
|
if (is_photo_empty) {
|
|
drop_user_photos(user_id, true, "on_get_user_full");
|
|
}
|
|
|
|
// User must be updated before UserFull
|
|
if (u->is_changed) {
|
|
LOG(ERROR) << "Receive inconsistent chatPhoto and chatPhotoInfo for " << user_id;
|
|
update_user(u, user_id);
|
|
}
|
|
|
|
user_full->is_update_user_full_sent = true;
|
|
update_user_full(user_full, user_id, "on_get_user_full");
|
|
|
|
// update peer settings after UserFull is created and updated to not update twice need_phone_number_privacy_exception
|
|
td_->messages_manager_->on_get_peer_settings(DialogId(user_id), std::move(user->settings_));
|
|
}
|
|
|
|
FileSourceId UserManager::get_user_full_file_source_id(UserId user_id) {
|
|
if (!user_id.is_valid()) {
|
|
return FileSourceId();
|
|
}
|
|
|
|
auto user_full = get_user_full(user_id);
|
|
if (user_full != nullptr) {
|
|
VLOG(file_references) << "Don't need to create file source for full " << user_id;
|
|
// user full was already added, source ID was registered and shouldn't be needed
|
|
return user_full->is_update_user_full_sent ? FileSourceId() : user_full->file_source_id;
|
|
}
|
|
|
|
auto &source_id = user_full_file_source_ids_[user_id];
|
|
if (!source_id.is_valid()) {
|
|
source_id = td_->file_reference_manager_->create_user_full_file_source(user_id);
|
|
}
|
|
VLOG(file_references) << "Return " << source_id << " for full " << user_id;
|
|
return source_id;
|
|
}
|
|
|
|
void UserManager::save_user_full(const UserFull *user_full, UserId user_id) {
|
|
if (!G()->use_chat_info_database()) {
|
|
return;
|
|
}
|
|
|
|
LOG(INFO) << "Trying to save to database full " << user_id;
|
|
CHECK(user_full != nullptr);
|
|
G()->td_db()->get_sqlite_pmc()->set(get_user_full_database_key(user_id), get_user_full_database_value(user_full),
|
|
Auto());
|
|
}
|
|
|
|
string UserManager::get_user_full_database_key(UserId user_id) {
|
|
return PSTRING() << "usf" << user_id.get();
|
|
}
|
|
|
|
string UserManager::get_user_full_database_value(const UserFull *user_full) {
|
|
return log_event_store(*user_full).as_slice().str();
|
|
}
|
|
|
|
void UserManager::on_load_user_full_from_database(UserId user_id, string value) {
|
|
LOG(INFO) << "Successfully loaded full " << user_id << " of size " << value.size() << " from database";
|
|
// G()->td_db()->get_sqlite_pmc()->erase(get_user_full_database_key(user_id), Auto());
|
|
// return;
|
|
|
|
if (get_user_full(user_id) != nullptr || value.empty()) {
|
|
return;
|
|
}
|
|
|
|
UserFull *user_full = add_user_full(user_id);
|
|
auto status = log_event_parse(*user_full, value);
|
|
if (status.is_error()) {
|
|
// can't happen unless database is broken
|
|
LOG(ERROR) << "Repair broken full " << user_id << ' ' << format::as_hex_dump<4>(Slice(value));
|
|
|
|
// just clean all known data about the user and pretend that there was nothing in the database
|
|
users_full_.erase(user_id);
|
|
G()->td_db()->get_sqlite_pmc()->erase(get_user_full_database_key(user_id), Auto());
|
|
return;
|
|
}
|
|
|
|
Dependencies dependencies;
|
|
dependencies.add(user_id);
|
|
if (user_full->business_info != nullptr) {
|
|
user_full->business_info->add_dependencies(dependencies);
|
|
}
|
|
dependencies.add(user_full->personal_channel_id);
|
|
if (!dependencies.resolve_force(td_, "on_load_user_full_from_database")) {
|
|
users_full_.erase(user_id);
|
|
G()->td_db()->get_sqlite_pmc()->erase(get_user_full_database_key(user_id), Auto());
|
|
return;
|
|
}
|
|
|
|
if (user_full->need_phone_number_privacy_exception && is_user_contact(user_id)) {
|
|
user_full->need_phone_number_privacy_exception = false;
|
|
}
|
|
|
|
User *u = get_user(user_id);
|
|
CHECK(u != nullptr);
|
|
drop_user_full_photos(user_full, user_id, u->photo.id, "on_load_user_full_from_database");
|
|
if (!user_full->photo.is_empty()) {
|
|
register_user_photo(u, user_id, user_full->photo);
|
|
}
|
|
if (user_id == get_my_id() && !user_full->fallback_photo.is_empty()) {
|
|
register_suggested_profile_photo(user_full->fallback_photo);
|
|
}
|
|
|
|
td_->group_call_manager_->on_update_dialog_about(DialogId(user_id), user_full->about, false);
|
|
|
|
user_full->is_update_user_full_sent = true;
|
|
update_user_full(user_full, user_id, "on_load_user_full_from_database", true);
|
|
|
|
if (is_user_deleted(u)) {
|
|
drop_user_full(user_id);
|
|
} else if (user_full->expires_at == 0.0) {
|
|
reload_user_full(user_id, Auto(), "on_load_user_full_from_database");
|
|
}
|
|
}
|
|
|
|
int64 UserManager::get_user_full_profile_photo_id(const UserFull *user_full) {
|
|
if (!user_full->personal_photo.is_empty()) {
|
|
return user_full->personal_photo.id.get();
|
|
}
|
|
if (!user_full->photo.is_empty()) {
|
|
return user_full->photo.id.get();
|
|
}
|
|
return user_full->fallback_photo.id.get();
|
|
}
|
|
|
|
void UserManager::drop_user_full_photos(UserFull *user_full, UserId user_id, int64 expected_photo_id,
|
|
const char *source) {
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
LOG(INFO) << "Expect full photo " << expected_photo_id << " from " << source;
|
|
for (auto photo_ptr : {&user_full->personal_photo, &user_full->photo, &user_full->fallback_photo}) {
|
|
if (photo_ptr->is_empty()) {
|
|
continue;
|
|
}
|
|
if (expected_photo_id == 0) {
|
|
// if profile photo is empty, we must drop the full photo
|
|
*photo_ptr = Photo();
|
|
user_full->is_changed = true;
|
|
} else if (expected_photo_id != photo_ptr->id.get()) {
|
|
LOG(INFO) << "Drop full photo " << photo_ptr->id.get();
|
|
// if full profile photo is unknown, we must drop the full photo
|
|
*photo_ptr = Photo();
|
|
user_full->is_changed = true;
|
|
} else {
|
|
// nothing to drop
|
|
break;
|
|
}
|
|
}
|
|
if (expected_photo_id != get_user_full_profile_photo_id(user_full)) {
|
|
user_full->expires_at = 0.0;
|
|
}
|
|
if (user_full->is_update_user_full_sent) {
|
|
update_user_full(user_full, user_id, "drop_user_full_photos");
|
|
}
|
|
}
|
|
|
|
void UserManager::drop_user_photos(UserId user_id, bool is_empty, const char *source) {
|
|
LOG(INFO) << "Drop user photos to " << (is_empty ? "empty" : "unknown") << " from " << source;
|
|
auto user_photos = user_photos_.get_pointer(user_id);
|
|
if (user_photos != nullptr) {
|
|
int32 new_count = is_empty ? 0 : -1;
|
|
if (user_photos->count == new_count) {
|
|
CHECK(user_photos->photos.empty());
|
|
CHECK(user_photos->offset == user_photos->count);
|
|
} else {
|
|
LOG(INFO) << "Drop photos of " << user_id << " to " << (is_empty ? "empty" : "unknown") << " from " << source;
|
|
user_photos->photos.clear();
|
|
user_photos->count = new_count;
|
|
user_photos->offset = user_photos->count;
|
|
}
|
|
}
|
|
}
|
|
|
|
void UserManager::drop_user_full(UserId user_id) {
|
|
auto user_full = get_user_full_force(user_id, "drop_user_full");
|
|
|
|
drop_user_photos(user_id, false, "drop_user_full");
|
|
|
|
if (user_full == nullptr) {
|
|
return;
|
|
}
|
|
|
|
user_full->expires_at = 0.0;
|
|
|
|
user_full->photo = Photo();
|
|
user_full->personal_photo = Photo();
|
|
user_full->fallback_photo = Photo();
|
|
// user_full->is_blocked = false;
|
|
// user_full->is_blocked_for_stories = false;
|
|
user_full->can_be_called = false;
|
|
user_full->supports_video_calls = false;
|
|
user_full->has_private_calls = false;
|
|
user_full->need_phone_number_privacy_exception = false;
|
|
user_full->wallpaper_overridden = false;
|
|
user_full->about = string();
|
|
user_full->description = string();
|
|
user_full->description_photo = Photo();
|
|
user_full->description_animation_file_id = FileId();
|
|
user_full->menu_button = nullptr;
|
|
user_full->commands.clear();
|
|
user_full->common_chat_count = 0;
|
|
user_full->personal_channel_id = ChannelId();
|
|
user_full->business_info = nullptr;
|
|
user_full->private_forward_name.clear();
|
|
user_full->group_administrator_rights = {};
|
|
user_full->broadcast_administrator_rights = {};
|
|
user_full->premium_gift_options.clear();
|
|
user_full->voice_messages_forbidden = false;
|
|
user_full->has_pinned_stories = false;
|
|
user_full->read_dates_private = false;
|
|
user_full->contact_require_premium = false;
|
|
user_full->birthdate = {};
|
|
user_full->sponsored_enabled = false;
|
|
user_full->has_preview_medias = false;
|
|
user_full->is_changed = true;
|
|
|
|
update_user_full(user_full, user_id, "drop_user_full");
|
|
td_->group_call_manager_->on_update_dialog_about(DialogId(user_id), user_full->about, true);
|
|
}
|
|
|
|
bool UserManager::have_secret_chat(SecretChatId secret_chat_id) const {
|
|
return secret_chats_.count(secret_chat_id) > 0;
|
|
}
|
|
|
|
const UserManager::SecretChat *UserManager::get_secret_chat(SecretChatId secret_chat_id) const {
|
|
return secret_chats_.get_pointer(secret_chat_id);
|
|
}
|
|
|
|
UserManager::SecretChat *UserManager::get_secret_chat(SecretChatId secret_chat_id) {
|
|
return secret_chats_.get_pointer(secret_chat_id);
|
|
}
|
|
|
|
UserManager::SecretChat *UserManager::add_secret_chat(SecretChatId secret_chat_id) {
|
|
CHECK(secret_chat_id.is_valid());
|
|
auto &secret_chat_ptr = secret_chats_[secret_chat_id];
|
|
if (secret_chat_ptr == nullptr) {
|
|
secret_chat_ptr = make_unique<SecretChat>();
|
|
}
|
|
return secret_chat_ptr.get();
|
|
}
|
|
|
|
bool UserManager::have_secret_chat_force(SecretChatId secret_chat_id, const char *source) {
|
|
return get_secret_chat_force(secret_chat_id, source) != nullptr;
|
|
}
|
|
|
|
UserManager::SecretChat *UserManager::get_secret_chat_force(SecretChatId secret_chat_id, const char *source) {
|
|
if (!secret_chat_id.is_valid()) {
|
|
return nullptr;
|
|
}
|
|
|
|
SecretChat *c = get_secret_chat(secret_chat_id);
|
|
if (c != nullptr) {
|
|
if (!have_user_force(c->user_id, source)) {
|
|
LOG(ERROR) << "Can't find " << c->user_id << " from " << secret_chat_id << " from " << source;
|
|
}
|
|
return c;
|
|
}
|
|
if (!G()->use_chat_info_database()) {
|
|
return nullptr;
|
|
}
|
|
if (loaded_from_database_secret_chats_.count(secret_chat_id)) {
|
|
return nullptr;
|
|
}
|
|
|
|
LOG(INFO) << "Trying to load " << secret_chat_id << " from database from " << source;
|
|
on_load_secret_chat_from_database(
|
|
secret_chat_id, G()->td_db()->get_sqlite_sync_pmc()->get(get_secret_chat_database_key(secret_chat_id)), true);
|
|
return get_secret_chat(secret_chat_id);
|
|
}
|
|
|
|
bool UserManager::get_secret_chat(SecretChatId secret_chat_id, bool force, Promise<Unit> &&promise) {
|
|
if (!secret_chat_id.is_valid()) {
|
|
promise.set_error(Status::Error(400, "Invalid secret chat identifier"));
|
|
return false;
|
|
}
|
|
|
|
if (!have_secret_chat(secret_chat_id)) {
|
|
if (!force && G()->use_chat_info_database()) {
|
|
send_closure_later(actor_id(this), &UserManager::load_secret_chat_from_database, nullptr, secret_chat_id,
|
|
std::move(promise));
|
|
return false;
|
|
}
|
|
|
|
promise.set_error(Status::Error(400, "Secret chat not found"));
|
|
return false;
|
|
}
|
|
|
|
promise.set_value(Unit());
|
|
return true;
|
|
}
|
|
|
|
void UserManager::save_secret_chat(SecretChat *c, SecretChatId secret_chat_id, bool from_binlog) {
|
|
if (!G()->use_chat_info_database()) {
|
|
return;
|
|
}
|
|
CHECK(c != nullptr);
|
|
if (!c->is_saved) {
|
|
if (!from_binlog) {
|
|
auto log_event = SecretChatLogEvent(secret_chat_id, c);
|
|
auto storer = get_log_event_storer(log_event);
|
|
if (c->log_event_id == 0) {
|
|
c->log_event_id = binlog_add(G()->td_db()->get_binlog(), LogEvent::HandlerType::SecretChatInfos, storer);
|
|
} else {
|
|
binlog_rewrite(G()->td_db()->get_binlog(), c->log_event_id, LogEvent::HandlerType::SecretChatInfos, storer);
|
|
}
|
|
}
|
|
|
|
save_secret_chat_to_database(c, secret_chat_id);
|
|
return;
|
|
}
|
|
}
|
|
|
|
string UserManager::get_secret_chat_database_key(SecretChatId secret_chat_id) {
|
|
return PSTRING() << "sc" << secret_chat_id.get();
|
|
}
|
|
|
|
string UserManager::get_secret_chat_database_value(const SecretChat *c) {
|
|
return log_event_store(*c).as_slice().str();
|
|
}
|
|
|
|
void UserManager::save_secret_chat_to_database(SecretChat *c, SecretChatId secret_chat_id) {
|
|
CHECK(c != nullptr);
|
|
if (c->is_being_saved) {
|
|
return;
|
|
}
|
|
if (loaded_from_database_secret_chats_.count(secret_chat_id)) {
|
|
save_secret_chat_to_database_impl(c, secret_chat_id, get_secret_chat_database_value(c));
|
|
return;
|
|
}
|
|
if (load_secret_chat_from_database_queries_.count(secret_chat_id) != 0) {
|
|
return;
|
|
}
|
|
|
|
load_secret_chat_from_database_impl(secret_chat_id, Auto());
|
|
}
|
|
|
|
void UserManager::save_secret_chat_to_database_impl(SecretChat *c, SecretChatId secret_chat_id, string value) {
|
|
CHECK(c != nullptr);
|
|
CHECK(load_secret_chat_from_database_queries_.count(secret_chat_id) == 0);
|
|
CHECK(!c->is_being_saved);
|
|
c->is_being_saved = true;
|
|
c->is_saved = true;
|
|
LOG(INFO) << "Trying to save to database " << secret_chat_id;
|
|
G()->td_db()->get_sqlite_pmc()->set(get_secret_chat_database_key(secret_chat_id), std::move(value),
|
|
PromiseCreator::lambda([secret_chat_id](Result<> result) {
|
|
send_closure(G()->user_manager(), &UserManager::on_save_secret_chat_to_database,
|
|
secret_chat_id, result.is_ok());
|
|
}));
|
|
}
|
|
|
|
void UserManager::on_save_secret_chat_to_database(SecretChatId secret_chat_id, bool success) {
|
|
if (G()->close_flag()) {
|
|
return;
|
|
}
|
|
|
|
SecretChat *c = get_secret_chat(secret_chat_id);
|
|
CHECK(c != nullptr);
|
|
CHECK(c->is_being_saved);
|
|
CHECK(load_secret_chat_from_database_queries_.count(secret_chat_id) == 0);
|
|
c->is_being_saved = false;
|
|
|
|
if (!success) {
|
|
LOG(ERROR) << "Failed to save " << secret_chat_id << " to database";
|
|
c->is_saved = false;
|
|
} else {
|
|
LOG(INFO) << "Successfully saved " << secret_chat_id << " to database";
|
|
}
|
|
if (c->is_saved) {
|
|
if (c->log_event_id != 0) {
|
|
binlog_erase(G()->td_db()->get_binlog(), c->log_event_id);
|
|
c->log_event_id = 0;
|
|
}
|
|
} else {
|
|
save_secret_chat(c, secret_chat_id, c->log_event_id != 0);
|
|
}
|
|
}
|
|
|
|
void UserManager::load_secret_chat_from_database(SecretChat *c, SecretChatId secret_chat_id, Promise<Unit> promise) {
|
|
if (loaded_from_database_secret_chats_.count(secret_chat_id)) {
|
|
promise.set_value(Unit());
|
|
return;
|
|
}
|
|
|
|
CHECK(c == nullptr || !c->is_being_saved);
|
|
load_secret_chat_from_database_impl(secret_chat_id, std::move(promise));
|
|
}
|
|
|
|
void UserManager::load_secret_chat_from_database_impl(SecretChatId secret_chat_id, Promise<Unit> promise) {
|
|
LOG(INFO) << "Load " << secret_chat_id << " from database";
|
|
auto &load_secret_chat_queries = load_secret_chat_from_database_queries_[secret_chat_id];
|
|
load_secret_chat_queries.push_back(std::move(promise));
|
|
if (load_secret_chat_queries.size() == 1u) {
|
|
G()->td_db()->get_sqlite_pmc()->get(
|
|
get_secret_chat_database_key(secret_chat_id), PromiseCreator::lambda([secret_chat_id](string value) {
|
|
send_closure(G()->user_manager(), &UserManager::on_load_secret_chat_from_database, secret_chat_id,
|
|
std::move(value), false);
|
|
}));
|
|
}
|
|
}
|
|
|
|
void UserManager::on_load_secret_chat_from_database(SecretChatId secret_chat_id, string value, bool force) {
|
|
if (G()->close_flag() && !force) {
|
|
// the secret chat is in Binlog and will be saved after restart
|
|
return;
|
|
}
|
|
|
|
CHECK(secret_chat_id.is_valid());
|
|
if (!loaded_from_database_secret_chats_.insert(secret_chat_id).second) {
|
|
return;
|
|
}
|
|
|
|
auto it = load_secret_chat_from_database_queries_.find(secret_chat_id);
|
|
vector<Promise<Unit>> promises;
|
|
if (it != load_secret_chat_from_database_queries_.end()) {
|
|
promises = std::move(it->second);
|
|
CHECK(!promises.empty());
|
|
load_secret_chat_from_database_queries_.erase(it);
|
|
}
|
|
|
|
LOG(INFO) << "Successfully loaded " << secret_chat_id << " of size " << value.size() << " from database";
|
|
// G()->td_db()->get_sqlite_pmc()->erase(get_secret_chat_database_key(secret_chat_id), Auto());
|
|
// return;
|
|
|
|
SecretChat *c = get_secret_chat(secret_chat_id);
|
|
if (c == nullptr) {
|
|
if (!value.empty()) {
|
|
c = add_secret_chat(secret_chat_id);
|
|
|
|
if (log_event_parse(*c, value).is_error()) {
|
|
LOG(ERROR) << "Failed to load " << secret_chat_id << " from database";
|
|
secret_chats_.erase(secret_chat_id);
|
|
} else {
|
|
c->is_saved = true;
|
|
update_secret_chat(c, secret_chat_id, true, true);
|
|
}
|
|
}
|
|
} else {
|
|
CHECK(!c->is_saved); // secret chat can't be saved before load completes
|
|
CHECK(!c->is_being_saved);
|
|
auto new_value = get_secret_chat_database_value(c);
|
|
if (value != new_value) {
|
|
save_secret_chat_to_database_impl(c, secret_chat_id, std::move(new_value));
|
|
} else if (c->log_event_id != 0) {
|
|
binlog_erase(G()->td_db()->get_binlog(), c->log_event_id);
|
|
c->log_event_id = 0;
|
|
}
|
|
}
|
|
|
|
// TODO load users asynchronously
|
|
if (c != nullptr && !have_user_force(c->user_id, "on_load_secret_chat_from_database")) {
|
|
LOG(ERROR) << "Can't find " << c->user_id << " from " << secret_chat_id;
|
|
}
|
|
|
|
set_promises(promises);
|
|
}
|
|
|
|
void UserManager::create_new_secret_chat(UserId user_id, Promise<td_api::object_ptr<td_api::chat>> &&promise) {
|
|
TRY_RESULT_PROMISE(promise, input_user, get_input_user(user_id));
|
|
if (input_user->get_id() != telegram_api::inputUser::ID) {
|
|
return promise.set_error(Status::Error(400, "Can't create secret chat with the user"));
|
|
}
|
|
auto user = static_cast<const telegram_api::inputUser *>(input_user.get());
|
|
|
|
send_closure(
|
|
G()->secret_chats_manager(), &SecretChatsManager::create_chat, UserId(user->user_id_), user->access_hash_,
|
|
PromiseCreator::lambda(
|
|
[actor_id = actor_id(this), promise = std::move(promise)](Result<SecretChatId> r_secret_chat_id) mutable {
|
|
if (r_secret_chat_id.is_error()) {
|
|
return promise.set_error(r_secret_chat_id.move_as_error());
|
|
}
|
|
send_closure(actor_id, &UserManager::on_create_new_secret_chat, r_secret_chat_id.ok(), std::move(promise));
|
|
}));
|
|
}
|
|
|
|
void UserManager::on_create_new_secret_chat(SecretChatId secret_chat_id,
|
|
Promise<td_api::object_ptr<td_api::chat>> &&promise) {
|
|
TRY_STATUS_PROMISE(promise, G()->close_status());
|
|
CHECK(secret_chat_id.is_valid());
|
|
DialogId dialog_id(secret_chat_id);
|
|
td_->dialog_manager_->force_create_dialog(dialog_id, "on_create_new_secret_chat");
|
|
promise.set_value(td_->messages_manager_->get_chat_object(dialog_id, "on_create_new_secret_chat"));
|
|
}
|
|
|
|
void UserManager::update_user(User *u, UserId user_id, bool from_binlog, bool from_database) {
|
|
CHECK(u != nullptr);
|
|
|
|
if (u->is_being_updated) {
|
|
LOG(ERROR) << "Detected recursive update of " << user_id;
|
|
}
|
|
u->is_being_updated = true;
|
|
SCOPE_EXIT {
|
|
u->is_being_updated = false;
|
|
};
|
|
|
|
if (user_id == get_my_id()) {
|
|
if (td_->option_manager_->get_option_boolean("is_premium") != u->is_premium) {
|
|
td_->option_manager_->set_option_boolean("is_premium", u->is_premium);
|
|
send_closure(td_->config_manager_, &ConfigManager::request_config, true);
|
|
if (!td_->auth_manager_->is_bot()) {
|
|
td_->reaction_manager_->reload_reaction_list(ReactionListType::Top, "update_user is_premium");
|
|
td_->messages_manager_->update_is_translatable(u->is_premium);
|
|
}
|
|
}
|
|
}
|
|
if (u->is_name_changed || u->is_username_changed || u->is_is_contact_changed) {
|
|
update_contacts_hints(u, user_id, from_database);
|
|
u->is_username_changed = false;
|
|
}
|
|
if (u->is_is_contact_changed) {
|
|
td_->messages_manager_->on_dialog_user_is_contact_updated(DialogId(user_id), u->is_contact);
|
|
send_closure_later(td_->story_manager_actor_, &StoryManager::on_dialog_active_stories_order_updated,
|
|
DialogId(user_id), "is_contact");
|
|
if (is_user_contact(u, user_id, false)) {
|
|
auto user_full = get_user_full(user_id);
|
|
if (user_full != nullptr && user_full->need_phone_number_privacy_exception) {
|
|
on_update_user_full_need_phone_number_privacy_exception(user_full, user_id, false);
|
|
update_user_full(user_full, user_id, "update_user");
|
|
}
|
|
}
|
|
u->is_is_contact_changed = false;
|
|
}
|
|
if (u->is_is_mutual_contact_changed) {
|
|
if (!from_database && u->is_update_user_sent) {
|
|
send_closure_later(td_->story_manager_actor_, &StoryManager::reload_dialog_expiring_stories, DialogId(user_id));
|
|
}
|
|
u->is_is_mutual_contact_changed = false;
|
|
}
|
|
if (u->is_is_deleted_changed) {
|
|
td_->messages_manager_->on_dialog_user_is_deleted_updated(DialogId(user_id), u->is_deleted);
|
|
if (u->is_deleted) {
|
|
auto user_full = get_user_full(user_id); // must not load user_full from database before sending updateUser
|
|
if (user_full != nullptr) {
|
|
u->is_full_info_changed = false;
|
|
drop_user_full(user_id);
|
|
}
|
|
}
|
|
u->is_is_deleted_changed = false;
|
|
}
|
|
if (u->is_is_premium_changed) {
|
|
send_closure_later(td_->story_manager_actor_, &StoryManager::on_dialog_active_stories_order_updated,
|
|
DialogId(user_id), "is_premium");
|
|
u->is_is_premium_changed = false;
|
|
}
|
|
if (u->is_name_changed) {
|
|
auto messages_manager = td_->messages_manager_.get();
|
|
messages_manager->on_dialog_title_updated(DialogId(user_id));
|
|
for_each_secret_chat_with_user(user_id, [messages_manager](SecretChatId secret_chat_id) {
|
|
messages_manager->on_dialog_title_updated(DialogId(secret_chat_id));
|
|
});
|
|
u->is_name_changed = false;
|
|
}
|
|
if (u->is_photo_changed) {
|
|
auto messages_manager = td_->messages_manager_.get();
|
|
messages_manager->on_dialog_photo_updated(DialogId(user_id));
|
|
for_each_secret_chat_with_user(user_id, [messages_manager](SecretChatId secret_chat_id) {
|
|
messages_manager->on_dialog_photo_updated(DialogId(secret_chat_id));
|
|
});
|
|
u->is_photo_changed = false;
|
|
}
|
|
if (u->is_accent_color_changed) {
|
|
auto messages_manager = td_->messages_manager_.get();
|
|
messages_manager->on_dialog_accent_colors_updated(DialogId(user_id));
|
|
for_each_secret_chat_with_user(user_id, [messages_manager](SecretChatId secret_chat_id) {
|
|
messages_manager->on_dialog_accent_colors_updated(DialogId(secret_chat_id));
|
|
});
|
|
u->is_accent_color_changed = false;
|
|
}
|
|
if (u->is_phone_number_changed) {
|
|
if (!u->phone_number.empty() && !td_->auth_manager_->is_bot()) {
|
|
resolved_phone_numbers_[u->phone_number] = user_id;
|
|
}
|
|
u->is_phone_number_changed = false;
|
|
}
|
|
auto unix_time = G()->unix_time();
|
|
if (u->is_status_changed && user_id != get_my_id()) {
|
|
auto left_time = get_user_was_online(u, user_id, unix_time) - G()->server_time();
|
|
if (left_time >= 0 && left_time < 30 * 86400) {
|
|
left_time += 2.0; // to guarantee expiration
|
|
LOG(DEBUG) << "Set online timeout for " << user_id << " in " << left_time << " seconds";
|
|
user_online_timeout_.set_timeout_in(user_id.get(), left_time);
|
|
} else {
|
|
LOG(DEBUG) << "Cancel online timeout for " << user_id;
|
|
user_online_timeout_.cancel_timeout(user_id.get());
|
|
}
|
|
}
|
|
if (u->is_stories_hidden_changed) {
|
|
send_closure_later(td_->story_manager_actor_, &StoryManager::on_dialog_active_stories_order_updated,
|
|
DialogId(user_id), "stories_hidden");
|
|
u->is_stories_hidden_changed = false;
|
|
}
|
|
if (!td_->auth_manager_->is_bot()) {
|
|
if (u->restriction_reasons.empty()) {
|
|
restricted_user_ids_.erase(user_id);
|
|
} else {
|
|
restricted_user_ids_.insert(user_id);
|
|
}
|
|
}
|
|
|
|
auto effective_emoji_status = u->emoji_status.get_effective_emoji_status(u->is_premium, unix_time);
|
|
if (effective_emoji_status != u->last_sent_emoji_status) {
|
|
if (!u->last_sent_emoji_status.is_empty()) {
|
|
user_emoji_status_timeout_.cancel_timeout(user_id.get());
|
|
}
|
|
u->last_sent_emoji_status = effective_emoji_status;
|
|
if (!u->last_sent_emoji_status.is_empty()) {
|
|
auto until_date = u->last_sent_emoji_status.get_until_date();
|
|
auto left_time = until_date - unix_time;
|
|
if (left_time >= 0 && left_time < 30 * 86400) {
|
|
user_emoji_status_timeout_.set_timeout_in(user_id.get(), left_time);
|
|
}
|
|
}
|
|
u->is_changed = true;
|
|
|
|
auto messages_manager = td_->messages_manager_.get();
|
|
messages_manager->on_dialog_emoji_status_updated(DialogId(user_id));
|
|
for_each_secret_chat_with_user(user_id, [messages_manager](SecretChatId secret_chat_id) {
|
|
messages_manager->on_dialog_emoji_status_updated(DialogId(secret_chat_id));
|
|
});
|
|
u->is_emoji_status_changed = false;
|
|
} else if (u->is_emoji_status_changed) {
|
|
LOG(DEBUG) << "Emoji status for " << user_id << " has changed";
|
|
u->need_save_to_database = true;
|
|
u->is_emoji_status_changed = false;
|
|
}
|
|
|
|
if (u->is_deleted) {
|
|
td_->inline_queries_manager_->remove_recent_inline_bot(user_id, Promise<>());
|
|
}
|
|
if (from_binlog || from_database) {
|
|
td_->dialog_manager_->on_dialog_usernames_received(DialogId(user_id), u->usernames, true);
|
|
}
|
|
|
|
LOG(DEBUG) << "Update " << user_id << ": need_save_to_database = " << u->need_save_to_database
|
|
<< ", is_changed = " << u->is_changed << ", is_status_changed = " << u->is_status_changed
|
|
<< ", from_binlog = " << from_binlog << ", from_database = " << from_database;
|
|
u->need_save_to_database |= u->is_changed;
|
|
if (u->need_save_to_database) {
|
|
if (!from_database) {
|
|
u->is_saved = false;
|
|
}
|
|
u->need_save_to_database = false;
|
|
}
|
|
if (u->is_changed) {
|
|
send_closure(G()->td(), &Td::send_update, get_update_user_object(user_id, u));
|
|
u->is_changed = false;
|
|
u->is_status_changed = false;
|
|
u->is_update_user_sent = true;
|
|
}
|
|
if (u->is_status_changed) {
|
|
if (!from_database) {
|
|
u->is_status_saved = false;
|
|
}
|
|
CHECK(u->is_update_user_sent);
|
|
send_closure(
|
|
G()->td(), &Td::send_update,
|
|
td_api::make_object<td_api::updateUserStatus>(user_id.get(), get_user_status_object(user_id, u, unix_time)));
|
|
u->is_status_changed = false;
|
|
}
|
|
if (u->is_online_status_changed) {
|
|
td_->dialog_participant_manager_->update_user_online_member_count(user_id);
|
|
u->is_online_status_changed = false;
|
|
}
|
|
|
|
if (!from_database) {
|
|
save_user(u, user_id, from_binlog);
|
|
}
|
|
|
|
if (u->cache_version != User::CACHE_VERSION && !u->is_repaired &&
|
|
have_input_peer_user(u, user_id, AccessRights::Read) && !G()->close_flag()) {
|
|
u->is_repaired = true;
|
|
|
|
LOG(INFO) << "Repairing cache of " << user_id;
|
|
reload_user(user_id, Promise<Unit>(), "update_user");
|
|
}
|
|
|
|
if (u->is_full_info_changed) {
|
|
u->is_full_info_changed = false;
|
|
auto user_full = get_user_full(user_id);
|
|
if (user_full != nullptr) {
|
|
user_full->need_send_update = true;
|
|
update_user_full(user_full, user_id, "update_user is_full_info_changed");
|
|
reload_user_full(user_id, Promise<Unit>(), "update_user");
|
|
}
|
|
}
|
|
}
|
|
|
|
void UserManager::update_secret_chat(SecretChat *c, SecretChatId secret_chat_id, bool from_binlog, bool from_database) {
|
|
CHECK(c != nullptr);
|
|
|
|
if (c->is_being_updated) {
|
|
LOG(ERROR) << "Detected recursive update of " << secret_chat_id;
|
|
}
|
|
c->is_being_updated = true;
|
|
SCOPE_EXIT {
|
|
c->is_being_updated = false;
|
|
};
|
|
|
|
LOG(DEBUG) << "Update " << secret_chat_id << ": need_save_to_database = " << c->need_save_to_database
|
|
<< ", is_changed = " << c->is_changed;
|
|
c->need_save_to_database |= c->is_changed;
|
|
if (c->need_save_to_database) {
|
|
if (!from_database) {
|
|
c->is_saved = false;
|
|
}
|
|
c->need_save_to_database = false;
|
|
|
|
DialogId dialog_id(secret_chat_id);
|
|
send_closure_later(G()->messages_manager(), &MessagesManager::force_create_dialog, dialog_id, "update secret chat",
|
|
true, true);
|
|
if (c->is_state_changed) {
|
|
send_closure_later(G()->messages_manager(), &MessagesManager::on_update_secret_chat_state, secret_chat_id,
|
|
c->state);
|
|
c->is_state_changed = false;
|
|
}
|
|
if (c->is_ttl_changed) {
|
|
send_closure_later(G()->messages_manager(), &MessagesManager::on_update_dialog_message_ttl,
|
|
DialogId(secret_chat_id), MessageTtl(c->ttl));
|
|
c->is_ttl_changed = false;
|
|
}
|
|
}
|
|
if (c->is_changed) {
|
|
send_closure(G()->td(), &Td::send_update, get_update_secret_chat_object(secret_chat_id, c));
|
|
c->is_changed = false;
|
|
}
|
|
|
|
if (!from_database) {
|
|
save_secret_chat(c, secret_chat_id, from_binlog);
|
|
}
|
|
}
|
|
|
|
void UserManager::update_user_full(UserFull *user_full, UserId user_id, const char *source, bool from_database) {
|
|
CHECK(user_full != nullptr);
|
|
|
|
if (user_full->is_being_updated) {
|
|
LOG(ERROR) << "Detected recursive update of full " << user_id << " from " << source;
|
|
}
|
|
user_full->is_being_updated = true;
|
|
SCOPE_EXIT {
|
|
user_full->is_being_updated = false;
|
|
};
|
|
|
|
unavailable_user_fulls_.erase(user_id); // don't needed anymore
|
|
if (user_full->is_common_chat_count_changed) {
|
|
td_->common_dialog_manager_->drop_common_dialogs_cache(user_id);
|
|
user_full->is_common_chat_count_changed = false;
|
|
}
|
|
if (true) {
|
|
vector<FileId> file_ids;
|
|
if (!user_full->personal_photo.is_empty()) {
|
|
append(file_ids, photo_get_file_ids(user_full->personal_photo));
|
|
}
|
|
if (!user_full->fallback_photo.is_empty()) {
|
|
append(file_ids, photo_get_file_ids(user_full->fallback_photo));
|
|
}
|
|
if (!user_full->description_photo.is_empty()) {
|
|
append(file_ids, photo_get_file_ids(user_full->description_photo));
|
|
}
|
|
if (user_full->description_animation_file_id.is_valid()) {
|
|
file_ids.push_back(user_full->description_animation_file_id);
|
|
}
|
|
if (user_full->business_info != nullptr) {
|
|
append(file_ids, user_full->business_info->get_file_ids(td_));
|
|
}
|
|
if (user_full->registered_file_ids != file_ids) {
|
|
auto &file_source_id = user_full->file_source_id;
|
|
if (!file_source_id.is_valid()) {
|
|
file_source_id = user_full_file_source_ids_.get(user_id);
|
|
if (file_source_id.is_valid()) {
|
|
VLOG(file_references) << "Move " << file_source_id << " inside of " << user_id;
|
|
user_full_file_source_ids_.erase(user_id);
|
|
} else {
|
|
VLOG(file_references) << "Need to create new file source for full " << user_id;
|
|
file_source_id = td_->file_reference_manager_->create_user_full_file_source(user_id);
|
|
}
|
|
}
|
|
|
|
td_->file_manager_->change_files_source(file_source_id, user_full->registered_file_ids, file_ids);
|
|
user_full->registered_file_ids = std::move(file_ids);
|
|
}
|
|
}
|
|
|
|
user_full->need_send_update |= user_full->is_changed;
|
|
user_full->need_save_to_database |= user_full->is_changed;
|
|
user_full->is_changed = false;
|
|
if (user_full->need_send_update || user_full->need_save_to_database) {
|
|
LOG(INFO) << "Update full " << user_id << " from " << source;
|
|
}
|
|
if (user_full->need_send_update) {
|
|
{
|
|
auto u = get_user(user_id);
|
|
CHECK(u == nullptr || u->is_update_user_sent);
|
|
}
|
|
if (!user_full->is_update_user_full_sent) {
|
|
LOG(ERROR) << "Send partial updateUserFullInfo for " << user_id << " from " << source;
|
|
user_full->is_update_user_full_sent = true;
|
|
}
|
|
send_closure(G()->td(), &Td::send_update,
|
|
td_api::make_object<td_api::updateUserFullInfo>(get_user_id_object(user_id, "updateUserFullInfo"),
|
|
get_user_full_info_object(user_id, user_full)));
|
|
user_full->need_send_update = false;
|
|
|
|
if (user_id == get_my_id() && !user_full->birthdate.is_empty() && !td_->auth_manager_->is_bot()) {
|
|
dismiss_suggested_action(SuggestedAction{SuggestedAction::Type::BirthdaySetup}, Promise<Unit>());
|
|
}
|
|
}
|
|
if (user_full->need_save_to_database) {
|
|
if (!from_database) {
|
|
save_user_full(user_full, user_id);
|
|
}
|
|
user_full->need_save_to_database = false;
|
|
}
|
|
}
|
|
|
|
td_api::object_ptr<td_api::UserStatus> UserManager::get_user_status_object(UserId user_id, const User *u,
|
|
int32 unix_time) const {
|
|
if (u->is_bot) {
|
|
return td_api::make_object<td_api::userStatusOnline>(std::numeric_limits<int32>::max());
|
|
}
|
|
|
|
int32 was_online = get_user_was_online(u, user_id, unix_time);
|
|
switch (was_online) {
|
|
case -6:
|
|
case -3:
|
|
return td_api::make_object<td_api::userStatusLastMonth>(was_online == -6);
|
|
case -5:
|
|
case -2:
|
|
return td_api::make_object<td_api::userStatusLastWeek>(was_online == -5);
|
|
case -4:
|
|
case -1:
|
|
return td_api::make_object<td_api::userStatusRecently>(was_online == -4);
|
|
case 0:
|
|
return td_api::make_object<td_api::userStatusEmpty>();
|
|
default: {
|
|
int32 time = G()->unix_time();
|
|
if (was_online > time) {
|
|
return td_api::make_object<td_api::userStatusOnline>(was_online);
|
|
} else {
|
|
return td_api::make_object<td_api::userStatusOffline>(was_online);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
bool UserManager::get_user_has_unread_stories(const User *u) {
|
|
CHECK(u != nullptr);
|
|
return u->max_active_story_id.get() > u->max_read_story_id.get();
|
|
}
|
|
|
|
td_api::object_ptr<td_api::updateUser> UserManager::get_update_user_object(UserId user_id, const User *u) const {
|
|
if (u == nullptr) {
|
|
return get_update_unknown_user_object(user_id);
|
|
}
|
|
return td_api::make_object<td_api::updateUser>(get_user_object(user_id, u));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::updateUser> UserManager::get_update_unknown_user_object(UserId user_id) const {
|
|
auto have_access = user_id == get_my_id() || user_messages_.count(user_id) != 0;
|
|
return td_api::make_object<td_api::updateUser>(td_api::make_object<td_api::user>(
|
|
user_id.get(), "", "", nullptr, "", td_api::make_object<td_api::userStatusEmpty>(), nullptr,
|
|
td_->theme_manager_->get_accent_color_id_object(AccentColorId(user_id)), 0, -1, 0, nullptr, false, false, false,
|
|
false, false, false, "", false, false, false, false, false, have_access,
|
|
td_api::make_object<td_api::userTypeUnknown>(), "", false));
|
|
}
|
|
|
|
int64 UserManager::get_user_id_object(UserId user_id, const char *source) const {
|
|
if (user_id.is_valid() && get_user(user_id) == nullptr && unknown_users_.count(user_id) == 0) {
|
|
LOG(ERROR) << "Have no information about " << user_id << " from " << source;
|
|
unknown_users_.insert(user_id);
|
|
send_closure(G()->td(), &Td::send_update, get_update_unknown_user_object(user_id));
|
|
}
|
|
return user_id.get();
|
|
}
|
|
|
|
void UserManager::get_user_id_object_async(UserId user_id, Promise<int64> &&promise) {
|
|
promise.set_value(get_user_id_object(user_id, "get_user_id_object_async"));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::user> UserManager::get_user_object(UserId user_id) const {
|
|
return get_user_object(user_id, get_user(user_id));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::user> UserManager::get_user_object(UserId user_id, const User *u) const {
|
|
if (u == nullptr) {
|
|
return nullptr;
|
|
}
|
|
td_api::object_ptr<td_api::UserType> type;
|
|
if (u->is_deleted) {
|
|
type = td_api::make_object<td_api::userTypeDeleted>();
|
|
} else if (u->is_bot) {
|
|
type = td_api::make_object<td_api::userTypeBot>(
|
|
u->can_be_edited_bot, u->can_join_groups, u->can_read_all_group_messages, u->has_main_app, u->is_inline_bot,
|
|
u->inline_query_placeholder, u->need_location_bot, u->is_business_bot, u->can_be_added_to_attach_menu,
|
|
u->bot_active_users);
|
|
} else {
|
|
type = td_api::make_object<td_api::userTypeRegular>();
|
|
}
|
|
|
|
auto emoji_status = u->last_sent_emoji_status.get_emoji_status_object();
|
|
auto have_access = user_id == get_my_id() || have_input_peer_user(u, user_id, AccessRights::Know);
|
|
auto restricts_new_chats = u->contact_require_premium && !u->is_mutual_contact;
|
|
return td_api::make_object<td_api::user>(
|
|
user_id.get(), u->first_name, u->last_name, u->usernames.get_usernames_object(), u->phone_number,
|
|
get_user_status_object(user_id, u, G()->unix_time()),
|
|
get_profile_photo_object(td_->file_manager_.get(), u->photo),
|
|
td_->theme_manager_->get_accent_color_id_object(u->accent_color_id, AccentColorId(user_id)),
|
|
u->background_custom_emoji_id.get(),
|
|
td_->theme_manager_->get_profile_accent_color_id_object(u->profile_accent_color_id),
|
|
u->profile_background_custom_emoji_id.get(), std::move(emoji_status), u->is_contact, u->is_mutual_contact,
|
|
u->is_close_friend, u->is_verified, u->is_premium, u->is_support,
|
|
get_restriction_reason_description(u->restriction_reasons), u->is_scam, u->is_fake,
|
|
u->max_active_story_id.is_valid(), get_user_has_unread_stories(u), restricts_new_chats, have_access,
|
|
std::move(type), u->language_code, u->attach_menu_enabled);
|
|
}
|
|
|
|
vector<int64> UserManager::get_user_ids_object(const vector<UserId> &user_ids, const char *source) const {
|
|
return transform(user_ids, [this, source](UserId user_id) { return get_user_id_object(user_id, source); });
|
|
}
|
|
|
|
td_api::object_ptr<td_api::users> UserManager::get_users_object(int32 total_count,
|
|
const vector<UserId> &user_ids) const {
|
|
if (total_count == -1) {
|
|
total_count = narrow_cast<int32>(user_ids.size());
|
|
}
|
|
return td_api::make_object<td_api::users>(total_count, get_user_ids_object(user_ids, "get_users_object"));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::userFullInfo> UserManager::get_user_full_info_object(UserId user_id) const {
|
|
return get_user_full_info_object(user_id, get_user_full(user_id));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::userFullInfo> UserManager::get_user_full_info_object(UserId user_id,
|
|
const UserFull *user_full) const {
|
|
CHECK(user_full != nullptr);
|
|
td_api::object_ptr<td_api::botInfo> bot_info;
|
|
const User *u = get_user(user_id);
|
|
bool is_bot = is_user_bot(u);
|
|
bool is_premium = is_user_premium(u);
|
|
td_api::object_ptr<td_api::formattedText> bio_object;
|
|
if (is_bot) {
|
|
auto menu_button = get_bot_menu_button_object(td_, user_full->menu_button.get());
|
|
auto commands =
|
|
transform(user_full->commands, [](const auto &command) { return command.get_bot_command_object(); });
|
|
bot_info = td_api::make_object<td_api::botInfo>(
|
|
user_full->about, user_full->description,
|
|
get_photo_object(td_->file_manager_.get(), user_full->description_photo),
|
|
td_->animations_manager_->get_animation_object(user_full->description_animation_file_id),
|
|
std::move(menu_button), std::move(commands),
|
|
user_full->group_administrator_rights == AdministratorRights()
|
|
? nullptr
|
|
: user_full->group_administrator_rights.get_chat_administrator_rights_object(),
|
|
user_full->broadcast_administrator_rights == AdministratorRights()
|
|
? nullptr
|
|
: user_full->broadcast_administrator_rights.get_chat_administrator_rights_object(),
|
|
user_full->has_preview_medias, nullptr, nullptr, nullptr, nullptr);
|
|
if (u != nullptr && u->can_be_edited_bot && u->usernames.has_editable_username()) {
|
|
auto bot_username = u->usernames.get_editable_username();
|
|
bot_info->edit_commands_link_ = td_api::make_object<td_api::internalLinkTypeBotStart>(
|
|
"botfather", PSTRING() << bot_username << "-commands", true);
|
|
bot_info->edit_description_link_ = td_api::make_object<td_api::internalLinkTypeBotStart>(
|
|
"botfather", PSTRING() << bot_username << "-intro", true);
|
|
bot_info->edit_description_media_link_ = td_api::make_object<td_api::internalLinkTypeBotStart>(
|
|
"botfather", PSTRING() << bot_username << "-intropic", true);
|
|
bot_info->edit_settings_link_ =
|
|
td_api::make_object<td_api::internalLinkTypeBotStart>("botfather", bot_username, true);
|
|
}
|
|
} else {
|
|
FormattedText bio;
|
|
bio.text = user_full->about;
|
|
bio.entities = find_entities(bio.text, true, true);
|
|
if (!is_premium) {
|
|
td::remove_if(bio.entities, [&](const MessageEntity &entity) {
|
|
if (entity.type == MessageEntity::Type::EmailAddress) {
|
|
return true;
|
|
}
|
|
if (entity.type == MessageEntity::Type::Url &&
|
|
!LinkManager::is_internal_link(utf8_utf16_substr(bio.text, entity.offset, entity.length))) {
|
|
return true;
|
|
}
|
|
return false;
|
|
});
|
|
}
|
|
bio_object = get_formatted_text_object(this, bio, true, 0);
|
|
}
|
|
auto voice_messages_forbidden = is_premium ? user_full->voice_messages_forbidden : false;
|
|
auto block_list_id = BlockListId(user_full->is_blocked, user_full->is_blocked_for_stories);
|
|
auto business_info = is_premium && user_full->business_info != nullptr
|
|
? user_full->business_info->get_business_info_object(td_)
|
|
: nullptr;
|
|
int64 personal_chat_id = 0;
|
|
if (user_full->personal_channel_id.is_valid()) {
|
|
DialogId dialog_id(user_full->personal_channel_id);
|
|
td_->dialog_manager_->force_create_dialog(dialog_id, "get_user_full_info_object", true);
|
|
personal_chat_id = td_->dialog_manager_->get_chat_id_object(dialog_id, "get_user_full_info_object");
|
|
}
|
|
return td_api::make_object<td_api::userFullInfo>(
|
|
get_chat_photo_object(td_->file_manager_.get(), user_full->personal_photo),
|
|
get_chat_photo_object(td_->file_manager_.get(), user_full->photo),
|
|
get_chat_photo_object(td_->file_manager_.get(), user_full->fallback_photo), block_list_id.get_block_list_object(),
|
|
user_full->can_be_called, user_full->supports_video_calls, user_full->has_private_calls,
|
|
!user_full->private_forward_name.empty(), voice_messages_forbidden, user_full->has_pinned_stories,
|
|
user_full->sponsored_enabled, user_full->need_phone_number_privacy_exception, user_full->wallpaper_overridden,
|
|
std::move(bio_object), user_full->birthdate.get_birthdate_object(), personal_chat_id,
|
|
get_premium_payment_options_object(user_full->premium_gift_options), user_full->common_chat_count,
|
|
std::move(business_info), std::move(bot_info));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::updateContactCloseBirthdays> UserManager::get_update_contact_close_birthdays() const {
|
|
return td_api::make_object<td_api::updateContactCloseBirthdays>(
|
|
transform(contact_birthdates_.users_, [this](const std::pair<UserId, Birthdate> &user) {
|
|
return td_api::make_object<td_api::closeBirthdayUser>(get_user_id_object(user.first, "closeBirthdayUser"),
|
|
user.second.get_birthdate_object());
|
|
}));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::SecretChatState> UserManager::get_secret_chat_state_object(SecretChatState state) {
|
|
switch (state) {
|
|
case SecretChatState::Waiting:
|
|
return td_api::make_object<td_api::secretChatStatePending>();
|
|
case SecretChatState::Active:
|
|
return td_api::make_object<td_api::secretChatStateReady>();
|
|
case SecretChatState::Closed:
|
|
case SecretChatState::Unknown:
|
|
return td_api::make_object<td_api::secretChatStateClosed>();
|
|
default:
|
|
UNREACHABLE();
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
td_api::object_ptr<td_api::updateSecretChat> UserManager::get_update_secret_chat_object(SecretChatId secret_chat_id,
|
|
const SecretChat *secret_chat) {
|
|
if (secret_chat == nullptr) {
|
|
return get_update_unknown_secret_chat_object(secret_chat_id);
|
|
}
|
|
return td_api::make_object<td_api::updateSecretChat>(get_secret_chat_object(secret_chat_id, secret_chat));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::updateSecretChat> UserManager::get_update_unknown_secret_chat_object(
|
|
SecretChatId secret_chat_id) {
|
|
return td_api::make_object<td_api::updateSecretChat>(td_api::make_object<td_api::secretChat>(
|
|
secret_chat_id.get(), 0, get_secret_chat_state_object(SecretChatState::Unknown), false, string(), 0));
|
|
}
|
|
|
|
int32 UserManager::get_secret_chat_id_object(SecretChatId secret_chat_id, const char *source) const {
|
|
if (secret_chat_id.is_valid() && get_secret_chat(secret_chat_id) == nullptr &&
|
|
unknown_secret_chats_.count(secret_chat_id) == 0) {
|
|
LOG(ERROR) << "Have no information about " << secret_chat_id << " from " << source;
|
|
unknown_secret_chats_.insert(secret_chat_id);
|
|
send_closure(G()->td(), &Td::send_update, get_update_unknown_secret_chat_object(secret_chat_id));
|
|
}
|
|
return secret_chat_id.get();
|
|
}
|
|
|
|
td_api::object_ptr<td_api::secretChat> UserManager::get_secret_chat_object(SecretChatId secret_chat_id) {
|
|
return get_secret_chat_object(secret_chat_id, get_secret_chat(secret_chat_id));
|
|
}
|
|
|
|
td_api::object_ptr<td_api::secretChat> UserManager::get_secret_chat_object(SecretChatId secret_chat_id,
|
|
const SecretChat *secret_chat) {
|
|
if (secret_chat == nullptr) {
|
|
return nullptr;
|
|
}
|
|
get_user_force(secret_chat->user_id, "get_secret_chat_object");
|
|
return get_secret_chat_object_const(secret_chat_id, secret_chat);
|
|
}
|
|
|
|
td_api::object_ptr<td_api::secretChat> UserManager::get_secret_chat_object_const(SecretChatId secret_chat_id,
|
|
const SecretChat *secret_chat) const {
|
|
return td_api::make_object<td_api::secretChat>(secret_chat_id.get(),
|
|
get_user_id_object(secret_chat->user_id, "secretChat"),
|
|
get_secret_chat_state_object(secret_chat->state),
|
|
secret_chat->is_outbound, secret_chat->key_hash, secret_chat->layer);
|
|
}
|
|
|
|
void UserManager::get_current_state(vector<td_api::object_ptr<td_api::Update>> &updates) const {
|
|
for (auto user_id : unknown_users_) {
|
|
if (!have_min_user(user_id)) {
|
|
updates.push_back(get_update_unknown_user_object(user_id));
|
|
}
|
|
}
|
|
for (auto secret_chat_id : unknown_secret_chats_) {
|
|
if (!have_secret_chat(secret_chat_id)) {
|
|
updates.push_back(get_update_unknown_secret_chat_object(secret_chat_id));
|
|
}
|
|
}
|
|
|
|
users_.foreach([&](const UserId &user_id, const unique_ptr<User> &user) {
|
|
updates.push_back(get_update_user_object(user_id, user.get()));
|
|
});
|
|
// secret chat objects contain user_id, so they must be sent after users
|
|
secret_chats_.foreach([&](const SecretChatId &secret_chat_id, const unique_ptr<SecretChat> &secret_chat) {
|
|
updates.push_back(
|
|
td_api::make_object<td_api::updateSecretChat>(get_secret_chat_object_const(secret_chat_id, secret_chat.get())));
|
|
});
|
|
|
|
users_full_.foreach([&](const UserId &user_id, const unique_ptr<UserFull> &user_full) {
|
|
updates.push_back(td_api::make_object<td_api::updateUserFullInfo>(
|
|
user_id.get(), get_user_full_info_object(user_id, user_full.get())));
|
|
});
|
|
|
|
if (!contact_birthdates_.users_.empty()) {
|
|
updates.push_back(get_update_contact_close_birthdays());
|
|
}
|
|
}
|
|
|
|
} // namespace td
|