Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2026-02-13 02:30:26 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix tts

Browse Source
This commit is contained in:
源文雨
2022-12-14 20:26:37 +08:00
parent b6026ee76a
commit e9e0498b34
2 changed files with 25 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
plugin/ai_reply/ai_tts.go
View File

@@ -3,8 +3,9 @@ package aireply
import (
"errors"
"net/url"
"sync"
"github.com/RomiChan/syncx"
"github.com/sirupsen/logrus"
zero "github.com/wdvxdr1123/ZeroBot"
"github.com/FloatTech/AnimeAPI/aireply"
@@ -99,9 +100,8 @@ var ttsModes = func() []string {
}()
type ttsmode struct {
sync.Mutex `json:"-"`
APIKey string // APIKey is for genshin vits
mode map[int64]int64 `json:"-"` // mode grp index
APIKey string // APIKey is for genshin vits
mode syncx.Map[int64, int64] `json:"-"` // mode grp index
}
func list(list []string, num int) string {
@@ -119,16 +119,14 @@ func list(list []string, num int) string {
func newttsmode() *ttsmode {
t := &ttsmode{}
t.Lock()
defer t.Unlock()
m, ok := control.Lookup("tts")
t.mode = make(map[int64]int64, 2*len(genshin.SoundList))
t.mode[defaultttsindexkey] = 0
t.mode = syncx.Map[int64, int64]{}
t.mode.Store(defaultttsindexkey, 0)
if ok {
index := m.GetData(defaultttsindexkey)
msk := index & 0xff
if msk >= 0 && (msk < int64(len(genshin.SoundList)) || msk == baiduttsindex || msk == mockingbirdttsindex) {
t.mode[defaultttsindexkey] = index
t.mode.Store(defaultttsindexkey, index)
}
}
return t
@@ -136,27 +134,20 @@ func newttsmode() *ttsmode {
func (t *ttsmode) getAPIKey(ctx *zero.Ctx) string {
if t.APIKey == "" {
t.Lock()
m := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
gid := ctx.Event.GroupID
if gid == 0 {
gid = -ctx.Event.UserID
}
_ = m.Manager.GetExtra(gid, &t)
t.Unlock()
}
logrus.Debugln("[tts] get api key:", t.APIKey)
return url.QueryEscape(t.APIKey)
}
func (t *ttsmode) setAPIKey(m *ctrl.Control[*zero.Ctx], grp int64, key string) error {
t.Lock()
defer t.Unlock()
err := m.Manager.SetExtra(grp, &key)
if err != nil {
return err
}
t.APIKey = key
return nil
return m.Manager.SetExtra(grp, t)
}
func (t *ttsmode) setSoundMode(ctx *zero.Ctx, name string, baiduper, mockingsynt int) error {
@@ -186,9 +177,7 @@ func (t *ttsmode) setSoundMode(ctx *zero.Ctx, name string, baiduper, mockingsynt
}
}
m := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
t.Lock()
defer t.Unlock()
t.mode[gid] = index
t.mode.Store(gid, index)
return m.SetData(gid, (m.GetData(gid)&^0xffff00)|((index<<8)&0xff00)|((int64(baiduper)<<16)&0x0f0000)|((int64(mockingsynt)<<20)&0xf00000))
}
@@ -197,16 +186,14 @@ func (t *ttsmode) getSoundMode(ctx *zero.Ctx) (tts.TTS, error) {
if gid == 0 {
gid = -ctx.Event.UserID
}
t.Lock()
defer t.Unlock()
i, ok := t.mode[gid]
i, ok := t.mode.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(genshin.SoundList)) && m != baiduttsindex && m != mockingbirdttsindex) {
i = t.mode[defaultttsindexkey]
i, _ = t.mode.Load(defaultttsindexkey)
m = i & 0xff
}
mode := ttsModes[m]
@@ -222,8 +209,13 @@ func (t *ttsmode) getSoundMode(ctx *zero.Ctx) (tts.TTS, error) {
return nil, err
}
default: // 原神
ins = genshin.NewGenshin(int(m), t.getAPIKey(ctx))
ttsins[mode] = ins
k := t.getAPIKey(ctx)
if k != "" {
ins = genshin.NewGenshin(int(m), t.getAPIKey(ctx))
ttsins[mode] = ins
} else {
return nil, errors.New("no valid speaker")
}
}
}
return ins, nil
@@ -234,8 +226,6 @@ func (t *ttsmode) resetSoundMode(ctx *zero.Ctx) error {
if gid == 0 {
gid = -ctx.Event.UserID
}
t.Lock()
defer t.Unlock()
m := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
index := m.GetData(defaultttsindexkey)
return m.SetData(gid, (m.GetData(gid)&0xff)|((index&^0xff)<<8)) // 重置数据
@@ -263,12 +253,10 @@ func (t *ttsmode) setDefaultSoundMode(name string, baiduper, mockingsynt int) er
return errors.New("语音人物" + name + "未注册index")
}
}
t.Lock()
defer t.Unlock()
m, ok := control.Lookup("tts")
if !ok {
return errors.New("[tts] service not found")
}
t.mode[defaultttsindexkey] = index
t.mode.Store(defaultttsindexkey, index)
return m.SetData(defaultttsindexkey, (index&0xff)|((int64(baiduper)<<8)&0x0f00)|((int64(mockingsynt)<<12)&0xf000))
}