❌ revert(protobuf): Remove protobuf for now

packages/relay/internal/messages.go

@@ -1,17 +1,27 @@
 package relay
 
 import (
+	"bytes"
+	"compress/gzip"
+	"encoding/json"
+	"fmt"
 	"github.com/pion/webrtc/v4"
 	"time"
 )
 
-// OnMessageCallback is a callback for messages of given type
+// OnMessageCallback is a callback for binary messages of given type
 type OnMessageCallback func(data []byte)
 
 // MessageBase is the base type for WS/DC messages.
 type MessageBase struct {
-	PayloadType string          `json:"payload_type"`
-	Latency     *LatencyTracker `json:"latency,omitempty"`
+	PayloadType    string         `json:"payload_type"`
+	LatencyTracker LatencyTracker `json:"latency_tracker,omitempty"`
+}
+
+// MessageInput represents an input message.
+type MessageInput struct {
+	MessageBase
+	Data string `json:"data"`
 }
 
 // MessageLog represents a log message.
@@ -83,6 +93,50 @@ type MessageAnswer struct {
 	AnswerType AnswerType `json:"answer_type"`
 }
 
+// EncodeMessage encodes a message to be sent with gzip compression
+func EncodeMessage(msg interface{}) ([]byte, error) {
+	// Marshal the message to JSON
+	data, err := json.Marshal(msg)
+	if err != nil {
+		return nil, fmt.Errorf("failed to encode message: %w", err)
+	}
+
+	// Gzip compress the JSON
+	var compressedData bytes.Buffer
+	writer := gzip.NewWriter(&compressedData)
+	_, err = writer.Write(data)
+	if err != nil {
+		return nil, fmt.Errorf("failed to compress message: %w", err)
+	}
+	if err := writer.Close(); err != nil {
+		return nil, fmt.Errorf("failed to finalize compression: %w", err)
+	}
+
+	return compressedData.Bytes(), nil
+}
+
+// DecodeMessage decodes a message received with gzip decompression
+func DecodeMessage(data []byte, target interface{}) error {
+	// Gzip decompress the data
+	reader, err := gzip.NewReader(bytes.NewReader(data))
+	if err != nil {
+		return fmt.Errorf("failed to initialize decompression: %w", err)
+	}
+	defer func(reader *gzip.Reader) {
+		if err = reader.Close(); err != nil {
+			fmt.Printf("failed to close reader: %v\n", err)
+		}
+	}(reader)
+
+	// Decode the JSON
+	err = json.NewDecoder(reader).Decode(target)
+	if err != nil {
+		return fmt.Errorf("failed to decode message: %w", err)
+	}
+
+	return nil
+}
+
 // SendLogMessageWS sends a log message to the given WebSocket connection.
 func (ws *SafeWebSocket) SendLogMessageWS(level, message string) error {
 	msg := MessageLog{
@@ -91,7 +145,12 @@ func (ws *SafeWebSocket) SendLogMessageWS(level, message string) error {
 		Message:     message,
 		Time:        time.Now().Format(time.RFC3339),
 	}
-	return ws.SendJSON(msg)
+	encoded, err := EncodeMessage(msg)
+	if err != nil {
+		return fmt.Errorf("failed to encode log message: %w", err)
+	}
+
+	return ws.SendBinary(encoded)
 }
 
 // SendMetricsMessageWS sends a metrics message to the given WebSocket connection.
@@ -103,7 +162,12 @@ func (ws *SafeWebSocket) SendMetricsMessageWS(usageCPU, usageMemory float64, upt
 		Uptime:          uptime,
 		PipelineLatency: pipelineLatency,
 	}
-	return ws.SendJSON(msg)
+	encoded, err := EncodeMessage(msg)
+	if err != nil {
+		return fmt.Errorf("failed to encode metrics message: %w", err)
+	}
+
+	return ws.SendBinary(encoded)
 }
 
 // SendICECandidateMessageWS sends an ICE candidate message to the given WebSocket connection.
@@ -112,7 +176,12 @@ func (ws *SafeWebSocket) SendICECandidateMessageWS(candidate webrtc.ICECandidate
 		MessageBase: MessageBase{PayloadType: "ice"},
 		Candidate:   candidate,
 	}
-	return ws.SendJSON(msg)
+	encoded, err := EncodeMessage(msg)
+	if err != nil {
+		return fmt.Errorf("failed to encode ICE candidate message: %w", err)
+	}
+
+	return ws.SendBinary(encoded)
 }
 
 // SendSDPMessageWS sends an SDP message to the given WebSocket connection.
@@ -121,7 +190,12 @@ func (ws *SafeWebSocket) SendSDPMessageWS(sdp webrtc.SessionDescription) error {
 		MessageBase: MessageBase{PayloadType: "sdp"},
 		SDP:         sdp,
 	}
-	return ws.SendJSON(msg)
+	encoded, err := EncodeMessage(msg)
+	if err != nil {
+		return fmt.Errorf("failed to encode SDP message: %w", err)
+	}
+
+	return ws.SendBinary(encoded)
 }
 
 // SendAnswerMessageWS sends an answer message to the given WebSocket connection.
@@ -130,5 +204,24 @@ func (ws *SafeWebSocket) SendAnswerMessageWS(answer AnswerType) error {
 		MessageBase: MessageBase{PayloadType: "answer"},
 		AnswerType:  answer,
 	}
-	return ws.SendJSON(msg)
+	encoded, err := EncodeMessage(msg)
+	if err != nil {
+		return fmt.Errorf("failed to encode answer message: %w", err)
+	}
+
+	return ws.SendBinary(encoded)
+}
+
+// SendInputMessageDC sends an input message to the given DataChannel connection.
+func (ndc *NestriDataChannel) SendInputMessageDC(data string) error {
+	msg := MessageInput{
+		MessageBase: MessageBase{PayloadType: "input"},
+		Data:        data,
+	}
+	encoded, err := EncodeMessage(msg)
+	if err != nil {
+		return fmt.Errorf("failed to encode input message: %w", err)
+	}
+
+	return ndc.SendBinary(encoded)
 }