From e0e9829051726f9ba4272e9335da32cb2eecedd6 Mon Sep 17 00:00:00 2001
From: Payton Swick <payton@automattic.com>
Date: Fri, 18 Dec 2020 13:27:42 -0500
Subject: [PATCH] Checkout: Update G Suite nudge to replace CartData with
 withShoppingCart (#48438)

* Switch GSuiteNudge to use withShoppingCart

* Wrap GSuiteNudge in CalypsoShoppingCartProvider

* Replace handleCheckoutCompleteRedirect in GSuiteNudge

* Add isMounted to GSuiteNudge

* Fix CalypsoShoppingCartProvider import

* Fix isMounted check

* Remove removePlanFromCart as the cart will be empty when you arrive
---
 client/my-sites/checkout/controller.jsx       |  5 +-
 .../my-sites/checkout/gsuite-nudge/index.jsx  | 57 ++++++++++++-------
 2 files changed, 40 insertions(+), 22 deletions(-)

diff --git a/client/my-sites/checkout/controller.jsx b/client/my-sites/checkout/controller.jsx
index 999d866461e1f..ed82cceb896ac 100644
--- a/client/my-sites/checkout/controller.jsx
+++ b/client/my-sites/checkout/controller.jsx
@@ -35,6 +35,7 @@ import UpsellNudge, {
 	CONCIERGE_SUPPORT_SESSION,
 	CONCIERGE_QUICKSTART_SESSION,
 } from './upsell-nudge';
+import CalypsoShoppingCartProvider from 'calypso/my-sites/checkout/calypso-shopping-cart-provider';
 
 export function checkout( context, next ) {
 	const { feature, plan, domainOrProduct, purchaseId } = context.params;
@@ -176,13 +177,13 @@ export function gsuiteNudge( context, next ) {
 	}
 
 	context.primary = (
-		<CheckoutContainer purchaseId={ Number( receiptId ) }>
+		<CalypsoShoppingCartProvider>
 			<GSuiteNudge
 				domain={ domain }
 				receiptId={ Number( receiptId ) }
 				selectedSiteId={ selectedSite.ID }
 			/>
-		</CheckoutContainer>
+		</CalypsoShoppingCartProvider>
 	);
 
 	next();
diff --git a/client/my-sites/checkout/gsuite-nudge/index.jsx b/client/my-sites/checkout/gsuite-nudge/index.jsx
index 9b16320b04c3c..3ccdd05a535f5 100644
--- a/client/my-sites/checkout/gsuite-nudge/index.jsx
+++ b/client/my-sites/checkout/gsuite-nudge/index.jsx
@@ -7,6 +7,7 @@ import { connect } from 'react-redux';
 import { localize } from 'i18n-calypso';
 import { get, some, compact } from 'lodash';
 import page from 'page';
+import { withShoppingCart } from '@automattic/shopping-cart';
 
 /**
  * Internal dependencies
@@ -23,10 +24,11 @@ import QuerySites from 'calypso/components/data/query-sites';
 import { getSiteSlug, getSiteTitle } from 'calypso/state/sites/selectors';
 import { getReceiptById } from 'calypso/state/receipts/selectors';
 import isEligibleForDotcomChecklist from 'calypso/state/selectors/is-eligible-for-dotcom-checklist';
-import { addItems, removeItem } from 'calypso/lib/cart/actions';
-import { getAllCartItems } from 'calypso/lib/cart-values/cart-items';
 import { isDotComPlan } from 'calypso/lib/products-values';
 import PageViewTracker from 'calypso/lib/analytics/page-view-tracker';
+import { fillInSingleCartItemAttributes } from 'calypso/lib/cart-values';
+import { getProductsList } from 'calypso/state/products-list/selectors/get-products-list';
+import getThankYouPageUrl from 'calypso/my-sites/checkout/composite-checkout/hooks/use-get-thank-you-url/get-thank-you-page-url';
 
 /**
  * Style dependencies
@@ -40,30 +42,44 @@ export class GSuiteNudge extends React.Component {
 		selectedSiteId: PropTypes.number.isRequired,
 	};
 
+	isMounted = false;
+
+	componentDidMount() {
+		this.isMounted = true;
+	}
+
+	componentWillUnmount() {
+		this.isMounted = false;
+	}
+
 	handleSkipClick = () => {
-		this.props.handleCheckoutCompleteRedirect();
+		const getThankYouPageUrlArguments = {
+			siteSlug: this.props.siteSlug,
+			receiptId: this.props.receiptId,
+			cart: this.props.cart,
+		};
+		const url = getThankYouPageUrl( getThankYouPageUrlArguments );
+		page.redirect( url );
 	};
 
 	handleAddEmailClick = ( cartItems ) => {
-		const { siteSlug, receiptId } = this.props;
-		this.removePlanFromCart();
+		const { siteSlug, receiptId, productsList } = this.props;
 
-		addItems(
-			// add `receipt_for_domain` to cartItem extras
-			cartItems.map( ( item ) => ( {
-				...item,
-				extra: { ...item.extra, receipt_for_domain: receiptId },
-			} ) )
-		);
-
-		page( `/checkout/${ siteSlug }` );
+		this.props.shoppingCartManager
+			.addProductsToCart(
+				// add `receipt_for_domain` to cartItem extras
+				cartItems
+					.map( ( item ) => ( {
+						...item,
+						extra: { ...item.extra, receipt_for_domain: receiptId },
+					} ) )
+					.map( ( item ) => fillInSingleCartItemAttributes( item, productsList ) )
+			)
+			.then( () => {
+				this.isMounted && page( `/checkout/${ siteSlug }` );
+			} );
 	};
 
-	removePlanFromCart() {
-		const items = getAllCartItems( this.props.cart );
-		items.filter( isDotComPlan ).forEach( ( item ) => removeItem( item, false ) );
-	}
-
 	render() {
 		const { domain, receiptId, selectedSiteId, siteSlug, siteTitle, translate } = this.props;
 
@@ -109,5 +125,6 @@ export default connect( ( state, props ) => {
 		siteSlug: getSiteSlug( state, props.selectedSiteId ),
 		siteTitle: getSiteTitle( state, props.selectedSiteId ),
 		isEligibleForChecklist,
+		productsList: getProductsList( state ),
 	};
-} )( localize( GSuiteNudge ) );
+} )( withShoppingCart( localize( GSuiteNudge ) ) );