release: fixed get STreams

p4xx07 · Jun 4, 2024 · 66f003a · 66f003a
1 parent 4afc69f
commit 66f003a
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/ffprobe/probe.go b/ffprobe/probe.go
@@ -100,15 +100,23 @@ func (p *Probe) GetStreams(streamType string) []Stream {
 			result = append(result, v)
 		}
 	}
-	return nil
+	return result
 }
 
 func (p *Probe) GetFirstVideoStream() *Stream {
-	return &p.GetStreams("video")[0]
+	streams := p.GetVideoStreams()
+	if len(streams) == 0 {
+		return nil
+	}
+	return &streams[0]
 }
 
 func (p *Probe) GetFirstAudioStream() *Stream {
-	return &p.GetStreams("audio")[0]
+	streams := p.GetAudioStreams()
+	if len(streams) == 0 {
+		return nil
+	}
+	return &streams[0]
 }
 
 func (p *Probe) GetVideoStreams() []Stream {