dkforest

commit c5c7f4ca53a8cf18958172edcf4eeadcc4d8d50c
parent e42ebdeb70cce872d559641424e0d5271e9bad86
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Sat, 30 Dec 2023 16:32:19 -0500

cleanup

Diffstat:
M
pkg/database/tableUsers.go
 | 
5
+++++
M
pkg/web/handlers/interceptors/slashInterceptor.go
 | 
6
++----

2 files changed, 7 insertions(+), 4 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) SetPmMode(db *DkfDB, pmMode int64) {
+	u.PmMode = pmMode
+	db.db.Exec(`UPDATE users SET pm_mode = ? WHERE id = ?`, u.PmMode, u.ID)
+}
+
 func (u *User) ToggleDisplayHellbanned(db *DkfDB) {
 	u.DisplayHellbanned = !u.DisplayHellbanned
 	db.db.Exec(`UPDATE users SET display_hellbanned = ? WHERE id = ?`, u.DisplayHellbanned, u.ID)
diff --git a/pkg/web/handlers/interceptors/slashInterceptor.go b/pkg/web/handlers/interceptors/slashInterceptor.go
@@ -1197,16 +1197,14 @@ func handleListPmWhitelistCmd(c *command.Command) (handled bool) {
 
 func handleSetPmModeCmd(c *command.Command) (handled bool) {
 	if c.Message == "/setpmmode whitelist" {
-		c.AuthUser.PmMode = database.PmModeWhitelist
-		c.AuthUser.DoSave(c.DB)
+		c.AuthUser.SetPmMode(c.DB, database.PmModeWhitelist)
 		msg := `pm mode set to "whitelist"`
 		c.ZeroProcMsg(msg)
 		c.Err = command.ErrRedirect
 		return true
 
 	} else if c.Message == "/setpmmode standard" {
-		c.AuthUser.PmMode = database.PmModeStandard
-		c.AuthUser.DoSave(c.DB)
+		c.AuthUser.SetPmMode(c.DB, database.PmModeStandard)
 		msg := `pm mode set to "standard"`
 		c.ZeroProcMsg(msg)
 		c.Err = command.ErrRedirect