dkforest

commit c6263a7abc05918de50aa2ee4a961369386c430d
parent 3235610e176c94a0f544cc21e18dbc537fa12fb4
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Mon,  1 Jan 2024 22:28:55 -0500

cleanup

Diffstat:
M
pkg/database/tableUsers.go
 | 
16
++++++++++++++++
M
pkg/web/handlers/poker.go
 | 
6
++----
M
pkg/web/handlers/settings.go
 | 
6
++----

3 files changed, 20 insertions(+), 8 deletions(-)
diff --git a/pkg/database/tableUsers.go b/pkg/database/tableUsers.go
@@ -363,6 +363,22 @@ func (u *User) DoSave(db *DkfDB) {
 	}
 }
 
+func (u *User) SetAgePublicKey(db *DkfDB, agePublicKey string) {
+	db.db.Model(u).Select("AgePublicKey").Updates(User{AgePublicKey: agePublicKey})
+}
+
+func (u *User) SetApiKey(db *DkfDB, apiKey string) {
+	db.db.Model(u).Select("ApiKey").Updates(User{ApiKey: apiKey})
+}
+
+func (u *User) SetPokerReferralToken(db *DkfDB, pokerReferralToken *string) {
+	db.db.Model(u).Select("PokerReferralToken").Updates(User{PokerReferralToken: pokerReferralToken})
+}
+
+func (u *User) SetPokerReferredBy(db *DkfDB, pokerReferredBy *UserID) {
+	db.db.Model(u).Select("PokerReferredBy").Updates(User{PokerReferredBy: pokerReferredBy})
+}
+
 func (u *User) SetSignupMetadata(db *DkfDB, signupMetadata string) {
 	db.db.Model(u).Select("SignupMetadata").Updates(User{SignupMetadata: signupMetadata})
 }
diff --git a/pkg/web/handlers/poker.go b/pkg/web/handlers/poker.go
@@ -352,8 +352,7 @@ func PokerRakeBackHandler(c echo.Context) error {
 			logrus.Error(err)
 			return hutils.RedirectReferer(c)
 		}
-		authUser.PokerReferralToken = &token
-		authUser.DoSave(db)
+		authUser.SetPokerReferralToken(db, &token)
 		return hutils.RedirectReferer(c)
 
 	} else if formName == "set_referrer" {
@@ -379,8 +378,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 hutils.RedirectReferer(c)
 	}
 	return hutils.RedirectReferer(c)
diff --git a/pkg/web/handlers/settings.go b/pkg/web/handlers/settings.go
@@ -592,8 +592,7 @@ func SettingsAPIHandler(c echo.Context) error {
 		}
 		if formName == "confirm" {
 			token := utils.GenerateToken16()
-			authUser.ApiKey = token
-			authUser.DoSave(db)
+			authUser.SetApiKey(db, token)
 			return c.Redirect(http.StatusFound, "/settings/api")
 		}
 		data.NeedConfirm = true
@@ -900,8 +899,7 @@ func AddAgeHandler(c echo.Context) error {
 				return c.Render(http.StatusOK, "age_code", data)
 			}
 			ageTokenCache.Delete(authUser.ID)
-			authUser.AgePublicKey = token.PKey
-			authUser.DoSave(db)
+			authUser.SetAgePublicKey(db, token.PKey)
 			return c.Redirect(http.StatusFound, "/settings/age")
 		}
 	}