Add to Wishlist from PDP

packages/peregrine/lib/talons/ProductFullDetail/__tests__/useProductFullDetail.spec.js

@@ -1,18 +1,37 @@
 import React from 'react';
-import { useMutation } from '@apollo/client';
+import { act } from 'react-test-renderer';
+import { useMutation, useQuery } from '@apollo/client';
 
 import createTestInstance from '../../../util/createTestInstance';
 import { useProductFullDetail } from '../useProductFullDetail';
+import { useUserContext } from '../../../context/user';
 
 jest.mock('@apollo/client', () => ({
     useMutation: jest.fn().mockImplementation(() => [
         jest.fn(),
         {
             error: null
         }
-    ])
+    ]),
+    useQuery: jest.fn().mockImplementation(() => ({
+        data: {
+            storeConfig: {
+                magento_wishlist_general_is_enabled: true
+            }
+        },
+        loading: false,
+        error: false
+    }))
 }));
 
+jest.mock('@magento/peregrine/lib/context/user', () => {
+    const userState = { isSignedIn: false };
+    const userApi = {};
+    const useUserContext = jest.fn(() => [userState, userApi]);
+
+    return { useUserContext };
+});
+
 jest.mock('@magento/peregrine/lib/context/cart', () => {
     const cartState = { cartId: 'ThisIsMyCart' };
     const cartApi = {};
@@ -38,10 +57,123 @@ const defaultProps = {
                     value: 99
                 }
             }
-        }
+        },
+        sku: 'MySimpleProductSku'
     }
 };
 
+describe('shouldShowWishlistButton', () => {
+    test('is false if not signed in', () => {
+        useUserContext.mockReturnValueOnce([{ isSignedIn: false }]);
+        const tree = createTestInstance(<Component {...defaultProps} />);
+
+        const { root } = tree;
+        const { talonProps } = root.findByType('i').props;
+
+        expect(talonProps.shouldShowWishlistButton).toBeFalsy();
+    });
+
+    test('is false if wishlist is disabled in config', () => {
+        useUserContext.mockReturnValueOnce([{ isSignedIn: true }]);
+        useQuery.mockReturnValueOnce({
+            data: {
+                storeConfig: {
+                    magento_wishlist_general_is_enabled: false
+                }
+            },
+            loading: false,
+            error: false
+        });
+        const tree = createTestInstance(<Component {...defaultProps} />);
+
+        const { root } = tree;
+        const { talonProps } = root.findByType('i').props;
+
+        expect(talonProps.shouldShowWishlistButton).toBeFalsy();
+    });
+
+    test('is true if signed in and wishlist is enabled', () => {
+        useUserContext.mockReturnValueOnce([{ isSignedIn: true }]);
+        useQuery.mockReturnValueOnce({
+            data: {
+                storeConfig: {
+                    magento_wishlist_general_is_enabled: true
+                }
+            },
+            loading: false,
+            error: false
+        });
+        const tree = createTestInstance(<Component {...defaultProps} />);
+
+        const { root } = tree;
+        const { talonProps } = root.findByType('i').props;
+
+        expect(talonProps.shouldShowWishlistButton).toBeTruthy();
+    });
+});
+
+describe('wishlistItemOptions', () => {
+    test('returns quantity and sku for all products', () => {
+        const tree = createTestInstance(<Component {...defaultProps} />);
+
+        const { root } = tree;
+        const { talonProps } = root.findByType('i').props;
+
+        expect(talonProps.wishlistItemOptions).toMatchObject({
+            quantity: 1,
+            sku: defaultProps.product.sku
+        });
+    });
+
+    test('returns selected_options for ConfigurableProducts', () => {
+        const optionId = 1;
+        const selectionId = 2;
+        const uid = 'foo';
+
+        const props = {
+            ...defaultProps,
+            product: {
+                ...defaultProps.product,
+                sku: 'MyConfigurableProductSku',
+                __typename: 'ConfigurableProduct',
+                configurable_options: [
+                    {
+                        attribute_id: optionId,
+                        values: [{ uid, value_index: selectionId }]
+                    }
+                ],
+                variants: []
+            }
+        };
+        const tree = createTestInstance(<Component {...props} />);
+
+        const { root } = tree;
+
+        expect(
+            root.findByType('i').props.talonProps.wishlistItemOptions
+        ).toMatchObject({
+            quantity: 1,
+            sku: props.product.sku,
+            selected_options: []
+        });
+
+        act(() => {
+            root.findByType('i').props.talonProps.handleSelectionChange(
+                optionId,
+                selectionId
+            );
+        });
+
+        expect(
+            root.findByType('i').props.talonProps.wishlistItemOptions
+        ).toMatchObject({
+            quantity: 1,
+            sku: props.product.sku,
+            selected_options: [uid]
+        });
+    });
+});
+
 test('returns undefined category if there are no categories for the product', () => {
     const props = {
         ...defaultProps,

packages/peregrine/lib/talons/ProductFullDetail/useProductFullDetail.js

@@ -1,6 +1,7 @@
 import { useCallback, useState, useMemo } from 'react';
-import { useMutation } from '@apollo/client';
+import { useMutation, useQuery } from '@apollo/client';
 import { useCartContext } from '@magento/peregrine/lib/context/cart';
+import { useUserContext } from '@magento/peregrine/lib/context/user';
 
 import { appendOptionsToPayload } from '@magento/peregrine/lib/util/appendOptionsToPayload';
 import { findMatchingVariant } from '@magento/peregrine/lib/util/findMatchingProductVariant';
@@ -150,6 +151,7 @@ const SUPPORTED_PRODUCT_TYPES = ['SimpleProduct', 'ConfigurableProduct'];
 /**
  * @param {GraphQLQuery} props.addConfigurableProductToCartMutation - configurable product mutation
  * @param {GraphQLQuery} props.addSimpleProductToCartMutation - configurable product mutation
+ * @param {GraphQLQuery} props.getWishlistConfig - queries for whether wishlists are enabled.
  * @param {Object} props.product - the product, see RootComponents/Product
  *
  * @returns {{
@@ -169,6 +171,7 @@ export const useProductFullDetail = props => {
     const {
         addConfigurableProductToCartMutation,
         addSimpleProductToCartMutation,
+        getWishlistConfig,
         product
     } = props;
 
@@ -179,6 +182,11 @@ export const useProductFullDetail = props => {
     );
 
     const [{ cartId }] = useCartContext();
+    const [{ isSignedIn }] = useUserContext();
+
+    const { data: storeConfigData } = useQuery(getWishlistConfig, {
+        fetchPolicy: 'cache-and-network'
+    });
 
     const [
         addConfigurableProductToCart,
@@ -222,6 +230,39 @@ export const useProductFullDetail = props => {
         [product, optionCodes, optionSelections]
     );
 
+    // The map of ids to values (and their uids)
+    // For example:
+    // { "179" => [{ uid: "abc", value_index: 1 }, { uid: "def", value_index: 2 }]}
+    const attributeIdToValuesMap = useMemo(() => {
+        const map = new Map();
+        // For simple items, this will be an empty map.
+        const options = product.configurable_options || [];
+        for (const { attribute_id, values } of options) {
+            map.set(attribute_id, values);
+        }
+        return map;
+    }, [product.configurable_options]);
+
+    // An array of selected option uids. Useful for passing to mutations.
+    // For example:
+    // ["abc", "def"]
+    const selectedOptionsArray = useMemo(() => {
+        const selectedOptions = [];
+
+        optionSelections.forEach((value, key) => {
+            const values = attributeIdToValuesMap.get(key);
+
+            const selectedValue = values.find(
+                item => item.value_index === value
+            );
+
+            if (selectedValue) {
+                selectedOptions.push(selectedValue.uid);
+            }
+        });
+        return selectedOptions;
+    }, [attributeIdToValuesMap, optionSelections]);
+
     const handleAddToCart = useCallback(
         async formValues => {
             const { quantity } = formValues;
@@ -310,6 +351,19 @@ export const useProductFullDetail = props => {
         [errorAddingConfigurableProduct, errorAddingSimpleProduct]
     );
 
+    const wishlistItemOptions = useMemo(() => {
+        const options = {
+            quantity: 1,
+            sku: product.sku
+        };
+
+        if (productType === 'ConfigurableProduct') {
+            options.selected_options = selectedOptionsArray;
+        }
+
+        return options;
+    }, [product, productType, selectedOptionsArray]);
+
     return {
         breadcrumbCategoryId,
         errorMessage: derivedErrorMessage,
@@ -319,6 +373,11 @@ export const useProductFullDetail = props => {
             isMissingOptions || isAddConfigurableLoading || isAddSimpleLoading,
         isSupportedProductType,
         mediaGalleryEntries,
-        productDetails
+        shouldShowWishlistButton:
+            isSignedIn &&
+            storeConfigData &&
+            !!storeConfigData.storeConfig.magento_wishlist_general_is_enabled,
+        productDetails,
+        wishlistItemOptions
     };
 };

packages/peregrine/lib/talons/RootComponents/Product/productDetailFragment.gql.js

@@ -42,6 +42,7 @@ export const ProductDetailsFragment = gql`
                 id
                 label
                 values {
+                    uid
                     default_label
                     label
                     store_label