Introduced wc_get_pay_buttons() function

Author: claudiosanches

includes/wc-template-functions.php

@@ -3664,3 +3664,29 @@ function woocommerce_product_reviews_tab() {
 		wc_deprecated_function( 'woocommerce_product_reviews_tab', '2.4' );
 	}
 }
+
+/**
+ * Display pay buttons HTML.
+ *
+ * @since 3.9.0
+ */
+function wc_get_pay_buttons() {
+	$supported_gateways = array();
+	$available_gateways = WC()->payment_gateways()->get_available_payment_gateways();
+
+	foreach ( $available_gateways as $gateway ) {
+		if ( $gateway->supports( 'pay_button' ) ) {
+			$supported_gateways[] = $gateway->get_pay_button_id();
+		}
+	}
+
+	if ( ! $supported_gateways ) {
+		return;
+	}
+
+	echo '<div class="woocommerce-pay-buttons">';
+	foreach ( $supported_gateways as $pay_button_id ) {
+		echo sprintf( '<div class="woocommerce-pay-button__%1$s %1$s" id="%1$s"></div>', esc_attr( $pay_button_id ) );
+	}
+	echo '</div>';
+}

includes/wc-template-hooks.php

@@ -207,13 +207,17 @@
  * @see woocommerce_checkout_coupon_form()
  * @see woocommerce_order_review()
  * @see woocommerce_checkout_payment()
+ * @see wc_checkout_privacy_policy_text()
+ * @see wc_terms_and_conditions_page_content()
+ * @see wc_get_pay_buttons()
  */
 add_action( 'woocommerce_before_checkout_form', 'woocommerce_checkout_login_form', 10 );
 add_action( 'woocommerce_before_checkout_form', 'woocommerce_checkout_coupon_form', 10 );
 add_action( 'woocommerce_checkout_order_review', 'woocommerce_order_review', 10 );
 add_action( 'woocommerce_checkout_order_review', 'woocommerce_checkout_payment', 20 );
 add_action( 'woocommerce_checkout_terms_and_conditions', 'wc_checkout_privacy_policy_text', 20 );
 add_action( 'woocommerce_checkout_terms_and_conditions', 'wc_terms_and_conditions_page_content', 30 );
+add_action( 'woocommerce_checkout_before_customer_details', 'wc_get_pay_buttons', 30 );
 
 /**
  * Cart widget
@@ -227,10 +231,13 @@
  *
  * @see woocommerce_cross_sell_display()
  * @see woocommerce_cart_totals()
+ * @see wc_get_pay_buttons()
  * @see woocommerce_button_proceed_to_checkout()
+ * @see wc_empty_cart_message()
  */
 add_action( 'woocommerce_cart_collaterals', 'woocommerce_cross_sell_display' );
 add_action( 'woocommerce_cart_collaterals', 'woocommerce_cart_totals', 10 );
+add_action( 'woocommerce_proceed_to_checkout', 'wc_get_pay_buttons', 10 );
 add_action( 'woocommerce_proceed_to_checkout', 'woocommerce_button_proceed_to_checkout', 20 );
 add_action( 'woocommerce_cart_is_empty', 'wc_empty_cart_message', 10 );
 

tests/unit-tests/templates/functions.php

@@ -142,4 +142,40 @@ public function test_wc_query_string_form_fields() {
 		$expected_html = '<input type="hidden" name="test_something" value="something else" />';
 		$this->assertEquals( $expected_html, $actual_html );
 	}
+
+	/**
+	 * Test test_wc_get_pay_buttons().
+	 */
+	public function test_wc_get_pay_buttons() {
+		// Test default.
+		ob_start();
+		wc_get_pay_buttons();
+		$actual_html = ob_get_clean();
+
+		$this->assertEquals( '', $actual_html );
+
+		// Include a payment gateway that supports "pay button".
+		add_filter(
+			'woocommerce_payment_gateways',
+			function( $gateways ) {
+				$gateways[] = 'WC_Mock_Payment_Gateway';
+
+				return $gateways;
+			}
+		);
+		WC()->payment_gateways()->init();
+
+		// Test pay buttons HTML.
+		ob_start();
+		wc_get_pay_buttons();
+		$actual_html = ob_get_clean();
+
+		$gateway       = new WC_Mock_Payment_Gateway();
+		$expected_html = sprintf(
+			'<div class="woocommerce-pay-buttons"><div class="woocommerce-pay-button__%1$s %1$s" id="%1$s"></div></div>',
+			$gateway->get_pay_button_id()
+		);
+
+		$this->assertEquals( $expected_html, $actual_html );
+	}
 }