Restructure protobufs and use them everywhere

packages/relay/internal/shared/participant.go

@@ -2,43 +2,59 @@ package shared
 
 import (
 	"fmt"
+	"log/slog"
 	"relay/internal/common"
 	"relay/internal/connections"
 
+	"github.com/libp2p/go-libp2p/core/peer"
 	"github.com/oklog/ulid/v2"
+	"github.com/pion/rtp"
 	"github.com/pion/webrtc/v4"
 )
 
 type Participant struct {
 	ID             ulid.ULID
+	SessionID      string  // Track session for reconnection
+	PeerID         peer.ID // libp2p peer ID
 	PeerConnection *webrtc.PeerConnection
 	DataChannel    *connections.NestriDataChannel
+
+	// Per-viewer tracks and channels
+	VideoTrack *webrtc.TrackLocalStaticRTP
+	AudioTrack *webrtc.TrackLocalStaticRTP
+	VideoChan  chan *rtp.Packet
+	AudioChan  chan *rtp.Packet
 }
 
-func NewParticipant() (*Participant, error) {
+func NewParticipant(sessionID string, peerID peer.ID) (*Participant, error) {
 	id, err := common.NewULID()
 	if err != nil {
 		return nil, fmt.Errorf("failed to create ULID for Participant: %w", err)
 	}
 	return &Participant{
-		ID: id,
+		ID:        id,
+		SessionID: sessionID,
+		PeerID:    peerID,
+		VideoChan: make(chan *rtp.Packet, 500),
+		AudioChan: make(chan *rtp.Packet, 100),
 	}, nil
 }
 
-func (p *Participant) addTrack(trackLocal *webrtc.TrackLocalStaticRTP) error {
-	rtpSender, err := p.PeerConnection.AddTrack(trackLocal)
-	if err != nil {
-		return err
+// Close cleans up participant resources
+func (p *Participant) Close() {
+	if p.VideoChan != nil {
+		close(p.VideoChan)
+		p.VideoChan = nil
 	}
-
-	go func() {
-		rtcpBuffer := make([]byte, 1400)
-		for {
-			if _, _, rtcpErr := rtpSender.Read(rtcpBuffer); rtcpErr != nil {
-				break
-			}
+	if p.AudioChan != nil {
+		close(p.AudioChan)
+		p.AudioChan = nil
+	}
+	if p.PeerConnection != nil {
+		err := p.PeerConnection.Close()
+		if err != nil {
+			slog.Error("Failed to close Participant PeerConnection", err)
 		}
-	}()
-
-	return nil
+		p.PeerConnection = nil
+	}
 }