Add td_api::addContact.

GitOrigin-RevId: 6b80aad27805780043a5ca4e7680bbfea65ddaad

td/generate/scheme/td_api.tl

@@ -3548,7 +3548,11 @@ unblockUser user_id:int32 = Ok;
 getBlockedUsers offset:int32 limit:int32 = Users;
 
 
-//@description Adds new contacts or edits existing contacts; contacts' user identifiers are ignored @contacts The list of contacts to import or edit, contact's vCard are ignored and are not imported
+//@description Adds a user to the contacts list or edits an existing contact by their user_id @contact The contact to add or edit; phone number can be empty and needs to be specified only if known, vCard is ignored
+//@share_phone_number True, if the user needs to be allowed to see current user's phone number. A corresponding rule to userPrivacySettingShowPhoneNumber will be added if needed
+addContact contact:contact share_phone_number:Bool = Ok;
+
+//@description Adds new contacts or edits existing contacts by their phone numbers; contacts' user identifiers are ignored @contacts The list of contacts to import or edit; contacts' vCard are ignored and are not imported
 importContacts contacts:vector<contact> = ImportedContacts;
 
 //@description Returns all user contacts

td/telegram/ContactsManager.cpp

@@ -488,6 +488,42 @@ class GetContactsStatusesQuery : public Td::ResultHandler {
   }
 };
 
+class AddContactQuery : public Td::ResultHandler {
+  Promise<Unit> promise_;
+
+ public:
+  explicit AddContactQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
+  }
+
+  void send(tl_object_ptr<telegram_api::InputUser> &&input_user, const string &first_name, const string &last_name,
+            const string &phone_number, bool share_phone_number) {
+    int32 flags = 0;
+    if (share_phone_number) {
+      flags |= telegram_api::contacts_addContact::ADD_PHONE_PRIVACY_EXCEPTION_MASK;
+    }
+    send_query(G()->net_query_creator().create(create_storer(telegram_api::contacts_addContact(
+        flags, false /*ignored*/, std::move(input_user), first_name, last_name, phone_number))));
+  }
+
+  void on_result(uint64 id, BufferSlice packet) override {
+    auto result_ptr = fetch_result<telegram_api::contacts_addContact>(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 AddContactQuery: " << 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->contacts_manager_->reload_contacts(true);
+  }
+};
+
 class ImportContactsQuery : public Td::ResultHandler {
   Promise<Unit> promise_;
   vector<Contact> input_contacts_;
@@ -593,7 +629,7 @@ class DeleteContactsQuery : public Td::ResultHandler {
   explicit DeleteContactsQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
   }
 
-  void send(vector<UserId> &&user_ids, vector<tl_object_ptr<telegram_api::InputUser>> &&input_users) {
+  void send(vector<tl_object_ptr<telegram_api::InputUser>> &&input_users) {
     send_query(
         G()->net_query_creator().create(create_storer(telegram_api::contacts_deleteContacts(std::move(input_users)))));
   }
@@ -3941,6 +3977,37 @@ void ContactsManager::reload_contacts(bool force) {
   }
 }
 
+void ContactsManager::add_contact(td_api::object_ptr<td_api::contact> &&contact, bool share_phone_number,
+                                  Promise<Unit> &&promise) {
+  if (contact == nullptr) {
+    return promise.set_error(Status::Error(400, "Added contact must be non-empty"));
+  }
+
+  if (G()->close_flag()) {
+    return promise.set_error(Status::Error(500, "Request aborted"));
+  }
+
+  if (!are_contacts_loaded_) {
+    load_contacts(PromiseCreator::lambda([actor_id = actor_id(this), contact = std::move(contact), share_phone_number,
+                                          promise = std::move(promise)](Result<Unit> &&) mutable {
+      send_closure(actor_id, &ContactsManager::add_contact, std::move(contact), share_phone_number, std::move(promise));
+    }));
+    return;
+  }
+
+  LOG(INFO) << "Add " << oneline(to_string(contact)) << " with share_phone_number = " << share_phone_number;
+
+  UserId user_id{contact->user_id_};
+  auto input_user = get_input_user(user_id);
+  if (input_user == nullptr) {
+    return promise.set_error(Status::Error(3, "User not found"));
+  }
+
+  td_->create_handler<AddContactQuery>(std::move(promise))
+      ->send(std::move(input_user), contact->first_name_, contact->last_name_, contact->phone_number_,
+             share_phone_number);
+}
+
 std::pair<vector<UserId>, vector<int32>> ContactsManager::import_contacts(
     const vector<tl_object_ptr<td_api::contact>> &contacts, int64 &random_id, Promise<Unit> &&promise) {
   if (!are_contacts_loaded_) {
@@ -3961,7 +4028,7 @@ std::pair<vector<UserId>, vector<int32>> ContactsManager::import_contacts(
   }
   for (auto &contact : contacts) {
     if (contact == nullptr) {
-      promise.set_error(Status::Error(400, "Imported contacts should not be empty"));
+      promise.set_error(Status::Error(400, "Imported contacts must be non-empty"));
       return {};
     }
   }
@@ -4004,8 +4071,7 @@ void ContactsManager::remove_contacts(vector<UserId> user_ids, Promise<Unit> &&p
     return promise.set_value(Unit());
   }
 
-  td_->create_handler<DeleteContactsQuery>(std::move(promise))
-      ->send(std::move(to_delete_user_ids), std::move(input_users));
+  td_->create_handler<DeleteContactsQuery>(std::move(promise))->send(std::move(input_users));
 }
 
 void ContactsManager::remove_contacts_by_phone_number(vector<string> user_phone_numbers, vector<UserId> user_ids,

td/telegram/ContactsManager.h

@@ -260,6 +260,8 @@ class ContactsManager : public Actor {
 
   tl_object_ptr<td_api::users> get_blocked_users_object(int64 random_id);
 
+  void add_contact(td_api::object_ptr<td_api::contact> &&contact, bool share_phone_number, Promise<Unit> &&promise);
+
   std::pair<vector<UserId>, vector<int32>> import_contacts(const vector<tl_object_ptr<td_api::contact>> &contacts,
                                                            int64 &random_id, Promise<Unit> &&promise);
 

td/telegram/Td.cpp

@@ -6331,6 +6331,18 @@ void Td::on_request(uint64 id, const td_api::getBlockedUsers &request) {
   CREATE_REQUEST(GetBlockedUsersRequest, request.offset_, request.limit_);
 }
 
+void Td::on_request(uint64 id, td_api::addContact &request) {
+  CHECK_IS_USER();
+  if (request.contact_ == nullptr) {
+    return send_error_raw(id, 5, "Contact must not be empty");
+  }
+  CLEAN_INPUT_STRING(request.contact_->phone_number_);
+  CLEAN_INPUT_STRING(request.contact_->first_name_);
+  CLEAN_INPUT_STRING(request.contact_->last_name_);
+  CREATE_OK_REQUEST_PROMISE();
+  contacts_manager_->add_contact(std::move(request.contact_), request.share_phone_number_, std::move(promise));
+}
+
 void Td::on_request(uint64 id, td_api::importContacts &request) {
   CHECK_IS_USER();
   for (auto &contact : request.contacts_) {

td/telegram/Td.h

@@ -726,6 +726,8 @@ class Td final : public NetQueryCallback {
 
   void on_request(uint64 id, const td_api::getBlockedUsers &request);
 
+  void on_request(uint64 id, td_api::addContact &request);
+
   void on_request(uint64 id, td_api::importContacts &request);
 
   void on_request(uint64 id, const td_api::getContacts &request);

td/telegram/cli.cpp

@@ -1597,6 +1597,15 @@ class CliClient final : public Actor {
         auto limit = to_integer<int32>(args);
         send_request(td_api::make_object<td_api::searchContacts>("", limit));
       }
+    } else if (op == "AddContact") {
+      string user_id;
+      string first_name;
+      string last_name;
+      std::tie(user_id, args) = split(args);
+      std::tie(first_name, last_name) = split(args);
+
+      send_request(td_api::make_object<td_api::addContact>(
+          td_api::make_object<td_api::contact>(string(), first_name, last_name, string(), as_user_id(user_id)), false));
     } else if (op == "ImportContacts" || op == "cic") {
       vector<string> contacts_str = full_split(args, ';');
       vector<tl_object_ptr<td_api::contact>> contacts;