iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

AddHandler in WebSockets.

Browse Source
This commit is contained in:
Miroslav Šedivý
2020-12-12 17:40:10 +01:00
parent af463b1015
commit 5b69413e01
2 changed files with 29 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/websocket/handler/handler.go
View File

@ -3,7 +3,6 @@ package handler
import (
"encoding/json"
"github.com/pkg/errors"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
@ -38,10 +37,11 @@ type MessageHandlerCtx struct {
capture types.CaptureManager
}
func (h *MessageHandlerCtx) Message(session types.Session, raw []byte) error {
func (h *MessageHandlerCtx) Message(session types.Session, raw []byte) bool {
header := message.Message{}
if err := json.Unmarshal(raw, &header); err != nil {
return err
h.logger.Error().Err(err).Msg("message parsing has failed")
return false
}
var err error
@ -87,8 +87,12 @@ func (h *MessageHandlerCtx) Message(session types.Session, raw []byte) error {
return h.keyboardModifiers(session, payload)
})
default:
return errors.Errorf("unknown message event %s", header.Event)
return false
}
return errors.Wrapf(err, "%s failed", header.Event)
if err != nil {
h.logger.Error().Err(err).Msg("message handler has failed")
}
return true
}