1 //
2 // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2021
3 //
4 // Distributed under the Boost Software License, Version 1.0. (See accompanying
5 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
6 //
7 #include "td/telegram/Dependencies.h"
8 
9 #include "td/telegram/ContactsManager.h"
10 #include "td/telegram/MessagesManager.h"
11 #include "td/telegram/Td.h"
12 #include "td/telegram/WebPagesManager.h"
13 
14 #include "td/utils/common.h"
15 #include "td/utils/logging.h"
16 
17 namespace td {
18 
add_dialog_and_dependencies(Dependencies & dependencies,DialogId dialog_id)19 void add_dialog_and_dependencies(Dependencies &dependencies, DialogId dialog_id) {
20   if (dialog_id.is_valid() && dependencies.dialog_ids.insert(dialog_id).second) {
21     add_dialog_dependencies(dependencies, dialog_id);
22   }
23 }
24 
add_dialog_dependencies(Dependencies & dependencies,DialogId dialog_id)25 void add_dialog_dependencies(Dependencies &dependencies, DialogId dialog_id) {
26   switch (dialog_id.get_type()) {
27     case DialogType::User:
28       dependencies.user_ids.insert(dialog_id.get_user_id());
29       break;
30     case DialogType::Chat:
31       dependencies.chat_ids.insert(dialog_id.get_chat_id());
32       break;
33     case DialogType::Channel:
34       dependencies.channel_ids.insert(dialog_id.get_channel_id());
35       break;
36     case DialogType::SecretChat:
37       dependencies.secret_chat_ids.insert(dialog_id.get_secret_chat_id());
38       break;
39     case DialogType::None:
40       break;
41     default:
42       UNREACHABLE();
43   }
44 }
45 
add_message_sender_dependencies(Dependencies & dependencies,DialogId dialog_id)46 void add_message_sender_dependencies(Dependencies &dependencies, DialogId dialog_id) {
47   if (dialog_id.get_type() == DialogType::User) {
48     dependencies.user_ids.insert(dialog_id.get_user_id());
49   } else {
50     add_dialog_and_dependencies(dependencies, dialog_id);
51   }
52 }
53 
resolve_dependencies_force(Td * td,const Dependencies & dependencies,const char * source)54 bool resolve_dependencies_force(Td *td, const Dependencies &dependencies, const char *source) {
55   bool success = true;
56   for (auto user_id : dependencies.user_ids) {
57     if (user_id.is_valid() && !td->contacts_manager_->have_user_force(user_id)) {
58       LOG(ERROR) << "Can't find " << user_id << " from " << source;
59       success = false;
60     }
61   }
62   for (auto chat_id : dependencies.chat_ids) {
63     if (chat_id.is_valid() && !td->contacts_manager_->have_chat_force(chat_id)) {
64       LOG(ERROR) << "Can't find " << chat_id << " from " << source;
65       success = false;
66     }
67   }
68   for (auto channel_id : dependencies.channel_ids) {
69     if (channel_id.is_valid() && !td->contacts_manager_->have_channel_force(channel_id)) {
70       LOG(ERROR) << "Can't find " << channel_id << " from " << source;
71       success = false;
72     }
73   }
74   for (auto secret_chat_id : dependencies.secret_chat_ids) {
75     if (secret_chat_id.is_valid() && !td->contacts_manager_->have_secret_chat_force(secret_chat_id)) {
76       LOG(ERROR) << "Can't find " << secret_chat_id << " from " << source;
77       success = false;
78     }
79   }
80   for (auto dialog_id : dependencies.dialog_ids) {
81     if (dialog_id.is_valid() && !td->messages_manager_->have_dialog_force(dialog_id, source)) {
82       LOG(ERROR) << "Can't find " << dialog_id << " from " << source;
83       td->messages_manager_->force_create_dialog(dialog_id, "resolve_dependencies_force", true);
84       success = false;
85     }
86   }
87   for (auto web_page_id : dependencies.web_page_ids) {
88     if (web_page_id.is_valid() && !td->web_pages_manager_->have_web_page_force(web_page_id)) {
89       LOG(INFO) << "Can't find " << web_page_id << " from " << source;
90       success = false;
91     }
92   }
93   return success;
94 }
95 
96 }  // namespace td
97