/* 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 "data/data_user.h" #include "data/data_chat.h" #include "data/data_channel.h" #include "data/data_photo.h" #include "data/data_folder.h" #include "data/data_session.h" #include "data/data_file_origin.h" #include "data/data_histories.h" #include "base/unixtime.h" #include "base/crc32hash.h" #include "lang/lang_keys.h" #include "observer_peer.h" #include "apiwrap.h" #include "boxes/confirm_box.h" #include "main/main_session.h" #include "main/main_account.h" #include "main/main_app_config.h" #include "core/application.h" #include "mainwindow.h" #include "window/window_session_controller.h" #include "ui/image/image.h" #include "ui/empty_userpic.h" #include "ui/text_options.h" #include "history/history.h" #include "history/view/history_view_element.h" #include "history/history_item.h" #include "facades.h" #include "app.h" namespace { constexpr auto kUpdateFullPeerTimeout = crl::time(5000); // Not more than once in 5 seconds. constexpr auto kUserpicSize = 160; using UpdateFlag = Notify::PeerUpdate::Flag; } // 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)]; } PeerId FakePeerIdForJustName(const QString &name) { return peerFromUser(name.isEmpty() ? 777 : base::crc32(name.constData(), name.size() * sizeof(QChar))); } } // namespace Data PeerClickHandler::PeerClickHandler(not_null peer) : _peer(peer) { } void PeerClickHandler::onClick(ClickContext context) const { if (context.button == Qt::LeftButton && App::wnd()) { const auto controller = App::wnd()->sessionController(); const auto currentPeer = controller->activeChatCurrent().peer(); if (_peer && _peer->isChannel() && currentPeer != _peer) { const auto clickedChannel = _peer->asChannel(); if (!clickedChannel->isPublic() && !clickedChannel->amIn() && (!currentPeer->isChannel() || currentPeer->asChannel()->linkedChat() != clickedChannel)) { Ui::show(Box(_peer->isMegagroup() ? tr::lng_group_not_accessible(tr::now) : tr::lng_channel_not_accessible(tr::now))); } else { controller->showPeerHistory( _peer, Window::SectionShow::Way::Forward); } } else { Ui::showPeerProfile(_peer); } } } PeerData::PeerData(not_null owner, PeerId id) : id(id) , _owner(owner) , _userpicEmpty(createEmptyUserpic()) { _nameText.setText(st::msgNameStyle, QString(), Ui::NameTextOptions()); } Data::Session &PeerData::owner() const { return *_owner; } Main::Session &PeerData::session() const { return _owner->session(); } Main::Account &PeerData::account() const { return session().account(); } void PeerData::updateNameDelayed( const QString &newName, const QString &newNameOrPhone, const QString &newUsername) { if (name == newName && nameVersion > 1) { if (isUser()) { if (asUser()->nameOrPhone == newNameOrPhone && asUser()->username == newUsername) { return; } } else if (isChannel()) { if (asChannel()->username == newUsername) { return; } } else if (isChat()) { return; } } name = newName; _nameText.setText(st::msgNameStyle, name, Ui::NameTextOptions()); refreshEmptyUserpic(); Notify::PeerUpdate update(this); if (nameVersion++ > 1) { 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(); if (update.flags) { 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 = owner().processPhoto(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() { _userpic->load(userpicOrigin()); } bool PeerData::userpicLoaded() const { return _userpic->loaded(); } bool PeerData::useEmptyUserpic() const { return !_userpicLocation.valid() || !_userpic || !_userpic->loaded(); } InMemoryKey PeerData::userpicUniqueKey() const { if (useEmptyUserpic()) { if (!_userpicEmpty) { refreshEmptyUserpic(); } return _userpicEmpty->uniqueKey(); } return inMemoryKey(_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)); } Data::FileOrigin PeerData::userpicOrigin() const { return Data::FileOriginPeerPhoto(id); } Data::FileOrigin PeerData::userpicPhotoOrigin() const { return (isUser() && userpicPhotoId()) ? Data::FileOriginUserPhoto(bareId(), userpicPhotoId()) : Data::FileOrigin(); } void PeerData::updateUserpic( PhotoId photoId, MTP::DcId dcId, const MTPFileLocation &location) { const auto size = kUserpicSize; const auto loc = location.match([&]( const MTPDfileLocationToBeDeprecated &deprecated) { return StorageImageLocation( StorageFileLocation( dcId, isSelf() ? peerToUser(id) : 0, MTP_inputPeerPhotoFileLocation( MTP_flags(0), input, deprecated.vvolume_id(), deprecated.vlocal_id())), size, size); }); setUserpicChecked(photoId, loc, Images::Create(loc)); } void PeerData::clearUserpic() { const auto photoId = PhotoId(0); const auto loc = StorageImageLocation(); const auto photo = [&] { if (isNotificationsUser()) { auto image = Core::App().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()) { // #feed // if (const auto feed = channel->feed()) { // owner().notifyFeedUpdated( // feed, // Data::FeedUpdateFlag::ChannelPhoto); // } //} } } auto PeerData::unavailableReasons() const -> const std::vector & { static const auto result = std::vector(); return result; } QString PeerData::computeUnavailableReason() const { const auto &list = unavailableReasons(); const auto &config = session().account().appConfig(); const auto skip = config.get>( "ignore_restriction_reasons", std::vector()); auto &&filtered = ranges::view::all( list ) | ranges::view::filter([&](const Data::UnavailableReason &reason) { return ranges::find(skip, reason.reason) == end(skip); }); const auto first = filtered.begin(); return (first != filtered.end()) ? first->text : QString(); } 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->amIn() && !chat->amRestricted(ChatRestriction::f_pin_messages); } else if (const auto channel = asChannel()) { return channel->isMegagroup() ? !channel->amRestricted(ChatRestriction::f_pin_messages) : ((channel->adminRights() & ChatAdminRight::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); } } bool PeerData::canExportChatHistory() const { for (const auto &block : _owner->history(id)->blocks) { for (const auto &message : block->messages) { if (!message->data()->serviceMsg()) { return true; } } } return false; } bool PeerData::setAbout(const QString &newAbout) { if (_about == newAbout) { return false; } _about = newAbout; Notify::peerUpdatedDelayed(this, UpdateFlag::AboutChanged); return true; } void PeerData::checkFolder(FolderId folderId) { const auto folder = folderId ? owner().folderLoaded(folderId) : nullptr; if (const auto history = owner().historyLoaded(this)) { if (folder && history->folder() != folder) { owner().histories().requestDialogEntry(history); } } } 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()) { const auto english = qsl("Saved messages"); const auto localized = tr::lng_saved_messages(tr::now); appendToIndex(english); if (localized != english) { appendToIndex(localized); } } } 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; void PeerData::updateFull() { if (!_lastFullUpdate || crl::now() > _lastFullUpdate + kUpdateFullPeerTimeout) { updateFullForced(); } } void PeerData::updateFullForced() { session().api().requestFullPeer(this); if (const auto channel = asChannel()) { if (!channel->amCreator() && !channel->inviter) { session().api().requestSelfParticipant(channel); } } } void PeerData::fullUpdated() { _lastFullUpdate = crl::now(); } UserData *PeerData::asUser() { return isUser() ? static_cast(this) : nullptr; } const UserData *PeerData::asUser() const { return isUser() ? static_cast(this) : nullptr; } ChatData *PeerData::asChat() { return isChat() ? static_cast(this) : nullptr; } const ChatData *PeerData::asChat() const { return isChat() ? static_cast(this) : nullptr; } ChannelData *PeerData::asChannel() { return isChannel() ? static_cast(this) : nullptr; } const ChannelData *PeerData::asChannel() const { return isChannel() ? static_cast(this) : nullptr; } ChannelData *PeerData::asMegagroup() { return isMegagroup() ? static_cast(this) : nullptr; } const ChannelData *PeerData::asMegagroup() const { return isMegagroup() ? static_cast(this) : nullptr; } ChatData *PeerData::asChatNotMigrated() { if (const auto chat = asChat()) { return chat->migrateTo() ? nullptr : chat; } return nullptr; } const ChatData *PeerData::asChatNotMigrated() const { if (const auto chat = asChat()) { return chat->migrateTo() ? nullptr : chat; } return nullptr; } ChannelData *PeerData::asChannelOrMigrated() { if (const auto channel = asChannel()) { return channel; } return migrateTo(); } const ChannelData *PeerData::asChannelOrMigrated() const { if (const auto channel = asChannel()) { return channel; } return migrateTo(); } ChatData *PeerData::migrateFrom() const { if (const auto megagroup = asMegagroup()) { return megagroup->amIn() ? megagroup->getMigrateFromChat() : nullptr; } return nullptr; } ChannelData *PeerData::migrateTo() const { if (const auto chat = asChat()) { if (const auto result = chat->getMigrateToChannel()) { return result->amIn() ? result : nullptr; } } return nullptr; } not_null PeerData::migrateToOrMe() { if (const auto channel = migrateTo()) { return channel; } return this; } not_null PeerData::migrateToOrMe() const { if (const auto channel = migrateTo()) { return channel; } return this; } const Ui::Text::String &PeerData::topBarNameText() const { if (const auto to = migrateTo()) { return to->topBarNameText(); } else if (const auto user = asUser()) { if (!user->phoneText.isEmpty()) { return user->phoneText; } } return _nameText; } const Ui::Text::String &PeerData::nameText() const { if (const auto to = migrateTo()) { return to->nameText(); } return _nameText; } const QString &PeerData::shortName() const { if (const auto user = asUser()) { return user->firstName.isEmpty() ? user->lastName : user->firstName; } return name; } QString PeerData::userName() const { if (const auto user = asUser()) { return user->username; } else if (const auto channel = asChannel()) { return channel->username; } return QString(); } bool PeerData::isVerified() const { if (const auto user = asUser()) { return user->isVerified(); } else if (const auto channel = asChannel()) { return channel->isVerified(); } return false; } bool PeerData::isScam() const { if (const auto user = asUser()) { return user->isScam(); } else if (const auto channel = asChannel()) { return channel->isScam(); } return false; } bool PeerData::isMegagroup() const { return isChannel() ? asChannel()->isMegagroup() : false; } bool PeerData::canWrite() const { if (const auto user = asUser()) { return user->canWrite(); } else if (const auto channel = asChannel()) { return channel->canWrite(); } else if (const auto chat = asChat()) { return chat->canWrite(); } return false; } Data::RestrictionCheckResult PeerData::amRestricted( ChatRestriction right) const { using Result = Data::RestrictionCheckResult; const auto allowByAdminRights = [](auto right, auto chat) -> bool { if (right == ChatRestriction::f_invite_users) { return chat->adminRights() & ChatAdminRight::f_invite_users; } else if (right == ChatRestriction::f_change_info) { return chat->adminRights() & ChatAdminRight::f_change_info; } else if (right == ChatRestriction::f_pin_messages) { return chat->adminRights() & ChatAdminRight::f_pin_messages; } else { return chat->hasAdminRights(); } }; if (const auto channel = asChannel()) { const auto defaultRestrictions = channel->defaultRestrictions() | (channel->isPublic() ? (ChatRestriction::f_pin_messages | ChatRestriction::f_change_info) : ChatRestrictions(0)); return (channel->amCreator() || allowByAdminRights(right, channel)) ? Result::Allowed() : (defaultRestrictions & right) ? Result::WithEveryone() : (channel->restrictions() & right) ? Result::Explicit() : Result::Allowed(); } else if (const auto chat = asChat()) { return (chat->amCreator() || allowByAdminRights(right, chat)) ? Result::Allowed() : (chat->defaultRestrictions() & right) ? Result::WithEveryone() : Result::Allowed(); } return Result::Allowed(); } bool PeerData::canRevokeFullHistory() const { return isUser() && !isSelf() && Global::RevokePrivateInbox() && (Global::RevokePrivateTimeLimit() == 0x7FFFFFFF); } bool PeerData::slowmodeApplied() const { if (const auto channel = asChannel()) { return !channel->amCreator() && !channel->hasAdminRights() && (channel->flags() & MTPDchannel::Flag::f_slowmode_enabled); } return false; } int PeerData::slowmodeSecondsLeft() const { if (const auto channel = asChannel()) { if (const auto seconds = channel->slowmodeSeconds()) { if (const auto last = channel->slowmodeLastMessage()) { const auto now = base::unixtime::now(); return std::max(seconds - (now - last), 0); } } } return 0; } bool PeerData::canSendPolls() const { if (const auto user = asUser()) { return user->isBot(); } else if (const auto chat = asChat()) { return chat->canSendPolls(); } else if (const auto channel = asChannel()) { return channel->canSendPolls(); } return false; } namespace Data { std::vector ListOfRestrictions() { using Flag = ChatRestriction; return { Flag::f_send_messages, Flag::f_send_media, Flag::f_send_stickers | Flag::f_send_gifs | Flag::f_send_games | Flag::f_send_inline, Flag::f_embed_links, Flag::f_send_polls, Flag::f_invite_users, Flag::f_pin_messages, Flag::f_change_info, }; } std::optional RestrictionError( not_null peer, ChatRestriction restriction) { using Flag = ChatRestriction; if (const auto restricted = peer->amRestricted(restriction)) { const auto all = restricted.isWithEveryone(); const auto channel = peer->asChannel(); if (!all && channel) { auto restrictedUntil = channel->restrictedUntil(); if (restrictedUntil > 0 && !ChannelData::IsRestrictedForever(restrictedUntil)) { auto restrictedUntilDateTime = base::unixtime::parse(channel->restrictedUntil()); auto date = restrictedUntilDateTime.toString(qsl("dd.MM.yy")); auto time = restrictedUntilDateTime.toString(cTimeFormat()); switch (restriction) { case Flag::f_send_polls: return tr::lng_restricted_send_polls_until( tr::now, lt_date, date, lt_time, time); case Flag::f_send_messages: return tr::lng_restricted_send_message_until( tr::now, lt_date, date, lt_time, time); case Flag::f_send_media: return tr::lng_restricted_send_media_until( tr::now, lt_date, date, lt_time, time); case Flag::f_send_stickers: return tr::lng_restricted_send_stickers_until( tr::now, lt_date, date, lt_time, time); case Flag::f_send_gifs: return tr::lng_restricted_send_gifs_until( tr::now, lt_date, date, lt_time, time); case Flag::f_send_inline: case Flag::f_send_games: return tr::lng_restricted_send_inline_until( tr::now, lt_date, date, lt_time, time); } Unexpected("Restriction in Data::RestrictionErrorKey."); } } switch (restriction) { case Flag::f_send_polls: return all ? tr::lng_restricted_send_polls_all(tr::now) : tr::lng_restricted_send_polls(tr::now); case Flag::f_send_messages: return all ? tr::lng_restricted_send_message_all(tr::now) : tr::lng_restricted_send_message(tr::now); case Flag::f_send_media: return all ? tr::lng_restricted_send_media_all(tr::now) : tr::lng_restricted_send_media(tr::now); case Flag::f_send_stickers: return all ? tr::lng_restricted_send_stickers_all(tr::now) : tr::lng_restricted_send_stickers(tr::now); case Flag::f_send_gifs: return all ? tr::lng_restricted_send_gifs_all(tr::now) : tr::lng_restricted_send_gifs(tr::now); case Flag::f_send_inline: case Flag::f_send_games: return all ? tr::lng_restricted_send_inline_all(tr::now) : tr::lng_restricted_send_inline(tr::now); } Unexpected("Restriction in Data::RestrictionErrorKey."); } return std::nullopt; } } // namespace Data