quantity

cart-grain.go

@@ -29,6 +29,7 @@ type CartItem struct {
 	Sku      string `json:"sku"`
 	Name     string `json:"name"`
 	Price    int64  `json:"price"`
+	Quantity int    `json:"qty"`
 	Image    string `json:"image"`
 }
 
@@ -64,7 +65,7 @@ func (c *CartGrain) GetLastChange() int64 {
 	return *c.storageMessages[len(c.storageMessages)-1].TimeStamp
 }
 
-func getItemData(sku string) (*messages.AddItem, error) {
+func getItemData(sku string, qty int) (*messages.AddItem, error) {
 	item, err := FetchItem(sku)
 	if err != nil {
 		return nil, err
@@ -87,7 +88,7 @@ func getItemData(sku string) (*messages.AddItem, error) {
 	}
 
 	return &messages.AddItem{
-		Quantity: 1,
+		Quantity: int32(qty),
 		Price:    int64(price),
 		Sku:      sku,
 		Name:     item.Title,
@@ -95,8 +96,8 @@ func getItemData(sku string) (*messages.AddItem, error) {
 	}, nil
 }
 
-func (c *CartGrain) AddItem(sku string) ([]byte, error) {
+func (c *CartGrain) AddItem(sku string, qty int) ([]byte, error) {
-	cartItem, err := getItemData(sku)
+	cartItem, err := getItemData(sku, qty)
 	if err != nil {
 		return nil, err
 	}
@@ -130,21 +131,36 @@ func (c *CartGrain) HandleMessage(message *Message, isReplay bool) ([]byte, erro
 		if !ok {
 			err = fmt.Errorf("expected AddRequest")
 		} else {
-			return c.AddItem(msg.Sku)
+			return c.AddItem(msg.Sku, 1) // extent AddRequest to include quantity
 		}
 	case AddItemType:
 		msg, ok := message.Content.(*messages.AddItem)
 		if !ok {
 			err = fmt.Errorf("expected AddItem")
 		} else {
+			found := false
+			if msg.Quantity < 1 {
+				return nil, fmt.Errorf("invalid quantity")
+			}
+			for _, item := range c.Items {
+				if item.Sku == msg.Sku {
+					found = true
+					item.Quantity += int(msg.Quantity)
+					c.TotalPrice += item.Price * int64(msg.Quantity)
+					break
+				}
+			}
+			if !found {
 				c.lastItemId++
 				c.Items = append(c.Items, CartItem{
 					Id: c.lastItemId,
+
 					Sku:   msg.Sku,
 					Name:  msg.Name,
 					Price: msg.Price,
 					Image: msg.Image,
 				})
+			}
 			c.TotalPrice += msg.Price * int64(msg.Quantity)
 		}
 	case RemoveItemType: