mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
package handler
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"demodesk/neko/internal/types"
|
|
"demodesk/neko/internal/types/event"
|
|
"demodesk/neko/internal/types/message"
|
|
"demodesk/neko/internal/utils"
|
|
)
|
|
|
|
func New(
|
|
sessions types.SessionManager,
|
|
desktop types.DesktopManager,
|
|
capture types.CaptureManager,
|
|
webrtc types.WebRTCManager,
|
|
) *MessageHandlerCtx {
|
|
logger := log.With().Str("module", "handler").Logger()
|
|
|
|
return &MessageHandlerCtx{
|
|
logger: logger,
|
|
sessions: sessions,
|
|
desktop: desktop,
|
|
capture: capture,
|
|
webrtc: webrtc,
|
|
}
|
|
}
|
|
|
|
type MessageHandlerCtx struct {
|
|
logger zerolog.Logger
|
|
sessions types.SessionManager
|
|
webrtc types.WebRTCManager
|
|
desktop types.DesktopManager
|
|
capture types.CaptureManager
|
|
}
|
|
|
|
func (h *MessageHandlerCtx) Message(session types.Session, raw []byte) error {
|
|
header := message.Message{}
|
|
if err := json.Unmarshal(raw, &header); err != nil {
|
|
return err
|
|
}
|
|
|
|
var err error
|
|
switch header.Event {
|
|
// Signal Events
|
|
case event.SIGNAL_ANSWER:
|
|
payload := &message.SignalAnswer{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.signalAnswer(session, payload)
|
|
})
|
|
|
|
// Control Events
|
|
case event.CONTROL_RELEASE:
|
|
err = h.controlRelease(session)
|
|
case event.CONTROL_REQUEST:
|
|
err = h.controlRequest(session)
|
|
|
|
// Screen Events
|
|
case event.SCREEN_SET:
|
|
payload := &message.ScreenSize{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.screenSet(session, payload)
|
|
})
|
|
|
|
// Clipboard Events
|
|
case event.CLIPBOARD_SET:
|
|
payload := &message.ClipboardData{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.clipboardSet(session, payload)
|
|
})
|
|
|
|
// Keyboard Events
|
|
case event.KEYBOARD_MODIFIERS:
|
|
payload := &message.KeyboardLayout{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.keyboardLayout(session, payload)
|
|
})
|
|
case event.KEYBOARD_LAYOUT:
|
|
payload := &message.KeyboardModifiers{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.keyboardModifiers(session, payload)
|
|
})
|
|
default:
|
|
return errors.Errorf("unknown message event %s", header.Event)
|
|
}
|
|
|
|
return errors.Wrapf(err, "%s failed", header.Event)
|
|
}
|