dkforest

commit 838bdc36643ef1a7203e6b23ba17e6e1c9c3ff38
parent b6e50f310216b0e55e24a62954e06add87c21730
Author: n0tr1v <n0tr1v@protonmail.com>
Date:   Fri, 30 Dec 2022 20:07:56 -0800

clear groups on reset

Diffstat:
M
pkg/database/tableChatRoomGroups.go
 | 
5
+++++
M
pkg/web/handlers/api/v1/werewolf.go
 | 
3
+++

2 files changed, 8 insertions(+), 0 deletions(-)
diff --git a/pkg/database/tableChatRoomGroups.go b/pkg/database/tableChatRoomGroups.go
@@ -72,6 +72,11 @@ func RmUserFromRoomGroup(roomID RoomID, groupID GroupID, userID UserID) (err err
 	return
 }
 
+func ClearRoomGroup(roomID RoomID, groupID GroupID) (err error) {
+	err = DB.Delete(&ChatRoomUserGroup{}, "group_id = ? AND room_id = ?", groupID, roomID).Error
+	return
+}
+
 func GetRoomGroups(roomID RoomID) (out []ChatRoomGroup, err error) {
 	err = DB.Find(&out, "room_id = ?", roomID).Error
 	return
diff --git a/pkg/web/handlers/api/v1/werewolf.go b/pkg/web/handlers/api/v1/werewolf.go
@@ -674,6 +674,9 @@ func (b *Werewolf) reset() {
 	b.healerCh = make(chan string)
 	b.votesCh = make(chan string)
 	b.readyCh = make(chan bool)
+	_ = database.ClearRoomGroup(b.roomID, b.werewolfGroupID)
+	_ = database.ClearRoomGroup(b.roomID, b.spectatorGroupID)
+	_ = database.ClearRoomGroup(b.roomID, b.deadGroupID)
 }
 
 func NewWerewolf() *Werewolf {