refactor: move play tts to core

internal/translate/apertium/translator.go

@@ -18,7 +18,7 @@ const (
 type Translator struct {
 	*core.Server
 	*core.Language
-	*core.TTSLock
+	*core.TTS
 	core.EngineName
 }
 
@@ -26,7 +26,7 @@ func NewTranslator() *Translator {
 	return &Translator{
 		Server:     new(core.Server),
 		Language:   new(core.Language),
-		TTSLock:    core.NewTTSLock(),
+		TTS:        core.NewTTS(),
 		EngineName: core.NewEngineName("Apertium"),
 	}
 }

internal/translate/argos/translator.go

@@ -17,7 +17,7 @@ const (
 type Translator struct {
 	*core.Server
 	*core.Language
-	*core.TTSLock
+	*core.TTS
 	core.EngineName
 }
 
@@ -25,7 +25,7 @@ func NewTranslator() *Translator {
 	return &Translator{
 		Server:     new(core.Server),
 		Language:   new(core.Language),
-		TTSLock:    core.NewTTSLock(),
+		TTS:        core.NewTTS(),
 		EngineName: core.NewEngineName("Argos"),
 	}
 }

internal/translate/bing/translator.go

@@ -9,11 +9,8 @@ import (
 	"net/url"
 	"regexp"
 	"strings"
-	"time"
 
 	"github.com/eeeXun/gtt/internal/translate/core"
-	"github.com/hajimehoshi/go-mp3"
-	"github.com/hajimehoshi/oto/v2"
 )
 
 const (
@@ -27,7 +24,7 @@ const (
 type Translator struct {
 	*core.Server
 	*core.Language
-	*core.TTSLock
+	*core.TTS
 	core.EngineName
 }
 
@@ -42,7 +39,7 @@ func NewTranslator() *Translator {
 	return &Translator{
 		Server:     new(core.Server),
 		Language:   new(core.Language),
-		TTSLock:    core.NewTTSLock(),
+		TTS:        core.NewTTS(),
 		EngineName: core.NewEngineName("Bing"),
 	}
 }
@@ -192,22 +189,5 @@ func (t *Translator) PlayTTS(lang, message string) error {
 	if err != nil {
 		return err
 	}
-	decoder, err := mp3.NewDecoder(res.Body)
+	return t.Play(res.Body)
-	if err != nil {
-		return err
-	}
-	otoCtx, readyChan, err := oto.NewContext(decoder.SampleRate(), 2, 2)
-	if err != nil {
-		return err
-	}
-	<-readyChan
-	player := otoCtx.NewPlayer(decoder)
-	player.Play()
-	for player.IsPlaying() {
-		if t.IsStopped() {
-			return player.Close()
-		}
-		time.Sleep(time.Millisecond)
-	}
-	return player.Close()
 }

internal/translate/chatgpt/translator.go

@@ -18,7 +18,7 @@ const (
 type Translator struct {
 	*core.Server
 	*core.Language
-	*core.TTSLock
+	*core.TTS
 	core.EngineName
 }
 
@@ -26,7 +26,7 @@ func NewTranslator() *Translator {
 	return &Translator{
 		Server:     new(core.Server),
 		Language:   new(core.Language),
-		TTSLock:    core.NewTTSLock(),
+		TTS:        core.NewTTS(),
 		EngineName: core.NewEngineName("ChatGPT"),
 	}
 }

internal/translate/core/lock.go

@@ -1,35 +0,0 @@
-package core
-
-type TTSLock struct {
-	using bool
-	stop  bool
-}
-
-func NewTTSLock() *TTSLock {
-	return &TTSLock{
-		stop:  true,
-		using: false,
-	}
-}
-
-func (l *TTSLock) LockAvailable() bool {
-	return l.stop && !l.using
-}
-
-func (l *TTSLock) AcquireLock() {
-	l.stop = false
-	l.using = true
-}
-
-func (l *TTSLock) IsStopped() bool {
-	return l.stop
-}
-
-func (l *TTSLock) StopTTS() {
-	l.stop = true
-}
-
-func (l *TTSLock) ReleaseLock() {
-	l.stop = true
-	l.using = false
-}

internal/translate/core/tts.go

@@ -0,0 +1,64 @@
+package core
+
+import (
+	"io"
+	"time"
+
+	"github.com/hajimehoshi/go-mp3"
+	"github.com/hajimehoshi/oto/v2"
+)
+
+type TTS struct {
+	stop  bool
+	using bool
+}
+
+func NewTTS() *TTS {
+	return &TTS{
+		stop:  true,
+		using: false,
+	}
+}
+
+func (s *TTS) LockAvailable() bool {
+	return s.stop && !s.using
+}
+
+func (s *TTS) AcquireLock() {
+	s.stop = false
+	s.using = true
+}
+
+func (s *TTS) IsStopped() bool {
+	return s.stop
+}
+
+func (s *TTS) StopTTS() {
+	s.stop = true
+}
+
+func (s *TTS) ReleaseLock() {
+	s.stop = true
+	s.using = false
+}
+
+func (s *TTS) Play(body io.Reader) error {
+	decoder, err := mp3.NewDecoder(body)
+	if err != nil {
+		return err
+	}
+	otoCtx, readyChan, err := oto.NewContext(decoder.SampleRate(), 2, 2)
+	if err != nil {
+		return err
+	}
+	<-readyChan
+	player := otoCtx.NewPlayer(decoder)
+	player.Play()
+	for player.IsPlaying() {
+		if s.IsStopped() {
+			return player.Close()
+		}
+		time.Sleep(time.Millisecond)
+	}
+	return player.Close()
+}

internal/translate/deepl/translator.go

@@ -18,7 +18,7 @@ const (
 type Translator struct {
 	*core.Server
 	*core.Language
-	*core.TTSLock
+	*core.TTS
 	core.EngineName
 }
 
@@ -26,7 +26,7 @@ func NewTranslator() *Translator {
 	return &Translator{
 		Server:     new(core.Server),
 		Language:   new(core.Language),
-		TTSLock:    core.NewTTSLock(),
+		TTS:        core.NewTTS(),
 		EngineName: core.NewEngineName("DeepL"),
 	}
 }

internal/translate/deeplx/translator.go

@@ -13,7 +13,7 @@ import (
 type Translator struct {
 	*core.Server
 	*core.Language
-	*core.TTSLock
+	*core.TTS
 	core.EngineName
 }
 
@@ -21,7 +21,7 @@ func NewTranslator() *Translator {
 	return &Translator{
 		Server:     new(core.Server),
 		Language:   new(core.Language),
-		TTSLock:    core.NewTTSLock(),
+		TTS:        core.NewTTS(),
 		EngineName: core.NewEngineName("DeepLX"),
 	}
 }

internal/translate/google/translator.go

@@ -7,11 +7,8 @@ import (
 	"io"
 	"net/http"
 	"net/url"
-	"time"
 
 	"github.com/eeeXun/gtt/internal/translate/core"
-	"github.com/hajimehoshi/go-mp3"
-	"github.com/hajimehoshi/oto/v2"
 )
 
 const (
@@ -22,7 +19,7 @@ const (
 type Translator struct {
 	*core.Server
 	*core.Language
-	*core.TTSLock
+	*core.TTS
 	core.EngineName
 }
 
@@ -30,7 +27,7 @@ func NewTranslator() *Translator {
 	return &Translator{
 		Server:     new(core.Server),
 		Language:   new(core.Language),
-		TTSLock:    core.NewTTSLock(),
+		TTS:        core.NewTTS(),
 		EngineName: core.NewEngineName("Google"),
 	}
 }
@@ -135,22 +132,5 @@ func (t *Translator) PlayTTS(lang, message string) error {
 	if res.StatusCode == 400 {
 		return errors.New(t.GetEngineName() + " does not support text to speech of " + lang)
 	}
-	decoder, err := mp3.NewDecoder(res.Body)
+	return t.Play(res.Body)
-	if err != nil {
-		return err
-	}
-	otoCtx, readyChan, err := oto.NewContext(decoder.SampleRate(), 2, 2)
-	if err != nil {
-		return err
-	}
-	<-readyChan
-	player := otoCtx.NewPlayer(decoder)
-	player.Play()
-	for player.IsPlaying() {
-		if t.IsStopped() {
-			return player.Close()
-		}
-		time.Sleep(time.Millisecond)
-	}
-	return player.Close()
 }

internal/translate/reverso/translator.go

@@ -9,11 +9,8 @@ import (
 	"io"
 	"net/http"
 	"regexp"
-	"time"
 
 	"github.com/eeeXun/gtt/internal/translate/core"
-	"github.com/hajimehoshi/go-mp3"
-	"github.com/hajimehoshi/oto/v2"
 )
 
 const (
@@ -24,7 +21,7 @@ const (
 type Translator struct {
 	*core.Server
 	*core.Language
-	*core.TTSLock
+	*core.TTS
 	core.EngineName
 }
 
@@ -32,7 +29,7 @@ func NewTranslator() *Translator {
 	return &Translator{
 		Server:     new(core.Server),
 		Language:   new(core.Language),
-		TTSLock:    core.NewTTSLock(),
+		TTS:        core.NewTTS(),
 		EngineName: core.NewEngineName("Reverso"),
 	}
 }
@@ -131,22 +128,5 @@ func (t *Translator) PlayTTS(lang, message string) error {
 	if err != nil {
 		return err
 	}
-	decoder, err := mp3.NewDecoder(res.Body)
+	return t.Play(res.Body)
-	if err != nil {
-		return err
-	}
-	otoCtx, readyChan, err := oto.NewContext(decoder.SampleRate(), 2, 2)
-	if err != nil {
-		return err
-	}
-	<-readyChan
-	player := otoCtx.NewPlayer(decoder)
-	player.Play()
-	for player.IsPlaying() {
-		if t.IsStopped() {
-			return player.Close()
-		}
-		time.Sleep(time.Millisecond)
-	}
-	return player.Close()
 }