From 4b6ce4bb145bd87efd27592e26cd7eba4035e11f Mon Sep 17 00:00:00 2001
From: Dion Hulse <dion@wordpress.org>
Date: Tue, 10 Dec 2024 17:19:53 +1000
Subject: [PATCH] Revalidation: Allow for the data-2fa-required element to be a
 <button> or other element.

---
 revalidation/script.js | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/revalidation/script.js b/revalidation/script.js
index 48fa36c2..b320538a 100644
--- a/revalidation/script.js
+++ b/revalidation/script.js
@@ -29,7 +29,7 @@ window.wp = window.wp || {};
 
 	// Does the provided URL look like a revalidation url?
 	const urlLooksLikeRevalidationURL = function( url ) {
-		return url.includes( 'wp-login.php' ) && url.includes( 'action=revalidate_2fa' );
+		return url && url.includes( 'wp-login.php' ) && url.includes( 'action=revalidate_2fa' );
 	};
 
 	// Display a modal dialog asking to revalidate.
@@ -159,11 +159,11 @@ window.wp = window.wp || {};
 	};
 
 	// Attach event listeners to all revalidate links and those that require 2FA sessions.
-	document.querySelectorAll( 'a[href*="action=revalidate_2fa"], a[data-2fa-required]' ).forEach(
+	document.querySelectorAll( 'a[href*="action=revalidate_2fa"], [data-2fa-required]' ).forEach(
 		(el) => el.addEventListener( 'click', maybeRevalidateOnLinkNavigate )
 	);
 
 	// Watch for revalidation completion.
 	window.addEventListener( 'message', messageHandler );
 
-} )( wporgTwoFactorRevalidation, window.wp );
\ No newline at end of file
+} )( wporgTwoFactorRevalidation, window.wp );