Support similar bots section.

Telegram/SourceFiles/api/api_chat_participants.cpp

@@ -211,11 +211,10 @@ void ApplyBotsList(
 		Data::PeerUpdate::Flag::FullInfo);
 }
 
-[[nodiscard]] ChatParticipants::Channels ParseSimilar(
+[[nodiscard]] ChatParticipants::Peers ParseSimilarChannels(
 		not_null<Main::Session*> session,
 		const MTPmessages_Chats &chats) {
-	auto result = ChatParticipants::Channels();
-	std::vector<not_null<ChannelData*>>();
+	auto result = ChatParticipants::Peers();
 	chats.match([&](const auto &data) {
 		const auto &list = data.vchats().v;
 		result.list.reserve(list.size());
@@ -234,10 +233,29 @@ void ApplyBotsList(
 	return result;
 }
 
-[[nodiscard]] ChatParticipants::Channels ParseSimilar(
+[[nodiscard]] ChatParticipants::Peers ParseSimilarChannels(
 		not_null<ChannelData*> channel,
 		const MTPmessages_Chats &chats) {
-	return ParseSimilar(&channel->session(), chats);
+	return ParseSimilarChannels(&channel->session(), chats);
+}
+
+[[nodiscard]] ChatParticipants::Peers ParseSimilarBots(
+		not_null<Main::Session*> session,
+		const MTPusers_Users &users) {
+	auto result = ChatParticipants::Peers();
+	users.match([&](const auto &data) {
+		const auto &list = data.vusers().v;
+		result.list.reserve(list.size());
+		for (const auto &user : list) {
+			result.list.push_back(session->data().processUser(user));
+		}
+		if constexpr (MTPDusers_usersSlice::Is<decltype(data)>()) {
+			if (session->premiumPossible()) {
+				result.more = data.vcount().v - data.vusers().v.size();
+			}
+		}
+	});
+	return result;
 }
 
 } // namespace
@@ -782,52 +800,65 @@ void ChatParticipants::unblock(
 	_kickRequests.emplace(kick, requestId);
 }
 
-void ChatParticipants::loadSimilarChannels(not_null<ChannelData*> channel) {
-	if (!channel->isBroadcast()) {
-		return;
-	} else if (const auto i = _similar.find(channel); i != end(_similar)) {
+void ChatParticipants::loadSimilarPeers(not_null<PeerData*> peer) {
+	if (const auto i = _similar.find(peer); i != end(_similar)) {
 		if (i->second.requestId
-			|| !i->second.channels.more
-			|| !channel->session().premium()) {
+			|| !i->second.peers.more
+			|| !peer->session().premium()) {
 			return;
 		}
 	}
-	using Flag = MTPchannels_GetChannelRecommendations::Flag;
-	_similar[channel].requestId = _api.request(
-		MTPchannels_GetChannelRecommendations(
-			MTP_flags(Flag::f_channel),
-			channel->inputChannel)
-	).done([=](const MTPmessages_Chats &result) {
-		auto &similar = _similar[channel];
-		similar.requestId = 0;
-		auto parsed = ParseSimilar(channel, result);
-		if (similar.channels == parsed) {
-			return;
-		}
-		similar.channels = std::move(parsed);
-		if (const auto history = channel->owner().historyLoaded(channel)) {
-			if (const auto item = history->joinedMessageInstance()) {
-				history->owner().requestItemResize(item);
+	if (const auto channel = peer->asBroadcast()) {
+		using Flag = MTPchannels_GetChannelRecommendations::Flag;
+		_similar[peer].requestId = _api.request(
+			MTPchannels_GetChannelRecommendations(
+				MTP_flags(Flag::f_channel),
+				channel->inputChannel)
+		).done([=](const MTPmessages_Chats &result) {
+			auto &similar = _similar[channel];
+			similar.requestId = 0;
+			auto parsed = ParseSimilarChannels(channel, result);
+			if (similar.peers == parsed) {
+				return;
 			}
-		}
-		_similarLoaded.fire_copy(channel);
-	}).send();
+			similar.peers = std::move(parsed);
+			if (const auto history = channel->owner().historyLoaded(channel)) {
+				if (const auto item = history->joinedMessageInstance()) {
+					history->owner().requestItemResize(item);
+				}
+			}
+			_similarLoaded.fire_copy(channel);
+		}).send();
+	} else if (const auto bot = peer->asBot()) {
+		_similar[peer].requestId = _api.request(
+			MTPbots_GetBotRecommendations(bot->inputUser)
+		).done([=](const MTPusers_Users &result) {
+			auto &similar = _similar[peer];
+			similar.requestId = 0;
+			auto parsed = ParseSimilarBots(&peer->session(), result);
+			if (similar.peers == parsed) {
+				return;
+			}
+			similar.peers = std::move(parsed);
+			_similarLoaded.fire_copy(peer);
+		}).send();
+	}
 }
 
-auto ChatParticipants::similar(not_null<ChannelData*> channel)
--> const Channels & {
-	const auto i = channel->isBroadcast()
-		? _similar.find(channel)
+auto ChatParticipants::similar(not_null<PeerData*> peer)
+-> const Peers & {
+	const auto i = (peer->isBroadcast() || peer->isBot())
+		? _similar.find(peer)
 		: end(_similar);
 	if (i != end(_similar)) {
-		return i->second.channels;
+		return i->second.peers;
 	}
-	static const auto empty = Channels();
+	static const auto empty = Peers();
 	return empty;
 }
 
 auto ChatParticipants::similarLoaded() const
--> rpl::producer<not_null<ChannelData*>> {
+-> rpl::producer<not_null<PeerData*>> {
 	return _similarLoaded.events();
 }
 
@@ -841,15 +872,15 @@ void ChatParticipants::loadRecommendations() {
 			MTP_inputChannelEmpty())
 	).done([=](const MTPmessages_Chats &result) {
 		_recommendations.requestId = 0;
-		auto parsed = ParseSimilar(_session, result);
-		_recommendations.channels = std::move(parsed);
-		_recommendations.channels.more = 0;
+		auto parsed = ParseSimilarChannels(_session, result);
+		_recommendations.peers = std::move(parsed);
+		_recommendations.peers.more = 0;
 		_recommendationsLoaded = true;
 	}).send();
 }
 
-const ChatParticipants::Channels &ChatParticipants::recommendations() const {
-	return _recommendations.channels;
+const ChatParticipants::Peers &ChatParticipants::recommendations() const {
+	return _recommendations.peers;
 }
 
 rpl::producer<> ChatParticipants::recommendationsLoaded() const {

Telegram/SourceFiles/api/api_chat_participants.h

@@ -138,27 +138,27 @@ public:
 		not_null<ChannelData*> channel,
 		not_null<PeerData*> participant);
 
-	void loadSimilarChannels(not_null<ChannelData*> channel);
+	void loadSimilarPeers(not_null<PeerData*> peer);
 
-	struct Channels {
-		std::vector<not_null<ChannelData*>> list;
+	struct Peers {
+		std::vector<not_null<PeerData*>> list;
 		int more = 0;
 
 		friend inline bool operator==(
-			const Channels &,
-			const Channels &) = default;
+			const Peers &,
+			const Peers &) = default;
 	};
-	[[nodiscard]] const Channels &similar(not_null<ChannelData*> channel);
+	[[nodiscard]] const Peers &similar(not_null<PeerData*> peer);
 	[[nodiscard]] auto similarLoaded() const
-		-> rpl::producer<not_null<ChannelData*>>;
+		-> rpl::producer<not_null<PeerData*>>;
 
 	void loadRecommendations();
-	[[nodiscard]] const Channels &recommendations() const;
+	[[nodiscard]] const Peers &recommendations() const;
 	[[nodiscard]] rpl::producer<> recommendationsLoaded() const;
 
 private:
-	struct SimilarChannels {
-		Channels channels;
+	struct SimilarPeers {
+		Peers peers;
 		mtpRequestId requestId = 0;
 	};
 
@@ -186,10 +186,10 @@ private:
 		not_null<PeerData*>>;
 	base::flat_map<KickRequest, mtpRequestId> _kickRequests;
 
-	base::flat_map<not_null<ChannelData*>, SimilarChannels> _similar;
-	rpl::event_stream<not_null<ChannelData*>> _similarLoaded;
+	base::flat_map<not_null<PeerData*>, SimilarPeers> _similar;
+	rpl::event_stream<not_null<PeerData*>> _similarLoaded;
 
-	SimilarChannels _recommendations;
+	SimilarPeers _recommendations;
 	rpl::variable<bool> _recommendationsLoaded = false;
 
 };