feat: append optional space after at (#103)

Co-authored-by: 源文雨 <[email protected]>

Author: Sn0wo2

bot.go

@@ -28,6 +28,7 @@ type Config struct {
 	MaxProcessTime  time.Duration `json:"max_process_time"`   // 事件最大处理时间 (默认4min)
 	MarkMessage     bool          `json:"mark_message"`       // 自动标记消息为已读
 	KeepAtMeMessage bool          `json:"keep_at_me_message"` // 是否保留at me的原始消息
+	AddSpaceAfterAt bool          `json:"at_space"`           // 是否在At消息后没有空格时自动添加空格
 	Driver          []Driver      `json:"-"`                  // 通信驱动
 }
 

context.go

@@ -113,6 +113,21 @@ func (ctx *Ctx) Send(msg interface{}) message.ID {
 
 // SendChain 快捷发送消息/合并转发-消息链
 func (ctx *Ctx) SendChain(msg ...message.Segment) message.ID {
+	if BotConfig.AddSpaceAfterAt && len(msg) > 0 {
+		newMsg := make(message.Message, 0, len(msg)*2)
+		for i := 0; i < len(msg)-1; i++ {
+			newMsg = append(newMsg, msg[i])
+			if msg[i].Type != "at" {
+				continue
+			}
+			if msg[i+1].Type != "text" ||
+				(len(msg[i+1].Data["text"]) > 0 && msg[i+1].Data["text"][0] != ' ') {
+				newMsg = append(newMsg, message.Text(" "))
+			}
+		}
+		newMsg = append(newMsg, msg[len(msg)-1])
+		msg = newMsg
+	}
 	return ctx.Send((message.Message)(msg))
 }