From 0ddf8927732997ff1a54cb1b7adc959013879d85 Mon Sep 17 00:00:00 2001
From: Christos <chriskmnds@gmail.com>
Date: Wed, 23 Oct 2024 02:06:55 +0300
Subject: [PATCH] Plans (state): `useCurrentPlan` in plans main component
 (#95355)

* Plan (state): useCurrentPlan in plans main component

---------

Co-authored-by: Jeremy Yip <jeremy.yip@automattic.com>
---
 client/my-sites/plans/main.jsx                | 20 ++++++++++---------
 .../plans/woo-express-plans-page/index.tsx    |  9 ++++-----
 2 files changed, 15 insertions(+), 14 deletions(-)

diff --git a/client/my-sites/plans/main.jsx b/client/my-sites/plans/main.jsx
index acf275455ecb5f..879c539355c181 100644
--- a/client/my-sites/plans/main.jsx
+++ b/client/my-sites/plans/main.jsx
@@ -14,7 +14,7 @@ import {
 	PLAN_WOOEXPRESS_SMALL_MONTHLY,
 } from '@automattic/calypso-products';
 import page from '@automattic/calypso-router';
-import { WpcomPlansUI } from '@automattic/data-stores';
+import { WpcomPlansUI, Plans } from '@automattic/data-stores';
 import { englishLocales } from '@automattic/i18n-utils';
 import { withShoppingCart } from '@automattic/shopping-cart';
 import { useDispatch } from '@wordpress/data';
@@ -39,6 +39,7 @@ import { PerformanceTrackerStop } from 'calypso/lib/performance-tracking';
 import PlansNavigation from 'calypso/my-sites/plans/navigation';
 import P2PlansMain from 'calypso/my-sites/plans/p2-plans-main';
 import PlansFeaturesMain from 'calypso/my-sites/plans-features-main';
+import { useSelector } from 'calypso/state';
 import { getByPurchaseId } from 'calypso/state/purchases/selectors';
 import { canCurrentUser } from 'calypso/state/selectors/can-current-user';
 import getCurrentLocaleSlug from 'calypso/state/selectors/get-current-locale-slug';
@@ -48,7 +49,6 @@ import isEligibleForWpComMonthlyPlan from 'calypso/state/selectors/is-eligible-f
 import isSiteWPForTeams from 'calypso/state/selectors/is-site-wpforteams';
 import siteHasFeature from 'calypso/state/selectors/site-has-feature';
 import { fetchSitePlans } from 'calypso/state/sites/plans/actions';
-import { getCurrentPlan } from 'calypso/state/sites/plans/selectors';
 import { isJetpackSite } from 'calypso/state/sites/selectors';
 import { getSelectedSite, getSelectedSiteId } from 'calypso/state/ui/selectors';
 import CalypsoShoppingCartProvider from '../checkout/calypso-shopping-cart-provider';
@@ -148,7 +148,7 @@ function DescriptionMessage( { isDomainUpsell, isFreePlan, yourDomainName, siteS
 	);
 }
 
-class Plans extends Component {
+class PlansComponent extends Component {
 	static propTypes = {
 		context: PropTypes.object.isRequired,
 		redirectToAddDomainFlow: PropTypes.bool,
@@ -511,9 +511,8 @@ class Plans extends Component {
 }
 
 const ConnectedPlans = connect(
-	( state ) => {
-		const selectedSiteId = getSelectedSiteId( state );
-		const currentPlan = getCurrentPlan( state, selectedSiteId );
+	( state, props ) => {
+		const { currentPlan, selectedSiteId } = props;
 		const currentPlanIntervalType = getIntervalTypeForTerm(
 			getPlan( currentPlan?.productSlug )?.term
 		);
@@ -521,7 +520,7 @@ const ConnectedPlans = connect(
 		return {
 			currentPlan,
 			currentPlanIntervalType,
-			purchase: currentPlan ? getByPurchaseId( state, currentPlan.id ) : null,
+			purchase: currentPlan ? getByPurchaseId( state, currentPlan.purchaseId ) : null,
 			selectedSite: getSelectedSite( state ),
 			canAccessPlans: canCurrentUser( state, getSelectedSiteId( state ), 'manage_options' ),
 			isWPForTeamsSite: isSiteWPForTeams( state, selectedSiteId ),
@@ -543,12 +542,15 @@ const ConnectedPlans = connect(
 	( dispatch ) => ( {
 		fetchSitePlans: ( siteId ) => dispatch( fetchSitePlans( siteId ) ),
 	} )
-)( withCartKey( withShoppingCart( localize( Plans ) ) ) );
+)( withCartKey( withShoppingCart( localize( PlansComponent ) ) ) );
 
 export default function PlansWrapper( props ) {
+	const selectedSiteId = useSelector( getSelectedSiteId );
+	const currentPlan = Plans.useCurrentPlan( { siteId: selectedSiteId } );
+
 	return (
 		<CalypsoShoppingCartProvider>
-			<ConnectedPlans { ...props } />
+			<ConnectedPlans { ...props } currentPlan={ currentPlan } selectedSiteId={ selectedSiteId } />
 		</CalypsoShoppingCartProvider>
 	);
 }
diff --git a/client/my-sites/plans/woo-express-plans-page/index.tsx b/client/my-sites/plans/woo-express-plans-page/index.tsx
index fb812498caf630..50ae405772769d 100644
--- a/client/my-sites/plans/woo-express-plans-page/index.tsx
+++ b/client/my-sites/plans/woo-express-plans-page/index.tsx
@@ -11,20 +11,19 @@ import {
 } from '@automattic/calypso-products';
 import page from '@automattic/calypso-router';
 import { Button, Card } from '@automattic/components';
-import { Plans, type SiteDetails } from '@automattic/data-stores';
+import { Plans, type SiteDetails, SitePlan } from '@automattic/data-stores';
 import { formatCurrency } from '@automattic/format-currency';
 import clsx from 'clsx';
 import { useTranslate } from 'i18n-calypso';
 import { useCallback } from 'react';
 import BodySectionCssClass from 'calypso/layout/body-section-css-class';
-import { SitePlanData } from 'calypso/my-sites/checkout/src/hooks/product-variants';
 import useCheckPlanAvailabilityForPurchase from 'calypso/my-sites/plans-features-main/hooks/use-check-plan-availability-for-purchase';
 import { WooExpressPlans } from '../ecommerce-trial/wooexpress-plans';
 
 import './style.scss';
 
 interface WooExpressPlansPageProps {
-	currentPlan: SitePlanData;
+	currentPlan: SitePlan;
 	interval?: 'monthly' | 'yearly';
 	selectedSite: SiteDetails;
 	showIntervalToggle: boolean;
@@ -74,8 +73,8 @@ const WooExpressPlansPage = ( {
 	const planInterval = isAnnualSubscription ? 'yearly' : 'monthly';
 
 	const goToSubscriptionPage = () => {
-		if ( selectedSite?.slug && currentPlan?.id ) {
-			page( `/purchases/subscriptions/${ selectedSite.slug }/${ currentPlan.id }` );
+		if ( selectedSite?.slug && currentPlan?.purchaseId ) {
+			page( `/purchases/subscriptions/${ selectedSite.slug }/${ currentPlan.purchaseId }` );
 		}
 	};