mirror of
https://github.com/Mrs4s/go-cqhttp.git
synced 2025-05-05 11:33:48 +08:00
feature new cqcode converter
This commit is contained in:
Mrs4s
parent
6599bf6aa6
commit
09eeff3ce6
110
coolq/cqcode.go
110
coolq/cqcode.go
@ -316,65 +316,69 @@ func ToStringMessage(e []message.IMessageElement, code int64, raw ...bool) (r st
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (bot *CQBot) ConvertStringMessage(m string, group bool) (r []message.IMessageElement) {
|
func (bot *CQBot) ConvertStringMessage(m string, group bool) (r []message.IMessageElement) {
|
||||||
i := matchReg.FindAllStringSubmatchIndex(m, -1)
|
c := newCQCodeConverter(m, bot, group)
|
||||||
si := 0
|
return c.convert()
|
||||||
for _, idx := range i {
|
/*
|
||||||
if idx[0] > si {
|
i := matchReg.FindAllStringSubmatchIndex(m, -1)
|
||||||
text := m[si:idx[0]]
|
si := 0
|
||||||
r = append(r, message.NewText(CQCodeUnescapeText(text)))
|
for _, idx := range i {
|
||||||
}
|
if idx[0] > si {
|
||||||
code := m[idx[0]:idx[1]]
|
text := m[si:idx[0]]
|
||||||
si = idx[1]
|
r = append(r, message.NewText(CQCodeUnescapeText(text)))
|
||||||
t := typeReg.FindAllStringSubmatch(code, -1)[0][1]
|
}
|
||||||
ps := paramReg.FindAllStringSubmatch(code, -1)
|
code := m[idx[0]:idx[1]]
|
||||||
d := make(map[string]string)
|
si = idx[1]
|
||||||
for _, p := range ps {
|
t := typeReg.FindAllStringSubmatch(code, -1)[0][1]
|
||||||
d[p[1]] = CQCodeUnescapeValue(p[2])
|
ps := paramReg.FindAllStringSubmatch(code, -1)
|
||||||
}
|
d := make(map[string]string)
|
||||||
if t == "reply" {
|
for _, p := range ps {
|
||||||
if len(r) > 0 {
|
d[p[1]] = CQCodeUnescapeValue(p[2])
|
||||||
if _, ok := r[0].(*message.ReplyElement); ok {
|
}
|
||||||
log.Warnf("警告: 一条信息只能包含一个 Reply 元素.")
|
if t == "reply" {
|
||||||
continue
|
if len(r) > 0 {
|
||||||
|
if _, ok := r[0].(*message.ReplyElement); ok {
|
||||||
|
log.Warnf("警告: 一条信息只能包含一个 Reply 元素.")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mid, err := strconv.Atoi(d["id"])
|
||||||
|
if err == nil {
|
||||||
|
org := bot.GetMessage(int32(mid))
|
||||||
|
if org != nil {
|
||||||
|
r = append([]message.IMessageElement{
|
||||||
|
&message.ReplyElement{
|
||||||
|
ReplySeq: org["message-id"].(int32),
|
||||||
|
Sender: org["sender"].(message.Sender).Uin,
|
||||||
|
Time: org["time"].(int32),
|
||||||
|
Elements: bot.ConvertStringMessage(org["message"].(string), group),
|
||||||
|
},
|
||||||
|
}, r...)
|
||||||
|
continue
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
mid, err := strconv.Atoi(d["id"])
|
elem, err := bot.ToElement(t, d, group)
|
||||||
if err == nil {
|
if err != nil {
|
||||||
org := bot.GetMessage(int32(mid))
|
if !IgnoreInvalidCQCode {
|
||||||
if org != nil {
|
log.Warnf("转换CQ码 %v 到MiraiGo Element时出现错误: %v 将原样发送.", code, err)
|
||||||
r = append([]message.IMessageElement{
|
r = append(r, message.NewText(code))
|
||||||
&message.ReplyElement{
|
} else {
|
||||||
ReplySeq: org["message-id"].(int32),
|
log.Warnf("转换CQ码 %v 到MiraiGo Element时出现错误: %v 将忽略.", code, err)
|
||||||
Sender: org["sender"].(message.Sender).Uin,
|
|
||||||
Time: org["time"].(int32),
|
|
||||||
Elements: bot.ConvertStringMessage(org["message"].(string), group),
|
|
||||||
},
|
|
||||||
}, r...)
|
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
switch i := elem.(type) {
|
||||||
|
case message.IMessageElement:
|
||||||
|
r = append(r, i)
|
||||||
|
case []message.IMessageElement:
|
||||||
|
r = append(r, i...)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
elem, err := bot.ToElement(t, d, group)
|
if si != len(m) {
|
||||||
if err != nil {
|
r = append(r, message.NewText(CQCodeUnescapeText(m[si:])))
|
||||||
if !IgnoreInvalidCQCode {
|
|
||||||
log.Warnf("转换CQ码 %v 到MiraiGo Element时出现错误: %v 将原样发送.", code, err)
|
|
||||||
r = append(r, message.NewText(code))
|
|
||||||
} else {
|
|
||||||
log.Warnf("转换CQ码 %v 到MiraiGo Element时出现错误: %v 将忽略.", code, err)
|
|
||||||
}
|
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
switch i := elem.(type) {
|
return
|
||||||
case message.IMessageElement:
|
*/
|
||||||
r = append(r, i)
|
|
||||||
case []message.IMessageElement:
|
|
||||||
r = append(r, i...)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if si != len(m) {
|
|
||||||
r = append(r, message.NewText(CQCodeUnescapeText(m[si:])))
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (bot *CQBot) ConvertObjectMessage(m gjson.Result, group bool) (r []message.IMessageElement) {
|
func (bot *CQBot) ConvertObjectMessage(m gjson.Result, group bool) (r []message.IMessageElement) {
|
||||||
|
|||||||
166
coolq/cqcode_c.go
Normal file
166
coolq/cqcode_c.go
Normal file
@ -0,0 +1,166 @@
|
|||||||
|
package coolq
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"github.com/Mrs4s/MiraiGo/message"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
"math"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CQCodeConverter struct {
|
||||||
|
message []rune
|
||||||
|
bot *CQBot
|
||||||
|
group bool
|
||||||
|
|
||||||
|
index int
|
||||||
|
stat state
|
||||||
|
elements []message.IMessageElement
|
||||||
|
tempText []rune
|
||||||
|
cqCode []rune
|
||||||
|
}
|
||||||
|
|
||||||
|
type state uint8
|
||||||
|
|
||||||
|
const (
|
||||||
|
S0 state = iota
|
||||||
|
S1
|
||||||
|
)
|
||||||
|
|
||||||
|
func newCQCodeConverter(message string, bot *CQBot, group bool) *CQCodeConverter {
|
||||||
|
return &CQCodeConverter{message: []rune(message), bot: bot, group: group, stat: S0}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) hasNext() bool {
|
||||||
|
return c.index < len(c.message)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) next() rune {
|
||||||
|
r := c.message[c.index]
|
||||||
|
c.index++
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) move(steps int) {
|
||||||
|
c.index += steps
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) peek(idx int) rune {
|
||||||
|
return c.message[idx]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) peekN(count int) string {
|
||||||
|
lastIdx := int(math.Min(float64(c.index+count), float64(len(c.message)-1)))
|
||||||
|
return string(c.message[c.index:lastIdx])
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) isCQCodeBegin(r rune) bool {
|
||||||
|
return r == '[' && c.peekN(3) == "CQ:"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) saveTempText() {
|
||||||
|
if len(c.tempText) != 0 {
|
||||||
|
c.elements = append(c.elements, message.NewText(CQCodeUnescapeValue(string(c.tempText))))
|
||||||
|
}
|
||||||
|
c.tempText = []rune{}
|
||||||
|
c.cqCode = []rune{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) saveCQCode() {
|
||||||
|
defer func() {
|
||||||
|
c.cqCode = []rune{}
|
||||||
|
c.tempText = []rune{}
|
||||||
|
}()
|
||||||
|
reader := strings.NewReader(string(c.cqCode))
|
||||||
|
buf := bufio.NewReader(reader)
|
||||||
|
t, _ := buf.ReadString(',')
|
||||||
|
t = t[0 : len(t)-1]
|
||||||
|
params := make(map[string]string)
|
||||||
|
for buf.Buffered() > 0 {
|
||||||
|
p, _ := buf.ReadString(',')
|
||||||
|
if strings.HasSuffix(p, ",") {
|
||||||
|
p = p[0 : len(p)-1]
|
||||||
|
}
|
||||||
|
p = strings.TrimSpace(p)
|
||||||
|
if p == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
data := strings.SplitN(p, "=", 2)
|
||||||
|
if len(data) == 2 {
|
||||||
|
params[data[0]] = data[1]
|
||||||
|
} else {
|
||||||
|
params[p] = ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if t == "reply" { // reply 特殊处理
|
||||||
|
if len(c.elements) > 0 {
|
||||||
|
if _, ok := c.elements[0].(*message.ReplyElement); ok {
|
||||||
|
log.Warnf("警告: 一条信息只能包含一个 Reply 元素.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mid, err := strconv.Atoi(params["id"])
|
||||||
|
if err == nil {
|
||||||
|
org := c.bot.GetMessage(int32(mid))
|
||||||
|
if org != nil {
|
||||||
|
c.elements = append([]message.IMessageElement{
|
||||||
|
&message.ReplyElement{
|
||||||
|
ReplySeq: org["message-id"].(int32),
|
||||||
|
Sender: org["sender"].(message.Sender).Uin,
|
||||||
|
Time: org["time"].(int32),
|
||||||
|
Elements: c.bot.ConvertStringMessage(org["message"].(string), c.group),
|
||||||
|
},
|
||||||
|
}, c.elements...)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
elem, err := c.bot.ToElement(t, params, c.group)
|
||||||
|
if err != nil {
|
||||||
|
org := "[" + string(c.cqCode) + "]"
|
||||||
|
if !IgnoreInvalidCQCode {
|
||||||
|
log.Warnf("转换CQ码 %v 时出现错误: %v 将原样发送.", org, err)
|
||||||
|
c.elements = append(c.elements, message.NewText(org))
|
||||||
|
} else {
|
||||||
|
log.Warnf("转换CQ码 %v 时出现错误: %v 将忽略.", org, err)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
switch i := elem.(type) {
|
||||||
|
case message.IMessageElement:
|
||||||
|
c.elements = append(c.elements, i)
|
||||||
|
case []message.IMessageElement:
|
||||||
|
c.elements = append(c.elements, i...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *CQCodeConverter) convert() []message.IMessageElement {
|
||||||
|
for c.hasNext() {
|
||||||
|
ch := c.next()
|
||||||
|
switch c.stat {
|
||||||
|
case S0:
|
||||||
|
if c.isCQCodeBegin(ch) {
|
||||||
|
c.saveTempText()
|
||||||
|
c.tempText = append(c.tempText, []rune("[CQ:")...)
|
||||||
|
c.move(3)
|
||||||
|
c.stat = S1
|
||||||
|
} else {
|
||||||
|
c.tempText = append(c.tempText, ch)
|
||||||
|
}
|
||||||
|
case S1:
|
||||||
|
if c.isCQCodeBegin(ch) {
|
||||||
|
c.move(-1)
|
||||||
|
c.stat = S0
|
||||||
|
} else if ch == ']' {
|
||||||
|
c.saveCQCode()
|
||||||
|
c.stat = S0
|
||||||
|
} else {
|
||||||
|
c.cqCode = append(c.cqCode, ch)
|
||||||
|
c.tempText = append(c.tempText, ch)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
c.saveTempText()
|
||||||
|
return c.elements
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user