dkforest

commit 77696c79483f9166d27d7b52cf909439bde94648
parent c5c7f4ca53a8cf18958172edcf4eeadcc4d8d50c
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Sat, 30 Dec 2023 16:59:05 -0500

cleanup

Diffstat:
M
pkg/database/tableUsers.go
 | 
5
+++++
M
pkg/web/handlers/poker.go
 | 
3
+--

2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/pkg/database/tableUsers.go b/pkg/database/tableUsers.go
@@ -362,6 +362,11 @@ func (u *User) DoSave(db *DkfDB) {
 	}
 }
 
+func (u *User) SetPokerReferredBy(db *DkfDB, userID *UserID) {
+	u.PokerReferredBy = userID
+	db.db.Exec(`UPDATE users SET poker_referred_by = ? WHERE id = ?`, u.PokerReferredBy, u.ID)
+}
+
 func (u *User) SetPmMode(db *DkfDB, pmMode int64) {
 	u.PmMode = pmMode
 	db.db.Exec(`UPDATE users SET pm_mode = ? WHERE id = ?`, u.PmMode, u.ID)
diff --git a/pkg/web/handlers/poker.go b/pkg/web/handlers/poker.go
@@ -381,8 +381,7 @@ func PokerRakeBackHandler(c echo.Context) error {
 			data.SetReferralError = "Yon can't give yourself the rake back"
 			return c.Render(http.StatusOK, "poker-rake-back", data)
 		}
-		authUser.PokerReferredBy = &referrer.ID
-		authUser.DoSave(db)
+		authUser.SetPokerReferredBy(db, &referrer.ID)
 		return c.Redirect(http.StatusFound, c.Request().Referer())
 	}
 	return c.Redirect(http.StatusFound, c.Request().Referer())