chore: build

backend/gestion/db/db_parrain.go

@@ -1,8 +1,11 @@
 package db
 
 import (
+	"database/sql"
 	"fmt"
 	"gestion/models"
+
+	"gorm.io/gorm"
 )
 
 func (d *Database) SetClientParrain(clientUsername, parrainUsername string) error {
@@ -18,13 +21,44 @@ func (d *Database) SetClientParrain(clientUsername, parrainUsername string) erro
 	return nil
 }
 
+// SetClientParrainAndCredit assigne un parrain à un client et crédite le parrain
+// dans une seule transaction, pour éviter un lien parrain enregistré sans le crédit associé.
+func (d *Database) SetClientParrainAndCredit(clientUsername, parrainUsername string, creditAmount float64) error {
+	return d.GDB.Transaction(func(tx *gorm.DB) error {
+		result := tx.Model(&models.Client{}).
+			Where("username = ? AND (parrain IS NULL OR parrain = '')", clientUsername).
+			Update("parrain", parrainUsername)
+		if result.Error != nil {
+			return result.Error
+		}
+		if result.RowsAffected == 0 {
+			return fmt.Errorf("client introuvable ou parrain déjà défini")
+		}
+
+		if creditAmount > 0 {
+			result = tx.Model(&models.Client{}).Where("username = ?", parrainUsername).
+				Updates(map[string]any{"referral_balance": gorm.Expr("referral_balance + ?", creditAmount)})
+			if result.Error != nil {
+				return result.Error
+			}
+			if result.RowsAffected == 0 {
+				return fmt.Errorf("parrain non trouvé")
+			}
+		}
+		return nil
+	})
+}
+
 func (d *Database) GetClientParrain(clientUsername string) (string, error) {
-	var parrain string
+	var parrain sql.NullString
 	err := d.GDB.Table("clients").
 		Select("parrain").
 		Where("username = ?", clientUsername).
 		Scan(&parrain).Error
-	return parrain, err
+	if err != nil {
+		return "", err
+	}
+	return parrain.String, nil
 }
 
 func (d *Database) GetClientsByParrain(parrainUsername string) ([]models.Client, error) {