Initial commit

pkg/webrtc/helper.go

@@ -0,0 +1,85 @@
+package webrtc
+
+import (
+	"github.com/AlexxIT/go2rtc/pkg/streamer"
+	"github.com/pion/ice/v2"
+	"github.com/pion/stun"
+	"github.com/pion/webrtc/v3"
+	"net"
+	"strconv"
+)
+
+func NewCandidate(address string) (string, error) {
+	host, port, err := net.SplitHostPort(address)
+	if err != nil {
+		return "", err
+	}
+
+	i, err := strconv.Atoi(port)
+	if err != nil {
+		return "", err
+	}
+
+	cand, err := ice.NewCandidateHost(&ice.CandidateHostConfig{
+		Network:   "tcp",
+		Address:   host,
+		Port:      i,
+		Component: ice.ComponentRTP,
+		TCPType:   ice.TCPTypePassive,
+	})
+	if err != nil {
+		return "", err
+	}
+
+	return "candidate:" + cand.Marshal(), nil
+}
+
+// GetPublicIP example from https://github.com/pion/stun
+func GetPublicIP() (net.IP, error) {
+	c, err := stun.Dial("udp", "stun.l.google.com:19302")
+	if err != nil {
+		return nil, err
+	}
+
+	var res stun.Event
+
+	message := stun.MustBuild(stun.TransactionID, stun.BindingRequest)
+	if err = c.Do(message, func(e stun.Event) { res = e }); err != nil {
+		return nil, err
+	}
+
+	if res.Error != nil {
+		return nil, res.Error
+	}
+
+	var xorAddr stun.XORMappedAddress
+	if err = xorAddr.GetFrom(res.Message); err != nil {
+		return nil, err
+	}
+
+	return xorAddr.IP, nil
+}
+
+func MimeType(codec *streamer.Codec) string {
+	switch codec.Name {
+	case streamer.CodecH264:
+		return webrtc.MimeTypeH264
+	case streamer.CodecH265:
+		return webrtc.MimeTypeH265
+	case streamer.CodecVP8:
+		return webrtc.MimeTypeVP8
+	case streamer.CodecVP9:
+		return webrtc.MimeTypeVP9
+	case streamer.CodecAV1:
+		return webrtc.MimeTypeAV1
+	case streamer.CodecPCMU:
+		return webrtc.MimeTypePCMU
+	case streamer.CodecPCMA:
+		return webrtc.MimeTypePCMA
+	case streamer.CodecOpus:
+		return webrtc.MimeTypeOpus
+	case streamer.CodecG722:
+		return webrtc.MimeTypeG722
+	}
+	panic("not implemented")
+}