From 9176b07e2662456c81627598092ed3d5d85d1b55 Mon Sep 17 00:00:00 2001
From: Julian Lechner <jll38@njit.edu>
Date: Tue, 7 May 2024 23:11:27 -0400
Subject: [PATCH] Add to cart w/ context

---
 frontend/src/components/shared/cart/Cart.tsx  |  4 ++
 .../ui/buttons/AddToCart/AddToCart.tsx        | 47 +++++++++++--
 .../ui/cards/ProductCard/ProductCard.tsx      |  4 +-
 frontend/src/pages/product/ProductPage.tsx    | 18 +++--
 frontend/src/types/CartTypes.ts               | 34 +++++++--
 frontend/src/types/types.ts                   | 70 ++++++++++++-------
 6 files changed, 136 insertions(+), 41 deletions(-)

diff --git a/frontend/src/components/shared/cart/Cart.tsx b/frontend/src/components/shared/cart/Cart.tsx
index 337b18b..ad8d9c7 100644
--- a/frontend/src/components/shared/cart/Cart.tsx
+++ b/frontend/src/components/shared/cart/Cart.tsx
@@ -8,6 +8,10 @@ import { CartContext } from "../../../App";
 export default function Cart() {
   const [open, setOpen] = useState(false);
   const { cart, setCart } = useContext(CartContext);
+
+  React.useEffect(() => {
+    console.log(cart);
+  }, [cart]);
   return (
     <>
       <button
diff --git a/frontend/src/components/ui/buttons/AddToCart/AddToCart.tsx b/frontend/src/components/ui/buttons/AddToCart/AddToCart.tsx
index 001330b..ef65e4c 100644
--- a/frontend/src/components/ui/buttons/AddToCart/AddToCart.tsx
+++ b/frontend/src/components/ui/buttons/AddToCart/AddToCart.tsx
@@ -1,15 +1,50 @@
-import React from "react";
+import React, { useContext, useState } from "react";
 import { Button } from "@mui/material";
-import { styled } from '@mui/material/styles';
+import { styled } from "@mui/material/styles";
+import { ICartContext, ICartItem } from "../../../../types/CartTypes";
+import { CartContext } from "../../../../App";
+import { Product } from "../../../../types/types";
 
 interface IAddToCart {
-  route?: string;
+  product: Product;
+  size: string;
+  color: string;
 }
-export default function AddToCart({ route }: IAddToCart) {
+export default function AddToCart({ product, size, color }: IAddToCart) {
+  const { cart, setCart } = useContext(CartContext);
+
+  function addToCart() {
+    // Check if the cart already has the item
+    const existingCartItemIndex = cart.findIndex(
+      (item) =>
+        item.product.product_id === product.product_id &&
+        item.size === size &&
+        item.color === color
+    );
+
+    if (existingCartItemIndex >= 0) {
+      // If item exists, increase its quantity
+      const newCart = [...cart];
+      newCart[existingCartItemIndex].quantity += 1;
+      setCart(newCart);
+    } else {
+      // If item does not exist, add it to the cart
+      const newCartItem = {
+        product,
+        quantity: 1,
+        size,
+        color,
+      };
+      setCart([...cart, newCartItem]);
+    }
+  }
+
   return (
     <>
-      {route ? (
-        <Button variant="contained">Add To Cart</Button>
+      {product ? (
+        <Button variant="contained" onClick={addToCart}>
+          Add To Cart
+        </Button>
       ) : (
         <Button variant="contained">Out of Stock</Button>
       )}
diff --git a/frontend/src/components/ui/cards/ProductCard/ProductCard.tsx b/frontend/src/components/ui/cards/ProductCard/ProductCard.tsx
index c71de9e..c4b00aa 100644
--- a/frontend/src/components/ui/cards/ProductCard/ProductCard.tsx
+++ b/frontend/src/components/ui/cards/ProductCard/ProductCard.tsx
@@ -1,4 +1,4 @@
-import React from "react";
+import React, { useState, createContext } from "react";
 import Box from "@mui/material/Box";
 import Card from "@mui/material/Card";
 import CardActions from "@mui/material/CardActions";
@@ -9,6 +9,8 @@ import CardMedia from "@mui/material/CardMedia";
 import { CardActionArea } from "@mui/material";
 
 import { AWS_S3_BASE_URL } from "../../../../constants";
+
+
 export default function ProductCard({ product }: any) {
   return (
     <Card
diff --git a/frontend/src/pages/product/ProductPage.tsx b/frontend/src/pages/product/ProductPage.tsx
index 1c2cfe0..b9a747e 100644
--- a/frontend/src/pages/product/ProductPage.tsx
+++ b/frontend/src/pages/product/ProductPage.tsx
@@ -1,4 +1,4 @@
-import React from "react";
+import React, { useContext, useState } from "react";
 import { useParams } from "react-router-dom";
 
 import { getProductInfo } from "../../services/Product";
@@ -10,16 +10,24 @@ import { Typography } from "@mui/material";
 import { TextField, Button } from "@mui/material";
 import { AWS_S3_BASE_URL } from "../../constants";
 
+import { ICartContext, ICartItem } from "../../types/CartTypes";
+import { CartContext } from "../../App";
+import { Product } from "../../types/types";
+
 export default function ProductPage() {
   const { productID } = useParams();
-  const [productInfo, setProductInfo] = React.useState<any>();
-  const [loading, setLoading] = React.useState<boolean>(true);
+  const [productInfo, setProductInfo] = useState<Product>();
+  const [color, setColor] = useState<string>('red');
+  const [size, setSize] = useState<string>('sm');
+  const [loading, setLoading] = useState<boolean>(true);
+  const { cart, setCart } = useContext(CartContext);
 
   React.useEffect(() => {
     const fetchProduct = async () => {
       if (productID) {
         const response = await getProductInfo(productID);
-        setProductInfo(response);
+        const product = new Product(response);
+        setProductInfo(product);
         setLoading(false);
       }
     };
@@ -74,7 +82,7 @@ export default function ProductPage() {
               </div>
               <hr></hr>
               <div className="flex flex-col gap-2">
-                <AddToCart route={"/"} />
+                <AddToCart product={productInfo} size={size} color={color} />
               </div>
               <hr></hr>
               <div className="flex flex-col gap-2">
diff --git a/frontend/src/types/CartTypes.ts b/frontend/src/types/CartTypes.ts
index 44080ea..bbaed76 100644
--- a/frontend/src/types/CartTypes.ts
+++ b/frontend/src/types/CartTypes.ts
@@ -1,10 +1,15 @@
-import React, { createContext, useState, useContext, Dispatch, SetStateAction } from 'react';
-import { IProduct } from './types';
+import React, {
+  createContext,
+  useState,
+  useContext,
+  Dispatch,
+  SetStateAction,
+} from "react";
+import { Product } from "./types";
 
 // Define the type for the cart
 export interface ICartItem {
-  id: number;
-  product: IProduct;
+  product: Product;
   quantity: number;
   size: string;
   color: string;
@@ -14,4 +19,23 @@ export interface ICartItem {
 export interface ICartContext {
   cart: ICartItem[];
   setCart: Dispatch<SetStateAction<ICartItem[]>>;
-}
\ No newline at end of file
+}
+
+export class CartItem implements ICartItem {
+  product: Product;
+  quantity: number;
+  size: string;
+  color: string;
+
+  constructor(
+    product: Product,
+    quantity: number,
+    size: string,
+    color: string
+  ) {
+    this.product = product;
+    this.quantity = quantity;
+    this.size = size;
+    this.color = color;
+  }
+}
diff --git a/frontend/src/types/types.ts b/frontend/src/types/types.ts
index 701ccfd..3ea6d53 100644
--- a/frontend/src/types/types.ts
+++ b/frontend/src/types/types.ts
@@ -3,39 +3,61 @@ export interface IProduct {
   product_type: string;
   product_name: string;
   price: number;
-  description: string;
+  sale_price: number;
+  blurb: string;
   stock_quantity: number;
   image_url: string;
 }
 
-export interface IOrderItem {
-    id: number;
-    order_id: number;
-    product_id: number;
-    quantity: number;
+export class Product implements IProduct {
+  product_id: string;
+  product_type: string;
+  product_name: string;
+  sale_price: number;
+  price: number;
+  blurb: string;
+  stock_quantity: number;
+  image_url: string;
+
+  constructor(product: IProduct) {
+    this.product_id = product.product_id;
+    this.product_type = product.product_type;
+    this.product_name = product.product_name;
+    this.price = product.price;
+    this.sale_price = product.sale_price;
+    this.blurb = product.blurb;
+    this.stock_quantity = product.stock_quantity;
+    this.image_url = product.image_url;
   }
-  
+}
+
+export interface IOrderItem {
+  id: number;
+  order_id: number;
+  product_id: number;
+  quantity: number;
+}
+
 export interface IOrder {
-    id: number;
-    user_id: number;
-    created_at: Date;
-    items: IOrderItem[];
-  }
+  id: number;
+  user_id: number;
+  created_at: Date;
+  items: IOrderItem[];
+}
 
 export interface IReview {
-    id: number;
-    product_id: number;
-    author_id: number;
-    content: string;
-    rating: number;
+  id: number;
+  product_id: number;
+  author_id: number;
+  content: string;
+  rating: number;
 }
 
 export interface IUser {
-    id: number;
-    username: string;
-    email: string;
-    is_active?: boolean; // Optional
+  id: number;
+  username: string;
+  email: string;
+  is_active?: boolean; // Optional
 }
-  
-  
-export default {}
\ No newline at end of file
+
+export default {};