// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2019 // // 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/ContactsManager.h" #include "td/telegram/td_api.h" #include "td/telegram/telegram_api.h" #include "td/telegram/telegram_api.hpp" #include "td/telegram/AuthManager.h" #include "td/telegram/ConfigShared.h" #include "td/telegram/FileReferenceManager.h" #include "td/telegram/files/FileManager.h" #include "td/telegram/files/FileType.h" #include "td/telegram/Global.h" #include "td/telegram/InlineQueriesManager.h" #include "td/telegram/logevent/LogEvent.h" #include "td/telegram/MessagesManager.h" #include "td/telegram/misc.h" #include "td/telegram/net/NetQuery.h" #include "td/telegram/NotificationManager.h" #include "td/telegram/Photo.h" #include "td/telegram/Photo.hpp" #include "td/telegram/SecretChatActor.h" #include "td/telegram/StickersManager.h" #include "td/telegram/Td.h" #include "td/telegram/TdDb.h" #include "td/telegram/TopDialogManager.h" #include "td/telegram/UpdatesManager.h" #include "td/telegram/Version.h" #include "td/actor/PromiseFuture.h" #include "td/actor/SleepActor.h" #include "td/db/binlog/BinlogEvent.h" #include "td/db/binlog/BinlogHelper.h" #include "td/db/SqliteKeyValueAsync.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/StringBuilder.h" #include "td/utils/Time.h" #include "td/utils/tl_helpers.h" #include #include #include #include namespace td { class SetAccountTtlQuery : public Td::ResultHandler { Promise promise_; public: explicit SetAccountTtlQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(int32 account_ttl) { send_query(G()->net_query_creator().create( create_storer(telegram_api::account_setAccountTTL(make_tl_object(account_ttl))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.move_as_ok(); if (!result) { return on_error(id, Status::Error(500, "Internal Server Error")); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class GetAccountTtlQuery : public Td::ResultHandler { Promise promise_; public: explicit GetAccountTtlQuery(Promise &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::account_getAccountTTL()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetAccountTtlQuery: " << to_string(ptr); promise_.set_value(std::move(ptr->days_)); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class GetAuthorizationsQuery : public Td::ResultHandler { Promise> promise_; static constexpr int32 AUTHORIZATION_FLAG_IS_CURRENT = 1 << 0; static constexpr int32 AUTHORIZATION_FLAG_IS_OFFICIAL_APPLICATION = 1 << 1; public: explicit GetAuthorizationsQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::account_getAuthorizations()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetAuthorizationsQuery: " << to_string(ptr); auto results = make_tl_object(); results->sessions_.reserve(ptr->authorizations_.size()); for (auto &authorization : ptr->authorizations_) { CHECK(authorization != nullptr); bool is_current = (authorization->flags_ & AUTHORIZATION_FLAG_IS_CURRENT) != 0; bool is_official_application = (authorization->flags_ & AUTHORIZATION_FLAG_IS_OFFICIAL_APPLICATION) != 0; results->sessions_.push_back(make_tl_object( authorization->hash_, is_current, authorization->api_id_, authorization->app_name_, authorization->app_version_, is_official_application, authorization->device_model_, authorization->platform_, authorization->system_version_, authorization->date_created_, authorization->date_active_, authorization->ip_, authorization->country_, authorization->region_)); } promise_.set_value(std::move(results)); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class ResetAuthorizationQuery : public Td::ResultHandler { Promise promise_; public: explicit ResetAuthorizationQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(int64 authorization_id) { send_query( G()->net_query_creator().create(create_storer(telegram_api::account_resetAuthorization(authorization_id)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.move_as_ok(); LOG_IF(WARNING, !result) << "Failed to terminate session"; promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class ResetAuthorizationsQuery : public Td::ResultHandler { Promise promise_; public: explicit ResetAuthorizationsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::auth_resetAuthorizations()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.move_as_ok(); LOG_IF(WARNING, !result) << "Failed to terminate all sessions"; promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class GetWebAuthorizationsQuery : public Td::ResultHandler { Promise> promise_; public: explicit GetWebAuthorizationsQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::account_getWebAuthorizations()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetWebAuthorizationsQuery: " << to_string(ptr); td->contacts_manager_->on_get_users(std::move(ptr->users_), "GetWebAuthorizationsQuery"); auto results = make_tl_object(); results->websites_.reserve(ptr->authorizations_.size()); for (auto &authorization : ptr->authorizations_) { CHECK(authorization != nullptr); UserId bot_user_id(authorization->bot_id_); if (!bot_user_id.is_valid()) { LOG(ERROR) << "Receive invalid bot " << bot_user_id; bot_user_id = UserId(); } results->websites_.push_back(make_tl_object( authorization->hash_, authorization->domain_, td->contacts_manager_->get_user_id_object(bot_user_id, "GetWebAuthorizationsQuery"), authorization->browser_, authorization->platform_, authorization->date_created_, authorization->date_active_, authorization->ip_, authorization->region_)); } promise_.set_value(std::move(results)); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class ResetWebAuthorizationQuery : public Td::ResultHandler { Promise promise_; public: explicit ResetWebAuthorizationQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(int64 hash) { send_query(G()->net_query_creator().create(create_storer(telegram_api::account_resetWebAuthorization(hash)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.move_as_ok(); LOG_IF(WARNING, !result) << "Failed to disconnect website"; promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class ResetWebAuthorizationsQuery : public Td::ResultHandler { Promise promise_; public: explicit ResetWebAuthorizationsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::account_resetWebAuthorizations()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.move_as_ok(); LOG_IF(WARNING, !result) << "Failed to disconnect all websites"; promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class BlockUserQuery : public Td::ResultHandler { public: void send(tl_object_ptr &&user) { send_query(G()->net_query_creator().create(create_storer(telegram_api::contacts_block(std::move(user))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.ok(); LOG_IF(WARNING, !result) << "Block user has failed"; } void on_error(uint64 id, Status status) override { LOG(WARNING) << "Receive error for blockUser: " << status; status.ignore(); } }; class UnblockUserQuery : public Td::ResultHandler { public: void send(tl_object_ptr &&user) { send_query(G()->net_query_creator().create(create_storer(telegram_api::contacts_unblock(std::move(user))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.ok(); LOG_IF(WARNING, !result) << "Unblock user has failed"; } void on_error(uint64 id, Status status) override { LOG(WARNING) << "Receive error for unblockUser: " << status; status.ignore(); } }; class GetBlockedUsersQuery : public Td::ResultHandler { Promise promise_; int32 offset_; int32 limit_; int64 random_id_; public: explicit GetBlockedUsersQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(int32 offset, int32 limit, int64 random_id) { offset_ = offset; limit_ = limit; random_id_ = random_id; send_query(G()->net_query_creator().create(create_storer(telegram_api::contacts_getBlocked(offset, limit)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetBlockedUsersQuery: " << to_string(ptr); int32 constructor_id = ptr->get_id(); switch (constructor_id) { case telegram_api::contacts_blocked::ID: { auto blocked_users = move_tl_object_as(ptr); td->contacts_manager_->on_get_users(std::move(blocked_users->users_), "GetBlockedUsersQuery"); td->contacts_manager_->on_get_blocked_users_result(offset_, limit_, random_id_, narrow_cast(blocked_users->blocked_.size()), std::move(blocked_users->blocked_)); break; } case telegram_api::contacts_blockedSlice::ID: { auto blocked_users = move_tl_object_as(ptr); td->contacts_manager_->on_get_users(std::move(blocked_users->users_), "GetBlockedUsersQuery"); td->contacts_manager_->on_get_blocked_users_result(offset_, limit_, random_id_, blocked_users->count_, std::move(blocked_users->blocked_)); break; } default: UNREACHABLE(); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_failed_get_blocked_users(random_id_); promise_.set_error(std::move(status)); } }; class GetContactsQuery : public Td::ResultHandler { public: void send(int32 hash) { LOG(INFO) << "Reload contacts with hash " << hash; send_query(G()->net_query_creator().create(create_storer(telegram_api::contacts_getContacts(hash)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for getContacts: " << to_string(ptr); td->contacts_manager_->on_get_contacts(std::move(ptr)); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_contacts_failed(std::move(status)); td->updates_manager_->get_difference("GetContactsQuery"); } }; class GetContactsStatusesQuery : public Td::ResultHandler { public: void send() { LOG(INFO) << "Reload contacts statuses"; send_query(G()->net_query_creator().create(create_storer(telegram_api::contacts_getStatuses()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } td->contacts_manager_->on_get_contacts_statuses(result_ptr.move_as_ok()); } void on_error(uint64 id, Status status) override { LOG(ERROR) << "Receive error for getContactsStatuses: " << status; } }; class ImportContactsQuery : public Td::ResultHandler { Promise promise_; vector input_contacts_; vector imported_user_ids_; vector unimported_contact_invites_; int64 random_id_; public: explicit ImportContactsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(vector input_contacts, int64 random_id) { random_id_ = random_id; size_t size = input_contacts.size(); if (size == 0) { td->contacts_manager_->on_imported_contacts(random_id, std::move(imported_user_ids_), std::move(unimported_contact_invites_)); promise_.set_value(Unit()); return; } imported_user_ids_.resize(size); unimported_contact_invites_.resize(size); input_contacts_ = std::move(input_contacts); vector> contacts; contacts.reserve(size); for (size_t i = 0; i < size; i++) { contacts.push_back(input_contacts_[i].get_input_phone_contact(static_cast(i))); } send_query( G()->net_query_creator().create(create_storer(telegram_api::contacts_importContacts(std::move(contacts))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for importContacts: " << to_string(ptr); td->contacts_manager_->on_get_users(std::move(ptr->users_), "ImportContactsQuery"); for (auto &imported_contact : ptr->imported_) { int64 client_id = imported_contact->client_id_; if (client_id < 0 || client_id >= static_cast(imported_user_ids_.size())) { LOG(ERROR) << "Wrong client_id " << client_id << " returned"; continue; } imported_user_ids_[static_cast(client_id)] = UserId(imported_contact->user_id_); } for (auto &popular_contact : ptr->popular_invites_) { int64 client_id = popular_contact->client_id_; if (client_id < 0 || client_id >= static_cast(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; } unimported_contact_invites_[static_cast(client_id)] = popular_contact->importers_; } if (!ptr->retry_contacts_.empty()) { int64 total_size = static_cast(input_contacts_.size()); vector> contacts; contacts.reserve(ptr->retry_contacts_.size()); for (auto &client_id : ptr->retry_contacts_) { if (client_id < 0 || client_id >= total_size) { LOG(ERROR) << "Wrong client_id " << client_id << " returned"; continue; } size_t i = static_cast(client_id); contacts.push_back(input_contacts_[i].get_input_phone_contact(client_id)); } send_query( G()->net_query_creator().create(create_storer(telegram_api::contacts_importContacts(std::move(contacts))))); return; } td->contacts_manager_->on_imported_contacts(random_id_, std::move(imported_user_ids_), std::move(unimported_contact_invites_)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->contacts_manager_->reload_contacts(true); } }; class DeleteContactsQuery : public Td::ResultHandler { Promise promise_; vector user_ids_; public: explicit DeleteContactsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(vector &&user_ids, vector> &&input_users) { user_ids_ = std::move(user_ids); send_query( G()->net_query_creator().create(create_storer(telegram_api::contacts_deleteContacts(std::move(input_users))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.ok(); if (!result) { return on_error(id, Status::Error(500, "Some contacts can't be deleted")); } td->contacts_manager_->on_deleted_contacts(user_ids_); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->contacts_manager_->reload_contacts(true); } }; class DeleteContactsByPhoneNumberQuery : public Td::ResultHandler { Promise promise_; vector user_ids_; public: explicit DeleteContactsByPhoneNumberQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(vector &&user_phone_numbers, vector &&user_ids) { user_ids_ = std::move(user_ids); send_query(G()->net_query_creator().create( create_storer(telegram_api::contacts_deleteByPhones(std::move(user_phone_numbers))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.ok(); if (!result) { return on_error(id, Status::Error(500, "Some contacts can't be deleted")); } td->contacts_manager_->on_deleted_contacts(user_ids_); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->contacts_manager_->reload_contacts(true); } }; class ResetContactsQuery : public Td::ResultHandler { Promise promise_; public: explicit ResetContactsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::contacts_resetSaved()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.ok(); if (!result) { LOG(ERROR) << "Failed to delete imported contacts"; td->contacts_manager_->reload_contacts(true); } else { td->contacts_manager_->on_update_contacts_reset(); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->contacts_manager_->reload_contacts(true); } }; class UploadProfilePhotoQuery : public Td::ResultHandler { Promise promise_; FileId file_id_; public: explicit UploadProfilePhotoQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(FileId file_id, tl_object_ptr &&input_file) { CHECK(input_file != nullptr); CHECK(file_id.is_valid()); file_id_ = file_id; send_query( G()->net_query_creator().create(create_storer(telegram_api::photos_uploadProfilePhoto(std::move(input_file))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for uploadProfilePhoto: " << to_string(ptr); td->contacts_manager_->on_get_users(std::move(ptr->users_), "UploadProfilePhotoQuery"); // ignore ptr->photo_ td->file_manager_->delete_partial_remote_location(file_id_); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->file_manager_->delete_partial_remote_location(file_id_); td->updates_manager_->get_difference("UploadProfilePhotoQuery"); } }; class UpdateProfilePhotoQuery : public Td::ResultHandler { Promise promise_; FileId file_id_; string file_reference_; public: explicit UpdateProfilePhotoQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(FileId file_id, tl_object_ptr &&input_photo) { CHECK(input_photo != nullptr); file_id_ = file_id; file_reference_ = FileManager::extract_file_reference(input_photo); send_query(G()->net_query_creator().create( create_storer(telegram_api::photos_updateProfilePhoto(std::move(input_photo))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } LOG(DEBUG) << "Receive result for updateProfilePhoto " << to_string(result_ptr.ok()); td->contacts_manager_->on_update_user_photo(td->contacts_manager_->get_my_id(), result_ptr.move_as_ok()); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (FileReferenceManager::is_file_reference_error(status)) { if (file_id_.is_valid()) { td->file_manager_->delete_file_reference(file_id_, file_reference_); td->contacts_manager_->upload_profile_photo(file_id_, std::move(promise_)); return; } else { LOG(ERROR) << "Receive file reference error, but file_id = " << file_id_; } } promise_.set_error(std::move(status)); } }; class DeleteProfilePhotoQuery : public Td::ResultHandler { Promise promise_; int64 profile_photo_id_; public: explicit DeleteProfilePhotoQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(int64 profile_photo_id) { profile_photo_id_ = profile_photo_id; vector> input_photo_ids; input_photo_ids.push_back(make_tl_object(profile_photo_id, 0, BufferSlice())); send_query( G()->net_query_creator().create(create_storer(telegram_api::photos_deletePhotos(std::move(input_photo_ids))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto result = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for deleteProfilePhoto " << format::as_array(result); if (result.size() != 1u) { LOG(WARNING) << "Photo can't be deleted"; return on_error(id, Status::Error(7, "Photo can't be deleted")); } td->contacts_manager_->on_delete_profile_photo(profile_photo_id_, std::move(promise_)); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class UpdateProfileQuery : public Td::ResultHandler { Promise promise_; int32 flags_; string first_name_; string last_name_; string about_; public: explicit UpdateProfileQuery(Promise &&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( create_storer(telegram_api::account_updateProfile(flags, first_name, last_name, about)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } LOG(DEBUG) << "Receive result for updateProfile " << to_string(result_ptr.ok()); td->contacts_manager_->on_get_user(result_ptr.move_as_ok(), "UpdateProfileQuery"); td->contacts_manager_->on_update_profile_success(flags_, first_name_, last_name_, about_); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class CheckUsernameQuery : public Td::ResultHandler { Promise promise_; public: explicit CheckUsernameQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(const string &username) { send_query(G()->net_query_creator().create(create_storer(telegram_api::account_checkUsername(username)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } promise_.set_value(result_ptr.move_as_ok()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class UpdateUsernameQuery : public Td::ResultHandler { Promise promise_; public: explicit UpdateUsernameQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(const string &username) { send_query(G()->net_query_creator().create(create_storer(telegram_api::account_updateUsername(username)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } LOG(DEBUG) << "Receive result for updateUsername " << to_string(result_ptr.ok()); td->contacts_manager_->on_get_user(result_ptr.move_as_ok(), "UpdateUsernameQuery"); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (status.message() == "USERNAME_NOT_MODIFIED" && !td->auth_manager_->is_bot()) { promise_.set_value(Unit()); return; } promise_.set_error(std::move(status)); } }; class ToggleChatAdminsQuery : public Td::ResultHandler { Promise promise_; public: explicit ToggleChatAdminsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChatId chat_id, bool everyone_is_administrator) { send_query(G()->net_query_creator().create( create_storer(telegram_api::messages_toggleChatAdmins(chat_id.get(), !everyone_is_administrator)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for toggleChatAdmins: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (status.message() == "CHAT_NOT_MODIFIED" && !td->auth_manager_->is_bot()) { promise_.set_value(Unit()); return; } promise_.set_error(std::move(status)); } }; class CheckChannelUsernameQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; string username_; public: explicit CheckChannelUsernameQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, const string &username) { channel_id_ = channel_id; tl_object_ptr input_channel; if (channel_id.is_valid()) { input_channel = td->contacts_manager_->get_input_channel(channel_id); } else { input_channel = make_tl_object(); } CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_checkUsername(std::move(input_channel), username)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } promise_.set_value(result_ptr.move_as_ok()); } void on_error(uint64 id, Status status) override { if (channel_id_.is_valid()) { td->contacts_manager_->on_get_channel_error(channel_id_, status, "CheckChannelUsernameQuery"); } promise_.set_error(std::move(status)); } }; class UpdateChannelUsernameQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; string username_; public: explicit UpdateChannelUsernameQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, const string &username) { channel_id_ = channel_id; username_ = username; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_updateUsername(std::move(input_channel), username)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.ok(); LOG(DEBUG) << "Receive result for updateChannelUsername " << result; if (!result) { return on_error(id, Status::Error(500, "Supergroup username is not updated")); } td->contacts_manager_->on_update_channel_username(channel_id_, std::move(username_)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (status.message() == "USERNAME_NOT_MODIFIED" || status.message() == "CHAT_NOT_MODIFIED") { td->contacts_manager_->on_update_channel_username(channel_id_, std::move(username_)); if (!td->auth_manager_->is_bot()) { promise_.set_value(Unit()); return; } } else { td->contacts_manager_->on_get_channel_error(channel_id_, status, "UpdateChannelUsernameQuery"); } promise_.set_error(std::move(status)); } }; class SetChannelStickerSetQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; int64 sticker_set_id_; public: explicit SetChannelStickerSetQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, int64 sticker_set_id, telegram_api::object_ptr &&input_sticker_set) { channel_id_ = channel_id; sticker_set_id_ = sticker_set_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_setStickers(std::move(input_channel), std::move(input_sticker_set))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.ok(); LOG(DEBUG) << "Receive result for setChannelStickerSet " << result; if (!result) { return on_error(id, Status::Error(500, "Supergroup sticker set not updated")); } td->contacts_manager_->on_update_channel_sticker_set(channel_id_, sticker_set_id_); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (status.message() == "CHAT_NOT_MODIFIED") { td->contacts_manager_->on_update_channel_sticker_set(channel_id_, sticker_set_id_); if (!td->auth_manager_->is_bot()) { promise_.set_value(Unit()); return; } } else { td->contacts_manager_->on_get_channel_error(channel_id_, status, "SetChannelStickerSetQuery"); } promise_.set_error(std::move(status)); } }; class ToggleChannelInvitesQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit ToggleChannelInvitesQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, bool anyone_can_invite) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_toggleInvites(std::move(input_channel), anyone_can_invite)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for toggleChannelInvites: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (status.message() == "CHAT_NOT_MODIFIED") { if (!td->auth_manager_->is_bot()) { promise_.set_value(Unit()); return; } } else { td->contacts_manager_->on_get_channel_error(channel_id_, status, "ToggleChannelInvitesQuery"); } promise_.set_error(std::move(status)); } }; class ToggleChannelSignaturesQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit ToggleChannelSignaturesQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, bool sign_messages) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_toggleSignatures(std::move(input_channel), sign_messages)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for toggleChannelSignatures: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (status.message() == "CHAT_NOT_MODIFIED") { if (!td->auth_manager_->is_bot()) { promise_.set_value(Unit()); return; } } else { td->contacts_manager_->on_get_channel_error(channel_id_, status, "ToggleChannelSignaturesQuery"); } promise_.set_error(std::move(status)); } }; class ToggleChannelIsAllHistoryAvailableQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; bool is_all_history_available_; public: explicit ToggleChannelIsAllHistoryAvailableQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, bool is_all_history_available) { channel_id_ = channel_id; is_all_history_available_ = is_all_history_available; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create(create_storer( telegram_api::channels_togglePreHistoryHidden(std::move(input_channel), !is_all_history_available)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for togglePreHistoryHidden: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); td->contacts_manager_->on_update_channel_is_all_history_available(channel_id_, is_all_history_available_); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (status.message() == "CHAT_NOT_MODIFIED") { if (!td->auth_manager_->is_bot()) { promise_.set_value(Unit()); return; } } else { td->contacts_manager_->on_get_channel_error(channel_id_, status, "ToggleChannelIsAllHistoryAvailableQuery"); } promise_.set_error(std::move(status)); } }; class EditChannelAboutQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; string about_; public: explicit EditChannelAboutQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, const string &about) { channel_id_ = channel_id; about_ = about; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_editAbout(std::move(input_channel), about)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.ok(); LOG(DEBUG) << "Receive result for editChannelAbout " << result; if (!result) { return on_error(id, Status::Error(500, "Supergroup description is not updated")); } td->contacts_manager_->on_update_channel_description(channel_id_, std::move(about_)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { if (status.message() == "CHAT_ABOUT_NOT_MODIFIED" || status.message() == "CHAT_NOT_MODIFIED") { td->contacts_manager_->on_update_channel_description(channel_id_, std::move(about_)); if (!td->auth_manager_->is_bot()) { promise_.set_value(Unit()); return; } } else { td->contacts_manager_->on_get_channel_error(channel_id_, status, "EditChannelAboutQuery"); } promise_.set_error(std::move(status)); } }; class ReportChannelSpamQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit ReportChannelSpamQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, UserId user_id, const vector &message_ids) { LOG(INFO) << "Send reportChannelSpamQuery in " << channel_id << " with messages " << format::as_array(message_ids) << " and " << user_id; channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); auto input_user = td->contacts_manager_->get_input_user(user_id); CHECK(input_user != nullptr); send_query(G()->net_query_creator().create(create_storer(telegram_api::channels_reportSpam( std::move(input_channel), std::move(input_user), MessagesManager::get_server_message_ids(message_ids))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.move_as_ok(); LOG_IF(INFO, !result) << "Report spam has failed"; promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "ReportChannelSpamQuery"); status.ignore(); } }; class DeleteChannelQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit DeleteChannelQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query( G()->net_query_creator().create(create_storer(telegram_api::channels_deleteChannel(std::move(input_channel))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for deleteChannel: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "DeleteChannelQuery"); promise_.set_error(std::move(status)); } }; class AddChatUserQuery : public Td::ResultHandler { Promise promise_; public: explicit AddChatUserQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChatId chat_id, tl_object_ptr &&input_user, int32 forward_limit) { send_query(G()->net_query_creator().create( create_storer(telegram_api::messages_addChatUser(chat_id.get(), std::move(input_user), forward_limit)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for addChatUser: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->updates_manager_->get_difference("AddChatUserQuery"); } }; class EditChatAdminQuery : public Td::ResultHandler { Promise promise_; ChatId chat_id_; public: explicit EditChatAdminQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChatId chat_id, tl_object_ptr &&input_user, bool is_administrator) { chat_id_ = chat_id; send_query(G()->net_query_creator().create( create_storer(telegram_api::messages_editChatAdmin(chat_id.get(), std::move(input_user), is_administrator)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } bool result = result_ptr.move_as_ok(); if (!result) { LOG(ERROR) << "Receive false as result of messages.editChatAdmin"; return on_error(id, Status::Error(400, "Can't edit chat administrators")); } // result will come in the updates promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->updates_manager_->get_difference("EditChatAdminQuery"); } }; class ExportChatInviteLinkQuery : public Td::ResultHandler { Promise promise_; ChatId chat_id_; public: explicit ExportChatInviteLinkQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChatId chat_id) { chat_id_ = chat_id; send_query(G()->net_query_creator().create(create_storer(telegram_api::messages_exportChatInvite(chat_id.get())))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for exportChatInvite: " << to_string(ptr); td->contacts_manager_->on_get_chat_invite_link(chat_id_, std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->updates_manager_->get_difference("ExportChatInviteLinkQuery"); } }; class ExportChannelInviteLinkQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit ExportChannelInviteLinkQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query( G()->net_query_creator().create(create_storer(telegram_api::channels_exportInvite(std::move(input_channel))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for exportChannelInvite: " << to_string(ptr); td->contacts_manager_->on_get_channel_invite_link(channel_id_, std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "ExportChannelInviteLinkQuery"); promise_.set_error(std::move(status)); td->updates_manager_->get_difference("ExportChannelInviteLinkQuery"); } }; class CheckDialogInviteLinkQuery : public Td::ResultHandler { Promise promise_; string invite_link_; public: explicit CheckDialogInviteLinkQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(const string &invite_link) { invite_link_ = invite_link; send_query(G()->net_query_creator().create(create_storer( telegram_api::messages_checkChatInvite(ContactsManager::get_dialog_invite_link_hash(invite_link_).str())))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for checkChatInvite: " << to_string(ptr); td->contacts_manager_->on_get_dialog_invite_link_info(invite_link_, std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class ImportDialogInviteLinkQuery : public Td::ResultHandler { Promise promise_; string invite_link_; public: explicit ImportDialogInviteLinkQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(const string &invite_link) { invite_link_ = invite_link; send_query(G()->net_query_creator().create(create_storer( telegram_api::messages_importChatInvite(ContactsManager::get_dialog_invite_link_hash(invite_link).str())))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for importChatInvite: " << to_string(ptr); auto dialog_ids = td->updates_manager_->get_chats(ptr.get()); if (dialog_ids.size() != 1u) { LOG(ERROR) << "Receive wrong result for ImportDialogInviteLinkQuery: " << to_string(ptr); return on_error(id, Status::Error(500, "Internal Server Error")); } td->updates_manager_->on_get_updates(std::move(ptr)); td->contacts_manager_->invalidate_invite_link_info(invite_link_); promise_.set_value(std::move(dialog_ids[0])); } void on_error(uint64 id, Status status) override { td->contacts_manager_->invalidate_invite_link_info(invite_link_); promise_.set_error(std::move(status)); } }; class DeleteChatUserQuery : public Td::ResultHandler { Promise promise_; public: explicit DeleteChatUserQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChatId chat_id, tl_object_ptr &&input_user) { send_query(G()->net_query_creator().create( create_storer(telegram_api::messages_deleteChatUser(chat_id.get(), std::move(input_user))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for deleteChatUser: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->updates_manager_->get_difference("DeleteChatUserQuery"); } }; class JoinChannelQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit JoinChannelQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query( G()->net_query_creator().create(create_storer(telegram_api::channels_joinChannel(std::move(input_channel))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for joinChannel: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "JoinChannelQuery"); promise_.set_error(std::move(status)); td->updates_manager_->get_difference("JoinChannelQuery"); } }; class InviteToChannelQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit InviteToChannelQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, vector> &&input_users) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_inviteToChannel(std::move(input_channel), std::move(input_users))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for inviteToChannel: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); td->contacts_manager_->invalidate_channel_full(channel_id_, false); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "InviteToChannelQuery"); promise_.set_error(std::move(status)); td->updates_manager_->get_difference("InviteToChannelQuery"); } }; class EditChannelAdminQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit EditChannelAdminQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, tl_object_ptr &&input_user, DialogParticipantStatus status) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create(create_storer(telegram_api::channels_editAdmin( std::move(input_channel), std::move(input_user), status.get_channel_admin_rights())))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for editChannelAdmin: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); td->contacts_manager_->invalidate_channel_full(channel_id_, false); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "EditChannelAdminQuery"); promise_.set_error(std::move(status)); td->updates_manager_->get_difference("EditChannelAdminQuery"); } }; class EditChannelBannedQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit EditChannelBannedQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, tl_object_ptr &&input_user, DialogParticipantStatus status) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query(G()->net_query_creator().create(create_storer(telegram_api::channels_editBanned( std::move(input_channel), std::move(input_user), status.get_channel_banned_rights())))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for editChannelBanned: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); td->contacts_manager_->invalidate_channel_full(channel_id_, false); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "EditChannelBannedQuery"); promise_.set_error(std::move(status)); td->updates_manager_->get_difference("EditChannelBannedQuery"); } }; class LeaveChannelQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit LeaveChannelQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id) { channel_id_ = channel_id; auto input_channel = td->contacts_manager_->get_input_channel(channel_id); CHECK(input_channel != nullptr); send_query( G()->net_query_creator().create(create_storer(telegram_api::channels_leaveChannel(std::move(input_channel))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for leaveChannel: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "LeaveChannelQuery"); promise_.set_error(std::move(status)); td->updates_manager_->get_difference("LeaveChannelQuery"); } }; class MigrateChatQuery : public Td::ResultHandler { Promise promise_; public: explicit MigrateChatQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChatId chat_id) { send_query(G()->net_query_creator().create(create_storer(telegram_api::messages_migrateChat(chat_id.get())))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for migrateChat: " << to_string(ptr); td->updates_manager_->on_get_updates(std::move(ptr)); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); td->updates_manager_->get_difference("MigrateChatQuery"); } }; class GetCreatedPublicChannelsQuery : public Td::ResultHandler { Promise promise_; public: explicit GetCreatedPublicChannelsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::channels_getAdminedPublicChannels()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto chats_ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetCreatedPublicChannelsQuery " << to_string(chats_ptr); int32 constructor_id = chats_ptr->get_id(); switch (constructor_id) { case telegram_api::messages_chats::ID: { auto chats = move_tl_object_as(chats_ptr); td->contacts_manager_->on_get_created_public_channels(std::move(chats->chats_)); break; } case telegram_api::messages_chatsSlice::ID: { auto chats = move_tl_object_as(chats_ptr); LOG(ERROR) << "Receive chatsSlice in result of GetCreatedPublicChannelsQuery"; td->contacts_manager_->on_get_created_public_channels(std::move(chats->chats_)); break; } default: UNREACHABLE(); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class GetUsersQuery : public Td::ResultHandler { Promise promise_; public: explicit GetUsersQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(vector> &&input_users) { send_query(G()->net_query_creator().create(create_storer(telegram_api::users_getUsers(std::move(input_users))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } td->contacts_manager_->on_get_users(result_ptr.move_as_ok(), "GetUsersQuery"); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class GetFullUserQuery : public Td::ResultHandler { UserId user_id_; public: void send(UserId user_id, tl_object_ptr &&input_user) { user_id_ = user_id; send_query(G()->net_query_creator().create(create_storer(telegram_api::users_getFullUser(std::move(input_user))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } LOG(DEBUG) << "Receive result for getFullUser " << to_string(result_ptr.ok()); td->contacts_manager_->on_get_user_full(result_ptr.move_as_ok()); td->contacts_manager_->on_get_user_full_success(user_id_); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_user_full_fail(user_id_, std::move(status)); } }; class GetUserPhotosQuery : public Td::ResultHandler { Promise promise_; UserId user_id_; int32 offset_; int32 limit_; public: explicit GetUserPhotosQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(UserId user_id, tl_object_ptr &&input_user, int32 offset, int32 limit, int64 photo_id) { user_id_ = user_id; offset_ = offset; limit_ = limit; LOG(INFO) << "Get " << user_id << " profile photos with offset " << offset << " and limit " << limit << " from photo " << photo_id; send_query(G()->net_query_creator().create( create_storer(telegram_api::photos_getUserPhotos(std::move(input_user), offset, photo_id, limit)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, 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(ptr); td->contacts_manager_->on_get_users(std::move(photos->users_), "GetUserPhotosQuery"); int32 photos_size = narrow_cast(photos->photos_.size()); td->contacts_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(ptr); td->contacts_manager_->on_get_users(std::move(photos->users_), "GetUserPhotosQuery"); td->contacts_manager_->on_get_user_photos(user_id_, offset_, limit_, photos->count_, std::move(photos->photos_)); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class GetChatsQuery : public Td::ResultHandler { Promise promise_; public: explicit GetChatsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(vector &&chat_ids) { send_query(G()->net_query_creator().create(create_storer(telegram_api::messages_getChats(std::move(chat_ids))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto chats_ptr = result_ptr.move_as_ok(); int32 constructor_id = chats_ptr->get_id(); switch (constructor_id) { case telegram_api::messages_chats::ID: { auto chats = move_tl_object_as(chats_ptr); td->contacts_manager_->on_get_chats(std::move(chats->chats_), "GetChatsQuery"); break; } case telegram_api::messages_chatsSlice::ID: { auto chats = move_tl_object_as(chats_ptr); LOG(ERROR) << "Receive chatsSlice in result of GetChatsQuery"; td->contacts_manager_->on_get_chats(std::move(chats->chats_), "GetChatsQuery"); break; } default: UNREACHABLE(); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; class GetFullChatQuery : public Td::ResultHandler { ChatId chat_id_; public: void send(ChatId chat_id) { chat_id_ = chat_id; LOG(INFO) << "Send getFullChat query to get " << chat_id; send_query(G()->net_query_creator().create(create_storer(telegram_api::messages_getFullChat(chat_id.get())))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); // LOG(INFO) << "Receive result for getFullChat query: " << to_string(ptr); td->contacts_manager_->on_get_users(std::move(ptr->users_), "GetFullChatQuery"); td->contacts_manager_->on_get_chats(std::move(ptr->chats_), "GetFullChatQuery"); td->contacts_manager_->on_get_chat_full(std::move(ptr->full_chat_)); td->contacts_manager_->on_get_chat_full_success(chat_id_); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_chat_full_fail(chat_id_, std::move(status)); } }; class GetChannelsQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit GetChannelsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(tl_object_ptr &&input_channel) { CHECK(input_channel != nullptr); if (input_channel->get_id() == telegram_api::inputChannel::ID) { channel_id_ = ChannelId(static_cast(input_channel.get())->channel_id_); } vector> input_channels; input_channels.push_back(std::move(input_channel)); send_query( G()->net_query_creator().create(create_storer(telegram_api::channels_getChannels(std::move(input_channels))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } // LOG(INFO) << "Receive result for getChannels query: " << to_string(result_ptr.ok()); auto chats_ptr = result_ptr.move_as_ok(); int32 constructor_id = chats_ptr->get_id(); switch (constructor_id) { case telegram_api::messages_chats::ID: { auto chats = move_tl_object_as(chats_ptr); td->contacts_manager_->on_get_chats(std::move(chats->chats_), "GetChannelsQuery"); break; } case telegram_api::messages_chatsSlice::ID: { auto chats = move_tl_object_as(chats_ptr); LOG(ERROR) << "Receive chatsSlice in result of GetChannelsQuery"; td->contacts_manager_->on_get_chats(std::move(chats->chats_), "GetChannelsQuery"); break; } default: UNREACHABLE(); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "GetChannelsQuery"); promise_.set_error(std::move(status)); } }; class GetFullChannelQuery : public Td::ResultHandler { ChannelId channel_id_; public: void send(ChannelId channel_id, tl_object_ptr &&input_channel) { channel_id_ = channel_id; send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_getFullChannel(std::move(input_channel))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); // LOG(INFO) << "Receive result for getFullChannel query: " << to_string(ptr); td->contacts_manager_->on_get_users(std::move(ptr->users_), "GetFullChannelQuery"); td->contacts_manager_->on_get_chats(std::move(ptr->chats_), "GetFullChannelQuery"); td->contacts_manager_->on_get_chat_full(std::move(ptr->full_chat_)); td->contacts_manager_->on_get_channel_full_success(channel_id_); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "GetFullChannelQuery"); td->contacts_manager_->on_get_channel_full_fail(channel_id_, std::move(status)); } }; class GetChannelParticipantQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; UserId user_id_; public: explicit GetChannelParticipantQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, UserId user_id, tl_object_ptr &&input_user) { auto input_channel = td->contacts_manager_->get_input_channel(channel_id); if (input_channel == nullptr) { return promise_.set_error(Status::Error(3, "Supergroup not found")); } CHECK(input_user != nullptr); channel_id_ = channel_id; user_id_ = user_id; send_query(G()->net_query_creator().create( create_storer(telegram_api::channels_getParticipant(std::move(input_channel), std::move(input_user))))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto participant = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetChannelParticipantQuery: " << to_string(participant); td->contacts_manager_->on_get_users(std::move(participant->users_), "GetChannelParticipantQuery"); promise_.set_value( td->contacts_manager_->get_dialog_participant(channel_id_, std::move(participant->participant_))); } void on_error(uint64 id, Status status) override { if (status.message() == "USER_NOT_PARTICIPANT") { promise_.set_value({user_id_, UserId(), 0, DialogParticipantStatus::Left()}); return; } td->contacts_manager_->on_get_channel_error(channel_id_, status, "GetChannelParticipantQuery"); promise_.set_error(std::move(status)); } }; class GetChannelParticipantsQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; ChannelParticipantsFilter filter_{nullptr}; int32 offset_; int32 limit_; int64 random_id_; public: explicit GetChannelParticipantsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, ChannelParticipantsFilter filter, int32 offset, int32 limit, int64 random_id) { auto input_channel = td->contacts_manager_->get_input_channel(channel_id); if (input_channel == nullptr) { return promise_.set_error(Status::Error(3, "Supergroup not found")); } channel_id_ = channel_id; filter_ = std::move(filter); offset_ = offset; limit_ = limit; random_id_ = random_id; send_query(G()->net_query_creator().create(create_storer(telegram_api::channels_getParticipants( std::move(input_channel), filter_.get_input_channel_participants_filter(), offset, limit, 0)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto participants_ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetChannelParticipantsQuery with filter " << to_string(filter_.get_input_channel_participants_filter()) << ": " << to_string(participants_ptr); switch (participants_ptr->get_id()) { case telegram_api::channels_channelParticipants::ID: { auto participants = telegram_api::move_object_as(participants_ptr); td->contacts_manager_->on_get_users(std::move(participants->users_), "GetChannelParticipantsQuery"); td->contacts_manager_->on_get_channel_participants_success(channel_id_, std::move(filter_), offset_, limit_, random_id_, participants->count_, std::move(participants->participants_)); break; } case telegram_api::channels_channelParticipantsNotModified::ID: LOG(ERROR) << "Receive channelParticipantsNotModified"; break; default: UNREACHABLE(); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "GetChannelParticipantsQuery"); td->contacts_manager_->on_get_channel_participants_fail(channel_id_, std::move(filter_), offset_, limit_, random_id_); promise_.set_error(std::move(status)); } }; class GetChannelAdministratorsQuery : public Td::ResultHandler { Promise promise_; ChannelId channel_id_; public: explicit GetChannelAdministratorsQuery(Promise &&promise) : promise_(std::move(promise)) { } void send(ChannelId channel_id, int32 hash) { auto input_channel = td->contacts_manager_->get_input_channel(channel_id); if (input_channel == nullptr) { return promise_.set_error(Status::Error(3, "Supergroup not found")); } channel_id_ = channel_id; send_query(G()->net_query_creator().create(create_storer(telegram_api::channels_getParticipants( std::move(input_channel), telegram_api::make_object(), 0, std::numeric_limits::max(), hash)))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto participants_ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetChannelAdministratorsQuery: " << to_string(participants_ptr); switch (participants_ptr->get_id()) { case telegram_api::channels_channelParticipants::ID: { auto participants = telegram_api::move_object_as(participants_ptr); td->contacts_manager_->on_get_users(std::move(participants->users_), "GetChannelAdministratorsQuery"); vector administrator_user_ids; administrator_user_ids.reserve(participants->participants_.size()); for (auto &participant : participants->participants_) { UserId user_id; downcast_call(*participant, [&user_id](auto &participant) { user_id = UserId(participant.user_id_); }); if (user_id.is_valid()) { administrator_user_ids.push_back(user_id); } } td->contacts_manager_->on_update_dialog_administrators(DialogId(channel_id_), std::move(administrator_user_ids), true); break; } case telegram_api::channels_channelParticipantsNotModified::ID: break; default: UNREACHABLE(); } promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { td->contacts_manager_->on_get_channel_error(channel_id_, status, "GetChannelAdministratorsQuery"); promise_.set_error(std::move(status)); } }; class GetSupportUserQuery : public Td::ResultHandler { Promise promise_; public: explicit GetSupportUserQuery(Promise &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::help_getSupport()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto ptr = result_ptr.move_as_ok(); LOG(INFO) << "Receive result for GetSupportUserQuery: " << to_string(ptr); td->contacts_manager_->on_get_user(std::move(ptr->user_), "GetSupportUserQuery", false, true); promise_.set_value(Unit()); } void on_error(uint64 id, Status status) override { promise_.set_error(std::move(status)); } }; StringBuilder &operator<<(StringBuilder &string_builder, ContactsManager::LinkState link_state) { switch (link_state) { case ContactsManager::LinkState::Unknown: string_builder << "unknown"; break; case ContactsManager::LinkState::None: string_builder << "none"; break; case ContactsManager::LinkState::KnowsPhoneNumber: string_builder << "knows phone number"; break; case ContactsManager::LinkState::Contact: string_builder << "contact"; break; } return string_builder; } bool ContactsManager::UserFull::is_bot_info_expired(int32 bot_info_version) const { return bot_info_version != -1 && (bot_info == nullptr || bot_info->version != bot_info_version); } bool ContactsManager::UserFull::is_expired() const { return expires_at < Time::now(); } bool ContactsManager::ChannelFull::is_expired() const { return expires_at < Time::now(); } class ContactsManager::UploadProfilePhotoCallback : public FileManager::UploadCallback { public: void on_upload_ok(FileId file_id, tl_object_ptr input_file) override { send_closure_later(G()->contacts_manager(), &ContactsManager::on_upload_profile_photo, file_id, std::move(input_file)); } void on_upload_encrypted_ok(FileId file_id, tl_object_ptr input_file) override { UNREACHABLE(); } void on_upload_secure_ok(FileId file_id, tl_object_ptr input_file) override { UNREACHABLE(); } void on_upload_error(FileId file_id, Status error) override { send_closure_later(G()->contacts_manager(), &ContactsManager::on_upload_profile_photo_error, file_id, std::move(error)); } }; const CSlice ContactsManager::INVITE_LINK_URLS[3] = {"t.me/joinchat/", "telegram.me/joinchat/", "telegram.dog/joinchat/"}; ContactsManager::ContactsManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) { upload_profile_photo_callback_ = std::make_shared(); my_id_ = load_my_id(); if (G()->parameters().use_chat_info_db) { 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(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(saved_contact_count_string); } } else { G()->td_db()->get_binlog_pmc()->erase("next_contacts_sync_date"); G()->td_db()->get_binlog_pmc()->erase("saved_contact_count"); } was_online_local_ = to_integer(G()->td_db()->get_binlog_pmc()->get("my_was_online_local")); was_online_remote_ = to_integer(G()->td_db()->get_binlog_pmc()->get("my_was_online_remote")); if (was_online_local_ >= G()->unix_time_cached() && !td_->is_online()) { was_online_local_ = G()->unix_time_cached() - 1; } user_online_timeout_.set_callback(on_user_online_timeout_callback); user_online_timeout_.set_callback_data(static_cast(this)); channel_unban_timeout_.set_callback(on_channel_unban_timeout_callback); channel_unban_timeout_.set_callback_data(static_cast(this)); } void ContactsManager::tear_down() { parent_.reset(); } UserId ContactsManager::load_my_id() { auto id_string = G()->td_db()->get_binlog_pmc()->get("my_id"); if (!id_string.empty()) { UserId my_id(to_integer(id_string)); if (my_id.is_valid()) { return my_id; } my_id = UserId(to_integer(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(); } void ContactsManager::on_user_online_timeout_callback(void *contacts_manager_ptr, int64 user_id_long) { auto contacts_manager = static_cast(contacts_manager_ptr); UserId user_id(narrow_cast(user_id_long)); auto u = contacts_manager->get_user(user_id); CHECK(u != nullptr); LOG(INFO) << "Update " << user_id << " online status to offline"; send_closure_later( G()->td(), &Td::send_update, make_tl_object(user_id.get(), contacts_manager->get_user_status_object(user_id, u))); } void ContactsManager::on_channel_unban_timeout_callback(void *contacts_manager_ptr, int64 channel_id_long) { auto td = static_cast(contacts_manager_ptr)->td_; send_closure_later(td->actor_id(td), &Td::on_channel_unban_timeout, channel_id_long); } void ContactsManager::on_channel_unban_timeout(ChannelId channel_id) { auto c = get_channel(channel_id); CHECK(c != nullptr); auto old_status = c->status; c->status.update_restrictions(); if (c->status == old_status) { LOG_IF(ERROR, c->status.is_restricted() || c->status.is_banned()) << "Status of " << channel_id << " wasn't updated: " << c->status; } else { c->need_send_update = true; } LOG(INFO) << "Update " << channel_id << " status"; c->is_status_changed = true; invalidate_channel_full(channel_id, false); update_channel(c, channel_id); // always call, because in case of failure we need to reactivate timeout } template void ContactsManager::store_link_state(const LinkState &link_state, StorerT &storer) { store(static_cast(link_state), storer); } template void ContactsManager::parse_link_state(LinkState &link_state, ParserT &parser) { uint32 link_state_uint32; parse(link_state_uint32, parser); link_state = static_cast(static_cast(link_state_uint32)); } template void ContactsManager::User::store(StorerT &storer) const { using td::store; bool has_last_name = !last_name.empty(); bool has_username = !username.empty(); bool has_photo = photo.small_file_id.is_valid(); bool is_restricted = !restriction_reason.empty(); bool has_language_code = !language_code.empty(); bool have_access_hash = access_hash != -1; 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(has_username); STORE_FLAG(has_photo); STORE_FLAG(is_restricted); STORE_FLAG(has_language_code); STORE_FLAG(have_access_hash); END_STORE_FLAGS(); store(first_name, storer); if (has_last_name) { store(last_name, storer); } if (has_username) { store(username, storer); } store(phone_number, storer); if (have_access_hash) { store(access_hash, storer); } if (has_photo) { store(photo, storer); } store_link_state(inbound, storer); store_link_state(outbound, storer); store(was_online, storer); if (is_restricted) { store(restriction_reason, 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); } } template void ContactsManager::User::parse(ParserT &parser) { using td::parse; bool has_last_name; bool has_username; bool has_photo; bool is_restricted; bool has_language_code; bool have_access_hash; 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(has_username); PARSE_FLAG(has_photo); PARSE_FLAG(is_restricted); PARSE_FLAG(has_language_code); PARSE_FLAG(have_access_hash); END_PARSE_FLAGS(); parse(first_name, parser); if (has_last_name) { parse(last_name, parser); } if (has_username) { parse(username, parser); } parse(phone_number, parser); if (parser.version() < static_cast(Version::FixMinUsers)) { have_access_hash = is_received; } if (have_access_hash) { parse(access_hash, parser); } if (has_photo) { parse(photo, parser); } parse_link_state(inbound, parser); parse_link_state(outbound, parser); parse(was_online, parser); if (is_restricted) { parse(restriction_reason, 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 (first_name.empty() && last_name.empty()) { first_name = phone_number; } } template void ContactsManager::Chat::store(StorerT &storer) const { using td::store; bool has_photo = photo.small_file_id.is_valid(); BEGIN_STORE_FLAGS(); STORE_FLAG(left); STORE_FLAG(kicked); STORE_FLAG(is_creator); STORE_FLAG(is_administrator); STORE_FLAG(everyone_is_administrator); STORE_FLAG(can_edit); STORE_FLAG(is_active); STORE_FLAG(has_photo); END_STORE_FLAGS(); store(title, storer); if (has_photo) { store(photo, storer); } store(participant_count, storer); store(date, storer); store(migrated_to_channel_id, storer); store(version, storer); } template void ContactsManager::Chat::parse(ParserT &parser) { using td::parse; bool has_photo; BEGIN_PARSE_FLAGS(); PARSE_FLAG(left); PARSE_FLAG(kicked); PARSE_FLAG(is_creator); PARSE_FLAG(is_administrator); PARSE_FLAG(everyone_is_administrator); PARSE_FLAG(can_edit); PARSE_FLAG(is_active); PARSE_FLAG(has_photo); END_PARSE_FLAGS(); parse(title, parser); if (has_photo) { parse(photo, parser); } parse(participant_count, parser); parse(date, parser); parse(migrated_to_channel_id, parser); parse(version, parser); } template void ContactsManager::Channel::store(StorerT &storer) const { using td::store; bool has_photo = photo.small_file_id.is_valid(); bool has_username = !username.empty(); bool is_restricted = !restriction_reason.empty(); bool use_new_rights = true; bool have_participant_count = participant_count != 0; BEGIN_STORE_FLAGS(); STORE_FLAG(false); STORE_FLAG(false); STORE_FLAG(anyone_can_invite); STORE_FLAG(sign_messages); STORE_FLAG(false); STORE_FLAG(false); STORE_FLAG(false); STORE_FLAG(is_megagroup); STORE_FLAG(is_verified); STORE_FLAG(has_photo); STORE_FLAG(has_username); STORE_FLAG(is_restricted); STORE_FLAG(use_new_rights); STORE_FLAG(have_participant_count); END_STORE_FLAGS(); store(status, storer); store(access_hash, storer); store(title, storer); if (has_photo) { store(photo, storer); } if (has_username) { store(username, storer); } store(date, storer); if (is_restricted) { store(restriction_reason, storer); } if (have_participant_count) { store(participant_count, storer); } } template void ContactsManager::Channel::parse(ParserT &parser) { using td::parse; bool has_photo; bool has_username; bool is_restricted; bool left; bool kicked; bool is_creator; bool can_edit; bool can_moderate; bool use_new_rights; bool have_participant_count; BEGIN_PARSE_FLAGS(); PARSE_FLAG(left); PARSE_FLAG(kicked); PARSE_FLAG(anyone_can_invite); PARSE_FLAG(sign_messages); PARSE_FLAG(is_creator); PARSE_FLAG(can_edit); PARSE_FLAG(can_moderate); PARSE_FLAG(is_megagroup); PARSE_FLAG(is_verified); PARSE_FLAG(has_photo); PARSE_FLAG(has_username); PARSE_FLAG(is_restricted); PARSE_FLAG(use_new_rights); PARSE_FLAG(have_participant_count); END_PARSE_FLAGS(); if (use_new_rights) { parse(status, parser); } else { if (kicked) { status = DialogParticipantStatus::Banned(0); } else if (left) { status = DialogParticipantStatus::Left(); } else if (is_creator) { status = DialogParticipantStatus::Creator(true); } else if (can_edit || can_moderate) { status = DialogParticipantStatus::ChannelAdministrator(false, is_megagroup); } else { status = DialogParticipantStatus::Member(); } } parse(access_hash, parser); parse(title, parser); if (has_photo) { parse(photo, parser); } if (has_username) { parse(username, parser); } parse(date, parser); if (is_restricted) { parse(restriction_reason, parser); } if (have_participant_count) { parse(participant_count, parser); } } template void ContactsManager::SecretChat::store(StorerT &storer) const { using td::store; bool has_layer = layer > SecretChatActor::DEFAULT_LAYER; BEGIN_STORE_FLAGS(); STORE_FLAG(is_outbound); STORE_FLAG(has_layer); 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); } } template void ContactsManager::SecretChat::parse(ParserT &parser) { using td::parse; bool has_layer; BEGIN_PARSE_FLAGS(); PARSE_FLAG(is_outbound); PARSE_FLAG(has_layer); END_PARSE_FLAGS(); if (parser.version() >= static_cast(Version::AddAccessHashToSecretChat)) { parse(access_hash, parser); } parse(user_id, parser); parse(state, parser); parse(ttl, parser); parse(date, parser); if (parser.version() >= static_cast(Version::AddKeyHashToSecretChat)) { parse(key_hash, parser); } if (has_layer) { parse(layer, parser); } else { layer = SecretChatActor::DEFAULT_LAYER; } } tl_object_ptr ContactsManager::get_input_user(UserId user_id) const { if (user_id == get_my_id()) { return make_tl_object(); } const User *u = get_user(user_id); if (u == nullptr || u->access_hash == -1) { if (td_->auth_manager_->is_bot() && user_id.is_valid()) { return make_tl_object(user_id.get(), 0); } return nullptr; } return make_tl_object(user_id.get(), u->access_hash); } bool ContactsManager::have_input_user(UserId user_id) const { if (user_id == get_my_id()) { return true; } const User *u = get_user(user_id); if (u == nullptr || u->access_hash == -1) { if (td_->auth_manager_->is_bot() && user_id.is_valid()) { return true; } return false; } return true; } tl_object_ptr ContactsManager::get_input_channel(ChannelId channel_id) const { const Channel *c = get_channel(channel_id); if (c == nullptr) { if (td_->auth_manager_->is_bot() && channel_id.is_valid()) { return make_tl_object(channel_id.get(), 0); } return nullptr; } return make_tl_object(channel_id.get(), c->access_hash); } bool ContactsManager::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), access_rights); } bool ContactsManager::have_input_peer_user(const User *user, AccessRights access_rights) { if (user == nullptr) { return false; } if (user->access_hash == -1) { return false; } if (access_rights == AccessRights::Read) { return true; } if (user->is_deleted) { return false; } return true; } tl_object_ptr ContactsManager::get_input_peer_user(UserId user_id, AccessRights access_rights) const { if (user_id == get_my_id()) { return make_tl_object(); } const User *u = get_user(user_id); if (!have_input_peer_user(u, access_rights)) { return nullptr; } return make_tl_object(user_id.get(), u->access_hash); } bool ContactsManager::have_input_peer_chat(ChatId chat_id, AccessRights access_rights) const { return have_input_peer_chat(get_chat(chat_id), access_rights); } bool ContactsManager::have_input_peer_chat(const Chat *chat, AccessRights access_rights) { if (chat == nullptr) { return false; } if (access_rights == AccessRights::Read) { return true; } if (chat->left) { return false; } if (access_rights == AccessRights::Write && !chat->is_active) { return false; } return true; } tl_object_ptr ContactsManager::get_input_peer_chat(ChatId chat_id, AccessRights access_rights) const { auto c = get_chat(chat_id); if (!have_input_peer_chat(c, access_rights)) { return nullptr; } return make_tl_object(chat_id.get()); } bool ContactsManager::have_input_peer_channel(ChannelId channel_id, AccessRights access_rights) const { const Channel *c = get_channel(channel_id); return have_input_peer_channel(c, access_rights); } tl_object_ptr ContactsManager::get_input_peer_channel(ChannelId channel_id, AccessRights access_rights) const { const Channel *c = get_channel(channel_id); if (!have_input_peer_channel(c, access_rights)) { return nullptr; } return make_tl_object(channel_id.get(), c->access_hash); } bool ContactsManager::have_input_peer_channel(const Channel *c, AccessRights access_rights) { if (c == nullptr) { return false; } if (c->status.is_creator()) { return true; } if (c->status.is_banned()) { return false; } if (!c->username.empty() && access_rights == AccessRights::Read) { return true; } if (!c->status.is_member()) { return false; } return true; } bool ContactsManager::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 ContactsManager::have_input_encrypted_peer(const SecretChat *secret_chat, AccessRights access_rights) { if (secret_chat == nullptr) { return false; } if (access_rights == AccessRights::Read) { return true; } return secret_chat->state == SecretChatState::Active; } tl_object_ptr ContactsManager::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 make_tl_object(secret_chat_id.get(), sc->access_hash); } const DialogPhoto *ContactsManager::get_user_dialog_photo(UserId user_id) { auto u = get_user(user_id); if (u == nullptr) { return nullptr; } auto it = pending_user_photos_.find(user_id); if (it != pending_user_photos_.end()) { do_update_user_photo(u, user_id, std::move(it->second)); pending_user_photos_.erase(it); update_user(u, user_id); } return &u->photo; } const DialogPhoto *ContactsManager::get_chat_dialog_photo(ChatId chat_id) const { auto c = get_chat(chat_id); if (c == nullptr) { return nullptr; } return &c->photo; } const DialogPhoto *ContactsManager::get_channel_dialog_photo(ChannelId channel_id) const { auto c = get_channel(channel_id); if (c == nullptr) { return nullptr; } return &c->photo; } const DialogPhoto *ContactsManager::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); } string ContactsManager::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 ContactsManager::get_chat_title(ChatId chat_id) const { auto c = get_chat(chat_id); if (c == nullptr) { return string(); } return c->title; } string ContactsManager::get_channel_title(ChannelId channel_id) const { auto c = get_channel(channel_id); if (c == nullptr) { return string(); } return c->title; } string ContactsManager::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); } int32 ContactsManager::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 ContactsManager::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; } bool ContactsManager::default_can_report_spam_in_secret_chat(SecretChatId secret_chat_id) const { auto c = get_secret_chat(secret_chat_id); if (c == nullptr) { return true; } if (c->is_outbound) { return false; } auto u = get_user(c->user_id); if (u == nullptr) { return true; } if (u->outbound == LinkState::Contact) { return false; } return true; } string ContactsManager::get_user_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->username; } string ContactsManager::get_secret_chat_username(SecretChatId secret_chat_id) const { auto c = get_secret_chat(secret_chat_id); if (c == nullptr) { return string(); } return get_user_username(c->user_id); } string ContactsManager::get_channel_username(ChannelId channel_id) const { auto c = get_channel(channel_id); if (c == nullptr) { return string(); } return c->username; } UserId ContactsManager::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 ContactsManager::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 ContactsManager::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 ContactsManager::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; } UserId ContactsManager::get_my_id() const { LOG_IF(ERROR, !my_id_.is_valid()) << "Wrong or unknown my id returned"; return my_id_; } void ContactsManager::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; } 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())); G()->shared_config().set_option_integer("my_id", my_id_.get()); } } void ContactsManager::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); if (u != nullptr) { int32 new_online; int32 now = G()->unix_time(); if (is_online) { new_online = now + 300; } else { new_online = now - 1; } 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; u->is_status_changed = true; } } 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->is_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); } } } ContactsManager::MyOnlineStatusInfo ContactsManager::get_my_online_status() const { MyOnlineStatusInfo status_info; status_info.is_online_local = td_->is_online(); status_info.is_online_remote = was_online_remote_ > G()->unix_time_cached(); status_info.was_online_local = was_online_local_; status_info.was_online_remote = was_online_remote_; return status_info; } UserId ContactsManager::get_service_notifications_user_id() { UserId user_id(777000); if (!have_user_force(user_id)) { LOG(FATAL) << "Failed to load service notification user"; } return user_id; } void ContactsManager::check_dialog_username(DialogId dialog_id, const string &username, Promise &&promise) { if (dialog_id != DialogId() && !dialog_id.is_valid()) { return promise.set_error(Status::Error(3, "Chat not found")); } switch (dialog_id.get_type()) { case DialogType::User: { if (dialog_id.get_user_id() != get_my_id()) { return promise.set_error(Status::Error(3, "Can't check username for private chat with other user")); } break; } case DialogType::Channel: { auto c = get_channel(dialog_id.get_channel_id()); if (c == nullptr) { return promise.set_error(Status::Error(6, "Chat not found")); } if (!get_channel_status(c).is_creator()) { return promise.set_error(Status::Error(6, "Not enough rights to change username")); } if (username == c->username) { return promise.set_value(CheckDialogUsernameResult::Ok); } break; } case DialogType::None: break; case DialogType::Chat: case DialogType::SecretChat: if (username.empty()) { return promise.set_value(CheckDialogUsernameResult::Ok); } return promise.set_error(Status::Error(3, "Chat can't have username")); default: UNREACHABLE(); return; } if (username.empty()) { return promise.set_value(CheckDialogUsernameResult::Ok); } if (!is_valid_username(username)) { return promise.set_value(CheckDialogUsernameResult::Invalid); } auto request_promise = PromiseCreator::lambda([promise = std::move(promise)](Result result) mutable { if (result.is_error()) { auto error = result.move_as_error(); if (error.message() == "CHANNEL_PUBLIC_GROUP_NA") { return promise.set_value(CheckDialogUsernameResult::PublicGroupsUnavailable); } if (error.message() == "CHANNELS_ADMIN_PUBLIC_TOO_MUCH") { return promise.set_value(CheckDialogUsernameResult::PublicDialogsTooMuch); } if (error.message() == "USERNAME_INVALID") { return promise.set_value(CheckDialogUsernameResult::Invalid); } return promise.set_error(std::move(error)); } promise.set_value(result.ok() ? CheckDialogUsernameResult::Ok : CheckDialogUsernameResult::Occupied); }); switch (dialog_id.get_type()) { case DialogType::User: return td_->create_handler(std::move(request_promise))->send(username); case DialogType::Channel: return td_->create_handler(std::move(request_promise)) ->send(dialog_id.get_channel_id(), username); case DialogType::None: return td_->create_handler(std::move(request_promise))->send(ChannelId(), username); case DialogType::Chat: case DialogType::SecretChat: default: UNREACHABLE(); } } td_api::object_ptr ContactsManager::get_check_chat_username_result_object( CheckDialogUsernameResult result) { switch (result) { case CheckDialogUsernameResult::Ok: return td_api::make_object(); case CheckDialogUsernameResult::Invalid: return td_api::make_object(); case CheckDialogUsernameResult::Occupied: return td_api::make_object(); case CheckDialogUsernameResult::PublicDialogsTooMuch: return td_api::make_object(); case CheckDialogUsernameResult::PublicGroupsUnavailable: return td_api::make_object(); default: UNREACHABLE(); return nullptr; } } void ContactsManager::set_account_ttl(int32 account_ttl, Promise &&promise) const { td_->create_handler(std::move(promise))->send(account_ttl); } void ContactsManager::get_account_ttl(Promise &&promise) const { td_->create_handler(std::move(promise))->send(); } void ContactsManager::get_active_sessions(Promise> &&promise) const { td_->create_handler(std::move(promise))->send(); } void ContactsManager::terminate_session(int64 session_id, Promise &&promise) const { td_->create_handler(std::move(promise))->send(session_id); } void ContactsManager::terminate_all_other_sessions(Promise &&promise) const { td_->create_handler(std::move(promise))->send(); } void ContactsManager::get_connected_websites(Promise> &&promise) const { td_->create_handler(std::move(promise))->send(); } void ContactsManager::disconnect_website(int64 website_id, Promise &&promise) const { td_->create_handler(std::move(promise))->send(website_id); } void ContactsManager::disconnect_all_websites(Promise &&promise) const { td_->create_handler(std::move(promise))->send(); } Status ContactsManager::block_user(UserId user_id) { if (user_id == get_my_id()) { return Status::Error(5, "Can't block self"); } auto user = get_input_user(user_id); if (user == nullptr) { return Status::Error(5, "User not found"); } td_->create_handler()->send(std::move(user)); on_update_user_blocked(user_id, true); return Status::OK(); } Status ContactsManager::unblock_user(UserId user_id) { if (user_id == get_my_id()) { return Status::Error(5, "Can't unblock self"); } auto user = get_input_user(user_id); if (user == nullptr) { return Status::Error(5, "User not found"); } td_->create_handler()->send(std::move(user)); on_update_user_blocked(user_id, false); return Status::OK(); } bool ContactsManager::is_valid_username(const string &username) { if (username.size() < 5 || username.size() > 32) { return false; } if (!is_alpha(username[0])) { return false; } for (auto c : username) { if (!is_alpha(c) && !is_digit(c) && c != '_') { return false; } } if (username.back() == '_') { return false; } for (size_t i = 1; i < username.size(); i++) { if (username[i - 1] == '_' && username[i] == '_') { return false; } } if (username.find("admin") == 0 || username.find("telegram") == 0 || username.find("support") == 0 || username.find("security") == 0 || username.find("settings") == 0 || username.find("contacts") == 0 || username.find("service") == 0 || username.find("telegraph") == 0) { return false; } return true; } int64 ContactsManager::get_blocked_users(int32 offset, int32 limit, Promise &&promise) { LOG(INFO) << "Get blocked users with offset = " << offset << " and limit = " << limit; if (offset < 0) { promise.set_error(Status::Error(3, "Parameter offset must be non-negative")); return 0; } if (limit <= 0) { promise.set_error(Status::Error(3, "Parameter limit must be positive")); return 0; } int64 random_id; do { random_id = Random::secure_int64(); } while (random_id == 0 || found_blocked_users_.find(random_id) != found_blocked_users_.end()); found_blocked_users_[random_id]; // reserve place for result td_->create_handler(std::move(promise))->send(offset, limit, random_id); return random_id; } void ContactsManager::on_get_blocked_users_result(int32 offset, int32 limit, int64 random_id, int32 total_count, vector> &&blocked_users) { LOG(INFO) << "Receive " << blocked_users.size() << " blocked users out of " << total_count; auto it = found_blocked_users_.find(random_id); CHECK(it != found_blocked_users_.end()); auto &result = it->second.second; CHECK(result.empty()); for (auto &blocked_user : blocked_users) { CHECK(blocked_user != nullptr); UserId user_id(blocked_user->user_id_); if (have_user(user_id)) { result.push_back(user_id); } else { LOG(ERROR) << "Have no info about " << user_id; } } it->second.first = total_count; } void ContactsManager::on_failed_get_blocked_users(int64 random_id) { auto it = found_blocked_users_.find(random_id); CHECK(it != found_blocked_users_.end()); found_blocked_users_.erase(it); } tl_object_ptr ContactsManager::get_blocked_users_object(int64 random_id) { auto it = found_blocked_users_.find(random_id); CHECK(it != found_blocked_users_.end()); auto result = get_users_object(it->second.first, it->second.second); found_blocked_users_.erase(it); return result; } void ContactsManager::load_contacts(Promise &&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"; promise.set_value(Unit()); return; } load_contacts_queries_.push_back(std::move(promise)); if (load_contacts_queries_.size() == 1u) { if (G()->parameters().use_chat_info_db && 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()->contacts_manager(), &ContactsManager::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"; } } int32 ContactsManager::get_contacts_hash() { if (!are_contacts_loaded_) { return 0; } vector 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); if (u != nullptr && u->outbound == LinkState::Contact) { user_ids.insert(std::upper_bound(user_ids.begin(), user_ids.end(), my_id.get()), my_id.get()); } vector numbers; numbers.reserve(user_ids.size() + 1); numbers.push_back(saved_contact_count_); for (auto user_id : user_ids) { numbers.push_back(narrow_cast(user_id)); } return get_vector_hash(numbers); } void ContactsManager::reload_contacts(bool force) { if (!td_->auth_manager_->is_bot() && next_contacts_sync_date_ != std::numeric_limits::max() && (next_contacts_sync_date_ < G()->unix_time() || force)) { next_contacts_sync_date_ = std::numeric_limits::max(); td_->create_handler()->send(get_contacts_hash()); } } std::pair, vector> ContactsManager::import_contacts( const vector> &contacts, int64 &random_id, Promise &&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; } for (auto &contact : contacts) { if (contact == nullptr) { promise.set_error(Status::Error(400, "Imported contacts should not be empty")); return {}; } } do { random_id = Random::secure_int64(); } while (random_id == 0 || imported_contacts_.find(random_id) != imported_contacts_.end()); imported_contacts_[random_id]; // reserve place for result td_->create_handler(std::move(promise)) ->send(transform(contacts, [](const tl_object_ptr &contact) { return Contact(contact->phone_number_, contact->first_name_, contact->last_name_, string(), 0); }), random_id); return {}; } void ContactsManager::remove_contacts(vector user_ids, Promise &&promise) { LOG(INFO) << "Delete contacts: " << format::as_array(user_ids); if (!are_contacts_loaded_) { load_contacts(std::move(promise)); return; } vector to_delete_user_ids; vector> input_users; for (auto &user_id : user_ids) { const User *u = get_user(user_id); if (u != nullptr && u->outbound == LinkState::Contact) { auto input_user = get_input_user(user_id); if (input_user != nullptr) { to_delete_user_ids.push_back(user_id); input_users.push_back(std::move(input_user)); } } } if (input_users.empty()) { return promise.set_value(Unit()); } td_->create_handler(std::move(promise)) ->send(std::move(to_delete_user_ids), std::move(input_users)); } void ContactsManager::remove_contacts_by_phone_number(vector user_phone_numbers, vector user_ids, Promise &&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(std::move(promise)) ->send(std::move(user_phone_numbers), std::move(user_ids)); } int32 ContactsManager::get_imported_contact_count(Promise &&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 ContactsManager::load_imported_contacts(Promise &&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()->parameters().use_chat_info_db) { 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()->contacts_manager(), &ContactsManager::on_load_imported_contacts_from_database, std::move(value)); })); } else { LOG(INFO) << "Have no previously imported contacts"; send_closure_later(G()->contacts_manager(), &ContactsManager::on_load_imported_contacts_from_database, string()); } } else { LOG(INFO) << "Load imported contacts request has already been sent"; } } void ContactsManager::on_load_imported_contacts_from_database(string value) { 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 { log_event_parse(all_imported_contacts_, value).ensure(); LOG(INFO) << "Successfully loaded " << all_imported_contacts_.size() << " imported contacts from database"; } load_imported_contact_users_multipromise_.add_promise(PromiseCreator::lambda([](Result<> result) { if (result.is_ok()) { send_closure_later(G()->contacts_manager(), &ContactsManager::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 ContactsManager::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; auto promises = std::move(load_imported_contacts_queries_); load_imported_contacts_queries_.clear(); for (auto &promise : promises) { promise.set_value(Unit()); } } std::pair, vector> ContactsManager::change_imported_contacts( vector> &&contacts, int64 &random_id, Promise &&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()->parameters().use_chat_info_db) { 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 {}; } for (auto &contact : contacts) { if (contact == nullptr) { promise.set_error(Status::Error(400, "Contacts should not be empty")); return {}; } } auto new_contacts = transform(std::move(contacts), [](tl_object_ptr &&contact) { return Contact(std::move(contact->phone_number_), std::move(contact->first_name_), std::move(contact->last_name_), string(), 0); }); vector new_contacts_unique_id(new_contacts.size()); vector unique_new_contacts; unique_new_contacts.reserve(new_contacts.size()); std::unordered_map different_new_contacts; std::unordered_set different_new_phone_numbers; size_t unique_size = 0; for (size_t i = 0; i < new_contacts.size(); i++) { auto it_success = different_new_contacts.emplace(std::move(new_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 to_delete; vector 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> to_add; for (auto &new_contact : different_new_contacts) { to_add.first.push_back(new_contact.second); to_add.second.push_back(std::move(new_contact.first)); } if (to_add.first.empty() && to_delete.empty()) { for (size_t i = 0; i < new_contacts.size(); i++) { auto unique_id = new_contacts_unique_id[i]; new_contacts[i].set_user_id(unique_new_contacts[unique_id].get_user_id()); } promise.set_value(Unit()); return {transform(new_contacts, [&](const Contact &contact) { return contact.get_user_id(); }), vector(new_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()->contacts_manager(), &ContactsManager::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 ContactsManager::on_clear_imported_contacts(vector &&contacts, vector contacts_unique_id, std::pair, vector> &&to_add, Promise &&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); td_->create_handler(std::move(promise))->send(std::move(to_add.second), 0); } void ContactsManager::clear_imported_contacts(Promise &&promise) { LOG(INFO) << "Delete imported contacts"; if (saved_contact_count_ == 0) { promise.set_value(Unit()); return; } td_->create_handler(std::move(promise))->send(); } void ContactsManager::on_update_contacts_reset() { /* UserId my_id = get_my_id(); for (auto &p : users_) { UserId user_id = p.first; User &u = p.second; bool is_contact = u.outbound == LinkState::Contact; if (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_links(&u, user_id, LinkState::KnowsPhoneNumber, u.inbound); update_user(&u, user_id); CHECK(u.outbound != LinkState::Contact); if (user_id != my_id) { CHECK(!contacts_hints_.has_key(user_id.get())); } } } */ saved_contact_count_ = 0; if (G()->parameters().use_chat_info_db) { 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_) { CHECK(all_imported_contacts_.empty()); if (load_imported_contacts_queries_.empty()) { LOG(INFO) << "Imported contacts was never loaded, just clear them"; } else { LOG(INFO) << "Imported contacts are being loaded, clear them also when they will be loaded"; need_clear_imported_contacts_ = true; } } else { if (!are_imported_contacts_changing_) { LOG(INFO) << "Imported contacts was loaded, but aren't changing now, just clear them"; all_imported_contacts_.clear(); } else { LOG(INFO) << "Imported contacts are changing now, clear them also after they will be loaded"; need_clear_imported_contacts_ = true; } } reload_contacts(true); } std::pair> ContactsManager::search_contacts(const string &query, int32 limit, Promise &&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> result; if (query.empty()) { result = contacts_hints_.search_empty(limit); } else { result = contacts_hints_.search(query, limit); } vector user_ids; user_ids.reserve(result.second.size()); for (auto key : result.second) { user_ids.emplace_back(narrow_cast(key)); } promise.set_value(Unit()); return {narrow_cast(result.first), std::move(user_ids)}; } void ContactsManager::set_profile_photo(const tl_object_ptr &input_photo, Promise &&promise) { auto r_file_id = td_->file_manager_->get_input_file_id(FileType::Photo, input_photo, DialogId(get_my_id()), false, false); if (r_file_id.is_error()) { // TODO promise.set_error(std::move(status)); return promise.set_error(Status::Error(7, r_file_id.error().message())); } FileId file_id = r_file_id.ok(); CHECK(file_id.is_valid()); FileView file_view = td_->file_manager_->get_file_view(file_id); CHECK(!file_view.is_encrypted()); if (file_view.has_remote_location() && !file_view.remote_location().is_web()) { td_->create_handler(std::move(promise)) ->send(file_id, file_view.remote_location().as_input_photo()); return; } upload_profile_photo(td_->file_manager_->dup_file_id(file_id), std::move(promise)); } void ContactsManager::upload_profile_photo(FileId file_id, Promise &&promise) { CHECK(file_id.is_valid()); CHECK(uploaded_profile_photos_.find(file_id) == uploaded_profile_photos_.end()); uploaded_profile_photos_.emplace(file_id, std::move(promise)); LOG(INFO) << "Ask to upload profile photo " << file_id; td_->file_manager_->upload(file_id, upload_profile_photo_callback_, 1, 0); } void ContactsManager::delete_profile_photo(int64 profile_photo_id, Promise &&promise) { const User *u = get_user(get_my_id()); if (u != nullptr && u->photo.id == profile_photo_id) { td_->create_handler(std::move(promise)) ->send(FileId(), make_tl_object()); return; } td_->create_handler(std::move(promise))->send(profile_photo_id); } void ContactsManager::set_name(const string &first_name, const string &last_name, Promise &&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(7, "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(std::move(promise))->send(flags, new_first_name, new_last_name, ""); } void ContactsManager::set_bio(const string &bio, Promise &&promise) { 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(std::move(promise))->send(flags, "", "", new_bio); } void ContactsManager::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(my_user_id); if (user_full != nullptr && user_full->is_inited) { user_full->about = about; user_full->is_changed = true; update_user_full(user_full, my_user_id); } } } void ContactsManager::set_username(const string &username, Promise &&promise) { if (!username.empty() && !is_valid_username(username)) { return promise.set_error(Status::Error(400, "Username is invalid")); } td_->create_handler(std::move(promise))->send(username); } void ContactsManager::toggle_chat_administrators(ChatId chat_id, bool everyone_is_administrator, Promise &&promise) { auto c = get_chat(chat_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Group not found")); } if (!get_chat_status(c).is_creator()) { return promise.set_error(Status::Error(6, "Not enough rights to toggle basic group administrators")); } td_->create_handler(std::move(promise))->send(chat_id, everyone_is_administrator); } void ContactsManager::set_channel_username(ChannelId channel_id, const string &username, Promise &&promise) { auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Supergroup not found")); } if (!get_channel_status(c).is_creator()) { return promise.set_error(Status::Error(6, "Not enough rights to change supergroup username")); } if (!username.empty() && !is_valid_username(username)) { return promise.set_error(Status::Error(400, "Username is invalid")); } if (!username.empty() && c->username.empty()) { auto channel_full = get_channel_full(channel_id); if (channel_full != nullptr && !channel_full->can_set_username) { return promise.set_error(Status::Error(3, "Can't set supergroup username")); } } td_->create_handler(std::move(promise))->send(channel_id, username); } void ContactsManager::set_channel_sticker_set(ChannelId channel_id, int64 sticker_set_id, Promise &&promise) { auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Supergroup not found")); } if (!c->is_megagroup) { return promise.set_error(Status::Error(6, "Chat sticker set can be set only for supergroups")); } if (!get_channel_status(c).can_change_info_and_settings()) { return promise.set_error(Status::Error(6, "Not enough rights to change supergroup sticker set")); } telegram_api::object_ptr input_sticker_set; if (sticker_set_id == 0) { input_sticker_set = telegram_api::make_object(); } else { input_sticker_set = td_->stickers_manager_->get_input_sticker_set(sticker_set_id); if (input_sticker_set == nullptr) { return promise.set_error(Status::Error(3, "Sticker set not found")); } } auto channel_full = get_channel_full(channel_id); if (channel_full != nullptr && !channel_full->can_set_sticker_set) { return promise.set_error(Status::Error(3, "Can't set supergroup sticker set")); } td_->create_handler(std::move(promise)) ->send(channel_id, sticker_set_id, std::move(input_sticker_set)); } void ContactsManager::toggle_channel_invites(ChannelId channel_id, bool anyone_can_invite, Promise &&promise) { auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Supergroup not found")); } if (!get_channel_status(c).can_change_info_and_settings()) { return promise.set_error(Status::Error(6, "Not enough rights to toggle supergroup invites")); } if (get_channel_type(c) != ChannelType::Megagroup) { return promise.set_error(Status::Error(6, "Invites by any member can be enabled in the supergroups only")); } td_->create_handler(std::move(promise))->send(channel_id, anyone_can_invite); } void ContactsManager::toggle_channel_sign_messages(ChannelId channel_id, bool sign_messages, Promise &&promise) { auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Supergroup not found")); } if (get_channel_type(c) == ChannelType::Megagroup) { return promise.set_error(Status::Error(6, "Message signatures can't be toggled in supergroups")); } if (!get_channel_status(c).can_change_info_and_settings()) { return promise.set_error(Status::Error(6, "Not enough rights to toggle channel sign messages")); } td_->create_handler(std::move(promise))->send(channel_id, sign_messages); } void ContactsManager::toggle_channel_is_all_history_available(ChannelId channel_id, bool is_all_history_available, Promise &&promise) { auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Supergroup not found")); } if (!get_channel_status(c).can_change_info_and_settings()) { return promise.set_error(Status::Error(6, "Not enough rights to toggle all supergroup history availability")); } if (get_channel_type(c) != ChannelType::Megagroup) { return promise.set_error(Status::Error(6, "Message history can be hidden in the supergroups only")); } td_->create_handler(std::move(promise)) ->send(channel_id, is_all_history_available); } void ContactsManager::set_channel_description(ChannelId channel_id, const string &description, Promise &&promise) { auto new_description = strip_empty_characters(description, MAX_DESCRIPTION_LENGTH); auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Supergroup not found")); } if (!get_channel_status(c).can_change_info_and_settings()) { return promise.set_error(Status::Error(6, "Not enough rights to set supergroup description")); } td_->create_handler(std::move(promise))->send(channel_id, new_description); } void ContactsManager::report_channel_spam(ChannelId channel_id, UserId user_id, const vector &message_ids, Promise &&promise) { auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Supergroup not found")); } if (!c->is_megagroup) { return promise.set_error(Status::Error(6, "Spam can be reported only in supergroups")); } if (!have_input_user(user_id)) { return promise.set_error(Status::Error(6, "Have no access to the user")); } if (user_id == get_my_id()) { return promise.set_error(Status::Error(6, "Can't report self")); } if (message_ids.empty()) { return promise.set_error(Status::Error(6, "Message list is empty")); } vector server_message_ids; for (auto &message_id : message_ids) { if (!message_id.is_valid()) { return promise.set_error(Status::Error(6, "Message not found")); } if (message_id.is_server()) { server_message_ids.push_back(message_id); } } if (server_message_ids.empty()) { return promise.set_value(Unit()); } td_->create_handler(std::move(promise))->send(channel_id, user_id, server_message_ids); } void ContactsManager::delete_channel(ChannelId channel_id, Promise &&promise) { auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Supergroup not found")); } if (!get_channel_status(c).is_creator()) { return promise.set_error(Status::Error(6, "Not enough rights to delete the supergroup")); } td_->create_handler(std::move(promise))->send(channel_id); } void ContactsManager::add_chat_participant(ChatId chat_id, UserId user_id, int32 forward_limit, Promise &&promise) { const Chat *c = get_chat(chat_id); if (c == nullptr) { return promise.set_error(Status::Error(3, "Chat info not found")); } if (!c->is_active) { return promise.set_error(Status::Error(3, "Chat is deactivated")); } if (forward_limit < 0) { return promise.set_error(Status::Error(3, "Can't forward negative number of messages")); } if (user_id != get_my_id()) { if (!get_chat_status(c).can_invite_users()) { return promise.set_error(Status::Error(3, "Not enough rights to invite members to the group chat")); } } else if (c->kicked) { return promise.set_error(Status::Error(3, "User was kicked from the chat")); } // TODO upper bound on forward_limit auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(3, "User not found")); } // TODO invoke after td_->create_handler(std::move(promise))->send(chat_id, std::move(input_user), forward_limit); } void ContactsManager::add_channel_participant(ChannelId channel_id, UserId user_id, Promise &&promise, DialogParticipantStatus old_status) { if (td_->auth_manager_->is_bot()) { return promise.set_error(Status::Error(400, "Bots can't add new chat members")); } const Channel *c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(3, "Chat info not found")); } auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(3, "User not found")); } if (user_id == get_my_id()) { // join the channel if (get_channel_status(c).is_banned()) { return promise.set_error(Status::Error(3, "Can't return to kicked from chat")); } td_->create_handler(std::move(promise))->send(channel_id); return; } if (!(c->anyone_can_invite && get_channel_status(c).is_member()) && !get_channel_status(c).can_invite_users()) { return promise.set_error(Status::Error(3, "Not enough rights to invite members to the supergroup chat")); } speculative_add_channel_user(channel_id, user_id, DialogParticipantStatus::Member(), old_status); vector> input_users; input_users.push_back(std::move(input_user)); td_->create_handler(std::move(promise))->send(channel_id, std::move(input_users)); } void ContactsManager::add_channel_participants(ChannelId channel_id, const vector &user_ids, Promise &&promise) { if (td_->auth_manager_->is_bot()) { return promise.set_error(Status::Error(400, "Bots can't add new chat members")); } const Channel *c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(3, "Chat info not found")); } if (!(c->anyone_can_invite && get_channel_status(c).is_member()) && !get_channel_status(c).can_invite_users()) { return promise.set_error(Status::Error(3, "Not enough rights to invite members to the supergroup chat")); } vector> input_users; for (auto user_id : user_ids) { auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(3, "User not found")); } if (user_id == get_my_id()) { // can't invite self continue; } input_users.push_back(std::move(input_user)); } if (input_users.empty()) { return promise.set_value(Unit()); } td_->create_handler(std::move(promise))->send(channel_id, std::move(input_users)); } void ContactsManager::change_channel_participant_status(ChannelId channel_id, UserId user_id, DialogParticipantStatus status, Promise &&promise) { auto c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Chat info not found")); } auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(6, "User not found")); } if (user_id == get_my_id()) { // fast path is needed, because get_channel_status may return Creator, while GetChannelParticipantQuery returning Left return change_channel_participant_status_impl(channel_id, user_id, std::move(status), get_channel_status(c), std::move(promise)); } auto on_result_promise = PromiseCreator::lambda([actor_id = actor_id(this), channel_id, user_id, status, promise = std::move(promise)](Result r_dialog_participant) mutable { // ResultHandlers are cleared before managers, so it is safe to capture this if (r_dialog_participant.is_error()) { return promise.set_error(r_dialog_participant.move_as_error()); } send_closure(actor_id, &ContactsManager::change_channel_participant_status_impl, channel_id, user_id, std::move(status), r_dialog_participant.ok().status, std::move(promise)); }); td_->create_handler(std::move(on_result_promise)) ->send(channel_id, user_id, std::move(input_user)); } void ContactsManager::change_channel_participant_status_impl(ChannelId channel_id, UserId user_id, DialogParticipantStatus status, DialogParticipantStatus old_status, Promise &&promise) { if (old_status == status) { return promise.set_value(Unit()); } LOG(INFO) << "Change status of " << user_id << " in " << channel_id << " from " << old_status << " to " << status; bool need_add = false; bool need_promote = false; bool need_restrict = false; if (status.is_creator() || old_status.is_creator()) { if (!old_status.is_creator()) { return promise.set_error(Status::Error(3, "Can't add creator to the chat")); } if (status.is_member()) { // creator member -> not creator member // creator not member -> creator member // creator not member -> not creator member if (old_status.is_member()) { return promise.set_error(Status::Error(3, "Can't demote chat creator")); } need_add = true; } else { // creator member -> creator not member // creator member -> not creator not member // creator not member -> not creator not member if (!old_status.is_member()) { return promise.set_error(Status::Error(3, "Can't restrict chat creator")); } need_restrict = true; } } else if (status.is_administrator()) { need_promote = true; } else if (!status.is_member() || status.is_restricted()) { if (status.is_member() && !old_status.is_member()) { // TODO there is no way in server API to invite someone and change restrictions // we need to first add user and change restrictions again after that // but if restrictions aren't changed, then adding is enough auto copy_old_status = old_status; copy_old_status.set_is_member(true); if (copy_old_status == status) { need_add = true; } else { need_restrict = true; } } else { need_restrict = true; } } else { // regular member if (old_status.is_administrator()) { need_promote = true; } else if (old_status.is_restricted() || old_status.is_banned()) { need_restrict = true; } else { CHECK(!old_status.is_member()); need_add = true; } } if (need_promote) { return promote_channel_participant(channel_id, user_id, std::move(status), std::move(old_status), std::move(promise)); } else if (need_restrict) { return restrict_channel_participant(channel_id, user_id, std::move(status), std::move(old_status), std::move(promise)); } else { CHECK(need_add); return add_channel_participant(channel_id, user_id, std::move(promise), std::move(old_status)); } } void ContactsManager::promote_channel_participant(ChannelId channel_id, UserId user_id, DialogParticipantStatus status, DialogParticipantStatus old_status, Promise &&promise) { LOG(INFO) << "Promote " << user_id << " in " << channel_id << " from " << old_status << " to " << status; const Channel *c = get_channel(channel_id); CHECK(c != nullptr); if (user_id == get_my_id()) { if (status.is_administrator()) { return promise.set_error(Status::Error(3, "Can't promote self")); } CHECK(status.is_member()); // allow to demote self. TODO is it allowed server-side? } else { if (!get_channel_status(c).can_promote_members()) { return promise.set_error(Status::Error(3, "Not enough rights")); } } auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(3, "User not found")); } speculative_add_channel_user(channel_id, user_id, status, old_status); td_->create_handler(std::move(promise))->send(channel_id, std::move(input_user), status); } void ContactsManager::change_chat_participant_status(ChatId chat_id, UserId user_id, DialogParticipantStatus status, Promise &&promise) { if (!status.is_member()) { return delete_chat_participant(chat_id, user_id, std::move(promise)); } auto c = get_chat(chat_id); if (c == nullptr) { return promise.set_error(Status::Error(6, "Chat info not found")); } if (!get_chat_status(c).can_promote_members()) { return promise.set_error(Status::Error(3, "Need creator rights in the group chat")); } if (c->everyone_is_administrator) { return promise.set_error(Status::Error(3, "Administrators editing is disabled in the group chat")); } if (user_id == get_my_id()) { return promise.set_error(Status::Error(3, "Can't change chat member status of self")); } auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(3, "User not found")); } if (status.is_creator()) { return promise.set_error(Status::Error(3, "Can't add creator to the group chat")); } if (status.is_restricted()) { return promise.set_error(Status::Error(3, "Can't restrict users in a basic group chat")); } td_->create_handler(std::move(promise)) ->send(chat_id, std::move(input_user), status.is_administrator()); } void ContactsManager::export_chat_invite_link(ChatId chat_id, Promise &&promise) { const Chat *c = get_chat(chat_id); if (c == nullptr) { return promise.set_error(Status::Error(3, "Chat info not found")); } if (!c->is_active) { return promise.set_error(Status::Error(3, "Chat is deactivated")); } if (!get_chat_status(c).can_export_dialog_invite_link()) { return promise.set_error(Status::Error(3, "Not enough rights to export chat invite link")); } td_->create_handler(std::move(promise))->send(chat_id); } void ContactsManager::export_channel_invite_link(ChannelId channel_id, Promise &&promise) { const Channel *c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(3, "Chat info not found")); } if (!get_channel_status(c).can_export_dialog_invite_link()) { return promise.set_error(Status::Error(3, "Not enough rights to export chat invite link")); } td_->create_handler(std::move(promise))->send(channel_id); } void ContactsManager::check_dialog_invite_link(const string &invite_link, Promise &&promise) const { auto it = invite_link_infos_.find(invite_link); if (it != invite_link_infos_.end()) { return promise.set_value(Unit()); } if (!is_valid_invite_link(invite_link)) { return promise.set_error(Status::Error(3, "Wrong invite link")); } td_->create_handler(std::move(promise))->send(invite_link); } void ContactsManager::import_dialog_invite_link(const string &invite_link, Promise &&promise) { if (!is_valid_invite_link(invite_link)) { return promise.set_error(Status::Error(3, "Wrong invite link")); } td_->create_handler(std::move(promise))->send(invite_link); } string ContactsManager::get_chat_invite_link(ChatId chat_id) const { auto chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { auto it = chat_invite_links_.find(chat_id); return it == chat_invite_links_.end() ? string() : it->second; } return chat_full->invite_link; } string ContactsManager::get_channel_invite_link( ChannelId channel_id) { // should be non-const to update ChannelFull cache auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr) { auto it = channel_invite_links_.find(channel_id); return it == channel_invite_links_.end() ? string() : it->second; } return channel_full->invite_link; } void ContactsManager::delete_chat_participant(ChatId chat_id, UserId user_id, Promise &&promise) { const Chat *c = get_chat(chat_id); if (c == nullptr) { return promise.set_error(Status::Error(3, "Chat info not found")); } if (!c->is_active) { return promise.set_error(Status::Error(3, "Chat is deactivated")); } auto my_id = get_my_id(); if (c->left) { if (user_id == my_id) { return promise.set_value(Unit()); } else { return promise.set_error(Status::Error(3, "Not in the chat")); } } if (user_id != my_id) { auto my_status = get_chat_status(c); if (!my_status.is_creator()) { // creator can delete anyone auto participant = get_chat_participant(chat_id, user_id); if (participant != nullptr) { // if have no information about participant, just send request to the server if (c->everyone_is_administrator) { // if all are administrators, only invited by me participants can be deleted if (participant->inviter_user_id != my_id) { return promise.set_error(Status::Error(3, "Need to be inviter of a user to kick it from a basic group")); } } else { // otherwise, only creator can kick administrators if (participant->status.is_administrator()) { return promise.set_error( Status::Error(3, "Only the creator of a basic group can kick group administrators")); } // regular users can be kicked by administrators and their inviters if (!my_status.is_administrator() && participant->inviter_user_id != my_id) { return promise.set_error(Status::Error(3, "Need to be inviter of a user to kick it from a basic group")); } } } } } auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(3, "User not found")); } // TODO invoke after td_->create_handler(std::move(promise))->send(chat_id, std::move(input_user)); } void ContactsManager::restrict_channel_participant(ChannelId channel_id, UserId user_id, DialogParticipantStatus status, DialogParticipantStatus old_status, Promise &&promise) { LOG(INFO) << "Restrict " << user_id << " in " << channel_id << " from " << old_status << " to " << status; const Channel *c = get_channel(channel_id); if (c == nullptr) { return promise.set_error(Status::Error(3, "Chat info not found")); } if (!c->status.is_member()) { if (user_id == get_my_id()) { if (status.is_member()) { return promise.set_error(Status::Error(3, "Can't unrestrict self")); } return promise.set_value(Unit()); } else { return promise.set_error(Status::Error(3, "Not in the chat")); } } auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(3, "User not found")); } if (user_id == get_my_id()) { if (status.is_restricted() || status.is_banned()) { return promise.set_error(Status::Error(3, "Can't restrict self")); } if (status.is_member()) { return promise.set_error(Status::Error(3, "Can't unrestrict self")); } // leave the channel td_->create_handler(std::move(promise))->send(channel_id); return; } if (status.is_creator()) { return promise.set_error(Status::Error(3, "Not enough rights to restrict chat creator")); } if (!get_channel_status(c).can_restrict_members()) { return promise.set_error(Status::Error(3, "Not enough rights to restrict/unrestrict chat member")); } if (old_status.is_member() && !status.is_member() && !status.is_banned()) { // we can't make participant Left without kicking it first auto on_result_promise = PromiseCreator::lambda([channel_id, user_id, status, promise = std::move(promise)](Result<> result) mutable { if (result.is_error()) { return promise.set_error(result.move_as_error()); } create_actor( "RestrictChannelParticipantSleepActor", 1.0, PromiseCreator::lambda([channel_id, user_id, status, promise = std::move(promise)](Result<> result) mutable { if (result.is_error()) { return promise.set_error(result.move_as_error()); } send_closure(G()->contacts_manager(), &ContactsManager::restrict_channel_participant, channel_id, user_id, status, DialogParticipantStatus::Banned(0), std::move(promise)); })) .release(); }); promise = std::move(on_result_promise); status = DialogParticipantStatus::Banned(0); } speculative_add_channel_user(channel_id, user_id, status, old_status); td_->create_handler(std::move(promise))->send(channel_id, std::move(input_user), status); } ChannelId ContactsManager::migrate_chat_to_megagroup(ChatId chat_id, Promise &promise) { auto c = get_chat(chat_id); if (c == nullptr) { promise.set_error(Status::Error(3, "Chat info not found")); return ChannelId(); } if (!c->is_creator) { promise.set_error(Status::Error(3, "Need creator rights in the chat")); return ChannelId(); } if (c->migrated_to_channel_id.is_valid()) { return c->migrated_to_channel_id; } td_->create_handler(std::move(promise))->send(chat_id); return ChannelId(); } vector ContactsManager::get_created_public_dialogs(Promise &&promise) { if (created_public_channels_inited_) { promise.set_value(Unit()); return transform(created_public_channels_, [&](ChannelId channel_id) { DialogId dialog_id(channel_id); td_->messages_manager_->force_create_dialog(dialog_id, "get_created_public_dialogs"); return dialog_id; }); } td_->create_handler(std::move(promise))->send(); return {}; } void ContactsManager::on_get_created_public_channels(vector> &&chats) { created_public_channels_inited_ = true; created_public_channels_.clear(); for (auto &chat : chats) { switch (chat->get_id()) { case telegram_api::chatEmpty::ID: LOG(ERROR) << "Receive chatEmpty as created public channel"; break; case telegram_api::chat::ID: LOG(ERROR) << "Receive chat as created public channel"; break; case telegram_api::chatForbidden::ID: LOG(ERROR) << "Receive chatForbidden as created public channel"; break; case telegram_api::channel::ID: { auto c = static_cast(chat.get()); ChannelId channel_id(c->id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; continue; } created_public_channels_.push_back(channel_id); break; } case telegram_api::channelForbidden::ID: { auto c = static_cast(chat.get()); ChannelId channel_id(c->id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; continue; } created_public_channels_.push_back(channel_id); break; } default: UNREACHABLE(); } on_get_chat(std::move(chat), "on_get_created_public_channels"); } } void ContactsManager::on_imported_contacts(int64 random_id, vector imported_contact_user_ids, vector unimported_contact_invites) { LOG(INFO) << "Contacts import with random_id " << random_id << " has finished: " << format::as_array(imported_contact_user_ids); if (random_id == 0) { // 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 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_imported_contacts"); // to ensure updateUser all_imported_contacts_[unique_id].set_user_id(imported_contact_user_ids[i]); unique_id_to_unimported_contact_invites[unique_id] = unimported_contact_invites[i]; } if (G()->parameters().use_chat_info_db) { 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()); } })); } 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(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 ContactsManager::on_deleted_contacts(const vector &deleted_contact_user_ids) { LOG(INFO) << "Contacts deletion has finished for " << deleted_contact_user_ids; for (auto user_id : deleted_contact_user_ids) { LOG(INFO) << "Drop contact with " << user_id; auto u = get_user(user_id); CHECK(u != nullptr); on_update_user_links(u, user_id, LinkState::KnowsPhoneNumber, u->inbound); update_user(u, user_id); CHECK(u->outbound != LinkState::Contact); CHECK(!contacts_hints_.has_key(user_id.get())); } } void ContactsManager::save_next_contacts_sync_date() { if (!G()->parameters().use_chat_info_db) { return; } G()->td_db()->get_binlog_pmc()->set("next_contacts_sync_date", to_string(next_contacts_sync_date_)); } void ContactsManager::on_get_contacts(tl_object_ptr &&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()->send(); return; } auto contacts = move_tl_object_as(new_contacts); std::unordered_set 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(); for (auto &p : users_) { UserId user_id = p.first; User &u = p.second; bool is_contact = u.outbound == LinkState::Contact; bool should_be_contact = contact_user_ids.count(user_id) == 1; if (is_contact != should_be_contact) { if (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_links(&u, user_id, LinkState::KnowsPhoneNumber, u.inbound); update_user(&u, user_id); CHECK(u.outbound != LinkState::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::max()); } void ContactsManager::save_contacts_to_database() { if (!G()->parameters().use_chat_info_db || !are_contacts_loaded_) { return; } LOG(INFO) << "Schedule save contacts to database"; vector user_ids = transform(contacts_hints_.search_empty(100000).second, [](int64 key) { return UserId(narrow_cast(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) << "Save 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()->contacts_manager(), &ContactsManager::save_next_contacts_sync_date); } })); } })); } void ContactsManager::on_get_contacts_failed(Status error) { CHECK(error.is_error()); next_contacts_sync_date_ = G()->unix_time() + Random::fast(5, 10); auto promises = std::move(load_contacts_queries_); load_contacts_queries_.clear(); for (auto &promise : promises) { promise.set_error(error.clone()); } } void ContactsManager::on_load_contacts_from_database(string value) { if (value.empty()) { reload_contacts(true); return; } vector user_ids; log_event_parse(user_ids, value).ensure(); LOG(INFO) << "Successfully loaded " << user_ids.size() << " contacts from database"; load_contact_users_multipromise_.add_promise( PromiseCreator::lambda([expected_contact_count = user_ids.size()](Result<> result) { if (result.is_ok()) { send_closure(G()->contacts_manager(), &ContactsManager::on_get_contacts_finished, expected_contact_count); } })); 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 ContactsManager::on_get_contacts_finished(size_t expected_contact_count) { LOG(INFO) << "Finished to get " << contacts_hints_.size() << " contacts out of " << expected_contact_count; are_contacts_loaded_ = true; auto promises = std::move(load_contacts_queries_); load_contacts_queries_.clear(); for (auto &promise : promises) { promise.set_value(Unit()); } if (expected_contact_count != contacts_hints_.size()) { save_contacts_to_database(); } } void ContactsManager::on_get_contacts_statuses(vector> &&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 ContactsManager::on_update_online_status_privacy() { td_->create_handler()->send(); } void ContactsManager::on_get_contacts_link(tl_object_ptr &&link) { UserId user_id = get_user_id(link->user_); on_get_user(std::move(link->user_), "on_get_contacts_link"); on_update_user_links(user_id, std::move(link->my_link_), std::move(link->foreign_link_)); } UserId ContactsManager::get_user_id(const tl_object_ptr &user) { CHECK(user != nullptr); switch (user->get_id()) { case telegram_api::userEmpty::ID: return UserId(static_cast(user.get())->id_); case telegram_api::user::ID: return UserId(static_cast(user.get())->id_); default: UNREACHABLE(); return UserId(); } } ChatId ContactsManager::get_chat_id(const tl_object_ptr &chat) { CHECK(chat != nullptr); switch (chat->get_id()) { case telegram_api::chatEmpty::ID: return ChatId(static_cast(chat.get())->id_); case telegram_api::chat::ID: return ChatId(static_cast(chat.get())->id_); case telegram_api::chatForbidden::ID: return ChatId(static_cast(chat.get())->id_); default: return ChatId(); } } ChannelId ContactsManager::get_channel_id(const tl_object_ptr &chat) { CHECK(chat != nullptr); switch (chat->get_id()) { case telegram_api::channel::ID: return ChannelId(static_cast(chat.get())->id_); case telegram_api::channelForbidden::ID: return ChannelId(static_cast(chat.get())->id_); default: return ChannelId(); } } void ContactsManager::on_get_user(tl_object_ptr &&user_ptr, const char *source, bool is_me, bool is_support) { 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(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); if (u == nullptr) { 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(user_ptr); UserId user_id(user->id_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; return; } int32 flags = user->flags_; LOG(INFO) << "Receive " << user_id << " with flags " << flags << " from " << source; if (is_me && (flags & USER_FLAG_IS_ME) == 0) { LOG(ERROR) << user_id << " doesn't have flag IS_ME, but must have it when received from " << source; flags |= USER_FLAG_IS_ME; } bool is_bot = (flags & USER_FLAG_IS_BOT) != 0; if (flags & USER_FLAG_IS_ME) { set_my_id(user_id); td_->auth_manager_->set_is_bot(is_bot); if (!is_bot) { G()->shared_config().set_option_string("my_phone_number", user->phone_); } } else { /* if (!(flags & USER_FLAG_HAS_ACCESS_HASH) && !(flags & USER_FLAG_IS_DELETED) && !(flags & USER_FLAG_IS_INACCESSIBLE)) { LOG(ERROR) << user_id << " has no access_hash"; return; } */ } if (is_support) { support_user_id_ = user_id; } bool have_access_hash = (flags & USER_FLAG_HAS_ACCESS_HASH) != 0; bool is_received = (flags & USER_FLAG_IS_INACCESSIBLE) == 0; User *u = add_user(user_id, "on_get_user"); if ((have_access_hash || u->access_hash == -1) && u->access_hash != user->access_hash_) { LOG(DEBUG) << "Access hash has changed for " << user_id << " from " << u->access_hash << " to " << user->access_hash_; u->access_hash = user->access_hash_; u->is_changed = true; } if (is_received) { on_update_user_phone_number(u, user_id, std::move(user->phone_)); } on_update_user_photo(u, user_id, std::move(user->photo_)); if (is_received) { on_update_user_online(u, user_id, std::move(user->status_)); LinkState out, in; if (flags & USER_FLAG_IS_MUTUAL_CONTACT) { out = LinkState::Contact; in = LinkState::Contact; } else if (flags & USER_FLAG_IS_CONTACT) { out = LinkState::Contact; in = LinkState::Unknown; } else if (flags & USER_FLAG_HAS_PHONE_NUMBER) { out = LinkState::KnowsPhoneNumber; in = LinkState::Unknown; } else { out = LinkState::None; in = LinkState::Unknown; } on_update_user_links(u, user_id, out, in); } if (is_received || !u->is_received) { on_update_user_name(u, user_id, std::move(user->first_name_), std::move(user->last_name_), std::move(user->username_)); } bool is_verified = (flags & USER_FLAG_IS_VERIFIED) != 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; string restriction_reason = std::move(user->restriction_reason_); bool is_inline_bot = (flags & USER_FLAG_IS_INLINE_BOT) != 0; string inline_query_placeholder = user->bot_inline_placeholder_; bool need_location_bot = (flags & USER_FLAG_NEED_LOCATION_BOT) != 0; bool has_bot_info_version = (flags & USER_FLAG_HAS_BOT_INFO_VERSION) != 0; LOG_IF(ERROR, !can_join_groups && !is_bot) << "Receive not bot which can't join groups from " << source; LOG_IF(ERROR, can_read_all_group_messages && !is_bot) << "Receive not bot which can read all group messages from " << source; LOG_IF(ERROR, is_inline_bot && !is_bot) << "Receive not bot which is inline bot from " << source; LOG_IF(ERROR, need_location_bot && !is_inline_bot) << "Receive not inline bot which needs user location from " << source; if (is_received && !u->is_received) { u->is_received = true; LOG(DEBUG) << "Receive " << user_id; u->need_send_update = true; } if (is_deleted) { // just in case is_verified = false; is_bot = false; can_join_groups = false; can_read_all_group_messages = false; is_inline_bot = false; inline_query_placeholder = string(); need_location_bot = false; has_bot_info_version = false; } LOG_IF(ERROR, has_bot_info_version && !is_bot) << "Receive not bot 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_bot != u->is_bot || can_join_groups != u->can_join_groups || can_read_all_group_messages != u->can_read_all_group_messages || restriction_reason != u->restriction_reason || is_inline_bot != u->is_inline_bot || inline_query_placeholder != u->inline_query_placeholder || need_location_bot != u->need_location_bot) { LOG_IF(ERROR, is_bot != u->is_bot && !is_deleted && !u->is_deleted) << "User.is_bot has changed from " << source; u->is_verified = is_verified; u->is_bot = is_bot; u->can_join_groups = can_join_groups; u->can_read_all_group_messages = can_read_all_group_messages; u->restriction_reason = std::move(restriction_reason); u->is_inline_bot = is_inline_bot; u->inline_query_placeholder = std::move(inline_query_placeholder); u->need_location_bot = need_location_bot; LOG(DEBUG) << "Info has changed for " << user_id; u->need_send_update = 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->is_changed = true; } if (is_deleted != u->is_deleted) { u->is_deleted = is_deleted; if (u->is_deleted) { invalidate_user_full(user_id); } LOG(DEBUG) << "is_deleted has changed for " << user_id; u->need_send_update = 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 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; u->need_send_update = true; } update_user(u, user_id); } class ContactsManager::UserLogEvent { public: UserId user_id; User u; UserLogEvent() = default; UserLogEvent(UserId user_id, const User &u) : user_id(user_id), u(u) { } template void store(StorerT &storer) const { td::store(user_id, storer); td::store(u, storer); } template void parse(ParserT &parser) { td::parse(user_id, parser); td::parse(u, parser); } }; void ContactsManager::save_user(User *u, UserId user_id, bool from_binlog) { if (!G()->parameters().use_chat_info_db) { 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 logevent = UserLogEvent(user_id, *u); auto storer = LogEventStorerImpl(logevent); if (u->logevent_id == 0) { u->logevent_id = binlog_add(G()->td_db()->get_binlog(), LogEvent::HandlerType::Users, storer); } else { binlog_rewrite(G()->td_db()->get_binlog(), u->logevent_id, LogEvent::HandlerType::Users, storer); } } save_user_to_database(u, user_id); } } void ContactsManager::on_binlog_user_event(BinlogEvent &&event) { if (!G()->parameters().use_chat_info_db) { binlog_erase(G()->td_db()->get_binlog(), event.id_); return; } UserLogEvent log_event; log_event_parse(log_event, event.data_).ensure(); auto user_id = log_event.user_id; LOG(INFO) << "Add " << user_id << " from binlog"; User *u = add_user(user_id, "on_binlog_user_event"); if (!(u->first_name.empty() && u->last_name.empty()) && Slice(u->debug_source) == Slice("on_binlog_user_event")) { LOG(ERROR) << "Skip adding already added " << user_id; binlog_erase(G()->td_db()->get_binlog(), event.id_); return; // TODO fix bug in Binlog and remove that fix } CHECK(u->first_name.empty() && u->last_name.empty()) << user_id << " " << u->debug_source; *u = std::move(log_event.u); // users come from binlog before all other events, so just add them u->logevent_id = event.id_; update_user(u, user_id, true, false); } string ContactsManager::get_user_database_key(UserId user_id) { return PSTRING() << "us" << user_id.get(); } string ContactsManager::get_user_database_value(const User *u) { return log_event_store(*u).as_slice().str(); } void ContactsManager::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 ContactsManager::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()->contacts_manager(), &ContactsManager::on_save_user_to_database, user_id, result.is_ok()); })); } void ContactsManager::on_save_user_to_database(UserId user_id, bool success) { User *u = get_user(user_id); CHECK(u != nullptr); CHECK(u->is_being_saved) << 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->is_changed << " " << u->need_send_update << " " << u->is_status_changed << " " << u->is_name_changed << " " << u->is_username_changed << " " << u->is_photo_changed << " " << u->is_outbound_link_changed; 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->logevent_id != 0) { binlog_erase(G()->td_db()->get_binlog(), u->logevent_id); u->logevent_id = 0; } } else { save_user(u, user_id, u->logevent_id != 0); } } void ContactsManager::load_user_from_database(User *u, UserId user_id, Promise 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 ContactsManager::load_user_from_database_impl(UserId user_id, Promise 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()->contacts_manager(), &ContactsManager::on_load_user_from_database, user_id, std::move(value)); })); } } void ContactsManager::on_load_user_from_database(UserId user_id, string value) { if (!loaded_from_database_users_.insert(user_id).second) { return; } auto it = load_user_from_database_queries_.find(user_id); vector> 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, "on_load_user_from_database"); log_event_parse(*u, value).ensure(); 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->logevent_id != 0) { binlog_erase(G()->td_db()->get_binlog(), u->logevent_id); u->logevent_id = 0; } } for (auto &promise : promises) { promise.set_value(Unit()); } } bool ContactsManager::have_user_force(UserId user_id) { return get_user_force(user_id) != nullptr; } ContactsManager::User *ContactsManager::get_user_force(UserId user_id) { auto u = get_user_force_impl(user_id); if (user_id == UserId(777000) && (u == nullptr || !u->is_received)) { int32 flags = telegram_api::user::ACCESS_HASH_MASK | telegram_api::user::FIRST_NAME_MASK | telegram_api::user::PHONE_MASK | telegram_api::user::PHOTO_MASK | telegram_api::user::VERIFIED_MASK; auto profile_photo = telegram_api::make_object( 3337190045231018, telegram_api::make_object(1, 702229962, 26779, 5859320227133863146, BufferSlice()), telegram_api::make_object(1, 702229962, 26781, -3695031185685824216, BufferSlice())); if (G()->is_test_dc()) { profile_photo = nullptr; flags -= telegram_api::user::PHOTO_MASK; } auto user = telegram_api::make_object( flags, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, 777000, 1, "Telegram", "Updates", string(), "42777", std::move(profile_photo), nullptr, 0, string(), string(), string()); on_get_user(std::move(user), "get_user_force"); u = get_user(user_id); CHECK(u != nullptr && u->is_received); } return u; } ContactsManager::User *ContactsManager::get_user_force_impl(UserId user_id) { if (!user_id.is_valid()) { return nullptr; } User *u = get_user(user_id); if (u != nullptr) { return u; } if (!G()->parameters().use_chat_info_db) { return nullptr; } if (loaded_from_database_users_.count(user_id)) { return nullptr; } LOG(INFO) << "Trying to load " << user_id << " from database"; on_load_user_from_database(user_id, G()->td_db()->get_sqlite_sync_pmc()->get(get_user_database_key(user_id))); return get_user(user_id); } class ContactsManager::ChatLogEvent { public: ChatId chat_id; Chat c; ChatLogEvent() = default; ChatLogEvent(ChatId chat_id, const Chat &c) : chat_id(chat_id), c(c) { } template void store(StorerT &storer) const { td::store(chat_id, storer); td::store(c, storer); } template void parse(ParserT &parser) { td::parse(chat_id, parser); td::parse(c, parser); } }; void ContactsManager::save_chat(Chat *c, ChatId chat_id, bool from_binlog) { if (!G()->parameters().use_chat_info_db) { return; } CHECK(c != nullptr); if (!c->is_saved) { if (!from_binlog) { auto logevent = ChatLogEvent(chat_id, *c); auto storer = LogEventStorerImpl(logevent); if (c->logevent_id == 0) { c->logevent_id = binlog_add(G()->td_db()->get_binlog(), LogEvent::HandlerType::Chats, storer); } else { binlog_rewrite(G()->td_db()->get_binlog(), c->logevent_id, LogEvent::HandlerType::Chats, storer); } } save_chat_to_database(c, chat_id); return; } } void ContactsManager::on_binlog_chat_event(BinlogEvent &&event) { if (!G()->parameters().use_chat_info_db) { binlog_erase(G()->td_db()->get_binlog(), event.id_); return; } ChatLogEvent log_event; log_event_parse(log_event, event.data_).ensure(); auto chat_id = log_event.chat_id; LOG(INFO) << "Add " << chat_id << " from binlog"; Chat *c = add_chat(chat_id); if (c->left || !c->kicked) { LOG(ERROR) << "Skip adding already added " << chat_id; binlog_erase(G()->td_db()->get_binlog(), event.id_); return; // TODO fix bug in Binlog and remove that fix } CHECK(!c->left && c->kicked); *c = std::move(log_event.c); // chats come from binlog before all other events, so just add them c->logevent_id = event.id_; update_chat(c, chat_id, true, false); } string ContactsManager::get_chat_database_key(ChatId chat_id) { return PSTRING() << "gr" << chat_id.get(); } string ContactsManager::get_chat_database_value(const Chat *c) { return log_event_store(*c).as_slice().str(); } void ContactsManager::save_chat_to_database(Chat *c, ChatId chat_id) { CHECK(c != nullptr); if (c->is_being_saved) { return; } if (loaded_from_database_chats_.count(chat_id)) { save_chat_to_database_impl(c, chat_id, get_chat_database_value(c)); return; } if (load_chat_from_database_queries_.count(chat_id) != 0) { return; } load_chat_from_database_impl(chat_id, Auto()); } void ContactsManager::save_chat_to_database_impl(Chat *c, ChatId chat_id, string value) { CHECK(c != nullptr); CHECK(load_chat_from_database_queries_.count(chat_id) == 0); c->is_being_saved = true; c->is_saved = true; LOG(INFO) << "Trying to save to database " << chat_id; G()->td_db()->get_sqlite_pmc()->set( get_chat_database_key(chat_id), std::move(value), PromiseCreator::lambda([chat_id](Result<> result) { send_closure(G()->contacts_manager(), &ContactsManager::on_save_chat_to_database, chat_id, result.is_ok()); })); } void ContactsManager::on_save_chat_to_database(ChatId chat_id, bool success) { Chat *c = get_chat(chat_id); CHECK(c != nullptr); CHECK(c->is_being_saved); CHECK(load_chat_from_database_queries_.count(chat_id) == 0); c->is_being_saved = false; if (!success) { LOG(ERROR) << "Failed to save " << chat_id << " to database"; c->is_saved = false; } else { LOG(INFO) << "Successfully saved " << chat_id << " to database"; } if (c->is_saved) { if (c->logevent_id != 0) { binlog_erase(G()->td_db()->get_binlog(), c->logevent_id); c->logevent_id = 0; } } else { save_chat(c, chat_id, c->logevent_id != 0); } } void ContactsManager::load_chat_from_database(Chat *c, ChatId chat_id, Promise promise) { if (loaded_from_database_chats_.count(chat_id)) { promise.set_value(Unit()); return; } CHECK(c == nullptr || !c->is_being_saved); load_chat_from_database_impl(chat_id, std::move(promise)); } void ContactsManager::load_chat_from_database_impl(ChatId chat_id, Promise promise) { LOG(INFO) << "Load " << chat_id << " from database"; auto &load_chat_queries = load_chat_from_database_queries_[chat_id]; load_chat_queries.push_back(std::move(promise)); if (load_chat_queries.size() == 1u) { G()->td_db()->get_sqlite_pmc()->get(get_chat_database_key(chat_id), PromiseCreator::lambda([chat_id](string value) { send_closure(G()->contacts_manager(), &ContactsManager::on_load_chat_from_database, chat_id, std::move(value)); })); } } void ContactsManager::on_load_chat_from_database(ChatId chat_id, string value) { if (!loaded_from_database_chats_.insert(chat_id).second) { return; } auto it = load_chat_from_database_queries_.find(chat_id); vector> promises; if (it != load_chat_from_database_queries_.end()) { promises = std::move(it->second); CHECK(!promises.empty()); load_chat_from_database_queries_.erase(it); } LOG(INFO) << "Successfully loaded " << chat_id << " of size " << value.size() << " from database"; // G()->td_db()->get_sqlite_pmc()->erase(get_chat_database_key(chat_id), Auto()); // return; Chat *c = get_chat(chat_id); if (c == nullptr) { if (!value.empty()) { c = add_chat(chat_id); log_event_parse(*c, value).ensure(); c->is_saved = true; update_chat(c, chat_id, true, true); } } else { CHECK(!c->is_saved); // chat can't be saved before load completes CHECK(!c->is_being_saved); auto new_value = get_chat_database_value(c); if (value != new_value) { save_chat_to_database_impl(c, chat_id, std::move(new_value)); } else if (c->logevent_id != 0) { binlog_erase(G()->td_db()->get_binlog(), c->logevent_id); c->logevent_id = 0; } } if (c != nullptr && c->migrated_to_channel_id.is_valid() && !have_channel_force(c->migrated_to_channel_id)) { LOG(ERROR) << "Can't find " << c->migrated_to_channel_id << " from " << chat_id; } for (auto &promise : promises) { promise.set_value(Unit()); } } bool ContactsManager::have_chat_force(ChatId chat_id) { return get_chat_force(chat_id) != nullptr; } ContactsManager::Chat *ContactsManager::get_chat_force(ChatId chat_id) { if (!chat_id.is_valid()) { return nullptr; } Chat *c = get_chat(chat_id); if (c != nullptr) { if (c->migrated_to_channel_id.is_valid() && !have_channel_force(c->migrated_to_channel_id)) { LOG(ERROR) << "Can't find " << c->migrated_to_channel_id << " from " << chat_id; } return c; } if (!G()->parameters().use_chat_info_db) { return nullptr; } if (loaded_from_database_chats_.count(chat_id)) { return nullptr; } LOG(INFO) << "Trying to load " << chat_id << " from database"; on_load_chat_from_database(chat_id, G()->td_db()->get_sqlite_sync_pmc()->get(get_chat_database_key(chat_id))); return get_chat(chat_id); } class ContactsManager::ChannelLogEvent { public: ChannelId channel_id; Channel c; ChannelLogEvent() = default; ChannelLogEvent(ChannelId channel_id, const Channel &c) : channel_id(channel_id), c(c) { } template void store(StorerT &storer) const { td::store(channel_id, storer); td::store(c, storer); } template void parse(ParserT &parser) { td::parse(channel_id, parser); td::parse(c, parser); } }; void ContactsManager::save_channel(Channel *c, ChannelId channel_id, bool from_binlog) { if (!G()->parameters().use_chat_info_db) { return; } CHECK(c != nullptr); if (!c->is_saved) { if (!from_binlog) { auto logevent = ChannelLogEvent(channel_id, *c); auto storer = LogEventStorerImpl(logevent); if (c->logevent_id == 0) { c->logevent_id = binlog_add(G()->td_db()->get_binlog(), LogEvent::HandlerType::Channels, storer); } else { binlog_rewrite(G()->td_db()->get_binlog(), c->logevent_id, LogEvent::HandlerType::Channels, storer); } } save_channel_to_database(c, channel_id); return; } } void ContactsManager::on_binlog_channel_event(BinlogEvent &&event) { if (!G()->parameters().use_chat_info_db) { binlog_erase(G()->td_db()->get_binlog(), event.id_); return; } ChannelLogEvent log_event; log_event_parse(log_event, event.data_).ensure(); auto channel_id = log_event.channel_id; LOG(INFO) << "Add " << channel_id << " from binlog"; Channel *c = add_channel(channel_id, "on_binlog_channel_event"); if (!c->status.is_banned() && Slice(c->debug_source) == Slice("on_binlog_channel_event")) { LOG(ERROR) << "Skip adding already added " << channel_id; binlog_erase(G()->td_db()->get_binlog(), event.id_); return; // TODO fix bug in Binlog and remove that fix } CHECK(c->status.is_banned()) << channel_id << " " << c->debug_source; *c = std::move(log_event.c); // channels come from binlog before all other events, so just add them c->logevent_id = event.id_; update_channel(c, channel_id, true, false); } string ContactsManager::get_channel_database_key(ChannelId channel_id) { return PSTRING() << "ch" << channel_id.get(); } string ContactsManager::get_channel_database_value(const Channel *c) { return log_event_store(*c).as_slice().str(); } void ContactsManager::save_channel_to_database(Channel *c, ChannelId channel_id) { CHECK(c != nullptr); if (c->is_being_saved) { return; } if (loaded_from_database_channels_.count(channel_id)) { save_channel_to_database_impl(c, channel_id, get_channel_database_value(c)); return; } if (load_channel_from_database_queries_.count(channel_id) != 0) { return; } load_channel_from_database_impl(channel_id, Auto()); } void ContactsManager::save_channel_to_database_impl(Channel *c, ChannelId channel_id, string value) { CHECK(c != nullptr); CHECK(load_channel_from_database_queries_.count(channel_id) == 0); c->is_being_saved = true; c->is_saved = true; LOG(INFO) << "Trying to save to database " << channel_id; G()->td_db()->get_sqlite_pmc()->set( get_channel_database_key(channel_id), std::move(value), PromiseCreator::lambda([channel_id](Result<> result) { send_closure(G()->contacts_manager(), &ContactsManager::on_save_channel_to_database, channel_id, result.is_ok()); })); } void ContactsManager::on_save_channel_to_database(ChannelId channel_id, bool success) { Channel *c = get_channel(channel_id); CHECK(c != nullptr); CHECK(c->is_being_saved); CHECK(load_channel_from_database_queries_.count(channel_id) == 0); c->is_being_saved = false; if (!success) { LOG(ERROR) << "Failed to save " << channel_id << " to database"; c->is_saved = false; } else { LOG(INFO) << "Successfully saved " << channel_id << " to database"; } if (c->is_saved) { if (c->logevent_id != 0) { binlog_erase(G()->td_db()->get_binlog(), c->logevent_id); c->logevent_id = 0; } } else { save_channel(c, channel_id, c->logevent_id != 0); } } void ContactsManager::load_channel_from_database(Channel *c, ChannelId channel_id, Promise promise) { if (loaded_from_database_channels_.count(channel_id)) { promise.set_value(Unit()); return; } CHECK(c == nullptr || !c->is_being_saved); load_channel_from_database_impl(channel_id, std::move(promise)); } void ContactsManager::load_channel_from_database_impl(ChannelId channel_id, Promise promise) { LOG(INFO) << "Load " << channel_id << " from database"; auto &load_channel_queries = load_channel_from_database_queries_[channel_id]; load_channel_queries.push_back(std::move(promise)); if (load_channel_queries.size() == 1u) { G()->td_db()->get_sqlite_pmc()->get( get_channel_database_key(channel_id), PromiseCreator::lambda([channel_id](string value) { send_closure(G()->contacts_manager(), &ContactsManager::on_load_channel_from_database, channel_id, std::move(value)); })); } } void ContactsManager::on_load_channel_from_database(ChannelId channel_id, string value) { if (!loaded_from_database_channels_.insert(channel_id).second) { return; } auto it = load_channel_from_database_queries_.find(channel_id); vector> promises; if (it != load_channel_from_database_queries_.end()) { promises = std::move(it->second); CHECK(!promises.empty()); load_channel_from_database_queries_.erase(it); } LOG(INFO) << "Successfully loaded " << channel_id << " of size " << value.size() << " from database"; // G()->td_db()->get_sqlite_pmc()->erase(get_channel_database_key(channel_id), Auto()); // return; Channel *c = get_channel(channel_id); if (c == nullptr) { if (!value.empty()) { c = add_channel(channel_id, "on_load_channel_from_database"); log_event_parse(*c, value).ensure(); c->is_saved = true; update_channel(c, channel_id, true, true); } } else { CHECK(!c->is_saved); // channel can't be saved before load completes CHECK(!c->is_being_saved); auto new_value = get_channel_database_value(c); if (value != new_value) { save_channel_to_database_impl(c, channel_id, std::move(new_value)); } else if (c->logevent_id != 0) { binlog_erase(G()->td_db()->get_binlog(), c->logevent_id); c->logevent_id = 0; } } for (auto &promise : promises) { promise.set_value(Unit()); } } bool ContactsManager::have_channel_force(ChannelId channel_id) { return get_channel_force(channel_id) != nullptr; } ContactsManager::Channel *ContactsManager::get_channel_force(ChannelId channel_id) { if (!channel_id.is_valid()) { return nullptr; } Channel *c = get_channel(channel_id); if (c != nullptr) { return c; } if (!G()->parameters().use_chat_info_db) { return nullptr; } if (loaded_from_database_channels_.count(channel_id)) { return nullptr; } LOG(INFO) << "Trying to load " << channel_id << " from database"; on_load_channel_from_database(channel_id, G()->td_db()->get_sqlite_sync_pmc()->get(get_channel_database_key(channel_id))); return get_channel(channel_id); } class ContactsManager::SecretChatLogEvent { public: SecretChatId secret_chat_id; SecretChat c; SecretChatLogEvent() = default; SecretChatLogEvent(SecretChatId secret_chat_id, const SecretChat &c) : secret_chat_id(secret_chat_id), c(c) { } template void store(StorerT &storer) const { td::store(secret_chat_id, storer); td::store(c, storer); } template void parse(ParserT &parser) { td::parse(secret_chat_id, parser); td::parse(c, parser); } }; void ContactsManager::save_secret_chat(SecretChat *c, SecretChatId secret_chat_id, bool from_binlog) { if (!G()->parameters().use_chat_info_db) { return; } CHECK(c != nullptr); if (!c->is_saved) { if (!from_binlog) { auto logevent = SecretChatLogEvent(secret_chat_id, *c); auto storer = LogEventStorerImpl(logevent); if (c->logevent_id == 0) { c->logevent_id = binlog_add(G()->td_db()->get_binlog(), LogEvent::HandlerType::SecretChatInfos, storer); } else { binlog_rewrite(G()->td_db()->get_binlog(), c->logevent_id, LogEvent::HandlerType::SecretChatInfos, storer); } } save_secret_chat_to_database(c, secret_chat_id); return; } } void ContactsManager::on_binlog_secret_chat_event(BinlogEvent &&event) { if (!G()->parameters().use_chat_info_db) { binlog_erase(G()->td_db()->get_binlog(), event.id_); return; } SecretChatLogEvent log_event; log_event_parse(log_event, event.data_).ensure(); auto secret_chat_id = log_event.secret_chat_id; LOG(INFO) << "Add " << secret_chat_id << " from binlog"; SecretChat *c = add_secret_chat(secret_chat_id); CHECK(c->date == 0); *c = std::move(log_event.c); // secret chats come from binlog before all other events, so just add them c->logevent_id = event.id_; update_secret_chat(c, secret_chat_id, true, false); } string ContactsManager::get_secret_chat_database_key(SecretChatId secret_chat_id) { return PSTRING() << "sc" << secret_chat_id.get(); } string ContactsManager::get_secret_chat_database_value(const SecretChat *c) { return log_event_store(*c).as_slice().str(); } void ContactsManager::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 ContactsManager::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); 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()->contacts_manager(), &ContactsManager::on_save_secret_chat_to_database, secret_chat_id, result.is_ok()); })); } void ContactsManager::on_save_secret_chat_to_database(SecretChatId secret_chat_id, bool success) { 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->logevent_id != 0) { binlog_erase(G()->td_db()->get_binlog(), c->logevent_id); c->logevent_id = 0; } } else { save_secret_chat(c, secret_chat_id, c->logevent_id != 0); } } void ContactsManager::load_secret_chat_from_database(SecretChat *c, SecretChatId secret_chat_id, Promise 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 ContactsManager::load_secret_chat_from_database_impl(SecretChatId secret_chat_id, Promise 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()->contacts_manager(), &ContactsManager::on_load_secret_chat_from_database, secret_chat_id, std::move(value)); })); } } void ContactsManager::on_load_secret_chat_from_database(SecretChatId secret_chat_id, string value) { 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> 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); log_event_parse(*c, value).ensure(); 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->logevent_id != 0) { binlog_erase(G()->td_db()->get_binlog(), c->logevent_id); c->logevent_id = 0; } } // TODO load users asynchronously if (c != nullptr && !have_user_force(c->user_id)) { LOG(ERROR) << "Can't find " << c->user_id << " from " << secret_chat_id; } for (auto &promise : promises) { promise.set_value(Unit()); } } bool ContactsManager::have_secret_chat_force(SecretChatId secret_chat_id) { return get_secret_chat_force(secret_chat_id) != nullptr; } ContactsManager::SecretChat *ContactsManager::get_secret_chat_force(SecretChatId secret_chat_id) { 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)) { LOG(ERROR) << "Can't find " << c->user_id << " from " << secret_chat_id; } return c; } if (!G()->parameters().use_chat_info_db) { return nullptr; } if (loaded_from_database_secret_chats_.count(secret_chat_id)) { return nullptr; } LOG(INFO) << "Trying to load " << secret_chat_id << " from database"; 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))); return get_secret_chat(secret_chat_id); } void ContactsManager::update_user(User *u, UserId user_id, bool from_binlog, bool from_database) { CHECK(u != nullptr); if (u->is_name_changed || u->is_username_changed || u->is_outbound_link_changed) { update_contacts_hints(u, user_id, from_database); } if (u->is_name_changed) { td_->messages_manager_->on_dialog_title_updated(DialogId(user_id)); auto it = secret_chats_with_user_.find(user_id); if (it != secret_chats_with_user_.end()) { for (auto secret_chat_id : it->second) { td_->messages_manager_->on_dialog_title_updated(DialogId(secret_chat_id)); } } } if (u->is_photo_changed) { td_->messages_manager_->on_dialog_photo_updated(DialogId(user_id)); auto it = secret_chats_with_user_.find(user_id); if (it != secret_chats_with_user_.end()) { for (auto secret_chat_id : it->second) { td_->messages_manager_->on_dialog_photo_updated(DialogId(secret_chat_id)); } } add_user_photo_id(u, user_id, u->photo.id, dialog_photo_get_file_ids(u->photo)); UserFull *user_full = get_user_full(user_id); if (user_full != nullptr) { user_full->photos.clear(); if (u->photo.id == 0) { user_full->photo_count = 0; } else { user_full->photo_count = -1; } user_full->photos_offset = user_full->photo_count; } } if (u->is_status_changed && user_id != get_my_id()) { auto left_time = u->was_online - G()->server_time_cached(); 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; 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()); } } u->is_name_changed = false; u->is_username_changed = false; u->is_photo_changed = false; u->is_outbound_link_changed = false; if (u->is_deleted) { td_->inline_queries_manager_->remove_recent_inline_bot(user_id, Promise<>()); /* DialogId dialog_id(user_id); for (auto category : {TopDialogCategory::Correspondent, TopDialogCategory::BotPM, TopDialogCategory::BotInline}) { send_closure(G()->top_dialog_manager(), &TopDialogManager::delete_dialog, category, dialog_id, get_input_peer_user(user_id, AccessRights::Read)); } */ } LOG(DEBUG) << "Update " << user_id << ": is_changed = " << u->is_changed << ", need_send_update = " << u->need_send_update << ", is_status_changed = " << u->is_status_changed; if (u->is_changed || u->need_send_update) { if (!from_database) { u->is_saved = false; } if (u->need_send_update) { send_closure(G()->td(), &Td::send_update, make_tl_object(get_user_object(user_id, u))); u->need_send_update = false; u->is_status_changed = false; } u->is_changed = false; } if (u->is_status_changed) { if (!from_database) { u->is_status_saved = false; } send_closure(G()->td(), &Td::send_update, make_tl_object(user_id.get(), get_user_status_object(user_id, u))); u->is_status_changed = false; } if (!from_database) { save_user(u, user_id, from_binlog); } } void ContactsManager::update_chat(Chat *c, ChatId chat_id, bool from_binlog, bool from_database) { CHECK(c != nullptr); if (c->is_photo_changed) { auto file_ids = dialog_photo_get_file_ids(c->photo); if (!file_ids.empty()) { if (!c->photo_source_id.is_valid()) { c->photo_source_id = td_->file_reference_manager_->create_chat_photo_file_source(chat_id); } for (auto file_id : file_ids) { td_->file_manager_->add_file_source(file_id, c->photo_source_id); } } td_->messages_manager_->on_dialog_photo_updated(DialogId(chat_id)); } if (c->is_title_changed) { td_->messages_manager_->on_dialog_title_updated(DialogId(chat_id)); } c->is_photo_changed = false; c->is_title_changed = false; LOG(DEBUG) << "Update " << chat_id << ": is_changed = " << c->is_changed << ", need_send_update = " << c->need_send_update; if (c->is_changed || c->need_send_update) { if (!from_database) { c->is_saved = false; } c->is_changed = false; if (c->need_send_update) { send_closure(G()->td(), &Td::send_update, make_tl_object(get_basic_group_object(chat_id, c))); c->need_send_update = false; } } if (!from_database) { save_chat(c, chat_id, from_binlog); } } void ContactsManager::update_channel(Channel *c, ChannelId channel_id, bool from_binlog, bool from_database) { CHECK(c != nullptr); if (c->is_photo_changed) { auto file_ids = dialog_photo_get_file_ids(c->photo); if (!file_ids.empty()) { if (!c->photo_source_id.is_valid()) { c->photo_source_id = td_->file_reference_manager_->create_channel_photo_file_source(channel_id); } for (auto file_id : file_ids) { td_->file_manager_->add_file_source(file_id, c->photo_source_id); } } td_->messages_manager_->on_dialog_photo_updated(DialogId(channel_id)); } if (c->is_title_changed) { td_->messages_manager_->on_dialog_title_updated(DialogId(channel_id)); } if (c->is_status_changed) { c->status.update_restrictions(); auto until_date = c->status.get_until_date(); int32 left_time = 0; if (until_date > 0) { left_time = until_date - G()->unix_time_cached() + 1; CHECK(left_time > 0); } if (left_time > 0 && left_time < 366 * 86400) { channel_unban_timeout_.set_timeout_in(channel_id.get(), left_time); } else { channel_unban_timeout_.cancel_timeout(channel_id.get()); } } if (c->is_username_changed) { if (c->status.is_creator() && created_public_channels_inited_) { if (c->username.empty()) { created_public_channels_.erase( std::remove(created_public_channels_.begin(), created_public_channels_.end(), channel_id), created_public_channels_.end()); } else { if (std::find(created_public_channels_.begin(), created_public_channels_.end(), channel_id) == created_public_channels_.end()) { created_public_channels_.push_back(channel_id); } } } } c->is_photo_changed = false; c->is_title_changed = false; c->is_status_changed = false; c->is_username_changed = false; LOG(DEBUG) << "Update " << channel_id << ": is_changed = " << c->is_changed << ", need_send_update = " << c->need_send_update; if (c->is_changed || c->need_send_update) { if (!from_database) { c->is_saved = false; } c->is_changed = false; if (c->need_send_update) { send_closure(G()->td(), &Td::send_update, make_tl_object(get_supergroup_object(channel_id, c))); c->need_send_update = false; } } if (!from_database) { save_channel(c, channel_id, from_binlog); } bool have_read_access = have_input_peer_channel(c, AccessRights::Read); bool is_member = c->status.is_member(); if (c->had_read_access && !have_read_access) { send_closure_later(G()->messages_manager(), &MessagesManager::delete_dialog, DialogId(channel_id)); } else if (!from_database && c->was_member != is_member) { DialogId dialog_id(channel_id); send_closure_later(G()->messages_manager(), &MessagesManager::force_create_dialog, dialog_id, "update channel", true); } c->had_read_access = have_read_access; c->was_member = is_member; } void ContactsManager::update_secret_chat(SecretChat *c, SecretChatId secret_chat_id, bool from_binlog, bool from_database) { CHECK(c != nullptr); LOG(DEBUG) << "Update " << secret_chat_id << ": is_changed = " << c->is_changed << ", need_send_update = " << c->need_send_update; if (c->is_changed || c->need_send_update) { if (!from_database) { c->is_saved = false; } c->is_changed = false; if (c->need_send_update) { send_closure(G()->td(), &Td::send_update, make_tl_object(get_secret_chat_object(secret_chat_id, c))); c->need_send_update = false; } DialogId dialog_id(secret_chat_id); send_closure_later(G()->messages_manager(), &MessagesManager::force_create_dialog, dialog_id, "update secret chat", 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 (!from_database) { save_secret_chat(c, secret_chat_id, from_binlog); } } void ContactsManager::update_user_full(UserFull *user_full, UserId user_id) { CHECK(user_full != nullptr); if (user_full->is_changed) { user_full->is_changed = false; if (user_full->is_inited) { send_closure(G()->td(), &Td::send_update, make_tl_object(get_user_id_object(user_id, "updateUserFullInfo"), get_user_full_info_object(user_id, user_full))); } } } void ContactsManager::update_chat_full(ChatFull *chat_full, ChatId chat_id) { CHECK(chat_full != nullptr); if (chat_full->is_changed) { vector administrator_user_ids; vector bot_user_ids; for (auto &participant : chat_full->participants) { auto user_id = participant.user_id; if (participant.status.is_administrator()) { administrator_user_ids.push_back(user_id); } if (is_user_bot(user_id)) { bot_user_ids.push_back(user_id); } } on_update_dialog_administrators(DialogId(chat_id), std::move(administrator_user_ids), chat_full->version != -1); td_->messages_manager_->on_dialog_bots_updated(DialogId(chat_id), std::move(bot_user_ids)); chat_full->is_changed = false; send_closure( G()->td(), &Td::send_update, make_tl_object(get_basic_group_id_object(chat_id, "update_chat_full"), get_basic_group_full_info_object(chat_full))); } } void ContactsManager::update_channel_full(ChannelFull *channel_full, ChannelId channel_id) { CHECK(channel_full != nullptr); if (channel_full->is_changed) { if (channel_full->participant_count < channel_full->administrator_count) { channel_full->administrator_count = channel_full->participant_count; } channel_full->is_changed = false; send_closure( G()->td(), &Td::send_update, make_tl_object(get_supergroup_id_object(channel_id, "update_channel_full"), get_supergroup_full_info_object(channel_full))); } } void ContactsManager::on_get_users(vector> &&users, const char *source) { for (auto &user : users) { on_get_user(std::move(user), source); } } void ContactsManager::on_get_user_full(tl_object_ptr &&user_full) { UserId user_id = get_user_id(user_full->user_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; return; } on_get_user(std::move(user_full->user_), "on_get_user_full"); const User *u = get_user(user_id); if (u == nullptr) { return; } on_update_user_links(user_id, std::move(user_full->link_->my_link_), std::move(user_full->link_->foreign_link_)); td_->messages_manager_->on_update_dialog_notify_settings(DialogId(user_id), std::move(user_full->notify_settings_)); MessageId pinned_message_id; if ((user_full->flags_ & USER_FULL_FLAG_HAS_PINNED_MESSAGE) != 0) { pinned_message_id = MessageId(ServerMessageId(user_full->pinned_msg_id_)); } td_->messages_manager_->on_update_dialog_pinned_message_id(DialogId(user_id), pinned_message_id); UserFull *user = &users_full_[user_id]; user->expires_at = Time::now() + USER_FULL_EXPIRE_TIME; user->is_inited = true; on_update_user_full_is_blocked(user, user_id, (user_full->flags_ & USER_FULL_FLAG_IS_BLOCKED) != 0); bool can_be_called = user_full->phone_calls_available_ && !user_full->phone_calls_private_; bool has_private_calls = user_full->phone_calls_private_; if (user->can_be_called != can_be_called || user->has_private_calls != has_private_calls || user->about != user_full->about_ || user->common_chat_count != user_full->common_chats_count_) { user->can_be_called = can_be_called; user->has_private_calls = has_private_calls; user->about = std::move(user_full->about_); user->common_chat_count = user_full->common_chats_count_; user->is_changed = true; } int32 photo_id = user_full->profile_photo_ == nullptr ? telegram_api::photoEmpty::ID : user_full->profile_photo_->get_id(); if (photo_id == telegram_api::photoEmpty::ID) { user->photo_count = 0; user->photos_offset = 0; user->photos.clear(); } else { CHECK(photo_id == telegram_api::photo::ID); // Photo profile_photo = // get_photo(td_->file_manager_.get(), move_tl_object_as(user_full->profile_photo_)); } if ((user_full->flags_ & USER_FULL_FLAG_HAS_BOT_INFO) != 0 && !u->is_deleted) { on_update_user_full_bot_info(user, user_id, u->bot_info_version, std::move(user_full->bot_info_)); } update_user_full(user, user_id); } void ContactsManager::on_get_user_photos(UserId user_id, int32 offset, int32 limit, int32 total_count, vector> photos) { int32 photo_count = narrow_cast(photos.size()); if (total_count < 0 || total_count < photo_count) { LOG(ERROR) << "Wrong photos total_count " << total_count << ". Receive " << photo_count << " photos"; total_count = photo_count; } LOG_IF(ERROR, limit < photo_count) << "Requested not more than " << limit << " photos, but " << photo_count << " returned"; 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(photo_ptr); auto profile_photo = convert_photo_to_profile_photo(server_photo); if (profile_photo) { get_profile_photo(td_->file_manager_.get(), std::move(profile_photo)); } else { LOG(ERROR) << "Failed to get profile photo from " << to_string(server_photo); } auto photo = get_photo(td_->file_manager_.get(), std::move(server_photo), DialogId()); add_user_photo_id(u, user_id, photo.id, photo_get_file_ids(photo)); } } return; } UserFull *user = &users_full_[user_id]; user->photo_count = total_count; CHECK(user->getting_photos_now); user->getting_photos_now = false; if (user->photos_offset == -1) { user->photos_offset = 0; CHECK(user->photos.empty()); } if (offset != narrow_cast(user->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.size(); user->photos.clear(); user->photos_offset = offset; } for (auto &photo : photos) { int32 photo_id = photo->get_id(); if (photo_id == telegram_api::photoEmpty::ID) { LOG(ERROR) << "Have got empty profile photo in getUserPhotos request for " << user_id << " with offset " << offset << " and limit " << limit << ". Receive " << photo_count << " photos out of " << total_count << " photos"; continue; } CHECK(photo_id == telegram_api::photo::ID); user->photos.push_back( get_photo(td_->file_manager_.get(), telegram_api::move_object_as(photo), DialogId())); add_user_photo_id(u, user_id, user->photos.back().id, photo_get_file_ids(user->photos.back())); } } bool ContactsManager::on_update_bot_info(tl_object_ptr &&bot_info) { CHECK(bot_info != nullptr); UserId user_id(bot_info->user_id_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; return false; } const User *u = get_user_force(user_id); if (u == nullptr) { LOG(ERROR) << "Have no " << user_id; return false; } if (u->is_deleted) { return false; } UserFull *user_full = &users_full_[user_id]; bool result = on_update_user_full_bot_info(user_full, user_id, u->bot_info_version, std::move(bot_info)); update_user_full(user_full, user_id); return result; } bool ContactsManager::on_update_user_full_bot_info(UserFull *user_full, UserId user_id, int32 bot_info_version, tl_object_ptr &&bot_info) { CHECK(user_full != nullptr); CHECK(bot_info != nullptr); if (user_full->bot_info != nullptr && user_full->bot_info->version > bot_info_version) { LOG(WARNING) << "Ignore outdated version of BotInfo for " << user_id << " with version " << bot_info_version << ", current version is " << user_full->bot_info->version; return false; } if (user_full->bot_info != nullptr && user_full->bot_info->version == bot_info_version) { LOG(DEBUG) << "Ignore already known version of BotInfo for " << user_id << " with version " << bot_info_version; return false; } vector> commands; commands.reserve(bot_info->commands_.size()); for (auto &command : bot_info->commands_) { commands.emplace_back(std::move(command->command_), std::move(command->description_)); } user_full->bot_info = td::make_unique(bot_info_version, std::move(bot_info->description_), std::move(commands)); user_full->is_changed = true; return true; } void ContactsManager::on_get_chat(tl_object_ptr &&chat, const char *source) { LOG(DEBUG) << "Receive from " << source << ' ' << to_string(chat); downcast_call(*chat, [this, source](auto &c) { this->on_chat_update(c, source); }); } void ContactsManager::on_get_chats(vector> &&chats, const char *source) { for (auto &chat : chats) { auto constuctor_id = chat->get_id(); if (constuctor_id == telegram_api::channel::ID || constuctor_id == telegram_api::channelForbidden::ID) { // apply info about megagroups before corresponding chats on_get_chat(std::move(chat), source); chat = nullptr; } } for (auto &chat : chats) { if (chat != nullptr) { on_get_chat(std::move(chat), source); chat = nullptr; } } } void ContactsManager::on_get_chat_full(tl_object_ptr &&chat_full_ptr) { LOG(INFO) << "Receive " << to_string(chat_full_ptr); if (chat_full_ptr->get_id() == telegram_api::chatFull::ID) { auto chat_full = move_tl_object_as(chat_full_ptr); ChatId chat_id(chat_full->id_); if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id; return; } MessageId pinned_message_id; if ((chat_full->flags_ & CHAT_FULL_FLAG_HAS_PINNED_MESSAGE) != 0) { pinned_message_id = MessageId(ServerMessageId(chat_full->pinned_msg_id_)); } td_->messages_manager_->on_update_dialog_pinned_message_id(DialogId(chat_id), pinned_message_id); ChatFull *chat = &chats_full_[chat_id]; on_update_chat_full_invite_link(chat, std::move(chat_full->exported_invite_)); // Ignoring chat_full->photo for (auto &bot_info : chat_full->bot_info_) { if (on_update_bot_info(std::move(bot_info))) { chat->is_changed = true; } } on_get_chat_participants(std::move(chat_full->participants_)); td_->messages_manager_->on_update_dialog_notify_settings(DialogId(chat_id), std::move(chat_full->notify_settings_)); update_chat_full(chat, chat_id); } else { CHECK(chat_full_ptr->get_id() == telegram_api::channelFull::ID); auto channel_full = move_tl_object_as(chat_full_ptr); ChannelId channel_id(channel_full->id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; return; } td_->messages_manager_->on_update_dialog_notify_settings(DialogId(channel_id), std::move(channel_full->notify_settings_)); // Ignoring channel_full->photo if (!have_channel(channel_id)) { LOG(ERROR) << channel_id << " not found"; return; } auto participant_count = (channel_full->flags_ & CHANNEL_FULL_FLAG_HAS_PARTICIPANT_COUNT) != 0 ? channel_full->participants_count_ : 0; auto administrator_count = (channel_full->flags_ & CHANNEL_FULL_FLAG_HAS_ADMINISTRATOR_COUNT) != 0 ? channel_full->admins_count_ : 0; auto restricted_count = (channel_full->flags_ & CHANNEL_FULL_FLAG_HAS_BANNED_COUNT) != 0 ? channel_full->banned_count_ : 0; auto banned_count = (channel_full->flags_ & CHANNEL_FULL_FLAG_HAS_BANNED_COUNT) != 0 ? channel_full->kicked_count_ : 0; auto can_get_participants = (channel_full->flags_ & CHANNEL_FULL_FLAG_CAN_GET_PARTICIPANTS) != 0; auto can_set_username = (channel_full->flags_ & CHANNEL_FULL_FLAG_CAN_SET_USERNAME) != 0; auto can_set_sticker_set = (channel_full->flags_ & CHANNEL_FULL_FLAG_CAN_SET_STICKERS) != 0; auto is_all_history_available = (channel_full->flags_ & CHANNEL_FULL_FLAG_IS_ALL_HISTORY_HIDDEN) == 0; int64 sticker_set_id = channel_full->stickerset_ == nullptr ? 0 : td_->stickers_manager_->on_get_sticker_set(std::move(channel_full->stickerset_), true); ChannelFull *channel = &channels_full_[channel_id]; channel->expires_at = Time::now() + CHANNEL_FULL_EXPIRE_TIME; if (channel->description != channel_full->about_ || channel->participant_count != participant_count || channel->administrator_count != administrator_count || channel->restricted_count != restricted_count || channel->banned_count != banned_count || channel->can_get_participants != can_get_participants || channel->can_set_username != can_set_username || channel->can_set_sticker_set != can_set_sticker_set || channel->sticker_set_id != sticker_set_id || channel->is_all_history_available != is_all_history_available) { channel->description = std::move(channel_full->about_); channel->participant_count = participant_count; channel->administrator_count = administrator_count; channel->restricted_count = restricted_count; channel->banned_count = banned_count; channel->can_get_participants = can_get_participants; channel->can_set_username = can_set_username; channel->can_set_sticker_set = can_set_sticker_set; channel->sticker_set_id = sticker_set_id; channel->is_all_history_available = is_all_history_available; channel->is_changed = true; if (participant_count != 0) { auto c = get_channel(channel_id); if (c != nullptr && c->participant_count != participant_count) { c->participant_count = participant_count; c->need_send_update = true; update_channel(c, channel_id); } } } td_->messages_manager_->on_read_channel_outbox(channel_id, MessageId(ServerMessageId(channel_full->read_outbox_max_id_))); if ((channel_full->flags_ & CHANNEL_FULL_FLAG_HAS_AVAILABLE_MIN_MESSAGE_ID) != 0) { td_->messages_manager_->on_update_channel_max_unavailable_message_id( channel_id, MessageId(ServerMessageId(channel_full->available_min_id_))); } td_->messages_manager_->on_read_channel_inbox( channel_id, MessageId(ServerMessageId(channel_full->read_inbox_max_id_)), channel_full->unread_count_); on_update_channel_full_invite_link(channel, std::move(channel_full->exported_invite_)); MessageId pinned_message_id; if ((channel_full->flags_ & CHANNEL_FULL_FLAG_HAS_PINNED_MESSAGE) != 0) { pinned_message_id = MessageId(ServerMessageId(channel_full->pinned_msg_id_)); } td_->messages_manager_->on_update_dialog_pinned_message_id(DialogId(channel_id), pinned_message_id); for (auto &bot_info : channel_full->bot_info_) { on_update_bot_info(std::move(bot_info)); } ChatId migrated_from_chat_id; MessageId migrated_from_max_message_id; if ((channel_full->flags_ & CHANNEL_FULL_FLAG_MIGRATED_FROM) != 0) { migrated_from_chat_id = ChatId(channel_full->migrated_from_chat_id_); migrated_from_max_message_id = MessageId(ServerMessageId(channel_full->migrated_from_max_id_)); } if (channel->migrated_from_chat_id != migrated_from_chat_id || channel->migrated_from_max_message_id != migrated_from_max_message_id) { channel->migrated_from_chat_id = migrated_from_chat_id; channel->migrated_from_max_message_id = migrated_from_max_message_id; channel->is_changed = true; } update_channel_full(channel, channel_id); } } bool ContactsManager::is_update_about_username_change_received(UserId user_id) const { const User *u = get_user(user_id); if (u != nullptr) { return u->inbound == LinkState::Contact; } else { return false; } } void ContactsManager::on_update_user_name(UserId user_id, string &&first_name, string &&last_name, string &&username) { if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; return; } User *u = get_user_force(user_id); if (u != nullptr) { on_update_user_name(u, user_id, std::move(first_name), std::move(last_name), std::move(username)); update_user(u, user_id); } else { LOG(INFO) << "Ignore update user name about unknown " << user_id; } } void ContactsManager::on_update_user_name(User *u, UserId user_id, string &&first_name, string &&last_name, string &&username) { 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->need_send_update = true; } td_->messages_manager_->on_dialog_username_updated(DialogId(user_id), u->username, username); if (u->username != username) { u->username = std::move(username); u->is_username_changed = true; LOG(DEBUG) << "Username has changed for " << user_id; u->need_send_update = true; } } void ContactsManager::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); 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 ContactsManager::on_update_user_phone_number(User *u, UserId user_id, string &&phone_number) { if (u->phone_number != phone_number) { u->phone_number = std::move(phone_number); LOG(DEBUG) << "Phone number has changed for " << user_id; u->need_send_update = true; } } void ContactsManager::on_update_user_photo(UserId user_id, tl_object_ptr &&photo_ptr) { if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; return; } User *u = get_user_force(user_id); if (u != nullptr) { on_update_user_photo(u, user_id, std::move(photo_ptr)); update_user(u, user_id); } else { LOG(INFO) << "Ignore update user photo about unknown " << user_id; } } void ContactsManager::on_update_user_photo(User *u, UserId user_id, tl_object_ptr &&photo) { if (td_->auth_manager_->is_bot() && !G()->parameters().use_file_db && !u->is_photo_inited) { bool is_empty = photo == nullptr || photo->get_id() == telegram_api::userProfilePhotoEmpty::ID; pending_user_photos_[user_id] = std::move(photo); UserFull *user_full = get_user_full(user_id); if (user_full != nullptr) { user_full->photos.clear(); if (is_empty) { user_full->photo_count = 0; } else { user_full->photo_count = -1; } user_full->photos_offset = user_full->photo_count; } return; } do_update_user_photo(u, user_id, std::move(photo)); } void ContactsManager::do_update_user_photo(User *u, UserId user_id, tl_object_ptr &&photo) { u->is_photo_inited = true; ProfilePhoto new_photo = get_profile_photo(td_->file_manager_.get(), std::move(photo)); if (new_photo != u->photo) { u->photo = new_photo; u->is_photo_changed = true; LOG(DEBUG) << "Photo has changed for " << user_id; u->need_send_update = true; } } void ContactsManager::add_user_photo_id(User *u, UserId user_id, int64 photo_id, const vector &photo_file_ids) { if (photo_id > 0 && !photo_file_ids.empty() && u->photo_ids.insert(photo_id).second) { FileSourceId file_source_id; auto it = user_profile_photo_file_source_ids_.find(std::make_pair(user_id, photo_id)); if (it != user_profile_photo_file_source_ids_.end()) { VLOG(file_references) << "Move " << it->second << " inside of " << user_id; file_source_id = it->second; user_profile_photo_file_source_ids_.erase(it); } else { VLOG(file_references) << "Need to create new file source for " << 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 ContactsManager::on_update_user_online(UserId user_id, tl_object_ptr &&status) { if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; return; } User *u = get_user_force(user_id); 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 ContactsManager::on_update_user_online(User *u, UserId user_id, tl_object_ptr &&status) { 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(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(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) { new_online = -1; } else if (id == telegram_api::userStatusLastWeek::ID) { new_online = -2; is_offline = true; } else if (id == telegram_api::userStatusLastMonth::ID) { new_online = -3; is_offline = true; } else { CHECK(id == telegram_api::userStatusEmpty::ID); new_online = 0; } if (new_online != u->was_online) { LOG(DEBUG) << "Update " << user_id << " online from " << u->was_online << " to " << new_online; u->was_online = new_online; u->is_status_changed = true; if (user_id == get_my_id()) { my_was_online_local_ = 0; if (is_offline) { td_->on_online_updated(false, false); } } } } void ContactsManager::on_update_user_blocked(UserId user_id, bool is_blocked) { LOG(INFO) << "Receive update user blocked with " << user_id << " and is_blocked = " << is_blocked; if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; return; } UserFull *user_full = get_user_full(user_id); if (user_full == nullptr) { return; } on_update_user_full_is_blocked(user_full, user_id, is_blocked); update_user_full(user_full, user_id); } void ContactsManager::on_update_user_full_is_blocked(UserFull *user_full, UserId user_id, bool is_blocked) { CHECK(user_full != nullptr); if (user_full->is_inited && user_full->is_blocked != is_blocked) { user_full->is_blocked = is_blocked; user_full->is_changed = true; } } void ContactsManager::on_delete_profile_photo(int64 profile_photo_id, Promise promise) { UserId my_id = get_my_id(); UserFull *user_full = get_user_full(my_id); if (user_full != nullptr) { // drop photo cache user_full->photos.clear(); user_full->photo_count = -1; user_full->photos_offset = -1; } auto input_user = get_input_user(my_id); CHECK(input_user != nullptr); vector> users; users.push_back(std::move(input_user)); td_->create_handler(std::move(promise))->send(std::move(users)); } ContactsManager::LinkState ContactsManager::get_link_state(tl_object_ptr &&link) { int32 id = link->get_id(); switch (id) { case telegram_api::contactLinkUnknown::ID: return LinkState::Unknown; case telegram_api::contactLinkNone::ID: return LinkState::None; case telegram_api::contactLinkHasPhone::ID: return LinkState::KnowsPhoneNumber; case telegram_api::contactLinkContact::ID: return LinkState::Contact; default: UNREACHABLE(); } return LinkState::Unknown; } void ContactsManager::on_update_user_links(UserId user_id, tl_object_ptr &&outbound, tl_object_ptr &&inbound) { if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; return; } User *u = get_user_force(user_id); if (u != nullptr) { on_update_user_links(u, user_id, get_link_state(std::move(outbound)), get_link_state(std::move(inbound))); update_user(u, user_id); } else { LOG(INFO) << "Ignore update user links about unknown " << user_id; } } void ContactsManager::on_update_user_links(User *u, UserId user_id, LinkState outbound, LinkState inbound) { UserId my_id = get_my_id(); if (user_id == my_id) { if (outbound == LinkState::None && !td_->auth_manager_->is_bot()) { outbound = LinkState::KnowsPhoneNumber; } inbound = outbound; } LOG(DEBUG) << "Update " << user_id << " links from (" << u->outbound << ", " << u->inbound << ") to (" << outbound << ", " << inbound << ")"; bool need_send_update = false; if (outbound != u->outbound && outbound != LinkState::Unknown) { need_send_update |= outbound != LinkState::None || u->outbound != LinkState::Unknown; LOG(DEBUG) << "Set outbound link to " << outbound << ", need_send_update = " << need_send_update; u->outbound = outbound; u->is_outbound_link_changed = true; u->is_changed = true; } if (inbound != u->inbound && inbound != LinkState::Unknown) { need_send_update |= inbound != LinkState::None || u->inbound != LinkState::Unknown; LOG(DEBUG) << "Set inbound link to " << inbound << ", need_send_update = " << need_send_update; u->inbound = inbound; u->is_changed = true; } if (u->inbound == LinkState::Contact && u->outbound != LinkState::Contact) { u->inbound = LinkState::KnowsPhoneNumber; u->is_changed = true; need_send_update = true; } if (need_send_update) { LOG(DEBUG) << "Links have changed for " << user_id; u->need_send_update = true; } } void ContactsManager::invalidate_user_full(UserId user_id) { auto user_full = get_user_full(user_id); if (user_full == nullptr) { return; } user_full->expires_at = 0.0; user_full->photos.clear(); user_full->photo_count = -1; user_full->photos_offset = -1; user_full->is_inited = true; user_full->is_blocked = false; user_full->can_be_called = false; user_full->has_private_calls = false; user_full->about = string(); user_full->common_chat_count = 0; user_full->bot_info = nullptr; user_full->is_changed = true; update_user_full(user_full, user_id); } void ContactsManager::on_get_chat_participants(tl_object_ptr &&participants_ptr) { switch (participants_ptr->get_id()) { case telegram_api::chatParticipantsForbidden::ID: { auto participants = move_tl_object_as(participants_ptr); ChatId chat_id(participants->chat_id_); if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id; return; } if (!have_chat_force(chat_id)) { LOG(ERROR) << chat_id << " not found"; return; } invalidate_chat_full(chat_id); break; } case telegram_api::chatParticipants::ID: { auto participants = move_tl_object_as(participants_ptr); ChatId chat_id(participants->chat_id_); if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id; return; } const Chat *c = get_chat_force(chat_id); if (c == nullptr) { LOG(ERROR) << chat_id << " not found"; return; } ChatFull *chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { LOG(INFO) << "Ignore update of members for unknown " << chat_id; return; } UserId new_creator_user_id; vector new_participants; new_participants.reserve(participants->participants_.size()); for (auto &participant_ptr : participants->participants_) { DialogParticipant dialog_participant; switch (participant_ptr->get_id()) { case telegram_api::chatParticipant::ID: { auto participant = move_tl_object_as(participant_ptr); dialog_participant = {UserId(participant->user_id_), UserId(participant->inviter_id_), participant->date_, DialogParticipantStatus::Member()}; break; } case telegram_api::chatParticipantCreator::ID: { auto participant = move_tl_object_as(participant_ptr); new_creator_user_id = UserId(participant->user_id_); dialog_participant = {new_creator_user_id, new_creator_user_id, c->date, DialogParticipantStatus::Creator(true)}; break; } case telegram_api::chatParticipantAdmin::ID: { auto participant = move_tl_object_as(participant_ptr); dialog_participant = {UserId(participant->user_id_), UserId(participant->inviter_id_), participant->date_, DialogParticipantStatus::GroupAdministrator(c->is_creator)}; break; } default: UNREACHABLE(); } LOG_IF(ERROR, !have_user(dialog_participant.user_id)) << "Have no information about " << dialog_participant.user_id; LOG_IF(ERROR, !have_user(dialog_participant.inviter_user_id)) << "Have no information about " << dialog_participant.inviter_user_id; if (dialog_participant.joined_date < c->date) { LOG_IF(ERROR, dialog_participant.joined_date < c->date - 30 && c->date >= 1486000000) << "Wrong join date = " << dialog_participant.joined_date << " for " << dialog_participant.user_id << ", " << chat_id << " was created at " << c->date; dialog_participant.joined_date = c->date; } new_participants.push_back(std::move(dialog_participant)); } if (new_creator_user_id.is_valid()) { LOG_IF(ERROR, !have_user(new_creator_user_id)) << "Have no information about group creator " << new_creator_user_id; if (chat_full->creator_user_id.is_valid() && chat_full->creator_user_id != new_creator_user_id) { LOG(ERROR) << "Group creator has changed from " << chat_full->creator_user_id << " to " << new_creator_user_id; } } if (chat_full->creator_user_id != new_creator_user_id) { chat_full->creator_user_id = new_creator_user_id; chat_full->is_changed = true; } on_update_chat_full_participants(chat_full, chat_id, std::move(new_participants), participants->version_); update_chat_full(chat_full, chat_id); break; } default: UNREACHABLE(); } } const DialogParticipant *ContactsManager::get_chat_participant(ChatId chat_id, UserId user_id) const { auto chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { return nullptr; } return get_chat_participant(chat_full, user_id); } const DialogParticipant *ContactsManager::get_chat_participant(const ChatFull *chat_full, UserId user_id) const { for (auto &dialog_participant : chat_full->participants) { if (dialog_participant.user_id == user_id) { return &dialog_participant; } } return nullptr; } DialogParticipant ContactsManager::get_dialog_participant( ChannelId channel_id, tl_object_ptr &&participant_ptr) const { switch (participant_ptr->get_id()) { case telegram_api::channelParticipant::ID: { auto participant = move_tl_object_as(participant_ptr); return {UserId(participant->user_id_), UserId(), participant->date_, DialogParticipantStatus::Member()}; } case telegram_api::channelParticipantSelf::ID: { auto participant = move_tl_object_as(participant_ptr); return {UserId(participant->user_id_), UserId(participant->inviter_id_), participant->date_, get_channel_status(channel_id)}; } case telegram_api::channelParticipantCreator::ID: { auto participant = move_tl_object_as(participant_ptr); return {UserId(participant->user_id_), UserId(), 0, DialogParticipantStatus::Creator(true)}; } case telegram_api::channelParticipantAdmin::ID: { auto participant = move_tl_object_as(participant_ptr); bool can_be_edited = (participant->flags_ & telegram_api::channelParticipantAdmin::CAN_EDIT_MASK) != 0; return {UserId(participant->user_id_), UserId(participant->promoted_by_), participant->date_, get_dialog_participant_status(can_be_edited, std::move(participant->admin_rights_))}; } case telegram_api::channelParticipantBanned::ID: { auto participant = move_tl_object_as(participant_ptr); auto is_member = (participant->flags_ & telegram_api::channelParticipantBanned::LEFT_MASK) == 0; return {UserId(participant->user_id_), UserId(participant->kicked_by_), participant->date_, get_dialog_participant_status(is_member, std::move(participant->banned_rights_))}; } default: UNREACHABLE(); return DialogParticipant(); } } tl_object_ptr ContactsManager::get_chat_member_object( const DialogParticipant &dialog_participant) const { UserId participant_user_id = dialog_participant.user_id; return make_tl_object( get_user_id_object(participant_user_id, "chatMember.user_id"), get_user_id_object(dialog_participant.inviter_user_id, "chatMember.inviter_user_id"), dialog_participant.joined_date, dialog_participant.status.get_chat_member_status_object(), get_bot_info_object(participant_user_id)); } bool ContactsManager::on_get_channel_error(ChannelId channel_id, const Status &status, const string &source) { LOG(INFO) << "Receive " << status << " in " << channel_id << " from " << source; if (status.code() == 401 || status.message() == CSlice("SESSION_REVOKED") || status.message() == CSlice("USER_DEACTIVATED")) { // authorization is lost return true; } if (status.code() == 420 || status.code() == 429) { // flood wait return true; } if (status.message() == CSlice("BOT_METHOD_INVALID")) { LOG(ERROR) << "Receive BOT_METHOD_INVALID from " << source; return true; } if (status.message() == "CHANNEL_PRIVATE" || status.message() == "CHANNEL_PUBLIC_GROUP_NA") { if (!channel_id.is_valid()) { LOG(ERROR) << "Receive " << status.message() << " in invalid " << channel_id << " from " << source; return false; } auto c = get_channel(channel_id); if (c == nullptr) { if (td_->auth_manager_->is_bot() && source == "GetChannelsQuery") { // get channel from server by its identifier return true; } LOG(ERROR) << "Receive " << status.message() << " in not found " << channel_id << " from " << source; return false; } auto debug_channel_object = oneline(to_string(get_supergroup_object(channel_id, c))); if (c->status.is_member()) { LOG(INFO) << "Emulate leaving " << channel_id; // TODO we also may try to write to public channel int32 flags = 0; if (c->is_megagroup) { flags |= CHANNEL_FLAG_IS_MEGAGROUP; } else { flags |= CHANNEL_FLAG_IS_BROADCAST; } telegram_api::channelForbidden update(flags, false /*ignored*/, false /*ignored*/, channel_id.get(), c->access_hash, c->title, 0); on_chat_update(update, "CHANNEL_PRIVATE"); } else if (!c->username.empty()) { LOG(INFO) << "Drop username of " << channel_id; on_update_channel_username(c, channel_id, ""); update_channel(c, channel_id); } LOG_IF(ERROR, have_input_peer_channel(c, AccessRights::Read)) << "Have read access to channel after receiving CHANNEL_PRIVATE. Channel state: " << oneline(to_string(get_supergroup_object(channel_id, c))) << ". Previous channel state: " << debug_channel_object; return true; } return false; } void ContactsManager::on_get_channel_participants_success( ChannelId channel_id, ChannelParticipantsFilter filter, int32 offset, int32 limit, int64 random_id, int32 total_count, vector> &&participants) { LOG(INFO) << "Receive " << participants.size() << " members in " << channel_id; auto it = received_channel_participants_.find(random_id); CHECK(it != received_channel_participants_.end()); it->second.first = total_count; auto &result = it->second.second; CHECK(result.empty()); for (auto &participant_ptr : participants) { result.push_back(get_dialog_participant(channel_id, std::move(participant_ptr))); } if (offset == 0 && static_cast(participants.size()) < limit) { if (filter.is_administrators() || filter.is_bots()) { auto user_ids = transform(result, [](const DialogParticipant &participant) { return participant.user_id; }); if (filter.is_administrators()) { on_update_dialog_administrators(DialogId(channel_id), std::move(user_ids), true); } else { td_->messages_manager_->on_dialog_bots_updated(DialogId(channel_id), std::move(user_ids)); } } } } void ContactsManager::on_get_channel_participants_fail(ChannelId channel_id, ChannelParticipantsFilter filter, int32 offset, int32 limit, int64 random_id) { // clean up received_channel_participants_.erase(random_id); } bool ContactsManager::speculative_add_count(int32 &count, int32 new_count) { new_count += count; if (new_count < 0) { new_count = 0; } if (new_count == count) { return false; } count = new_count; return true; } void ContactsManager::speculative_add_channel_participants(ChannelId channel_id, int32 new_participant_count, bool by_me) { if (by_me) { // Currently ignore all changes made by the current user, because they may be already counted invalidate_channel_full(channel_id, false); // just in case return; } auto c = get_channel(channel_id); if (c != nullptr && c->participant_count != 0 && speculative_add_count(c->participant_count, new_participant_count)) { c->need_send_update = true; update_channel(c, channel_id); } auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr) { return; } channel_full->is_changed |= speculative_add_count(channel_full->participant_count, new_participant_count); update_channel_full(channel_full, channel_id); } void ContactsManager::speculative_add_channel_user(ChannelId channel_id, UserId user_id, DialogParticipantStatus status, DialogParticipantStatus old_status) { auto c = get_channel(channel_id); if (c != nullptr && c->participant_count != 0 && speculative_add_count(c->participant_count, status.is_member() - old_status.is_member())) { c->need_send_update = true; update_channel(c, channel_id); } if (status.is_administrator() != old_status.is_administrator()) { DialogId dialog_id(channel_id); auto administrators_it = dialog_administrators_.find(dialog_id); if (administrators_it != dialog_administrators_.end()) { auto user_ids = administrators_it->second; auto it = std::find(user_ids.begin(), user_ids.end(), user_id); bool is_found = it != user_ids.end(); if (status.is_administrator() != is_found) { if (is_found) { user_ids.erase(it); } else { user_ids.push_back(user_id); } on_update_dialog_administrators(dialog_id, std::move(user_ids), true); } } } auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr) { return; } channel_full->is_changed |= speculative_add_count(channel_full->participant_count, status.is_member() - old_status.is_member()); channel_full->is_changed |= speculative_add_count(channel_full->administrator_count, status.is_administrator() - old_status.is_administrator()); channel_full->is_changed |= speculative_add_count(channel_full->restricted_count, status.is_restricted() - old_status.is_restricted()); channel_full->is_changed |= speculative_add_count(channel_full->banned_count, status.is_banned() - old_status.is_banned()); update_channel_full(channel_full, channel_id); } void ContactsManager::invalidate_channel_full(ChannelId channel_id, bool drop_invite_link) { LOG(INFO) << "Invalidate channel full for " << channel_id; // drop channel full cache auto channel_full = get_channel_full(channel_id); if (channel_full != nullptr) { channel_full->expires_at = 0.0; if (drop_invite_link) { on_update_channel_full_invite_link(channel_full, nullptr); } update_channel_full(channel_full, channel_id); } else if (drop_invite_link) { auto it = channel_invite_links_.find(channel_id); if (it != channel_invite_links_.end()) { invalidate_invite_link_info(it->second); } } } void ContactsManager::on_get_chat_invite_link(ChatId chat_id, tl_object_ptr &&invite_link_ptr) { CHECK(chat_id.is_valid()); if (!have_chat(chat_id)) { LOG(ERROR) << chat_id << " not found"; return; } auto chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { update_invite_link(chat_invite_links_[chat_id], std::move(invite_link_ptr)); return; } on_update_chat_full_invite_link(chat_full, std::move(invite_link_ptr)); update_chat_full(chat_full, chat_id); } void ContactsManager::on_update_chat_full_invite_link( ChatFull *chat_full, tl_object_ptr &&invite_link_ptr) { CHECK(chat_full != nullptr); if (update_invite_link(chat_full->invite_link, std::move(invite_link_ptr))) { chat_full->is_changed = true; } } void ContactsManager::on_get_channel_invite_link(ChannelId channel_id, tl_object_ptr &&invite_link_ptr) { CHECK(channel_id.is_valid()); if (!have_channel(channel_id)) { LOG(ERROR) << channel_id << " not found"; return; } auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr) { update_invite_link(channel_invite_links_[channel_id], std::move(invite_link_ptr)); return; } on_update_channel_full_invite_link(channel_full, std::move(invite_link_ptr)); update_channel_full(channel_full, channel_id); } void ContactsManager::on_update_channel_full_invite_link( ChannelFull *channel_full, tl_object_ptr &&invite_link_ptr) { CHECK(channel_full != nullptr); if (update_invite_link(channel_full->invite_link, std::move(invite_link_ptr))) { channel_full->is_changed = true; } } void ContactsManager::on_get_dialog_invite_link_info(const string &invite_link, tl_object_ptr &&chat_invite_ptr) { auto &invite_link_info = invite_link_infos_[invite_link]; if (invite_link_info == nullptr) { invite_link_info = make_unique(); } CHECK(chat_invite_ptr != nullptr); switch (chat_invite_ptr->get_id()) { case telegram_api::chatInviteAlready::ID: { auto chat_invite_already = move_tl_object_as(chat_invite_ptr); auto chat_id = get_chat_id(chat_invite_already->chat_); if (chat_id != ChatId() && !chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id; chat_id = ChatId(); } auto channel_id = get_channel_id(chat_invite_already->chat_); if (channel_id != ChannelId() && !channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; channel_id = ChannelId(); } on_get_chat(std::move(chat_invite_already->chat_), "chatInviteAlready"); CHECK(chat_id == ChatId() || channel_id == ChannelId()); invite_link_info->chat_id = chat_id; invite_link_info->channel_id = channel_id; if (chat_id.is_valid()) { on_get_chat_invite_link(chat_id, make_tl_object(invite_link)); } if (channel_id.is_valid()) { on_get_channel_invite_link(channel_id, make_tl_object(invite_link)); } break; } case telegram_api::chatInvite::ID: { auto chat_invite = move_tl_object_as(chat_invite_ptr); invite_link_info->chat_id = ChatId(); invite_link_info->channel_id = ChannelId(); invite_link_info->title = chat_invite->title_; invite_link_info->photo = get_dialog_photo(td_->file_manager_.get(), std::move(chat_invite->photo_)); invite_link_info->participant_count = chat_invite->participants_count_; invite_link_info->participant_user_ids.clear(); for (auto &user : chat_invite->participants_) { auto user_id = get_user_id(user); if (!user_id.is_valid()) { LOG(ERROR) << "Receive invalid " << user_id; } else { on_get_user(std::move(user), "chatInvite"); } invite_link_info->participant_user_ids.push_back(user_id); } invite_link_info->is_chat = (chat_invite->flags_ & CHAT_INVITE_FLAG_IS_CHANNEL) == 0; invite_link_info->is_channel = (chat_invite->flags_ & CHAT_INVITE_FLAG_IS_CHANNEL) != 0; bool is_broadcast = (chat_invite->flags_ & CHAT_INVITE_FLAG_IS_BROADCAST) != 0; bool is_public = (chat_invite->flags_ & CHAT_INVITE_FLAG_IS_PUBLIC) != 0; bool is_megagroup = (chat_invite->flags_ & CHAT_INVITE_FLAG_IS_MEGAGROUP) != 0; if (!invite_link_info->is_channel) { if (is_broadcast || is_public || is_megagroup) { LOG(ERROR) << "Receive wrong chat invite: " << to_string(chat_invite); is_public = is_megagroup = false; } } else { LOG_IF(ERROR, is_broadcast == is_megagroup) << "Receive wrong chat invite: " << to_string(chat_invite); } invite_link_info->is_public = is_public; invite_link_info->is_megagroup = is_megagroup; break; } default: UNREACHABLE(); } } bool ContactsManager::is_valid_invite_link(const string &invite_link) { return !get_dialog_invite_link_hash(invite_link).empty(); } Slice ContactsManager::get_dialog_invite_link_hash(const string &invite_link) { auto lower_cased_invite_link_str = to_lower(invite_link); Slice lower_cased_invite_link = lower_cased_invite_link_str; size_t offset = 0; if (begins_with(lower_cased_invite_link, "https://")) { offset = 8; } else if (begins_with(lower_cased_invite_link, "http://")) { offset = 7; } lower_cased_invite_link.remove_prefix(offset); for (auto &url : INVITE_LINK_URLS) { if (begins_with(lower_cased_invite_link, url)) { return Slice(invite_link).substr(url.size() + offset); } } return Slice(); } bool ContactsManager::update_invite_link(string &invite_link, tl_object_ptr &&invite_link_ptr) { string new_invite_link; if (invite_link_ptr != nullptr) { switch (invite_link_ptr->get_id()) { case telegram_api::chatInviteEmpty::ID: // link is empty break; case telegram_api::chatInviteExported::ID: { auto chat_invite_exported = move_tl_object_as(invite_link_ptr); new_invite_link = std::move(chat_invite_exported->link_); break; } default: UNREACHABLE(); } } if (new_invite_link != invite_link) { if (!invite_link.empty()) { invite_link_infos_.erase(invite_link); } LOG_IF(ERROR, !new_invite_link.empty() && !is_valid_invite_link(new_invite_link)) << "Unsupported invite link " << new_invite_link; invite_link = std::move(new_invite_link); return true; } return false; } void ContactsManager::invalidate_invite_link_info(const string &invite_link) { LOG(INFO) << "Invalidate info about invite link " << invite_link; invite_link_infos_.erase(invite_link); } void ContactsManager::repair_chat_participants(ChatId chat_id) { send_get_chat_full_query(chat_id, Auto()); } void ContactsManager::on_update_chat_add_user(ChatId chat_id, UserId inviter_user_id, UserId user_id, int32 date, int32 version) { if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id; return; } if (!have_user(user_id)) { LOG(ERROR) << "Can't find " << user_id; return; } if (!have_user(inviter_user_id)) { LOG(ERROR) << "Can't find " << inviter_user_id; return; } LOG(INFO) << "Receive updateChatParticipantAdd to " << chat_id << " with " << user_id << " invited by " << inviter_user_id << " at " << date << " with version " << version; ChatFull *chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { LOG(INFO) << "Ignoring update about members of " << chat_id; return; } auto c = get_chat(chat_id); if (c == nullptr) { LOG(ERROR) << "Receive updateChatParticipantAdd for unknown " << chat_id << ". Couldn't apply it"; repair_chat_participants(chat_id); return; } if (c->left) { // possible if updates come out of order LOG(WARNING) << "Receive updateChatParticipantAdd for left " << chat_id << ". Couldn't apply it"; repair_chat_participants(chat_id); // just in case return; } if (on_update_chat_full_participants_short(chat_full, chat_id, version)) { for (auto &participant : chat_full->participants) { if (participant.user_id == user_id) { if (participant.inviter_user_id != inviter_user_id) { LOG(ERROR) << user_id << " was readded to " << chat_id << " by " << inviter_user_id << ", previously invited by " << participant.inviter_user_id; participant.inviter_user_id = inviter_user_id; participant.joined_date = date; repair_chat_participants(chat_id); } else { // Possible if update comes twice LOG(INFO) << user_id << " was readded to " << chat_id; } return; } } chat_full->participants.push_back(DialogParticipant{user_id, inviter_user_id, date, user_id == chat_full->creator_user_id ? DialogParticipantStatus::Creator(true) : DialogParticipantStatus::Member()}); chat_full->is_changed = true; update_chat_full(chat_full, chat_id); // Chat is already updated if (chat_full->version == c->version && narrow_cast(chat_full->participants.size()) != c->participant_count) { LOG(ERROR) << "Number of members of " << chat_id << " with version " << c->version << " is " << c->participant_count << " but there are " << chat_full->participants.size() << " members in the ChatFull"; repair_chat_participants(chat_id); } } } void ContactsManager::on_update_chat_edit_administrator(ChatId chat_id, UserId user_id, bool is_administrator, int32 version) { if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id; return; } if (!have_user(user_id)) { LOG(ERROR) << "Can't find " << user_id; return; } LOG(INFO) << "Receive updateChatParticipantAdmin in " << chat_id << " with " << user_id << ", administrator rights " << (is_administrator ? "enabled" : "disabled") << " with version " << version; auto c = get_chat_force(chat_id); if (c == nullptr) { LOG(INFO) << "Ignoring update about members of unknown " << chat_id; return; } if (c->left) { // possible if updates come out of order LOG(WARNING) << "Receive updateChatParticipantAdmin for left " << chat_id << ". Couldn't apply it"; repair_chat_participants(chat_id); // just in case return; } if (version <= -1) { LOG(ERROR) << "Receive wrong version " << version << " for " << chat_id; return; } CHECK(c->version >= 0); if (version > c->version) { if (version != c->version + 1) { LOG(ERROR) << "Administrators of " << chat_id << " with version " << c->version << " has changed but new version is " << version; repair_chat_participants(chat_id); return; } c->version = version; c->is_changed = true; if (user_id == get_my_id()) { on_update_chat_rights(c, chat_id, c->is_creator, is_administrator, c->everyone_is_administrator); } update_chat(c, chat_id); } ChatFull *chat_full = get_chat_full(chat_id); if (chat_full != nullptr) { if (chat_full->version + 1 == version) { for (auto &participant : chat_full->participants) { if (participant.user_id == user_id) { participant.status = is_administrator ? DialogParticipantStatus::GroupAdministrator(c->is_creator) : DialogParticipantStatus::Member(); chat_full->is_changed = true; update_chat_full(chat_full, chat_id); return; } } } // can't find chat member or version have increased too much repair_chat_participants(chat_id); } } void ContactsManager::on_update_chat_delete_user(ChatId chat_id, UserId user_id, int32 version) { if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id; return; } if (!have_user(user_id)) { LOG(ERROR) << "Can't find " << user_id; return; } LOG(INFO) << "Receive updateChatParticipantDelete from " << chat_id << " with " << user_id << " and version " << version; ChatFull *chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { LOG(INFO) << "Ignoring update about members of " << chat_id; return; } const Chat *c = get_chat_force(chat_id); if (c == nullptr) { LOG(ERROR) << "Receive updateChatParticipantDelete for unknown " << chat_id; repair_chat_participants(chat_id); return; } if (user_id == get_my_id()) { LOG_IF(WARNING, !c->left) << "User was removed from " << chat_id << " but it is not left the group. Possible if updates comes out of order"; return; } if (c->left) { // possible if updates come out of order LOG(INFO) << "Receive updateChatParticipantDelete for left " << chat_id; repair_chat_participants(chat_id); return; } if (on_update_chat_full_participants_short(chat_full, chat_id, version)) { for (size_t i = 0; i < chat_full->participants.size(); i++) { if (chat_full->participants[i].user_id == user_id) { chat_full->participants[i] = chat_full->participants.back(); chat_full->participants.resize(chat_full->participants.size() - 1); chat_full->is_changed = true; update_chat_full(chat_full, chat_id); if (static_cast(chat_full->participants.size()) != c->participant_count) { repair_chat_participants(chat_id); } return; } } LOG(ERROR) << "Can't find group member " << user_id << " in " << chat_id << " to delete him"; repair_chat_participants(chat_id); } } void ContactsManager::on_update_chat_everyone_is_administrator(ChatId chat_id, bool everyone_is_administrator, int32 version) { if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id; return; } LOG(INFO) << "Receive updateChatAdmins in " << chat_id << " with version " << version << " and everyone_is_administrator = " << everyone_is_administrator << ". Current version is " << version; auto c = get_chat_force(chat_id); if (c == nullptr) { LOG(INFO) << "Ignoring update about unknown " << chat_id; return; } if (c->left) { // possible if updates come out of order LOG(WARNING) << "Receive updateChatAdmins for left " << chat_id << ". Couldn't apply it"; repair_chat_participants(chat_id); // just in case return; } if (version <= -1) { LOG(ERROR) << "Receive wrong version " << version << " for " << chat_id; return; } CHECK(c->version >= 0); if (version > c->version) { if (version != c->version + 1) { LOG(WARNING) << "Anyone can edit of " << chat_id << " with version " << c->version << " has changed but new version is " << version; repair_chat_participants(chat_id); return; } LOG_IF(ERROR, everyone_is_administrator == c->everyone_is_administrator) << "Receive updateChatAdmins in " << chat_id << " with version " << version << " and everyone_is_administrator = " << everyone_is_administrator << ", but everyone_is_administrator is not changed. Current version is " << c->version; c->version = version; c->is_changed = true; on_update_chat_rights(c, chat_id, c->is_creator, c->is_administrator, everyone_is_administrator); update_chat(c, chat_id); } } void ContactsManager::on_update_chat_left(Chat *c, ChatId chat_id, bool left, bool kicked) { if (c->left != left || c->kicked != kicked) { bool drop_invite_link = c->left != left; c->left = left; c->kicked = kicked; if (c->left) { c->participant_count = 0; c->version = -1; invalidate_chat_full(chat_id); } if (drop_invite_link) { auto it = chat_invite_links_.find(chat_id); if (it != chat_invite_links_.end()) { invalidate_invite_link_info(it->second); } } c->need_send_update = true; } } void ContactsManager::on_update_chat_rights(Chat *c, ChatId chat_id, bool is_creator, bool is_administrator, bool everyone_is_administrator) { if (c->is_creator != is_creator || c->is_administrator != is_administrator || c->everyone_is_administrator != everyone_is_administrator) { c->is_creator = is_creator; c->is_administrator = is_administrator; c->everyone_is_administrator = everyone_is_administrator; c->can_edit = is_creator || is_administrator || everyone_is_administrator; c->need_send_update = true; } } void ContactsManager::on_update_chat_participant_count(Chat *c, ChatId chat_id, int32 participant_count, int32 version, const string &debug_str) { if (version <= -1) { LOG(ERROR) << "Receive wrong version " << version << " from " << debug_str; return; } if (version < c->version) { // some outdated data LOG(INFO) << "Receive member count of " << chat_id << " with version " << version << " from " << debug_str << ", but current version is " << c->version; return; } if (c->participant_count != participant_count) { if (version == c->version) { // version is not changed when deleted user is removed from the chat LOG_IF(ERROR, c->participant_count != participant_count + 1) << "Member count of " << chat_id << " has changed from " << c->participant_count << " to " << participant_count << ", but version " << c->version << " remains unchanged in " << debug_str; repair_chat_participants(chat_id); } c->participant_count = participant_count; c->version = version; c->need_send_update = true; return; } if (version > c->version) { c->version = version; c->is_changed = true; } } void ContactsManager::on_update_chat_photo(Chat *c, ChatId chat_id, tl_object_ptr &&chat_photo_ptr) { DialogPhoto new_chat_photo = get_dialog_photo(td_->file_manager_.get(), std::move(chat_photo_ptr)); if (new_chat_photo != c->photo) { if (c->photo_source_id.is_valid()) { for (auto file_id : dialog_photo_get_file_ids(c->photo)) { td_->file_manager_->remove_file_source(file_id, c->photo_source_id); } } c->photo = new_chat_photo; c->is_photo_changed = true; c->is_changed = true; } } void ContactsManager::on_update_chat_title(Chat *c, ChatId chat_id, string &&title) { if (c->title != title) { c->title = std::move(title); c->is_title_changed = true; c->is_changed = true; } } void ContactsManager::on_update_chat_active(Chat *c, ChatId chat_id, bool is_active) { if (c->is_active != is_active) { c->is_active = is_active; c->need_send_update = true; } } void ContactsManager::on_update_chat_migrated_to_channel_id(Chat *c, ChatId chat_id, ChannelId migrated_to_channel_id) { if (c->migrated_to_channel_id != migrated_to_channel_id && migrated_to_channel_id.is_valid()) { LOG_IF(ERROR, c->migrated_to_channel_id.is_valid()) << "Group upgraded to supergroup has changed from " << c->migrated_to_channel_id << " to " << migrated_to_channel_id; c->migrated_to_channel_id = migrated_to_channel_id; c->need_send_update = true; } } bool ContactsManager::on_update_chat_full_participants_short(ChatFull *chat_full, ChatId chat_id, int32 version) { if (version <= -1) { LOG(ERROR) << "Receive wrong version " << version << " for " << chat_id; return false; } if (chat_full->version == -1) { // chat members are unknown, nothing to update return false; } if (chat_full->version + 1 == version) { chat_full->version = version; return true; } LOG(ERROR) << "Member count of " << chat_id << " with version " << chat_full->version << " has changed but new version is " << version; repair_chat_participants(chat_id); return false; } void ContactsManager::on_update_chat_full_participants(ChatFull *chat_full, ChatId chat_id, vector participants, int32 version) { if (version <= -1) { LOG(ERROR) << "Receive members with wrong version " << version; return; } if (version < chat_full->version) { // some outdated data LOG(WARNING) << "Receive members of " << chat_id << " with version " << version << " but current version is " << chat_full->version; return; } if (chat_full->participants.size() != participants.size() && version == chat_full->version) { LOG(INFO) << "Members of " << chat_id << " has changed"; // this is possible in very rare situations repair_chat_participants(chat_id); } chat_full->participants = std::move(participants); chat_full->version = version; chat_full->is_changed = true; } void ContactsManager::invalidate_chat_full(ChatId chat_id) { ChatFull *chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { auto it = chat_invite_links_.find(chat_id); if (it != chat_invite_links_.end()) { invalidate_invite_link_info(it->second); } return; } LOG(INFO) << "Invalidate groupFull of " << chat_id; //chat_full->creator_user_id = UserId(); chat_full->participants.clear(); chat_full->version = -1; update_invite_link(chat_full->invite_link, nullptr); chat_full->is_changed = true; update_chat_full(chat_full, chat_id); } void ContactsManager::on_update_channel_photo(Channel *c, ChannelId channel_id, tl_object_ptr &&chat_photo_ptr) { DialogPhoto new_chat_photo = get_dialog_photo(td_->file_manager_.get(), std::move(chat_photo_ptr)); if (new_chat_photo != c->photo) { if (c->photo_source_id.is_valid()) { for (auto file_id : dialog_photo_get_file_ids(c->photo)) { td_->file_manager_->remove_file_source(file_id, c->photo_source_id); } } c->photo = new_chat_photo; c->is_photo_changed = true; c->is_changed = true; } } void ContactsManager::on_update_channel_title(Channel *c, ChannelId channel_id, string &&title) { if (c->title != title) { c->title = std::move(title); c->is_title_changed = true; c->is_changed = true; } } void ContactsManager::on_update_channel_status(Channel *c, ChannelId channel_id, DialogParticipantStatus &&status) { if (c->status != status) { LOG(INFO) << "Update " << channel_id << " status from " << c->status << " to " << status; bool drop_invite_link = c->status.is_administrator() != status.is_administrator() || c->status.is_member() != status.is_member(); c->status = status; c->is_status_changed = true; c->need_send_update = true; invalidate_channel_full(channel_id, drop_invite_link); } } void ContactsManager::on_update_channel_username(ChannelId channel_id, string &&username) { if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; return; } Channel *c = get_channel_force(channel_id); if (c != nullptr) { on_update_channel_username(c, channel_id, std::move(username)); update_channel(c, channel_id); } else { LOG(ERROR) << "Ignore update channel username about unknown " << channel_id; } } void ContactsManager::on_update_channel_username(Channel *c, ChannelId channel_id, string &&username) { td_->messages_manager_->on_dialog_username_updated(DialogId(channel_id), c->username, username); if (c->username != username) { if (c->username.empty() || username.empty()) { // moving channel from private to public can change availability of chat members invalidate_channel_full(channel_id, true); } c->username = std::move(username); c->is_username_changed = true; c->need_send_update = true; } } void ContactsManager::on_update_channel_description(ChannelId channel_id, string &&description) { if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; return; } auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr) { return; } if (channel_full->description != description) { channel_full->description = std::move(description); channel_full->is_changed = true; update_channel_full(channel_full, channel_id); } } void ContactsManager::on_update_channel_sticker_set(ChannelId channel_id, int64 sticker_set_id) { if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; return; } auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr) { return; } if (channel_full->sticker_set_id != sticker_set_id) { channel_full->sticker_set_id = sticker_set_id; channel_full->is_changed = true; update_channel_full(channel_full, channel_id); } } void ContactsManager::on_update_channel_is_all_history_available(ChannelId channel_id, bool is_all_history_available) { if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; return; } auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr) { return; } if (channel_full->is_all_history_available != is_all_history_available) { channel_full->is_all_history_available = is_all_history_available; channel_full->is_changed = true; update_channel_full(channel_full, channel_id); } } void ContactsManager::update_contacts_hints(const User *u, UserId user_id, bool from_database) { bool is_contact = u->outbound == LinkState::Contact && user_id != get_my_id(); 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 ? u->first_name + " " + u->last_name + " " + u->username : ""; if (new_value != old_value) { if (is_contact) { contacts_hints_.add(key, new_value); } else { contacts_hints_.remove(key); } } if (G()->parameters().use_chat_info_db) { // update contacts database if (!are_contacts_loaded_) { if (!from_database && load_contacts_queries_.empty()) { search_contacts("", std::numeric_limits::max(), Auto()); } } else { if (old_value.empty() == is_contact) { save_contacts_to_database(); } } } } bool ContactsManager::have_user(UserId user_id) const { auto u = get_user(user_id); return u != nullptr && u->is_received; } bool ContactsManager::have_min_user(UserId user_id) const { return users_.count(user_id) > 0; } bool ContactsManager::is_user_deleted(UserId user_id) const { auto u = get_user(user_id); return u == nullptr || u->is_deleted; } bool ContactsManager::is_user_bot(UserId user_id) const { auto u = get_user(user_id); return u != nullptr && !u->is_deleted && u->is_bot; } Result ContactsManager::get_bot_data(UserId user_id) const { auto p = users_.find(user_id); if (p == users_.end()) { return Status::Error(5, "Bot not found"); } auto bot = &p->second; if (!bot->is_bot) { return Status::Error(5, "User is not a bot"); } if (bot->is_deleted) { return Status::Error(5, "Bot is deleted"); } if (!bot->is_received) { return Status::Error(5, "Bot is inaccessible"); } BotData bot_data; bot_data.username = bot->username; bot_data.can_join_groups = bot->can_join_groups; bot_data.can_read_all_group_messages = bot->can_read_all_group_messages; bot_data.is_inline = bot->is_inline_bot; bot_data.need_location = bot->need_location_bot; return bot_data; } const ContactsManager::User *ContactsManager::get_user(UserId user_id) const { auto p = users_.find(user_id); if (p == users_.end()) { return nullptr; } else { return &p->second; } } ContactsManager::User *ContactsManager::get_user(UserId user_id) { auto p = users_.find(user_id); if (p == users_.end()) { return nullptr; } else { return &p->second; } } void ContactsManager::send_get_me_query(Td *td, Promise &&promise) { vector> users; users.push_back(make_tl_object()); td->create_handler(std::move(promise))->send(std::move(users)); } UserId ContactsManager::get_me(Promise &&promise) { auto my_id = get_my_id(); if (!have_user_force(my_id)) { send_get_me_query(td_, std::move(promise)); return UserId(); } promise.set_value(Unit()); return my_id; } bool ContactsManager::get_user(UserId user_id, int left_tries, Promise &&promise) { if (!user_id.is_valid()) { promise.set_error(Status::Error(6, "Invalid user id")); return false; } if (user_id == UserId(777000)) { get_user_force(user_id); // preload 777000 synchronously } // TODO support loading user from database and merging it with min-user in memory if (!have_min_user(user_id)) { // TODO UserLoader if (left_tries > 2 && G()->parameters().use_chat_info_db) { send_closure_later(actor_id(this), &ContactsManager::load_user_from_database, nullptr, user_id, std::move(promise)); return false; } auto input_user = get_input_user(user_id); if (left_tries == 1 || input_user == nullptr) { promise.set_error(Status::Error(6, "User not found")); return false; } vector> users; users.push_back(std::move(input_user)); td_->create_handler(std::move(promise))->send(std::move(users)); return false; } promise.set_value(Unit()); return true; } ContactsManager::User *ContactsManager::add_user(UserId user_id, const char *source) { CHECK(user_id.is_valid()); User *u = &users_[user_id]; if (u->debug_source == nullptr) { u->debug_source = source; } return u; } const ContactsManager::UserFull *ContactsManager::get_user_full(UserId user_id) const { auto p = users_full_.find(user_id); if (p == users_full_.end()) { return nullptr; } else { return &p->second; } } ContactsManager::UserFull *ContactsManager::get_user_full(UserId user_id) { auto p = users_full_.find(user_id); if (p == users_full_.end()) { return nullptr; } else { return &p->second; } } bool ContactsManager::get_user_full(UserId user_id, Promise &&promise) { auto user = get_user(user_id); if (user == nullptr) { promise.set_error(Status::Error(6, "User not found")); return false; } auto user_full = get_user_full(user_id); if (user_full == nullptr || !user_full->is_inited) { auto input_user = get_input_user(user_id); if (input_user == nullptr) { promise.set_error(Status::Error(6, "Can't get info about unaccessible user")); return false; } send_get_user_full_query(user_id, std::move(input_user), std::move(promise)); return false; } if (user_full->is_expired() || user_full->is_bot_info_expired(user->bot_info_version)) { auto input_user = get_input_user(user_id); CHECK(input_user != nullptr); if (td_->auth_manager_->is_bot()) { send_get_user_full_query(user_id, std::move(input_user), std::move(promise)); return false; } else { send_get_user_full_query(user_id, std::move(input_user), Auto()); } } promise.set_value(Unit()); return true; } void ContactsManager::send_get_user_full_query(UserId user_id, tl_object_ptr &&input_user, Promise &&promise) { auto &promises = get_user_full_queries_[user_id]; promises.push_back(std::move(promise)); if (promises.size() != 1) { // query has already been sent, just wait for the result return; } td_->create_handler()->send(user_id, std::move(input_user)); } void ContactsManager::on_get_user_full_success(UserId user_id) { auto it = get_user_full_queries_.find(user_id); CHECK(it != get_user_full_queries_.end()); CHECK(!it->second.empty()); auto promises = std::move(it->second); get_user_full_queries_.erase(it); for (auto &promise : promises) { promise.set_value(Unit()); } } void ContactsManager::on_get_user_full_fail(UserId user_id, Status &&error) { auto it = get_user_full_queries_.find(user_id); CHECK(it != get_user_full_queries_.end()); CHECK(!it->second.empty()); auto promises = std::move(it->second); get_user_full_queries_.erase(it); for (auto &promise : promises) { promise.set_error(error.clone()); } } std::pair> ContactsManager::get_user_profile_photos(UserId user_id, int32 offset, int32 limit, Promise &&promise) { std::pair> result; result.first = -1; if (offset < 0) { promise.set_error(Status::Error(3, "Parameter offset must be non-negative")); return result; } if (limit <= 0) { promise.set_error(Status::Error(3, "Parameter limit must be positive")); return result; } if (limit > MAX_GET_PROFILE_PHOTOS) { limit = MAX_GET_PROFILE_PHOTOS; } auto input_user = get_input_user(user_id); if (input_user == nullptr) { promise.set_error(Status::Error(6, "User not found")); return result; } auto user_full = &users_full_[user_id]; if (user_full->getting_photos_now) { promise.set_error(Status::Error(400, "Request for new profile photos has already been sent")); return result; } if (user_full->photo_count != -1) { // know photo count CHECK(user_full->photos_offset != -1); result.first = user_full->photo_count; if (offset >= user_full->photo_count) { // offset if too big promise.set_value(Unit()); return result; } if (limit > user_full->photo_count - offset) { limit = user_full->photo_count - offset; } int32 cache_begin = user_full->photos_offset; int32 cache_end = cache_begin + narrow_cast(user_full->photos.size()); if (cache_begin <= offset && offset + limit <= cache_end) { // answer query from cache for (int i = 0; i < limit; i++) { result.second.push_back(&user_full->photos[i + offset - cache_begin]); } promise.set_value(Unit()); return result; } if (cache_begin <= offset && offset < cache_end) { // adjust offset to the end of cache limit = offset + limit - cache_end; offset = cache_end; } } user_full->getting_photos_now = true; if (limit < MAX_GET_PROFILE_PHOTOS / 5) { limit = MAX_GET_PROFILE_PHOTOS / 5; // make limit reasonable } td_->create_handler(std::move(promise))->send(user_id, std::move(input_user), offset, limit, 0); return result; } void ContactsManager::reload_user_profile_photo(UserId user_id, int64 photo_id, Promise &&promise) { get_user_force(user_id); auto input_user = get_input_user(user_id); if (input_user == nullptr) { return promise.set_error(Status::Error(6, "User info not found")); } // 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(std::move(promise))->send(user_id, std::move(input_user), -1, 1, photo_id); } FileSourceId ContactsManager::get_user_profile_photo_file_source_id(UserId user_id, int64 photo_id) { 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; } bool ContactsManager::have_chat(ChatId chat_id) const { return chats_.count(chat_id) > 0; } const ContactsManager::Chat *ContactsManager::get_chat(ChatId chat_id) const { auto p = chats_.find(chat_id); if (p == chats_.end()) { return nullptr; } else { return &p->second; } } ContactsManager::Chat *ContactsManager::get_chat(ChatId chat_id) { auto p = chats_.find(chat_id); if (p == chats_.end()) { return nullptr; } else { return &p->second; } } ContactsManager::Chat *ContactsManager::add_chat(ChatId chat_id) { auto c = get_chat(chat_id); if (c != nullptr) { return c; } CHECK(chat_id.is_valid()); c = &chats_[chat_id]; auto it = chat_photo_file_source_ids_.find(chat_id); if (it != chat_photo_file_source_ids_.end()) { VLOG(file_references) << "Move " << it->second << " inside of " << chat_id; c->photo_source_id = it->second; chat_photo_file_source_ids_.erase(it); } return c; } bool ContactsManager::get_chat(ChatId chat_id, int left_tries, Promise &&promise) { if (!chat_id.is_valid()) { promise.set_error(Status::Error(6, "Invalid basic group id")); return false; } if (!have_chat(chat_id)) { if (left_tries > 2 && G()->parameters().use_chat_info_db) { send_closure_later(actor_id(this), &ContactsManager::load_chat_from_database, nullptr, chat_id, std::move(promise)); return false; } if (left_tries > 1) { td_->create_handler(std::move(promise))->send(vector{chat_id.get()}); return false; } promise.set_error(Status::Error(6, "Group not found")); return false; } promise.set_value(Unit()); return true; } void ContactsManager::reload_chat(ChatId chat_id, Promise &&promise) { if (!chat_id.is_valid()) { return promise.set_error(Status::Error(6, "Invalid basic group id")); } // this request will be needed only to download the chat photo, // so there is no reason to combine different requests into one request td_->create_handler(std::move(promise))->send(vector{chat_id.get()}); } const ContactsManager::ChatFull *ContactsManager::get_chat_full(ChatId chat_id) const { auto p = chats_full_.find(chat_id); if (p == chats_full_.end()) { return nullptr; } else { return &p->second; } } ContactsManager::ChatFull *ContactsManager::get_chat_full(ChatId chat_id) { auto p = chats_full_.find(chat_id); if (p == chats_full_.end()) { return nullptr; } else { return &p->second; } } bool ContactsManager::is_chat_full_outdated(ChatFull *chat_full, Chat *c, ChatId chat_id) { CHECK(c != nullptr); CHECK(chat_full != nullptr); if (chat_full->version != c->version) { LOG(INFO) << "Have outdated ChatFull " << chat_id << " with current version " << (chat_full ? chat_full->version : -123456789) << " and chat version " << c->version; return true; } for (auto &participant : chat_full->participants) { auto user = get_user(participant.user_id); if (user != nullptr && user->bot_info_version != -1) { auto user_full = get_user_full(participant.user_id); if (user_full == nullptr || user_full->is_bot_info_expired(user->bot_info_version)) { LOG(INFO) << "Have outdated botInfo for " << participant.user_id << " with version " << (user_full && user_full->bot_info ? user_full->bot_info->version : -123456789) << ", but current version is " << user->bot_info_version; return true; } } } return false; } bool ContactsManager::get_chat_full(ChatId chat_id, Promise &&promise) { auto chat = get_chat(chat_id); if (chat == nullptr) { promise.set_error(Status::Error(6, "Group not found")); return false; } auto chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { LOG(INFO) << "Full " << chat_id << " not found"; send_get_chat_full_query(chat_id, std::move(promise)); return false; } if (is_chat_full_outdated(chat_full, chat, chat_id)) { LOG(INFO) << "Have outdated full " << chat_id; if (td_->auth_manager_->is_bot()) { send_get_chat_full_query(chat_id, std::move(promise)); return false; } else { send_get_chat_full_query(chat_id, Auto()); } } promise.set_value(Unit()); return true; } void ContactsManager::send_get_chat_full_query(ChatId chat_id, Promise &&promise) { auto &promises = get_chat_full_queries_[chat_id]; promises.push_back(std::move(promise)); if (promises.size() != 1) { // query has already been sent, just wait for the result return; } td_->create_handler()->send(chat_id); } void ContactsManager::on_get_chat_full_success(ChatId chat_id) { auto it = get_chat_full_queries_.find(chat_id); CHECK(it != get_chat_full_queries_.end()); CHECK(!it->second.empty()); auto promises = std::move(it->second); get_chat_full_queries_.erase(it); for (auto &promise : promises) { promise.set_value(Unit()); } } void ContactsManager::on_get_chat_full_fail(ChatId chat_id, Status &&error) { auto it = get_chat_full_queries_.find(chat_id); CHECK(it != get_chat_full_queries_.end()); CHECK(!it->second.empty()); auto promises = std::move(it->second); get_chat_full_queries_.erase(it); for (auto &promise : promises) { promise.set_error(error.clone()); } } bool ContactsManager::get_chat_is_active(ChatId chat_id) const { auto c = get_chat(chat_id); if (c == nullptr) { return false; } return c->is_active; } DialogParticipantStatus ContactsManager::get_chat_status(ChatId chat_id) const { auto c = get_chat(chat_id); if (c == nullptr) { return DialogParticipantStatus::Banned(0); } return get_chat_status(c); } DialogParticipantStatus ContactsManager::get_chat_status(const Chat *c) { if (c->kicked || !c->is_active) { return DialogParticipantStatus::Banned(0); } if (c->left) { return DialogParticipantStatus::Left(); } if (c->is_creator) { return DialogParticipantStatus::Creator(true); } if (c->can_edit) { return DialogParticipantStatus::GroupAdministrator(false); } return DialogParticipantStatus::Member(); } bool ContactsManager::is_appointed_chat_administrator(ChatId chat_id) const { auto c = get_chat(chat_id); if (c == nullptr) { return false; } if (c->everyone_is_administrator) { return c->is_creator; } else { return c->can_edit; } } FileSourceId ContactsManager::get_chat_photo_file_source_id(ChatId chat_id) { auto c = get_chat(chat_id); auto &source_id = c == nullptr ? chat_photo_file_source_ids_[chat_id] : c->photo_source_id; if (!source_id.is_valid()) { source_id = td_->file_reference_manager_->create_chat_photo_file_source(chat_id); } return source_id; } ChannelType ContactsManager::get_channel_type(ChannelId channel_id) const { auto c = get_channel(channel_id); if (c == nullptr) { return ChannelType::Unknown; } return get_channel_type(c); } ChannelType ContactsManager::get_channel_type(const Channel *c) { if (c->is_megagroup) { return ChannelType::Megagroup; } return ChannelType::Broadcast; } int32 ContactsManager::get_channel_date(ChannelId channel_id) const { auto c = get_channel(channel_id); if (c == nullptr) { return 0; } return c->date; } DialogParticipantStatus ContactsManager::get_channel_status(ChannelId channel_id) const { auto c = get_channel(channel_id); if (c == nullptr) { return DialogParticipantStatus::Banned(0); } return get_channel_status(c); } DialogParticipantStatus ContactsManager::get_channel_status(const Channel *c) { c->status.update_restrictions(); return c->status; } bool ContactsManager::get_channel_sign_messages(ChannelId channel_id) const { auto c = get_channel(channel_id); if (c == nullptr) { return false; } return get_channel_sign_messages(c); } bool ContactsManager::get_channel_sign_messages(const Channel *c) { return c->sign_messages; } FileSourceId ContactsManager::get_channel_photo_file_source_id(ChannelId channel_id) { auto c = get_channel(channel_id); auto &source_id = c == nullptr ? channel_photo_file_source_ids_[channel_id] : c->photo_source_id; if (!source_id.is_valid()) { source_id = td_->file_reference_manager_->create_channel_photo_file_source(channel_id); } return source_id; } bool ContactsManager::have_channel(ChannelId channel_id) const { return channels_.count(channel_id) > 0; } bool ContactsManager::have_min_channel(ChannelId channel_id) const { return min_channels_.count(channel_id) > 0; } const ContactsManager::Channel *ContactsManager::get_channel(ChannelId channel_id) const { auto p = channels_.find(channel_id); if (p == channels_.end()) { return nullptr; } else { return &p->second; } } ContactsManager::Channel *ContactsManager::get_channel(ChannelId channel_id) { auto p = channels_.find(channel_id); if (p == channels_.end()) { return nullptr; } else { return &p->second; } } ContactsManager::Channel *ContactsManager::add_channel(ChannelId channel_id, const char *source) { auto c = get_channel(channel_id); if (c != nullptr) { return c; } CHECK(channel_id.is_valid()); c = &channels_[channel_id]; auto it = channel_photo_file_source_ids_.find(channel_id); if (it != channel_photo_file_source_ids_.end()) { VLOG(file_references) << "Move " << it->second << " inside of " << channel_id; c->photo_source_id = it->second; channel_photo_file_source_ids_.erase(it); } c->debug_source = source; return c; } bool ContactsManager::get_channel(ChannelId channel_id, int left_tries, Promise &&promise) { if (!channel_id.is_valid()) { promise.set_error(Status::Error(6, "Invalid supergroup id")); return false; } if (!have_channel(channel_id)) { if (left_tries > 2 && G()->parameters().use_chat_info_db) { send_closure_later(actor_id(this), &ContactsManager::load_channel_from_database, nullptr, channel_id, std::move(promise)); return false; } if (left_tries > 1 && td_->auth_manager_->is_bot()) { td_->create_handler(std::move(promise))->send(get_input_channel(channel_id)); return false; } promise.set_error(Status::Error(6, "Supergroup not found")); return false; } promise.set_value(Unit()); return true; } void ContactsManager::reload_channel(ChannelId channel_id, Promise &&promise) { if (!channel_id.is_valid()) { return promise.set_error(Status::Error(6, "Invalid supergroup id")); } have_channel_force(channel_id); auto input_channel = get_input_channel(channel_id); if (input_channel == nullptr) { return promise.set_error(Status::Error(6, "Supergroup info not found")); } // this request will be needed only to download the channel photo, // so there is no reason to combine different requests into one request td_->create_handler(std::move(promise))->send(std::move(input_channel)); } const ContactsManager::ChannelFull *ContactsManager::get_channel_full(ChannelId channel_id) const { auto p = channels_full_.find(channel_id); if (p == channels_full_.end()) { return nullptr; } else { return &p->second; } } ContactsManager::ChannelFull *ContactsManager::get_channel_full(ChannelId channel_id) { auto p = channels_full_.find(channel_id); if (p == channels_full_.end()) { return nullptr; } auto channel_full = &p->second; if (channel_full->is_expired()) { auto input_channel = get_input_channel(channel_id); CHECK(input_channel != nullptr); send_get_channel_full_query(channel_id, std::move(input_channel), Auto()); } return channel_full; } bool ContactsManager::get_channel_full(ChannelId channel_id, Promise &&promise) { auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr) { auto input_channel = get_input_channel(channel_id); if (input_channel == nullptr) { promise.set_error(Status::Error(6, "Supergroup not found")); return false; } send_get_channel_full_query(channel_id, std::move(input_channel), std::move(promise)); return false; } if (channel_full->is_expired()) { if (td_->auth_manager_->is_bot()) { auto input_channel = get_input_channel(channel_id); CHECK(input_channel != nullptr); send_get_channel_full_query(channel_id, std::move(input_channel), std::move(promise)); return false; } else { // request has already been sent in get_channel_full // send_get_channel_full_query(channel_id, std::move(input_channel), Auto()); } } promise.set_value(Unit()); return true; } void ContactsManager::send_get_channel_full_query(ChannelId channel_id, tl_object_ptr &&input_channel, Promise &&promise) { auto &promises = get_channel_full_queries_[channel_id]; promises.push_back(std::move(promise)); if (promises.size() != 1) { // query has already been sent, just wait for the result return; } td_->create_handler()->send(channel_id, std::move(input_channel)); } void ContactsManager::on_get_channel_full_success(ChannelId channel_id) { auto it = get_channel_full_queries_.find(channel_id); CHECK(it != get_channel_full_queries_.end()); CHECK(!it->second.empty()); auto promises = std::move(it->second); get_channel_full_queries_.erase(it); for (auto &promise : promises) { promise.set_value(Unit()); } } void ContactsManager::on_get_channel_full_fail(ChannelId channel_id, Status &&error) { auto it = get_channel_full_queries_.find(channel_id); CHECK(it != get_channel_full_queries_.end()); CHECK(!it->second.empty()); auto promises = std::move(it->second); get_channel_full_queries_.erase(it); for (auto &promise : promises) { promise.set_error(error.clone()); } } bool ContactsManager::have_secret_chat(SecretChatId secret_chat_id) const { return secret_chats_.count(secret_chat_id) > 0; } ContactsManager::SecretChat *ContactsManager::add_secret_chat(SecretChatId secret_chat_id) { CHECK(secret_chat_id.is_valid()); return &secret_chats_[secret_chat_id]; } const ContactsManager::SecretChat *ContactsManager::get_secret_chat(SecretChatId secret_chat_id) const { auto it = secret_chats_.find(secret_chat_id); if (it == secret_chats_.end()) { return nullptr; } return &it->second; } ContactsManager::SecretChat *ContactsManager::get_secret_chat(SecretChatId secret_chat_id) { auto it = secret_chats_.find(secret_chat_id); if (it == secret_chats_.end()) { return nullptr; } return &it->second; } bool ContactsManager::get_secret_chat(SecretChatId secret_chat_id, bool force, Promise &&promise) { if (!secret_chat_id.is_valid()) { promise.set_error(Status::Error(6, "Invalid secret chat id")); return false; } if (!have_secret_chat(secret_chat_id)) { if (!force && G()->parameters().use_chat_info_db) { send_closure_later(actor_id(this), &ContactsManager::load_secret_chat_from_database, nullptr, secret_chat_id, std::move(promise)); return false; } promise.set_error(Status::Error(6, "Secret chat not found")); return false; } promise.set_value(Unit()); return true; } void ContactsManager::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) { 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->is_changed = 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]; old_secret_chat_ids.erase(std::remove(old_secret_chat_ids.begin(), old_secret_chat_ids.end(), secret_chat_id), old_secret_chat_ids.end()); } secret_chat->user_id = user_id; secret_chats_with_user_[secret_chat->user_id].push_back(secret_chat_id); secret_chat->need_send_update = true; } if (state != SecretChatState::Unknown && state != secret_chat->state) { secret_chat->state = state; secret_chat->need_send_update = true; secret_chat->is_state_changed = true; } if (is_outbound != secret_chat->is_outbound) { secret_chat->is_outbound = is_outbound; secret_chat->need_send_update = true; } if (ttl != -1 && ttl != secret_chat->ttl) { secret_chat->ttl = ttl; secret_chat->need_send_update = true; } if (date != 0 && date != secret_chat->date) { secret_chat->date = date; secret_chat->is_changed = true; } if (!key_hash.empty() && key_hash != secret_chat->key_hash) { secret_chat->key_hash = std::move(key_hash); secret_chat->need_send_update = true; } if (layer != 0 && layer != secret_chat->layer) { secret_chat->layer = layer; secret_chat->need_send_update = true; } update_secret_chat(secret_chat, secret_chat_id); } std::pair> ContactsManager::search_among_users(const vector &user_ids, const string &query, int32 limit) { Hints hints; // TODO cache Hints UserId my_user_id = get_my_id(); for (auto user_id : user_ids) { auto u = get_user(user_id); if (u == nullptr) { continue; } hints.add(user_id.get(), u->first_name + " " + u->last_name + " " + u->username); auto was_online = u->was_online; if (user_id == my_user_id && my_was_online_local_ != 0) { was_online = my_was_online_local_; } hints.set_rating(user_id.get(), -was_online); } auto result = hints.search(query, limit, true); return {narrow_cast(result.first), transform(result.second, [](int64 key) { return UserId(narrow_cast(key)); })}; } DialogParticipant ContactsManager::get_chat_participant(ChatId chat_id, UserId user_id, bool force, Promise &&promise) { LOG(INFO) << "Trying to get " << user_id << " as member of " << chat_id; if (force) { promise.set_value(Unit()); } else if (!get_chat_full(chat_id, std::move(promise))) { return DialogParticipant(); } // promise is already set auto result = get_chat_participant(chat_id, user_id); if (result == nullptr) { return {user_id, UserId(), 0, DialogParticipantStatus::Left()}; } return *result; } std::pair> ContactsManager::search_chat_participants(ChatId chat_id, const string &query, int32 limit, DialogParticipantsFilter filter, bool force, Promise &&promise) { if (limit < 0) { promise.set_error(Status::Error(3, "Parameter limit must be non-negative")); return {}; } if (force) { promise.set_value(Unit()); } else if (!get_chat_full(chat_id, std::move(promise))) { return {}; } // promise is already set auto chat_full = get_chat_full(chat_id); if (chat_full == nullptr) { return {}; } auto is_dialog_participant_suitable = [this](const DialogParticipant &participant, DialogParticipantsFilter filter) { switch (filter) { case DialogParticipantsFilter::Administrators: return participant.status.is_administrator(); case DialogParticipantsFilter::Members: return participant.status.is_member(); // should be always true case DialogParticipantsFilter::Restricted: return participant.status.is_restricted(); // should be always false case DialogParticipantsFilter::Banned: return participant.status.is_banned(); // should be always false case DialogParticipantsFilter::Bots: return is_user_bot(participant.user_id); default: UNREACHABLE(); return false; } }; vector user_ids; for (auto &participant : chat_full->participants) { if (is_dialog_participant_suitable(participant, filter)) { user_ids.push_back(participant.user_id); } } int32 total_count; std::tie(total_count, user_ids) = search_among_users(user_ids, query, limit); return {total_count, transform(user_ids, [&](UserId user_id) { return *get_chat_participant(chat_full, user_id); })}; } DialogParticipant ContactsManager::get_channel_participant(ChannelId channel_id, UserId user_id, int64 &random_id, bool force, Promise &&promise) { LOG(INFO) << "Trying to get " << user_id << " as member of " << channel_id; if (random_id != 0) { // request has already been sent before auto it = received_channel_participant_.find(random_id); CHECK(it != received_channel_participant_.end()); auto result = std::move(it->second); received_channel_participant_.erase(it); promise.set_value(Unit()); return result; } auto input_user = get_input_user(user_id); if (input_user == nullptr) { promise.set_error(Status::Error(6, "User not found")); return DialogParticipant(); } if (!td_->auth_manager_->is_bot() && is_user_bot(user_id)) { // get BotInfo through UserFull auto user = get_user(user_id); auto user_full = get_user_full(user_id); if (user_full == nullptr || user_full->is_bot_info_expired(user->bot_info_version)) { if (force) { LOG(ERROR) << "Can't find cached UserFull"; } else { send_get_user_full_query(user_id, std::move(input_user), std::move(promise)); return DialogParticipant(); } } } do { random_id = Random::secure_int64(); } while (random_id == 0 || received_channel_participant_.find(random_id) != received_channel_participant_.end()); received_channel_participant_[random_id]; // reserve place for result LOG(DEBUG) << "Get info about " << user_id << " membership in the " << channel_id; auto on_result_promise = PromiseCreator::lambda( [this, random_id, promise = std::move(promise)](Result r_dialog_participant) mutable { // ResultHandlers are cleared before managers, so it is safe to capture this auto it = received_channel_participant_.find(random_id); CHECK(it != received_channel_participant_.end()); if (r_dialog_participant.is_error()) { received_channel_participant_.erase(it); promise.set_error(r_dialog_participant.move_as_error()); } else { it->second = r_dialog_participant.move_as_ok(); promise.set_value(Unit()); } }); td_->create_handler(std::move(on_result_promise)) ->send(channel_id, user_id, std::move(input_user)); return DialogParticipant(); } std::pair> ContactsManager::get_channel_participants( ChannelId channel_id, const tl_object_ptr &filter, const string &additional_query, int32 offset, int32 limit, int32 additional_limit, int64 &random_id, bool force, Promise &&promise) { if (random_id != 0) { // request has already been sent before auto it = received_channel_participants_.find(random_id); CHECK(it != received_channel_participants_.end()); auto result = std::move(it->second); received_channel_participants_.erase(it); promise.set_value(Unit()); if (additional_query.empty()) { return result; } auto user_ids = transform(result.second, [](const auto &participant) { return participant.user_id; }); std::pair> result_user_ids = search_among_users(user_ids, additional_query, additional_limit); result.first = result_user_ids.first; std::unordered_set result_user_ids_set(result_user_ids.second.begin(), result_user_ids.second.end()); auto all_participants = std::move(result.second); result.second.clear(); for (auto &participant : all_participants) { if (result_user_ids_set.count(participant.user_id)) { result.second.push_back(std::move(participant)); result_user_ids_set.erase(participant.user_id); } } return result; } std::pair> result; if (limit <= 0) { promise.set_error(Status::Error(3, "Parameter limit must be positive")); return result; } if (offset < 0) { promise.set_error(Status::Error(3, "Parameter offset must be non-negative")); return result; } auto channel_full = get_channel_full(channel_id); if (channel_full == nullptr || (!force && channel_full->is_expired())) { if (force) { LOG(ERROR) << "Can't find cached ChannelFull"; } else { auto input_channel = get_input_channel(channel_id); if (input_channel == nullptr) { promise.set_error(Status::Error(6, "Supergroup not found")); } else { send_get_channel_full_query(channel_id, std::move(input_channel), std::move(promise)); } return result; } } if (channel_full != nullptr && !channel_full->is_expired() && !channel_full->can_get_participants) { promise.set_error(Status::Error(3, "Supergroup members are unavailable")); return result; } do { random_id = Random::secure_int64(); } while (random_id == 0 || received_channel_participants_.find(random_id) != received_channel_participants_.end()); received_channel_participants_[random_id]; // reserve place for result LOG(DEBUG) << "Get members of the " << channel_id << " with offset = " << offset << " and limit = " << limit; td_->create_handler(std::move(promise)) ->send(channel_id, ChannelParticipantsFilter(filter), offset, limit, random_id); return result; } vector ContactsManager::get_dialog_administrators(DialogId dialog_id, int left_tries, Promise &&promise) { auto it = dialog_administrators_.find(dialog_id); if (it != dialog_administrators_.end()) { promise.set_value(Unit()); if (left_tries >= 2) { auto hash = get_vector_hash(transform(it->second, [](UserId user_id) { return static_cast(user_id.get()); })); reload_dialog_administrators(dialog_id, hash, Auto()); // update administrators cache } return it->second; } if (left_tries >= 3) { load_dialog_administrators(dialog_id, std::move(promise)); return {}; } if (left_tries >= 2) { reload_dialog_administrators(dialog_id, 0, std::move(promise)); return {}; } LOG(ERROR) << "Have no known administrators in " << dialog_id; promise.set_value(Unit()); return {}; } string ContactsManager::get_dialog_administrators_database_key(DialogId dialog_id) { return PSTRING() << "admin" << (-dialog_id.get()); } void ContactsManager::load_dialog_administrators(DialogId dialog_id, Promise &&promise) { if (G()->parameters().use_chat_info_db) { LOG(INFO) << "Load administrators of " << dialog_id << " from database"; G()->td_db()->get_sqlite_pmc()->get( get_dialog_administrators_database_key(dialog_id), PromiseCreator::lambda([dialog_id, promise = std::move(promise)](string value) mutable { send_closure(G()->contacts_manager(), &ContactsManager::on_load_dialog_administrators_from_database, dialog_id, std::move(value), std::move(promise)); })); } else { promise.set_value(Unit()); } } void ContactsManager::on_load_dialog_administrators_from_database(DialogId dialog_id, string value, Promise &&promise) { if (value.empty()) { promise.set_value(Unit()); return; } vector user_ids; log_event_parse(user_ids, value).ensure(); LOG(INFO) << "Successfully loaded " << user_ids.size() << " administrators in " << dialog_id << " from database"; MultiPromiseActorSafe load_users_multipromise{"LoadUsersMultiPromiseActor"}; load_users_multipromise.add_promise( PromiseCreator::lambda([dialog_id, user_ids, promise = std::move(promise)](Result<> result) mutable { send_closure(G()->contacts_manager(), &ContactsManager::on_load_administrator_users_finished, dialog_id, std::move(user_ids), std::move(result), std::move(promise)); })); auto lock_promise = load_users_multipromise.get_promise(); for (auto user_id : user_ids) { get_user(user_id, 3, load_users_multipromise.get_promise()); } lock_promise.set_value(Unit()); } void ContactsManager::on_load_administrator_users_finished(DialogId dialog_id, vector user_ids, Result<> result, Promise promise) { if (result.is_ok()) { dialog_administrators_.emplace(dialog_id, std::move(user_ids)); } promise.set_value(Unit()); } void ContactsManager::on_update_dialog_administrators(DialogId dialog_id, vector administrator_user_ids, bool have_access) { LOG(INFO) << "Update administrators in " << dialog_id << " to " << format::as_array(administrator_user_ids); if (have_access) { std::sort(administrator_user_ids.begin(), administrator_user_ids.end(), [](UserId lhs, UserId rhs) { return lhs.get() < rhs.get(); }); auto it = dialog_administrators_.find(dialog_id); if (it != dialog_administrators_.end()) { if (it->second == administrator_user_ids) { return; } it->second = std::move(administrator_user_ids); } else { it = dialog_administrators_.emplace(dialog_id, std::move(administrator_user_ids)).first; } if (G()->parameters().use_chat_info_db) { LOG(INFO) << "Save administrators of " << dialog_id << " to database"; G()->td_db()->get_sqlite_pmc()->set(get_dialog_administrators_database_key(dialog_id), log_event_store(it->second).as_slice().str(), Auto()); } } else { dialog_administrators_.erase(dialog_id); if (G()->parameters().use_chat_info_db) { G()->td_db()->get_sqlite_pmc()->erase(get_dialog_administrators_database_key(dialog_id), Auto()); } } } void ContactsManager::reload_dialog_administrators(DialogId dialog_id, int32 hash, Promise &&promise) { switch (dialog_id.get_type()) { case DialogType::Chat: get_chat_full(dialog_id.get_chat_id(), std::move(promise)); break; case DialogType::Channel: td_->create_handler(std::move(promise))->send(dialog_id.get_channel_id(), hash); break; default: UNREACHABLE(); } } void ContactsManager::on_chat_update(telegram_api::chatEmpty &chat, const char *source) { ChatId chat_id(chat.id_); if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id << " from " << source; return; } if (!have_chat(chat_id)) { LOG(ERROR) << "Have no information about " << chat_id << " but received chatEmpty from " << source; } } void ContactsManager::on_chat_update(telegram_api::chat &chat, const char *source) { auto debug_str = PSTRING() << " from " << source << " in " << oneline(to_string(chat)); ChatId chat_id(chat.id_); if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id << debug_str; return; } bool has_left = 0 != (chat.flags_ & CHAT_FLAG_USER_HAS_LEFT); bool was_kicked = 0 != (chat.flags_ & CHAT_FLAG_USER_WAS_KICKED); if (was_kicked) { LOG_IF(ERROR, has_left) << "Kicked and left" << debug_str; // only one of the flags can be set has_left = true; } bool is_creator = 0 != (chat.flags_ & CHAT_FLAG_USER_IS_CREATOR); bool is_administrator = 0 != (chat.flags_ & CHAT_FLAG_IS_ADMINISTRATOR); bool everyone_is_administrator = 0 == (chat.flags_ & CHAT_FLAG_ADMINISTRATORS_ENABLED); bool is_active = 0 == (chat.flags_ & CHAT_FLAG_IS_DEACTIVATED); ChannelId migrated_to_channel_id; if (chat.flags_ & CHAT_FLAG_WAS_MIGRATED) { switch (chat.migrated_to_->get_id()) { case telegram_api::inputChannelEmpty::ID: { LOG(ERROR) << "Receive empty upgraded to supergroup for " << chat_id << debug_str; break; } case telegram_api::inputChannel::ID: { auto input_channel = move_tl_object_as(chat.migrated_to_); migrated_to_channel_id = ChannelId(input_channel->channel_id_); if (!have_channel(migrated_to_channel_id)) { if (!migrated_to_channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << migrated_to_channel_id << debug_str; } else { // temporarily create the channel Channel *c = add_channel(migrated_to_channel_id, "on_chat_update"); c->access_hash = input_channel->access_hash_; c->title = chat.title_; c->status = DialogParticipantStatus::Left(); c->is_megagroup = true; // we definitely need to call update_channel, because client should know about every added channel update_channel(c, migrated_to_channel_id); // get info about the channel td_->create_handler(Promise<>())->send(std::move(input_channel)); } } break; } default: UNREACHABLE(); } } Chat *c = add_chat(chat_id); on_update_chat_title(c, chat_id, std::move(chat.title_)); if (!has_left) { on_update_chat_participant_count(c, chat_id, chat.participants_count_, chat.version_, debug_str); } if (c->date != chat.date_) { LOG_IF(ERROR, c->date != 0) << "Chat creation date has changed from " << c->date << " to " << chat.date_ << debug_str; c->date = chat.date_; c->is_changed = true; } on_update_chat_left(c, chat_id, has_left, was_kicked); on_update_chat_rights(c, chat_id, is_creator, is_administrator, everyone_is_administrator); on_update_chat_photo(c, chat_id, std::move(chat.photo_)); on_update_chat_active(c, chat_id, is_active); on_update_chat_migrated_to_channel_id(c, chat_id, migrated_to_channel_id); LOG_IF(INFO, !is_active && !migrated_to_channel_id.is_valid()) << chat_id << " is deactivated in " << debug_str; update_chat(c, chat_id); } void ContactsManager::on_chat_update(telegram_api::chatForbidden &chat, const char *source) { ChatId chat_id(chat.id_); if (!chat_id.is_valid()) { LOG(ERROR) << "Receive invalid " << chat_id << " from " << source; return; } Chat *c = add_chat(chat_id); bool is_uninited = c->left == false && c->kicked == true; on_update_chat_title(c, chat_id, std::move(chat.title_)); // chat participant count will be updated in on_update_chat_left // leave rights as is on_update_chat_photo(c, chat_id, nullptr); if (c->date != 0) { c->date = 0; // removed in 38-th layer c->is_changed = true; } on_update_chat_left(c, chat_id, true, true); if (is_uninited) { on_update_chat_active(c, chat_id, true); on_update_chat_migrated_to_channel_id(c, chat_id, ChannelId()); } else { // leave active and migrated to as is } update_chat(c, chat_id); } void ContactsManager::on_chat_update(telegram_api::channel &channel, const char *source) { ChannelId channel_id(channel.id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id << " from " << source << ": " << to_string(channel); return; } if (channel.flags_ == 0 && channel.access_hash_ == 0 && channel.title_.empty()) { Channel *c = get_channel_force(channel_id); LOG(ERROR) << "Receive empty " << to_string(channel) << " from " << source << ", have " << to_string(get_supergroup_object(channel_id, c)); if (c == nullptr) { min_channels_.insert(channel_id); } return; } bool is_min = (channel.flags_ & CHANNEL_FLAG_IS_MIN) != 0; bool has_access_hash = (channel.flags_ & CHANNEL_FLAG_HAS_ACCESS_HASH) != 0; auto access_hash = has_access_hash ? channel.access_hash_ : 0; bool anyone_can_invite = (channel.flags_ & CHANNEL_FLAG_ANYONE_CAN_INVITE) != 0; bool sign_messages = (channel.flags_ & CHANNEL_FLAG_SIGN_MESSAGES) != 0; bool is_megagroup = (channel.flags_ & CHANNEL_FLAG_IS_MEGAGROUP) != 0; bool is_verified = (channel.flags_ & CHANNEL_FLAG_IS_VERIFIED) != 0; string restriction_reason = std::move(channel.restriction_reason_); int32 participant_count = (channel.flags_ & CHANNEL_FLAG_HAS_PARTICIPANT_COUNT) != 0 ? channel.participants_count_ : 0; { bool is_broadcast = (channel.flags_ & CHANNEL_FLAG_IS_BROADCAST) != 0; LOG_IF(ERROR, is_broadcast == is_megagroup) << "Receive wrong channel flag is_broadcast == is_megagroup == " << is_megagroup << " from " << source << ": " << oneline(to_string(channel)); } if (!is_megagroup && anyone_can_invite) { LOG(ERROR) << "Anyone can invite new members to the " << channel_id << " from " << source; anyone_can_invite = false; } if (is_megagroup) { LOG_IF(ERROR, sign_messages) << "Need to sign messages in the supergroup " << channel_id << " from " << source; sign_messages = true; } DialogParticipantStatus status = [&]() { bool has_left = (channel.flags_ & CHANNEL_FLAG_USER_HAS_LEFT) != 0; bool is_creator = (channel.flags_ & CHANNEL_FLAG_USER_IS_CREATOR) != 0; if (is_creator) { return DialogParticipantStatus::Creator(!has_left); } else if (channel.admin_rights_ != nullptr) { return get_dialog_participant_status(false, std::move(channel.admin_rights_)); } else if (channel.banned_rights_ != nullptr) { return get_dialog_participant_status(!has_left, std::move(channel.banned_rights_)); } else if (has_left) { return DialogParticipantStatus::Left(); } else { return DialogParticipantStatus::Member(); } }(); if (is_min) { // TODO there can be better support for min channels Channel *c = get_channel_force(channel_id); if (c != nullptr) { LOG(DEBUG) << "Receive known min " << channel_id; on_update_channel_title(c, channel_id, std::move(channel.title_)); on_update_channel_username(c, channel_id, std::move(channel.username_)); on_update_channel_photo(c, channel_id, std::move(channel.photo_)); if (c->anyone_can_invite != anyone_can_invite || c->is_megagroup != is_megagroup || c->is_verified != is_verified) { c->anyone_can_invite = anyone_can_invite; c->is_megagroup = is_megagroup; c->is_verified = is_verified; c->need_send_update = true; invalidate_channel_full(channel_id, false); } update_channel(c, channel_id); } else { min_channels_.insert(channel_id); } return; } if (!has_access_hash) { LOG(ERROR) << "Receive non-min " << channel_id << " without access_hash from " << source; return; } Channel *c = add_channel(channel_id, "on_channel"); if (c->status.is_banned()) { // possibly uninited channel min_channels_.erase(channel_id); } if (c->access_hash != access_hash) { c->access_hash = access_hash; c->is_changed = true; } on_update_channel_title(c, channel_id, std::move(channel.title_)); if (c->date != channel.date_) { c->date = channel.date_; c->need_send_update = true; } on_update_channel_photo(c, channel_id, std::move(channel.photo_)); on_update_channel_status(c, channel_id, std::move(status)); on_update_channel_username(c, channel_id, std::move(channel.username_)); // uses status, must be called after if (participant_count != 0 && participant_count != c->participant_count) { c->participant_count = participant_count; c->need_send_update = true; } if (c->anyone_can_invite != anyone_can_invite || c->sign_messages != sign_messages || c->is_megagroup != is_megagroup || c->is_verified != is_verified || c->restriction_reason != restriction_reason) { c->anyone_can_invite = anyone_can_invite; c->sign_messages = sign_messages; c->is_megagroup = is_megagroup; c->is_verified = is_verified; c->restriction_reason = std::move(restriction_reason); c->need_send_update = true; invalidate_channel_full(channel_id, false); } update_channel(c, channel_id); } void ContactsManager::on_chat_update(telegram_api::channelForbidden &channel, const char *source) { ChannelId channel_id(channel.id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id << " from " << source << ": " << to_string(channel); return; } if (channel.flags_ == 0 && channel.access_hash_ == 0 && channel.title_.empty()) { Channel *c = get_channel_force(channel_id); LOG(ERROR) << "Receive empty " << to_string(channel) << " from " << source << ", have " << to_string(get_supergroup_object(channel_id, c)); if (c == nullptr) { min_channels_.insert(channel_id); } return; } Channel *c = add_channel(channel_id, "on_channel_forbidden"); if (c->status.is_banned()) { // possibly uninited channel min_channels_.erase(channel_id); } if (c->access_hash != channel.access_hash_) { c->access_hash = channel.access_hash_; c->is_changed = true; } on_update_channel_title(c, channel_id, std::move(channel.title_)); on_update_channel_photo(c, channel_id, nullptr); if (c->date != 0) { c->date = 0; c->need_send_update = true; } int32 unban_date = (channel.flags_ & CHANNEL_FLAG_HAS_UNBAN_DATE) != 0 ? channel.until_date_ : 0; on_update_channel_status(c, channel_id, DialogParticipantStatus::Banned(unban_date)); on_update_channel_username(c, channel_id, ""); // don't know if channel username is empty, but update it anyway bool anyone_can_invite = false; bool sign_messages = false; bool is_megagroup = (channel.flags_ & CHANNEL_FLAG_IS_MEGAGROUP) != 0; bool is_verified = false; string restriction_reason; { bool is_broadcast = (channel.flags_ & CHANNEL_FLAG_IS_BROADCAST) != 0; LOG_IF(ERROR, is_broadcast == is_megagroup) << "Receive wrong channel flag is_broadcast == is_megagroup == " << is_megagroup << " from " << source << ": " << oneline(to_string(channel)); } if (is_megagroup) { sign_messages = true; } if (c->participant_count != 0) { c->participant_count = 0; c->need_send_update = true; } if (c->anyone_can_invite != anyone_can_invite || c->sign_messages != sign_messages || c->is_megagroup != is_megagroup || c->is_verified != is_verified || c->restriction_reason != restriction_reason) { c->anyone_can_invite = anyone_can_invite; c->sign_messages = sign_messages; c->is_megagroup = is_megagroup; c->is_verified = is_verified; c->restriction_reason = std::move(restriction_reason); c->need_send_update = true; invalidate_channel_full(channel_id, false); } update_channel(c, channel_id); } void ContactsManager::on_upload_profile_photo(FileId file_id, tl_object_ptr input_file) { LOG(INFO) << "File " << file_id << " has been uploaded"; auto it = uploaded_profile_photos_.find(file_id); CHECK(it != uploaded_profile_photos_.end()); auto promise = std::move(it->second); uploaded_profile_photos_.erase(it); FileView file_view = td_->file_manager_->get_file_view(file_id); if (file_view.has_remote_location()) { if (file_view.remote_location().is_web()) { // TODO reupload promise.set_error(Status::Error(400, "Can't use web photo as profile photo")); return; } td_->create_handler(std::move(promise)) ->send(file_id, file_view.remote_location().as_input_photo()); return; } CHECK(input_file != nullptr); td_->create_handler(std::move(promise))->send(file_id, std::move(input_file)); } void ContactsManager::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); uploaded_profile_photos_.erase(it); promise.set_error(std::move(status)); // TODO check that status has valid error code } tl_object_ptr ContactsManager::get_user_status_object(UserId user_id, const User *u) const { if (u->is_bot) { return make_tl_object(std::numeric_limits::max()); } int32 was_online = u->was_online; if (user_id == get_my_id() && my_was_online_local_ != 0) { was_online = my_was_online_local_; } switch (was_online) { case -3: return make_tl_object(); case -2: return make_tl_object(); case -1: return make_tl_object(); case 0: return make_tl_object(); default: { int32 time = G()->unix_time(); if (was_online > time) { return make_tl_object(was_online); } else { return make_tl_object(was_online); } } } } int32 ContactsManager::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 info about " << user_id << " from " << source; unknown_users_.insert(user_id); send_closure(G()->td(), &Td::send_update, td_api::make_object(td_api::make_object( user_id.get(), "", "", "", "", td_api::make_object(), get_profile_photo_object(td_->file_manager_.get(), nullptr), get_link_state_object(LinkState::Unknown), get_link_state_object(LinkState::Unknown), false, "", false, td_api::make_object(), ""))); } return user_id.get(); } tl_object_ptr ContactsManager::get_user_object(UserId user_id) const { return get_user_object(user_id, get_user(user_id)); } tl_object_ptr ContactsManager::get_user_object(UserId user_id, const User *u) const { if (u == nullptr) { return nullptr; } tl_object_ptr type; if (u->is_deleted) { type = make_tl_object(); } else if (u->is_bot) { type = make_tl_object(u->can_join_groups, u->can_read_all_group_messages, u->is_inline_bot, u->inline_query_placeholder, u->need_location_bot); } else { type = make_tl_object(); } return make_tl_object( user_id.get(), u->first_name, u->last_name, u->username, u->phone_number, get_user_status_object(user_id, u), get_profile_photo_object(td_->file_manager_.get(), &u->photo), get_link_state_object(u->outbound), get_link_state_object(u->inbound), u->is_verified, u->restriction_reason, u->is_received, std::move(type), u->language_code); } vector ContactsManager::get_user_ids_object(const vector &user_ids) const { return transform(user_ids, [this](UserId user_id) { return get_user_id_object(user_id, "get_user_ids_object"); }); } tl_object_ptr ContactsManager::get_users_object(int32 total_count, const vector &user_ids) const { if (total_count == -1) { total_count = narrow_cast(user_ids.size()); } return td_api::make_object(total_count, get_user_ids_object(user_ids)); } tl_object_ptr ContactsManager::get_user_full_info_object(UserId user_id) const { return get_user_full_info_object(user_id, get_user_full(user_id)); } tl_object_ptr ContactsManager::get_user_full_info_object(UserId user_id, const UserFull *user_full) const { CHECK(user_full != nullptr); bool is_bot = is_user_bot(user_id); return make_tl_object(user_full->is_blocked, user_full->can_be_called, user_full->has_private_calls, is_bot ? string() : user_full->about, is_bot ? user_full->about : string(), user_full->common_chat_count, get_bot_info_object(user_full->bot_info.get())); } int32 ContactsManager::get_basic_group_id_object(ChatId chat_id, const char *source) const { if (chat_id.is_valid() && get_chat(chat_id) == nullptr && unknown_chats_.count(chat_id) == 0) { LOG(ERROR) << "Have no info about " << chat_id << " from " << source; unknown_chats_.insert(chat_id); send_closure( G()->td(), &Td::send_update, td_api::make_object(td_api::make_object( chat_id.get(), 0, DialogParticipantStatus::Banned(0).get_chat_member_status_object(), true, true, 0))); } return chat_id.get(); } tl_object_ptr ContactsManager::get_basic_group_object(ChatId chat_id) { return get_basic_group_object(chat_id, get_chat(chat_id)); } tl_object_ptr ContactsManager::get_basic_group_object(ChatId chat_id, const Chat *chat) { if (chat == nullptr) { return nullptr; } if (chat->migrated_to_channel_id.is_valid()) { get_channel_force(chat->migrated_to_channel_id); } return get_basic_group_object_const(chat_id, chat); } tl_object_ptr ContactsManager::get_basic_group_object_const(ChatId chat_id, const Chat *chat) const { return make_tl_object( chat_id.get(), chat->participant_count, get_chat_status(chat).get_chat_member_status_object(), chat->everyone_is_administrator, chat->is_active, get_supergroup_id_object(chat->migrated_to_channel_id, "get_basic_group_object")); } tl_object_ptr ContactsManager::get_basic_group_full_info_object(ChatId chat_id) const { return get_basic_group_full_info_object(get_chat_full(chat_id)); } tl_object_ptr ContactsManager::get_basic_group_full_info_object( const ChatFull *chat_full) const { CHECK(chat_full != nullptr); return make_tl_object( get_user_id_object(chat_full->creator_user_id, "basicGroupFullInfo"), transform(chat_full->participants, [this](const DialogParticipant &chat_participant) { return get_chat_member_object(chat_participant); }), chat_full->invite_link); } int32 ContactsManager::get_supergroup_id_object(ChannelId channel_id, const char *source) const { if (channel_id.is_valid() && get_channel(channel_id) == nullptr && unknown_channels_.count(channel_id) == 0) { LOG(ERROR) << "Have no info about " << channel_id << " received from " << source; unknown_channels_.insert(channel_id); send_closure(G()->td(), &Td::send_update, td_api::make_object(td_api::make_object( channel_id.get(), string(), 0, DialogParticipantStatus::Banned(0).get_chat_member_status_object(), 0, false, false, true, false, ""))); } return channel_id.get(); } tl_object_ptr ContactsManager::get_supergroup_object(ChannelId channel_id) const { return get_supergroup_object(channel_id, get_channel(channel_id)); } tl_object_ptr ContactsManager::get_supergroup_object(ChannelId channel_id, const Channel *channel) const { if (channel == nullptr) { return nullptr; } return make_tl_object( channel_id.get(), channel->username, channel->date, get_channel_status(channel).get_chat_member_status_object(), channel->participant_count, channel->anyone_can_invite, channel->sign_messages, !channel->is_megagroup, channel->is_verified, channel->restriction_reason); } tl_object_ptr ContactsManager::get_supergroup_full_info_object(ChannelId channel_id) const { return get_supergroup_full_info_object(get_channel_full(channel_id)); } tl_object_ptr ContactsManager::get_supergroup_full_info_object( const ChannelFull *channel_full) const { CHECK(channel_full != nullptr); return make_tl_object( channel_full->description, channel_full->participant_count, channel_full->administrator_count, channel_full->restricted_count, channel_full->banned_count, channel_full->can_get_participants, channel_full->can_set_username, channel_full->can_set_sticker_set, channel_full->is_all_history_available, channel_full->sticker_set_id, channel_full->invite_link, get_basic_group_id_object(channel_full->migrated_from_chat_id, "get_supergroup_full_info_object"), channel_full->migrated_from_max_message_id.get()); } tl_object_ptr ContactsManager::get_secret_chat_state_object(SecretChatState state) { switch (state) { case SecretChatState::Waiting: return make_tl_object(); case SecretChatState::Active: return make_tl_object(); case SecretChatState::Closed: case SecretChatState::Unknown: return make_tl_object(); default: UNREACHABLE(); return nullptr; } } int32 ContactsManager::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 info about " << secret_chat_id << " from " << source; unknown_secret_chats_.insert(secret_chat_id); send_closure( G()->td(), &Td::send_update, td_api::make_object(td_api::make_object( secret_chat_id.get(), 0, get_secret_chat_state_object(SecretChatState::Unknown), false, 0, string(), 0))); } return secret_chat_id.get(); } tl_object_ptr ContactsManager::get_secret_chat_object(SecretChatId secret_chat_id) { return get_secret_chat_object(secret_chat_id, get_secret_chat(secret_chat_id)); } tl_object_ptr ContactsManager::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); return get_secret_chat_object_const(secret_chat_id, secret_chat); } tl_object_ptr ContactsManager::get_secret_chat_object_const(SecretChatId secret_chat_id, const SecretChat *secret_chat) const { return td_api::make_object( 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->ttl, secret_chat->key_hash, secret_chat->layer); } tl_object_ptr ContactsManager::get_link_state_object(LinkState link) { switch (link) { case LinkState::Unknown: case LinkState::None: return make_tl_object(); case LinkState::KnowsPhoneNumber: return make_tl_object(); case LinkState::Contact: return make_tl_object(); default: UNREACHABLE(); } return make_tl_object(); } tl_object_ptr ContactsManager::get_bot_info_object(const BotInfo *bot_info) { if (bot_info == nullptr) { return nullptr; } vector> commands; for (auto &command : bot_info->commands) { commands.push_back(make_tl_object(command.first, command.second)); } return make_tl_object(bot_info->description, std::move(commands)); } tl_object_ptr ContactsManager::get_bot_info_object(UserId user_id) const { auto user_full = get_user_full(user_id); if (user_full == nullptr || user_full->bot_info == nullptr) { return nullptr; } return get_bot_info_object(user_full->bot_info.get()); } tl_object_ptr ContactsManager::get_chat_invite_link_info_object( const string &invite_link) const { auto it = invite_link_infos_.find(invite_link); if (it == invite_link_infos_.end()) { return nullptr; } auto invite_link_info = it->second.get(); CHECK(invite_link_info != nullptr); DialogId dialog_id; string title; const DialogPhoto *photo = nullptr; int32 participant_count = 0; vector member_user_ids; bool is_public = false; td_api::object_ptr chat_type; if (invite_link_info->chat_id != ChatId()) { CHECK(invite_link_info->channel_id == ChannelId()); auto chat_id = invite_link_info->chat_id; const Chat *c = get_chat(chat_id); dialog_id = DialogId(invite_link_info->chat_id); if (c != nullptr) { title = c->title; photo = &c->photo; participant_count = c->participant_count; } else { LOG(ERROR) << "Have no information about " << chat_id; } chat_type = td_api::make_object( get_basic_group_id_object(chat_id, "get_chat_invite_link_info_object")); } else if (invite_link_info->channel_id != ChannelId()) { CHECK(invite_link_info->chat_id == ChatId()); auto channel_id = invite_link_info->channel_id; const Channel *c = get_channel(channel_id); dialog_id = DialogId(invite_link_info->channel_id); bool is_megagroup = false; if (c != nullptr) { title = c->title; photo = &c->photo; is_public = !c->username.empty(); is_megagroup = c->is_megagroup; participant_count = c->participant_count; } else { LOG(ERROR) << "Have no information about " << channel_id; } chat_type = td_api::make_object( get_supergroup_id_object(channel_id, "get_chat_invite_link_info_object"), !is_megagroup); } else { title = invite_link_info->title; photo = &invite_link_info->photo; participant_count = invite_link_info->participant_count; member_user_ids = get_user_ids_object(invite_link_info->participant_user_ids); is_public = invite_link_info->is_public; if (invite_link_info->is_chat) { chat_type = td_api::make_object(0); } else { chat_type = td_api::make_object(0, !invite_link_info->is_megagroup); } } if (dialog_id != DialogId()) { td_->messages_manager_->force_create_dialog(dialog_id, "get_chat_invite_link_info_object"); } return make_tl_object(dialog_id.get(), std::move(chat_type), title, get_chat_photo_object(td_->file_manager_.get(), photo), participant_count, std::move(member_user_ids), is_public); } UserId ContactsManager::get_support_user(Promise &&promise) { if (support_user_id_.is_valid()) { promise.set_value(Unit()); return support_user_id_; } td_->create_handler(std::move(promise))->send(); return UserId(); } void ContactsManager::after_get_difference() { get_user(get_my_id(), 3, Promise()); } void ContactsManager::get_current_state(vector> &updates) const { for (auto &it : users_) { updates.push_back(td_api::make_object(get_user_object(it.first, &it.second))); } for (auto &it : channels_) { updates.push_back(td_api::make_object(get_supergroup_object(it.first, &it.second))); } for (auto &it : chats_) { // chat object can contain channel_id, so it must be sent after channels updates.push_back( td_api::make_object(get_basic_group_object_const(it.first, &it.second))); } for (auto &it : secret_chats_) { // secret chat object contains user_id, so it must be sent after users updates.push_back( td_api::make_object(get_secret_chat_object_const(it.first, &it.second))); } for (auto &it : users_full_) { if (!it.second.is_inited) { continue; } updates.push_back(td_api::make_object(get_user_id_object(it.first, "get_current_state"), get_user_full_info_object(it.first, &it.second))); } for (auto &it : channels_full_) { updates.push_back(td_api::make_object( get_supergroup_id_object(it.first, "get_current_state"), get_supergroup_full_info_object(&it.second))); } for (auto &it : chats_full_) { updates.push_back(td_api::make_object( get_basic_group_id_object(it.first, "get_current_state"), get_basic_group_full_info_object(&it.second))); } } } // namespace td