Fixed Api requests for statistics with non-default DC.

Telegram/SourceFiles/api/api_statistics.cpp

@@ -187,40 +187,37 @@ namespace {
 
 } // namespace
 
-Statistics::Statistics(not_null<ApiWrap*> api)
-: _api(&api->instance()) {
+Statistics::Statistics(not_null<ChannelData*> channel)
+: _channel(channel)
+, _api(&channel->session().api().instance()) {
 }
 
-rpl::producer<rpl::no_value, QString> Statistics::request(
-		not_null<PeerData*> peer) {
+rpl::producer<rpl::no_value, QString> Statistics::request() {
 	return [=](auto consumer) {
 		auto lifetime = rpl::lifetime();
-		const auto channel = peer->asChannel();
-		if (!channel) {
-			return lifetime;
-		}
 
-		if (!channel->isMegagroup()) {
+		const auto dcId = _channel->owner().statsDcId(_channel);
+		if (!_channel->isMegagroup()) {
 			_api.request(MTPstats_GetBroadcastStats(
 				MTP_flags(MTPstats_GetBroadcastStats::Flags(0)),
-				channel->inputChannel
+				_channel->inputChannel
 			)).done([=](const MTPstats_BroadcastStats &result) {
 				_channelStats = ChannelStatisticsFromTL(result.data());
 				consumer.put_done();
 			}).fail([=](const MTP::Error &error) {
 				consumer.put_error_copy(error.type());
-			}).send();
+			}).toDC(MTP::ShiftDcId(dcId, MTP::kStatsDcShift)).send();
 		} else {
 			_api.request(MTPstats_GetMegagroupStats(
 				MTP_flags(MTPstats_GetMegagroupStats::Flags(0)),
-				channel->inputChannel
+				_channel->inputChannel
 			)).done([=](const MTPstats_MegagroupStats &result) {
 				_supergroupStats = SupergroupStatisticsFromTL(result.data());
-				peer->owner().processUsers(result.data().vusers());
+				_channel->owner().processUsers(result.data().vusers());
 				consumer.put_done();
 			}).fail([=](const MTP::Error &error) {
 				consumer.put_error_copy(error.type());
-			}).send();
+			}).toDC(MTP::ShiftDcId(dcId, MTP::kStatsDcShift)).send();
 		}
 
 		return lifetime;
@@ -228,15 +225,11 @@ rpl::producer<rpl::no_value, QString> Statistics::request(
 }
 
 Statistics::GraphResult Statistics::requestZoom(
-		not_null<PeerData*> peer,
 		const QString &token,
 		float64 x) {
 	return [=](auto consumer) {
 		auto lifetime = rpl::lifetime();
-		const auto channel = peer->asChannel();
-		if (!channel) {
-			return lifetime;
-		}
+		const auto dcId = _channel->owner().statsDcId(_channel);
 		const auto wasEmpty = _zoomDeque.empty();
 		_zoomDeque.push_back([=] {
 			_api.request(MTPstats_LoadAsyncGraph(
@@ -256,7 +249,7 @@ Statistics::GraphResult Statistics::requestZoom(
 				}
 			}).fail([=](const MTP::Error &error) {
 				consumer.put_error_copy(error.type());
-			}).send();
+			}).toDC(MTP::ShiftDcId(dcId, MTP::kStatsDcShift)).send();
 		});
 		if (wasEmpty) {
 			_zoomDeque.front()();
@@ -266,32 +259,6 @@ Statistics::GraphResult Statistics::requestZoom(
 	};
 }
 
-Statistics::GraphResult Statistics::requestMessage(
-		not_null<PeerData*> peer,
-		MsgId msgId) {
-	return [=](auto consumer) {
-		auto lifetime = rpl::lifetime();
-		const auto channel = peer->asChannel();
-		if (!channel) {
-			return lifetime;
-		}
-
-		_api.request(MTPstats_GetMessageStats(
-			MTP_flags(MTPstats_GetMessageStats::Flags(0)),
-			channel->inputChannel,
-			MTP_int(msgId.bare)
-		)).done([=](const MTPstats_MessageStats &result) {
-			consumer.put_next(
-				StatisticalGraphFromTL(result.data().vviews_graph()));
-			consumer.put_done();
-		}).fail([=](const MTP::Error &error) {
-			consumer.put_error_copy(error.type());
-		}).send();
-
-		return lifetime;
-	};
-}
-
 Data::ChannelStatistics Statistics::channelStats() const {
 	return _channelStats;
 }
@@ -314,6 +281,7 @@ void PublicForwards::request(
 	if (_requestId) {
 		return;
 	}
+	const auto dcId = _channel->owner().statsDcId(_channel);
 	const auto offsetPeer = _channel->owner().peer(token.fullId.peer);
 	const auto tlOffsetPeer = offsetPeer
 		? offsetPeer->input
@@ -396,7 +364,7 @@ void PublicForwards::request(
 		});
 	}).fail([=] {
 		_requestId = 0;
-	}).send();
+	}).toDC(MTP::ShiftDcId(dcId, MTP::kStatsDcShift)).send();
 }
 
 MessageStatistics::MessageStatistics(
@@ -416,6 +384,7 @@ void MessageStatistics::request(Fn<void(Data::MessageStatistics)> done) {
 	if (_channel->isMegagroup()) {
 		return;
 	}
+	const auto dcId = _channel->owner().statsDcId(_channel);
 
 	const auto requestFirstPublicForwards = [=](
 			const Data::StatisticalGraph &messageGraph,
@@ -484,7 +453,7 @@ void MessageStatistics::request(Fn<void(Data::MessageStatistics)> done) {
 			StatisticalGraphFromTL(result.data().vviews_graph()));
 	}).fail([=](const MTP::Error &error) {
 		requestPrivateForwards({});
-	}).send();
+	}).toDC(MTP::ShiftDcId(dcId, MTP::kStatsDcShift)).send();
 
 }
 

Telegram/SourceFiles/api/api_statistics.h

@@ -11,7 +11,6 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
 #include "data/data_statistics.h"
 #include "mtproto/sender.h"
 
-class ApiWrap;
 class ChannelData;
 class PeerData;
 
@@ -19,23 +18,19 @@ namespace Api {
 
 class Statistics final {
 public:
-	explicit Statistics(not_null<ApiWrap*> api);
+	explicit Statistics(not_null<ChannelData*> channel);
 
-	[[nodiscard]] rpl::producer<rpl::no_value, QString> request(
-		not_null<PeerData*> peer);
+	[[nodiscard]] rpl::producer<rpl::no_value, QString> request();
 	using GraphResult = rpl::producer<Data::StatisticalGraph, QString>;
 	[[nodiscard]] GraphResult requestZoom(
-		not_null<PeerData*> peer,
 		const QString &token,
 		float64 x);
-	[[nodiscard]] GraphResult requestMessage(
-		not_null<PeerData*> peer,
-		MsgId msgId);
 
 	[[nodiscard]] Data::ChannelStatistics channelStats() const;
 	[[nodiscard]] Data::SupergroupStatistics supergroupStats() const;
 
 private:
+	const not_null<ChannelData*> _channel;
 	Data::ChannelStatistics _channelStats;
 	Data::SupergroupStatistics _supergroupStats;
 	MTP::Sender _api;