Enqueue logout script only if logout cookie is set

Author: psrpinto

src/Sessions/logout.php

@@ -2,14 +2,20 @@
 
 namespace Automattic\Chatrix\Sessions;
 
+use const Automattic\Chatrix\LOGOUT_COOKIE_NAME;
+use const Automattic\Chatrix\SCRIPT_HANDLE_LOGOUT;
+
 function init_logout() {
 	// Set a cookie when user logs out.
 	add_action(
 		'wp_logout',
 		function () {
 			// Expire in 10 minutes.
 			$expiration = time() + ( 10 * 60 );
-			setcookie( 'chatrix-logout', 'true', $expiration, COOKIEPATH, COOKIE_DOMAIN );
+			setcookie( LOGOUT_COOKIE_NAME, 'true', $expiration, COOKIEPATH, COOKIE_DOMAIN );
+
+			// Enqueue the script that will perform logout on the client.
+			wp_enqueue_script( SCRIPT_HANDLE_LOGOUT );
 		}
 	);
 }

src/plugin.php

@@ -10,6 +10,7 @@
 const SCRIPT_HANDLE_APP    = 'chatrix-app';
 const SCRIPT_HANDLE_LOGOUT = 'chatrix-logout';
 const CONFIG_VARIABLE      = 'ChatrixConfig';
+const LOGOUT_COOKIE_NAME   = 'chatrix-logout';
 
 function main() {
 	init_logout();
@@ -51,7 +52,11 @@ function () {
 					automattic_chatrix_version(),
 					false
 				);
-				wp_enqueue_script( SCRIPT_HANDLE_LOGOUT );
+
+				// Enqueue the logout script only if the logout cookie is set.
+				if ( isset( $_COOKIE[ LOGOUT_COOKIE_NAME ] ) ) {
+					wp_enqueue_script( SCRIPT_HANDLE_LOGOUT );
+				}
 			}
 		}
 	);