From b74a9f92f2135a8a6d51a2e9ec3648b3d5db1580 Mon Sep 17 00:00:00 2001
From: Mike Jolley <mike.jolley@me.com>
Date: Thu, 6 Oct 2022 13:00:30 +0100
Subject: [PATCH] Introduce Toggle between Shipping Options and Pickup
 Locations (https://github.com/woocommerce/woocommerce-blocks/pull/7308)

* prefersCollection controls shipping visibility

* Toggle shipping visibility based on prefersCollection

* Remove log

* Toggle in admin
---
 .../context/hooks/use-checkout-address.ts     | 14 ++++++++---
 .../checkout-collection-method/edit.tsx       | 25 ++++++++++++++++---
 .../checkout-shipping-address-block/edit.tsx  |  3 +++
 .../checkout-shipping-methods-block/edit.tsx  |  9 ++++++-
 4 files changed, 42 insertions(+), 9 deletions(-)

diff --git a/plugins/woocommerce-blocks/assets/js/base/context/hooks/use-checkout-address.ts b/plugins/woocommerce-blocks/assets/js/base/context/hooks/use-checkout-address.ts
index 6acfa8a7150..5fb7e2899ab 100644
--- a/plugins/woocommerce-blocks/assets/js/base/context/hooks/use-checkout-address.ts
+++ b/plugins/woocommerce-blocks/assets/js/base/context/hooks/use-checkout-address.ts
@@ -39,8 +39,12 @@ interface CheckoutAddress {
  */
 export const useCheckoutAddress = (): CheckoutAddress => {
 	const { needsShipping } = useShippingData();
-	const useShippingAsBilling = useSelect( ( select ) =>
-		select( CHECKOUT_STORE_KEY ).getUseShippingAsBilling()
+	const { useShippingAsBilling, prefersCollection } = useSelect(
+		( select ) => ( {
+			useShippingAsBilling:
+				select( CHECKOUT_STORE_KEY ).getUseShippingAsBilling(),
+			prefersCollection: select( CHECKOUT_STORE_KEY ).prefersCollection(),
+		} )
 	);
 	const { __internalSetUseShippingAsBilling } =
 		useDispatch( CHECKOUT_STORE_KEY );
@@ -89,8 +93,10 @@ export const useCheckoutAddress = (): CheckoutAddress => {
 		defaultAddressFields,
 		useShippingAsBilling,
 		setUseShippingAsBilling: __internalSetUseShippingAsBilling,
-		showShippingFields: ! forcedBillingAddress && needsShipping,
-		showBillingFields: ! needsShipping || ! useShippingAsBilling,
+		showShippingFields:
+			! forcedBillingAddress && needsShipping && ! prefersCollection,
+		showBillingFields:
+			! needsShipping || ! useShippingAsBilling || prefersCollection,
 		forcedBillingAddress,
 	};
 };
diff --git a/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-collection-method/edit.tsx b/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-collection-method/edit.tsx
index 07c7f17128d..b2019fa4719 100644
--- a/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-collection-method/edit.tsx
+++ b/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-collection-method/edit.tsx
@@ -20,6 +20,8 @@ import {
 import { useShippingData } from '@woocommerce/base-context/hooks';
 import { innerBlockAreas } from '@woocommerce/blocks-checkout';
 import type { CartShippingPackageShippingRate } from '@woocommerce/type-defs/cart';
+import { useDispatch, useSelect } from '@wordpress/data';
+import { CHECKOUT_STORE_KEY } from '@woocommerce/block-data';
 
 /**
  * Internal dependencies
@@ -155,7 +157,13 @@ export const Edit = ( {
 	};
 	setAttributes: ( attributes: Record< string, unknown > ) => void;
 } ): JSX.Element => {
-	const [ currentView, changeView ] = useState( 'shipping' );
+	const { setPrefersCollection } = useDispatch( CHECKOUT_STORE_KEY );
+	const { prefersCollection } = useSelect( ( select ) => {
+		const checkoutStore = select( CHECKOUT_STORE_KEY );
+		return {
+			prefersCollection: checkoutStore.prefersCollection(),
+		};
+	} );
 	const { showPrice, showIcon, className, localPickupText, shippingText } =
 		attributes;
 	const { shippingRates } = useShippingData();
@@ -165,6 +173,15 @@ export const Edit = ( {
 	const shippingStartingPrice = getShippingStartingPrice(
 		shippingRates[ 0 ]?.shipping_rates
 	);
+
+	const changeView = ( method: string ) => {
+		if ( method === 'pickup' ) {
+			setPrefersCollection( true );
+		} else {
+			setPrefersCollection( false );
+		}
+	};
+
 	return (
 		<FormStepBlock
 			attributes={ attributes }
@@ -215,10 +232,10 @@ export const Edit = ( {
 				className="wc-block-checkout__collection-method-container"
 				label="options"
 				onChange={ changeView }
-				checked={ currentView }
+				checked={ prefersCollection ? 'pickup' : 'shipping' }
 			>
 				<ShippingSelector
-					checked={ currentView }
+					checked={ prefersCollection ? 'pickup' : 'shipping' }
 					rate={ shippingStartingPrice }
 					showPrice={ showPrice }
 					showIcon={ showIcon }
@@ -226,7 +243,7 @@ export const Edit = ( {
 					toggleText={ shippingText }
 				/>
 				<LocalPickupSelector
-					checked={ currentView }
+					checked={ prefersCollection ? 'pickup' : 'shipping' }
 					rate={ localPickupStartingPrice }
 					showPrice={ showPrice }
 					showIcon={ showIcon }
diff --git a/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-shipping-address-block/edit.tsx b/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-shipping-address-block/edit.tsx
index 3b491a8bd84..f4d3142c682 100644
--- a/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-shipping-address-block/edit.tsx
+++ b/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-shipping-address-block/edit.tsx
@@ -3,6 +3,7 @@
  */
 import classnames from 'classnames';
 import { useBlockProps } from '@wordpress/block-editor';
+import { useCheckoutAddress } from '@woocommerce/base-context/hooks';
 import { innerBlockAreas } from '@woocommerce/blocks-checkout';
 import { useCheckoutAddress } from '@woocommerce/base-context/hooks';
 /**
@@ -41,9 +42,11 @@ export const Edit = ( {
 	const { addressFieldControls: Controls } =
 		useCheckoutBlockControlsContext();
 	const { showShippingFields } = useCheckoutAddress();
+
 	if ( ! showShippingFields ) {
 		return null;
 	}
+
 	return (
 		<FormStepBlock
 			setAttributes={ setAttributes }
diff --git a/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-shipping-methods-block/edit.tsx b/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-shipping-methods-block/edit.tsx
index 33c641169c2..6ae76c5394a 100644
--- a/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-shipping-methods-block/edit.tsx
+++ b/plugins/woocommerce-blocks/assets/js/blocks/checkout/inner-blocks/checkout-shipping-methods-block/edit.tsx
@@ -8,6 +8,7 @@ import { PanelBody, ExternalLink } from '@wordpress/components';
 import { ADMIN_URL, getSetting } from '@woocommerce/settings';
 import ExternalLinkCard from '@woocommerce/editor-components/external-link-card';
 import { innerBlockAreas } from '@woocommerce/blocks-checkout';
+import { useCheckoutAddress } from '@woocommerce/base-context/hooks';
 import Noninteractive from '@woocommerce/base-components/noninteractive';
 
 /**
@@ -37,7 +38,7 @@ export const Edit = ( {
 		className: string;
 	};
 	setAttributes: ( attributes: Record< string, unknown > ) => void;
-} ): JSX.Element => {
+} ): JSX.Element | null => {
 	const globalShippingMethods = getSetting(
 		'globalShippingMethods'
 	) as shippingAdminLink[];
@@ -45,6 +46,12 @@ export const Edit = ( {
 		'activeShippingZones'
 	) as shippingAdminLink[];
 
+	const { showShippingFields } = useCheckoutAddress();
+
+	if ( ! showShippingFields ) {
+		return null;
+	}
+
 	return (
 		<FormStepBlock
 			attributes={ attributes }