3 files changed, 29 insertions, 15 deletions
diff --git a/src-migrate/modules/cart/components/ItemAction.tsx b/src-migrate/modules/cart/components/ItemAction.tsx
index e73d507b..e5e7f314 100644
--- a/src-migrate/modules/cart/components/ItemAction.tsx
+++ b/src-migrate/modules/cart/components/ItemAction.tsx
@@ -11,6 +11,7 @@ import { deleteUserCart, upsertUserCart } from '~/services/cart'
 
 import { useDebounce } from 'usehooks-ts'
 import { useCartStore } from '../stores/useCartStore'
+import { useProductCartContext } from '@/contexts/ProductCartContext'
 
 
 type Props = {
@@ -19,7 +20,7 @@ type Props = {
 
 const CartItemAction = ({ item }: Props) => {
   const auth = getAuth()
-
+  const { setRefreshCart } = useProductCartContext()
   const [isLoadDelete, setIsLoadDelete] = useState<boolean>(false)
   const [isLoadQuantity, setIsLoadQuantity] = useState<boolean>(false)
 
@@ -36,6 +37,7 @@ const CartItemAction = ({ item }: Props) => {
     await deleteUserCart(auth.id, [item.cart_id])
     await loadCart(auth.id)
     setIsLoadDelete(false)
+    setRefreshCart(true)
   }
 
   const decreaseQty = () => { setQuantity((quantity) => quantity -= 1) }
diff --git a/src-migrate/modules/cart/components/ItemSelect.tsx b/src-migrate/modules/cart/components/ItemSelect.tsx
index b904a1de..d4a1b537 100644
--- a/src-migrate/modules/cart/components/ItemSelect.tsx
+++ b/src-migrate/modules/cart/components/ItemSelect.tsx
@@ -13,23 +13,25 @@ type Props = {
 
 const CartItemSelect = ({ item }: Props) => {
   const auth = getAuth()
-  const { loadCart } = useCartStore()
+  const { updateCartItem, cart } = useCartStore()
 
   const [isLoad, setIsLoad] = useState<boolean>(false)
 
   const handleChange = async (e: React.ChangeEvent<HTMLInputElement>) => {
-    if (typeof auth !== 'object') return
-
-    setIsLoad(true)
-    await upsertUserCart({
-      userId: auth.id,
-      type: item.cart_type,
-      id: item.id,
-      qty: item.quantity,
-      selected: e.target.checked
-    })
-    await loadCart(auth.id)
-    setIsLoad(false)
+    if (typeof auth !== 'object' || !cart) return
+    
+    setIsLoad(true);
+    const updatedCartItems = cart.products.map(cartItem =>
+      cartItem.id === item.id
+        ? { ...cartItem, selected: e.target.checked }
+        : cartItem
+    );
+
+    // Update the entire cart
+    const updatedCart = { ...cart, products: updatedCartItems };
+    updateCartItem(updatedCart);
+
+    setIsLoad(false);
   }
 
   return (
diff --git a/src-migrate/modules/cart/stores/useCartStore.ts b/src-migrate/modules/cart/stores/useCartStore.ts
index 3d9a0aed..3b50ec32 100644
--- a/src-migrate/modules/cart/stores/useCartStore.ts
+++ b/src-migrate/modules/cart/stores/useCartStore.ts
@@ -1,5 +1,5 @@
 import { create } from 'zustand';
-import { CartProps } from '~/types/cart';
+import { CartItem, CartProps } from '~/types/cart';
 import { getUserCart } from '~/services/cart';
 
 type State = {
@@ -16,6 +16,7 @@ type State = {
 
 type Action = {
   loadCart: (userId: number) => Promise<void>;
+  updateCartItem: (updateCart: CartProps) => void;
 };
 
 export const useCartStore = create<State & Action>((set, get) => ({
@@ -39,6 +40,15 @@ export const useCartStore = create<State & Action>((set, get) => ({
     const summary = computeSummary(cart);
     set({ summary });
   },
+  updateCartItem: (updatedCart) => {
+    const cart = get().cart;
+    if (!cart) return;
+    
+    set({ cart: updatedCart });
+    const summary = computeSummary(updatedCart);
+    set({ summary });
+  },
+  
 }));
 
 const computeSummary = (cart: CartProps) => {