mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-18 20:50:12 +08:00
286 lines
7.1 KiB
Go
286 lines
7.1 KiB
Go
package aireply
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
|
|
"github.com/RomiChan/syncx"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
|
|
"github.com/FloatTech/AnimeAPI/aireply"
|
|
"github.com/FloatTech/AnimeAPI/tts"
|
|
"github.com/FloatTech/AnimeAPI/tts/baidutts"
|
|
"github.com/FloatTech/AnimeAPI/tts/genshin"
|
|
"github.com/FloatTech/AnimeAPI/tts/lolimi"
|
|
"github.com/FloatTech/AnimeAPI/tts/ttscn"
|
|
ctrl "github.com/FloatTech/zbpctrl"
|
|
"github.com/FloatTech/zbputils/control"
|
|
)
|
|
|
|
// 数据结构: [8 bits] [8 bits] [8 bits]
|
|
// [具体人物] [tts模式] [回复模式]
|
|
|
|
// defaultttsindexkey
|
|
// 数据结构: [8 bits] [8 bits]
|
|
// [具体人物] [tts模式]
|
|
|
|
// [tts模式]: 0~200 genshin 201 baidu 202 ttscn 203 lolimi
|
|
|
|
const (
|
|
baiduttsindex = 201 + iota
|
|
ttscnttsindex
|
|
lolimittsindex
|
|
)
|
|
|
|
// extrattsname is the tts other than genshin vits
|
|
var extrattsname = []string{"百度", "TTSCN", "桑帛云"}
|
|
|
|
var ttscnspeakers = [...]string{
|
|
"晓晓（女 - 年轻人）",
|
|
"云扬（男 - 年轻人）",
|
|
"晓辰（女 - 年轻人 - 抖音热门）",
|
|
"晓涵（女 - 年轻人）",
|
|
"晓墨（女 - 年轻人）",
|
|
"晓秋（女 - 中年人）",
|
|
"晓睿（女 - 老年）",
|
|
"晓双（女 - 儿童）",
|
|
"晓萱（女 - 年轻人）",
|
|
"晓颜（女 - 年轻人）",
|
|
"晓悠（女 - 儿童）",
|
|
"云希（男 - 年轻人 - 抖音热门）",
|
|
"云野（男 - 中年人）",
|
|
"晓梦（女 - 年轻人）",
|
|
"晓伊（女 - 儿童）",
|
|
"晓甄（女 - 年轻人）",
|
|
}
|
|
|
|
const defaultttsindexkey = -2905
|
|
|
|
var (
|
|
原 = newapikeystore("./data/tts/o.txt")
|
|
ཆཏ = newapikeystore("./data/tts/c.txt")
|
|
百 = newapikeystore("./data/tts/b.txt")
|
|
)
|
|
|
|
type replymode []string
|
|
|
|
func (r replymode) setReplyMode(ctx *zero.Ctx, name string) error {
|
|
gid := ctx.Event.GroupID
|
|
if gid == 0 {
|
|
gid = -ctx.Event.UserID
|
|
}
|
|
var ok bool
|
|
var index int64
|
|
for i, s := range r {
|
|
if s == name {
|
|
ok = true
|
|
index = int64(i)
|
|
break
|
|
}
|
|
}
|
|
if !ok {
|
|
return errors.New("no such mode")
|
|
}
|
|
m, ok := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
|
|
if !ok {
|
|
return errors.New("no such plugin")
|
|
}
|
|
return m.SetData(gid, (m.GetData(gid)&^0xff)|(index&0xff))
|
|
}
|
|
|
|
func (r replymode) getReplyMode(ctx *zero.Ctx) aireply.AIReply {
|
|
gid := ctx.Event.GroupID
|
|
if gid == 0 {
|
|
gid = -ctx.Event.UserID
|
|
}
|
|
m, ok := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
|
|
if ok {
|
|
switch m.GetData(gid) & 0xff {
|
|
case 0:
|
|
return aireply.NewLolimiAi(aireply.JingfengURL, aireply.JingfengBotName)
|
|
case 1:
|
|
return aireply.NewLolimiAi(aireply.MomoURL, aireply.MomoBotName)
|
|
case 2:
|
|
return aireply.NewQYK(aireply.QYKURL, aireply.QYKBotName)
|
|
case 3:
|
|
return aireply.NewXiaoAi(aireply.XiaoAiURL, aireply.XiaoAiBotName)
|
|
case 4:
|
|
k := ཆཏ.k
|
|
if k != "" {
|
|
return aireply.NewChatGPT(aireply.ChatGPTURL, k)
|
|
}
|
|
return aireply.NewLolimiAi(aireply.JingfengURL, aireply.JingfengBotName)
|
|
}
|
|
}
|
|
return aireply.NewLolimiAi(aireply.JingfengURL, aireply.JingfengBotName)
|
|
}
|
|
|
|
var ttsins = func() map[string]tts.TTS {
|
|
m := make(map[string]tts.TTS, 512)
|
|
for _, mode := range append(genshin.SoundList[:], extrattsname...) {
|
|
m[mode] = nil
|
|
}
|
|
return m
|
|
}()
|
|
|
|
var ttsModes = func() []string {
|
|
s := append(genshin.SoundList[:], make([]string, baiduttsindex-len(genshin.SoundList))...) // 0-200
|
|
s = append(s, extrattsname...) // 201 202 ...
|
|
return s
|
|
}()
|
|
|
|
type ttsmode syncx.Map[int64, int64]
|
|
|
|
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 newttsmode() *ttsmode {
|
|
t := &ttsmode{}
|
|
m, ok := control.Lookup("tts")
|
|
(*syncx.Map[int64, int64])(t).Store(defaultttsindexkey, 0)
|
|
if ok {
|
|
index := m.GetData(defaultttsindexkey)
|
|
msk := index & 0xff
|
|
if msk >= 0 && (msk < int64(len(ttsModes))) {
|
|
(*syncx.Map[int64, int64])(t).Store(defaultttsindexkey, index)
|
|
}
|
|
}
|
|
return t
|
|
}
|
|
|
|
func (t *ttsmode) setSoundMode(ctx *zero.Ctx, name string, character int) error {
|
|
gid := ctx.Event.GroupID
|
|
if gid == 0 {
|
|
gid = -ctx.Event.UserID
|
|
}
|
|
_, ok := ttsins[name]
|
|
if !ok {
|
|
return errors.New("不支持设置语音人物" + name)
|
|
}
|
|
var index = int64(-1)
|
|
for i, s := range genshin.SoundList {
|
|
if s == name {
|
|
index = int64(i + 1)
|
|
break
|
|
}
|
|
}
|
|
if index == -1 {
|
|
switch name {
|
|
case extrattsname[0]:
|
|
index = baiduttsindex
|
|
case extrattsname[1]:
|
|
index = ttscnttsindex
|
|
case extrattsname[2]:
|
|
index = lolimittsindex
|
|
default:
|
|
return errors.New("语音人物" + name + "未注册index")
|
|
}
|
|
}
|
|
m := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
|
|
//按原来的逻辑map存的是前16位
|
|
storeIndex := (m.GetData(gid) &^ 0xffff00) | ((index << 8) & 0xff00) | ((int64(character) << 16) & 0xff0000)
|
|
(*syncx.Map[int64, int64])(t).Store(gid, (storeIndex>>8)&0xffff)
|
|
return m.SetData(gid, storeIndex)
|
|
}
|
|
|
|
func (t *ttsmode) getSoundMode(ctx *zero.Ctx) (tts.TTS, error) {
|
|
gid := ctx.Event.GroupID
|
|
if gid == 0 {
|
|
gid = -ctx.Event.UserID
|
|
}
|
|
i, ok := (*syncx.Map[int64, int64])(t).Load(gid)
|
|
if !ok {
|
|
m := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
|
|
i = m.GetData(gid) >> 8
|
|
}
|
|
m := i & 0xff
|
|
if m <= 0 || (m >= int64(len(ttsModes))) {
|
|
i, _ = (*syncx.Map[int64, int64])(t).Load(defaultttsindexkey)
|
|
if i == 0 {
|
|
i = ctx.State["manager"].(*ctrl.Control[*zero.Ctx]).GetData(defaultttsindexkey)
|
|
(*syncx.Map[int64, int64])(t).Store(defaultttsindexkey, i)
|
|
}
|
|
m = i & 0xff
|
|
}
|
|
mode := ttsModes[m]
|
|
ins, ok := ttsins[mode]
|
|
if !ok || ins == nil {
|
|
switch mode {
|
|
case extrattsname[0]:
|
|
id, sec, _ := strings.Cut(百.k, ",")
|
|
ins = baidutts.NewBaiduTTS(int(i&0xff00)>>8, id, sec)
|
|
case extrattsname[1]:
|
|
var err error
|
|
ins, err = ttscn.NewTTSCN("中文（普通话，简体）", ttscnspeakers[int(i&0xff00)>>8], ttscn.KBRates[0])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
case extrattsname[2]:
|
|
ins = lolimi.NewLolimi(int(i&0xff00) >> 8)
|
|
default: // 原神
|
|
k := 原.k
|
|
if k != "" {
|
|
ins = genshin.NewGenshin(int(m-1), 原.k)
|
|
ttsins[mode] = ins
|
|
} else {
|
|
ins = lolimi.NewLolimi(int(i&0xff00) >> 8)
|
|
}
|
|
}
|
|
}
|
|
return ins, nil
|
|
}
|
|
|
|
func (t *ttsmode) resetSoundMode(ctx *zero.Ctx) error {
|
|
gid := ctx.Event.GroupID
|
|
if gid == 0 {
|
|
gid = -ctx.Event.UserID
|
|
}
|
|
m := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
|
|
// 只保留后面8位
|
|
(*syncx.Map[int64, int64])(t).Delete(gid)
|
|
return m.SetData(gid, (m.GetData(gid) & 0xff)) // 重置数据
|
|
}
|
|
|
|
func (t *ttsmode) setDefaultSoundMode(name string, character int) error {
|
|
_, ok := ttsins[name]
|
|
if !ok {
|
|
return errors.New("不支持设置语音人物" + name)
|
|
}
|
|
index := int64(-1)
|
|
for i, s := range genshin.SoundList {
|
|
if s == name {
|
|
index = int64(i + 1)
|
|
break
|
|
}
|
|
}
|
|
if index == -1 {
|
|
switch name {
|
|
case extrattsname[0]:
|
|
index = baiduttsindex
|
|
case extrattsname[1]:
|
|
index = ttscnttsindex
|
|
case extrattsname[2]:
|
|
index = lolimittsindex
|
|
default:
|
|
return errors.New("语音人物" + name + "未注册index")
|
|
}
|
|
}
|
|
m, ok := control.Lookup("tts")
|
|
if !ok {
|
|
return errors.New("[tts] service not found")
|
|
}
|
|
storeIndex := (index & 0xff) | ((int64(character) << 8) & 0xff00)
|
|
(*syncx.Map[int64, int64])(t).Store(defaultttsindexkey, storeIndex)
|
|
return m.SetData(defaultttsindexkey, storeIndex)
|
|
}
|