dkforest

commit ebe25fc36639a8d5490d2a6880dc13b2ebd88cf7
parent 4a321fdce2d89db39809beba28b0e46d770c9291
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Mon,  1 Jan 2024 17:43:43 -0500

cleanup

Diffstat:
M
pkg/database/tableUsers.go
 | 
4
++++
M
pkg/web/handlers/forum.go
 | 
9
+++------

2 files changed, 7 insertions(+), 6 deletions(-)
diff --git a/pkg/database/tableUsers.go b/pkg/database/tableUsers.go
@@ -363,6 +363,10 @@ func (u *User) DoSave(db *DkfDB) {
 	}
 }
 
+func (u *User) SetCanUseForum(db *DkfDB, canUseForum bool) {
+	db.db.Model(u).Select("CanUseForum").Updates(User{CanUseForum: canUseForum})
+}
+
 func (u *User) ResetLoginAttempts(db *DkfDB) {
 	db.db.Model(u).Select("LoginAttempts").Updates(User{LoginAttempts: 0})
 }
diff --git a/pkg/web/handlers/forum.go b/pkg/web/handlers/forum.go
@@ -199,8 +199,7 @@ func ThreadReplyHandler(c echo.Context) error {
 		}
 		if isForumSpam(data.Message) {
 			db.NewAudit(*authUser, fmt.Sprintf("spam forum thread reply %s (#%d)", authUser.Username, authUser.ID))
-			authUser.CanUseForum = false
-			authUser.DoSave(db)
+			authUser.SetCanUseForum(db, false)
 			return c.Redirect(http.StatusFound, "/")
 		}
 		message := database.MakeForumMessage(data.Message, authUser.ID, thread.ID)
@@ -271,8 +270,7 @@ func ThreadEditMessageHandler(c echo.Context) error {
 		}
 		if isForumSpam(data.Message) {
 			db.NewAudit(*authUser, fmt.Sprintf("spam forum edit msg %s (#%d)", authUser.Username, authUser.ID))
-			authUser.CanUseForum = false
-			authUser.DoSave(db)
+			authUser.SetCanUseForum(db, false)
 			return c.Redirect(http.StatusFound, "/")
 		}
 		msg.Message = data.Message
@@ -357,8 +355,7 @@ func NewThreadHandler(c echo.Context) error {
 		}
 		if isForumSpam(data.Message) {
 			db.NewAudit(*authUser, fmt.Sprintf("spam forum new thread %s (#%d)", authUser.Username, authUser.ID))
-			authUser.CanUseForum = false
-			authUser.DoSave(db)
+			authUser.SetCanUseForum(db, false)
 			return c.Redirect(http.StatusFound, "/")
 		}
 		thread := database.MakeForumThread(data.ThreadName, authUser.ID, 1)