tts

Author: lz1998

service/bot/api_handler.go

@@ -98,7 +98,7 @@ func preProcessGroupSendingMessage(cli *client.QQClient, groupCode int64, m *mes
 }
 
 func HandleSendPrivateMsg(cli *client.QQClient, req *onebot.SendPrivateMsgReq) *onebot.SendPrivateMsgResp {
-	miraiMsg := ProtoMsgToMiraiMsg(req.Message, req.AutoEscape)
+	miraiMsg := ProtoMsgToMiraiMsg(cli, req.Message, req.AutoEscape)
 	sendingMessage := &message.SendingMessage{Elements: miraiMsg}
 	log.Infof("Bot(%d) Private(%d) <- %s", cli.Uin, req.UserId, MiraiMsgToRawMsg(miraiMsg))
 	preProcessPrivateSendingMessage(cli, req.UserId, sendingMessage)
@@ -110,7 +110,7 @@ func HandleSendPrivateMsg(cli *client.QQClient, req *onebot.SendPrivateMsgReq) *
 }
 
 func HandleSendGroupMsg(cli *client.QQClient, req *onebot.SendGroupMsgReq) *onebot.SendGroupMsgResp {
-	miraiMsg := ProtoMsgToMiraiMsg(req.Message, req.AutoEscape)
+	miraiMsg := ProtoMsgToMiraiMsg(cli, req.Message, req.AutoEscape)
 	sendingMessage := &message.SendingMessage{Elements: miraiMsg}
 	log.Infof("Bot(%d) Group(%d) <- %s", cli.Uin, req.GroupId, MiraiMsgToRawMsg(miraiMsg))
 	preProcessGroupSendingMessage(cli, req.GroupId, sendingMessage)
@@ -126,7 +126,7 @@ func HandleSendGroupMsg(cli *client.QQClient, req *onebot.SendGroupMsgReq) *oneb
 }
 
 func HandleSendMsg(cli *client.QQClient, req *onebot.SendMsgReq) *onebot.SendMsgResp {
-	miraiMsg := ProtoMsgToMiraiMsg(req.Message, req.AutoEscape)
+	miraiMsg := ProtoMsgToMiraiMsg(cli, req.Message, req.AutoEscape)
 	sendingMessage := &message.SendingMessage{Elements: miraiMsg}
 	if req.GroupId != 0 {
 		preProcessGroupSendingMessage(cli, req.GroupId, sendingMessage)

service/bot/proto2mirai.go

@@ -5,6 +5,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/Mrs4s/MiraiGo/client"
 	"github.com/Mrs4s/MiraiGo/message"
 	"github.com/ProtobufBot/Go-Mirai-Client/pkg/util"
 	"github.com/ProtobufBot/Go-Mirai-Client/proto_gen/onebot"
@@ -17,7 +18,7 @@ func EmptyText() *message.TextElement {
 }
 
 // 消息列表，不自动把code变成msg
-func ProtoMsgToMiraiMsg(msgList []*onebot.Message, notConvertText bool) []message.IMessageElement {
+func ProtoMsgToMiraiMsg(cli *client.QQClient, msgList []*onebot.Message, notConvertText bool) []message.IMessageElement {
 	containReply := false // 每条消息只能包含一个reply
 	messageChain := make([]message.IMessageElement, 0)
 	for _, protoMsg := range msgList {
@@ -31,7 +32,7 @@ func ProtoMsgToMiraiMsg(msgList []*onebot.Message, notConvertText bool) []messag
 					log.Warnf("text不存在")
 					continue
 				}
-				messageChain = append(messageChain, RawMsgToMiraiMsg(text)...) // 转换xml码
+				messageChain = append(messageChain, RawMsgToMiraiMsg(cli, text)...) // 转换xml码
 			}
 		case "at":
 			messageChain = append(messageChain, ProtoAtToMiraiAt(protoMsg.Data))
@@ -56,6 +57,8 @@ func ProtoMsgToMiraiMsg(msgList []*onebot.Message, notConvertText bool) []messag
 			}
 		case "sleep":
 			ProtoSleep(protoMsg.Data)
+		case "tts":
+			messageChain = append(messageChain, ProtoTtsToMiraiTts(cli, protoMsg.Data))
 		default:
 			log.Errorf("不支持的消息类型 %+v", protoMsg)
 		}
@@ -274,3 +277,23 @@ func ProtoSleep(data map[string]string) {
 	}
 	time.Sleep(time.Duration(ms) * time.Millisecond)
 }
+
+func ProtoTtsToMiraiTts(cli *client.QQClient, data map[string]string) (m message.IMessageElement) {
+	defer func() {
+		if r := recover(); r != nil {
+			log.Warnf("tts 转换失败")
+			m = EmptyText()
+		}
+	}()
+	text, ok := data["text"]
+	if !ok {
+		log.Warnf("text不存在")
+		return EmptyText()
+	}
+	b, err := cli.GetTts(text)
+	if err != nil {
+		log.Warnf("failed to get tts")
+		return EmptyText()
+	}
+	return &message.VoiceElement{Data: b}
+}

service/bot/raw2mirai.go

@@ -1,11 +1,12 @@
 package bot
 
 import (
+	"encoding/xml"
 	"html"
 	"regexp"
 	"strings"
 
-	"encoding/xml"
+	"github.com/Mrs4s/MiraiGo/client"
 	"github.com/Mrs4s/MiraiGo/message"
 	log "github.com/sirupsen/logrus"
 )
@@ -17,7 +18,7 @@ type Node struct {
 
 var re = regexp.MustCompile("<[\\s\\S]+?/>")
 
-func RawMsgToMiraiMsg(str string) []message.IMessageElement {
+func RawMsgToMiraiMsg(cli *client.QQClient, str string) []message.IMessageElement {
 	containReply := false
 	var node Node
 	textList := re.Split(str, -1)
@@ -71,6 +72,8 @@ func RawMsgToMiraiMsg(str string) []message.IMessageElement {
 				}
 			case "sleep":
 				ProtoSleep(attrMap)
+			case "tts":
+				elemList = append(elemList, ProtoTtsToMiraiTts(cli, attrMap))
 			default:
 				log.Warnf("不支持的类型 %s", code)
 				elemList = append(elemList, message.NewText(code))