BIG core logic rewrite

pkg/mpegts/producer.go

@@ -1,20 +1,21 @@
 package mpegts
 
 import (
-	"github.com/AlexxIT/go2rtc/pkg/streamer"
+	"encoding/json"
+	"github.com/AlexxIT/go2rtc/pkg/core"
 )
 
-func (c *Client) GetMedias() []*streamer.Media {
+func (c *Client) GetMedias() []*core.Media {
 	return c.medias
 }
 
-func (c *Client) GetTrack(media *streamer.Media, codec *streamer.Codec) *streamer.Track {
-	for _, track := range c.tracks {
+func (c *Client) GetTrack(media *core.Media, codec *core.Codec) (*core.Receiver, error) {
+	for _, track := range c.receivers {
 		if track.Codec == codec {
-			return track
+			return track, nil
 		}
 	}
-	return nil
+	return nil, core.ErrCantGetTrack
 }
 
 func (c *Client) Start() error {
@@ -22,5 +23,19 @@ func (c *Client) Start() error {
 }
 
 func (c *Client) Stop() error {
+	for _, receiver := range c.receivers {
+		receiver.Close()
+	}
 	return c.Close()
 }
+
+func (c *Client) MarshalJSON() ([]byte, error) {
+	info := &core.Info{
+		Type:      "MPEG-TS active producer",
+		URL:       c.res.Request.URL.String(),
+		Medias:    c.medias,
+		Receivers: c.receivers,
+		Recv:      c.recv,
+	}
+	return json.Marshal(info)
+}