feat: airecord (#1180)

Co-authored-by: 源文雨 <41315874+fumiama@users.noreply.github.com>
2026-02-12 10:10:25 +00:00 · 2025-07-20 13:31:06 +08:00
parent cb0ffa0c17
commit 617d4f50a4
10 changed files with 267 additions and 20 deletions
--- a/plugin/airecord/record.go
+++ b/plugin/airecord/record.go
@@ -0,0 +1,134 @@
+// Package airecord 群应用：AI声聊
+package airecord
+
+import (
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/tidwall/gjson"
+
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+
+	"github.com/FloatTech/AnimeAPI/airecord"
+	ctrl "github.com/FloatTech/zbpctrl"
+	"github.com/FloatTech/zbputils/control"
+)
+
+func init() {
+	en := control.AutoRegister(&ctrl.Options[*zero.Ctx]{
+		DisableOnDefault: false,
+		Extra:            control.ExtraFromString("airecord"),
+		Brief:            "群应用：AI声聊",
+		Help: "- 设置AI语音群号1048452984(tips：机器人任意所在群聊即可)\n" +
+			"- 设置AI语音模型\n" +
+			"- 查看AI语音配置\n" +
+			"- 发送AI语音xxx",
+		PrivateDataFolder: "airecord",
+	})
+
+	en.OnPrefix("设置AI语音群号", zero.OnlyPrivate, zero.SuperUserPermission).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			u := strings.TrimSpace(ctx.State["args"].(string))
+			num, err := strconv.ParseInt(u, 10, 64)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: parse gid err: ", err))
+				return
+			}
+			err = airecord.SetCustomGID(num)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: set gid err: ", err))
+				return
+			}
+			ctx.SendChain(message.Text("设置AI语音群号为", num))
+		})
+	en.OnFullMatch("设置AI语音模型", zero.OnlyPrivate, zero.SuperUserPermission).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			next := zero.NewFutureEvent("message", 999, false, ctx.CheckSession())
+			recv, cancel := next.Repeat()
+			defer cancel()
+			jsonData := ctx.GetAICharacters(0, 1)
+
+			// 转换为字符串数组
+			var names []string
+			// 初始化两个映射表
+			nameToID := make(map[string]string)
+			nameToURL := make(map[string]string)
+			characters := jsonData.Get("#.characters")
+
+			// 遍历每个角色对象
+			characters.ForEach(func(_, group gjson.Result) bool {
+				group.ForEach(func(_, character gjson.Result) bool {
+					// 提取当前角色的三个字段
+					name := character.Get("character_name").String()
+					names = append(names, name)
+					// 存入映射表（重复名称会覆盖，保留最后出现的条目）
+					nameToID[name] = character.Get("character_id").String()
+					nameToURL[name] = character.Get("preview_url").String()
+					return true // 继续遍历
+				})
+				return true // 继续遍历
+			})
+			var builder strings.Builder
+			// 写入开头文本
+			builder.WriteString("请选择语音模型序号：\n")
+
+			// 遍历names数组，拼接序号和名称
+			for i, v := range names {
+				// 将数字转换为字符串（不依赖fmt）
+				numStr := strconv.Itoa(i)
+				// 拼接格式："序号. 名称\n"
+				builder.WriteString(numStr)
+				builder.WriteString(". ")
+				builder.WriteString(v)
+				builder.WriteString("\n")
+			}
+			// 获取最终字符串
+			ctx.SendChain(message.Text(builder.String()))
+			for {
+				select {
+				case <-time.After(time.Second * 120):
+					ctx.SendChain(message.Text("设置AI语音模型指令过期"))
+					return
+				case ct := <-recv:
+					msg := ct.Event.Message.ExtractPlainText()
+					num, err := strconv.Atoi(msg)
+					if err != nil {
+						ctx.SendChain(message.Text("请输入数字!"))
+						continue
+					}
+					if num < 0 || num >= len(names) {
+						ctx.SendChain(message.Text("序号非法!"))
+						continue
+					}
+					err = airecord.SetRecordModel(names[num], nameToID[names[num]])
+					if err != nil {
+						ctx.SendChain(message.Text("ERROR: set model err: ", err))
+						continue
+					}
+					ctx.SendChain(message.Text("已选择语音模型: ", names[num]))
+					ctx.SendChain(message.Record(nameToURL[names[num]]))
+					return
+				}
+			}
+		})
+	en.OnFullMatch("查看AI语音配置", zero.OnlyPrivate, zero.SuperUserPermission).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			ctx.SendChain(message.Text(airecord.PrintRecordConfig()))
+		})
+	en.OnPrefix("发送AI语音", zero.UserOrGrpAdmin).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			u := strings.TrimSpace(ctx.State["args"].(string))
+			recCfg := airecord.GetConfig()
+			record := ctx.GetAIRecord(recCfg.ModelID, recCfg.Customgid, u)
+			if record == "" {
+				id := ctx.SendGroupAIRecord(recCfg.ModelID, ctx.Event.GroupID, u)
+				if id == "" {
+					ctx.SendChain(message.Text("ERROR: get record err: empty record"))
+					return
+				}
+			}
+			ctx.SendChain(message.Record(record))
+		})
+}