mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
93 lines
2.3 KiB
Go
93 lines
2.3 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)
|
|
case event.CONTROL_GIVE:
|
|
payload := &message.Control{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.controlGive(session, payload)
|
|
})
|
|
case event.CONTROL_CLIPBOARD:
|
|
payload := &message.Clipboard{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.controlClipboard(session, payload)
|
|
})
|
|
case event.CONTROL_KEYBOARD:
|
|
payload := &message.Keyboard{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.controlKeyboard(session, payload)
|
|
})
|
|
|
|
// Screen Events
|
|
case event.SCREEN_RESOLUTION:
|
|
err = h.screenSize(session)
|
|
case event.SCREEN_CONFIGURATIONS:
|
|
err = h.screenConfigurations(session)
|
|
case event.SCREEN_SET:
|
|
payload := &message.ScreenResolution{}
|
|
err = utils.Unmarshal(payload, raw, func() error {
|
|
return h.screenSizeChange(session, payload)
|
|
})
|
|
default:
|
|
return errors.Errorf("unknown message event %s", header.Event)
|
|
}
|
|
|
|
return errors.Wrapf(err, "%s failed", header.Event)
|
|
}
|