neko/internal/websocket/handler/admin.go

149 lines
2.9 KiB
Go
Raw Normal View History

2020-11-02 04:09:48 +13:00
package handler
import (
2020-10-29 07:15:48 +13:00
"demodesk/neko/internal/types"
"demodesk/neko/internal/types/event"
"demodesk/neko/internal/types/message"
)
2020-11-02 04:09:48 +13:00
func (h *MessageHandlerCtx) adminLock(session types.Session) error {
if !session.Admin() {
h.logger.Debug().Msg("user not admin")
return nil
}
if h.locked {
h.logger.Debug().Msg("server already locked...")
return nil
}
h.locked = true
2020-11-17 06:12:25 +13:00
return h.sessions.Broadcast(
message.Admin{
Event: event.ADMIN_LOCK,
ID: session.ID(),
2020-11-17 06:12:25 +13:00
}, nil)
}
2020-11-02 04:09:48 +13:00
func (h *MessageHandlerCtx) adminUnlock(session types.Session) error {
if !session.Admin() {
h.logger.Debug().Msg("user not admin")
return nil
}
if !h.locked {
h.logger.Debug().Msg("server not locked...")
return nil
}
h.locked = false
2020-11-17 06:12:25 +13:00
return h.sessions.Broadcast(
message.Admin{
Event: event.ADMIN_UNLOCK,
ID: session.ID(),
2020-11-17 06:12:25 +13:00
}, nil)
}
2020-11-02 04:09:48 +13:00
func (h *MessageHandlerCtx) adminControl(session types.Session) error {
if !session.Admin() {
h.logger.Debug().Msg("user not admin")
return nil
}
host := h.sessions.GetHost()
h.sessions.SetHost(session)
if host != nil {
2020-11-17 06:12:25 +13:00
return h.sessions.Broadcast(
message.AdminTarget{
Event: event.ADMIN_CONTROL,
ID: session.ID(),
Target: host.ID(),
2020-11-17 06:12:25 +13:00
}, nil)
}
2020-11-17 06:12:25 +13:00
return h.sessions.Broadcast(
message.Admin{
Event: event.ADMIN_CONTROL,
ID: session.ID(),
}, nil)
}
2020-11-02 04:09:48 +13:00
func (h *MessageHandlerCtx) adminRelease(session types.Session) error {
if !session.Admin() {
h.logger.Debug().Msg("user not admin")
return nil
}
host := h.sessions.GetHost()
h.sessions.ClearHost()
if host != nil {
2020-11-17 06:12:25 +13:00
return h.sessions.Broadcast(
message.AdminTarget{
Event: event.ADMIN_RELEASE,
ID: session.ID(),
Target: host.ID(),
2020-11-17 06:12:25 +13:00
}, nil)
}
2020-11-17 06:12:25 +13:00
return h.sessions.Broadcast(
message.Admin{
Event: event.ADMIN_RELEASE,
ID: session.ID(),
}, nil)
}
2020-11-02 04:09:48 +13:00
func (h *MessageHandlerCtx) adminGive(session types.Session, payload *message.Admin) error {
if !session.Admin() {
h.logger.Debug().Msg("user not admin")
return nil
}
target, ok := h.sessions.Get(payload.ID)
if !ok {
2020-11-02 06:12:16 +13:00
h.logger.Debug().Str("id", payload.ID).Msg("can't find target session")
return nil
}
h.sessions.SetHost(target)
2020-11-17 06:12:25 +13:00
return h.sessions.Broadcast(
message.AdminTarget{
Event: event.CONTROL_GIVE,
ID: session.ID(),
Target: target.ID(),
2020-11-17 06:12:25 +13:00
}, nil)
}
2020-11-02 04:09:48 +13:00
func (h *MessageHandlerCtx) adminKick(session types.Session, payload *message.Admin) error {
if !session.Admin() {
h.logger.Debug().Msg("user not admin")
return nil
}
target, ok := h.sessions.Get(payload.ID)
if !ok {
2020-11-02 06:12:16 +13:00
h.logger.Debug().Str("id", payload.ID).Msg("can't find target session")
return nil
}
if target.Admin() {
h.logger.Debug().Msg("target is an admin, baling")
return nil
}
if err := target.Disconnect("kicked"); err != nil {
return err
}
2020-11-17 06:12:25 +13:00
return h.sessions.Broadcast(
message.AdminTarget{
Event: event.ADMIN_KICK,
Target: target.ID(),
ID: session.ID(),
2020-11-17 06:12:25 +13:00
}, []string{payload.ID})
}