package webrtc import ( "bytes" "encoding/binary" "strconv" "github.com/pion/webrtc/v3" ) const OP_MOVE = 0x01 const OP_SCROLL = 0x02 const OP_KEY_DOWN = 0x03 const OP_KEY_UP = 0x04 const OP_KEY_CLK = 0x05 type PayloadHeader struct { Event uint8 Length uint16 } type PayloadMove struct { PayloadHeader X uint16 Y uint16 } type PayloadScroll struct { PayloadHeader X int16 Y int16 } type PayloadKey struct { PayloadHeader Key uint64 } func (manager *WebRTCManager) handle(id string, msg webrtc.DataChannelMessage) error { if !manager.sessions.IsHost(id) { return nil } buffer := bytes.NewBuffer(msg.Data) header := &PayloadHeader{} hbytes := make([]byte, 3) if _, err := buffer.Read(hbytes); err != nil { return err } if err := binary.Read(bytes.NewBuffer(hbytes), binary.LittleEndian, header); err != nil { return err } buffer = bytes.NewBuffer(msg.Data) switch header.Event { case OP_MOVE: payload := &PayloadMove{} if err := binary.Read(buffer, binary.LittleEndian, payload); err != nil { return err } manager.remote.Move(int(payload.X), int(payload.Y)) break case OP_SCROLL: payload := &PayloadScroll{} if err := binary.Read(buffer, binary.LittleEndian, payload); err != nil { return err } manager.logger. Debug(). Str("x", strconv.Itoa(int(payload.X))). Str("y", strconv.Itoa(int(payload.Y))). Msg("scroll") manager.remote.Scroll(int(payload.X), int(payload.Y)) break case OP_KEY_DOWN: payload := &PayloadKey{} if err := binary.Read(buffer, binary.LittleEndian, payload); err != nil { return err } if payload.Key < 8 { err := manager.remote.ButtonDown(int(payload.Key)) if err != nil { manager.logger.Warn().Err(err).Msg("button down failed") return nil } manager.logger.Debug().Msgf("button down %d", payload.Key) } else { err := manager.remote.KeyDown(uint64(payload.Key)) if err != nil { manager.logger.Warn().Err(err).Msg("key down failed") return nil } manager.logger.Debug().Msgf("key down %d", payload.Key) } break case OP_KEY_UP: payload := &PayloadKey{} err := binary.Read(buffer, binary.LittleEndian, payload) if err != nil { return err } if payload.Key < 8 { err := manager.remote.ButtonUp(int(payload.Key)) if err != nil { manager.logger.Warn().Err(err).Msg("button up failed") return nil } manager.logger.Debug().Msgf("button up %d", payload.Key) } else { err := manager.remote.KeyUp(uint64(payload.Key)) if err != nil { manager.logger.Warn().Err(err).Msg("key up failed") return nil } manager.logger.Debug().Msgf("key up %d", payload.Key) } break case OP_KEY_CLK: // unused break } return nil }