From 2943168803a1d5f1359934e78c3e11437882d28b Mon Sep 17 00:00:00 2001
From: Loan Laux <loan@outgrow.io>
Date: Thu, 5 Aug 2021 19:01:24 +0200
Subject: [PATCH] add useAddShippingAddress hook + add phone and region fields
 in address

Signed-off-by: Loan Laux <loan@outgrow.io>
---
 .../PaymentMethodView/PaymentMethodView.tsx   | 15 ++-
 .../checkout/ShippingView/ShippingView.tsx    | 48 +++++++--
 components/ui/context.tsx                     |  7 +-
 .../cart/use-add-shipping-address.tsx         | 20 ++++
 framework/commerce/index.tsx                  |  1 +
 framework/commerce/types/cart.ts              | 23 +++++
 .../api/endpoints/cart/add-item.ts            | 17 ++--
 .../cart/use-add-shipping-address.tsx         | 97 +++++++++++++++++++
 framework/reactioncommerce/provider.ts        |  9 +-
 .../reactioncommerce/utils/mutations/index.ts |  1 +
 .../mutations/set-shipping-address-on-cart.ts | 13 +++
 11 files changed, 229 insertions(+), 22 deletions(-)
 create mode 100644 framework/commerce/cart/use-add-shipping-address.tsx
 create mode 100644 framework/reactioncommerce/cart/use-add-shipping-address.tsx
 create mode 100644 framework/reactioncommerce/utils/mutations/set-shipping-address-on-cart.ts

diff --git a/components/checkout/PaymentMethodView/PaymentMethodView.tsx b/components/checkout/PaymentMethodView/PaymentMethodView.tsx
index 873371add..ebf558fec 100644
--- a/components/checkout/PaymentMethodView/PaymentMethodView.tsx
+++ b/components/checkout/PaymentMethodView/PaymentMethodView.tsx
@@ -144,12 +144,21 @@ const PaymentMethodView: FC = () => {
             </div>
           </div>
           <div className={s.fieldset}>
-            <label className={s.label}>Country/Region</label>
+            <label className={s.label}>Region</label>
+            <input
+              className={s.input}
+              name='region'
+              onChange={updateAddressData}
+              value={address.region}
+            />
+          </div>
+          <div className={s.fieldset}>
+            <label className={s.label}>Country</label>
             <select
               className={s.select}
-              name='countryOrRegion'
+              name='country'
               onChange={updateAddressData}
-              value={address.countryOrRegion}
+              value={address.country}
             >
               {countries.map((country) => (
                 <option
diff --git a/components/checkout/ShippingView/ShippingView.tsx b/components/checkout/ShippingView/ShippingView.tsx
index ded5ddaa1..7ffd9a8c2 100644
--- a/components/checkout/ShippingView/ShippingView.tsx
+++ b/components/checkout/ShippingView/ShippingView.tsx
@@ -1,13 +1,22 @@
-import { FC, useState } from 'react'
+import { FC } from 'react'
 import cn from 'classnames'
 import s from './ShippingView.module.css'
 import Button from '@components/ui/Button'
 import { useUI } from '@components/ui/context'
 import SidebarLayout from '@components/common/SidebarLayout'
 import countries from '@lib/countries'
+import useAddShippingAddress from '@framework/cart/use-add-shipping-address'
 
 const PaymentMethodView: FC = () => {
-  const { paymentMethodDetails, setShippingAddress, setSidebarView, setUseBillingAddressForShipping, shippingAddress, useBillingAddressForShipping } = useUI()
+  const addShippingAddress = useAddShippingAddress()
+  const {
+    paymentMethodDetails,
+    setShippingAddress,
+    setSidebarView,
+    setUseBillingAddressForShipping,
+    shippingAddress,
+    useBillingAddressForShipping,
+  } = useUI()
 
   const handleUseBillingAddressForShipping = (event) => {
     setUseBillingAddressForShipping(event.target.value === 'true')
@@ -21,7 +30,7 @@ const PaymentMethodView: FC = () => {
         addressLine2: paymentMethodDetails.address?.addressLine2,
         postalCode: paymentMethodDetails.address?.postalCode,
         city: paymentMethodDetails.address?.city,
-        countryOrRegion: paymentMethodDetails.address?.countryOrRegion,
+        country: paymentMethodDetails.address?.country,
       })
     }
   }
@@ -32,7 +41,11 @@ const PaymentMethodView: FC = () => {
   })
 
   return (
-    <SidebarLayout handleBack={() => setSidebarView('CHECKOUT_VIEW')}>
+    <SidebarLayout handleBack={async () => {
+      // add shipping address to cart
+      await addShippingAddress({ address: shippingAddress })
+      setSidebarView('CHECKOUT_VIEW')
+    }}>
       <div className="px-4 sm:px-6 flex-1">
         <h2 className="pt-1 pb-8 text-2xl font-semibold tracking-wide cursor-pointer inline-block">
           Shipping
@@ -87,6 +100,17 @@ const PaymentMethodView: FC = () => {
               />
             </div>
           </div>
+          <div className={s.fieldset}>
+            <label className={s.label}>Phone Number</label>
+            <input
+              className={s.input}
+              name='phone'
+              onChange={updateAddressData}
+              value={shippingAddress.phone}
+              disabled={useBillingAddressForShipping}
+              type='tel'
+            />
+          </div>
           <div className={s.fieldset}>
             <label className={s.label}>Company (Optional)</label>
             <input
@@ -140,12 +164,22 @@ const PaymentMethodView: FC = () => {
             </div>
           </div>
           <div className={s.fieldset}>
-            <label className={s.label}>Country/Region</label>
+            <label className={s.label}>Region</label>
+            <input
+              className={s.input}
+              name='region'
+              onChange={updateAddressData}
+              value={shippingAddress.region}
+              disabled={useBillingAddressForShipping}
+            />
+          </div>
+          <div className={s.fieldset}>
+            <label className={s.label}>Country</label>
             <select
               className={s.select}
-              name="countryOrRegion"
+              name="country"
               onChange={updateAddressData}
-              value={shippingAddress.countryOrRegion}
+              value={shippingAddress.country}
               disabled={useBillingAddressForShipping}
 
             >
diff --git a/components/ui/context.tsx b/components/ui/context.tsx
index aa27f19d8..98f5da21e 100644
--- a/components/ui/context.tsx
+++ b/components/ui/context.tsx
@@ -28,7 +28,8 @@ export const initialState = {
       addressLine2: '',
       postalCode: '',
       city: '',
-      countryOrRegion: '',
+      country: '',
+      region: '',
     },
     paymentMethod: null,
   },
@@ -40,7 +41,9 @@ export const initialState = {
     addressLine2: '',
     postalCode: '',
     city: '',
-    countryOrRegion: '',
+    country: '',
+    region: '',
+    phone: '',
   },
   useBillingAddressForShipping: true,
 }
diff --git a/framework/commerce/cart/use-add-shipping-address.tsx b/framework/commerce/cart/use-add-shipping-address.tsx
new file mode 100644
index 000000000..1909ff180
--- /dev/null
+++ b/framework/commerce/cart/use-add-shipping-address.tsx
@@ -0,0 +1,20 @@
+import { useHook, useMutationHook } from '../utils/use-hook'
+import { mutationFetcher } from '../utils/default-fetcher'
+import type { HookFetcherFn, MutationHook } from '../utils/types'
+import type { AddShippingAddressHook } from '../types/cart'
+import type { Provider } from '..'
+
+export type UseAddShippingAddress<
+  H extends MutationHook<AddShippingAddressHook<any>> = MutationHook<AddShippingAddressHook>
+> = ReturnType<H['useHook']>
+
+export const fetcher: HookFetcherFn<AddShippingAddressHook> = mutationFetcher
+
+const fn = (provider: Provider) => provider.cart?.useAddShippingAddress!
+
+const useRemoveItem: UseAddShippingAddress = (input) => {
+  const hook = useHook(fn)
+  return useMutationHook({ fetcher, ...hook })(input)
+}
+
+export default useRemoveItem
diff --git a/framework/commerce/index.tsx b/framework/commerce/index.tsx
index 7ecb44dc1..1f4376231 100644
--- a/framework/commerce/index.tsx
+++ b/framework/commerce/index.tsx
@@ -26,6 +26,7 @@ export type Provider = CommerceConfig & {
   cart?: {
     useCart?: SWRHook<Cart.GetCartHook>
     useAddItem?: MutationHook<Cart.AddItemHook>
+    useAddShippingAddress?: MutationHook<Cart.AddShippingAddressHook>
     useUpdateItem?: MutationHook<Cart.UpdateItemHook>
     useRemoveItem?: MutationHook<Cart.RemoveItemHook>
   }
diff --git a/framework/commerce/types/cart.ts b/framework/commerce/types/cart.ts
index f9252961e..9090fa502 100644
--- a/framework/commerce/types/cart.ts
+++ b/framework/commerce/types/cart.ts
@@ -96,6 +96,20 @@ export type CartItemBody = {
   }
 }
 
+export type Address = {
+  cardholderName: string
+  firstName: string
+  lastName: string
+  company: string
+  addressLine1: string
+  addressLine2: string
+  postalCode: string
+  city: string
+  region: string
+  country: string
+  phone: string
+}
+
 /**
  * Hooks schema
  */
@@ -104,6 +118,7 @@ export type CartTypes = {
   cart?: Cart
   item: LineItem
   itemBody: CartItemBody
+  address: Address
 }
 
 export type CartHooks<T extends CartTypes = CartTypes> = {
@@ -144,6 +159,14 @@ export type RemoveItemHook<T extends CartTypes = CartTypes> = {
   actionInput: { id: string }
 }
 
+export type AddShippingAddressHook<T extends CartTypes = CartTypes> = {
+  data: T['cart'] | null
+  input: { address?: T['address'] }
+  fetcherInput: { address: Address }
+  body: { itemId: string }
+  actionInput: { id: string }
+}
+
 /**
  * API Schema
  */
diff --git a/framework/reactioncommerce/api/endpoints/cart/add-item.ts b/framework/reactioncommerce/api/endpoints/cart/add-item.ts
index 5af2e815f..d0d14f8fe 100644
--- a/framework/reactioncommerce/api/endpoints/cart/add-item.ts
+++ b/framework/reactioncommerce/api/endpoints/cart/add-item.ts
@@ -95,14 +95,13 @@ const addItem: CartEndpoint['handlers']['addItem'] = async ({
   }
 
   if (anonymousCartToken && reactionCustomerToken) {
-    console.log('reconciliating carts')(
-      ({ _id: cartId } = await reconcileCarts({
-        config,
-        cartId,
-        anonymousCartToken,
-        reactionCustomerToken,
-      }))
-    )
+    console.log('reconciliating carts')
+    ({ _id: cartId } = await reconcileCarts({
+      config,
+      cartId,
+      anonymousCartToken,
+      reactionCustomerToken,
+    }))
 
     // Clear the anonymous cart token cookie and update cart ID cookie
     res.setHeader('Set-Cookie', [
@@ -135,7 +134,7 @@ const addItem: CartEndpoint['handlers']['addItem'] = async ({
       headers: {
         ...authorizationHeaderParam,
       },
-    }
+    },
   )
 
   console.log('updatedCart', updatedCart)
diff --git a/framework/reactioncommerce/cart/use-add-shipping-address.tsx b/framework/reactioncommerce/cart/use-add-shipping-address.tsx
new file mode 100644
index 000000000..35074b006
--- /dev/null
+++ b/framework/reactioncommerce/cart/use-add-shipping-address.tsx
@@ -0,0 +1,97 @@
+import { useCallback } from 'react'
+
+import type {
+  MutationHookContext,
+  HookFetcherContext,
+} from '@commerce/utils/types'
+
+import { ValidationError } from '@commerce/utils/errors'
+
+import useAddShippingAddress, {
+  AddShippingAddressInput as AddShippingAddressInputBase,
+  UseAddShippingAddress,
+} from '@commerce/cart/use-add-shipping-address'
+
+import useCart from './use-cart'
+import {
+  setShippingAddressOnCartMutation,
+  getAnonymousCartToken,
+  getAnonymousCartId,
+  normalizeCart,
+} from '../utils'
+import { Cart, LineItem } from '../types'
+import { Mutation, MutationUpdateCartItemsQuantityArgs } from '../schema'
+import { AddShippingAddressBody } from '@commerce/types'
+
+export type AddShippingAddressFn<T = any> = T extends LineItem
+  ? (input?: AddShippingAddressInput<T>) => Promise<Cart | null>
+  : (input: AddShippingAddressInput<T>) => Promise<Cart | null>
+
+export type AddShippingAddressInput<T = any> = T extends LineItem
+  ? Partial<AddShippingAddressInputBase>
+  : AddShippingAddressInputBase
+
+export default useAddShippingAddress as UseAddShippingAddress<typeof handler>
+
+export const handler = {
+  fetchOptions: {
+    query: setShippingAddressOnCartMutation,
+  },
+  async fetcher({
+    input: { address },
+    options,
+    fetch,
+  }: HookFetcherContext<AddShippingAddressBody>) {
+    const { setShippingAddressOnCart } = await fetch<
+      Mutation,
+      MutationUpdateCartItemsQuantityArgs
+      >({
+      ...options,
+      variables: {
+        input: {
+          cartId: getAnonymousCartId(),
+          cartToken: getAnonymousCartToken(),
+          address: {
+            address1: address.addressLine1,
+            address2: address.addressLine2,
+            city: address.city,
+            company: address.company,
+            country: address.country,
+            firstName: address.firstName,
+            lastName: address.lastName,
+            fullName: `${address.firstName} ${address.lastName}`,
+            phone: address.phone,
+            postal: address.postalCode,
+            region: address.region,
+          },
+        },
+      },
+    })
+    return normalizeCart(setShippingAddressOnCart?.cart)
+  },
+  useHook: ({
+    fetch,
+  }: MutationHookContext<Cart | null, AddShippingAddressBody>) => <
+    T extends LineItem | undefined = undefined
+    >(
+    ctx: { address?: T } = {}
+  ) => {
+    const { mutate } = useCart()
+    const addShippingAddress: AddShippingAddressFn<LineItem> = async (input) => {
+      const address = input?.address ?? ctx.address
+
+      console.log('addShippingAddress input', input)
+      if (!address) {
+        throw new ValidationError({
+          message: 'Invalid input used for this operation',
+        })
+      }
+
+      const data = await fetch({ input: { address } })
+      await mutate(data, false)
+      return data
+    }
+
+    return useCallback(addShippingAddress as AddShippingAddressFn<T>, [fetch, mutate])
+  },
+}
diff --git a/framework/reactioncommerce/provider.ts b/framework/reactioncommerce/provider.ts
index 30d6b7c7f..896c15a8a 100644
--- a/framework/reactioncommerce/provider.ts
+++ b/framework/reactioncommerce/provider.ts
@@ -5,6 +5,7 @@ import {
 
 import { handler as useCart } from './cart/use-cart'
 import { handler as useAddItem } from './cart/use-add-item'
+import { handler as useAddShippingAddress } from './cart/use-add-shipping-address'
 import { handler as useUpdateItem } from './cart/use-update-item'
 import { handler as useRemoveItem } from './cart/use-remove-item'
 
@@ -22,7 +23,13 @@ export const reactionCommerceProvider = {
   anonymousCartTokenCookie: REACTION_ANONYMOUS_CART_TOKEN_COOKIE,
   cartIdCookie: REACTION_CART_ID_COOKIE,
   fetcher,
-  cart: { useCart, useAddItem, useUpdateItem, useRemoveItem },
+  cart: {
+    useCart,
+    useAddItem,
+    useAddShippingAddress,
+    useUpdateItem,
+    useRemoveItem,
+  },
   customer: { useCustomer },
   products: { useSearch },
   auth: { useLogin, useLogout, useSignup },
diff --git a/framework/reactioncommerce/utils/mutations/index.ts b/framework/reactioncommerce/utils/mutations/index.ts
index f27a0d976..c0fe1942d 100644
--- a/framework/reactioncommerce/utils/mutations/index.ts
+++ b/framework/reactioncommerce/utils/mutations/index.ts
@@ -4,4 +4,5 @@ export { default as createCartMutation } from './create-cart'
 export { default as authenticateMutation } from './authenticate'
 export { default as logoutMutation } from './logout'
 export { default as removeCartItemsMutation } from './remove-cart-items'
+export { default as setShippingAddressOnCartMutation } from './set-shipping-address-on-cart'
 export { default as updateCartItemsQuantityMutation } from './update-cart-items-quantity'
diff --git a/framework/reactioncommerce/utils/mutations/set-shipping-address-on-cart.ts b/framework/reactioncommerce/utils/mutations/set-shipping-address-on-cart.ts
new file mode 100644
index 000000000..5312ed0dc
--- /dev/null
+++ b/framework/reactioncommerce/utils/mutations/set-shipping-address-on-cart.ts
@@ -0,0 +1,13 @@
+import { cartPayloadFragment } from '@framework/utils/queries/get-cart-query'
+
+const setShippingAddressOnCartMutation = `
+  mutation setShippingAddressOnCartMutation($input: SetShippingAddressOnCartInput!) {
+    setShippingAddressOnCart(input: $input) {
+      cart {
+        ${cartPayloadFragment}
+      }
+    }
+  }
+`
+
+export default setShippingAddressOnCartMutation