Ensure order attribution inputs are not removed when shortcode cart i…

…s updated (#9038)
Automattic · Jul 9, 2024 · 1c18a0d · 1c18a0d
1 parent 6d4e054
commit 1c18a0d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/changelog/9001-tokenized-prb-order-origin b/changelog/9001-tokenized-prb-order-origin
@@ -0,0 +1,5 @@
+Significance: patch
+Type: fix
+Comment: Bug fix for PRB feature behind a flag.
+
+
diff --git a/includes/express-checkout/class-wc-payments-express-checkout-button-display-handler.php b/includes/express-checkout/class-wc-payments-express-checkout-button-display-handler.php
@@ -150,14 +150,29 @@ public function display_express_checkout_buttons() {
 			} else {
 				$this->payment_request_button_handler->display_payment_request_button_html();
 			}
+
+			if ( is_cart() ) {
+				add_action( 'woocommerce_after_cart', [ $this, 'add_order_attribution_inputs' ], 1 );
+			} else {
+				$this->add_order_attribution_inputs();
+			}
+
 			?>
-			<wc-order-attribution-inputs id="wcpay-express-checkout__order-attribution-inputs"></wc-order-attribution-inputs>
 			</div >
 			<?php
 			$this->display_express_checkout_separator_if_necessary( $separator_starts_hidden );
 		}
 	}
 
+	/**
+	 * Add order attribution inputs to the page.
+	 *
+	 * @return void
+	 */
+	public function add_order_attribution_inputs() {
+		echo '<wc-order-attribution-inputs id="wcpay-express-checkout__order-attribution-inputs"></wc-order-attribution-inputs>';
+	}
+
 	/**
 	 * Check if the pay-for-order flow is supported.
 	 *