/* 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 "data/data_peer.h" #include #include #include "data/data_peer_values.h" #include "data/data_channel_admins.h" #include "data/data_photo.h" #include "data/data_feed.h" #include "data/data_session.h" #include "history/history.h" #include "lang/lang_keys.h" #include "observer_peer.h" #include "mainwidget.h" #include "apiwrap.h" #include "boxes/confirm_box.h" #include "styles/style_history.h" #include "auth_session.h" #include "messenger.h" #include "mainwindow.h" #include "window/window_controller.h" #include "storage/localstorage.h" #include "ui/image/image.h" #include "ui/empty_userpic.h" #include "ui/text_options.h" namespace { constexpr auto kUpdateFullPeerTimeout = TimeMs(5000); // Not more than once in 5 seconds. constexpr auto kUserpicSize = 160; } // namespace namespace Data { int PeerColorIndex(int32 bareId) { const auto index = std::abs(bareId) % 7; const int map[] = { 0, 7, 4, 1, 6, 3, 5 }; return map[index]; } int PeerColorIndex(PeerId peerId) { return PeerColorIndex(peerToBareInt(peerId)); } style::color PeerUserpicColor(PeerId peerId) { const style::color colors[] = { st::historyPeer1UserpicBg, st::historyPeer2UserpicBg, st::historyPeer3UserpicBg, st::historyPeer4UserpicBg, st::historyPeer5UserpicBg, st::historyPeer6UserpicBg, st::historyPeer7UserpicBg, st::historyPeer8UserpicBg, }; return colors[PeerColorIndex(peerId)]; } } // namespace Data using UpdateFlag = Notify::PeerUpdate::Flag; PeerClickHandler::PeerClickHandler(not_null peer) : _peer(peer) { } void PeerClickHandler::onClick(ClickContext context) const { if (context.button == Qt::LeftButton && App::wnd()) { auto controller = App::wnd()->controller(); if (_peer && _peer->isChannel() && controller->activeChatCurrent().peer() != _peer) { if (!_peer->asChannel()->isPublic() && !_peer->asChannel()->amIn()) { Ui::show(Box(lang(_peer->isMegagroup() ? lng_group_not_accessible : lng_channel_not_accessible))); } else { controller->showPeerHistory( _peer, Window::SectionShow::Way::Forward); } } else { Ui::showPeerProfile(_peer); } } } PeerData::PeerData(const PeerId &id) : id(id) , _userpicEmpty(createEmptyUserpic()) { nameText.setText(st::msgNameStyle, QString(), Ui::NameTextOptions()); } void PeerData::updateNameDelayed( const QString &newName, const QString &newNameOrPhone, const QString &newUsername) { if (name == newName) { if (isUser()) { if (asUser()->nameOrPhone == newNameOrPhone && asUser()->username == newUsername) { return; } } else if (isChannel()) { if (asChannel()->username == newUsername) { return; } } else if (isChat()) { return; } } ++nameVersion; name = newName; nameText.setText(st::msgNameStyle, name, Ui::NameTextOptions()); refreshEmptyUserpic(); Notify::PeerUpdate update(this); update.flags |= UpdateFlag::NameChanged; update.oldNameFirstLetters = nameFirstLetters(); if (isUser()) { if (asUser()->username != newUsername) { asUser()->username = newUsername; update.flags |= UpdateFlag::UsernameChanged; } asUser()->setNameOrPhone(newNameOrPhone); } else if (isChannel()) { if (asChannel()->username != newUsername) { asChannel()->username = newUsername; if (newUsername.isEmpty()) { asChannel()->removeFlags( MTPDchannel::Flag::f_username); } else { asChannel()->addFlags(MTPDchannel::Flag::f_username); } update.flags |= UpdateFlag::UsernameChanged; } } fillNames(); Notify::PeerUpdated().notify(update, true); } std::unique_ptr PeerData::createEmptyUserpic() const { return std::make_unique( Data::PeerUserpicColor(id), name); } void PeerData::refreshEmptyUserpic() const { _userpicEmpty = useEmptyUserpic() ? createEmptyUserpic() : nullptr; } ClickHandlerPtr PeerData::createOpenLink() { return std::make_shared(this); } void PeerData::setUserpic( PhotoId photoId, const StorageImageLocation &location, ImagePtr userpic) { _userpicPhotoId = photoId; _userpic = userpic; _userpicLocation = location; refreshEmptyUserpic(); } void PeerData::setUserpicPhoto(const MTPPhoto &data) { const auto photoId = data.match([&](const MTPDphoto &data) { const auto photo = Auth().data().photo(data); photo->peer = this; return photo->id; }, [](const MTPDphotoEmpty &data) { return PhotoId(0); }); if (_userpicPhotoId != photoId) { _userpicPhotoId = photoId; Notify::peerUpdatedDelayed(this, UpdateFlag::PhotoChanged); } } ImagePtr PeerData::currentUserpic() const { if (_userpic) { _userpic->load(userpicOrigin()); if (_userpic->loaded()) { if (!useEmptyUserpic()) { _userpicEmpty = nullptr; } return _userpic; } } if (!_userpicEmpty) { refreshEmptyUserpic(); } return ImagePtr(); } void PeerData::paintUserpic(Painter &p, int x, int y, int size) const { if (auto userpic = currentUserpic()) { p.drawPixmap(x, y, userpic->pixCircled(userpicOrigin(), size, size)); } else { _userpicEmpty->paint(p, x, y, x + size + x, size); } } void PeerData::paintUserpicRounded(Painter &p, int x, int y, int size) const { if (auto userpic = currentUserpic()) { p.drawPixmap(x, y, userpic->pixRounded(userpicOrigin(), size, size, ImageRoundRadius::Small)); } else { _userpicEmpty->paintRounded(p, x, y, x + size + x, size); } } void PeerData::paintUserpicSquare(Painter &p, int x, int y, int size) const { if (auto userpic = currentUserpic()) { p.drawPixmap(x, y, userpic->pix(userpicOrigin(), size, size)); } else { _userpicEmpty->paintSquare(p, x, y, x + size + x, size); } } void PeerData::loadUserpic(bool loadFirst, bool prior) { _userpic->load(userpicOrigin(), loadFirst, prior); } bool PeerData::userpicLoaded() const { return _userpic->loaded(); } bool PeerData::useEmptyUserpic() const { return _userpicLocation.isNull() || !_userpic || !_userpic->loaded(); } StorageKey PeerData::userpicUniqueKey() const { if (useEmptyUserpic()) { if (!_userpicEmpty) { refreshEmptyUserpic(); } return _userpicEmpty->uniqueKey(); } return storageKey(_userpicLocation); } void PeerData::saveUserpic(const QString &path, int size) const { genUserpic(size).save(path, "PNG"); } void PeerData::saveUserpicRounded(const QString &path, int size) const { genUserpicRounded(size).save(path, "PNG"); } QPixmap PeerData::genUserpic(int size) const { if (auto userpic = currentUserpic()) { return userpic->pixCircled(userpicOrigin(), size, size); } auto result = QImage(QSize(size, size) * cIntRetinaFactor(), QImage::Format_ARGB32_Premultiplied); result.setDevicePixelRatio(cRetinaFactor()); result.fill(Qt::transparent); { Painter p(&result); paintUserpic(p, 0, 0, size); } return App::pixmapFromImageInPlace(std::move(result)); } QPixmap PeerData::genUserpicRounded(int size) const { if (auto userpic = currentUserpic()) { return userpic->pixRounded(userpicOrigin(), size, size, ImageRoundRadius::Small); } auto result = QImage(QSize(size, size) * cIntRetinaFactor(), QImage::Format_ARGB32_Premultiplied); result.setDevicePixelRatio(cRetinaFactor()); result.fill(Qt::transparent); { Painter p(&result); paintUserpicRounded(p, 0, 0, size); } return App::pixmapFromImageInPlace(std::move(result)); } void PeerData::updateUserpic( PhotoId photoId, const MTPFileLocation &location) { const auto size = kUserpicSize; const auto loc = StorageImageLocation::FromMTP(size, size, location); const auto photo = loc.isNull() ? ImagePtr() : Images::Create(loc); setUserpicChecked(photoId, loc, photo); } void PeerData::clearUserpic() { const auto photoId = PhotoId(0); const auto loc = StorageImageLocation(); const auto photo = [&] { if (id == peerFromUser(ServiceUserId)) { auto image = Messenger::Instance().logoNoMargin().scaledToWidth( kUserpicSize, Qt::SmoothTransformation); return _userpic ? _userpic : Images::Create(std::move(image), "PNG"); } return ImagePtr(); }(); setUserpicChecked(photoId, loc, photo); } void PeerData::setUserpicChecked( PhotoId photoId, const StorageImageLocation &location, ImagePtr userpic) { if (_userpicPhotoId != photoId || _userpic.get() != userpic.get() || _userpicLocation != location) { setUserpic(photoId, location, userpic); Notify::peerUpdatedDelayed(this, UpdateFlag::PhotoChanged); if (const auto channel = asChannel()) { if (const auto feed = channel->feed()) { Auth().data().notifyFeedUpdated( feed, Data::FeedUpdateFlag::ChannelPhoto); } } } } bool PeerData::canPinMessages() const { if (const auto user = asUser()) { return user->fullFlags() & MTPDuserFull::Flag::f_can_pin_message; } else if (const auto chat = asChat()) { return chat->adminsEnabled() ? chat->amAdmin() : chat->amIn(); } else if (const auto channel = asChannel()) { using AdminRight = ChannelData::AdminRight; if (channel->isMegagroup()) { return (channel->adminRights() & AdminRight::f_pin_messages) || channel->amCreator(); } return (channel->adminRights() & AdminRight::f_edit_messages) || channel->amCreator(); } Unexpected("Peer type in PeerData::canPinMessages."); } void PeerData::setPinnedMessageId(MsgId messageId) { const auto min = [&] { if (const auto channel = asChannel()) { return channel->availableMinId(); } return MsgId(0); }(); messageId = (messageId > min) ? messageId : MsgId(0); if (_pinnedMessageId != messageId) { _pinnedMessageId = messageId; Notify::peerUpdatedDelayed( this, Notify::PeerUpdate::Flag::PinnedMessageChanged); } } void PeerData::fillNames() { _nameWords.clear(); _nameFirstLetters.clear(); auto toIndexList = QStringList(); auto appendToIndex = [&](const QString &value) { if (!value.isEmpty()) { toIndexList.push_back(TextUtilities::RemoveAccents(value)); } }; appendToIndex(name); const auto appendTranslit = !toIndexList.isEmpty() && cRussianLetters().match(toIndexList.front()).hasMatch(); if (appendTranslit) { appendToIndex(translitRusEng(toIndexList.front())); } if (const auto user = asUser()) { if (user->nameOrPhone != name) { appendToIndex(user->nameOrPhone); } appendToIndex(user->username); if (isSelf()) { appendToIndex(lang(lng_saved_messages)); } } else if (const auto channel = asChannel()) { appendToIndex(channel->username); } auto toIndex = toIndexList.join(' '); toIndex += ' ' + rusKeyboardLayoutSwitch(toIndex); const auto namesList = TextUtilities::PrepareSearchWords(toIndex); for (const auto &name : namesList) { _nameWords.insert(name); _nameFirstLetters.insert(name[0]); } } PeerData::~PeerData() = default; const Text &BotCommand::descriptionText() const { if (_descriptionText.isEmpty() && !_description.isEmpty()) { _descriptionText.setText( st::defaultTextStyle, _description, Ui::NameTextOptions()); } return _descriptionText; } bool UserData::canShareThisContact() const { return canShareThisContactFast() || !Auth().data().findContactPhone(peerToUser(id)).isEmpty(); } void UserData::setContactStatus(ContactStatus status) { if (_contactStatus != status) { const auto changed = (_contactStatus == ContactStatus::Contact) != (status == ContactStatus::Contact); _contactStatus = status; if (changed) { Notify::peerUpdatedDelayed( this, Notify::PeerUpdate::Flag::UserIsContact); } } if (_contactStatus == ContactStatus::Contact && cReportSpamStatuses().value(id, dbiprsHidden) != dbiprsHidden) { cRefReportSpamStatuses().insert(id, dbiprsHidden); Local::writeReportSpamStatuses(); } } // see Local::readPeer as well void UserData::setPhoto(const MTPUserProfilePhoto &photo) { if (photo.type() == mtpc_userProfilePhoto) { const auto &data = photo.c_userProfilePhoto(); updateUserpic(data.vphoto_id.v, data.vphoto_small); } else { clearUserpic(); } } bool UserData::setAbout(const QString &newAbout) { if (_about == newAbout) { return false; } _about = newAbout; Notify::peerUpdatedDelayed(this, UpdateFlag::AboutChanged); return true; } void UserData::setRestrictionReason(const QString &text) { if (_restrictionReason != text) { _restrictionReason = text; Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::RestrictionReasonChanged); } } void UserData::setCommonChatsCount(int count) { if (_commonChatsCount != count) { _commonChatsCount = count; Notify::peerUpdatedDelayed(this, UpdateFlag::UserCommonChatsChanged); } } void UserData::setName(const QString &newFirstName, const QString &newLastName, const QString &newPhoneName, const QString &newUsername) { bool changeName = !newFirstName.isEmpty() || !newLastName.isEmpty(); QString newFullName; if (changeName && newFirstName.trimmed().isEmpty()) { firstName = newLastName; lastName = QString(); newFullName = firstName; } else { if (changeName) { firstName = newFirstName; lastName = newLastName; } newFullName = lastName.isEmpty() ? firstName : lng_full_name(lt_first_name, firstName, lt_last_name, lastName); } updateNameDelayed(newFullName, newPhoneName, newUsername); } void UserData::setPhone(const QString &newPhone) { if (_phone != newPhone) { _phone = newPhone; } } void UserData::setBotInfoVersion(int version) { if (version < 0) { if (botInfo) { if (!botInfo->commands.isEmpty()) { botInfo->commands.clear(); Notify::botCommandsChanged(this); } botInfo = nullptr; Notify::userIsBotChanged(this); } } else if (!botInfo) { botInfo = std::make_unique(); botInfo->version = version; Notify::userIsBotChanged(this); } else if (botInfo->version < version) { if (!botInfo->commands.isEmpty()) { botInfo->commands.clear(); Notify::botCommandsChanged(this); } botInfo->description.clear(); botInfo->version = version; botInfo->inited = false; } } void UserData::setBotInfo(const MTPBotInfo &info) { switch (info.type()) { case mtpc_botInfo: { const auto &d(info.c_botInfo()); if (peerFromUser(d.vuser_id.v) != id || !botInfo) return; QString desc = qs(d.vdescription); if (botInfo->description != desc) { botInfo->description = desc; botInfo->text = Text(st::msgMinWidth); } auto &v = d.vcommands.v; botInfo->commands.reserve(v.size()); auto changedCommands = false; int32 j = 0; for (int32 i = 0, l = v.size(); i < l; ++i) { if (v.at(i).type() != mtpc_botCommand) continue; QString cmd = qs(v.at(i).c_botCommand().vcommand), desc = qs(v.at(i).c_botCommand().vdescription); if (botInfo->commands.size() <= j) { botInfo->commands.push_back(BotCommand(cmd, desc)); changedCommands = true; } else { if (botInfo->commands[j].command != cmd) { botInfo->commands[j].command = cmd; changedCommands = true; } if (botInfo->commands[j].setDescription(desc)) { changedCommands = true; } } ++j; } while (j < botInfo->commands.size()) { botInfo->commands.pop_back(); changedCommands = true; } botInfo->inited = true; if (changedCommands) { Notify::botCommandsChanged(this); } } break; } } void UserData::setNameOrPhone(const QString &newNameOrPhone) { if (nameOrPhone != newNameOrPhone) { nameOrPhone = newNameOrPhone; phoneText.setText( st::msgNameStyle, nameOrPhone, Ui::NameTextOptions()); } } void UserData::madeAction(TimeId when) { if (botInfo || isServiceUser(id) || when <= 0) return; if (onlineTill <= 0 && -onlineTill < when) { onlineTill = -when - SetOnlineAfterActivity; Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::UserOnlineChanged); } else if (onlineTill > 0 && onlineTill < when + 1) { onlineTill = when + SetOnlineAfterActivity; Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::UserOnlineChanged); } } void UserData::setAccessHash(uint64 accessHash) { if (accessHash == kInaccessibleAccessHashOld) { _accessHash = 0; // _flags.add(MTPDuser_ClientFlag::f_inaccessible | 0); _flags.add(MTPDuser::Flag::f_deleted); } else { _accessHash = accessHash; } } void UserData::setBlockStatus(BlockStatus blockStatus) { if (blockStatus != _blockStatus) { _blockStatus = blockStatus; Notify::peerUpdatedDelayed(this, UpdateFlag::UserIsBlocked); } } void UserData::setCallsStatus(CallsStatus callsStatus) { if (callsStatus != _callsStatus) { _callsStatus = callsStatus; Notify::peerUpdatedDelayed(this, UpdateFlag::UserHasCalls); } } bool UserData::hasCalls() const { return (callsStatus() != CallsStatus::Disabled) && (callsStatus() != CallsStatus::Unknown); } void ChatData::setPhoto(const MTPChatPhoto &photo) { setPhoto(userpicPhotoId(), photo); } void ChatData::setPhoto(PhotoId photoId, const MTPChatPhoto &photo) { if (photo.type() == mtpc_chatPhoto) { const auto &data = photo.c_chatPhoto(); updateUserpic(photoId, data.vphoto_small); } else { clearUserpic(); } } void ChatData::setName(const QString &newName) { updateNameDelayed(newName.isEmpty() ? name : newName, QString(), QString()); } void ChatData::invalidateParticipants() { auto wasCanEdit = canEdit(); participants.clear(); admins.clear(); removeFlags(MTPDchat::Flag::f_admin); invitedByMe.clear(); botStatus = 0; if (wasCanEdit != canEdit()) { Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::ChatCanEdit); } Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::MembersChanged | Notify::PeerUpdate::Flag::AdminsChanged); } void ChatData::setInviteLink(const QString &newInviteLink) { if (newInviteLink != _inviteLink) { _inviteLink = newInviteLink; Notify::peerUpdatedDelayed(this, UpdateFlag::InviteLinkChanged); } } ChannelData::ChannelData(const PeerId &id) : PeerData(id) , inputChannel(MTP_inputChannel(MTP_int(bareId()), MTP_long(0))) { Data::PeerFlagValue( this, MTPDchannel::Flag::f_megagroup ) | rpl::start_with_next([this](bool megagroup) { if (megagroup) { if (!mgInfo) { mgInfo = std::make_unique(); } } else if (mgInfo) { mgInfo = nullptr; } }, _lifetime); } void ChannelData::setPhoto(const MTPChatPhoto &photo) { setPhoto(userpicPhotoId(), photo); } void ChannelData::setPhoto(PhotoId photoId, const MTPChatPhoto &photo) { if (photo.type() == mtpc_chatPhoto) { const auto &data = photo.c_chatPhoto(); updateUserpic(photoId, data.vphoto_small); } else { clearUserpic(); } } void ChannelData::setName(const QString &newName, const QString &newUsername) { updateNameDelayed(newName.isEmpty() ? name : newName, QString(), newUsername); } void PeerData::updateFull() { if (!_lastFullUpdate || getms(true) > _lastFullUpdate + kUpdateFullPeerTimeout) { updateFullForced(); } } void PeerData::updateFullForced() { Auth().api().requestFullPeer(this); if (auto channel = asChannel()) { if (!channel->amCreator() && !channel->inviter) { Auth().api().requestSelfParticipant(channel); } } } void PeerData::fullUpdated() { _lastFullUpdate = getms(true); } bool ChannelData::setAbout(const QString &newAbout) { if (_about == newAbout) { return false; } _about = newAbout; Notify::peerUpdatedDelayed(this, UpdateFlag::AboutChanged); return true; } void ChannelData::setInviteLink(const QString &newInviteLink) { if (newInviteLink != _inviteLink) { _inviteLink = newInviteLink; Notify::peerUpdatedDelayed(this, UpdateFlag::InviteLinkChanged); } } void ChannelData::setMembersCount(int newMembersCount) { if (_membersCount != newMembersCount) { if (isMegagroup() && !mgInfo->lastParticipants.empty()) { mgInfo->lastParticipantsStatus |= MegagroupInfo::LastParticipantsCountOutdated; mgInfo->lastParticipantsCount = membersCount(); } _membersCount = newMembersCount; Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::MembersChanged); } } void ChannelData::setAdminsCount(int newAdminsCount) { if (_adminsCount != newAdminsCount) { _adminsCount = newAdminsCount; Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::AdminsChanged); } } void ChannelData::setRestrictedCount(int newRestrictedCount) { if (_restrictedCount != newRestrictedCount) { _restrictedCount = newRestrictedCount; Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::BannedUsersChanged); } } void ChannelData::setKickedCount(int newKickedCount) { if (_kickedCount != newKickedCount) { _kickedCount = newKickedCount; Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::BannedUsersChanged); } } MTPChannelBannedRights ChannelData::KickedRestrictedRights() { using Flag = MTPDchannelBannedRights::Flag; auto flags = Flag::f_view_messages | Flag::f_send_messages | Flag::f_send_media | Flag::f_embed_links | Flag::f_send_stickers | Flag::f_send_gifs | Flag::f_send_games | Flag::f_send_inline; return MTP_channelBannedRights(MTP_flags(flags), MTP_int(std::numeric_limits::max())); } void ChannelData::applyEditAdmin(not_null user, const MTPChannelAdminRights &oldRights, const MTPChannelAdminRights &newRights) { auto flags = Notify::PeerUpdate::Flag::AdminsChanged | Notify::PeerUpdate::Flag::None; if (mgInfo) { // If rights are empty - still add participant? TODO check if (!base::contains(mgInfo->lastParticipants, user)) { mgInfo->lastParticipants.push_front(user); setMembersCount(membersCount() + 1); if (user->botInfo && !mgInfo->bots.contains(user)) { mgInfo->bots.insert(user); if (mgInfo->botStatus != 0 && mgInfo->botStatus < 2) { mgInfo->botStatus = 2; } } } // If rights are empty - still remove restrictions? TODO check if (mgInfo->lastRestricted.contains(user)) { mgInfo->lastRestricted.remove(user); if (restrictedCount() > 0) { setRestrictedCount(restrictedCount() - 1); } } auto userId = peerToUser(user->id); auto it = mgInfo->lastAdmins.find(user); if (newRights.c_channelAdminRights().vflags.v != 0) { auto lastAdmin = MegagroupInfo::Admin { newRights }; lastAdmin.canEdit = true; if (it == mgInfo->lastAdmins.cend()) { mgInfo->lastAdmins.emplace(user, lastAdmin); setAdminsCount(adminsCount() + 1); } else { it->second = lastAdmin; } Data::ChannelAdminChanges(this).feed(userId, true); } else { if (it != mgInfo->lastAdmins.cend()) { mgInfo->lastAdmins.erase(it); if (adminsCount() > 0) { setAdminsCount(adminsCount() - 1); } } Data::ChannelAdminChanges(this).feed(userId, false); } } if (oldRights.c_channelAdminRights().vflags.v && !newRights.c_channelAdminRights().vflags.v) { // We removed an admin. if (adminsCount() > 1) { setAdminsCount(adminsCount() - 1); } if (!isMegagroup() && user->botInfo && membersCount() > 1) { // Removing bot admin removes it from channel. setMembersCount(membersCount() - 1); } } else if (!oldRights.c_channelAdminRights().vflags.v && newRights.c_channelAdminRights().vflags.v) { // We added an admin. setAdminsCount(adminsCount() + 1); updateFullForced(); } Notify::peerUpdatedDelayed(this, flags); } void ChannelData::applyEditBanned(not_null user, const MTPChannelBannedRights &oldRights, const MTPChannelBannedRights &newRights) { auto flags = Notify::PeerUpdate::Flag::BannedUsersChanged | Notify::PeerUpdate::Flag::None; auto isKicked = (newRights.c_channelBannedRights().vflags.v & MTPDchannelBannedRights::Flag::f_view_messages); auto isRestricted = !isKicked && (newRights.c_channelBannedRights().vflags.v != 0); if (mgInfo) { // If rights are empty - still remove admin? TODO check if (mgInfo->lastAdmins.contains(user)) { mgInfo->lastAdmins.remove(user); if (adminsCount() > 1) { setAdminsCount(adminsCount() - 1); } else { flags |= Notify::PeerUpdate::Flag::AdminsChanged; } } auto it = mgInfo->lastRestricted.find(user); if (isRestricted) { if (it == mgInfo->lastRestricted.cend()) { mgInfo->lastRestricted.emplace(user, MegagroupInfo::Restricted { newRights }); setRestrictedCount(restrictedCount() + 1); } else { it->second.rights = newRights; } } else { if (it != mgInfo->lastRestricted.cend()) { mgInfo->lastRestricted.erase(it); if (restrictedCount() > 0) { setRestrictedCount(restrictedCount() - 1); } } if (isKicked) { auto i = ranges::find(mgInfo->lastParticipants, user); if (i != mgInfo->lastParticipants.end()) { mgInfo->lastParticipants.erase(i); } if (membersCount() > 1) { setMembersCount(membersCount() - 1); } else { mgInfo->lastParticipantsStatus |= MegagroupInfo::LastParticipantsCountOutdated; mgInfo->lastParticipantsCount = 0; } setKickedCount(kickedCount() + 1); if (mgInfo->bots.contains(user)) { mgInfo->bots.remove(user); if (mgInfo->bots.empty() && mgInfo->botStatus > 0) { mgInfo->botStatus = -1; } } flags |= Notify::PeerUpdate::Flag::MembersChanged; Auth().data().removeMegagroupParticipant(this, user); } } Data::ChannelAdminChanges(this).feed(peerToUser(user->id), false); } else { if (isKicked) { if (membersCount() > 1) { setMembersCount(membersCount() - 1); flags |= Notify::PeerUpdate::Flag::MembersChanged; } setKickedCount(kickedCount() + 1); } } Notify::peerUpdatedDelayed(this, flags); } bool ChannelData::isGroupAdmin(not_null user) const { if (auto info = mgInfo.get()) { return info->admins.contains(peerToUser(user->id)); } return false; } void ChannelData::setRestrictionReason(const QString &text) { if (_restrictionReason != text) { _restrictionReason = text; Notify::peerUpdatedDelayed(this, Notify::PeerUpdate::Flag::RestrictionReasonChanged); } } void ChannelData::setAvailableMinId(MsgId availableMinId) { if (_availableMinId != availableMinId) { _availableMinId = availableMinId; if (auto history = App::historyLoaded(this)) { history->clearUpTill(availableMinId); } if (pinnedMessageId() <= _availableMinId) { clearPinnedMessage(); } } } void ChannelData::setFeed(not_null feed) { setFeedPointer(feed); } void ChannelData::clearFeed() { setFeedPointer(nullptr); } void ChannelData::setFeedPointer(Data::Feed *feed) { if (_feed != feed) { const auto was = _feed; _feed = feed; if (was) { was->unregisterOne(this); } if (_feed) { _feed->registerOne(this); } } } bool ChannelData::canBanMembers() const { return (adminRights() & AdminRight::f_ban_users) || amCreator(); } bool ChannelData::canEditMessages() const { return (adminRights() & AdminRight::f_edit_messages) || amCreator(); } bool ChannelData::canDeleteMessages() const { return (adminRights() & AdminRight::f_delete_messages) || amCreator(); } bool ChannelData::anyoneCanAddMembers() const { return (flags() & MTPDchannel::Flag::f_democracy); } bool ChannelData::hiddenPreHistory() const { return (fullFlags() & MTPDchannelFull::Flag::f_hidden_prehistory); } bool ChannelData::canAddMembers() const { return (adminRights() & AdminRight::f_invite_users) || amCreator() || (anyoneCanAddMembers() && amIn() && !hasRestrictions()); } bool ChannelData::canAddAdmins() const { return (adminRights() & AdminRight::f_add_admins) || amCreator(); } bool ChannelData::canPublish() const { return (adminRights() & AdminRight::f_post_messages) || amCreator(); } bool ChannelData::canWrite() const { // Duplicated in Data::CanWriteValue(). return amIn() && (canPublish() || (!isBroadcast() && !restricted(Restriction::f_send_messages))); } bool ChannelData::canViewMembers() const { return fullFlags() & MTPDchannelFull::Flag::f_can_view_participants; } bool ChannelData::canViewAdmins() const { return (isMegagroup() || hasAdminRights() || amCreator()); } bool ChannelData::canViewBanned() const { return (hasAdminRights() || amCreator()); } bool ChannelData::canEditInformation() const { return (adminRights() & AdminRight::f_change_info) || amCreator(); } bool ChannelData::canEditInvites() const { return canEditInformation(); } bool ChannelData::canEditSignatures() const { return canEditInformation(); } bool ChannelData::canEditPreHistoryHidden() const { return canEditInformation(); } bool ChannelData::canEditUsername() const { return amCreator() && (fullFlags() & MTPDchannelFull::Flag::f_can_set_username); } bool ChannelData::canEditStickers() const { return (fullFlags() & MTPDchannelFull::Flag::f_can_set_stickers); } bool ChannelData::canDelete() const { constexpr auto kDeleteChannelMembersLimit = 1000; return amCreator() && (membersCount() <= kDeleteChannelMembersLimit); } bool ChannelData::canEditLastAdmin(not_null user) const { // Duplicated in ParticipantsBoxController::canEditAdmin :( if (mgInfo) { auto i = mgInfo->lastAdmins.find(user); if (i != mgInfo->lastAdmins.cend()) { return i->second.canEdit; } return (user != mgInfo->creator); } return false; } bool ChannelData::canEditAdmin(not_null user) const { // Duplicated in ParticipantsBoxController::canEditAdmin :( if (user->isSelf()) { return false; } else if (amCreator()) { return true; } else if (!canEditLastAdmin(user)) { return false; } return adminRights() & AdminRight::f_add_admins; } bool ChannelData::canRestrictUser(not_null user) const { // Duplicated in ParticipantsBoxController::canRestrictUser :( if (user->isSelf()) { return false; } else if (amCreator()) { return true; } else if (!canEditLastAdmin(user)) { return false; } return adminRights() & AdminRight::f_ban_users; } void ChannelData::setAdminRights(const MTPChannelAdminRights &rights) { if (rights.c_channelAdminRights().vflags.v == adminRights()) { return; } _adminRights.set(rights.c_channelAdminRights().vflags.v); if (isMegagroup()) { const auto self = Auth().user(); if (hasAdminRights()) { if (!amCreator()) { auto me = MegagroupInfo::Admin { rights }; me.canEdit = false; mgInfo->lastAdmins.emplace(self, me); } mgInfo->lastRestricted.remove(self); } else { mgInfo->lastAdmins.remove(self); } auto amAdmin = hasAdminRights() || amCreator(); Data::ChannelAdminChanges(this).feed(Auth().userId(), amAdmin); } Notify::peerUpdatedDelayed(this, UpdateFlag::ChannelRightsChanged | UpdateFlag::AdminsChanged | UpdateFlag::BannedUsersChanged); } void ChannelData::setRestrictedRights(const MTPChannelBannedRights &rights) { if (rights.c_channelBannedRights().vflags.v == restrictions() && rights.c_channelBannedRights().vuntil_date.v == _restrictedUntill) { return; } _restrictedUntill = rights.c_channelBannedRights().vuntil_date.v; _restrictions.set(rights.c_channelBannedRights().vflags.v); if (isMegagroup()) { const auto self = Auth().user(); if (hasRestrictions()) { if (!amCreator()) { auto me = MegagroupInfo::Restricted { rights }; mgInfo->lastRestricted.emplace(self, me); } mgInfo->lastAdmins.remove(self); Data::ChannelAdminChanges(this).feed(Auth().userId(), false); } else { mgInfo->lastRestricted.remove(self); } } Notify::peerUpdatedDelayed(this, UpdateFlag::ChannelRightsChanged | UpdateFlag::AdminsChanged | UpdateFlag::BannedUsersChanged); } uint64 PtsWaiter::ptsKey(PtsSkippedQueue queue, int32 pts) { return _queue.insert(uint64(uint32(pts)) << 32 | (++_skippedKey), queue).key(); } void PtsWaiter::setWaitingForSkipped(ChannelData *channel, int32 ms) { if (ms >= 0) { if (App::main()) { App::main()->ptsWaiterStartTimerFor(channel, ms); } _waitingForSkipped = true; } else { _waitingForSkipped = false; checkForWaiting(channel); } } void PtsWaiter::setWaitingForShortPoll(ChannelData *channel, int32 ms) { if (ms >= 0) { if (App::main()) { App::main()->ptsWaiterStartTimerFor(channel, ms); } _waitingForShortPoll = true; } else { _waitingForShortPoll = false; checkForWaiting(channel); } } void PtsWaiter::checkForWaiting(ChannelData *channel) { if (!_waitingForSkipped && !_waitingForShortPoll && App::main()) { App::main()->ptsWaiterStartTimerFor(channel, -1); } } void PtsWaiter::applySkippedUpdates(ChannelData *channel) { if (!_waitingForSkipped) return; setWaitingForSkipped(channel, -1); if (_queue.isEmpty()) return; ++_applySkippedLevel; for (auto i = _queue.cbegin(), e = _queue.cend(); i != e; ++i) { switch (i.value()) { case SkippedUpdate: Auth().api().applyUpdateNoPtsCheck(_updateQueue.value(i.key())); break; case SkippedUpdates: Auth().api().applyUpdatesNoPtsCheck(_updatesQueue.value(i.key())); break; } } --_applySkippedLevel; clearSkippedUpdates(); } void PtsWaiter::clearSkippedUpdates() { _queue.clear(); _updateQueue.clear(); _updatesQueue.clear(); _applySkippedLevel = 0; } bool PtsWaiter::updated(ChannelData *channel, int32 pts, int32 count, const MTPUpdates &updates) { if (_requesting || _applySkippedLevel) { return true; } else if (pts <= _good && count > 0) { return false; } else if (check(channel, pts, count)) { return true; } _updatesQueue.insert(ptsKey(SkippedUpdates, pts), updates); return false; } bool PtsWaiter::updated(ChannelData *channel, int32 pts, int32 count, const MTPUpdate &update) { if (_requesting || _applySkippedLevel) { return true; } else if (pts <= _good && count > 0) { return false; } else if (check(channel, pts, count)) { return true; } _updateQueue.insert(ptsKey(SkippedUpdate, pts), update); return false; } bool PtsWaiter::updated(ChannelData *channel, int32 pts, int32 count) { if (_requesting || _applySkippedLevel) { return true; } else if (pts <= _good && count > 0) { return false; } return check(channel, pts, count); } bool PtsWaiter::updateAndApply(ChannelData *channel, int32 pts, int32 count, const MTPUpdates &updates) { if (!updated(channel, pts, count, updates)) { return false; } if (!_waitingForSkipped || _queue.isEmpty()) { // Optimization - no need to put in queue and back. Auth().api().applyUpdatesNoPtsCheck(updates); } else { _updatesQueue.insert(ptsKey(SkippedUpdates, pts), updates); applySkippedUpdates(channel); } return true; } bool PtsWaiter::updateAndApply(ChannelData *channel, int32 pts, int32 count, const MTPUpdate &update) { if (!updated(channel, pts, count, update)) { return false; } if (!_waitingForSkipped || _queue.isEmpty()) { // Optimization - no need to put in queue and back. Auth().api().applyUpdateNoPtsCheck(update); } else { _updateQueue.insert(ptsKey(SkippedUpdate, pts), update); applySkippedUpdates(channel); } return true; } bool PtsWaiter::updateAndApply(ChannelData *channel, int32 pts, int32 count) { if (!updated(channel, pts, count)) { return false; } applySkippedUpdates(channel); return true; } bool PtsWaiter::check(ChannelData *channel, int32 pts, int32 count) { // return false if need to save that update and apply later if (!inited()) { init(pts); return true; } _last = qMax(_last, pts); _count += count; if (_last == _count) { _good = _last; return true; } else if (_last < _count) { setWaitingForSkipped(channel, 1); } else { setWaitingForSkipped(channel, WaitForSkippedTimeout); } return !count; }