MJPEG module refactoring

pkg/mjpeg/client.go

@@ -2,7 +2,6 @@ package mjpeg
 
 import (
 	"bufio"
-	"encoding/json"
 	"errors"
 	"github.com/AlexxIT/go2rtc/pkg/streamer"
 	"github.com/AlexxIT/go2rtc/pkg/tcp"
@@ -25,63 +24,13 @@ type Client struct {
 	closed bool
 	res    *http.Response
 
-	track *streamer.Track
-	recv  uint32
+	medias []*streamer.Media
+	track  *streamer.Track
+	recv   uint32
 }
 
 func NewClient(res *http.Response) *Client {
-	codec := &streamer.Codec{
-		Name: streamer.CodecJPEG, ClockRate: 90000, PayloadType: streamer.PayloadTypeRAW,
-	}
-	return &Client{
-		res:   res,
-		track: streamer.NewTrack(codec, streamer.DirectionSendonly),
-	}
-}
-
-func (c *Client) GetMedias() []*streamer.Media {
-	return []*streamer.Media{{
-		Kind:      streamer.KindVideo,
-		Direction: streamer.DirectionSendonly,
-		Codecs:    []*streamer.Codec{c.track.Codec},
-	}}
-}
-
-func (c *Client) GetTrack(media *streamer.Media, codec *streamer.Codec) *streamer.Track {
-	return c.track
-}
-
-func (c *Client) Start() error {
-	ct := c.res.Header.Get("Content-Type")
-
-	if ct == "image/jpeg" {
-		return c.startJPEG()
-	}
-
-	// added in go1.18
-	if _, s, ok := strings.Cut(ct, "boundary="); ok {
-		return c.startMJPEG(s)
-	}
-
-	return errors.New("wrong Content-Type: " + ct)
-}
-
-func (c *Client) Stop() error {
-	// important for close reader/writer gorutines
-	_ = c.res.Body.Close()
-	c.closed = true
-	return nil
-}
-
-func (c *Client) MarshalJSON() ([]byte, error) {
-	info := &streamer.Info{
-		Type:       "MJPEG source",
-		URL:        c.res.Request.URL.String(),
-		RemoteAddr: c.RemoteAddr,
-		UserAgent:  c.UserAgent,
-		Recv:       atomic.LoadUint32(&c.recv),
-	}
-	return json.Marshal(info)
+	return &Client{res: res}
 }
 
 func (c *Client) startJPEG() error {

pkg/mjpeg/producer.go

@@ -0,0 +1,64 @@
+package mjpeg
+
+import (
+	"encoding/json"
+	"errors"
+	"github.com/AlexxIT/go2rtc/pkg/streamer"
+	"strings"
+	"sync/atomic"
+)
+
+func (c *Client) GetMedias() []*streamer.Media {
+	if c.medias == nil {
+		c.medias = []*streamer.Media{{
+			Kind:      streamer.KindVideo,
+			Direction: streamer.DirectionSendonly,
+			Codecs: []*streamer.Codec{
+				{
+					Name: streamer.CodecJPEG, ClockRate: 90000, PayloadType: streamer.PayloadTypeRAW,
+				},
+			},
+		}}
+	}
+	return c.medias
+}
+
+func (c *Client) GetTrack(media *streamer.Media, codec *streamer.Codec) *streamer.Track {
+	if c.track == nil {
+		c.track = streamer.NewTrack(codec, streamer.DirectionSendonly)
+	}
+	return c.track
+}
+
+func (c *Client) Start() error {
+	ct := c.res.Header.Get("Content-Type")
+
+	if ct == "image/jpeg" {
+		return c.startJPEG()
+	}
+
+	// added in go1.18
+	if _, s, ok := strings.Cut(ct, "boundary="); ok {
+		return c.startMJPEG(s)
+	}
+
+	return errors.New("wrong Content-Type: " + ct)
+}
+
+func (c *Client) Stop() error {
+	// important for close reader/writer gorutines
+	_ = c.res.Body.Close()
+	c.closed = true
+	return nil
+}
+
+func (c *Client) MarshalJSON() ([]byte, error) {
+	info := &streamer.Info{
+		Type:       "MJPEG source",
+		URL:        c.res.Request.URL.String(),
+		RemoteAddr: c.RemoteAddr,
+		UserAgent:  c.UserAgent,
+		Recv:       atomic.LoadUint32(&c.recv),
+	}
+	return json.Marshal(info)
+}