diff --git a/plugin/ai_reply/ai_tts.go b/plugin/ai_reply/ai_tts.go index a8ea831c..f9b9ef99 100644 --- a/plugin/ai_reply/ai_tts.go +++ b/plugin/ai_reply/ai_tts.go @@ -1,9 +1,12 @@ package aireply import ( + "fmt" + "net/url" "regexp" "strconv" "sync" + "time" "github.com/pkumza/numcn" log "github.com/sirupsen/logrus" @@ -11,172 +14,246 @@ import ( "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 re = regexp.MustCompile(`(\-|\+)?\d+(\.\d+)?`) +// 每个角色的测试文案 +var testRecord = map[string]string{ + "派蒙": "哎，又是看不懂的东西。我完全不知道这些奇怪的问题和实验，能得到什么结果…", + "凯亚": "真是个急性子啊你。", + "安柏": "最初的鸟儿是不会飞翔的，飞翔是它们勇敢跃入峡谷的奖励。", + "丽莎": "嗨，小可爱，你是新来的助理吗？", + "琴": "蒲公英骑士，琴，申请入队。", + "香菱": "我是来自璃月的厨师香菱，最擅长的是做各种捞…捞，料理…哎呀，练了那么多次，还是会紧张，嘿。", + "枫原万叶": "飘摇风雨中，带刀归来赤脚行。", + "迪卢克": "在黎明来临之前，总要有人照亮黑暗。", + "温迪": "若你困于无风之地，我将为你奏响高天之歌。", + "可莉": "西风骑士团，火花骑士，可莉，前来报到！…呃—后面该说什么词来着？可莉背不下来啦...", + "早柚": "终末番,早柚，参上。 呼——", + "托马": "初次见面，异乡的旅人，你的名字我可是早就听说了。只要你不嫌弃，我托马，从今天起就是你的朋友了。", + "芭芭拉": "芭芭拉，闪耀登场~治疗就交给我吧，不会让你失望的！", + "优菈": "沉沦是很容易的一件事，但我仍想冻住这股潮流。", + "云堇": "曲高未必人不识，自有知音和清词。", + "钟离": "人间归离复归离，借一浮生逃浮生。", + "魈": "三眼五显仙人，魈，听召，前来守护", + "凝光": "就算古玩价值连城，给人的快乐，也只有刚拥有的一瞬", + "雷电将军": "浮世千百年来风景依旧，人之在世却如白露与泡影。", + "北斗": "不知道如何向前的话，总之先迈出第一步，后面的道路就会自然而然地展开了。", + "甘雨": "这项工作，该划掉了。", + "七七": "椰羊的奶，好喝!比一般的羊奶，好喝!", + "刻晴": "劳逸结合是不错，但也别放松过头。", + "神里绫华": "若知是梦何须醒，不比真如一相会。", + "雷泽": "你是朋友。我和你一起狩猎。", + "神里绫人": "此前听绫华屡次提起阁下，不料公务繁忙，直至今日才有机会相见。", + "罗莎莉亚": "哪怕如今你已经走上截然不同的道路，也不要否认从前的自己，从前的每一个你都是你脚下的基石，不要害怕过去，不要畏惧与它抗衡。", + "阿贝多": "用自己的双脚丈量土地，将未知变为知识。", + "八重神子": "我的神明，就托付给你了。", + "宵宫": "即使只是片刻的火花，也能在仰望黑夜的人心中留下久久不灭的美丽光芒。", + "荒泷一斗": "更好地活下去,绝不该靠牺牲同类换取，应该是,一起更好地活着,才对。", + "九条裟罗": "想要留住雪花。但在手心里，它只会融化的更快。", + "夜兰": "线人来信了，嗯，看来又出现了新的变数。", + "珊瑚宫心海": "成为了现任人神巫女之后，我也慢慢习惯了这样的生活，更重要的是我也因此和你相遇了，不是吗？", + "五郎": "海祇岛反抗军大将，五郎，前来助阵！", + "达达利亚": "许下的诺言就好好遵守，做错了事情就承担责任，这才是家人应有的样子吧。", + "莫娜": "正是因为无法更改，无可违逆，只能接受，命运才会被称之为命运。", + "班尼特": "只要有大家在，伤口就不会痛!", + "申鹤": "不知道你是喜欢人间的灯火，还是山林的月光？", + "行秋": "有时明月无人夜，独向昭潭制恶龙。", + "烟绯": "律法即是约束，也是工具。", + "久岐忍": "有麻烦事要处理的话，直接告诉我就好，我来摆平。", + "辛焱": "马上就要演出了，你也一起来嗨吗？", + "砂糖": "我是砂糖，炼金术的…研究员。", + "胡桃": "阴阳有序，命运无常，死亡难以预测，却也有它的规矩。", + "重云": "我名重云，家族久居璃月，世代以驱邪除魔为业。", + "菲谢尔": "我即断罪之皇女，真名为菲谢尔。应命运的召唤降临在此间——哎？你也是，异世界的旅人吗…？", + "诺艾尔": "我是诺艾尔，西风骑士团的女仆，从今天起会陪你一起去冒险。", + "迪奥娜": "猫尾酒馆的招牌调酒师，迪奥娜，我的出场费可是很贵的。", + "鹿野院平藏": "我叫鹿野院平藏，是天领奉行里破案最多最快的侦探……", +} + +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)) + } + // 发送语音 + if ID := ctx.SendChain(record); ID.ID() == 0 { + 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 - } - 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() }