add plugin

matcher.go

@@ -0,0 +1,130 @@
+package rei
+
+import (
+	"sort"
+	"sync"
+)
+
+type (
+	// Rule filter the event
+	Rule func(ctx *Ctx) bool
+	// Process 事件处理函数
+	Process func(ctx *Ctx)
+)
+
+// Matcher 是 ZeroBot 匹配和处理事件的最小单元
+type Matcher struct {
+	// Temp 是否为临时Matcher，临时 Matcher 匹配一次后就会删除当前 Matcher
+	Temp bool
+	// Block 是否阻断后续 Matcher，为 true 时当前Matcher匹配成功后，后续Matcher不参与匹配
+	Block bool
+	// Priority 优先级，越小优先级越高
+	Priority int
+	// Event 当前匹配到的事件
+	Event *Event
+	// Type 匹配的事件类型
+	Type string
+	// Rules 匹配规则
+	Rules []Rule
+	// Process 处理事件的函数
+	Process Process
+	// Engine 注册 Matcher 的 Engine，Engine可为一系列 Matcher 添加通用 Rule 和 其他钩子
+	Engine *Engine
+}
+
+var (
+	// 所有主匹配器列表
+	matcherMap = make(map[string][]*Matcher, 0)
+	// Matcher 修改读写锁
+	matcherLock = sync.RWMutex{}
+)
+
+// State store the context of a matcher.
+type State map[string]interface{}
+
+func sortMatcher(typ string) {
+	sort.Slice(matcherMap[typ], func(i, j int) bool { // 按优先级排序
+		return matcherMap[typ][i].Priority < matcherMap[typ][j].Priority
+	})
+}
+
+// SetBlock 设置是否阻断后面的 Matcher 触发
+func (m *Matcher) SetBlock(block bool) *Matcher {
+	m.Block = block
+	return m
+}
+
+// SetPriority 设置当前 Matcher 优先级
+func (m *Matcher) SetPriority(priority int) *Matcher {
+	matcherLock.Lock()
+	defer matcherLock.Unlock()
+	m.Priority = priority
+	sortMatcher(m.Type)
+	return m
+}
+
+// FirstPriority 设置当前 Matcher 优先级 - 0
+func (m *Matcher) FirstPriority() *Matcher {
+	return m.SetPriority(0)
+}
+
+// SecondPriority 设置当前 Matcher 优先级 - 1
+func (m *Matcher) SecondPriority() *Matcher {
+	return m.SetPriority(1)
+}
+
+// ThirdPriority 设置当前 Matcher 优先级 - 2
+func (m *Matcher) ThirdPriority() *Matcher {
+	return m.SetPriority(2)
+}
+
+// BindEngine bind the matcher to a engine
+func (m *Matcher) BindEngine(e *Engine) *Matcher {
+	m.Engine = e
+	return m
+}
+
+// StoreMatcher store a matcher to matcher list.
+func StoreMatcher(m *Matcher) *Matcher {
+	matcherLock.Lock()
+	defer matcherLock.Unlock()
+	matcherMap[m.Type] = append(matcherMap[m.Type], m)
+	sortMatcher(m.Type)
+	return m
+}
+
+// StoreTempMatcher store a matcher only triggered once.
+func StoreTempMatcher(m *Matcher) *Matcher {
+	m.Temp = true
+	StoreMatcher(m)
+	return m
+}
+
+// Delete remove the matcher from list
+func (m *Matcher) Delete() {
+	matcherLock.Lock()
+	defer matcherLock.Unlock()
+	for i, matcher := range matcherMap[m.Type] {
+		if m == matcher {
+			matcherMap[m.Type] = append(matcherMap[m.Type][:i], matcherMap[m.Type][i+1:]...)
+		}
+	}
+}
+
+func (m *Matcher) copy() *Matcher {
+	return &Matcher{
+		Type:     m.Type,
+		Rules:    m.Rules,
+		Block:    m.Block,
+		Priority: m.Priority,
+		Process:  m.Process,
+		Temp:     m.Temp,
+		Engine:   m.Engine,
+	}
+}
+
+// Handle 直接处理事件
+func (m *Matcher) Handle(handler Process) *Matcher {
+	m.Process = handler
+	return m
+}