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/SuggestedAction.h"
8
9 #include "td/telegram/ChannelId.h"
10 #include "td/telegram/ConfigManager.h"
11 #include "td/telegram/ConfigShared.h"
12 #include "td/telegram/ContactsManager.h"
13 #include "td/telegram/Global.h"
14 #include "td/telegram/Td.h"
15
16 #include "td/actor/actor.h"
17
18 #include "td/utils/algorithm.h"
19
20 #include <algorithm>
21
22 namespace td {
23
init(Type type)24 void SuggestedAction::init(Type type) {
25 type_ = type;
26 }
27
SuggestedAction(Slice action_str)28 SuggestedAction::SuggestedAction(Slice action_str) {
29 if (action_str == Slice("AUTOARCHIVE_POPULAR")) {
30 init(Type::EnableArchiveAndMuteNewChats);
31 } else if (action_str == Slice("VALIDATE_PASSWORD")) {
32 init(Type::CheckPassword);
33 } else if (action_str == Slice("VALIDATE_PHONE_NUMBER")) {
34 init(Type::CheckPhoneNumber);
35 } else if (action_str == Slice("NEWCOMER_TICKS")) {
36 init(Type::SeeTicksHint);
37 }
38 }
39
SuggestedAction(Slice action_str,DialogId dialog_id)40 SuggestedAction::SuggestedAction(Slice action_str, DialogId dialog_id) {
41 CHECK(dialog_id.is_valid());
42 if (action_str == Slice("CONVERT_GIGAGROUP")) {
43 type_ = Type::ConvertToGigagroup;
44 dialog_id_ = dialog_id;
45 }
46 }
47
SuggestedAction(const td_api::object_ptr<td_api::SuggestedAction> & suggested_action)48 SuggestedAction::SuggestedAction(const td_api::object_ptr<td_api::SuggestedAction> &suggested_action) {
49 if (suggested_action == nullptr) {
50 return;
51 }
52 switch (suggested_action->get_id()) {
53 case td_api::suggestedActionEnableArchiveAndMuteNewChats::ID:
54 init(Type::EnableArchiveAndMuteNewChats);
55 break;
56 case td_api::suggestedActionCheckPassword::ID:
57 init(Type::CheckPassword);
58 break;
59 case td_api::suggestedActionCheckPhoneNumber::ID:
60 init(Type::CheckPhoneNumber);
61 break;
62 case td_api::suggestedActionSeeTicksHint::ID:
63 init(Type::SeeTicksHint);
64 break;
65 case td_api::suggestedActionConvertToBroadcastGroup::ID: {
66 auto action = static_cast<const td_api::suggestedActionConvertToBroadcastGroup *>(suggested_action.get());
67 ChannelId channel_id(action->supergroup_id_);
68 if (channel_id.is_valid()) {
69 type_ = Type::ConvertToGigagroup;
70 dialog_id_ = DialogId(channel_id);
71 }
72 break;
73 }
74 case td_api::suggestedActionSetPassword::ID: {
75 auto action = static_cast<const td_api::suggestedActionSetPassword *>(suggested_action.get());
76 type_ = Type::SetPassword;
77 otherwise_relogin_days_ = action->authorization_delay_;
78 break;
79 }
80 default:
81 UNREACHABLE();
82 }
83 }
84
get_suggested_action_str() const85 string SuggestedAction::get_suggested_action_str() const {
86 switch (type_) {
87 case Type::EnableArchiveAndMuteNewChats:
88 return "AUTOARCHIVE_POPULAR";
89 case Type::CheckPassword:
90 return "VALIDATE_PASSWORD";
91 case Type::CheckPhoneNumber:
92 return "VALIDATE_PHONE_NUMBER";
93 case Type::SeeTicksHint:
94 return "NEWCOMER_TICKS";
95 case Type::ConvertToGigagroup:
96 return "CONVERT_GIGAGROUP";
97 default:
98 return string();
99 }
100 }
101
get_suggested_action_object() const102 td_api::object_ptr<td_api::SuggestedAction> SuggestedAction::get_suggested_action_object() const {
103 switch (type_) {
104 case Type::Empty:
105 return nullptr;
106 case Type::EnableArchiveAndMuteNewChats:
107 return td_api::make_object<td_api::suggestedActionEnableArchiveAndMuteNewChats>();
108 case Type::CheckPassword:
109 return td_api::make_object<td_api::suggestedActionCheckPassword>();
110 case Type::CheckPhoneNumber:
111 return td_api::make_object<td_api::suggestedActionCheckPhoneNumber>();
112 case Type::SeeTicksHint:
113 return td_api::make_object<td_api::suggestedActionSeeTicksHint>();
114 case Type::ConvertToGigagroup:
115 return td_api::make_object<td_api::suggestedActionConvertToBroadcastGroup>(dialog_id_.get_channel_id().get());
116 case Type::SetPassword:
117 return td_api::make_object<td_api::suggestedActionSetPassword>(otherwise_relogin_days_);
118 default:
119 UNREACHABLE();
120 return nullptr;
121 }
122 }
123
get_update_suggested_actions_object(const vector<SuggestedAction> & added_actions,const vector<SuggestedAction> & removed_actions)124 td_api::object_ptr<td_api::updateSuggestedActions> get_update_suggested_actions_object(
125 const vector<SuggestedAction> &added_actions, const vector<SuggestedAction> &removed_actions) {
126 auto get_object = [](const SuggestedAction &action) {
127 return action.get_suggested_action_object();
128 };
129 return td_api::make_object<td_api::updateSuggestedActions>(transform(added_actions, get_object),
130 transform(removed_actions, get_object));
131 }
132
update_suggested_actions(vector<SuggestedAction> & suggested_actions,vector<SuggestedAction> && new_suggested_actions)133 void update_suggested_actions(vector<SuggestedAction> &suggested_actions,
134 vector<SuggestedAction> &&new_suggested_actions) {
135 td::unique(new_suggested_actions);
136 if (new_suggested_actions == suggested_actions) {
137 return;
138 }
139
140 vector<SuggestedAction> added_actions;
141 vector<SuggestedAction> removed_actions;
142 auto old_it = suggested_actions.begin();
143 auto new_it = new_suggested_actions.begin();
144 while (old_it != suggested_actions.end() || new_it != new_suggested_actions.end()) {
145 if (old_it != suggested_actions.end() && (new_it == new_suggested_actions.end() || *old_it < *new_it)) {
146 removed_actions.push_back(*old_it++);
147 } else if (old_it == suggested_actions.end() || *new_it < *old_it) {
148 added_actions.push_back(*new_it++);
149 } else {
150 ++old_it;
151 ++new_it;
152 }
153 }
154 CHECK(!added_actions.empty() || !removed_actions.empty());
155 suggested_actions = std::move(new_suggested_actions);
156 send_closure(G()->td(), &Td::send_update, get_update_suggested_actions_object(added_actions, removed_actions));
157 }
158
remove_suggested_action(vector<SuggestedAction> & suggested_actions,SuggestedAction suggested_action)159 void remove_suggested_action(vector<SuggestedAction> &suggested_actions, SuggestedAction suggested_action) {
160 if (td::remove(suggested_actions, suggested_action)) {
161 send_closure(G()->td(), &Td::send_update, get_update_suggested_actions_object({}, {suggested_action}));
162 }
163 }
164
dismiss_suggested_action(SuggestedAction action,Promise<Unit> && promise)165 void dismiss_suggested_action(SuggestedAction action, Promise<Unit> &&promise) {
166 switch (action.type_) {
167 case SuggestedAction::Type::Empty:
168 return promise.set_error(Status::Error(400, "Action must be non-empty"));
169 case SuggestedAction::Type::EnableArchiveAndMuteNewChats:
170 case SuggestedAction::Type::CheckPassword:
171 case SuggestedAction::Type::CheckPhoneNumber:
172 case SuggestedAction::Type::SeeTicksHint:
173 return send_closure_later(G()->config_manager(), &ConfigManager::dismiss_suggested_action, std::move(action),
174 std::move(promise));
175 case SuggestedAction::Type::ConvertToGigagroup:
176 return send_closure_later(G()->contacts_manager(), &ContactsManager::dismiss_dialog_suggested_action,
177 std::move(action), std::move(promise));
178 case SuggestedAction::Type::SetPassword: {
179 if (action.otherwise_relogin_days_ <= 0) {
180 return promise.set_error(Status::Error(400, "Invalid authorization_delay specified"));
181 }
182 auto days = narrow_cast<int32>(G()->shared_config().get_option_integer("otherwise_relogin_days"));
183 if (days == action.otherwise_relogin_days_) {
184 vector<SuggestedAction> removed_actions{SuggestedAction{SuggestedAction::Type::SetPassword, DialogId(), days}};
185 send_closure(G()->td(), &Td::send_update, get_update_suggested_actions_object({}, removed_actions));
186 G()->shared_config().set_option_empty("otherwise_relogin_days");
187 }
188 return promise.set_value(Unit());
189 }
190 default:
191 UNREACHABLE();
192 return;
193 }
194 }
195
196 } // namespace td
197