PR: Add Last Withdrawal, and Total Amount of Deposits Greater OR Equals Conditions

db/interface.go

@@ -126,6 +126,9 @@ type Database interface {
 	ChangeWorkspaceDeleteStatus(workspace_uuid string, status bool) Workspace
 	UpdateWorkspaceForDeletion(uuid string) error
 	ProcessDeleteWorkspace(workspace_uuid string) error
+	GetLastWithdrawal(workspace_uuid string) NewPaymentHistory
+	GetSumOfDeposits(workspace_uuid string) uint
+	GetSumOfWithdrawal(workspace_uuid string) uint
 	DeleteAllUsersFromWorkspace(uuid string) error
 	GetFilterStatusCount() FilterStattuCount
 	UserHasManageBountyRoles(pubKeyFromAuth string, uuid string) bool

db/workspaces.go

@@ -264,7 +264,7 @@ func (db database) GetWorkspaceBudgetHistory(workspace_uuid string) []BudgetHist
 	return budgetHistory
 }
 
-func (db database) ProcessUpdateBudget(invoice NewInvoiceList) error {
+func (db database) ProcessUpdateBudget(non_tx_invoice NewInvoiceList) error {
 	// Start db transaction
 	tx := db.db.Begin()
 
@@ -280,15 +280,25 @@ func (db database) ProcessUpdateBudget(invoice NewInvoiceList) error {
 		return err
 	}
 
-	created := invoice.Created
-	workspace_uuid := invoice.WorkspaceUuid
+	created := non_tx_invoice.Created
+	workspace_uuid := non_tx_invoice.WorkspaceUuid
+
+	invoice := NewInvoiceList{}
+	tx.Where("payment_request = ?", non_tx_invoice.PaymentRequest).Find(&invoice)
+
+	if invoice.Status {
+		tx.Rollback()
+		return errors.New("cannot process already paid invoice")
+	}
 
 	if workspace_uuid == "" {
 		return errors.New("cannot Create a Workspace Without a Workspace uuid")
 	}
 
 	// Get payment history and update budget
-	paymentHistory := db.GetPaymentHistoryByCreated(created, workspace_uuid)
+	paymentHistory := NewPaymentHistory{}
+	tx.Model(&NewPaymentHistory{}).Where("created = ?", created).Where("workspace_uuid = ? ", workspace_uuid).Find(&paymentHistory)
+
 	if paymentHistory.WorkspaceUuid != "" && paymentHistory.Amount != 0 {
 		paymentHistory.Status = true
 
@@ -298,9 +308,10 @@ func (db database) ProcessUpdateBudget(invoice NewInvoiceList) error {
 		}
 
 		// get Workspace budget and add payment to total budget
-		WorkspaceBudget := db.GetWorkspaceBudget(workspace_uuid)
+		workspaceBudget := NewBountyBudget{}
+		tx.Model(&NewBountyBudget{}).Where("workspace_uuid = ?", workspace_uuid).Find(&workspaceBudget)
 
-		if WorkspaceBudget.WorkspaceUuid == "" {
+		if workspaceBudget.WorkspaceUuid == "" {
 			now := time.Now()
 			workBudget := NewBountyBudget{
 				WorkspaceUuid: workspace_uuid,
@@ -313,11 +324,11 @@ func (db database) ProcessUpdateBudget(invoice NewInvoiceList) error {
 				tx.Rollback()
 			}
 		} else {
-			totalBudget := WorkspaceBudget.TotalBudget
-			WorkspaceBudget.TotalBudget = totalBudget + paymentHistory.Amount
+			totalBudget := workspaceBudget.TotalBudget
+			workspaceBudget.TotalBudget = totalBudget + paymentHistory.Amount
 
-			if err = tx.Model(&NewBountyBudget{}).Where("workspace_uuid = ?", WorkspaceBudget.WorkspaceUuid).Updates(map[string]interface{}{
-				"total_budget": WorkspaceBudget.TotalBudget,
+			if err = tx.Model(&NewBountyBudget{}).Where("workspace_uuid = ?", workspaceBudget.WorkspaceUuid).Updates(map[string]interface{}{
+				"total_budget": workspaceBudget.TotalBudget,
 			}).Error; err != nil {
 				tx.Rollback()
 			}
@@ -333,34 +344,51 @@ func (db database) ProcessUpdateBudget(invoice NewInvoiceList) error {
 }
 
 func (db database) AddAndUpdateBudget(invoice NewInvoiceList) NewPaymentHistory {
+	// Start db transaction
+	tx := db.db.Begin()
+
 	created := invoice.Created
 	workspace_uuid := invoice.WorkspaceUuid
 
-	paymentHistory := db.GetPaymentHistoryByCreated(created, workspace_uuid)
+	paymentHistory := NewPaymentHistory{}
+	tx.Model(&NewPaymentHistory{}).Where("created = ?", created).Where("workspace_uuid = ? ", workspace_uuid).Find(&paymentHistory)
 
 	if paymentHistory.WorkspaceUuid != "" && paymentHistory.Amount != 0 {
 		paymentHistory.Status = true
 		db.db.Where("created = ?", created).Where("workspace_uuid = ? ", workspace_uuid).Updates(paymentHistory)
 
 		// get Workspace budget and add payment to total budget
-		WorkspaceBudget := db.GetWorkspaceBudget(workspace_uuid)
+		workspaceBudget := NewBountyBudget{}
+		tx.Model(&NewBountyBudget{}).Where("workspace_uuid = ?", workspace_uuid).Find(&workspaceBudget)
 
-		if WorkspaceBudget.WorkspaceUuid == "" {
+		if workspaceBudget.WorkspaceUuid == "" {
 			now := time.Now()
 			workBudget := NewBountyBudget{
 				WorkspaceUuid: workspace_uuid,
 				TotalBudget:   paymentHistory.Amount,
 				Created:       &now,
 				Updated:       &now,
 			}
-			db.CreateWorkspaceBudget(workBudget)
+
+			if err := tx.Create(&workBudget).Error; err != nil {
+				tx.Rollback()
+			}
 		} else {
-			totalBudget := WorkspaceBudget.TotalBudget
-			WorkspaceBudget.TotalBudget = totalBudget + paymentHistory.Amount
-			db.UpdateWorkspaceBudget(WorkspaceBudget)
+			totalBudget := workspaceBudget.TotalBudget
+			workspaceBudget.TotalBudget = totalBudget + paymentHistory.Amount
+
+			if err := tx.Model(&NewBountyBudget{}).Where("workspace_uuid = ?", workspaceBudget.WorkspaceUuid).Updates(map[string]interface{}{
+				"total_budget": workspaceBudget.TotalBudget,
+			}).Error; err != nil {
+				tx.Rollback()
+			}
 		}
+	} else {
+		tx.Rollback()
 	}
 
+	tx.Commit()
+
 	return paymentHistory
 }
 
@@ -487,7 +515,7 @@ func (db database) GetPaymentHistory(workspace_uuid string, r *http.Request) []N
 func (db database) GetPendingPaymentHistory() []NewPaymentHistory {
 	paymentHistories := []NewPaymentHistory{}
 
-	query := `SELECT * FROM payment_histories WHERE payment_status = '` + PaymentPending + `' AND status = true ORDER BY created DESC`
+	query := `SELECT * FROM payment_histories WHERE payment_status = '` + PaymentPending + `' AND status = true AND payment_type = 'payment' ORDER BY created DESC`
 
 	db.db.Raw(query).Find(&paymentHistories)
 	return paymentHistories
@@ -612,6 +640,26 @@ func (db database) DeleteAllUsersFromWorkspace(workspace_uuid string) error {
 	return nil
 }
 
+func (db database) GetLastWithdrawal(workspace_uuid string) NewPaymentHistory {
+	p := NewPaymentHistory{}
+	db.db.Model(&NewPaymentHistory{}).Where("workspace_uuid", workspace_uuid).Where("payment_type", "withdraw").Order("created DESC").Limit(1).Find(&p)
+	return p
+}
+
+func (db database) GetSumOfDeposits(workspace_uuid string) uint {
+	var depositAmount uint
+	db.db.Model(&NewPaymentHistory{}).Where("workspace_uuid = ?", workspace_uuid).Where("status = ?", true).Where("payment_type = ?", "deposit").Select("SUM(amount)").Row().Scan(&depositAmount)
+
+	return depositAmount
+}
+
+func (db database) GetSumOfWithdrawal(workspace_uuid string) uint {
+	var depositAmount uint
+	db.db.Model(&NewPaymentHistory{}).Where("workspace_uuid = ?", workspace_uuid).Where("status = ?", true).Where("payment_type = ?", "withdraw").Select("SUM(amount)").Row().Scan(&depositAmount)
+
+	return depositAmount
+}
+
 func (db database) GetFeaturePhasesBountiesCount(bountyType string, phaseUuid string) int64 {
 	var count int64