tdesktop/Telegram/SourceFiles/profile/profile_channel_controllers...

1964 lines
59 KiB
C++

/*
This file is part of Telegram Desktop,
the official desktop application for the Telegram messaging service.
For license and copyright information please follow this link:
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
*/
#include "profile/profile_channel_controllers.h"
#include "boxes/peer_list_controllers.h"
#include "boxes/edit_participant_box.h"
#include "boxes/confirm_box.h"
#include "boxes/add_contact_box.h"
#include "core/tl_help.h"
#include "base/overload.h"
#include "auth_session.h"
#include "apiwrap.h"
#include "lang/lang_keys.h"
#include "mainwidget.h"
#include "observer_peer.h"
#include "dialogs/dialogs_indexed_list.h"
#include "data/data_peer_values.h"
#include "data/data_session.h"
#include "ui/widgets/popup_menu.h"
#include "window/window_controller.h"
#include "history/history.h"
namespace Profile {
namespace {
constexpr auto kParticipantsFirstPageCount = 16;
constexpr auto kParticipantsPerPage = 200;
constexpr auto kSortByOnlineDelay = TimeMs(1000);
void RemoveAdmin(
not_null<ChannelData*> channel,
not_null<UserData*> user,
const MTPChannelAdminRights &oldRights,
base::lambda<void()> onDone) {
const auto newRights = MTP_channelAdminRights(MTP_flags(0));
auto done = [=](const MTPUpdates &result) {
Auth().api().applyUpdates(result);
channel->applyEditAdmin(user, oldRights, newRights);
onDone();
};
MTP::send(
MTPchannels_EditAdmin(
channel->inputChannel,
user->inputUser,
newRights),
rpcDone(std::move(done)));
}
} // namespace
ParticipantsBoxController::ParticipantsBoxController(
not_null<Window::Navigation*> navigation,
not_null<ChannelData*> channel,
Role role)
: PeerListController(CreateSearchController(channel, role, &_additional))
, _navigation(navigation)
, _channel(channel)
, _role(role) {
if (_channel->mgInfo) {
_additional.creator = _channel->mgInfo->creator;
}
if (_role == Role::Profile) {
setupSortByOnline();
setupListChangeViewers();
}
}
void ParticipantsBoxController::setupSortByOnline() {
_sortByOnlineTimer.setCallback([this] { sortByOnline(); });
using UpdateFlag = Notify::PeerUpdate::Flag;
subscribe(Notify::PeerUpdated(), Notify::PeerUpdatedHandler(
UpdateFlag::UserOnlineChanged,
[this](const Notify::PeerUpdate &update) {
if (auto row = delegate()->peerListFindRow(
update.peer->id)) {
row->refreshStatus();
sortByOnlineDelayed();
}
}));
}
void ParticipantsBoxController::setupListChangeViewers() {
if (!_channel->isMegagroup()) {
return;
}
Auth().data().megagroupParticipantAdded(
_channel
) | rpl::start_with_next([this](not_null<UserData*> user) {
if (delegate()->peerListFullRowsCount() > 0) {
if (delegate()->peerListRowAt(0)->peer() == user) {
return;
}
}
if (auto row = delegate()->peerListFindRow(user->id)) {
delegate()->peerListPartitionRows([user](const PeerListRow &row) {
return (row.peer() == user);
});
} else {
delegate()->peerListPrependRow(createRow(user));
delegate()->peerListRefreshRows();
sortByOnline();
}
}, lifetime());
Auth().data().megagroupParticipantRemoved(
_channel
) | rpl::start_with_next([this](not_null<UserData*> user) {
if (auto row = delegate()->peerListFindRow(user->id)) {
delegate()->peerListRemoveRow(row);
}
delegate()->peerListRefreshRows();
}, lifetime());
}
void ParticipantsBoxController::sortByOnlineDelayed() {
if (!_sortByOnlineTimer.isActive()) {
_sortByOnlineTimer.callOnce(kSortByOnlineDelay);
}
}
void ParticipantsBoxController::sortByOnline() {
if (_role != Role::Profile
|| !_channel->isMegagroup()
|| _channel->membersCount() > Global::ChatSizeMax()) {
_onlineCount = 0;
return;
}
auto now = unixtime();
delegate()->peerListSortRows([now](
const PeerListRow &a,
const PeerListRow &b) {
return Data::SortByOnlineValue(a.peer()->asUser(), now) >
Data::SortByOnlineValue(b.peer()->asUser(), now);
});
refreshOnlineCount();
}
void ParticipantsBoxController::refreshOnlineCount() {
Expects(_role == Role::Profile);
Expects(_channel->membersCount() <= Global::ChatSizeMax());
auto now = unixtime();
auto left = 0, right = delegate()->peerListFullRowsCount();
while (right > left) {
auto middle = (left + right) / 2;
auto row = delegate()->peerListRowAt(middle);
if (Data::OnlineTextActive(row->peer()->asUser(), now)) {
left = middle + 1;
} else {
right = middle;
}
}
_onlineCount = left;
}
std::unique_ptr<PeerListSearchController>
ParticipantsBoxController::CreateSearchController(
not_null<ChannelData*> channel,
Role role,
not_null<Additional*> additional) {
// In admins box complex search is used for adding new admins.
if (role != Role::Admins || channel->canAddAdmins()) {
return std::make_unique<ParticipantsBoxSearchController>(
channel,
role,
additional);
}
return nullptr;
}
void ParticipantsBoxController::Start(
not_null<Window::Navigation*> navigation,
not_null<ChannelData*> channel,
Role role) {
auto controller = std::make_unique<ParticipantsBoxController>(
navigation,
channel,
role);
auto initBox = [role, channel, controller = controller.get()](not_null<PeerListBox*> box) {
box->addButton(langFactory(lng_close), [box] { box->closeBox(); });
auto canAddNewItem = [role, channel] {
switch (role) {
case Role::Members: return !channel->isMegagroup() && channel->canAddMembers() && (channel->membersCount() < Global::ChatSizeMax());
case Role::Admins: return channel->canAddAdmins();
case Role::Restricted:
case Role::Kicked: return channel->canBanMembers();
}
Unexpected("Role value in ParticipantsBoxController::Start()");
};
auto addNewItemText = [role] {
switch (role) {
case Role::Members: return langFactory(lng_channel_add_members);
case Role::Admins: return langFactory(lng_channel_add_admin);
case Role::Restricted: return langFactory(lng_channel_add_restricted);
case Role::Kicked: return langFactory(lng_channel_add_banned);
}
Unexpected("Role value in ParticipantsBoxController::Start()");
};
if (canAddNewItem()) {
box->addLeftButton(addNewItemText(), [controller] { controller->addNewItem(); });
}
};
Ui::show(
Box<PeerListBox>(std::move(controller), std::move(initBox)),
LayerOption::KeepOther);
}
void ParticipantsBoxController::addNewItem() {
Expects(_role != Role::Profile);
if (_role == Role::Members) {
if (_channel->membersCount() >= Global::ChatSizeMax()) {
Ui::show(
Box<MaxInviteBox>(_channel),
LayerOption::KeepOther);
} else {
auto already = std::vector<not_null<UserData*>>();
already.reserve(delegate()->peerListFullRowsCount());
for (auto i = 0, count = delegate()->peerListFullRowsCount(); i != count; ++i) {
already.push_back(delegate()->peerListRowAt(i)->peer()->asUser());
}
AddParticipantsBoxController::Start(_channel, { already.begin(), already.end() });
}
return;
}
auto weak = base::make_weak(this);
_addBox = Ui::show(Box<PeerListBox>(std::make_unique<AddParticipantBoxController>(_channel, _role, [weak](not_null<UserData*> user, const MTPChannelAdminRights &rights) {
if (const auto strong = weak.get()) {
strong->editAdminDone(user, rights);
}
}, [weak](not_null<UserData*> user, const MTPChannelBannedRights &rights) {
if (const auto strong = weak.get()) {
strong->editRestrictedDone(user, rights);
}
}), [](not_null<PeerListBox*> box) {
box->addButton(langFactory(lng_cancel), [box] { box->closeBox(); });
}), LayerOption::KeepOther);
}
void ParticipantsBoxController::peerListSearchAddRow(not_null<PeerData*> peer) {
PeerListController::peerListSearchAddRow(peer);
if (_role == Role::Restricted && delegate()->peerListFullRowsCount() > 0) {
setDescriptionText(QString());
}
}
std::unique_ptr<PeerListRow> ParticipantsBoxController::createSearchRow(
not_null<PeerData*> peer) {
if (auto user = peer->asUser()) {
return createRow(user);
}
return nullptr;
}
std::unique_ptr<PeerListRow> ParticipantsBoxController::createRestoredRow(
not_null<PeerData*> peer) {
if (auto user = peer->asUser()) {
return createRow(user);
}
return nullptr;
}
std::unique_ptr<PeerListState> ParticipantsBoxController::saveState() const {
Expects(_role == Role::Profile);
auto result = PeerListController::saveState();
auto my = std::make_unique<SavedState>();
my->additional = _additional;
my->offset = _offset;
my->allLoaded = _allLoaded;
my->wasLoading = (_loadRequestId != 0);
if (auto search = searchController()) {
my->searchState = search->saveState();
}
if (_channel->isMegagroup()) {
auto weak = result.get();
Auth().data().megagroupParticipantAdded(
_channel
) | rpl::start_with_next([weak](not_null<UserData*> user) {
if (!weak->list.empty()) {
if (weak->list[0] == user) {
return;
}
}
auto pos = ranges::find(weak->list, user);
if (pos == weak->list.cend()) {
weak->list.push_back(user);
}
ranges::stable_partition(
weak->list,
[user](auto peer) { return (peer == user); });
}, my->lifetime);
Auth().data().megagroupParticipantRemoved(
_channel
) | rpl::start_with_next([weak](not_null<UserData*> user) {
weak->list.erase(std::remove(
weak->list.begin(),
weak->list.end(),
user), weak->list.end());
weak->filterResults.erase(std::remove(
weak->filterResults.begin(),
weak->filterResults.end(),
user), weak->filterResults.end());
}, my->lifetime);
}
result->controllerState = std::move(my);
return result;
}
void ParticipantsBoxController::restoreState(
std::unique_ptr<PeerListState> state) {
auto typeErasedState = state
? state->controllerState.get()
: nullptr;
if (auto my = dynamic_cast<SavedState*>(typeErasedState)) {
if (auto requestId = base::take(_loadRequestId)) {
request(requestId).cancel();
}
_additional = std::move(my->additional);
_offset = my->offset;
_allLoaded = my->allLoaded;
if (auto search = searchController()) {
search->restoreState(std::move(my->searchState));
}
if (my->wasLoading) {
loadMoreRows();
}
PeerListController::restoreState(std::move(state));
if (delegate()->peerListFullRowsCount() > 0) {
setNonEmptyDescription();
} else if (_allLoaded) {
setDescriptionText(lang(lng_blocked_list_not_found));
}
sortByOnline();
}
}
rpl::producer<int> ParticipantsBoxController::onlineCountValue() const {
return _onlineCount.value();
}
template <typename Callback>
void ParticipantsBoxController::HandleParticipant(const MTPChannelParticipant &participant, Role role, not_null<Additional*> additional, Callback callback) {
if ((role == Role::Profile
|| role == Role::Members
|| role == Role::Admins)
&& participant.type() == mtpc_channelParticipantAdmin) {
auto &admin = participant.c_channelParticipantAdmin();
if (auto user = App::userLoaded(admin.vuser_id.v)) {
additional->adminRights[user] = admin.vadmin_rights;
if (admin.is_can_edit()) {
additional->adminCanEdit.emplace(user);
} else {
additional->adminCanEdit.erase(user);
}
if (auto promoted = App::userLoaded(admin.vpromoted_by.v)) {
auto it = additional->adminPromotedBy.find(user);
if (it == additional->adminPromotedBy.end()) {
additional->adminPromotedBy.emplace(user, promoted);
} else {
it->second = promoted;
}
} else {
LOG(("API Error: No user %1 for admin promoted by.").arg(admin.vpromoted_by.v));
}
callback(user);
}
} else if ((role == Role::Profile
|| role == Role::Members
|| role == Role::Admins)
&& participant.type() == mtpc_channelParticipantCreator) {
auto &creator = participant.c_channelParticipantCreator();
if (auto user = App::userLoaded(creator.vuser_id.v)) {
additional->creator = user;
callback(user);
}
} else if ((role == Role::Profile
|| role == Role::Members
|| role == Role::Restricted
|| role == Role::Kicked)
&& participant.type() == mtpc_channelParticipantBanned) {
auto &banned = participant.c_channelParticipantBanned();
if (auto user = App::userLoaded(banned.vuser_id.v)) {
additional->restrictedRights[user] = banned.vbanned_rights;
if (auto kickedby = App::userLoaded(banned.vkicked_by.v)) {
auto it = additional->restrictedBy.find(user);
if (it == additional->restrictedBy.end()) {
additional->restrictedBy.emplace(user, kickedby);
} else {
it->second = kickedby;
}
}
callback(user);
}
} else if ((role == Role::Profile
|| role == Role::Members)
&& participant.type() == mtpc_channelParticipant) {
auto &member = participant.c_channelParticipant();
if (auto user = App::userLoaded(member.vuser_id.v)) {
callback(user);
}
} else if ((role == Role::Profile
|| role == Role::Members)
&& participant.type() == mtpc_channelParticipantSelf) {
auto &member = participant.c_channelParticipantSelf();
if (auto user = App::userLoaded(member.vuser_id.v)) {
callback(user);
}
} else {
LOG(("API Error: Bad participant type got while requesting for participants: %1").arg(participant.type()));
}
}
void ParticipantsBoxController::prepare() {
auto titleKey = [this] {
switch (_role) {
case Role::Admins: return lng_channel_admins;
case Role::Profile:
case Role::Members: return lng_profile_participants_section;
case Role::Restricted: return lng_restricted_list_title;
case Role::Kicked: return lng_banned_list_title;
}
Unexpected("Role in ParticipantsBoxController::prepare()");
};
delegate()->peerListSetSearchMode(PeerListSearchMode::Enabled);
delegate()->peerListSetTitle(langFactory(titleKey()));
setDescriptionText(lang(lng_contacts_loading));
setSearchNoResultsText(lang(lng_blocked_list_not_found));
loadMoreRows();
delegate()->peerListRefreshRows();
}
void ParticipantsBoxController::loadMoreRows() {
if (searchController() && searchController()->loadMoreRows()) {
return;
}
if (_loadRequestId || _allLoaded) {
return;
}
if (feedMegagroupLastParticipants()) {
return;
}
auto filter = [this] {
if (_role == Role::Members || _role == Role::Profile) {
return MTP_channelParticipantsRecent();
} else if (_role == Role::Admins) {
return MTP_channelParticipantsAdmins();
} else if (_role == Role::Restricted) {
return MTP_channelParticipantsBanned(MTP_string(QString()));
}
return MTP_channelParticipantsKicked(MTP_string(QString()));
}();
// First query is small and fast, next loads a lot of rows.
auto perPage = (_offset > 0) ? kParticipantsPerPage : kParticipantsFirstPageCount;
auto participantsHash = 0;
_loadRequestId = request(MTPchannels_GetParticipants(
_channel->inputChannel,
filter,
MTP_int(_offset),
MTP_int(perPage),
MTP_int(participantsHash)
)).done([this](const MTPchannels_ChannelParticipants &result) {
auto firstLoad = !_offset;
_loadRequestId = 0;
auto wasRecentRequest = firstLoad
&& (_role == Role::Members || _role == Role::Profile);
auto parseParticipants = [&](auto &&result, auto &&callback) {
if (wasRecentRequest) {
Auth().api().parseRecentChannelParticipants(
_channel,
result,
callback);
} else {
Auth().api().parseChannelParticipants(
_channel,
result,
callback);
}
};
parseParticipants(result, [&](
int availableCount,
const QVector<MTPChannelParticipant> &list) {
for (auto &participant : list) {
HandleParticipant(
participant,
_role,
&_additional,
[&](auto user) { appendRow(user); });
}
if (auto size = list.size()) {
_offset += size;
} else {
// To be sure - wait for a whole empty result list.
_allLoaded = true;
}
});
if (delegate()->peerListFullRowsCount() > 0) {
sortByOnline();
if (firstLoad) {
setNonEmptyDescription();
}
} else if (_allLoaded) {
setDescriptionText(lang(lng_blocked_list_not_found));
}
delegate()->peerListRefreshRows();
}).fail([this](const RPCError &error) {
_loadRequestId = 0;
}).send();
}
void ParticipantsBoxController::setNonEmptyDescription() {
setDescriptionText((_role == Role::Kicked)
? lang(_channel->isMegagroup()
? lng_group_blocked_list_about
: lng_channel_blocked_list_about)
: QString());
}
bool ParticipantsBoxController::feedMegagroupLastParticipants() {
if ((_role != Role::Members && _role != Role::Profile)
|| delegate()->peerListFullRowsCount() > 0) {
return false;
}
auto megagroup = _channel->asMegagroup();
if (!megagroup) {
return false;
}
auto info = megagroup->mgInfo.get();
//
// channelFull and channels_channelParticipants members count is desynced
// so we almost always have LastParticipantsCountOutdated that is set
// inside setMembersCount() and so we almost never use lastParticipants.
//
// => disable this check temporarily.
//
//if (info->lastParticipantsStatus != MegagroupInfo::LastParticipantsUpToDate) {
// _channel->updateFull();
// return false;
//}
if (info->lastParticipants.empty()) {
return false;
}
if (info->creator) {
_additional.creator = info->creator;
}
for_const (auto user, info->lastParticipants) {
auto admin = info->lastAdmins.find(user);
if (admin != info->lastAdmins.cend()) {
_additional.restrictedRights.erase(user);
if (admin->second.canEdit) {
_additional.adminCanEdit.emplace(user);
} else {
_additional.adminCanEdit.erase(user);
}
_additional.adminRights.emplace(user, admin->second.rights);
} else {
_additional.adminCanEdit.erase(user);
_additional.adminRights.erase(user);
auto restricted = info->lastRestricted.find(user);
if (restricted != info->lastRestricted.cend()) {
_additional.restrictedRights.emplace(user, restricted->second.rights);
} else {
_additional.restrictedRights.erase(user);
}
}
appendRow(user);
//
// Don't count lastParticipants in _offset, because we don't know
// their exact information (admin / creator / restricted), they
// could simply be added from the last messages authors.
//
//++_offset;
}
sortByOnline();
return true;
}
void ParticipantsBoxController::rowClicked(not_null<PeerListRow*> row) {
auto user = row->peer()->asUser();
Expects(user != nullptr);
if (_role == Role::Admins) {
showAdmin(user);
} else if (_role == Role::Restricted || _role == Role::Kicked) {
showRestricted(user);
} else {
_navigation->showPeerInfo(row->peer());
}
}
void ParticipantsBoxController::rowActionClicked(not_null<PeerListRow*> row) {
auto user = row->peer()->asUser();
Expects(user != nullptr);
if (_role == Role::Members || _role == Role::Profile) {
kickMember(user);
} else if (_role == Role::Admins) {
removeAdmin(user);
} else if (_role == Role::Restricted) {
showRestricted(user);
} else {
removeKicked(row, user);
}
}
bool ParticipantsBoxController::canEditAdminByRights(
not_null<UserData*> user) const {
if (_additional.adminRights.find(user)
!= _additional.adminRights.cend()) {
return (_additional.adminCanEdit.find(user)
!= _additional.adminCanEdit.cend());
}
return (user != _additional.creator);
}
bool ParticipantsBoxController::canEditAdmin(
not_null<UserData*> user) const {
if (user->isSelf()) {
return false;
} else if (_channel->amCreator()) {
return true;
} else if (!canEditAdminByRights(user)) {
return false;
}
return _channel->adminRights() & ChannelAdminRight::f_add_admins;
}
bool ParticipantsBoxController::canRestrictUser(
not_null<UserData*> user) const {
if (user->isSelf()) {
return false;
} else if (_channel->amCreator()) {
return true;
} else if (!canEditAdminByRights(user)) {
return false;
}
return _channel->adminRights() & ChannelAdminRight::f_ban_users;
}
Ui::PopupMenu *ParticipantsBoxController::rowContextMenu(
not_null<PeerListRow*> row) {
Expects(row->peer()->isUser());
auto user = row->peer()->asUser();
auto result = new Ui::PopupMenu(nullptr);
result->addAction(
lang(lng_context_view_profile),
[weak = base::make_weak(this), user] {
if (const auto strong = weak.get()) {
strong->_navigation->showPeerInfo(user);
}
});
if (canEditAdmin(user)) {
auto it = _additional.adminRights.find(user);
auto isCreator = (user == _additional.creator);
auto notAdmin = !isCreator && (it == _additional.adminRights.cend());
auto label = lang(notAdmin
? lng_context_promote_admin
: lng_context_edit_permissions);
result->addAction(
label,
[weak = base::make_weak(this), user] {
if (const auto strong = weak.get()) {
strong->showAdmin(user);
}
});
}
if (canRestrictUser(user)) {
auto isGroup = _channel->isMegagroup();
if (isGroup) {
result->addAction(
lang(lng_context_restrict_user),
[weak = base::make_weak(this), user]{
if (const auto strong = weak.get()) {
strong->showRestricted(user);
}
});
}
result->addAction(
lang(isGroup
? lng_context_remove_from_group
: lng_profile_kick),
[weak = base::make_weak(this), user] {
if (auto strong = weak.get()) {
strong->kickMember(user);
}
});
}
return result;
}
void ParticipantsBoxController::showAdmin(not_null<UserData*> user) {
auto it = _additional.adminRights.find(user);
auto isCreator = (user == _additional.creator);
auto notAdmin = !isCreator && (it == _additional.adminRights.cend());
auto currentRights = isCreator
? MTP_channelAdminRights(MTP_flags(~MTPDchannelAdminRights::Flag::f_add_admins | MTPDchannelAdminRights::Flag::f_add_admins))
: notAdmin ? MTP_channelAdminRights(MTP_flags(0)) : it->second;
auto weak = base::make_weak(this);
auto box = Box<EditAdminBox>(_channel, user, currentRights);
auto canEdit = (_additional.adminCanEdit.find(user) != _additional.adminCanEdit.end());
auto canSave = notAdmin ? _channel->canAddAdmins() : canEdit;
if (canSave) {
box->setSaveCallback([channel = _channel.get(), user, weak](const MTPChannelAdminRights &oldRights, const MTPChannelAdminRights &newRights) {
MTP::send(MTPchannels_EditAdmin(channel->inputChannel, user->inputUser, newRights), rpcDone([channel, user, weak, oldRights, newRights](const MTPUpdates &result) {
Auth().api().applyUpdates(result);
channel->applyEditAdmin(user, oldRights, newRights);
if (weak) {
weak->editAdminDone(user, newRights);
}
}));
});
}
_editBox = Ui::show(std::move(box), LayerOption::KeepOther);
}
void ParticipantsBoxController::editAdminDone(
not_null<UserData*> user,
const MTPChannelAdminRights &rights) {
if (_editBox) {
_editBox->closeBox();
}
if (_addBox) {
_addBox->closeBox();
}
auto notAdmin = (rights.c_channelAdminRights().vflags.v == 0);
if (notAdmin) {
_additional.adminRights.erase(user);
_additional.adminPromotedBy.erase(user);
_additional.adminCanEdit.erase(user);
if (_role == Role::Admins) {
removeRow(user);
}
} else {
// It won't be replaced if the entry already exists.
_additional.adminPromotedBy.emplace(user, App::self());
_additional.adminCanEdit.emplace(user);
_additional.adminRights[user] = rights;
_additional.kicked.erase(user);
_additional.restrictedRights.erase(user);
_additional.restrictedBy.erase(user);
if (_role == Role::Admins) {
prependRow(user);
} else if (_role == Role::Kicked || _role == Role::Restricted) {
removeRow(user);
}
}
recomputeTypeFor(user);
delegate()->peerListRefreshRows();
}
void ParticipantsBoxController::showRestricted(not_null<UserData*> user) {
auto it = _additional.restrictedRights.find(user);
auto restrictedRights = (it == _additional.restrictedRights.cend())
? MTP_channelBannedRights(MTP_flags(0), MTP_int(0))
: it->second;
auto weak = base::make_weak(this);
auto hasAdminRights = false;
auto box = Box<EditRestrictedBox>(_channel, user, hasAdminRights, restrictedRights);
if (_channel->canBanMembers()) {
box->setSaveCallback([megagroup = _channel.get(), user, weak](const MTPChannelBannedRights &oldRights, const MTPChannelBannedRights &newRights) {
MTP::send(MTPchannels_EditBanned(megagroup->inputChannel, user->inputUser, newRights), rpcDone([megagroup, user, weak, oldRights, newRights](const MTPUpdates &result) {
Auth().api().applyUpdates(result);
megagroup->applyEditBanned(user, oldRights, newRights);
if (weak) {
weak->editRestrictedDone(user, newRights);
}
}));
});
}
_editBox = Ui::show(std::move(box), LayerOption::KeepOther);
}
void ParticipantsBoxController::editRestrictedDone(not_null<UserData*> user, const MTPChannelBannedRights &rights) {
if (_editBox) {
_editBox->closeBox();
}
if (_addBox) {
_addBox->closeBox();
}
auto notBanned = (rights.c_channelBannedRights().vflags.v == 0);
auto fullBanned = rights.c_channelBannedRights().is_view_messages();
if (notBanned) {
_additional.kicked.erase(user);
_additional.restrictedRights.erase(user);
_additional.restrictedBy.erase(user);
if (_role == Role::Kicked || _role == Role::Restricted) {
removeRow(user);
}
} else {
_additional.adminRights.erase(user);
_additional.adminCanEdit.erase(user);
_additional.adminPromotedBy.erase(user);
_additional.restrictedBy.emplace(user, App::self());
if (fullBanned) {
_additional.kicked.emplace(user);
_additional.restrictedRights.erase(user);
if (_role == Role::Kicked) {
prependRow(user);
} else if (_role == Role::Admins
|| _role == Role::Restricted
|| _role == Role::Members) {
removeRow(user);
}
} else {
_additional.restrictedRights[user] = rights;
_additional.kicked.erase(user);
if (_role == Role::Restricted) {
prependRow(user);
} else if (_role == Role::Kicked
|| _role == Role::Admins
|| _role == Role::Members) {
removeRow(user);
}
}
}
recomputeTypeFor(user);
delegate()->peerListRefreshRows();
}
void ParticipantsBoxController::kickMember(not_null<UserData*> user) {
auto text = (_channel->isMegagroup() ? lng_profile_sure_kick : lng_profile_sure_kick_channel)(lt_user, user->firstName);
auto weak = base::make_weak(this);
_editBox = Ui::show(Box<ConfirmBox>(text, lang(lng_box_remove), [weak, user] {
if (const auto strong = weak.get()) {
strong->kickMemberSure(user);
}
}), LayerOption::KeepOther);
}
void ParticipantsBoxController::kickMemberSure(not_null<UserData*> user) {
if (_editBox) {
_editBox->closeBox();
}
auto alreadyIt = _additional.restrictedRights.find(user);
auto currentRights = (alreadyIt == _additional.restrictedRights.cend())
? MTP_channelBannedRights(MTP_flags(0), MTP_int(0))
: alreadyIt->second;
if (auto row = delegate()->peerListFindRow(user->id)) {
delegate()->peerListRemoveRow(row);
delegate()->peerListRefreshRows();
}
Auth().api().kickParticipant(_channel, user, currentRights);
}
void ParticipantsBoxController::removeAdmin(not_null<UserData*> user) {
const auto text = lng_profile_sure_remove_admin(lt_user, user->firstName);
const auto weak = base::make_weak(this);
_editBox = Ui::show(Box<ConfirmBox>(text, lang(lng_box_remove), [=] {
if (const auto strong = weak.get()) {
strong->removeAdminSure(user);
}
}), LayerOption::KeepOther);
}
void ParticipantsBoxController::removeAdminSure(not_null<UserData*> user) {
if (_editBox) {
_editBox->closeBox();
}
const auto oldRightsIt = _additional.adminRights.find(user);
if (oldRightsIt == _additional.adminRights.cend()) {
return;
}
const auto weak = base::make_weak(this);
RemoveAdmin(_channel, user, oldRightsIt->second, [=] {
if (const auto strong = weak.get()) {
const auto newRights = MTP_channelAdminRights(MTP_flags(0));
strong->editAdminDone(user, newRights);
}
});
}
void ParticipantsBoxController::removeKicked(
not_null<PeerListRow*> row,
not_null<UserData*> user) {
delegate()->peerListRemoveRow(row);
delegate()->peerListRefreshRows();
Auth().api().unblockParticipant(_channel, user);
}
bool ParticipantsBoxController::appendRow(not_null<UserData*> user) {
if (delegate()->peerListFindRow(user->id)) {
recomputeTypeFor(user);
return false;
}
delegate()->peerListAppendRow(createRow(user));
if (_role != Role::Kicked) {
setDescriptionText(QString());
}
return true;
}
bool ParticipantsBoxController::prependRow(not_null<UserData*> user) {
if (auto row = delegate()->peerListFindRow(user->id)) {
recomputeTypeFor(user);
refreshCustomStatus(row);
if (_role == Role::Admins) {
// Perhaps we've added a new admin from search.
delegate()->peerListPrependRowFromSearchResult(row);
}
return false;
}
delegate()->peerListPrependRow(createRow(user));
if (_role != Role::Kicked) {
setDescriptionText(QString());
}
return true;
}
bool ParticipantsBoxController::removeRow(not_null<UserData*> user) {
if (auto row = delegate()->peerListFindRow(user->id)) {
if (_role == Role::Admins) {
// Perhaps we are removing an admin from search results.
row->setCustomStatus(lang(lng_channel_admin_status_not_admin));
delegate()->peerListConvertRowToSearchResult(row);
} else {
delegate()->peerListRemoveRow(row);
}
if (_role != Role::Kicked
&& !delegate()->peerListFullRowsCount()) {
setDescriptionText(lang(lng_blocked_list_not_found));
}
return true;
}
return false;
}
std::unique_ptr<PeerListRow> ParticipantsBoxController::createRow(
not_null<UserData*> user) const {
if (_role == Role::Profile) {
return std::make_unique<Row>(user, computeType(user));
}
auto row = std::make_unique<PeerListRowWithLink>(user);
refreshCustomStatus(row.get());
if (_role == Role::Admins
&& canEditAdminByRights(user)
&& _additional.adminRights.find(user)
!= _additional.adminRights.cend()) {
row->setActionLink(lang(lng_profile_kick));
} else if (_role == Role::Restricted
|| (_role == Role::Admins
&& _additional.adminCanEdit.find(user)
!= _additional.adminCanEdit.cend())) {
// row->setActionLink(lang(lng_profile_edit_permissions));
} else if (_role == Role::Kicked) {
row->setActionLink(lang(lng_blocked_list_unblock));
} else if (_role == Role::Members) {
if (_channel->canBanMembers() && _additional.creator != user
&& (_additional.adminRights.find(user) == _additional.adminRights.cend()
|| _additional.adminCanEdit.find(user) != _additional.adminCanEdit.cend())) {
row->setActionLink(lang(lng_profile_kick));
}
}
return std::move(row);
}
auto ParticipantsBoxController::computeType(
not_null<UserData*> user) const -> Type {
auto isCreator = (user == _additional.creator);
auto isAdmin = (_additional.adminRights.find(user) != _additional.adminRights.cend());
auto result = Type();
result.rights = isCreator
? Rights::Creator
: isAdmin
? Rights::Admin
: Rights::Normal;
result.canRemove = canRestrictUser(user);
return result;
}
void ParticipantsBoxController::recomputeTypeFor(
not_null<UserData*> user) {
if (_role != Role::Profile) {
return;
}
if (auto row = delegate()->peerListFindRow(user->id)) {
static_cast<Row*>(row)->setType(computeType(user));
}
}
void ParticipantsBoxController::refreshCustomStatus(not_null<PeerListRow*> row) const {
auto user = row->peer()->asUser();
if (_role == Role::Admins) {
auto promotedBy = _additional.adminPromotedBy.find(user);
if (promotedBy == _additional.adminPromotedBy.cend()) {
if (user == _additional.creator) {
row->setCustomStatus(lang(lng_channel_admin_status_creator));
} else {
row->setCustomStatus(lang(lng_channel_admin_status_not_admin));
}
} else {
row->setCustomStatus(lng_channel_admin_status_promoted_by(lt_user, App::peerName(promotedBy->second)));
}
} else if (_role == Role::Kicked || _role == Role::Restricted) {
auto restrictedBy = _additional.restrictedBy.find(user);
if (restrictedBy == _additional.restrictedBy.cend()) {
row->setCustomStatus(lng_channel_banned_status_restricted_by(lt_user, "Unknown"));
} else {
row->setCustomStatus(lng_channel_banned_status_restricted_by(lt_user, App::peerName(restrictedBy->second)));
}
}
}
ParticipantsBoxSearchController::ParticipantsBoxSearchController(not_null<ChannelData*> channel, Role role, not_null<Additional*> additional)
: _channel(channel)
, _role(role)
, _additional(additional) {
_timer.setCallback([this] { searchOnServer(); });
}
void ParticipantsBoxSearchController::searchQuery(const QString &query) {
if (_query != query) {
_query = query;
_offset = 0;
_requestId = 0;
_allLoaded = false;
if (!_query.isEmpty() && !searchInCache()) {
_timer.callOnce(AutoSearchTimeout);
} else {
_timer.cancel();
}
}
}
auto ParticipantsBoxSearchController::saveState() const
-> std::unique_ptr<SavedStateBase> {
auto result = std::make_unique<SavedState>();
result->query = _query;
result->offset = _offset;
result->allLoaded = _allLoaded;
result->wasLoading = (_requestId != 0);
return std::move(result);
}
void ParticipantsBoxSearchController::restoreState(
std::unique_ptr<SavedStateBase> state) {
if (auto my = dynamic_cast<SavedState*>(state.get())) {
if (auto requestId = base::take(_requestId)) {
request(requestId).cancel();
}
_cache.clear();
_queries.clear();
_allLoaded = my->allLoaded;
_offset = my->offset;
_query = my->query;
if (my->wasLoading) {
searchOnServer();
}
}
}
void ParticipantsBoxSearchController::searchOnServer() {
Expects(!_query.isEmpty());
loadMoreRows();
}
bool ParticipantsBoxSearchController::isLoading() {
return _timer.isActive() || _requestId;
}
bool ParticipantsBoxSearchController::searchInCache() {
auto it = _cache.find(_query);
if (it != _cache.cend()) {
_requestId = 0;
searchDone(_requestId, it->second.result, it->second.requestedCount);
return true;
}
return false;
}
bool ParticipantsBoxSearchController::loadMoreRows() {
if (_query.isEmpty()) {
return false;
}
if (!_allLoaded && !isLoading()) {
auto filter = [this] {
switch (_role) {
case Role::Admins: // Search for members, appoint as admin on found.
case Role::Profile:
case Role::Members: return MTP_channelParticipantsSearch(MTP_string(_query));
case Role::Restricted: return MTP_channelParticipantsBanned(MTP_string(_query));
case Role::Kicked: return MTP_channelParticipantsKicked(MTP_string(_query));
}
Unexpected("Role in ParticipantsBoxSearchController::loadMoreRows()");
}();
// For search we request a lot of rows from the first query.
// (because we've waited for search request by timer already,
// so we don't expect it to be fast, but we want to fill cache).
auto perPage = kParticipantsPerPage;
auto participantsHash = 0;
_requestId = request(MTPchannels_GetParticipants(
_channel->inputChannel,
filter,
MTP_int(_offset),
MTP_int(perPage),
MTP_int(participantsHash)
)).done([this, perPage](const MTPchannels_ChannelParticipants &result, mtpRequestId requestId) {
searchDone(requestId, result, perPage);
}).fail([this](const RPCError &error, mtpRequestId requestId) {
if (_requestId == requestId) {
_requestId = 0;
_allLoaded = true;
delegate()->peerListSearchRefreshRows();
}
}).send();
auto entry = Query();
entry.text = _query;
entry.offset = _offset;
_queries.emplace(_requestId, entry);
}
return true;
}
void ParticipantsBoxSearchController::searchDone(
mtpRequestId requestId,
const MTPchannels_ChannelParticipants &result,
int requestedCount) {
auto query = _query;
if (requestId) {
Auth().api().parseChannelParticipants(_channel, result, [&](auto&&...) {
auto it = _queries.find(requestId);
if (it != _queries.cend()) {
query = it->second.text;
if (it->second.offset == 0) {
auto &entry = _cache[query];
entry.result = result;
entry.requestedCount = requestedCount;
}
_queries.erase(it);
}
});
}
if (_requestId != requestId) {
return;
}
_requestId = 0;
TLHelp::VisitChannelParticipants(result, base::overload([&](
const MTPDchannels_channelParticipants &data) {
auto &list = data.vparticipants.v;
if (list.size() < requestedCount) {
// We want cache to have full information about a query with small
// results count (if we don't need the second request). So we don't
// wait for an empty results list unlike the non-search peer list.
_allLoaded = true;
}
auto parseRole = (_role == Role::Admins) ? Role::Members : _role;
for_const (auto &participant, list) {
ParticipantsBoxController::HandleParticipant(participant, parseRole, _additional, [this](not_null<UserData*> user) {
delegate()->peerListSearchAddRow(user);
});
}
_offset += list.size();
}, [&](mtpTypeId type) {
_allLoaded = true;
}));
delegate()->peerListSearchRefreshRows();
}
AddParticipantBoxController::AddParticipantBoxController(not_null<ChannelData*> channel, Role role, AdminDoneCallback adminDoneCallback, BannedDoneCallback bannedDoneCallback) : PeerListController(std::make_unique<AddParticipantBoxSearchController>(channel, &_additional))
, _channel(channel)
, _role(role)
, _adminDoneCallback(std::move(adminDoneCallback))
, _bannedDoneCallback(std::move(bannedDoneCallback)) {
if (_channel->mgInfo) {
_additional.creator = _channel->mgInfo->creator;
}
}
std::unique_ptr<PeerListRow> AddParticipantBoxController::createSearchRow(not_null<PeerData*> peer) {
if (peer->isSelf()) {
return nullptr;
}
if (auto user = peer->asUser()) {
return createRow(user);
}
return nullptr;
}
void AddParticipantBoxController::prepare() {
delegate()->peerListSetSearchMode(PeerListSearchMode::Enabled);
auto title = [this] {
switch (_role) {
case Role::Admins: return langFactory(lng_channel_add_admin);
case Role::Restricted: return langFactory(lng_channel_add_restricted);
case Role::Kicked: return langFactory(lng_channel_add_banned);
}
Unexpected("Role in AddParticipantBoxController::prepare()");
};
delegate()->peerListSetTitle(title());
setDescriptionText(lang(lng_contacts_loading));
setSearchNoResultsText(lang(lng_blocked_list_not_found));
delegate()->peerListRefreshRows();
loadMoreRows();
}
void AddParticipantBoxController::loadMoreRows() {
if (searchController() && searchController()->loadMoreRows()) {
return;
}
if (_loadRequestId || _allLoaded) {
return;
}
// First query is small and fast, next loads a lot of rows.
auto perPage = (_offset > 0) ? kParticipantsPerPage : kParticipantsFirstPageCount;
auto participantsHash = 0;
_loadRequestId = request(MTPchannels_GetParticipants(
_channel->inputChannel,
MTP_channelParticipantsRecent(),
MTP_int(_offset),
MTP_int(perPage),
MTP_int(participantsHash)
)).done([this](const MTPchannels_ChannelParticipants &result) {
_loadRequestId = 0;
Auth().api().parseChannelParticipants(_channel, result, [&](
int availableCount,
const QVector<MTPChannelParticipant> &list) {
for (auto &participant : list) {
HandleParticipant(
participant,
&_additional,
[this](auto user) { appendRow(user); });
}
if (auto size = list.size()) {
_offset += size;
} else {
// To be sure - wait for a whole empty result list.
_allLoaded = true;
}
});
if (delegate()->peerListFullRowsCount() > 0) {
setDescriptionText(QString());
} else if (_allLoaded) {
setDescriptionText(lang(lng_blocked_list_not_found));
}
delegate()->peerListRefreshRows();
}).fail([this](const RPCError &error) {
_loadRequestId = 0;
}).send();
}
void AddParticipantBoxController::rowClicked(not_null<PeerListRow*> row) {
auto user = row->peer()->asUser();
switch (_role) {
case Role::Admins: return showAdmin(user);
case Role::Restricted: return showRestricted(user);
case Role::Kicked: return kickUser(user);
}
Unexpected("Role in AddParticipantBoxController::rowClicked()");
}
template <typename Callback>
bool AddParticipantBoxController::checkInfoLoaded(not_null<UserData*> user, Callback callback) {
if (_additional.infoNotLoaded.find(user) == _additional.infoNotLoaded.end()) {
return true;
}
// We don't know what this user status is in the group.
request(MTPchannels_GetParticipant(_channel->inputChannel, user->inputUser)).done([this, user, callback](const MTPchannels_ChannelParticipant &result) {
Expects(result.type() == mtpc_channels_channelParticipant);
auto &participant = result.c_channels_channelParticipant();
App::feedUsers(participant.vusers);
HandleParticipant(participant.vparticipant, &_additional, [](not_null<UserData*>) {});
_additional.infoNotLoaded.erase(user);
callback();
}).fail([this, user, callback](const RPCError &error) {
_additional.infoNotLoaded.erase(user);
_additional.external.emplace(user);
callback();
}).send();
return false;
}
void AddParticipantBoxController::showAdmin(not_null<UserData*> user, bool sure) {
if (!checkInfoLoaded(user, [this, user] { showAdmin(user); })) {
return;
}
if (sure && _editBox) {
// Close the confirmation box.
_editBox->closeBox();
}
// Check restrictions.
auto weak = base::make_weak(this);
auto alreadyIt = _additional.adminRights.find(user);
auto currentRights = (_additional.creator == user)
? MTP_channelAdminRights(MTP_flags(~MTPDchannelAdminRights::Flag::f_add_admins | MTPDchannelAdminRights::Flag::f_add_admins))
: MTP_channelAdminRights(MTP_flags(0));
if (alreadyIt != _additional.adminRights.end()) {
// The user is already an admin.
currentRights = alreadyIt->second;
} else if (_additional.kicked.find(user) != _additional.kicked.end()) {
// The user is banned.
if (_channel->canAddMembers()) {
if (_channel->canBanMembers()) {
if (!sure) {
_editBox = Ui::show(Box<ConfirmBox>(lang(lng_sure_add_admin_unban), [weak, user] {
if (weak) {
weak->showAdmin(user, true);
}
}), LayerOption::KeepOther);
return;
}
} else {
Ui::show(Box<InformBox>(
lang(lng_error_cant_add_admin_unban)),
LayerOption::KeepOther);
return;
}
} else {
Ui::show(Box<InformBox>(
lang(lng_error_cant_add_admin_invite)),
LayerOption::KeepOther);
return;
}
} else if (_additional.restrictedRights.find(user) != _additional.restrictedRights.end()) {
// The user is restricted.
if (_channel->canBanMembers()) {
if (!sure) {
_editBox = Ui::show(Box<ConfirmBox>(lang(lng_sure_add_admin_unban), [weak, user] {
if (weak) {
weak->showAdmin(user, true);
}
}), LayerOption::KeepOther);
return;
}
} else {
Ui::show(Box<InformBox>(
lang(lng_error_cant_add_admin_unban)),
LayerOption::KeepOther);
return;
}
} else if (_additional.external.find(user) != _additional.external.end()) {
// The user is not in the group yet.
if (_channel->canAddMembers()) {
if (!sure) {
const auto text = lang(_channel->isMegagroup()
? lng_sure_add_admin_invite
: lng_sure_add_admin_invite_channel);
_editBox = Ui::show(Box<ConfirmBox>(text, [weak, user] {
if (weak) {
weak->showAdmin(user, true);
}
}), LayerOption::KeepOther);
return;
}
} else {
Ui::show(
Box<InformBox>(lang(lng_error_cant_add_admin_invite)),
LayerOption::KeepOther);
return;
}
}
// Finally show the admin.
auto canNotEdit = (_additional.creator == user)
|| ((alreadyIt != _additional.adminRights.end())
&& (_additional.adminCanEdit.find(user) == _additional.adminCanEdit.end()));
auto box = Box<EditAdminBox>(_channel, user, currentRights);
if (!canNotEdit) {
box->setSaveCallback([channel = _channel.get(), user, weak](const MTPChannelAdminRights &oldRights, const MTPChannelAdminRights &newRights) {
MTP::send(MTPchannels_EditAdmin(channel->inputChannel, user->inputUser, newRights), rpcDone([channel, user, weak, oldRights, newRights](const MTPUpdates &result) {
Auth().api().applyUpdates(result);
channel->applyEditAdmin(user, oldRights, newRights);
if (weak) {
weak->editAdminDone(user, newRights);
}
}), rpcFail([channel, weak](const RPCError &error) {
if (MTP::isDefaultHandledError(error)) {
return false;
}
if (error.type() == qstr("USER_NOT_MUTUAL_CONTACT")) {
Ui::show(
Box<InformBox>(PeerFloodErrorText(
channel->isMegagroup()
? PeerFloodType::InviteGroup
: PeerFloodType::InviteChannel)),
LayerOption::KeepOther);
} else if (error.type() == qstr("BOT_GROUPS_BLOCKED")) {
Ui::show(
Box<InformBox>(lang(lng_error_cant_add_bot)),
LayerOption::KeepOther);
}
if (weak && weak->_editBox) {
weak->_editBox->closeBox();
}
return true;
}));
});
}
_editBox = Ui::show(std::move(box), LayerOption::KeepOther);
}
void AddParticipantBoxController::editAdminDone(
not_null<UserData*> user,
const MTPChannelAdminRights &rights) {
if (_editBox) _editBox->closeBox();
_additional.restrictedRights.erase(user);
_additional.restrictedBy.erase(user);
_additional.kicked.erase(user);
_additional.external.erase(user);
if (rights.c_channelAdminRights().vflags.v == 0) {
_additional.adminRights.erase(user);
_additional.adminPromotedBy.erase(user);
_additional.adminCanEdit.erase(user);
} else {
_additional.adminRights[user] = rights;
_additional.adminCanEdit.emplace(user);
auto it = _additional.adminPromotedBy.find(user);
if (it == _additional.adminPromotedBy.end()) {
_additional.adminPromotedBy.emplace(user, App::self());
}
}
if (_adminDoneCallback) {
_adminDoneCallback(user, rights);
}
}
void AddParticipantBoxController::showRestricted(not_null<UserData*> user, bool sure) {
if (!checkInfoLoaded(user, [this, user] { showRestricted(user); })) {
return;
}
if (sure && _editBox) {
// Close the confirmation box.
_editBox->closeBox();
}
// Check restrictions.
auto weak = base::make_weak(this);
auto alreadyIt = _additional.restrictedRights.find(user);
auto currentRights = MTP_channelBannedRights(MTP_flags(0), MTP_int(0));
auto hasAdminRights = false;
if (alreadyIt != _additional.restrictedRights.end()) {
// The user is already banned or restricted.
currentRights = alreadyIt->second;
} else if (_additional.adminRights.find(user) != _additional.adminRights.end() || _additional.creator == user) {
// The user is an admin or creator.
if (_additional.adminCanEdit.find(user) != _additional.adminCanEdit.end()) {
hasAdminRights = true;
if (!sure) {
_editBox = Ui::show(Box<ConfirmBox>(lang(lng_sure_ban_admin), [weak, user] {
if (weak) {
weak->showRestricted(user, true);
}
}), LayerOption::KeepOther);
return;
}
} else {
Ui::show(
Box<InformBox>(lang(lng_error_cant_ban_admin)),
LayerOption::KeepOther);
return;
}
}
// Finally edit the restricted.
auto box = Box<EditRestrictedBox>(_channel, user, hasAdminRights, currentRights);
box->setSaveCallback([user, weak](const MTPChannelBannedRights &oldRights, const MTPChannelBannedRights &newRights) {
if (weak) {
weak->restrictUserSure(user, oldRights, newRights);
}
});
_editBox = Ui::show(std::move(box), LayerOption::KeepOther);
}
void AddParticipantBoxController::restrictUserSure(not_null<UserData*> user, const MTPChannelBannedRights &oldRights, const MTPChannelBannedRights &newRights) {
auto weak = base::make_weak(this);
MTP::send(MTPchannels_EditBanned(_channel->inputChannel, user->inputUser, newRights), rpcDone([megagroup = _channel.get(), user, weak, oldRights, newRights](const MTPUpdates &result) {
Auth().api().applyUpdates(result);
megagroup->applyEditBanned(user, oldRights, newRights);
if (const auto strong = weak.get()) {
strong->editRestrictedDone(user, newRights);
}
}));
}
void AddParticipantBoxController::editRestrictedDone(
not_null<UserData*> user,
const MTPChannelBannedRights &rights) {
if (_editBox) _editBox->closeBox();
_additional.adminRights.erase(user);
_additional.adminCanEdit.erase(user);
_additional.adminPromotedBy.erase(user);
if (rights.c_channelBannedRights().vflags.v == 0) {
_additional.restrictedRights.erase(user);
_additional.restrictedBy.erase(user);
_additional.kicked.erase(user);
} else {
_additional.restrictedRights[user] = rights;
if (rights.c_channelBannedRights().vflags.v & MTPDchannelBannedRights::Flag::f_view_messages) {
_additional.kicked.emplace(user);
} else {
_additional.kicked.erase(user);
}
_additional.restrictedBy.emplace(user, App::self());
}
if (_bannedDoneCallback) {
_bannedDoneCallback(user, rights);
}
}
void AddParticipantBoxController::kickUser(not_null<UserData*> user, bool sure) {
if (!checkInfoLoaded(user, [this, user] { kickUser(user); })) {
return;
}
// Check restrictions.
auto weak = base::make_weak(this);
if (_additional.adminRights.find(user) != _additional.adminRights.end() || _additional.creator == user) {
// The user is an admin or creator.
if (_additional.adminCanEdit.find(user) != _additional.adminCanEdit.end()) {
if (!sure) {
_editBox = Ui::show(Box<ConfirmBox>(lang(lng_sure_ban_admin), [weak, user] {
if (weak) {
weak->kickUser(user, true);
}
}), LayerOption::KeepOther);
return;
}
} else {
Ui::show(
Box<InformBox>(lang(lng_error_cant_ban_admin)),
LayerOption::KeepOther);
return;
}
}
// Finally kick him.
if (!sure) {
auto text = lng_sure_ban_user_group(lt_user, App::peerName(user));
_editBox = Ui::show(Box<ConfirmBox>(text, [weak, user] {
if (weak) {
weak->kickUser(user, true);
}
}), LayerOption::KeepOther);
return;
}
auto currentRights = MTP_channelBannedRights(MTP_flags(0), MTP_int(0));
auto alreadyIt = _additional.restrictedRights.find(user);
if (alreadyIt != _additional.restrictedRights.end()) {
// The user is already banned or restricted.
currentRights = alreadyIt->second;
}
restrictUserSure(user, currentRights, ChannelData::KickedRestrictedRights());
}
bool AddParticipantBoxController::appendRow(not_null<UserData*> user) {
if (delegate()->peerListFindRow(user->id) || user->isSelf()) {
return false;
}
delegate()->peerListAppendRow(createRow(user));
return true;
}
bool AddParticipantBoxController::prependRow(not_null<UserData*> user) {
if (delegate()->peerListFindRow(user->id)) {
return false;
}
delegate()->peerListPrependRow(createRow(user));
return true;
}
std::unique_ptr<PeerListRow> AddParticipantBoxController::createRow(not_null<UserData*> user) const {
return std::make_unique<PeerListRow>(user);
}
template <typename Callback>
void AddParticipantBoxController::HandleParticipant(
const MTPChannelParticipant &participant,
not_null<Additional*> additional,
Callback callback) {
switch (participant.type()) {
case mtpc_channelParticipantAdmin: {
auto &admin = participant.c_channelParticipantAdmin();
if (auto user = App::userLoaded(admin.vuser_id.v)) {
additional->infoNotLoaded.erase(user);
additional->restrictedRights.erase(user);
additional->kicked.erase(user);
additional->restrictedBy.erase(user);
additional->adminRights[user] = admin.vadmin_rights;
if (admin.is_can_edit()) {
additional->adminCanEdit.emplace(user);
} else {
additional->adminCanEdit.erase(user);
}
if (auto promoted = App::userLoaded(admin.vpromoted_by.v)) {
auto it = additional->adminPromotedBy.find(user);
if (it == additional->adminPromotedBy.end()) {
additional->adminPromotedBy.emplace(user, promoted);
} else {
it->second = promoted;
}
} else {
LOG(("API Error: No user %1 for admin promoted by.").arg(admin.vpromoted_by.v));
}
callback(user);
}
} break;
case mtpc_channelParticipantCreator: {
auto &creator = participant.c_channelParticipantCreator();
if (auto user = App::userLoaded(creator.vuser_id.v)) {
additional->infoNotLoaded.erase(user);
additional->creator = user;
callback(user);
}
} break;
case mtpc_channelParticipantBanned: {
auto &banned = participant.c_channelParticipantBanned();
if (auto user = App::userLoaded(banned.vuser_id.v)) {
additional->infoNotLoaded.erase(user);
additional->adminRights.erase(user);
additional->adminCanEdit.erase(user);
additional->adminPromotedBy.erase(user);
if (banned.is_left()) {
additional->kicked.emplace(user);
} else {
additional->kicked.erase(user);
}
additional->restrictedRights[user] = banned.vbanned_rights;
if (auto kickedby = App::userLoaded(banned.vkicked_by.v)) {
auto it = additional->restrictedBy.find(user);
if (it == additional->restrictedBy.end()) {
additional->restrictedBy.emplace(user, kickedby);
} else {
it->second = kickedby;
}
}
callback(user);
}
} break;
case mtpc_channelParticipant: {
auto &data = participant.c_channelParticipant();
if (auto user = App::userLoaded(data.vuser_id.v)) {
additional->infoNotLoaded.erase(user);
additional->adminRights.erase(user);
additional->adminCanEdit.erase(user);
additional->adminPromotedBy.erase(user);
additional->restrictedRights.erase(user);
additional->kicked.erase(user);
additional->restrictedBy.erase(user);
callback(user);
}
} break;
default: Unexpected("Participant type in AddParticipantBoxController::HandleParticipant()");
}
}
AddParticipantBoxSearchController::AddParticipantBoxSearchController(not_null<ChannelData*> channel, not_null<Additional*> additional)
: _channel(channel)
, _additional(additional) {
_timer.setCallback([this] { searchOnServer(); });
}
void AddParticipantBoxSearchController::searchQuery(const QString &query) {
if (_query != query) {
_query = query;
_offset = 0;
_requestId = 0;
_participantsLoaded = false;
_chatsContactsAdded = false;
_globalLoaded = false;
if (!_query.isEmpty() && !searchParticipantsInCache()) {
_timer.callOnce(AutoSearchTimeout);
} else {
_timer.cancel();
}
}
}
void AddParticipantBoxSearchController::searchOnServer() {
Expects(!_query.isEmpty());
loadMoreRows();
}
bool AddParticipantBoxSearchController::isLoading() {
return _timer.isActive() || _requestId;
}
bool AddParticipantBoxSearchController::searchParticipantsInCache() {
auto it = _participantsCache.find(_query);
if (it != _participantsCache.cend()) {
_requestId = 0;
searchParticipantsDone(_requestId, it->second.result, it->second.requestedCount);
return true;
}
return false;
}
bool AddParticipantBoxSearchController::searchGlobalInCache() {
auto it = _globalCache.find(_query);
if (it != _globalCache.cend()) {
_requestId = 0;
searchGlobalDone(_requestId, it->second);
return true;
}
return false;
}
bool AddParticipantBoxSearchController::loadMoreRows() {
if (_query.isEmpty()) {
return false;
}
if (_globalLoaded) {
return true;
}
if (_participantsLoaded) {
if (!_chatsContactsAdded) {
addChatsContacts();
}
if (!isLoading() && !searchGlobalInCache()) {
requestGlobal();
}
} else if (!isLoading()) {
requestParticipants();
}
return true;
}
void AddParticipantBoxSearchController::requestParticipants() {
// For search we request a lot of rows from the first query.
// (because we've waited for search request by timer already,
// so we don't expect it to be fast, but we want to fill cache).
auto perPage = kParticipantsPerPage;
auto participantsHash = 0;
_requestId = request(MTPchannels_GetParticipants(
_channel->inputChannel,
MTP_channelParticipantsSearch(MTP_string(_query)),
MTP_int(_offset),
MTP_int(perPage),
MTP_int(participantsHash)
)).done([this, perPage](const MTPchannels_ChannelParticipants &result, mtpRequestId requestId) {
searchParticipantsDone(requestId, result, perPage);
}).fail([this](const RPCError &error, mtpRequestId requestId) {
if (_requestId == requestId) {
_requestId = 0;
_participantsLoaded = true;
loadMoreRows();
delegate()->peerListSearchRefreshRows();
}
}).send();
auto entry = Query();
entry.text = _query;
entry.offset = _offset;
_participantsQueries.emplace(_requestId, entry);
}
void AddParticipantBoxSearchController::searchParticipantsDone(mtpRequestId requestId, const MTPchannels_ChannelParticipants &result, int requestedCount) {
auto query = _query;
if (requestId) {
Auth().api().parseChannelParticipants(_channel, result, [&](auto&&...) {
auto it = _participantsQueries.find(requestId);
if (it != _participantsQueries.cend()) {
query = it->second.text;
if (it->second.offset == 0) {
auto &entry = _participantsCache[query];
entry.result = result;
entry.requestedCount = requestedCount;
}
_participantsQueries.erase(it);
}
});
}
if (_requestId != requestId) {
return;
}
_requestId = 0;
TLHelp::VisitChannelParticipants(result, base::overload([&](
const MTPDchannels_channelParticipants &data) {
auto &list = data.vparticipants.v;
if (list.size() < requestedCount) {
// We want cache to have full information about a query with small
// results count (if we don't need the second request). So we don't
// wait for an empty results list unlike the non-search peer list.
_participantsLoaded = true;
if (list.empty() && _offset == 0) {
// No results, so we want to request global search immediately.
loadMoreRows();
}
}
auto addUser = [&](auto user) {
delegate()->peerListSearchAddRow(user);
};
for (auto &participant : list) {
AddParticipantBoxController::HandleParticipant(
participant,
_additional,
addUser);
}
_offset += list.size();
}, [&](mtpTypeId type) {
_participantsLoaded = true;
}));
delegate()->peerListSearchRefreshRows();
}
void AddParticipantBoxSearchController::requestGlobal() {
if (_query.isEmpty()) {
_globalLoaded = true;
return;
}
auto perPage = SearchPeopleLimit;
_requestId = request(MTPcontacts_Search(
MTP_string(_query),
MTP_int(perPage)
)).done([=](const MTPcontacts_Found &result, mtpRequestId requestId) {
searchGlobalDone(requestId, result);
}).fail([=](const RPCError &error, mtpRequestId requestId) {
if (_requestId == requestId) {
_requestId = 0;
_globalLoaded = true;
delegate()->peerListSearchRefreshRows();
}
}).send();
_globalQueries.emplace(_requestId, _query);
}
void AddParticipantBoxSearchController::searchGlobalDone(mtpRequestId requestId, const MTPcontacts_Found &result) {
Expects(result.type() == mtpc_contacts_found);
auto &found = result.c_contacts_found();
auto query = _query;
if (requestId) {
App::feedUsers(found.vusers);
App::feedChats(found.vchats);
auto it = _globalQueries.find(requestId);
if (it != _globalQueries.cend()) {
query = it->second;
_globalCache[query] = result;
_globalQueries.erase(it);
}
}
const auto feedList = [&](const MTPVector<MTPPeer> &list) {
const auto contains = [](const auto &map, const auto &value) {
return map.find(value) != map.end();
};
for (const auto &mtpPeer : list.v) {
const auto peerId = peerFromMTP(mtpPeer);
if (const auto peer = App::peerLoaded(peerId)) {
if (const auto user = peer->asUser()) {
if (_additional->creator != user
&& !contains(_additional->adminRights, user)
&& !contains(_additional->restrictedRights, user)
&& !contains(_additional->external, user)
&& !contains(_additional->kicked, user)) {
_additional->infoNotLoaded.emplace(user);
}
delegate()->peerListSearchAddRow(user);
}
}
}
};
if (_requestId == requestId) {
_requestId = 0;
_globalLoaded = true;
feedList(found.vmy_results);
feedList(found.vresults);
delegate()->peerListSearchRefreshRows();
}
}
void AddParticipantBoxSearchController::addChatsContacts() {
_chatsContactsAdded = true;
auto wordList = TextUtilities::PrepareSearchWords(_query);
if (wordList.empty()) {
return;
}
auto getSmallestIndex = [&](Dialogs::IndexedList *list) -> const Dialogs::List* {
if (list->isEmpty()) {
return nullptr;
}
auto result = (const Dialogs::List*)nullptr;
for_const (auto &word, wordList) {
auto found = list->filtered(word[0]);
if (found->isEmpty()) {
return nullptr;
}
if (!result || result->size() > found->size()) {
result = found;
}
}
return result;
};
auto dialogsIndex = getSmallestIndex(App::main()->dialogsList());
auto contactsIndex = getSmallestIndex(App::main()->contactsNoDialogsList());
auto allWordsAreFound = [&](const base::flat_set<QString> &nameWords) {
auto hasNamePartStartingWith = [&](const QString &word) {
for (auto &nameWord : nameWords) {
if (nameWord.startsWith(word)) {
return true;
}
}
return false;
};
for_const (auto &word, wordList) {
if (!hasNamePartStartingWith(word)) {
return false;
}
}
return true;
};
auto filterAndAppend = [&](const Dialogs::List *list) {
if (!list) {
return;
}
for (const auto row : *list) {
if (const auto history = row->history()) {
if (const auto user = history->peer->asUser()) {
if (allWordsAreFound(user->nameWords())) {
delegate()->peerListSearchAddRow(user);
}
}
}
}
};
filterAndAppend(dialogsIndex);
filterAndAppend(contactsIndex);
delegate()->peerListSearchRefreshRows();
}
} // namespace Profile