neko/internal/webrtc/peer.go

46 lines
1.2 KiB
Go
Raw Normal View History

package webrtc
2021-02-06 05:57:33 +13:00
import (
"fmt"
"github.com/pion/webrtc/v3"
)
2020-11-26 06:41:40 +13:00
type WebRTCPeerCtx struct {
2021-02-06 08:35:30 +13:00
api *webrtc.API
engine *webrtc.MediaEngine
settings *webrtc.SettingEngine
connection *webrtc.PeerConnection
configuration *webrtc.Configuration
videoTracks map[string]*webrtc.TrackLocalStaticSample
videoSender *webrtc.RTPSender
}
2020-11-26 06:41:40 +13:00
func (webrtc_peer *WebRTCPeerCtx) SignalAnswer(sdp string) error {
return webrtc_peer.connection.SetRemoteDescription(webrtc.SessionDescription{
2020-11-02 04:09:48 +13:00
SDP: sdp,
Type: webrtc.SDPTypeAnswer,
})
}
2021-02-03 08:43:33 +13:00
func (webrtc_peer *WebRTCPeerCtx) SignalCandidate(candidate webrtc.ICECandidateInit) error {
return webrtc_peer.connection.AddICECandidate(candidate)
}
2021-02-06 05:57:33 +13:00
func (webrtc_peer *WebRTCPeerCtx) SetVideoID(videoID string) error {
track, ok := webrtc_peer.videoTracks[videoID]
if !ok {
return fmt.Errorf("videoID not found in available tracks")
}
2021-02-06 08:35:30 +13:00
return webrtc_peer.videoSender.ReplaceTrack(track)
}
2020-11-26 06:41:40 +13:00
func (webrtc_peer *WebRTCPeerCtx) Destroy() error {
if webrtc_peer.connection == nil || webrtc_peer.connection.ConnectionState() != webrtc.PeerConnectionStateConnected {
2020-11-02 04:09:48 +13:00
return nil
}
2021-02-03 06:28:32 +13:00
2020-11-26 06:41:40 +13:00
return webrtc_peer.connection.Close()
}