3pp-mirror/neko
1
0
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2025-05-15 18:17:42 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix locked room bug

Browse source
This commit is contained in:
m1k1o 2021-01-14 21:41:00 +01:00
parent 9e5019be7d
commit f093ef762b
3 changed files with 9 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

15
server/internal/websocket/handler.go
View file

@ -22,7 +22,7 @@ type MessageHandler struct {
locked bool
}
func (h *MessageHandler) Connected(id string, socket *WebSocket) (bool, string, error) {
func (h *MessageHandler) Connected(admin bool, socket *WebSocket) (bool, string, error) {
address := socket.Address()
if address == "" {
h.logger.Debug().Msg("no remote address")
@ -34,22 +34,15 @@ func (h *MessageHandler) Connected(id string, socket *WebSocket) (bool, string,
}
}
if h.locked {
session, ok := h.sessions.Get(id)
if !ok || !session.Admin() {
h.logger.Debug().Msg("server locked")
return false, "locked", nil
}
if h.locked && !admin {
h.logger.Debug().Msg("server locked")
return false, "locked", nil
}
return true, "", nil
}
func (h *MessageHandler) Disconnected(id string) error {
if h.locked && len(h.sessions.Admins()) == 0 {
h.locked = false
}
return h.sessions.Destroy(id)
}