update cart func and add status for table

Bahamin1 · Jun 21, 2024 · 0986962 · 0986962
1 parent ef80e7a
commit 0986962
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 5 deletions.
diff --git a/src/Dorder/Table.mo b/src/Dorder/Table.mo
@@ -21,7 +21,14 @@ module {
         reserveTime : ?Time.Time;
         userWantsToJoin : [Principal];
         seatedCustomers : [Principal];
-        order : ?Cart.Order;
+        status : TableStatus;
+        order : [Cart.CartItem];
+    };
+
+    public type TableStatus = {
+        #Open;
+        #OnOrder;
+        #Finalized;
     };
 
     public type TableMap = Map.Map<Nat, Table>;
@@ -45,9 +52,10 @@ module {
             capacity = capacity;
             reservedBy = null;
             reserveTime = null;
+            status = #Open;
             userWantsToJoin = [];
             seatedCustomers = [];
-            order = null;
+            order = [];
 
         };
 
@@ -72,9 +80,10 @@ module {
                             capacity = table.capacity;
                             reservedBy = ?reservedBy;
                             reserveTime = ?Time.now();
+                            status = #OnOrder;
                             userWantsToJoin = [];
                             seatedCustomers = [];
-                            order = null;
+                            order = [];
                         };
                         put(tables, tableId, updatedTable);
                         return #ok(tables);
@@ -102,9 +111,10 @@ module {
                             capacity = table.capacity;
                             reservedBy = null;
                             reserveTime = null;
+                            status = #Open;
                             userWantsToJoin = [];
                             seatedCustomers = [];
-                            order = null;
+                            order = [];
 
                         };
                         put(tables, tableId, updatedTable);

diff --git a/src/Dorder/main.mo b/src/Dorder/main.mo
@@ -437,9 +437,39 @@ shared ({ caller = manager }) actor class Dorder() = this {
       createdAt = Time.now();
     };
     Map.set(cartMap, phash, caller, newOrder);
-    return newOrder;
+    return #ok(newOrder);
 
   };
+
+  public shared ({ caller }) func addOrderToTable(items : [Menu.MenuItem], tableId : Nat) : async Result.Result<Text, Text> {
+    switch (Table.get(tableMap, tableId)) {
+      case (?table) {
+        if (table.status == #Finalized) {
+          return #err("Cannot add order. The table has finalized orders.");
+        };
+
+        let newOrder = {
+          id = Time.now();
+          items = items;
+          totalPrice = items.foldLeft<Nat>(0, func(acc, item) { acc + item.price });
+          tableId = ?tableId;
+          orderType = #OnTable;
+          orderStatus = #Pending;
+          orderedBy = caller;
+          orderedAt = Time.now();
+          finalized = false;
+        };
+
+        table.orders.add(newOrder);
+        Table.put(tableMap, tableId, table);
+
+        return #ok("Order added to table successfully.");
+      };
+      case (null) {
+        return #err("Table not found.");
+      };
+    };
+  };
 };
 
 //   public shared ({ caller }) func addToCart(menuId : Nat, quantity : Nat) : async Result.Result<Cart.CartMap, Text> {