508 lines
		
	
	
	
		
			13 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			508 lines
		
	
	
	
		
			13 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 "data/data_chat.h"
 | |
| 
 | |
| #include "data/data_user.h"
 | |
| #include "data/data_channel.h"
 | |
| #include "data/data_session.h"
 | |
| #include "data/data_changes.h"
 | |
| #include "data/data_group_call.h"
 | |
| #include "history/history.h"
 | |
| #include "main/main_session.h"
 | |
| #include "apiwrap.h"
 | |
| #include "api/api_invite_links.h"
 | |
| 
 | |
| namespace {
 | |
| 
 | |
| using UpdateFlag = Data::PeerUpdate::Flag;
 | |
| 
 | |
| } // namespace
 | |
| 
 | |
| ChatData::ChatData(not_null<Data::Session*> owner, PeerId id)
 | |
| : PeerData(owner, id)
 | |
| , inputChat(MTP_int(bareId())) {
 | |
| 	_flags.changes(
 | |
| 	) | rpl::start_with_next([=](const Flags::Change &change) {
 | |
| 		if (change.diff & MTPDchat::Flag::f_call_not_empty) {
 | |
| 			if (const auto history = this->owner().historyLoaded(this)) {
 | |
| 				history->updateChatListEntry();
 | |
| 			}
 | |
| 		}
 | |
| 	}, _lifetime);
 | |
| }
 | |
| 
 | |
| void ChatData::setPhoto(const MTPChatPhoto &photo) {
 | |
| 	setPhoto(userpicPhotoId(), photo);
 | |
| }
 | |
| 
 | |
| void ChatData::setPhoto(PhotoId photoId, const MTPChatPhoto &photo) {
 | |
| 	photo.match([&](const MTPDchatPhoto &data) {
 | |
| 		updateUserpic(photoId, data.vdc_id().v, data.vphoto_small());
 | |
| 	}, [&](const MTPDchatPhotoEmpty &) {
 | |
| 		clearUserpic();
 | |
| 	});
 | |
| }
 | |
| 
 | |
| auto ChatData::defaultAdminRights(not_null<UserData*> user) -> AdminRights {
 | |
| 	const auto isCreator = (creator == user->bareId())
 | |
| 		|| (user->isSelf() && amCreator());
 | |
| 	using Flag = AdminRight;
 | |
| 	return Flag::f_change_info
 | |
| 		| Flag::f_delete_messages
 | |
| 		| Flag::f_ban_users
 | |
| 		| Flag::f_invite_users
 | |
| 		| Flag::f_pin_messages
 | |
| 		| Flag::f_manage_call
 | |
| 		| (isCreator ? Flag::f_add_admins : Flag(0));
 | |
| }
 | |
| 
 | |
| bool ChatData::canWrite() const {
 | |
| 	// Duplicated in Data::CanWriteValue().
 | |
| 	return amIn() && !amRestricted(Restriction::f_send_messages);
 | |
| }
 | |
| 
 | |
| bool ChatData::canEditInformation() const {
 | |
| 	return amIn() && !amRestricted(Restriction::f_change_info);
 | |
| }
 | |
| 
 | |
| bool ChatData::canEditPermissions() const {
 | |
| 	return amIn()
 | |
| 		&& (amCreator() || (adminRights() & AdminRight::f_ban_users));
 | |
| }
 | |
| 
 | |
| bool ChatData::canEditUsername() const {
 | |
| 	return amCreator()
 | |
| 		&& (fullFlags() & MTPDchatFull::Flag::f_can_set_username);
 | |
| }
 | |
| 
 | |
| bool ChatData::canEditPreHistoryHidden() const {
 | |
| 	return amCreator();
 | |
| }
 | |
| 
 | |
| bool ChatData::canDeleteMessages() const {
 | |
| 	return amCreator()
 | |
| 		|| (adminRights() & AdminRight::f_delete_messages);
 | |
| }
 | |
| 
 | |
| bool ChatData::canAddMembers() const {
 | |
| 	return amIn() && !amRestricted(Restriction::f_invite_users);
 | |
| }
 | |
| 
 | |
| bool ChatData::canSendPolls() const {
 | |
| 	return amIn() && !amRestricted(Restriction::f_send_polls);
 | |
| }
 | |
| 
 | |
| bool ChatData::canAddAdmins() const {
 | |
| 	return amIn() && amCreator();
 | |
| }
 | |
| 
 | |
| bool ChatData::canBanMembers() const {
 | |
| 	return amCreator()
 | |
| 		|| (adminRights() & AdminRight::f_ban_users);
 | |
| }
 | |
| 
 | |
| bool ChatData::anyoneCanAddMembers() const {
 | |
| 	return !(defaultRestrictions() & Restriction::f_invite_users);
 | |
| }
 | |
| 
 | |
| void ChatData::setName(const QString &newName) {
 | |
| 	updateNameDelayed(newName.isEmpty() ? name : newName, QString(), QString());
 | |
| }
 | |
| 
 | |
| void ChatData::applyEditAdmin(not_null<UserData*> user, bool isAdmin) {
 | |
| 	if (isAdmin) {
 | |
| 		admins.emplace(user);
 | |
| 	} else {
 | |
| 		admins.remove(user);
 | |
| 	}
 | |
| 	session().changes().peerUpdated(this, UpdateFlag::Admins);
 | |
| }
 | |
| 
 | |
| void ChatData::invalidateParticipants() {
 | |
| 	participants.clear();
 | |
| 	admins.clear();
 | |
| 	setAdminRights(MTP_chatAdminRights(MTP_flags(0)));
 | |
| 	//setDefaultRestrictions(MTP_chatBannedRights(MTP_flags(0), MTP_int(0)));
 | |
| 	invitedByMe.clear();
 | |
| 	botStatus = 0;
 | |
| 	session().changes().peerUpdated(
 | |
| 		this,
 | |
| 		UpdateFlag::Members | UpdateFlag::Admins);
 | |
| }
 | |
| 
 | |
| void ChatData::setInviteLink(const QString &newInviteLink) {
 | |
| 	_inviteLink = newInviteLink;
 | |
| }
 | |
| 
 | |
| bool ChatData::canHaveInviteLink() const {
 | |
| 	return amCreator()
 | |
| 		|| (adminRights() & AdminRight::f_invite_users);
 | |
| }
 | |
| 
 | |
| void ChatData::setAdminRights(const MTPChatAdminRights &rights) {
 | |
| 	if (rights.c_chatAdminRights().vflags().v == adminRights()) {
 | |
| 		return;
 | |
| 	}
 | |
| 	_adminRights.set(rights.c_chatAdminRights().vflags().v);
 | |
| 	session().changes().peerUpdated(
 | |
| 		this,
 | |
| 		UpdateFlag::Rights | UpdateFlag::Admins | UpdateFlag::BannedUsers);
 | |
| }
 | |
| 
 | |
| void ChatData::setDefaultRestrictions(const MTPChatBannedRights &rights) {
 | |
| 	if (rights.c_chatBannedRights().vflags().v == defaultRestrictions()) {
 | |
| 		return;
 | |
| 	}
 | |
| 	_defaultRestrictions.set(rights.c_chatBannedRights().vflags().v);
 | |
| 	session().changes().peerUpdated(this, UpdateFlag::Rights);
 | |
| }
 | |
| 
 | |
| void ChatData::refreshBotStatus() {
 | |
| 	if (participants.empty()) {
 | |
| 		botStatus = 0;
 | |
| 	} else {
 | |
| 		const auto bot = ranges::none_of(participants, &UserData::isBot);
 | |
| 		botStatus = bot ? -1 : 2;
 | |
| 	}
 | |
| }
 | |
| 
 | |
| auto ChatData::applyUpdateVersion(int version) -> UpdateStatus {
 | |
| 	if (_version > version) {
 | |
| 		return UpdateStatus::TooOld;
 | |
| 	} else if (_version + 1 < version) {
 | |
| 		invalidateParticipants();
 | |
| 		session().api().requestFullPeer(this);
 | |
| 		return UpdateStatus::Skipped;
 | |
| 	}
 | |
| 	setVersion(version);
 | |
| 	return UpdateStatus::Good;
 | |
| }
 | |
| 
 | |
| ChannelData *ChatData::getMigrateToChannel() const {
 | |
| 	return _migratedTo;
 | |
| }
 | |
| 
 | |
| void ChatData::setMigrateToChannel(ChannelData *channel) {
 | |
| 	if (_migratedTo != channel) {
 | |
| 		_migratedTo = channel;
 | |
| 		if (channel->amIn()) {
 | |
| 			session().changes().peerUpdated(this, UpdateFlag::Migration);
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void ChatData::setGroupCall(const MTPInputGroupCall &call) {
 | |
| 	if (migrateTo()) {
 | |
| 		return;
 | |
| 	}
 | |
| 	call.match([&](const MTPDinputGroupCall &data) {
 | |
| 		if (_call && _call->id() == data.vid().v) {
 | |
| 			return;
 | |
| 		} else if (!_call && !data.vid().v) {
 | |
| 			return;
 | |
| 		} else if (!data.vid().v) {
 | |
| 			clearGroupCall();
 | |
| 			return;
 | |
| 		}
 | |
| 		const auto hasCall = (_call != nullptr);
 | |
| 		if (hasCall) {
 | |
| 			owner().unregisterGroupCall(_call.get());
 | |
| 		}
 | |
| 		_call = std::make_unique<Data::GroupCall>(
 | |
| 			this,
 | |
| 			data.vid().v,
 | |
| 			data.vaccess_hash().v);
 | |
| 		owner().registerGroupCall(_call.get());
 | |
| 		session().changes().peerUpdated(this, UpdateFlag::GroupCall);
 | |
| 		addFlags(MTPDchat::Flag::f_call_active);
 | |
| 	});
 | |
| }
 | |
| 
 | |
| void ChatData::clearGroupCall() {
 | |
| 	if (!_call) {
 | |
| 		return;
 | |
| 	} else if (const auto group = migrateTo(); group && !group->groupCall()) {
 | |
| 		group->migrateCall(base::take(_call));
 | |
| 	} else {
 | |
| 		owner().unregisterGroupCall(_call.get());
 | |
| 		_call = nullptr;
 | |
| 	}
 | |
| 	session().changes().peerUpdated(this, UpdateFlag::GroupCall);
 | |
| 	removeFlags(MTPDchat::Flag::f_call_active
 | |
| 		| MTPDchat::Flag::f_call_not_empty);
 | |
| }
 | |
| 
 | |
| void ChatData::setGroupCallDefaultJoinAs(PeerId peerId) {
 | |
| 	_callDefaultJoinAs = peerId;
 | |
| }
 | |
| 
 | |
| PeerId ChatData::groupCallDefaultJoinAs() const {
 | |
| 	return _callDefaultJoinAs;
 | |
| }
 | |
| 
 | |
| namespace Data {
 | |
| 
 | |
| void ApplyChatUpdate(
 | |
| 		not_null<ChatData*> chat,
 | |
| 		const MTPDupdateChatParticipants &update) {
 | |
| 	ApplyChatUpdate(chat, update.vparticipants());
 | |
| }
 | |
| 
 | |
| void ApplyChatUpdate(
 | |
| 		not_null<ChatData*> chat,
 | |
| 		const MTPDupdateChatParticipantAdd &update) {
 | |
| 	if (chat->applyUpdateVersion(update.vversion().v)
 | |
| 		!= ChatData::UpdateStatus::Good) {
 | |
| 		return;
 | |
| 	} else if (chat->count < 0) {
 | |
| 		return;
 | |
| 	}
 | |
| 	const auto user = chat->owner().userLoaded(update.vuser_id().v);
 | |
| 	const auto session = &chat->session();
 | |
| 	if (!user
 | |
| 		|| (!chat->participants.empty()
 | |
| 			&& chat->participants.contains(user))) {
 | |
| 		chat->invalidateParticipants();
 | |
| 		++chat->count;
 | |
| 		return;
 | |
| 	}
 | |
| 	if (chat->participants.empty()) {
 | |
| 		if (chat->count > 0) { // If the count is known.
 | |
| 			++chat->count;
 | |
| 		}
 | |
| 		chat->botStatus = 0;
 | |
| 	} else {
 | |
| 		chat->participants.emplace(user);
 | |
| 		if (update.vinviter_id().v == session->userId()) {
 | |
| 			chat->invitedByMe.insert(user);
 | |
| 		} else {
 | |
| 			chat->invitedByMe.remove(user);
 | |
| 		}
 | |
| 		++chat->count;
 | |
| 		if (user->isBot()) {
 | |
| 			chat->botStatus = 2;
 | |
| 			if (!user->botInfo->inited) {
 | |
| 				session->api().requestFullPeer(user);
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	session->changes().peerUpdated(chat, UpdateFlag::Members);
 | |
| }
 | |
| 
 | |
| void ApplyChatUpdate(
 | |
| 		not_null<ChatData*> chat,
 | |
| 		const MTPDupdateChatParticipantDelete &update) {
 | |
| 	if (chat->applyUpdateVersion(update.vversion().v)
 | |
| 		!= ChatData::UpdateStatus::Good) {
 | |
| 		return;
 | |
| 	} else if (chat->count <= 0) {
 | |
| 		return;
 | |
| 	}
 | |
| 	const auto user = chat->owner().userLoaded(update.vuser_id().v);
 | |
| 	if (!user
 | |
| 		|| (!chat->participants.empty()
 | |
| 			&& !chat->participants.contains(user))) {
 | |
| 		chat->invalidateParticipants();
 | |
| 		--chat->count;
 | |
| 		return;
 | |
| 	}
 | |
| 	if (chat->participants.empty()) {
 | |
| 		if (chat->count > 0) {
 | |
| 			chat->count--;
 | |
| 		}
 | |
| 		chat->botStatus = 0;
 | |
| 	} else {
 | |
| 		chat->participants.erase(user);
 | |
| 		chat->count--;
 | |
| 		chat->invitedByMe.remove(user);
 | |
| 		chat->admins.remove(user);
 | |
| 		if (user->isSelf()) {
 | |
| 			chat->setAdminRights(MTP_chatAdminRights(MTP_flags(0)));
 | |
| 		}
 | |
| 		if (const auto history = chat->owner().historyLoaded(chat)) {
 | |
| 			if (history->lastKeyboardFrom == user->id) {
 | |
| 				history->clearLastKeyboard();
 | |
| 			}
 | |
| 		}
 | |
| 		if (chat->botStatus > 0 && user->isBot()) {
 | |
| 			chat->refreshBotStatus();
 | |
| 		}
 | |
| 	}
 | |
| 	chat->session().changes().peerUpdated(chat, UpdateFlag::Members);
 | |
| }
 | |
| 
 | |
| void ApplyChatUpdate(
 | |
| 		not_null<ChatData*> chat,
 | |
| 		const MTPDupdateChatParticipantAdmin &update) {
 | |
| 	if (chat->applyUpdateVersion(update.vversion().v)
 | |
| 		!= ChatData::UpdateStatus::Good) {
 | |
| 		return;
 | |
| 	}
 | |
| 	const auto session = &chat->session();
 | |
| 	const auto user = chat->owner().userLoaded(update.vuser_id().v);
 | |
| 	if (!user) {
 | |
| 		chat->invalidateParticipants();
 | |
| 		return;
 | |
| 	}
 | |
| 	if (user->isSelf()) {
 | |
| 		chat->setAdminRights(MTP_chatAdminRights(mtpIsTrue(update.vis_admin())
 | |
| 			? MTP_flags(chat->defaultAdminRights(user))
 | |
| 			: MTP_flags(0)));
 | |
| 	}
 | |
| 	if (mtpIsTrue(update.vis_admin())) {
 | |
| 		if (chat->noParticipantInfo()) {
 | |
| 			session->api().requestFullPeer(chat);
 | |
| 		} else {
 | |
| 			chat->admins.emplace(user);
 | |
| 		}
 | |
| 	} else {
 | |
| 		chat->admins.erase(user);
 | |
| 	}
 | |
| 	session->changes().peerUpdated(chat, UpdateFlag::Admins);
 | |
| }
 | |
| 
 | |
| void ApplyChatUpdate(
 | |
| 		not_null<ChatData*> chat,
 | |
| 		const MTPDupdateChatDefaultBannedRights &update) {
 | |
| 	if (chat->applyUpdateVersion(update.vversion().v)
 | |
| 		!= ChatData::UpdateStatus::Good) {
 | |
| 		return;
 | |
| 	}
 | |
| 	chat->setDefaultRestrictions(update.vdefault_banned_rights());
 | |
| }
 | |
| 
 | |
| void ApplyChatUpdate(not_null<ChatData*> chat, const MTPDchatFull &update) {
 | |
| 	ApplyChatUpdate(chat, update.vparticipants());
 | |
| 
 | |
| 	if (const auto call = update.vcall()) {
 | |
| 		chat->setGroupCall(*call);
 | |
| 	} else {
 | |
| 		chat->clearGroupCall();
 | |
| 	}
 | |
| 	if (const auto as = update.vgroupcall_default_join_as()) {
 | |
| 		chat->setGroupCallDefaultJoinAs(peerFromMTP(*as));
 | |
| 	} else {
 | |
| 		chat->setGroupCallDefaultJoinAs(0);
 | |
| 	}
 | |
| 
 | |
| 	chat->setMessagesTTL(update.vttl_period().value_or_empty());
 | |
| 	if (const auto info = update.vbot_info()) {
 | |
| 		for (const auto &item : info->v) {
 | |
| 			item.match([&](const MTPDbotInfo &data) {
 | |
| 				const auto userId = data.vuser_id().v;
 | |
| 				if (const auto bot = chat->owner().userLoaded(userId)) {
 | |
| 					bot->setBotInfo(item);
 | |
| 					chat->session().api().fullPeerUpdated().notify(bot);
 | |
| 				}
 | |
| 			});
 | |
| 		}
 | |
| 	}
 | |
| 	chat->setFullFlags(update.vflags().v);
 | |
| 	if (const auto photo = update.vchat_photo()) {
 | |
| 		chat->setUserpicPhoto(*photo);
 | |
| 	} else {
 | |
| 		chat->setUserpicPhoto(MTP_photoEmpty(MTP_long(0)));
 | |
| 	}
 | |
| 	if (const auto invite = update.vexported_invite()) {
 | |
| 		chat->session().api().inviteLinks().setMyPermanent(chat, *invite);
 | |
| 	} else {
 | |
| 		chat->session().api().inviteLinks().clearMyPermanent(chat);
 | |
| 	}
 | |
| 	if (const auto pinned = update.vpinned_msg_id()) {
 | |
| 		SetTopPinnedMessageId(chat, pinned->v);
 | |
| 	}
 | |
| 	chat->checkFolder(update.vfolder_id().value_or_empty());
 | |
| 	chat->fullUpdated();
 | |
| 	chat->setAbout(qs(update.vabout()));
 | |
| 
 | |
| 	chat->session().api().applyNotifySettings(
 | |
| 		MTP_inputNotifyPeer(chat->input),
 | |
| 		update.vnotify_settings());
 | |
| }
 | |
| 
 | |
| void ApplyChatUpdate(
 | |
| 		not_null<ChatData*> chat,
 | |
| 		const MTPChatParticipants &participants) {
 | |
| 	const auto session = &chat->session();
 | |
| 	participants.match([&](const MTPDchatParticipantsForbidden &data) {
 | |
| 		if (const auto self = data.vself_participant()) {
 | |
| 			// self->
 | |
| 		}
 | |
| 		chat->count = -1;
 | |
| 		chat->invalidateParticipants();
 | |
| 	}, [&](const MTPDchatParticipants &data) {
 | |
| 		const auto status = chat->applyUpdateVersion(data.vversion().v);
 | |
| 		if (status == ChatData::UpdateStatus::TooOld) {
 | |
| 			return;
 | |
| 		}
 | |
| 		// Even if we skipped some updates, we got current participants
 | |
| 		// and we've requested peer from API to have current rights.
 | |
| 		chat->setVersion(data.vversion().v);
 | |
| 
 | |
| 		const auto &list = data.vparticipants().v;
 | |
| 		chat->count = list.size();
 | |
| 		chat->participants.clear();
 | |
| 		chat->invitedByMe.clear();
 | |
| 		chat->admins.clear();
 | |
| 		chat->setAdminRights(MTP_chatAdminRights(MTP_flags(0)));
 | |
| 		const auto selfUserId = session->userId();
 | |
| 		for (const auto &participant : list) {
 | |
| 			const auto userId = participant.match([&](const auto &data) {
 | |
| 				return data.vuser_id().v;
 | |
| 			});
 | |
| 			const auto user = chat->owner().userLoaded(userId);
 | |
| 			if (!user) {
 | |
| 				chat->invalidateParticipants();
 | |
| 				break;
 | |
| 			}
 | |
| 
 | |
| 			chat->participants.emplace(user);
 | |
| 
 | |
| 			const auto inviterId = participant.match([&](
 | |
| 					const MTPDchatParticipantCreator &data) {
 | |
| 				return 0;
 | |
| 			}, [&](const auto &data) {
 | |
| 				return data.vinviter_id().v;
 | |
| 			});
 | |
| 			if (inviterId == selfUserId) {
 | |
| 				chat->invitedByMe.insert(user);
 | |
| 			}
 | |
| 
 | |
| 			participant.match([&](const MTPDchatParticipantCreator &data) {
 | |
| 				chat->creator = userId;
 | |
| 			}, [&](const MTPDchatParticipantAdmin &data) {
 | |
| 				chat->admins.emplace(user);
 | |
| 				if (user->isSelf()) {
 | |
| 					chat->setAdminRights(MTP_chatAdminRights(
 | |
| 						MTP_flags(chat->defaultAdminRights(user))));
 | |
| 				}
 | |
| 			}, [](const MTPDchatParticipant &) {
 | |
| 			});
 | |
| 		}
 | |
| 		if (chat->participants.empty()) {
 | |
| 			return;
 | |
| 		}
 | |
| 		if (const auto history = chat->owner().historyLoaded(chat)) {
 | |
| 			if (history->lastKeyboardFrom) {
 | |
| 				const auto i = ranges::find(
 | |
| 					chat->participants,
 | |
| 					history->lastKeyboardFrom,
 | |
| 					&UserData::id);
 | |
| 				if (i == end(chat->participants)) {
 | |
| 					history->clearLastKeyboard();
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 		chat->refreshBotStatus();
 | |
| 		session->changes().peerUpdated(
 | |
| 			chat,
 | |
| 			UpdateFlag::Members | UpdateFlag::Admins);
 | |
| 	});
 | |
| }
 | |
| 
 | |
| } // namespace Data
 | 
