From bbb18c36140bd23a2a0b66b0918b77a1fed0f558 Mon Sep 17 00:00:00 2001
From: eeeXun <sdes96303@gmail.com>
Date: Sun, 19 Feb 2023 02:46:33 +0800
Subject: [PATCH] feat: tts

---
 .../translate/reversotranslate/language.go    | 35 ++++---
 internal/translate/reversotranslate/tts.go    | 98 +++++++++++--------
 2 files changed, 78 insertions(+), 55 deletions(-)

diff --git a/internal/translate/reversotranslate/language.go b/internal/translate/reversotranslate/language.go
index b516a97..750f255 100644
--- a/internal/translate/reversotranslate/language.go
+++ b/internal/translate/reversotranslate/language.go
@@ -58,20 +58,25 @@ var (
 		"Ukrainian":            "ukr",
 	}
 	voiceName = map[string]string{
-		"Arabic":     "Mehdi22k",
-		"Chinese":    "Lulu22k",
-		"Dutch":      "Femke22k",
-		"English":    "Heather22k",
-		"French":     "Alice22k",
-		"German":     "Claudia22k",
-		"Hebrew":     "he-IL-Asaf",
-		"Italian":    "Chiara22k",
-		"Japanese":   "Sakura22k",
-		"Polish":     "Ania22k",
-		"Portuguese": "Celia22k",
-		"Romanian":   "ro-RO-Andrei",
-		"Russian":    "Alyona22k",
-		"Spanish":    "Ines22k",
-		"Turkish":    "Ipek22k",
+		"Arabic":               "Mehdi22k",
+		"Chinese (Simplified)": "Lulu22k",
+		"Czech":                "Eliska22k",
+		"Danish":               "Mette22k",
+		"Dutch":                "Sofie22k",
+		"English":              "Heather22k",
+		"French":               "Alice22k",
+		"German":               "Andreas22k",
+		"Greek":                "Dimitris22k",
+		"Hebrew":               "he-IL-Asaf",
+		"Italian":              "Fabiana22k",
+		"Japanese":             "Sakura22k",
+		"Korean":               "Minji22k",
+		"Polish":               "Monika22k",
+		"Portuguese":           "Celia22k",
+		"Romanian":             "ro-RO-Andrei",
+		"Russian":              "Alyona22k",
+		"Spanish":              "Antonio22k",
+		"Swedish":              "Emma22k",
+		"Turkish":              "Ipek22k",
 	}
 )
diff --git a/internal/translate/reversotranslate/tts.go b/internal/translate/reversotranslate/tts.go
index 0c15bfd..47362ba 100644
--- a/internal/translate/reversotranslate/tts.go
+++ b/internal/translate/reversotranslate/tts.go
@@ -1,11 +1,23 @@
 // echo 'nice' | base64
-// curl -A "Mozilla/4.0" \
-// 	"https://voice.reverso.net/RestPronunciation.svc/v1/output=json/GetVoiceStream/voiceName=Heather22k?voiceSpeed=80&inputText=bmljZQ==" \
-// 	--output a.mp
+//
+//	curl -A "Mozilla/4.0" \
+//		"https://voice.reverso.net/RestPronunciation.svc/v1/output=json/GetVoiceStream/voiceName=Heather22k?voiceSpeed=80&inputText=bmljZQ==" \
+//		--output a.mp
 package reversotranslate
 
+import (
+	"encoding/base64"
+	"errors"
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/hajimehoshi/go-mp3"
+	"github.com/hajimehoshi/oto/v2"
+)
+
 const (
-	ttsURL = "https://translate.google.com.vn/translate_tts?ie=UTF-8&q=%s&tl=%s&client=tw-ob"
+	ttsURL = "https://voice.reverso.net/RestPronunciation.svc/v1/output=json/GetVoiceStream/voiceName=%s?voiceSpeed=80&inputText=%s"
 )
 
 func (t *ReversoTranslate) LockAvailable() bool {
@@ -21,42 +33,48 @@ func (t *ReversoTranslate) StopTTS() {
 }
 
 func (t *ReversoTranslate) PlayTTS(lang, message string) error {
-	// urlStr := fmt.Sprintf(
-	// 	ttsURL,
-	// 	url.QueryEscape(message),
-	// 	langCode[lang],
-	// )
-	// res, err := http.Get(urlStr)
-	// if err != nil {
-	// 	t.SoundLock.Release()
-	// 	return err
-	// }
-	// decoder, err := mp3.NewDecoder(res.Body)
-	// if err != nil {
-	// 	t.SoundLock.Release()
-	// 	return err
-	// }
-	// otoCtx, readyChan, err := oto.NewContext(decoder.SampleRate(), 2, 2)
-	// if err != nil {
-	// 	t.SoundLock.Release()
-	// 	return err
-	// }
-	// <-readyChan
-	// player := otoCtx.NewPlayer(decoder)
-	// player.Play()
-	// for player.IsPlaying() {
-	// 	if t.SoundLock.Stop {
-	// 		t.SoundLock.Release()
-	// 		return nil
-	// 	} else {
-	// 		time.Sleep(time.Millisecond)
-	// 	}
-	// }
-	// if err = player.Close(); err != nil {
-	// 	t.SoundLock.Release()
-	// 	return err
-	// }
-	//
+	name, ok := voiceName[lang]
+	if !ok {
+		return errors.New(t.EngineName + " does not support text to speech of " + lang)
+	}
+	urlStr := fmt.Sprintf(
+		ttsURL,
+		name,
+		base64.StdEncoding.EncodeToString([]byte(message)),
+	)
+	req, _ := http.NewRequest("GET", urlStr, nil)
+	req.Header.Add("User-Agent", "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36")
+	res, err := http.DefaultClient.Do(req)
+	if err != nil {
+		t.SoundLock.Release()
+		return err
+	}
+	decoder, err := mp3.NewDecoder(res.Body)
+	if err != nil {
+		t.SoundLock.Release()
+		return err
+	}
+	otoCtx, readyChan, err := oto.NewContext(decoder.SampleRate(), 2, 2)
+	if err != nil {
+		t.SoundLock.Release()
+		return err
+	}
+	<-readyChan
+	player := otoCtx.NewPlayer(decoder)
+	player.Play()
+	for player.IsPlaying() {
+		if t.SoundLock.Stop {
+			t.SoundLock.Release()
+			return nil
+		} else {
+			time.Sleep(time.Millisecond)
+		}
+	}
+	if err = player.Close(); err != nil {
+		t.SoundLock.Release()
+		return err
+	}
+
 	t.SoundLock.Release()
 	return nil
 }