mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 22:00:11 +08:00
46 lines
1.8 KiB
Go
46 lines
1.8 KiB
Go
// Package moegoe 日韩 VITS 模型拟声
|
||
package moegoe
|
||
|
||
import (
|
||
"fmt"
|
||
"net/url"
|
||
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
|
||
ctrl "github.com/FloatTech/zbpctrl"
|
||
"github.com/FloatTech/zbputils/control"
|
||
"github.com/FloatTech/zbputils/ctxext"
|
||
)
|
||
|
||
const (
|
||
jpapi = "https://moegoe.azurewebsites.net/api/speak?text=%s&id=%d"
|
||
krapi = "https://moegoe.azurewebsites.net/api/speakkr?text=%s&id=%d"
|
||
)
|
||
|
||
var speakers = map[string]uint{
|
||
"宁宁": 0, "爱瑠": 1, "芳乃": 2, "茉子": 3, "丛雨": 4, "小春": 5, "七海": 6,
|
||
"Sua": 0, "Mimiru": 1, "Arin": 2, "Yeonhwa": 3, "Yuhwa": 4, "Seonbae": 5,
|
||
}
|
||
|
||
func init() {
|
||
en := control.Register("moegoe", &ctrl.Options[*zero.Ctx]{
|
||
DisableOnDefault: false,
|
||
Help: "moegoe\n" +
|
||
"- 让[宁宁|爱瑠|芳乃|茉子|丛雨|小春|七海]说(日语)\n" +
|
||
"- 让[Sua|Mimiru|Arin|Yeonhwa|Yuhwa|Seonbae]说(韩语)",
|
||
}).ApplySingle(ctxext.DefaultSingle)
|
||
en.OnRegex("^让(宁宁|爱瑠|芳乃|茉子|丛雨|小春|七海)说([A-Za-z\\s\\d\u3005\u3040-\u30ff\u4e00-\u9fff\uff11-\uff19\uff21-\uff3a\uff41-\uff5a\uff66-\uff9d.。,，、:：;；!！?？]+)$").Limit(ctxext.LimitByGroup).SetBlock(true).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
text := ctx.State["regex_matched"].([]string)[2]
|
||
id := speakers[ctx.State["regex_matched"].([]string)[1]]
|
||
ctx.SendChain(message.Record(fmt.Sprintf(jpapi, url.QueryEscape(text), id)))
|
||
})
|
||
en.OnRegex("^让(Sua|Mimiru|Arin|Yeonhwa|Yuhwa|Seonbae)说([A-Za-z\\s\\d\u3131-\u3163\uac00-\ud7ff.。,，、:：;；!！?？]+)$").Limit(ctxext.LimitByGroup).SetBlock(true).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
text := ctx.State["regex_matched"].([]string)[2]
|
||
id := speakers[ctx.State["regex_matched"].([]string)[1]]
|
||
ctx.SendChain(message.Record(fmt.Sprintf(krapi, url.QueryEscape(text), id)))
|
||
})
|
||
}
|