From 8976fa163e71d09a5f7a3b9997109cef0edc5d17 Mon Sep 17 00:00:00 2001
From: Pavel Djundik <xPaw@users.noreply.github.com>
Date: Wed, 19 Feb 2020 20:20:28 +0200
Subject: [PATCH] Do not reconnect if STS cap is received in CAP NEW

---
 src/plugins/irc-events/cap.js | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/src/plugins/irc-events/cap.js b/src/plugins/irc-events/cap.js
index 3b1d7c8b..da20d3cc 100644
--- a/src/plugins/irc-events/cap.js
+++ b/src/plugins/irc-events/cap.js
@@ -7,14 +7,14 @@ module.exports = function(irc, network) {
 	const client = this;
 
 	irc.on("cap ls", (data) => {
-		handleSTS(data);
+		handleSTS(data, true);
 	});
 
 	irc.on("cap new", (data) => {
-		handleSTS(data);
+		handleSTS(data, false);
 	});
 
-	function handleSTS(data) {
+	function handleSTS(data, shouldReconnect) {
 		if (!Object.prototype.hasOwnProperty.call(data.capabilities, "sts")) {
 			return;
 		}
@@ -50,8 +50,12 @@ module.exports = function(irc, network) {
 				true
 			);
 
-			// Forcefully end the connection
-			irc.connection.end();
+			// Forcefully end the connection if STS is seen in CAP LS
+			// We will update the port and tls setting if we see CAP NEW,
+			// but will not force a reconnection
+			if (shouldReconnect) {
+				irc.connection.end();
+			}
 
 			// Update the port
 			network.port = port;
@@ -63,8 +67,10 @@ module.exports = function(irc, network) {
 			irc.options.tls = true;
 			irc.options.rejectUnauthorized = true;
 
-			// Start a new connection
-			irc.connect();
+			if (shouldReconnect) {
+				// Start a new connection
+				irc.connect();
+			}
 
 			client.save();
 		}