From e2e47fadf1594f095e34418e87165ae665c8223f Mon Sep 17 00:00:00 2001
From: Remco Tolsma <869674+remcotolsma@users.noreply.github.com>
Date: Fri, 3 May 2024 14:51:48 +0200
Subject: [PATCH] Fix mix customer ID and Mollie ID, see
 5c739d2fcdc448d27e8f9246d713951a147a638c.

---
 src/CustomerDataStore.php | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/CustomerDataStore.php b/src/CustomerDataStore.php
index de5a419..b48a10b 100644
--- a/src/CustomerDataStore.php
+++ b/src/CustomerDataStore.php
@@ -207,10 +207,21 @@ public function save_customer( Customer $customer, $data = [], $format = [] ) {
 	public function connect_mollie_customer_to_wp_user( $customer, \WP_User $user ) {
 		global $wpdb;
 
+		$customer_id = $wpdb->get_var(
+			$wpdb->prepare(
+				"SELECT id FROM $wpdb->pronamic_pay_mollie_customers WHERE mollie_id = %s;",
+				$customer->get_id()
+			)
+		);
+
+		if ( null === $customer_id ) {
+			return;
+		}
+
 		$row = $wpdb->get_row(
 			$wpdb->prepare(
 				"SELECT * FROM $wpdb->pronamic_pay_mollie_customer_users WHERE customer_id = %d AND user_id = %d;",
-				$customer->get_id(),
+				$customer_id,
 				$user->ID
 			)
 		);
@@ -220,7 +231,7 @@ public function connect_mollie_customer_to_wp_user( $customer, \WP_User $user )
 		}
 
 		$data = [
-			'customer_id' => $customer->get_id(),
+			'customer_id' => $customer_id,
 			'user_id'     => $user->ID,
 		];