From a8bae3231ff2593bf333c204461964ed33221625 Mon Sep 17 00:00:00 2001
From: RadRussianRus <radrussianrus@gmail.com>
Date: Fri, 26 Aug 2022 22:06:21 +0300
Subject: [PATCH] [Option][GUI] Upload speed boost

---
 Telegram/Resources/langs/rewrites/en.json     |  7 +++
 .../SourceFiles/kotato/kotato_settings.cpp    | 24 +++++++++
 .../kotato/kotato_settings_menu.cpp           | 51 +++++++++++++++++++
 .../mtproto/dedicated_file_loader.cpp         | 10 +++-
 Telegram/SourceFiles/mtproto/facade.h         |  8 +--
 Telegram/SourceFiles/storage/file_upload.cpp  | 23 ++++++---
 Telegram/SourceFiles/storage/file_upload.h    |  2 +-
 7 files changed, 113 insertions(+), 12 deletions(-)

diff --git a/Telegram/Resources/langs/rewrites/en.json b/Telegram/Resources/langs/rewrites/en.json
index 7135ec517..fd76c661c 100644
--- a/Telegram/Resources/langs/rewrites/en.json
+++ b/Telegram/Resources/langs/rewrites/en.json
@@ -45,6 +45,13 @@
 	"ktg_fonts_use_system_font": "Use system font",
 	"ktg_fonts_use_original_metrics": "Use Open Sans height",
 	"ktg_settings_network": "Network",
+	"ktg_settings_net_speed_boost": "Upload speed boost",
+	"ktg_net_speed_boost_title": "Upload speed boost",
+	"ktg_net_speed_boost_desc": "Warning: changing this parameter to high values on slow networks can make even worse. Use at your own risk.\n\nYou'll need to restart app to save changes.",
+	"ktg_net_speed_boost_default": "Disabled",
+	"ktg_net_speed_boost_slight": "Slight",
+	"ktg_net_speed_boost_medium": "Medium",
+	"ktg_net_speed_boost_big": "Big",
 	"ktg_settings_system": "System",
 	"ktg_settings_other": "Other",
 	"ktg_profile_copy_id": "Copy ID",
diff --git a/Telegram/SourceFiles/kotato/kotato_settings.cpp b/Telegram/SourceFiles/kotato/kotato_settings.cpp
index 48e5b16b4..d47b1ec7f 100644
--- a/Telegram/SourceFiles/kotato/kotato_settings.cpp
+++ b/Telegram/SourceFiles/kotato/kotato_settings.cpp
@@ -155,6 +155,25 @@ CheckHandler IntLimitMin(int min) {
 }
 
 
+CheckHandler NetSpeedBoostConv(CheckHandler wrapped = nullptr) {
+	return [=] (QVariant value) -> QVariant {
+		auto newValue = 0;
+		if (value.canConvert<int>()) {
+			newValue = value.value<int>();
+		} else if (value.canConvert<QString>()) {
+			const auto strValue = value.value<QString>();
+			if (strValue == "high") {
+				newValue = 3;
+			} else if (strValue == "medium") {
+				newValue = 2;
+			} else if (strValue == "low") {
+				newValue = 1;
+			}
+		}
+		return (wrapped) ? wrapped(newValue) : newValue;
+	};
+}
+
 struct Definition {
 	SettingScope scope = SettingScope::Global;
 	SettingStorage storage = SettingStorage::MainJson;
@@ -266,6 +285,10 @@ const std::map<QString, Definition, std::greater<QString>> DefinitionMap {
 		.type = SettingType::IntSetting,
 		.defaultValue = 2,
 		.limitHandler = IntLimit(0, 2, 2), }},
+	{ "net_speed_boost", {
+		.type = SettingType::IntSetting,
+		.defaultValue = 0,
+		.limitHandler = NetSpeedBoostConv(IntLimit(0, 3)), }},
 };
 
 using OldOptionKey = QString;
@@ -411,6 +434,7 @@ QByteArray GenerateSettingsJson(bool areDefault = false) {
 
 	if (areDefault) {
 		settings.insert(qsl("version"), QString::number(AppKotatoVersion));
+		settings.insert(qsl("net_speed_boost"), QJsonValue(QJsonValue::Null));
 	}
 
 	auto document = QJsonDocument();
diff --git a/Telegram/SourceFiles/kotato/kotato_settings_menu.cpp b/Telegram/SourceFiles/kotato/kotato_settings_menu.cpp
index 5ddefdb3a..a19344605 100644
--- a/Telegram/SourceFiles/kotato/kotato_settings_menu.cpp
+++ b/Telegram/SourceFiles/kotato/kotato_settings_menu.cpp
@@ -48,6 +48,26 @@ namespace Settings {
 namespace {
 
 
+QString NetBoostLabel(int boost) {
+	switch (boost) {
+		case 0:
+			return ktr("ktg_net_speed_boost_default");
+
+		case 1:
+			return ktr("ktg_net_speed_boost_slight");
+
+		case 2:
+			return ktr("ktg_net_speed_boost_medium");
+
+		case 3:
+			return ktr("ktg_net_speed_boost_big");
+
+		default:
+			Unexpected("Boost in Settings::NetBoostLabel.");
+	}
+	return QString();
+}
+
 QString ChatIdLabel(int option) {
 	switch (option) {
 		case 0:
@@ -211,6 +231,37 @@ void SetupKotatoNetwork(not_null<Ui::VerticalLayout*> container) {
 	AddSkip(container);
 	AddSubsectionTitle(container, rktr("ktg_settings_network"));
 
+	const auto netBoostButton = container->add(
+		object_ptr<Button>(
+			container,
+			rktr("ktg_settings_net_speed_boost"),
+			st::settingsButtonNoIcon));
+	auto netBoostText = rpl::single(
+		NetBoostLabel(::Kotato::JsonSettings::GetIntWithPending("net_speed_boost"))
+	) | rpl::then(
+		::Kotato::JsonSettings::EventsWithPending(
+			"net_speed_boost"
+		) | rpl::map([] {
+			return NetBoostLabel(::Kotato::JsonSettings::GetIntWithPending("net_speed_boost"));
+		})
+	);
+	CreateRightLabel(
+		netBoostButton,
+		std::move(netBoostText),
+		st::settingsButtonNoIcon,
+		rktr("ktg_settings_net_speed_boost"));
+	netBoostButton->addClickHandler([=] {
+		Ui::show(Box<::Kotato::RadioBox>(
+			ktr("ktg_net_speed_boost_title"),
+			ktr("ktg_net_speed_boost_desc"),
+			::Kotato::JsonSettings::GetIntWithPending("net_speed_boost"),
+			4,
+			NetBoostLabel,
+			[=] (int value) {
+				::Kotato::JsonSettings::SetAfterRestart("net_speed_boost", value);
+				::Kotato::JsonSettings::Write();
+			}, true));
+	});
 
 	AddSkip(container);
 }
diff --git a/Telegram/SourceFiles/mtproto/dedicated_file_loader.cpp b/Telegram/SourceFiles/mtproto/dedicated_file_loader.cpp
index 676aa047c..bee70ca91 100644
--- a/Telegram/SourceFiles/mtproto/dedicated_file_loader.cpp
+++ b/Telegram/SourceFiles/mtproto/dedicated_file_loader.cpp
@@ -7,6 +7,7 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
 */
 #include "mtproto/dedicated_file_loader.h"
 
+#include "kotato/kotato_settings.h"
 #include "mtproto/facade.h"
 #include "main/main_account.h" // Account::sessionChanges.
 #include "main/main_session.h" // Session::account.
@@ -80,6 +81,11 @@ std::optional<DedicatedLoader::File> ParseFile(
 	return DedicatedLoader::File{ name, size, fields.vdc_id().v, location };
 }
 
+int RequestCount() {
+	static const auto count = 2 + (2 * ::Kotato::JsonSettings::GetInt("net_speed_boost"));
+	return count;
+}
+
 } // namespace
 
 WeakInstance::WeakInstance(base::weak_ptr<Main::Session> session)
@@ -310,7 +316,7 @@ void DedicatedLoader::startLoading() {
 }
 
 void DedicatedLoader::sendRequest() {
-	if (_requests.size() >= kRequestsCount || _offset >= _size) {
+	if (_requests.size() >= RequestCount() || _offset >= _size) {
 		return;
 	}
 	const auto offset = _offset;
@@ -326,7 +332,7 @@ void DedicatedLoader::sendRequest() {
 		MTP::updaterDcId(_dcId));
 	_offset += kChunkSize;
 
-	if (_requests.size() < kRequestsCount) {
+	if (_requests.size() < RequestCount()) {
 		base::call_delayed(kNextRequestDelay, this, [=] { sendRequest(); });
 	}
 }
diff --git a/Telegram/SourceFiles/mtproto/facade.h b/Telegram/SourceFiles/mtproto/facade.h
index 6f31960da..989fdad4b 100644
--- a/Telegram/SourceFiles/mtproto/facade.h
+++ b/Telegram/SourceFiles/mtproto/facade.h
@@ -42,6 +42,8 @@ constexpr ShiftedDcId groupCallStreamDcId(DcId dcId) {
 
 constexpr auto kUploadSessionsCount = 2;
 
+constexpr auto kUploadSessionsCountMax = 8;
+
 namespace details {
 
 constexpr ShiftedDcId downloadDcId(DcId dcId, int index) {
@@ -92,7 +94,7 @@ inline DcId getTemporaryIdFromRealDcId(ShiftedDcId shiftedDcId) {
 namespace details {
 
 constexpr ShiftedDcId uploadDcId(DcId dcId, int index) {
-	static_assert(kUploadSessionsCount < kMaxMediaDcCount, "Too large MTPUploadSessionsCount!");
+	static_assert(kUploadSessionsCountMax < kMaxMediaDcCount, "Too large MTPUploadSessionsCount!");
 	return ShiftDcId(dcId, kBaseUploadDcShift + index);
 };
 
@@ -101,14 +103,14 @@ constexpr ShiftedDcId uploadDcId(DcId dcId, int index) {
 // send(req, callbacks, MTP::uploadDcId(index)) - for upload shifted dc id
 // uploading always to the main dc so BareDcId(result) == 0
 inline ShiftedDcId uploadDcId(int index) {
-	Expects(index >= 0 && index < kUploadSessionsCount);
+	Expects(index >= 0 && index < kUploadSessionsCountMax);
 
 	return details::uploadDcId(0, index);
 };
 
 constexpr bool isUploadDcId(ShiftedDcId shiftedDcId) {
 	return (shiftedDcId >= details::uploadDcId(0, 0))
-		&& (shiftedDcId < details::uploadDcId(0, kUploadSessionsCount - 1) + kDcShift);
+		&& (shiftedDcId < details::uploadDcId(0, kUploadSessionsCountMax - 1) + kDcShift);
 }
 
 inline ShiftedDcId destroyKeyNextDcId(ShiftedDcId shiftedDcId) {
diff --git a/Telegram/SourceFiles/storage/file_upload.cpp b/Telegram/SourceFiles/storage/file_upload.cpp
index f2a014f7e..23dd4928e 100644
--- a/Telegram/SourceFiles/storage/file_upload.cpp
+++ b/Telegram/SourceFiles/storage/file_upload.cpp
@@ -7,6 +7,7 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
 */
 #include "storage/file_upload.h"
 
+#include "kotato/kotato_settings.h"
 #include "api/api_editing.h"
 #include "api/api_send_progress.h"
 #include "storage/localimageloader.h"
@@ -56,6 +57,16 @@ constexpr auto kKillSessionTimeout = 15 * crl::time(1000);
 	return Core::IsMimeSticker(mime) ? "WEBP" : "JPG";
 }
 
+int UploadSessionsCount() {
+	static const auto count = 2 + (2 * ::Kotato::JsonSettings::GetInt("net_speed_boost"));
+	return count;
+}
+
+int UploadSessionsInterval() {
+	static const auto interval = 500 - (100 * ::Kotato::JsonSettings::GetInt("net_speed_boost"));
+	return interval;
+}
+
 } // namespace
 
 struct Uploader::File {
@@ -368,7 +379,7 @@ void Uploader::currentFailed() {
 	dcMap.clear();
 	uploadingId = FullMsgId();
 	sentSize = 0;
-	for (int i = 0; i < MTP::kUploadSessionsCount; ++i) {
+	for (int i = 0; i < UploadSessionsCount(); ++i) {
 		sentSizes[i] = 0;
 	}
 
@@ -395,13 +406,13 @@ void Uploader::notifyFailed(FullMsgId id, const File &file) {
 }
 
 void Uploader::stopSessions() {
-	for (int i = 0; i < MTP::kUploadSessionsCount; ++i) {
+	for (int i = 0; i < UploadSessionsCount(); ++i) {
 		_api->instance().stopSession(MTP::uploadDcId(i));
 	}
 }
 
 void Uploader::sendNext() {
-	if (sentSize >= kMaxUploadFileParallelSize || _pausedId.msg) {
+	if (sentSize >= (UploadSessionsCount() * 512 * 1024) || _pausedId.msg) {
 		return;
 	}
 
@@ -426,7 +437,7 @@ void Uploader::sendNext() {
 	auto &uploadingData = i->second;
 
 	auto todc = 0;
-	for (auto dc = 1; dc != MTP::kUploadSessionsCount; ++dc) {
+	for (auto dc = 1; dc != UploadSessionsCount(); ++dc) {
 		if (sentSizes[dc] < sentSizes[todc]) {
 			todc = dc;
 		}
@@ -619,7 +630,7 @@ void Uploader::sendNext() {
 
 		parts.erase(part);
 	}
-	_nextTimer.callOnce(kUploadRequestInterval);
+	_nextTimer.callOnce(crl::time(UploadSessionsInterval()));
 }
 
 void Uploader::cancel(const FullMsgId &msgId) {
@@ -676,7 +687,7 @@ void Uploader::clear() {
 	cancelRequests();
 	dcMap.clear();
 	sentSize = 0;
-	for (int i = 0; i < MTP::kUploadSessionsCount; ++i) {
+	for (int i = 0; i < UploadSessionsCount(); ++i) {
 		_api->instance().stopSession(MTP::uploadDcId(i));
 		sentSizes[i] = 0;
 	}
diff --git a/Telegram/SourceFiles/storage/file_upload.h b/Telegram/SourceFiles/storage/file_upload.h
index de5a84a19..7e4a83fc6 100644
--- a/Telegram/SourceFiles/storage/file_upload.h
+++ b/Telegram/SourceFiles/storage/file_upload.h
@@ -127,7 +127,7 @@ private:
 	base::flat_map<mtpRequestId, int32> docRequestsSent;
 	base::flat_map<mtpRequestId, int32> dcMap;
 	uint32 sentSize = 0; // FileSize: Right now any file size fits 32 bit.
-	uint32 sentSizes[MTP::kUploadSessionsCount] = { 0 };
+	uint32 sentSizes[MTP::kUploadSessionsCountMax] = { 0 };
 
 	FullMsgId uploadingId;
 	FullMsgId _pausedId;