Add remediation tool to fix incorrect analytics data from canceled authorizations

includes/class-wc-payments.php

@@ -319,6 +319,13 @@ class WC_Payments {
 	 */
 	private static $payment_method_service;
 
+	/**
+	 * Instance of WC_Payments_Remediate_Canceled_Auth_Fees, created in init function
+	 *
+	 * @var WC_Payments_Remediate_Canceled_Auth_Fees
+	 */
+	private static $fee_remediation;
+
 	/**
 	 * Entry point to the initialization logic.
 	 */
@@ -352,6 +359,7 @@ public static function init() {
 		add_action( 'admin_init', [ __CLASS__, 'add_woo_admin_notes' ] );
 		add_action( 'admin_init', [ __CLASS__, 'remove_deprecated_notes' ] );
 		add_action( 'init', [ __CLASS__, 'install_actions' ] );
+		add_action( 'upgrader_process_complete', [ __CLASS__, 'on_plugin_update' ], 10, 2 );
 
 		add_action( 'woocommerce_blocks_payment_method_type_registration', [ __CLASS__, 'register_checkout_gateway' ] );
 
@@ -492,6 +500,7 @@ public static function init() {
 		include_once __DIR__ . '/class-wc-payments-order-service.php';
 		include_once __DIR__ . '/class-wc-payments-order-success-page.php';
 		include_once __DIR__ . '/class-wc-payments-file-service.php';
+		include_once __DIR__ . '/migrations/class-wc-payments-remediate-canceled-auth-fees.php';
 		include_once __DIR__ . '/class-wc-payments-webhook-processing-service.php';
 		include_once __DIR__ . '/class-wc-payments-webhook-reliability-service.php';
 		include_once __DIR__ . '/fraud-prevention/class-fraud-prevention-service.php';
@@ -552,6 +561,7 @@ public static function init() {
 		self::$incentives_service                   = new WC_Payments_Incentives_Service( self::$database_cache );
 		self::$duplicate_payment_prevention_service = new Duplicate_Payment_Prevention_Service();
 		self::$duplicates_detection_service         = new Duplicates_Detection_Service();
+		self::$fee_remediation                      = new WC_Payments_Remediate_Canceled_Auth_Fees();
 
 		( new WooPay_Scheduler( self::$api_client ) )->init();
 
@@ -564,6 +574,7 @@ public static function init() {
 		self::$compatibility_service->init_hooks();
 		self::$customer_service->init_hooks();
 		self::$token_service->init_hooks();
+		self::$fee_remediation->init();
 
 		/**
 		 * FLAG: PAYMENT_METHODS_LIST
@@ -1509,6 +1520,43 @@ public static function update_plugin_version() {
 		update_option( 'woocommerce_woocommerce_payments_version', WCPAY_VERSION_NUMBER );
 	}
 
+	/**
+	 * Handle plugin updates.
+	 *
+	 * @param WP_Upgrader $upgrader WP_Upgrader instance.
+	 * @param array       $options  Array of update data.
+	 * @return void
+	 */
+	public static function on_plugin_update( $upgrader, $options ) {
+		if ( 'update' !== $options['action'] || 'plugin' !== $options['type'] ) {
+			return;
+		}
+
+		// Check if WooPayments was updated.
+		$plugins = isset( $options['plugins'] ) ? $options['plugins'] : [];
+		if ( ! in_array( plugin_basename( WCPAY_PLUGIN_FILE ), $plugins, true ) ) {
+			return;
+		}
+
+		// Ensure get_plugin_data() is available.
+		if ( ! function_exists( 'get_plugin_data' ) ) {
+			require_once ABSPATH . 'wp-admin/includes/plugin.php';
+		}
+
+		// Get new version.
+		$plugin_data = get_plugin_data( WCPAY_PLUGIN_FILE );
+		$new_version = $plugin_data['Version'];
+
+		// Initialize fee_remediation if not already set.
+		if ( ! isset( self::$fee_remediation ) ) {
+			include_once __DIR__ . '/migrations/class-wc-payments-remediate-canceled-auth-fees.php';
+			self::$fee_remediation = new WC_Payments_Remediate_Canceled_Auth_Fees();
+		}
+
+		// Trigger version gate check.
+		self::$fee_remediation->maybe_schedule_remediation( $new_version );
+	}
+
 	/**
 	 * Sets the plugin activation timestamp.
 	 *