dkforest

commit 07c25a6b3e99347a124c8ca9165a5ca56030e09c
parent 01cf950a7d6f272a72c0da5cd2f3d300c291d0e3
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Mon,  1 Jan 2024 17:52:08 -0500

cleanup

Diffstat:
M
pkg/database/tableUsers.go
 | 
8
++++++++
M
pkg/web/handlers/api/v1/handlers.go
 | 
3
+--
M
pkg/web/handlers/streamModals/codeModal.go
 | 
3
+--

3 files changed, 10 insertions(+), 4 deletions(-)
diff --git a/pkg/database/tableUsers.go b/pkg/database/tableUsers.go
@@ -363,6 +363,14 @@ func (u *User) DoSave(db *DkfDB) {
 	}
 }
 
+func (u *User) SetSyntaxHighlightCode(db *DkfDB, syntaxHighlightCode string) {
+	db.db.Model(u).Select("SyntaxHighlightCode").Updates(User{SyntaxHighlightCode: syntaxHighlightCode})
+}
+
+func (u *User) DecrGeneralMessagesCount(db *DkfDB) {
+	db.db.Model(u).Select("GeneralMessagesCount").Updates(User{GeneralMessagesCount: u.GeneralMessagesCount - 1})
+}
+
 func (u *User) SetVerified(db *DkfDB, verified bool) {
 	db.db.Model(u).Select("Verified").Updates(User{Verified: verified})
 }
diff --git a/pkg/web/handlers/api/v1/handlers.go b/pkg/web/handlers/api/v1/handlers.go
@@ -371,8 +371,7 @@ func ChatDeleteMessageHandler(c echo.Context) error {
 	}
 
 	if msg.OwnMessage(authUser.ID) && msg.RoomID == config.GeneralRoomID && !msg.IsPm() {
-		authUser.GeneralMessagesCount--
-		authUser.DoSave(db)
+		authUser.DecrGeneralMessagesCount(db)
 	}
 
 	if err := msg.Delete(db); err != nil {
diff --git a/pkg/web/handlers/streamModals/codeModal.go b/pkg/web/handlers/streamModals/codeModal.go
@@ -148,8 +148,7 @@ func (_ CodeModal) InterceptMsg(cmd *command.Command) {
 	if !isValidLang(lang) {
 		lang = ""
 	}
-	cmd.AuthUser.SyntaxHighlightCode = lang
-	cmd.AuthUser.DoSave(cmd.DB)
+	cmd.AuthUser.SetSyntaxHighlightCode(cmd.DB, lang)
 
 	cmd.ModMsg = isMod
 	if pm != "" {