Reconnection bugfixes and improvements (#30)

* Reconnect bugfix

* one line

* added to the readme

README.md

@@ -18,11 +18,10 @@
 * Respects Discords 15sec limit when it comes to updating your status
 * Stable or Insiders build detection
 * Debug mode detection
+* Easily manually reconnect to discord
 
 ## The rich presence won't show after my PC has been put to sleep / after I lost internet!
-It will only attempt to reconnect 20 times. After it hit that threshold you will have to manually enable it again.  
+It will only attempt to reconnect 20 times. After it hit that threshold you will have to manually click the `Reconnect to Discord` button in the bottom left of the window
-Just open the command pallette and execute the enable command for the extension / or reload the window.  
-You can also set the reconnectThreshold in the settings to something very high, for example 9999 or Infinity to never stop trying to reconnect.
 
 ## Contributing
 

src/extension.ts

@@ -25,8 +25,8 @@ let rpc: Client;
 const eventHandlers: Set<Disposable> = new Set();
 // Define the config variable and its type.
 let config;
-// Define the reconnect timer and its type.
+// Define the reconnecting var and its type.
-let reconnectTimer: NodeJS.Timer;
+let reconnecting: boolean;
 // Define the reconnect counter and its type.
 let reconnectCounter = 0;
 // Define the last known file name and its type.
@@ -66,7 +66,8 @@ export function activate(context: ExtensionContext) {
 	const reconnecter = commands.registerCommand('discord.reconnect', async () => {
 		if (rpc) try { await destroyRPC(); } catch {}
 		initRPC(config.get('clientID'), true);
-		window.showInformationMessage('Reconnecting to Discord RPC');
+
+		if (!config.get('silent')) window.showInformationMessage('Reconnecting to Discord RPC');
 
 		if (statusBarIcon) statusBarIcon.text = '$(pulse) Reconnecting';
 	});
@@ -88,18 +89,17 @@ function initRPC(clientID: string, loud?: boolean): void {
 
 	// Once the RPC Client is ready, set the activity.
 	rpc.once('ready', () => {
-		if (loud) window.showInformationMessage('Successfully reconnected to Discord RPC');
+		// Announce the reconnection
+		if (loud && !config.get('silent')) window.showInformationMessage('Successfully reconnected to Discord RPC');
 
 		// Remove icon if connected
-		if (statusBarIcon) statusBarIcon.dispose();
+		if (statusBarIcon) {
-
+			statusBarIcon.dispose();
-		// This is purely for safety measures.
+			statusBarIcon = null;
-		if (reconnectTimer) {
-			// Clear the reconnect interval.
-			clearInterval(reconnectTimer);
-			// Null reconnect variable.
-			reconnectTimer = null;
 		}
+
+		// Stop from reconnecing.
+		reconnecting = false;
 		// This is purely for safety measures.
 		if (activityTimer) {
 			// Clear the activity interval.
@@ -117,11 +117,12 @@ function initRPC(clientID: string, loud?: boolean): void {
 		rpc.transport.once('close', async () => {
 			if (!config.get('enabled')) return;
 			await destroyRPC();
-			// Set an interval for reconnecting.
+
-			reconnectTimer = setInterval(() => {
+			// Set the client to begin reconnecting
-				reconnectCounter++;
+			reconnecting = true;
 			initRPC(config.get('clientID'));
-			}, 5000);
+			// Create reconnecting button
+			createButon(true);
 		});
 
 		// Update the user's activity to the `activity` variable.
@@ -133,15 +134,24 @@ function initRPC(clientID: string, loud?: boolean): void {
 
 	// Log in to the RPC Client, and check whether or not it errors.
 	rpc.login(clientID).catch(async error => {
-		if (reconnectTimer) {
+		// Check if the client is reconnecting
-			// Destroy and dispose of everything after a default of 20 reconnect attempts
+		if (reconnecting) {
+			// Destroy and dispose of everything after the set reconnect attempts
 			if (reconnectCounter >= config.get('reconnectThreshold')) {
+				// Create reconnect button
 				createButon();
 				await destroyRPC();
 			} else {
+				// Increment the counter
+				reconnectCounter++;
+				// Create reconnecting button
+				createButon(true);
+				// Retry connection
+				initRPC(config.get('clientID'));
 				return;
 			}
 		}
+		// Announce failure
 		if (!config.get('silent')) {
 			if (error.message.includes('ENOENT')) window.showErrorMessage('No Discord Client detected!');
 			else window.showErrorMessage(`Couldn't connect to Discord via RPC: ${error.message}`);
@@ -151,14 +161,36 @@ function initRPC(clientID: string, loud?: boolean): void {
 }
 
 // Create reconnect button
-function createButon(): void {
+function createButon(isReconnecting?: boolean): void {
+	// Check if the button already
 	if (!statusBarIcon) {
+		// Create the icon
 		statusBarIcon = window.createStatusBarItem(StatusBarAlignment.Left);
+		// Check if the client is reconnecting
+		if (isReconnecting) {
+			// Show attempts left
+			const attempts = config.get('reconnectThreshold') - reconnectCounter;
+			statusBarIcon.text = `$(issue-reopened) Reconnecting: ${attempts} attempt${attempts === 1 ? '' : 's'} left`;
+			statusBarIcon.command = '';
+		} else {
+			// Show button to reconnect
 			statusBarIcon.text = '$(plug) Reconnect to Discord';
 			statusBarIcon.command = 'discord.reconnect';
+		}
+		// Show the button
 		statusBarIcon.show();
 	} else  {
+		// Check if the client is reconnecting
+		if (isReconnecting) {
+			// Show attempts left
+			const attempts = config.get('reconnectThreshold') - reconnectCounter;
+			statusBarIcon.text = `$(issue-reopened) Reconnecting: ${attempts} attempt${attempts === 1 ? '' : 's'} left`;
+			statusBarIcon.command = '';
+		} else {
+			// Show button to reconnect
 			statusBarIcon.text = '$(plug) Reconnect to Discord';
+			statusBarIcon.command = 'discord.reconnect';
+		}
 	}
 }
 
@@ -166,10 +198,8 @@ function createButon(): void {
 async function destroyRPC(): Promise<void> {
 	// Do not continue if RPC isn't initalized.
 	if (!rpc) return;
-	// Clear the reconnect interval.
+	// Stop reconnecting.
-	if (reconnectTimer) clearInterval(reconnectTimer);
+	reconnecting = false;
-	// Null reconnect variable.
-	reconnectTimer = null;
 	// Clear the activity interval.
 	if (activityTimer) clearInterval(activityTimer);
 	// Null the activity timer.