1 files changed, 114 insertions, 17 deletions

diff --git a/backend/src/handlers/cart_items.rs b/backend/src/handlers/cart_items.rs
index 25baaeb..e17f4c4 100644
--- a/backend/src/handlers/cart_items.rs
+++ b/backend/src/handlers/cart_items.rs
@@ -7,7 +7,7 @@ use actix_identity::Identity;
 use actix_web::{web, HttpResponse, Responder};
 use diesel::prelude::*;
 use log::{error, info};
-use serde::Deserialize;
+use serde::Serialize;
 
 pub async fn add_to_cart(
     cookie: Identity,
@@ -27,16 +27,41 @@ pub async fn add_to_cart(
         let new_cart_item = AddCartItem {
             cart_id: selected_user.id,
             product_id: item_details,
+            quantity: Some(1),
         };
         info!(
             "cart id: {:?}, product id {:?}",
             selected_user.id, item_details
         );
-        diesel::insert_into(cart_items)
-            .values((cart_id.eq(selected_user.id), product_id.eq(item_details)))
-            .execute(&conn)
-            .expect("Coundn't connect to DB");
-        HttpResponse::Ok().body("Inserted successfully!")
+        let current_entry = cart_items
+            .filter(cart_id.eq(selected_user.id))
+            .filter(product_id.eq(item_details))
+            .limit(1)
+            .first::<CartItem>(&conn);
+        match current_entry {
+            Ok(v) => {
+                info!("Item already present in cart, increasing quantity.");
+                let old_quantity = v.quantity.unwrap_or(1);
+                diesel::update(
+                    cart_items
+                        .filter(cart_id.eq(selected_user.id))
+                        .filter(product_id.eq(item_details)),
+                )
+                .set(quantity.eq(old_quantity + 1))
+                .execute(&conn)
+                .unwrap();
+                return HttpResponse::Ok()
+                    .body("Updated quantity successfully!");
+            }
+            Err(_) => {
+                info!("Item not present, adding to cart.");
+                diesel::insert_into(cart_items)
+                    .values(new_cart_item)
+                    .execute(&conn)
+                    .expect("Couldn't connect to DB");
+                HttpResponse::Ok().body("Inserted successfully!")
+            }
+        }
     } else {
         error!("Unauthorized add to cart action!");
         return HttpResponse::Unauthorized()
@@ -58,15 +83,44 @@ pub async fn remove_from_cart(
             .limit(1)
             .first::<Customer>(&conn)
             .expect("Couldn't connect to DB");
-
-        diesel::delete(
-            cart_items
-                .filter(cart_id.eq(selected_user.id))
-                .filter(product_id.eq(item_details)),
-        )
-        .execute(&conn)
-        .expect("Coundn't connect to DB");
-        HttpResponse::Ok().body("Removed successfully!")
+        let current_entry = cart_items
+            .filter(cart_id.eq(selected_user.id))
+            .filter(product_id.eq(item_details))
+            .limit(1)
+            .first::<CartItem>(&conn);
+        match current_entry {
+            Ok(v) => {
+                info!("Item already present in cart, increasing quantity.");
+                let old_quantity = v.quantity.unwrap_or(1);
+                if old_quantity == 1 {
+                    diesel::delete(
+                        cart_items
+                            .filter(cart_id.eq(selected_user.id))
+                            .filter(product_id.eq(item_details)),
+                    )
+                    .execute(&conn)
+                    .expect("Coundn't connect to DB");
+                } else {
+                    diesel::update(
+                        cart_items
+                            .filter(cart_id.eq(selected_user.id))
+                            .filter(product_id.eq(item_details)),
+                    )
+                    .set(quantity.eq(old_quantity - 1))
+                    .execute(&conn)
+                    .unwrap();
+                    return HttpResponse::Ok()
+                        .body("Updated quantity successfully!");
+                }
+                return HttpResponse::Ok()
+                    .body("Updated quantity successfully!");
+            }
+            Err(_) => {
+                info!("Item not present.");
+                return HttpResponse::InternalServerError()
+                    .body("Item not found!");
+            }
+        }
     } else {
         error!("Unauthorized add to cart action!");
         return HttpResponse::Unauthorized()
@@ -74,6 +128,12 @@ pub async fn remove_from_cart(
     }
 }
 
+#[derive(Serialize)]
+struct UserCartItem {
+    product_item: Product,
+    quantity: i32,
+}
+
 pub async fn get_user_cart_items(
     cookie: Identity,
     pool: web::Data<TPool>,
@@ -92,11 +152,15 @@ pub async fn get_user_cart_items(
         let cart_products = user_cart_items
             .into_iter()
             .map(|item| {
-                prod::product
+                let p = prod::product
                     .filter(prod::id.eq(item.product_id))
                     .limit(1)
                     .first::<Product>(&conn)
-                    .expect("Couldn't connect to db")
+                    .expect("Couldn't connect to db");
+                UserCartItem {
+                    product_item: p,
+                    quantity: item.quantity.unwrap_or(1),
+                }
             })
             .collect::<Vec<_>>();
         return HttpResponse::Ok().json(&cart_products);
@@ -105,3 +169,36 @@ pub async fn get_user_cart_items(
             .body("Need to be logged in to add to cart!");
     }
 }
+
+pub async fn get_user_cart_total(
+    cookie: Identity,
+    pool: web::Data<TPool>,
+) -> impl Responder {
+    let conn = pool.get().unwrap();
+    if let Some(uname) = cookie.identity() {
+        let selected_user = customer
+            .filter(username.eq(&uname))
+            .limit(1)
+            .first::<Customer>(&conn)
+            .expect("Couldn't connect to DB");
+        let user_cart_items = cart_items
+            .filter(cart_id.eq(selected_user.id))
+            .load::<CartItem>(&conn)
+            .expect("Couldn't connect to DB");
+        let cart_total: f32 = user_cart_items
+            .into_iter()
+            .map(|item| {
+                let p = prod::product
+                    .filter(prod::id.eq(item.product_id))
+                    .limit(1)
+                    .first::<Product>(&conn)
+                    .expect("Couldn't connect to db");
+                return p.price * item.quantity.unwrap_or(1) as f32;
+            })
+            .sum();
+        return HttpResponse::Ok().json(&cart_total);
+    } else {
+        return HttpResponse::Unauthorized()
+            .body("Need to be logged in to add to cart!");
+    }
+}