消息模板匹配器

engine.go

@@ -79,6 +79,14 @@ func OnMessage(rules ...Rule) *Matcher { return On("message", rules...) }
 // OnMessage 消息触发器
 func (e *Engine) OnMessage(rules ...Rule) *Matcher { return e.On("message", rules...) }
 
+// OnPattern 消息模板触发器
+func OnPattern(rules ...PatternSegment) *Matcher { return On("message", PatternRule(rules...)) }
+
+// OnPattern 消息模板触发器
+func (e *Engine) OnPattern(rules ...PatternSegment) *Matcher {
+	return e.On("message", PatternRule(rules...))
+}
+
 // OnNotice 系统提示触发器
 func OnNotice(rules ...Rule) *Matcher { return On("notice", rules...) }
 

rules.go

@@ -11,6 +11,11 @@ import (
 	"github.com/wdvxdr1123/ZeroBot/utils/helper"
 )
 
+const (
+	KEY_REGEX   = "regex_matched"
+	KEY_PATTERN = "pattern_matched"
+)
+
 // Type check the ctx.Event's type
 func Type(type_ string) Rule {
 	t := strings.SplitN(type_, "/", 3)
@@ -122,6 +127,79 @@ func RegexRule(regexPattern string) Rule {
 	}
 }
 
+type PatternSegment struct {
+	Type    string
+	Matcher func(ctx *Ctx, msg message.MessageSegment) bool
+}
+type Pattern []PatternSegment
+
+// PatternText KEY_PATTERN type []string
+func PatternText(regex string) PatternSegment {
+	re := regexp.MustCompile(regex)
+	return PatternSegment{
+		Type: "text",
+		Matcher: func(ctx *Ctx, msg message.MessageSegment) bool {
+			s := msg.Data["text"]
+			s = strings.Trim(s, " \n\r\t")
+			matchString := re.MatchString(s)
+			if matchString {
+				if _, ok := ctx.State[KEY_PATTERN]; !ok {
+					ctx.State[KEY_PATTERN] = make([]interface{}, 0)
+				}
+
+				ctx.State[KEY_PATTERN] = append(ctx.State[KEY_PATTERN].([]interface{}), re.FindStringSubmatch(s))
+			}
+			return matchString
+		},
+	}
+}
+
+// PatternAt KEY_PATTERN type string
+func PatternAt() PatternSegment {
+	return PatternSegment{
+		Type: "at",
+		Matcher: func(ctx *Ctx, msg message.MessageSegment) bool {
+			if _, ok := ctx.State[KEY_PATTERN]; !ok {
+				ctx.State[KEY_PATTERN] = make([]interface{}, 0)
+			}
+			ctx.State[KEY_PATTERN] = append(ctx.State[KEY_PATTERN].([]interface{}), msg.Data["qq"])
+			return true
+		},
+	}
+}
+
+// PatternImage KEY_PATTERN type msg.Data
+func PatternImage() PatternSegment {
+	return PatternSegment{
+		Type: "image",
+		Matcher: func(ctx *Ctx, msg message.MessageSegment) bool {
+			if _, ok := ctx.State[KEY_PATTERN]; !ok {
+				ctx.State[KEY_PATTERN] = make([]interface{}, 0)
+			}
+			ctx.State[KEY_PATTERN] = append(ctx.State[KEY_PATTERN].([]interface{}), msg.Data)
+			return true
+		},
+	}
+}
+func patternMatch(ctx *Ctx, pattern []PatternSegment, msgs []message.MessageSegment) bool {
+	if len(pattern) != len(msgs) {
+		return false
+	}
+	for i := 0; i < len(pattern); i++ {
+		if pattern[i].Type != (msgs[i].Type) || !pattern[i].Matcher(ctx, msgs[i]) {
+			return false
+		}
+	}
+	return true
+}
+
+// PatternRule check if the message can be matched by the pattern
+func PatternRule(pattern ...PatternSegment) Rule {
+	return func(ctx *Ctx) bool {
+		return patternMatch(ctx, pattern, ctx.Event.Message)
+	}
+}
+
 // ReplyRule check if the message is replying some message
 func ReplyRule(messageID int64) Rule {
 	return func(ctx *Ctx) bool {