From e938e0783d468128c1a7ee4de3d73690c12feb7f Mon Sep 17 00:00:00 2001
From: Lee Robinson <lrobinson2011@gmail.com>
Date: Sun, 28 Jul 2024 22:56:37 -0500
Subject: [PATCH] fix last bug

---
 components/product/gallery.tsx          | 18 ++++++++++----
 components/product/product-context.tsx  | 31 ++++++++++++++++---------
 components/product/variant-selector.tsx |  8 +++++--
 components/welcome-toast.tsx            |  1 -
 4 files changed, 40 insertions(+), 18 deletions(-)

diff --git a/components/product/gallery.tsx b/components/product/gallery.tsx
index 3bbb8fd6d..79659096c 100644
--- a/components/product/gallery.tsx
+++ b/components/product/gallery.tsx
@@ -2,11 +2,12 @@
 
 import { ArrowLeftIcon, ArrowRightIcon } from '@heroicons/react/24/outline';
 import { GridTileImage } from 'components/grid/tile';
-import { useProduct } from 'components/product/product-context';
+import { useProduct, useUpdateURL } from 'components/product/product-context';
 import Image from 'next/image';
 
 export function Gallery({ images }: { images: { src: string; altText: string }[] }) {
   const { state, updateImage } = useProduct();
+  const updateURL = useUpdateURL();
   const imageIndex = state.image ? parseInt(state.image) : 0;
 
   const nextImageIndex = imageIndex + 1 < images.length ? imageIndex + 1 : 0;
@@ -33,7 +34,10 @@ export function Gallery({ images }: { images: { src: string; altText: string }[]
           <div className="absolute bottom-[15%] flex w-full justify-center">
             <div className="mx-auto flex h-11 items-center rounded-full border border-white bg-neutral-50/80 text-neutral-500 backdrop-blur dark:border-black dark:bg-neutral-900/80">
               <button
-                formAction={() => updateImage(previousImageIndex.toString())}
+                formAction={() => {
+                  const newState = updateImage(previousImageIndex.toString());
+                  updateURL(newState);
+                }}
                 aria-label="Previous product image"
                 className={buttonClassName}
               >
@@ -41,7 +45,10 @@ export function Gallery({ images }: { images: { src: string; altText: string }[]
               </button>
               <div className="mx-1 h-6 w-px bg-neutral-500"></div>
               <button
-                formAction={() => updateImage(nextImageIndex.toString())}
+                formAction={() => {
+                  const newState = updateImage(nextImageIndex.toString());
+                  updateURL(newState);
+                }}
                 aria-label="Next product image"
                 className={buttonClassName}
               >
@@ -60,7 +67,10 @@ export function Gallery({ images }: { images: { src: string; altText: string }[]
             return (
               <li key={image.src} className="h-20 w-20">
                 <button
-                  formAction={() => updateImage(index.toString())}
+                  formAction={() => {
+                    const newState = updateImage(index.toString());
+                    updateURL(newState);
+                  }}
                   aria-label="Select product image"
                   className="h-full w-full"
                 >
diff --git a/components/product/product-context.tsx b/components/product/product-context.tsx
index 7be633b26..cc964c5a4 100644
--- a/components/product/product-context.tsx
+++ b/components/product/product-context.tsx
@@ -11,14 +11,13 @@ type ProductState = {
 
 type ProductContextType = {
   state: ProductState;
-  updateOption: (name: string, value: string) => void;
-  updateImage: (index: string) => void;
+  updateOption: (name: string, value: string) => ProductState;
+  updateImage: (index: string) => ProductState;
 };
 
 const ProductContext = createContext<ProductContextType | undefined>(undefined);
 
 export function ProductProvider({ children }: { children: React.ReactNode }) {
-  const router = useRouter();
   const searchParams = useSearchParams();
 
   const getInitialState = () => {
@@ -38,17 +37,15 @@ export function ProductProvider({ children }: { children: React.ReactNode }) {
   );
 
   const updateOption = (name: string, value: string) => {
-    setOptimisticState({ [name]: value });
-    const newParams = new URLSearchParams(window.location.search);
-    newParams.set(name, value);
-    router.push(`?${newParams.toString()}`, { scroll: false });
+    const newState = { [name]: value };
+    setOptimisticState(newState);
+    return { ...state, ...newState };
   };
 
   const updateImage = (index: string) => {
-    setOptimisticState({ image: index });
-    const newParams = new URLSearchParams(window.location.search);
-    newParams.set('image', index);
-    router.push(`?${newParams.toString()}`, { scroll: false });
+    const newState = { image: index };
+    setOptimisticState(newState);
+    return { ...state, ...newState };
   };
 
   const value = useMemo(
@@ -70,3 +67,15 @@ export function useProduct() {
   }
   return context;
 }
+
+export function useUpdateURL() {
+  const router = useRouter();
+
+  return (state: ProductState) => {
+    const newParams = new URLSearchParams(window.location.search);
+    Object.entries(state).forEach(([key, value]) => {
+      newParams.set(key, value);
+    });
+    router.push(`?${newParams.toString()}`, { scroll: false });
+  };
+}
diff --git a/components/product/variant-selector.tsx b/components/product/variant-selector.tsx
index ad432ca1f..e4b1dd0c5 100644
--- a/components/product/variant-selector.tsx
+++ b/components/product/variant-selector.tsx
@@ -1,7 +1,7 @@
 'use client';
 
 import clsx from 'clsx';
-import { useProduct } from 'components/product/product-context';
+import { useProduct, useUpdateURL } from 'components/product/product-context';
 import { ProductOption, ProductVariant } from 'lib/shopify/types';
 
 type Combination = {
@@ -18,6 +18,7 @@ export function VariantSelector({
   variants: ProductVariant[];
 }) {
   const { state, updateOption } = useProduct();
+  const updateURL = useUpdateURL();
   const hasNoOptionsOrJustOneOption =
     !options.length || (options.length === 1 && options[0]?.values.length === 1);
 
@@ -62,7 +63,10 @@ export function VariantSelector({
 
             return (
               <button
-                formAction={() => updateOption(optionNameLowerCase, value)}
+                formAction={() => {
+                  const newState = updateOption(optionNameLowerCase, value);
+                  updateURL(newState);
+                }}
                 key={value}
                 aria-disabled={!isAvailableForSale}
                 disabled={!isAvailableForSale}
diff --git a/components/welcome-toast.tsx b/components/welcome-toast.tsx
index 9475fafd3..e49d5f0db 100644
--- a/components/welcome-toast.tsx
+++ b/components/welcome-toast.tsx
@@ -8,7 +8,6 @@ export function WelcomeToast() {
     // ignore if screen height is too small
     if (window.innerHeight < 650) return;
     if (!document.cookie.includes('welcome-toast=2')) {
-      console.log('Welcome to Next.js Commerce!');
       toast('🛍️ Welcome to Next.js Commerce!', {
         id: 'welcome-toast',
         duration: Infinity,