MOL-664 Fix missing payment method title when checkout done with blocks

mollie · Feb 11, 2022 · 5f4a8e2 · 5f4a8e2
1 parent 107a640
commit 5f4a8e2
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/Gateway/GatewayModule.php b/src/Gateway/GatewayModule.php
@@ -211,15 +211,19 @@ static function () {
         $this->molliePayPalButtonHandling($paypalGateway, $notice, $logger, $pluginUrl);
         $checkoutBlockHandler = new CheckoutBlockService($container->get('settings.data_helper'));
         $checkoutBlockHandler->bootstrapAjaxRequest();
-        add_action( 'woocommerce_blocks_checkout_update_order_meta', function($order) use($gatewayInstances){
-            $orderPaymentMethod = $order->get_payment_method();
-            $title = $orderPaymentMethod? $gatewayInstances[$orderPaymentMethod]->title: '';
+        add_action( 'woocommerce_rest_checkout_process_payment_with_context', function($paymentContext){
+            if(strpos($paymentContext->payment_method, 'mollie_wc_gateway_') === false){
+                return;
+            }
+            $title = isset($paymentContext->payment_data['payment_method_title'])?$paymentContext->payment_data['payment_method_title']:false;
             if(!$title){
-                return $order;
+                return ;
             }
-            $order->update_meta_data('_payment_method_title', $title);
-            return $order;
+            $order = $paymentContext->order;
+            $order->set_payment_method_title( $title );
+            $order->save();
         } );
+
         return true;
     }