更换tts的语音api至原神语音api (#382)

plugin/ai_reply/ai_tts.go

@@ -1,182 +1,259 @@
 package aireply
 
 import (
+	"fmt"
+	"net/url"
 	"regexp"
 	"strconv"
 	"sync"
+	"time"
 
 	"github.com/pkumza/numcn"
 	log "github.com/sirupsen/logrus"
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/message"
 
 	"github.com/FloatTech/AnimeAPI/aireply"
-	"github.com/FloatTech/AnimeAPI/tts"
-	"github.com/FloatTech/AnimeAPI/tts/baidutts"
-	"github.com/FloatTech/AnimeAPI/tts/mockingbird"
-	fcext "github.com/FloatTech/floatbox/ctxext"
 	ctrl "github.com/FloatTech/zbpctrl"
 	"github.com/FloatTech/zbputils/control"
 	"github.com/FloatTech/zbputils/ctxext"
 )
 
-const ttsServiceName = "tts"
+const (
+	ttsServiceName = "tts"
+	cnapi          = "http://233366.proxy.nscc-gz.cn:8888?speaker=%s&text=%s"
+	// testString     = "这是测试语言......"
+)
+
+// 每个角色的测试文案
+var testRecord = map[string]string{
+	"派蒙":    "哎，又是看不懂的东西。我完全不知道这些奇怪的问题和实验，能得到什么结果…",
+	"凯亚":    "真是个急性子啊你。",
+	"安柏":    "最初的鸟儿是不会飞翔的，飞翔是它们勇敢跃入峡谷的奖励。",
+	"丽莎":    "嗨，小可爱，你是新来的助理吗？",
+	"琴":     "蒲公英骑士，琴，申请入队。",
+	"香菱":    "我是来自璃月的厨师香菱，最擅长的是做各种捞…捞，料理…哎呀，练了那么多次，还是会紧张，嘿。",
+	"枫原万叶":  "飘摇风雨中，带刀归来赤脚行。",
+	"迪卢克":   "在黎明来临之前，总要有人照亮黑暗。",
+	"温迪":    "若你困于无风之地，我将为你奏响高天之歌。",
+	"可莉":    "西风骑士团，火花骑士，可莉，前来报到！…呃—后面该说什么词来着？可莉背不下来啦...",
+	"早柚":    "终末番,早柚，参上。 呼——",
+	"托马":    "初次见面，异乡的旅人，你的名字我可是早就听说了。只要你不嫌弃，我托马，从今天起就是你的朋友了。",
+	"芭芭拉":   "芭芭拉，闪耀登场~治疗就交给我吧，不会让你失望的！",
+	"优菈":    "沉沦是很容易的一件事，但我仍想冻住这股潮流。",
+	"云堇":    "曲高未必人不识，自有知音和清词。",
+	"钟离":    "人间归离复归离，借一浮生逃浮生。",
+	"魈":     "三眼五显仙人，魈，听召，前来守护",
+	"凝光":    "就算古玩价值连城，给人的快乐，也只有刚拥有的一瞬",
+	"雷电将军":  "浮世千百年来风景依旧，人之在世却如白露与泡影。",
+	"北斗":    "不知道如何向前的话，总之先迈出第一步，后面的道路就会自然而然地展开了。",
+	"甘雨":    "这项工作，该划掉了。",
+	"七七":    "椰羊的奶，好喝!比一般的羊奶，好喝!",
+	"刻晴":    "劳逸结合是不错，但也别放松过头。",
+	"神里绫华":  "若知是梦何须醒，不比真如一相会。",
+	"雷泽":    "你是朋友。我和你一起狩猎。",
+	"神里绫人":  "此前听绫华屡次提起阁下，不料公务繁忙，直至今日才有机会相见。",
+	"罗莎莉亚":  "哪怕如今你已经走上截然不同的道路，也不要否认从前的自己，从前的每一个你都是你脚下的基石，不要害怕过去，不要畏惧与它抗衡。",
+	"阿贝多":   "用自己的双脚丈量土地，将未知变为知识。",
+	"八重神子":  "我的神明，就托付给你了。",
+	"宵宫":    "即使只是片刻的火花，也能在仰望黑夜的人心中留下久久不灭的美丽光芒。",
+	"荒泷一斗":  "更好地活下去,绝不该靠牺牲同类换取，应该是,一起更好地活着,才对。",
+	"九条裟罗":  "想要留住雪花。但在手心里，它只会融化的更快。",
+	"夜兰":    "线人来信了，嗯，看来又出现了新的变数。",
+	"珊瑚宫心海": "成为了现任人神巫女之后，我也慢慢习惯了这样的生活，更重要的是我也因此和你相遇了，不是吗？",
+	"五郎":    "海祇岛反抗军大将，五郎，前来助阵！",
+	"达达利亚":  "许下的诺言就好好遵守，做错了事情就承担责任，这才是家人应有的样子吧。",
+	"莫娜":    "正是因为无法更改，无可违逆，只能接受，命运才会被称之为命运。",
+	"班尼特":   "只要有大家在，伤口就不会痛!",
+	"申鹤":    "不知道你是喜欢人间的灯火，还是山林的月光？",
+	"行秋":    "有时明月无人夜，独向昭潭制恶龙。",
+	"烟绯":    "律法即是约束，也是工具。",
+	"久岐忍":   "有麻烦事要处理的话，直接告诉我就好，我来摆平。",
+	"辛焱":    "马上就要演出了，你也一起来嗨吗？",
+	"砂糖":    "我是砂糖，炼金术的…研究员。",
+	"胡桃":    "阴阳有序，命运无常，死亡难以预测，却也有它的规矩。",
+	"重云":    "我名重云，家族久居璃月，世代以驱邪除魔为业。",
+	"菲谢尔":   "我即断罪之皇女，真名为菲谢尔。应命运的召唤降临在此间——哎？你也是，异世界的旅人吗…？",
+	"诺艾尔":   "我是诺艾尔，西风骑士团的女仆，从今天起会陪你一起去冒险。",
+	"迪奥娜":   "猫尾酒馆的招牌调酒师，迪奥娜，我的出场费可是很贵的。",
+	"鹿野院平藏": "我叫鹿野院平藏，是天领奉行里破案最多最快的侦探……",
+}
 
-var re = regexp.MustCompile(`(\-|\+)?\d+(\.\d+)?`)
+var (
+	re        = regexp.MustCompile(`(\-|\+)?\d+(\.\d+)?`)
+	soundList = [...]string{
+		"派蒙", "凯亚", "安柏", "丽莎", "琴",
+		"香菱", "枫原万叶", "迪卢克", "温迪", "可莉",
+		"早柚", "托马", "芭芭拉", "优菈", "云堇",
+		"钟离", "魈", "凝光", "雷电将军", "北斗",
+		"甘雨", "七七", "刻晴", "神里绫华", "雷泽",
+		"神里绫人", "罗莎莉亚", "阿贝多", "八重神子", "宵宫",
+		"荒泷一斗", "九条裟罗", "夜兰", "珊瑚宫心海", "五郎",
+		"达达利亚", "莫娜", "班尼特", "申鹤", "行秋",
+		"烟绯", "久岐忍", "辛焱", "砂糖", "胡桃",
+		"重云", "菲谢尔", "诺艾尔", "迪奥娜", "鹿野院平藏",
+	}
+)
 
 type ttsInstances struct {
 	sync.RWMutex
-	m map[string]tts.TTS
-	l []string
+	defaultSoundMode string
+	soundMode        []string
 }
 
-func (t *ttsInstances) List() []string {
-	t.RLock()
-	cl := make([]string, len(t.l))
-	_ = copy(cl, t.l)
-	t.RUnlock()
-	return cl
+func list(list []string, num int) string {
+	s := ""
+	for i, value := range list {
+		s += value
+		if (i+1)%num == 0 {
+			s += "\n"
+		} else {
+			s += " | "
+		}
+	}
+	return s
 }
 
 func init() {
-	t := &ttsInstances{
-		m: map[string]tts.TTS{
-			"百度女声":   baidutts.NewBaiduTTS(0),
-			"百度男声":   baidutts.NewBaiduTTS(1),
-			"百度度逍遥":  baidutts.NewBaiduTTS(3),
-			"百度度丫丫":  baidutts.NewBaiduTTS(4),
-			"拟声鸟阿梓":  nil,
-			"拟声鸟文静":  nil,
-			"拟声鸟药水哥": nil,
-		},
-		l: []string{"拟声鸟阿梓", "拟声鸟文静", "拟声鸟药水哥", "百度女声", "百度男声", "百度度逍遥", "百度度丫丫"},
+	tts := &ttsInstances{
+		defaultSoundMode: "派蒙",
+		soundMode:        soundList[:],
 	}
 	engine := control.Register(ttsServiceName, &ctrl.Options[*zero.Ctx]{
 		DisableOnDefault: true,
-		Help: "语音回复(包括拟声鸟和百度)\n" +
+		Help: "语音回复(大家一起来炼丹)\n" +
 			"- @Bot 任意文本(任意一句话回复)\n" +
-			"- 设置语音模式[拟声鸟阿梓 | 拟声鸟文静 | 拟声鸟药水哥 | 百度女声 | 百度男声| 百度度逍遥 | 百度度丫丫]\n" +
-			"- 设置默认语音模式[拟声鸟阿梓 | 拟声鸟文静 | 拟声鸟药水哥 | 百度女声 | 百度男声| 百度度逍遥 | 百度度丫丫]\n",
+			"- 设置语音模式[原神人物]\n" +
+			"- 设置默认语音模式[原神人物]\n" +
+			"当前适用的原神人物含有以下：\n" + list(soundList[:], 5),
 	})
 	engine.OnMessage(zero.OnlyToMe).SetBlock(true).Limit(ctxext.LimitByUser).
 		Handle(func(ctx *zero.Ctx) {
 			msg := ctx.ExtractPlainText()
+			// 获取回复模式
 			r := aireply.NewAIReply(getReplyMode(ctx))
-			tts, err := t.new(t.getSoundMode(ctx))
-			if err != nil {
-				ctx.SendChain(message.Text("ERROR: ", err))
+			// 获取回复的文本
+			reply := r.TalkPlain(msg, zero.BotConfig.NickName[0])
+			// 获取角色
+			name := tts.getSoundMode(ctx)
+			if _, ok := testRecord[name]; !ok {
+				ctx.SendChain(message.Text("配置的语言人物数据丢失！请重新设置语言人物。"))
 				return
 			}
-			var reply string
-			if tts != nil {
-				rec, err := tts.Speak(ctx.Event.UserID, func() string {
-					reply = r.TalkPlain(msg, zero.BotConfig.NickName[0])
-					reply = re.ReplaceAllStringFunc(reply, func(s string) string {
-						f, err := strconv.ParseFloat(s, 64)
-						if err != nil {
-							log.Errorln("[tts]:", err)
-							return s
-						}
-						return numcn.EncodeFromFloat64(f)
-					})
-					log.Debugln("[tts]:", reply)
-					return reply
-				})
-				if err == nil {
-					ctx.SendChain(message.Record(rec))
-				} else {
-					ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(reply))
-				}
+			// 获取语言
+			record := message.Record(fmt.Sprintf(cnapi, url.QueryEscape(name), url.QueryEscape(
+				// 将数字转文字
+				re.ReplaceAllStringFunc(reply, func(s string) string {
+					f, err := strconv.ParseFloat(s, 64)
+					if err != nil {
+						log.Errorln("[tts]:", err)
+						return s
+					}
+					return numcn.EncodeFromFloat64(f)
+				}),
+			))).Add("cache", 0)
+			if record.Data == nil {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(reply))
 			}
-		})
-	engine.OnRegex(`^设置语音模式(.*)$`, fcext.ValueInList(func(ctx *zero.Ctx) string { return ctx.State["regex_matched"].([]string)[1] }, t)).SetBlock(true).
-		Handle(func(ctx *zero.Ctx) {
-			param := ctx.State["regex_matched"].([]string)[1]
-			err := t.setSoundMode(ctx, param)
-			if err != nil {
-				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(err))
-				return
+			// 发送语音
+			if ID := ctx.SendChain(record); ID.ID() == 0 {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(reply))
 			}
-			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("设置成功，当前模式为", param))
-		})
-	engine.OnRegex(`^设置默认语音模式(.*)$`, fcext.ValueInList(func(ctx *zero.Ctx) string { return ctx.State["regex_matched"].([]string)[1] }, t)).SetBlock(true).
-		Handle(func(ctx *zero.Ctx) {
-			param := ctx.State["regex_matched"].([]string)[1]
-			t.setDefaultSoundMode(param)
-			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("设置成功，默认模式为", param))
 		})
-}
-
-// new 语音简单工厂
-func (t *ttsInstances) new(name string) (ts tts.TTS, err error) {
-	t.RLock()
-	ts = t.m[name]
-	t.RUnlock()
-	if ts == nil {
-		switch name {
-		case "拟声鸟阿梓":
-			t.Lock()
-			ts, err = mockingbird.NewMockingBirdTTS(0)
-			t.Unlock()
-		case "拟声鸟文静":
-			t.Lock()
-			ts, err = mockingbird.NewMockingBirdTTS(1)
-			t.Unlock()
-		case "拟声鸟药水哥":
-			t.Lock()
-			ts, err = mockingbird.NewMockingBirdTTS(2)
-			t.Unlock()
+	engine.OnRegex(`^设置语音模式(.*)$`, func(ctx *zero.Ctx) bool {
+		param := ctx.State["regex_matched"].([]string)[1]
+		if _, ok := testRecord[param]; !ok {
+			return false
 		}
-	}
-	return
+		return true
+	}).SetBlock(true).Handle(func(ctx *zero.Ctx) {
+		param := ctx.State["regex_matched"].([]string)[1]
+		// 保存设置
+		err := tts.setSoundMode(ctx, param)
+		if err != nil {
+			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(err))
+			return
+		}
+		// 设置验证
+		name := tts.getSoundMode(ctx)
+		if _, ok := testRecord[name]; !ok {
+			ctx.SendChain(message.Text("配置的语言人物数据丢失！请重新设置语言人物。"))
+			return
+		}
+		record := message.Record(fmt.Sprintf(cnapi, url.QueryEscape(name), url.QueryEscape(testRecord[name]))).Add("cache", 0)
+		if ID := ctx.SendChain(record); ID.ID() == 0 {
+			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("设置失败！无法发送测试语言，请重试。"))
+			return
+		}
+		time.Sleep(time.Second * 2)
+		ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("设置成功，默认模式为", name))
+	})
+	engine.OnRegex(`^设置默认语音模式(.*)$`, func(ctx *zero.Ctx) bool {
+		param := ctx.State["regex_matched"].([]string)[1]
+		if _, ok := testRecord[param]; !ok {
+			return false
+		}
+		return true
+	}).SetBlock(true).Handle(func(ctx *zero.Ctx) {
+		param := ctx.State["regex_matched"].([]string)[1]
+		// 保存设置
+		tts.setDefaultSoundMode(param)
+		// 设置验证
+		name := tts.defaultSoundMode
+		record := message.Record(fmt.Sprintf(cnapi, url.QueryEscape(name), url.QueryEscape(testRecord[name]))).Add("cache", 0)
+		if ID := ctx.SendChain(record); ID.ID() == 0 {
+			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("设置失败！无法发送测试语言，请重试。"))
+			return
+		}
+		time.Sleep(time.Second * 2)
+		ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("设置成功，默认模式为", name))
+	})
 }
 
-func (t *ttsInstances) setSoundMode(ctx *zero.Ctx, name string) error {
+func (tts *ttsInstances) setSoundMode(ctx *zero.Ctx, name string) error {
 	gid := ctx.Event.GroupID
 	if gid == 0 {
 		gid = -ctx.Event.UserID
 	}
 	var index int64
-	t.RLock()
-	for i, s := range t.l {
+	tts.RLock()
+	for i, s := range tts.soundMode {
 		if s == name {
 			index = int64(i)
 			break
 		}
 	}
-	t.RUnlock()
+	tts.RUnlock()
 	m := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
 	return m.SetData(gid, index)
 }
 
-func (t *ttsInstances) getSoundMode(ctx *zero.Ctx) (name string) {
+func (tts *ttsInstances) getSoundMode(ctx *zero.Ctx) (name string) {
 	gid := ctx.Event.GroupID
 	if gid == 0 {
 		gid = -ctx.Event.UserID
 	}
 	m, ok := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
 	if ok {
-		t.RLock()
-		defer t.RUnlock()
+		tts.RLock()
+		defer tts.RUnlock()
 		index := m.GetData(gid)
-		if int(index) < len(t.l) {
-			return t.l[index]
+		if int(index) < len(tts.soundMode) {
+			return tts.soundMode[index]
 		}
 	}
-	return "拟声鸟阿梓"
+	return tts.defaultSoundMode
 }
 
-func (t *ttsInstances) setDefaultSoundMode(name string) {
-	var index int
-	t.RLock()
-	for _, s := range t.l {
+func (tts *ttsInstances) setDefaultSoundMode(name string) {
+	var index int64
+	tts.RLock()
+	for i, s := range tts.soundMode {
 		if s == name {
+			index = int64(i)
 			break
 		}
-		index++
 	}
-	t.RUnlock()
-	t.Lock()
-	t.l[0], t.l[index] = t.l[index], t.l[0]
-	t.Unlock()
+	tts.defaultSoundMode = tts.soundMode[index]
+	tts.RUnlock()
 }