From 5d533338d0c27cde618465afaa5d19b779e3d8ff Mon Sep 17 00:00:00 2001
From: Alexey Khit <alexey.khit@gmail.com>
Date: Mon, 14 Aug 2023 06:48:52 +0300
Subject: [PATCH] Fix incoming FLV source

---
 pkg/flv/client.go   | 16 ++++++++--------
 pkg/flv/producer.go |  5 +++++
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/pkg/flv/client.go b/pkg/flv/client.go
index 09fbe1fb..f08ecd49 100644
--- a/pkg/flv/client.go
+++ b/pkg/flv/client.go
@@ -19,6 +19,8 @@ type Client struct {
 	medias    []*core.Media
 	receivers []*core.Receiver
 
+	video, audio *core.Receiver
+
 	recv int
 }
 
@@ -120,8 +122,6 @@ func (c *Client) Describe() error {
 }
 
 func (c *Client) Play() error {
-	video, audio := core.VA(c.receivers)
-
 	for {
 		tagType, timeMS, b, err := c.Transport.ReadTag()
 		if err != nil {
@@ -132,31 +132,31 @@ func (c *Client) Play() error {
 
 		switch tagType {
 		case TagAudio:
-			if audio == nil || b[1] == 0 {
+			if c.audio == nil || b[1] == 0 {
 				continue
 			}
 
 			pkt := &rtp.Packet{
 				Header: rtp.Header{
-					Timestamp: TimeToRTP(timeMS, audio.Codec.ClockRate),
+					Timestamp: TimeToRTP(timeMS, c.audio.Codec.ClockRate),
 				},
 				Payload: b[2:],
 			}
-			audio.WriteRTP(pkt)
+			c.audio.WriteRTP(pkt)
 
 		case TagVideo:
 			// frame type 4b, codecID 4b, avc packet type 8b, composition time 24b
-			if video == nil || b[1] == 0 {
+			if c.video == nil || b[1] == 0 {
 				continue
 			}
 
 			pkt := &rtp.Packet{
 				Header: rtp.Header{
-					Timestamp: TimeToRTP(timeMS, video.Codec.ClockRate),
+					Timestamp: TimeToRTP(timeMS, c.video.Codec.ClockRate),
 				},
 				Payload: b[5:],
 			}
-			video.WriteRTP(pkt)
+			c.video.WriteRTP(pkt)
 		}
 	}
 }
diff --git a/pkg/flv/producer.go b/pkg/flv/producer.go
index efd19d8e..3aa3c10a 100644
--- a/pkg/flv/producer.go
+++ b/pkg/flv/producer.go
@@ -18,6 +18,11 @@ func (c *Client) GetTrack(media *core.Media, codec *core.Codec) (*core.Receiver,
 		}
 	}
 	track := core.NewReceiver(media, codec)
+	if media.Kind == core.KindVideo {
+		c.video = track
+	} else {
+		c.audio = track
+	}
 	c.receivers = append(c.receivers, track)
 	return track, nil
 }