Initial commit

pkg/fake/consumer.go

@@ -0,0 +1,47 @@
+package fake
+
+import (
+	"github.com/AlexxIT/go2rtc/pkg/streamer"
+	"github.com/pion/rtp"
+	"time"
+)
+
+type Consumer struct {
+	streamer.Element
+	Medias []*streamer.Media
+	Tracks []*streamer.Track
+
+	RecvPackets int
+	SendPackets int
+}
+
+func (c *Consumer) GetMedias() []*streamer.Media {
+	return c.Medias
+}
+
+func (c *Consumer) AddTrack(media *streamer.Media, track *streamer.Track) *streamer.Track {
+	switch track.Direction {
+	case streamer.DirectionSendonly:
+		track = track.Bind(func(packet *rtp.Packet) error {
+			if track.Codec.PayloadType != packet.PayloadType {
+				panic("wrong payload type")
+			}
+			c.RecvPackets++
+			return nil
+		})
+	case streamer.DirectionRecvonly:
+		go func() {
+			for {
+				pkt := &rtp.Packet{}
+				pkt.PayloadType = track.Codec.PayloadType
+				if err := track.WriteRTP(pkt); err != nil {
+					return
+				}
+				c.SendPackets++
+				time.Sleep(time.Second)
+			}
+		}()
+	}
+	c.Tracks = append(c.Tracks, track)
+	return track
+}

pkg/fake/producer.go

@@ -0,0 +1,62 @@
+package fake
+
+import (
+	"github.com/AlexxIT/go2rtc/pkg/streamer"
+	"github.com/pion/rtp"
+	"time"
+)
+
+type Producer struct {
+	streamer.Element
+	Medias []*streamer.Media
+	Tracks []*streamer.Track
+
+	RecvPackets int
+	SendPackets int
+}
+
+func (p *Producer) GetMedias() []*streamer.Media {
+	return p.Medias
+}
+
+func (p *Producer) GetTrack(media *streamer.Media, codec *streamer.Codec) *streamer.Track {
+	if !streamer.Contains(p.Medias, media, codec) {
+		panic("you shall not pass!")
+	}
+
+	track := &streamer.Track{Codec: codec, Direction: media.Direction}
+
+	switch media.Direction {
+	case streamer.DirectionSendonly:
+		track2 := track.Bind(func(packet *rtp.Packet) error {
+			p.RecvPackets++
+			return nil
+		})
+		p.Tracks = append(p.Tracks, track2)
+	case streamer.DirectionRecvonly:
+		p.Tracks = append(p.Tracks, track)
+	}
+
+	return track
+}
+
+func (p *Producer) Start() error {
+	for {
+		for _, track := range p.Tracks {
+			if track.Direction != streamer.DirectionSendonly {
+				continue
+			}
+			pkt := &rtp.Packet{}
+			pkt.PayloadType = track.Codec.PayloadType
+			if err := track.WriteRTP(pkt); err != nil {
+				return err
+			}
+			p.SendPackets++
+		}
+		time.Sleep(time.Second)
+	}
+}
+
+func (p *Producer) Stop() error {
+	panic("not implemented")
+}