working qty clamping

Author: maikirakiwi

main.go

@@ -89,6 +89,10 @@ func checkoutWebhookHandler(event stripe.Event) bool {
 			}
 			slog.Info("Link " + link.LinkID + " now used " + fmt.Sprintf("%d", link.Used) + " times")
 			db_res = DB.Conn.Save(&link)
+			if db_res.Error != nil {
+				slog.Error("Error while saving link paid event to database: " + db_res.Error.Error())
+				return false
+			}
 
 			if len(link.TrackingInventoryIDs) > 0 {
 				// Expand line_items field
@@ -106,11 +110,6 @@ func checkoutWebhookHandler(event stripe.Event) bool {
 					slog.Info(fmt.Sprintf("Inventory item %s was bought %d times in link %s", trackingItem, boughtItems[trackingItem], link.LinkID))
 				}
 
-				if db_res.Error != nil {
-					slog.Error("Error while saving link paid event to database: " + db_res.Error.Error())
-					return false
-				}
-
 			}
 
 		}

menu/createNewLink.go

@@ -63,6 +63,14 @@ func CreateNewLink() {
 	}
 
 	li, trackedItems := addItems(*settings)
+
+	for _, trackedItem := range trackedItems {
+		tracking := &models.Inventory{}
+		DB.Conn.Where(&models.Inventory{Product: trackedItem}).First(&tracking)
+		tracking.SessionLinkIDs = append(tracking.SessionLinkIDs, linkID)
+		DB.Conn.Save(&tracking)
+	}
+
 	params.LineItems = li
 	params.Mode = stripe.String("payment")
 	params.SuccessURL = stripe.String("http://" + settings.Domain + ":4242/success")

menu/inventory.go

@@ -180,11 +180,11 @@ func UpdateInventory(product string, decrementQuantity int64) {
 		slog.Error("Error while searching database for product " + product + err.Error())
 		return
 	}
-	inventory.Quantity -= decrementQuantity
+	inventory.Quantity = inventory.Quantity - decrementQuantity
 	if inventory.Quantity > 0 {
-		go ClampAdjustableQty(inventory.SessionLinkIDs, product, inventory.Quantity)
+		ClampAdjustableQty(inventory.SessionLinkIDs, product, inventory.Quantity)
 	} else {
-		go RemoveItemFromLink(inventory.SessionLinkIDs, product)
+		RemoveItemFromLink(inventory.SessionLinkIDs, product)
 	}
 
 	DB.Conn.Save(&inventory)
@@ -199,12 +199,30 @@ func ClampAdjustableQty(links []string, product string, qty int64) {
 			return
 		}
 		for _, li := range seslink.Params.LineItems {
-			if *li.PriceData.Product == product && *li.AdjustableQuantity.Maximum > qty {
-				li.AdjustableQuantity.Maximum = stripe.Int64(qty)
+			price := &models.Price{}
+			err := DB.Conn.Where(&models.Price{PriceID: *li.Price}).First(price).Error
+			if err != nil {
+				slog.Error("Error while getting price from DB: " + err.Error())
+				return
+			}
+			if price.Product == product {
+				if li.AdjustableQuantity != nil {
+					if *li.AdjustableQuantity.Maximum > qty {
+						li.AdjustableQuantity.Maximum = stripe.Int64(qty)
+					}
+					if *li.AdjustableQuantity.Minimum == *li.AdjustableQuantity.Maximum {
+						li.AdjustableQuantity = nil
+					}
+				}
+
+				if *li.Quantity > qty {
+					li.Quantity = stripe.Int64(qty)
+				}
 				break
 			}
 
 		}
+
 		DB.Conn.Save(&seslink)
 	}
 }
@@ -219,13 +237,19 @@ func RemoveItemFromLink(links []string, product string) {
 		}
 		if len(seslink.Params.LineItems) == 1 {
 			seslink.Active = false
+			DB.Conn.Save(&seslink)
 			continue
 		}
 		newLineItems := []*stripe.CheckoutSessionLineItemParams{}
 		for _, li := range seslink.Params.LineItems {
-			if *li.PriceData.Product != product {
+			price := &models.Price{}
+			err := DB.Conn.Where(&models.Price{PriceID: *li.Price}).First(price).Error
+			if err != nil {
+				slog.Error("Error while getting price from DB: " + err.Error())
+				return
+			}
+			if price.Product != product {
 				newLineItems = append(newLineItems, li)
-				break
 			}
 
 		}

menu/modifyExistingLink.go

@@ -74,11 +74,9 @@ func ModifyExistingLink() {
 	case "Change List Item Price":
 		prompt := promptui.Select{
 			Label: "Change Item Price",
-			Items: stripeapi.GetProductsInLinkReadable(allLinks[selection].LinkID),
+			Items: stripeapi.GetProductsInLink(allLinks[selection].LinkID),
 		}
 		itemSelection, _, _ := prompt.Run()
-		println("Selected: " + *allLinks[selection].Params.LineItems[itemSelection].PriceData.ProductData.Name)
-		println("Current Price: " + strconv.Itoa(int(*allLinks[selection].Params.LineItems[itemSelection].PriceData.UnitAmount)))
 		pricePrompt := promptui.Prompt{
 			Label: "Price in cents. (e.g. $1.99 = 199)",
 		}
@@ -104,12 +102,15 @@ func ModifyExistingLink() {
 			println("Error: %v\n", err)
 			return
 		}
-		newPrice, err := stripeapi.NewPriceIfNotExist(settings.DefaultCurrency, priceInt, *allLinks[selection].Params.LineItems[itemSelection].PriceData.Product)
+		price := &models.Price{}
+		DB.Conn.Where(&models.Price{PriceID: *allLinks[selection].Params.LineItems[itemSelection].Price}).First(price)
+		newPrice, err := stripeapi.NewPriceIfNotExist(settings.DefaultCurrency, priceInt, price.Product)
 		if err != nil {
 			println("Error: %v\n", err)
 			return
 		}
 		*allLinks[selection].Params.LineItems[itemSelection].Price = newPrice
+		DB.Conn.Save(&allLinks[selection])
 	case "Change Max Uses":
 		prompt := promptui.Prompt{
 			Label: "Change Max Uses",

models/models.go

@@ -93,7 +93,7 @@ type DropdownOption struct {
 
 type InvoicePDF struct {
 	gorm.Model       `json:"-"`
-	Nickname         string
+	Nickname         string `gorm:"unique"`
 	TaxID            string
 	CustomFieldName  string
 	CustomFieldValue string
@@ -104,7 +104,7 @@ type InvoicePDF struct {
 type Inventory struct {
 	gorm.Model
 	DisplayName    string
-	Product        string
+	Product        string `gorm:"unique"`
 	Quantity       int64
 	SessionLinkIDs IDs `gorm:"embedded"`
 }

stripe-api/stripe-api.go

@@ -50,19 +50,6 @@ func GetAllProduct(active bool) ([]*stripe.Product, error) {
 	return products, nil
 }
 
-func GetProductsInLinkReadable(LinkID string) []string {
-	link := models.SessionLink{}
-	err := DB.Conn.Where(&models.SessionLink{LinkID: LinkID}).First(&link)
-	if err.Error != nil {
-		return nil
-	}
-	var products []string
-	for _, li := range link.Params.LineItems {
-		products = append(products, *li.PriceData.ProductData.Name)
-	}
-	return products
-}
-
 func GetProductsInLink(LinkID string) []string {
 	link := models.SessionLink{}
 	err := DB.Conn.Where(&models.SessionLink{LinkID: LinkID}).First(&link)
@@ -71,7 +58,9 @@ func GetProductsInLink(LinkID string) []string {
 	}
 	var products []string
 	for _, li := range link.Params.LineItems {
-		products = append(products, *li.PriceData.Product)
+		price := &models.Price{}
+		DB.Conn.Where(&models.Price{PriceID: *li.Price}).First(price)
+		products = append(products, price.Product)
 	}
 	return products
 }