neko/internal/webrtc/peer.go

77 lines
1.7 KiB
Go
Raw Normal View History

package webrtc
2021-06-30 10:04:41 +12:00
import (
"sync"
"github.com/pion/webrtc/v3"
2021-08-30 04:59:46 +12:00
"github.com/rs/zerolog"
2021-06-30 10:04:41 +12:00
)
2020-11-26 06:41:40 +13:00
type WebRTCPeerCtx struct {
2021-06-30 10:04:41 +12:00
mu sync.Mutex
2021-08-30 04:59:46 +12:00
logger zerolog.Logger
2021-02-15 02:40:17 +13:00
connection *webrtc.PeerConnection
dataChannel *webrtc.DataChannel
changeVideo func(videoID string) error
iceTrickle bool
}
func (peer *WebRTCPeerCtx) CreateOffer(ICERestart bool) (*webrtc.SessionDescription, error) {
2021-06-30 10:04:41 +12:00
peer.mu.Lock()
defer peer.mu.Unlock()
2021-06-28 08:02:05 +12:00
offer, err := peer.connection.CreateOffer(&webrtc.OfferOptions{
ICERestart: ICERestart,
})
if err != nil {
return nil, err
}
if !peer.iceTrickle {
2021-06-28 08:02:05 +12:00
// Create channel that is blocked until ICE Gathering is complete
gatherComplete := webrtc.GatheringCompletePromise(peer.connection)
if err := peer.connection.SetLocalDescription(offer); err != nil {
return nil, err
}
<-gatherComplete
} else {
if err := peer.connection.SetLocalDescription(offer); err != nil {
return nil, err
}
}
return peer.connection.LocalDescription(), nil
}
2021-02-15 05:11:21 +13:00
func (peer *WebRTCPeerCtx) SignalAnswer(sdp string) error {
return peer.connection.SetRemoteDescription(webrtc.SessionDescription{
2021-02-15 02:40:17 +13:00
SDP: sdp,
2020-11-02 04:09:48 +13:00
Type: webrtc.SDPTypeAnswer,
})
}
2021-02-15 05:11:21 +13:00
func (peer *WebRTCPeerCtx) SignalCandidate(candidate webrtc.ICECandidateInit) error {
return peer.connection.AddICECandidate(candidate)
2021-02-03 08:43:33 +13:00
}
2021-02-15 05:11:21 +13:00
func (peer *WebRTCPeerCtx) SetVideoID(videoID string) error {
2021-06-30 10:04:41 +12:00
peer.mu.Lock()
defer peer.mu.Unlock()
2021-08-30 04:59:46 +12:00
peer.logger.Info().Str("video_id", videoID).Msg("change video id")
2021-02-15 05:11:21 +13:00
return peer.changeVideo(videoID)
}
2021-08-30 05:17:10 +12:00
func (peer *WebRTCPeerCtx) Destroy() {
2021-06-30 10:04:41 +12:00
peer.mu.Lock()
defer peer.mu.Unlock()
2021-09-18 10:56:03 +12:00
if peer.connection != nil {
err := peer.connection.Close()
peer.logger.Err(err).Msg("peer connection destroyed")
peer.connection = nil
}
}