refactor capture with broadcast.

server/internal/types/broadcast.go

@@ -1,11 +0,0 @@
-package types
-
-type BroadcastManager interface {
-	Shutdown() error
-	Start() error
-	Stop()
-	IsActive() bool
-	Create(url string) error
-	Destroy()
-	GetUrl() string
-}

server/internal/types/capture.go

@@ -1,14 +1,38 @@
 package types
 
+import (
+	"errors"
+
+	"m1k1o/neko/internal/types/codec"
+)
+
+var (
+	ErrCapturePipelineAlreadyExists = errors.New("capture pipeline already exists")
+)
+
+type BroadcastManager interface {
+	Start(url string) error
+	Stop()
+	Started() bool
+	Url() string
+}
+
+type StreamSinkManager interface {
+	Codec() codec.RTPCodec
+	OnSample(listener func(sample Sample))
+
+	AddListener() error
+	RemoveListener() error
+
+	ListenersCount() int
+	Started() bool
+}
+
 type CaptureManager interface {
-	VideoCodec() string
-	AudioCodec() string
 	Start()
 	Shutdown() error
-	OnVideoFrame(listener func(sample Sample))
-	OnAudioFrame(listener func(sample Sample))
-	StartStream()
-	StopStream()
-	Streaming() bool
-	ChangeResolution(width int, height int, rate int16) error
+
+	Broadcast() BroadcastManager
+	Audio() StreamSinkManager
+	Video() StreamSinkManager
 }

server/internal/types/codec/codecs.go

@@ -0,0 +1,170 @@
+package codec
+
+import (
+	"strings"
+
+	"github.com/pion/webrtc/v3"
+)
+
+var RTCPFeedback = []webrtc.RTCPFeedback{
+	{Type: webrtc.TypeRTCPFBTransportCC, Parameter: ""},
+	{Type: webrtc.TypeRTCPFBGoogREMB, Parameter: ""},
+
+	// https://www.iana.org/assignments/sdp-parameters/sdp-parameters.xhtml#sdp-parameters-19
+	{Type: webrtc.TypeRTCPFBCCM, Parameter: "fir"},
+
+	// https://www.iana.org/assignments/sdp-parameters/sdp-parameters.xhtml#sdp-parameters-15
+	{Type: webrtc.TypeRTCPFBNACK, Parameter: "pli"},
+	{Type: webrtc.TypeRTCPFBNACK, Parameter: ""},
+}
+
+func ParseRTC(codec webrtc.RTPCodecParameters) (RTPCodec, bool) {
+	codecName := strings.Split(codec.RTPCodecCapability.MimeType, "/")[1]
+	return ParseStr(codecName)
+}
+
+func ParseStr(codecName string) (codec RTPCodec, ok bool) {
+	ok = true
+
+	switch strings.ToLower(codecName) {
+	case VP8().Name:
+		codec = VP8()
+	case VP9().Name:
+		codec = VP9()
+	case H264().Name:
+		codec = H264()
+	case Opus().Name:
+		codec = Opus()
+	case G722().Name:
+		codec = G722()
+	case PCMU().Name:
+		codec = PCMU()
+	case PCMA().Name:
+		codec = PCMA()
+	default:
+		ok = false
+	}
+
+	return
+}
+
+type RTPCodec struct {
+	Name        string
+	PayloadType webrtc.PayloadType
+	Type        webrtc.RTPCodecType
+	Capability  webrtc.RTPCodecCapability
+}
+
+func (codec *RTPCodec) Register(engine *webrtc.MediaEngine) error {
+	return engine.RegisterCodec(webrtc.RTPCodecParameters{
+		RTPCodecCapability: codec.Capability,
+		PayloadType:        codec.PayloadType,
+	}, codec.Type)
+}
+
+func VP8() RTPCodec {
+	return RTPCodec{
+		Name:        "vp8",
+		PayloadType: 96,
+		Type:        webrtc.RTPCodecTypeVideo,
+		Capability: webrtc.RTPCodecCapability{
+			MimeType:     webrtc.MimeTypeVP8,
+			ClockRate:    90000,
+			Channels:     0,
+			SDPFmtpLine:  "",
+			RTCPFeedback: RTCPFeedback,
+		},
+	}
+}
+
+// TODO: Profile ID.
+func VP9() RTPCodec {
+	return RTPCodec{
+		Name:        "vp9",
+		PayloadType: 98,
+		Type:        webrtc.RTPCodecTypeVideo,
+		Capability: webrtc.RTPCodecCapability{
+			MimeType:     webrtc.MimeTypeVP9,
+			ClockRate:    90000,
+			Channels:     0,
+			SDPFmtpLine:  "profile-id=0",
+			RTCPFeedback: RTCPFeedback,
+		},
+	}
+}
+
+// TODO: Profile ID.
+func H264() RTPCodec {
+	return RTPCodec{
+		Name:        "h264",
+		PayloadType: 102,
+		Type:        webrtc.RTPCodecTypeVideo,
+		Capability: webrtc.RTPCodecCapability{
+			MimeType:     webrtc.MimeTypeH264,
+			ClockRate:    90000,
+			Channels:     0,
+			SDPFmtpLine:  "level-asymmetry-allowed=1;packetization-mode=1;profile-level-id=42e01f",
+			RTCPFeedback: RTCPFeedback,
+		},
+	}
+}
+
+func Opus() RTPCodec {
+	return RTPCodec{
+		Name:        "opus",
+		PayloadType: 111,
+		Type:        webrtc.RTPCodecTypeAudio,
+		Capability: webrtc.RTPCodecCapability{
+			MimeType:     webrtc.MimeTypeOpus,
+			ClockRate:    48000,
+			Channels:     2,
+			SDPFmtpLine:  "useinbandfec=1",
+			RTCPFeedback: []webrtc.RTCPFeedback{},
+		},
+	}
+}
+
+func G722() RTPCodec {
+	return RTPCodec{
+		Name:        "g722",
+		PayloadType: 9,
+		Type:        webrtc.RTPCodecTypeAudio,
+		Capability: webrtc.RTPCodecCapability{
+			MimeType:     webrtc.MimeTypeG722,
+			ClockRate:    8000,
+			Channels:     0,
+			SDPFmtpLine:  "",
+			RTCPFeedback: []webrtc.RTCPFeedback{},
+		},
+	}
+}
+
+func PCMU() RTPCodec {
+	return RTPCodec{
+		Name:        "pcmu",
+		PayloadType: 0,
+		Type:        webrtc.RTPCodecTypeAudio,
+		Capability: webrtc.RTPCodecCapability{
+			MimeType:     webrtc.MimeTypePCMU,
+			ClockRate:    8000,
+			Channels:     0,
+			SDPFmtpLine:  "",
+			RTCPFeedback: []webrtc.RTCPFeedback{},
+		},
+	}
+}
+
+func PCMA() RTPCodec {
+	return RTPCodec{
+		Name:        "pcma",
+		PayloadType: 8,
+		Type:        webrtc.RTPCodecTypeAudio,
+		Capability: webrtc.RTPCodecCapability{
+			MimeType:     webrtc.MimeTypePCMA,
+			ClockRate:    8000,
+			Channels:     0,
+			SDPFmtpLine:  "",
+			RTCPFeedback: []webrtc.RTCPFeedback{},
+		},
+	}
+}