Refactored track_player.go

pkg/model.go

@@ -1,10 +1,10 @@
 package gomus
 
 import (
-	"fmt"
-	"os"
+	"time"
 
 	tea "github.com/charmbracelet/bubbletea"
+	"github.com/faiface/beep/speaker"
 )
 
 var (
@@ -20,10 +20,10 @@ type Model struct {
 	cursor           int
 	currentlyPlaying int
 
-	trackPlayer
+	TrackPlayer
 	trackIndex
 
-	trackPlayerView trackPlayerView
+	trackPlayerView
 }
 
 func NewModel(args ModelArgs) Model {
@@ -35,7 +35,7 @@ func NewModel(args ModelArgs) Model {
 		currentlyPlaying: 0,
 
 		trackIndex:  ti,
-		trackPlayer: trackPlayer{},
+		TrackPlayer: TrackPlayer{},
 
 		trackPlayerView: tpv,
 	}
@@ -55,12 +55,15 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	case tea.KeyMsg:
 		switch msg.String() {
 		case "ctrl+c", "q":
-			m.trackPlayer.close()
+			m.TrackPlayer.Close()
 			return m, tea.Quit
 		case "enter":
 			t := m.trackPlayerView.trackList.SelectedItem().(track)
 			cmds = append(cmds, newTrackChangeCmd(t))
-			m.trackPlayer.play(t.getReader())
+			stream, format, err := t.GetStream()
+			check(err)
+			speaker.Init(format.SampleRate, format.SampleRate.N(time.Second/10))
+			m.TrackPlayer.Play(&stream)
 		case " ":
 			pauseState := m.TrackPlayer.TogglePause()
 			cmds = append(cmds, newTrackPauseCmd(pauseState))

pkg/track_fetcher.go

@@ -2,48 +2,52 @@ package gomus
 
 import (
 	"fmt"
-	"io"
 	"io/ioutil"
 	"os"
 	"path/filepath"
 	"strings"
 
+	"github.com/faiface/beep"
+	bflac "github.com/faiface/beep/flac"
 	"github.com/mewkiz/flac"
 	"github.com/mewkiz/flac/meta"
 )
 
 type track struct {
-	name      string
-	artist    string
-	trackPath string
+	Name      string
+	Artist    string
+	TrackPath string
 }
 
 func (t track) FilterValue() string { return "" }
+func (t track) fullName() string    { return fmt.Sprintf("%s - %s", t.Artist, t.Name) }
 
-func (t track) fullName() string {
-	return fmt.Sprintf("%s - %s", t.artist, t.name)
-}
-
-func (t track) getReader() io.Reader {
-	f, err := os.Open(t.trackPath)
+func (t track) GetStream() (beep.StreamSeekCloser, beep.Format, error) {
+	f, err := os.Open(t.TrackPath)
 	check(err)
 
-	return f
+	if strings.HasSuffix(t.TrackPath, ".flac") {
+		streamer, format, err := bflac.Decode(f)
+		check(err)
+		return streamer, format, nil
+	}
+
+	return nil, beep.Format{}, fmt.Errorf("Could not parse track")
 }
 
-func trackFromFlac(path string) track {
+func TrackFromFlac(path string) track {
 	s, err := flac.ParseFile(path)
 	check(err)
 
-	t := track{trackPath: path}
+	t := track{TrackPath: path}
 	for _, block := range s.Blocks {
 		if block.Header.Type == meta.TypeVorbisComment {
 			c := block.Body.(*meta.VorbisComment)
 			for _, tag := range c.Tags {
 				if tag[0] == "ARTIST" {
-					t.artist = tag[1]
+					t.Artist = tag[1]
 				} else if tag[0] == "TITLE" {
-					t.name = tag[1]
+					t.Name = tag[1]
 				}
 			}
 		}
@@ -64,7 +68,7 @@ func NewDirTrackIndex(path string) trackIndex {
 	for _, file := range files {
 		if !file.IsDir() && strings.HasSuffix(file.Name(), ".flac") {
 			p := filepath.Join(path, file.Name())
-			tracks = append(tracks, trackFromFlac(p))
+			tracks = append(tracks, TrackFromFlac(p))
 		}
 	}
 

pkg/track_player.go

@@ -1,32 +1,35 @@
 package gomus
 
 import (
-	"io"
-	"time"
-
 	"github.com/faiface/beep"
-	"github.com/faiface/beep/flac"
 	"github.com/faiface/beep/speaker"
 )
 
-type trackPlayer struct {
-	currentStream beep.StreamSeekCloser
+type TrackPlayer struct {
+	streamer   *beep.StreamSeekCloser
+	playerCtrl *beep.Ctrl
 }
 
-func (t trackPlayer) play(reader io.Reader) {
-	if t.currentStream != nil {
-		t.currentStream.Close()
+func (t *TrackPlayer) Play(streamer *beep.StreamSeekCloser) {
+	if t.streamer != nil {
+		t.Close()
 	}
-	streamer, format, err := flac.Decode(reader)
-	check(err)
 
-	err = speaker.Init(format.SampleRate, format.SampleRate.N(time.Second/10))
-	check(err)
+	ctrl := &beep.Ctrl{Streamer: beep.Loop(-1, *streamer), Paused: false}
+	speaker.Play(ctrl)
 
-	speaker.Play(streamer)
-	t.currentStream = streamer
+	t.playerCtrl = ctrl
+	t.streamer = streamer
 }
 
-func (t trackPlayer) close() {
-	t.currentStream.Close()
+func (t *TrackPlayer) TogglePause() bool {
+	speaker.Lock()
+	newState := !t.playerCtrl.Paused
+	t.playerCtrl.Paused = newState
+	speaker.Unlock()
+	return newState
+}
+
+func (t *TrackPlayer) Close() {
+	(*t.streamer).Close()
 }