move server to server directory.

2025-06-10 23:03:05 +02:00 · 2024-06-23 17:48:14 +02:00 · 2024-06-23 17:48:14 +02:00 · cfb423b13d
commit cfb423b13d
parent da45f62ca8
211 changed files with 18 additions and 10 deletions
--- a/server/internal/webrtc/handler.go
+++ b/server/internal/webrtc/handler.go
@ -0,0 +1,205 @@
+package webrtc
+
+import (
+	"bytes"
+	"encoding/binary"
+	"math"
+	"time"
+
+	"github.com/demodesk/neko/internal/webrtc/payload"
+	"github.com/demodesk/neko/pkg/types"
+	"github.com/pion/webrtc/v3"
+	"github.com/rs/zerolog"
+)
+
+func (manager *WebRTCManagerCtx) handle(
+	logger zerolog.Logger, data []byte,
+	dataChannel *webrtc.DataChannel,
+	session types.Session,
+) error {
+	isHost := session.IsHost()
+
+	//
+	// parse header
+	//
+
+	buffer := bytes.NewBuffer(data)
+
+	header := &payload.Header{}
+	if err := binary.Read(buffer, binary.BigEndian, header); err != nil {
+		return err
+	}
+
+	//
+	// parse body
+	//
+
+	// handle cursor move event
+	if header.Event == payload.OP_MOVE {
+		payload := &payload.Move{}
+		if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+			return err
+		}
+
+		x, y := int(payload.X), int(payload.Y)
+		if isHost {
+			// handle active cursor movement
+			manager.desktop.Move(x, y)
+			manager.curPosition.Set(x, y)
+		} else {
+			// handle inactive cursor movement
+			session.SetCursor(types.Cursor{
+				X: x,
+				Y: y,
+			})
+		}
+
+		return nil
+	} else if header.Event == payload.OP_PING {
+		ping := &payload.Ping{}
+		if err := binary.Read(buffer, binary.BigEndian, ping); err != nil {
+			return err
+		}
+
+		// create pong header
+		header := payload.Header{
+			Event:  payload.OP_PONG,
+			Length: 19,
+		}
+
+		// generate server timestamp
+		serverTs := uint64(time.Now().UnixMilli())
+
+		// generate pong payload
+		pong := payload.Pong{
+			Ping:      *ping,
+			ServerTs1: uint32(serverTs / math.MaxUint32),
+			ServerTs2: uint32(serverTs % math.MaxUint32),
+		}
+
+		buffer := &bytes.Buffer{}
+
+		if err := binary.Write(buffer, binary.BigEndian, header); err != nil {
+			return err
+		}
+
+		if err := binary.Write(buffer, binary.BigEndian, pong); err != nil {
+			return err
+		}
+
+		return dataChannel.Send(buffer.Bytes())
+	}
+
+	// continue only if session is host
+	if !isHost {
+		return nil
+	}
+
+	switch header.Event {
+	case payload.OP_SCROLL:
+		// TODO: remove this once the client is fixed
+		if header.Length == 4 {
+			payload := &payload.Scroll_Old{}
+			if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+				return err
+			}
+
+			manager.desktop.Scroll(int(payload.X), int(payload.Y), false)
+			logger.Trace().
+				Int16("x", payload.X).
+				Int16("y", payload.Y).
+				Msg("scroll")
+		} else {
+			payload := &payload.Scroll{}
+			if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+				return err
+			}
+
+			manager.desktop.Scroll(int(payload.DeltaX), int(payload.DeltaY), payload.ControlKey)
+			logger.Trace().
+				Int16("deltaX", payload.DeltaX).
+				Int16("deltaY", payload.DeltaY).
+				Bool("controlKey", payload.ControlKey).
+				Msg("scroll")
+		}
+	case payload.OP_KEY_DOWN:
+		payload := &payload.Key{}
+		if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+			return err
+		}
+
+		if err := manager.desktop.KeyDown(payload.Key); err != nil {
+			logger.Warn().Err(err).Uint32("key", payload.Key).Msg("key down failed")
+		} else {
+			logger.Trace().Uint32("key", payload.Key).Msg("key down")
+		}
+	case payload.OP_KEY_UP:
+		payload := &payload.Key{}
+		if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+			return err
+		}
+
+		if err := manager.desktop.KeyUp(payload.Key); err != nil {
+			logger.Warn().Err(err).Uint32("key", payload.Key).Msg("key up failed")
+		} else {
+			logger.Trace().Uint32("key", payload.Key).Msg("key up")
+		}
+	case payload.OP_BTN_DOWN:
+		payload := &payload.Key{}
+		if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+			return err
+		}
+
+		if err := manager.desktop.ButtonDown(payload.Key); err != nil {
+			logger.Warn().Err(err).Uint32("key", payload.Key).Msg("button down failed")
+		} else {
+			logger.Trace().Uint32("key", payload.Key).Msg("button down")
+		}
+	case payload.OP_BTN_UP:
+		payload := &payload.Key{}
+		if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+			return err
+		}
+
+		if err := manager.desktop.ButtonUp(payload.Key); err != nil {
+			logger.Warn().Err(err).Uint32("key", payload.Key).Msg("button up failed")
+		} else {
+			logger.Trace().Uint32("key", payload.Key).Msg("button up")
+		}
+	case payload.OP_TOUCH_BEGIN:
+		payload := &payload.Touch{}
+		if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+			return err
+		}
+
+		if err := manager.desktop.TouchBegin(payload.TouchId, int(payload.X), int(payload.Y), payload.Pressure); err != nil {
+			logger.Warn().Err(err).Uint32("touchId", payload.TouchId).Msg("touch begin failed")
+		} else {
+			logger.Trace().Uint32("touchId", payload.TouchId).Msg("touch begin")
+		}
+	case payload.OP_TOUCH_UPDATE:
+		payload := &payload.Touch{}
+		if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+			return err
+		}
+
+		if err := manager.desktop.TouchUpdate(payload.TouchId, int(payload.X), int(payload.Y), payload.Pressure); err != nil {
+			logger.Warn().Err(err).Uint32("touchId", payload.TouchId).Msg("touch update failed")
+		} else {
+			logger.Trace().Uint32("touchId", payload.TouchId).Msg("touch update")
+		}
+	case payload.OP_TOUCH_END:
+		payload := &payload.Touch{}
+		if err := binary.Read(buffer, binary.BigEndian, payload); err != nil {
+			return err
+		}
+
+		if err := manager.desktop.TouchEnd(payload.TouchId, int(payload.X), int(payload.Y), payload.Pressure); err != nil {
+			logger.Warn().Err(err).Uint32("touchId", payload.TouchId).Msg("touch end failed")
+		} else {
+			logger.Trace().Uint32("touchId", payload.TouchId).Msg("touch end")
+		}
+	}
+
+	return nil
+}