Add media filter for RTSP source #198

pkg/rtsp/client.go

@@ -147,6 +147,16 @@ func (c *Conn) Describe() error {
 		return err
 	}
 
+	if c.Media != "" {
+		clone := make([]*core.Media, 0, len(medias))
+		for _, media := range medias {
+			if strings.Contains(c.Media, media.Kind) {
+				clone = append(clone, media)
+			}
+		}
+		medias = clone
+	}
+
 	// TODO: rewrite more smart
 	if c.Medias == nil {
 		c.Medias = medias

pkg/rtsp/conn.go

@@ -23,6 +23,7 @@ type Conn struct {
 	// public
 
 	Backchannel bool
+	Media       string
 	PacketSize  uint16
 	SessionName string
 	Transport   string // custom transport support, ex. RTSP over WebSocket