dkforest

commit b4f535e8a19fcf60c735b8d58007dd19f128a3a7
parent 73b729f210c91c7dc57aec05e8ffc7d836295819
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Wed,  4 Oct 2023 13:40:50 -0400

cleanup

Diffstat:
M
pkg/database/tableUsers.go
 | 
15
+++++++--------

1 file changed, 7 insertions(+), 8 deletions(-)
diff --git a/pkg/database/tableUsers.go b/pkg/database/tableUsers.go
@@ -297,18 +297,17 @@ func (u *User) DoSave(db *DkfDB) {
 }
 
 func (u *User) HellBan(db *DkfDB) {
-	u.IsHellbanned = true
-	u.DoSave(db)
-	if err := db.db.Model(&ChatMessage{}).Where("user_id = ?", u.ID).Update("is_hellbanned", true).Error; err != nil {
-		logrus.Error(err)
-	}
-	MsgPubSub.Pub(RefreshTopic, ChatMessageType{Typ: ForceRefresh})
+	u.setHellBan(db, true)
 }
 
 func (u *User) UnHellBan(db *DkfDB) {
-	u.IsHellbanned = false
+	u.setHellBan(db, false)
+}
+
+func (u *User) setHellBan(db *DkfDB, hb bool) {
+	u.IsHellbanned = hb
 	u.DoSave(db)
-	if err := db.db.Model(&ChatMessage{}).Where("user_id = ?", u.ID).Update("is_hellbanned", false).Error; err != nil {
+	if err := db.db.Model(&ChatMessage{}).Where("user_id = ?", u.ID).Update("is_hellbanned", hb).Error; err != nil {
 		logrus.Error(err)
 	}
 	MsgPubSub.Pub(RefreshTopic, ChatMessageType{Typ: ForceRefresh})