update server

peer.go

@@ -2,55 +2,70 @@ package main
 
 import (
 	"encoding/json"
+	"log"
 
 	"github.com/google/uuid"
 	"github.com/gorilla/websocket"
 	"github.com/pion/webrtc/v4"
 )
 
-type Peer struct {
-	ID      string
-	Conn    *websocket.Conn
-	PC      *webrtc.PeerConnection
-	Room    *Room
-	Senders map[string]*webrtc.RTPSender
-}
-
 type Signal struct {
 	Type string          `json:"type"`
 	Data json.RawMessage `json:"data"`
 }
 
-func NewPeer(conn *websocket.Conn) *Peer {
-	pc, _ := webrtc.NewPeerConnection(webrtc.Configuration{})
+type Peer struct {
+	ID   string
+	Conn *websocket.Conn
+	PC   *webrtc.PeerConnection
+	Room *Room
+}
 
-	p := &Peer{
-		ID:      uuid.NewString(),
-		Conn:    conn,
-		PC:      pc,
-		Senders: make(map[string]*webrtc.RTPSender),
+func NewPeer(conn *websocket.Conn, room *Room) *Peer {
+	pc, err := webrtc.NewPeerConnection(webrtc.Configuration{
+		ICEServers: []webrtc.ICEServer{
+			{URLs: []string{"stun:stun.l.google.com:19302"}},
+		},
+	})
+	if err != nil {
+		panic(err)
 	}
 
-	pc.OnTrack(func(track *webrtc.TrackRemote, _ *webrtc.RTPReceiver) {
-		localTrack := NewTrack(track)
-
-		p.Room.ForwardTrack(p, localTrack)
-
-		for {
-			pkt, _, err := track.ReadRTP()
-			if err != nil {
-				return
-			}
-			localTrack.Write(pkt)
-		}
-	})
+	p := &Peer{
+		ID:   uuid.NewString(),
+		Conn: conn,
+		PC:   pc,
+		Room: room,
+	}
 
 	pc.OnICECandidate(func(c *webrtc.ICECandidate) {
 		if c == nil {
 			return
 		}
-		msg, _ := json.Marshal(c.ToJSON())
-		p.Send("ice", msg)
+		data, _ := json.Marshal(c.ToJSON())
+		p.Send("ice", data)
+	})
+
+	pc.OnTrack(func(remote *webrtc.TrackRemote, _ *webrtc.RTPReceiver) {
+		log.Println("track received", remote.Kind())
+
+		local := NewTrack(remote)
+
+		p.Room.mu.Lock()
+		for _, other := range p.Room.peers {
+			if other.ID != p.ID {
+				other.AddTrack(local)
+			}
+		}
+		p.Room.mu.Unlock()
+
+		for {
+			pkt, _, err := remote.ReadRTP()
+			if err != nil {
+				return
+			}
+			local.Write(pkt)
+		}
 	})
 
 	return p
@@ -66,7 +81,11 @@ func (p *Peer) AddTrack(track *Track) {
 }
 
 func (p *Peer) ReadLoop() {
-	defer p.Conn.Close()
+	defer func() {
+		p.Conn.Close()
+		p.Room.Remove(p.ID)
+		p.PC.Close()
+	}()
 
 	for {
 		var msg Signal
@@ -95,6 +114,5 @@ func (p *Peer) ReadLoop() {
 }
 
 func (p *Peer) Send(t string, data []byte) {
-	msg := Signal{Type: t, Data: data}
-	p.Conn.WriteJSON(msg)
+	p.Conn.WriteJSON(Signal{Type: t, Data: data})
 }