Rewrite stream info API

pkg/ivideon/client.go

@@ -15,6 +15,7 @@ import (
 	"net/http"
 	"strings"
 	"sync"
+	"sync/atomic"
 	"time"
 )
 
@@ -41,6 +42,8 @@ type Client struct {
 	buffer chan []byte
 	state  State
 	mu     sync.Mutex
+
+	recv uint32
 }
 
 func NewClient(id string) *Client {
@@ -109,6 +112,7 @@ func (c *Client) Handle() error {
 		c.mu.Lock()
 		if c.state == StateHandle {
 			c.buffer <- data
+			atomic.AddUint32(&c.recv, uint32(len(data)))
 		}
 		c.mu.Unlock()
 	}
@@ -140,6 +144,7 @@ func (c *Client) Handle() error {
 				c.mu.Lock()
 				if c.state == StateHandle {
 					c.buffer <- data
+					atomic.AddUint32(&c.recv, uint32(len(data)))
 				}
 				c.mu.Unlock()
 			}

pkg/ivideon/streamer.go

@@ -1,8 +1,10 @@
 package ivideon
 
 import (
+	"encoding/json"
 	"fmt"
 	"github.com/AlexxIT/go2rtc/pkg/streamer"
+	"sync/atomic"
 )
 
 func (c *Client) GetMedias() []*streamer.Media {
@@ -29,3 +31,19 @@ func (c *Client) Start() error {
 func (c *Client) Stop() error {
 	return c.Close()
 }
+
+func (c *Client) MarshalJSON() ([]byte, error) {
+	var tracks []*streamer.Track
+	for _, track := range c.tracks {
+		tracks = append(tracks, track)
+	}
+
+	info := &streamer.Info{
+		Type:   "Ivideon source",
+		URL:    c.ID,
+		Medias: c.medias,
+		Tracks: tracks,
+		Recv:   atomic.LoadUint32(&c.recv),
+	}
+	return json.Marshal(info)
+}