From d87422488c5236aae08fc830b79c86a1260f5a43 Mon Sep 17 00:00:00 2001
From: Dan Jones <danjones@goodevilgenius.org>
Date: Mon, 4 Sep 2023 09:09:49 -0500
Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20Convert=20SelectAudioStrea?=
 =?UTF-8?q?m=20into=20a=20method=20of=20Probe?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/root.go    |  5 ++---
 media/probe.go | 36 ++++++++++++++++++------------------
 2 files changed, 20 insertions(+), 21 deletions(-)

diff --git a/cmd/root.go b/cmd/root.go
index 14e9b96..b9b0f4b 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -9,7 +9,6 @@ import (
 
 	"codeberg.org/danjones000/strip-beats/app"
 	"codeberg.org/danjones000/strip-beats/input/boolean"
-	"codeberg.org/danjones000/strip-beats/media"
 	"github.com/spf13/cobra"
 )
 
@@ -25,8 +24,8 @@ var rootCmd = &cobra.Command{
 		// media.Watch(out.Format.Path)
 		show := boolean.Choose("Show stream?")
 		if show {
-			stream := media.SelectAudioStream(out)
-			fmt.Printf("Stream %d of %s: %+v\n", stream, out.Format.Path, out.Streams[stream])
+			stream := out.WantedAudioStream()
+			fmt.Printf("Stream %d of %s: %+v\n", stream.Index, out.Format.Path, stream)
 		}
 	},
 }
diff --git a/media/probe.go b/media/probe.go
index b8cb7cc..4c576bf 100644
--- a/media/probe.go
+++ b/media/probe.go
@@ -62,6 +62,24 @@ func (p Probe) GetFirstAcceptableAudio() *Stream {
 	return sts[0]
 }
 
+func (pr Probe) WantedAudioStream() *Stream {
+	count := pr.GetAudioCount()
+	if count < 1 {
+		return nil
+	}
+
+	if count == 1 {
+		return pr.GetFirstAudio()
+	}
+
+	pref := pr.GetFirstAcceptableAudio()
+	if pref != nil {
+		return pref
+	}
+
+	return pr.GetFirstAudio()
+}
+
 type Stream struct {
 	Index         int
 	CodecName     string `json:"codec_name"`
@@ -130,21 +148,3 @@ func ProbeFile(path string) Probe {
 	}
 	return ret
 }
-
-func SelectAudioStream(pr Probe) int {
-	count := pr.GetAudioCount()
-	if count < 1 {
-		return -1
-	}
-
-	if count == 1 {
-		return pr.GetFirstAudio().Index
-	}
-
-	pref := pr.GetFirstAcceptableAudio()
-	if pref != nil {
-		return pref.Index
-	}
-
-	return pr.GetFirstAudio().Index
-}