go-webrtc-net/listener.go

package webrtc

import (
	"context"
	"net"

	"github.com/pion/webrtc/v3"
)

// Listen creates a new WebRTC listener using the provided net.Listener for signaling
func Listen(lstn net.Listener) (net.Listener, error) {
	ctx, cancel := context.WithCancel(context.Background())

	l := &RTCListener{
		underlying: lstn,
		config: &webrtc.Configuration{
			ICEServers: []webrtc.ICEServer{
				{URLs: STUN_SERVER_URLS},
			},
		},
		acceptChan: make(chan net.Conn),
		ctx:        ctx,
		cancel:     cancel,
	}

	go l.acceptLoop()
	return l, nil
}

func (l *RTCListener) Accept() (net.Conn, error) {
	select {
	case conn := <-l.acceptChan:
		return conn, nil
	case <-l.ctx.Done():
		return nil, ErrConnectionClosed
	}
}

func (l *RTCListener) Close() error {
	l.mu.Lock()
	defer l.mu.Unlock()

	if l.closed {
		return nil
	}

	l.closed = true
	l.cancel()
	return l.underlying.Close()
}

func (l *RTCListener) Addr() net.Addr {
	return l.underlying.Addr()
}

func (l *RTCListener) acceptLoop() {
	for {
		conn, err := l.underlying.Accept()
		if err != nil {
			l.Close()
			return
		}
		select {
		case l.acceptChan <- conn:
		case <-l.ctx.Done():
			return
		}
	}
}