Update extensions.go

p4xx07 · May 30, 2023 · be4e970 · be4e970
1 parent 99d1b6b
commit be4e970
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ffprobe/extensions.go b/ffprobe/extensions.go
@@ -38,7 +38,11 @@ func (p *Probe) GetResolution() (width int, height int) {
 	return videoStream.Width, videoStream.Height
 }
 
-func (p *Probe) GetStreamDuration(codecType string) (float64, error) {
+func (p *Probe) GetStreamDuration(codecType ...string) (float64, error) {
+	selected := 'video'
+	if len(codecTypes) > 0{
+		selected = codecType[0]
+	}
 	var selected Stream
 	for _, stream := range p.Streams {
 		if stream.CodecType == codecType {