Check whether the dom elements exist before changing their attributes

woocommerce · Nov 24, 2023 · 863cfc7 · 863cfc7
1 parent 646299c
commit 863cfc7
Showing 1 changed file with 15 additions and 14 deletions.
diff --git a/client/classic/upe/payment-processing.js b/client/classic/upe/payment-processing.js
@@ -243,20 +243,21 @@ function setSelectedUPEPaymentType( paymentType ) {
 
 // Show or hide save payment information checkbox
 function showNewPaymentMethodCheckbox( show = true ) {
-	if ( show ) {
-		document.querySelector(
-			'.woocommerce-SavedPaymentMethods-saveNew'
-		).style.visibility = 'visible';
-	} else {
-		document.querySelector(
-			'.woocommerce-SavedPaymentMethods-saveNew'
-		).style.visibility = 'hidden';
-		document
-			.querySelector( 'input#wc-stripe-new-payment-method' )
-			.setAttribute( 'checked', false );
-		document
-			.querySelector( 'input#wc-stripe-new-payment-method' )
-			.dispatchEvent( new Event( 'change' ) );
+	const saveCardElement = document.querySelector(
+		'.woocommerce-SavedPaymentMethods-saveNew'
+	);
+
+	if ( saveCardElement ) {
+		saveCardElement.style.visibility = show ? 'visible' : 'hidden';
+	}
+
+	const stripeSaveCardCheckbox = document.querySelector(
+		'input#wc-stripe-new-payment-method'
+	);
+
+	if ( ! show && stripeSaveCardCheckbox ) {
+		stripeSaveCardCheckbox.setAttribute( 'checked', false );
+		stripeSaveCardCheckbox.dispatchEvent( new Event( 'change' ) );
 	}
 }