From 69f6cc2e7359154331f4d7f9f2fed61ced61bcf0 Mon Sep 17 00:00:00 2001
From: Philipp Stracker <p.stracker@syde.com>
Date: Fri, 20 Dec 2024 16:09:54 +0100
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20First=20steps=20to=20implement=20me?=
 =?UTF-8?q?rchant=20authentication?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Onboarding/Components/ConnectionButton.js | 23 +++++++--
 .../js/hooks/useHandleConnections.js          | 51 +++++++++++++++++--
 2 files changed, 66 insertions(+), 8 deletions(-)

diff --git a/modules/ppcp-settings/resources/js/Components/Screens/Onboarding/Components/ConnectionButton.js b/modules/ppcp-settings/resources/js/Components/Screens/Onboarding/Components/ConnectionButton.js
index e100881b8..7cbc504e0 100644
--- a/modules/ppcp-settings/resources/js/Components/Screens/Onboarding/Components/ConnectionButton.js
+++ b/modules/ppcp-settings/resources/js/Components/Screens/Onboarding/Components/ConnectionButton.js
@@ -34,7 +34,6 @@ const ButtonOrPlaceholder = ( {
 		buttonProps.href = href;
 		buttonProps.target = 'PPFrame';
 		buttonProps[ 'data-paypal-button' ] = 'true';
-		buttonProps[ 'data-paypal-onboard-complete' ] = 'onOnboardComplete';
 		buttonProps[ 'data-paypal-onboard-button' ] = 'true';
 	}
 
@@ -48,18 +47,34 @@ const ConnectionButton = ( {
 	showIcon = true,
 	className = '',
 } ) => {
-	const { onboardingUrl, scriptLoaded } =
-		useHandleOnboardingButton( isSandbox );
+	const {
+		onboardingUrl,
+		scriptLoaded,
+		setCompleteHandler,
+		removeCompleteHandler,
+	} = useHandleOnboardingButton( isSandbox );
 	const buttonClassName = classNames( 'ppcp-r-connection-button', className, {
 		'sandbox-mode': isSandbox,
 		'live-mode': ! isSandbox,
 	} );
+	const environment = isSandbox ? 'sandbox' : 'production';
 
 	useEffect( () => {
 		if ( scriptLoaded && onboardingUrl ) {
 			window.PAYPAL.apps.Signup.render();
+			setCompleteHandler( environment );
 		}
-	}, [ scriptLoaded, onboardingUrl ] );
+
+		return () => {
+			removeCompleteHandler();
+		};
+	}, [
+		scriptLoaded,
+		onboardingUrl,
+		environment,
+		setCompleteHandler,
+		removeCompleteHandler,
+	] );
 
 	return (
 		<BusyStateWrapper isBusy={ ! onboardingUrl }>
diff --git a/modules/ppcp-settings/resources/js/hooks/useHandleConnections.js b/modules/ppcp-settings/resources/js/hooks/useHandleConnections.js
index 0bd14cc17..9d860e1c8 100644
--- a/modules/ppcp-settings/resources/js/hooks/useHandleConnections.js
+++ b/modules/ppcp-settings/resources/js/hooks/useHandleConnections.js
@@ -1,10 +1,13 @@
 import { __ } from '@wordpress/i18n';
 import { useDispatch } from '@wordpress/data';
-import { useState, useEffect } from '@wordpress/element';
+import { useState, useEffect, useCallback, useRef } from '@wordpress/element';
 import { store as noticesStore } from '@wordpress/notices';
 
 import { CommonHooks, OnboardingHooks } from '../data';
 
+const PAYPAL_PARTNER_SDK_URL =
+	'https://www.paypal.com/webapps/merchantboarding/js/lib/lightbox/partner.js';
+
 const MESSAGES = {
 	CONNECTED: __( 'Connected to PayPal', 'woocommerce-paypal-payments' ),
 	POPUP_BLOCKED: __(
@@ -41,6 +44,7 @@ export const useHandleOnboardingButton = ( isSandbox ) => {
 	const products = OnboardingHooks.useDetermineProducts();
 	const [ onboardingUrl, setOnboardingUrl ] = useState( '' );
 	const [ scriptLoaded, setScriptLoaded ] = useState( false );
+	const timerRef = useRef( null );
 
 	useEffect( () => {
 		const fetchOnboardingUrl = async () => {
@@ -73,8 +77,7 @@ export const useHandleOnboardingButton = ( isSandbox ) => {
 
 		const script = document.createElement( 'script' );
 		script.id = 'partner-js';
-		script.src =
-			'https://www.paypal.com/webapps/merchantboarding/js/lib/lightbox/partner.js';
+		script.src = PAYPAL_PARTNER_SDK_URL;
 		script.onload = () => {
 			setScriptLoaded( true );
 		};
@@ -105,7 +108,47 @@ export const useHandleOnboardingButton = ( isSandbox ) => {
 		};
 	}, [ onboardingUrl ] );
 
-	return { onboardingUrl, scriptLoaded };
+	const setCompleteHandler = useCallback( ( environment ) => {
+		const onComplete = ( authCode, shareId ) => {
+			// TODO -- finish this!
+			console.log(
+				`${ environment }-boarding complete - AUTH: `,
+				authCode
+			);
+			console.log(
+				`${ environment }-boarding complete - SHARE:`,
+				shareId
+			);
+		};
+
+		const addHandler = () => {
+			const MiniBrowser = window.PAYPAL?.apps?.Signup?.MiniBrowser;
+			if ( ! MiniBrowser || MiniBrowser.onOnboardComplete ) {
+				return;
+			}
+
+			MiniBrowser.onOnboardComplete = onComplete;
+		};
+
+		// Ensure the onComplete handler is not removed by a PayPal init script.
+		timerRef.current = setInterval( addHandler, 250 );
+	}, [] );
+
+	const removeCompleteHandler = useCallback( () => {
+		if ( timerRef.current ) {
+			clearInterval( timerRef.current );
+			timerRef.current = null;
+		}
+
+		delete window.PAYPAL?.apps?.Signup?.MiniBrowser?.onOnboardComplete;
+	}, [] );
+
+	return {
+		onboardingUrl,
+		scriptLoaded,
+		setCompleteHandler,
+		removeCompleteHandler,
+	};
 };
 
 const useConnectionBase = () => {