shkong/go-cqhttp
1
0
Fork 0
mirror of https://github.com/Mrs4s/go-cqhttp.git synced 2025-05-05 19:43:49 +08:00
Code Issues Projects Releases Wiki Activity

remove convert

Browse Source
This commit is contained in:
wdvxdr 2020-12-03 22:19:46 +08:00
parent 947d165ac5
commit 70e8ea07bd
2 changed files with 31 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
coolq/bot.go
View File

@ -18,7 +18,6 @@ import (
jsoniter "github.com/json-iterator/go" jsoniter "github.com/json-iterator/go"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"github.com/tidwall/gjson"
) )
var json = jsoniter.ConfigCompatibleWithStandardLibrary var json = jsoniter.ConfigCompatibleWithStandardLibrary
@ -392,7 +391,7 @@ func (bot *CQBot) Release() {
} }
func (bot *CQBot) dispatchEventMessage(m MSG) { func (bot *CQBot) dispatchEventMessage(m MSG) {
if global.EventFilter != nil && global.EventFilter.Eval(gjson.Parse(m.ToJson())) == false { if global.EventFilter != nil && global.EventFilter.Eval(global.MSG(m)) == false {
log.Debug("Event filtered!") log.Debug("Event filtered!")
return return
} }

42
global/filter.go
View File

@ -1,6 +1,7 @@
package global package global
import ( import (
"fmt"
"io/ioutil" "io/ioutil"
"regexp" "regexp"
"strings" "strings"
@ -9,8 +10,28 @@ import (
"github.com/tidwall/gjson" "github.com/tidwall/gjson"
) )
type MSG map[string]interface{}
func (m MSG) Get(s string) MSG {
if v,ok := m[s];ok {
if msg,ok := v.(MSG);ok {
return msg
}
return MSG{"__str__": v} // 用这个名字应该没问题吧
}
return MSG{}
}
func (m MSG) String() string {
if str,ok:=m["__str__"];ok {
return fmt.Sprint(str)
}
str, _ := json.MarshalToString(m)
return str
}
type Filter interface { type Filter interface {
Eval(payload gjson.Result) bool Eval(payload MSG) bool
} }
type operationNode struct { type operationNode struct {
@ -31,7 +52,7 @@ func notOperatorConstruct(argument gjson.Result) *NotOperator {
return op return op
} }
func (op *NotOperator) Eval(payload gjson.Result) bool { func (op *NotOperator) Eval(payload MSG) bool {
return !op.operand.Eval(payload) return !op.operand.Eval(payload)
} }
@ -70,7 +91,7 @@ func andOperatorConstruct(argument gjson.Result) *AndOperator {
return op return op
} }
func (andOperator *AndOperator) Eval(payload gjson.Result) bool { func (andOperator *AndOperator) Eval(payload MSG) bool {
res := true res := true
for _, operand := range andOperator.operands { for _, operand := range andOperator.operands {
@ -106,7 +127,7 @@ func orOperatorConstruct(argument gjson.Result) *OrOperator {
return op return op
} }
func (op *OrOperator) Eval(payload gjson.Result) bool { func (op *OrOperator) Eval(payload MSG) bool {
res := false res := false
for _, operand := range op.operands { for _, operand := range op.operands {
res = res || operand.Eval(payload) res = res || operand.Eval(payload)
@ -127,7 +148,7 @@ func equalOperatorConstruct(argument gjson.Result) *EqualOperator {
return op return op
} }
func (op *EqualOperator) Eval(payload gjson.Result) bool { func (op *EqualOperator) Eval(payload MSG) bool {
return payload.String() == op.operand return payload.String() == op.operand
} }
@ -141,7 +162,7 @@ func notEqualOperatorConstruct(argument gjson.Result) *NotEqualOperator {
return op return op
} }
func (op *NotEqualOperator) Eval(payload gjson.Result) bool { func (op *NotEqualOperator) Eval(payload MSG) bool {
return !(payload.String() == op.operand) return !(payload.String() == op.operand)
} }
@ -167,7 +188,7 @@ func inOperatorConstruct(argument gjson.Result) *InOperator {
return op return op
} }
func (op *InOperator) Eval(payload gjson.Result) bool { func (op *InOperator) Eval(payload MSG) bool {
payloadStr := payload.String() payloadStr := payload.String()
if op.operandArray != nil { if op.operandArray != nil {
for _, value := range op.operandArray { for _, value := range op.operandArray {
@ -193,10 +214,7 @@ func containsOperatorConstruct(argument gjson.Result) *ContainsOperator {
return op return op
} }
func (op *ContainsOperator) Eval(payload gjson.Result) bool { func (op *ContainsOperator) Eval(payload MSG) bool {
if payload.IsObject() || payload.IsArray() {
return false
}
return strings.Contains(payload.String(), op.operand) return strings.Contains(payload.String(), op.operand)
} }
@ -213,7 +231,7 @@ func regexOperatorConstruct(argument gjson.Result) *RegexOperator {
return op return op
} }
func (op *RegexOperator) Eval(payload gjson.Result) bool { func (op *RegexOperator) Eval(payload MSG) bool {
matched := op.regex.MatchString(payload.String()) matched := op.regex.MatchString(payload.String())
return matched return matched
} }