add plugin

engine.go

@@ -0,0 +1,176 @@
+package rei
+
+// New 生成空引擎
+func NewEngine() *Engine {
+	return &Engine{
+		preHandler:  []Rule{},
+		midHandler:  []Rule{},
+		postHandler: []Process{},
+	}
+}
+
+var defaultEngine = NewEngine()
+
+// Engine is the pre_handler, mid_handler, post_handler manager
+type Engine struct {
+	preHandler  []Rule
+	midHandler  []Rule
+	postHandler []Process
+	matchers    []*Matcher
+}
+
+// Delete 移除该 Engine 注册的所有 Matchers
+func (e *Engine) Delete() {
+	for _, m := range e.matchers {
+		m.Delete()
+	}
+}
+
+// UsePreHandler 向该 Engine 添加新 PreHandler(Rule),
+// 会在 Rule 判断前触发，如果 preHandler
+// 没有通过，则 Rule, Matcher 不会触发
+//
+// 可用于分群组管理插件等
+func (e *Engine) UsePreHandler(rules ...Rule) {
+	e.preHandler = append(e.preHandler, rules...)
+}
+
+// UseMidHandler 向该 Engine 添加新 MidHandler(Rule),
+// 会在 Rule 判断后， Matcher 触发前触发，如果 midHandler
+// 没有通过，则 Matcher 不会触发
+//
+// 可用于速率限制等
+func (e *Engine) UseMidHandler(rules ...Rule) {
+	e.midHandler = append(e.midHandler, rules...)
+}
+
+// UsePostHandler 向该 Engine 添加新 PostHandler(Rule),
+// 会在 Matcher 触发后触发，如果 PostHandler 返回 false,
+// 则后续的 post handler 不会触发
+//
+// 可用于速率限制等
+func (e *Engine) UsePostHandler(handler ...Process) {
+	e.postHandler = append(e.postHandler, handler...)
+}
+
+// On 添加新的指定消息类型的匹配器(默认Engine)
+func On(typ string, rules ...Rule) *Matcher { return defaultEngine.On(typ, rules...) }
+
+// On 添加新的指定消息类型的匹配器
+func (e *Engine) On(typ string, rules ...Rule) *Matcher {
+	matcher := &Matcher{
+		Type:   typ,
+		Rules:  rules,
+		Engine: e,
+	}
+	e.matchers = append(e.matchers, matcher)
+	return StoreMatcher(matcher)
+}
+
+// OnMessage 消息触发器
+func (e *Engine) OnMessage(rules ...Rule) *Matcher { return e.On("Message", rules...) }
+
+// OnMessage 消息触发器
+func OnMessage(rules ...Rule) *Matcher { return On("Message", rules...) }
+
+// OnEditedMessage 修改消息触发器
+func (e *Engine) OnEditedMessage(rules ...Rule) *Matcher { return e.On("EditedMessage", rules...) }
+
+// OnEditedMessage 修改消息触发器
+func OnEditedMessage(rules ...Rule) *Matcher { return On("EditedMessage", rules...) }
+
+// OnChannelPost ...
+func (e *Engine) OnChannelPost(rules ...Rule) *Matcher { return e.On("ChannelPost", rules...) }
+
+// OnChannelPost ...
+func OnChannelPost(rules ...Rule) *Matcher { return On("ChannelPost", rules...) }
+
+// OnEditedChannelPost ...
+func (e *Engine) OnEditedChannelPost(rules ...Rule) *Matcher {
+	return e.On("EditedChannelPost", rules...)
+}
+
+// OnEditedChannelPost ...
+func OnEditedChannelPost(rules ...Rule) *Matcher {
+	return On("EditedChannelPost", rules...)
+}
+
+// OnInlineQuery ...
+func (e *Engine) OnInlineQuery(rules ...Rule) *Matcher { return e.On("InlineQuery", rules...) }
+
+// OnInlineQuery ...
+func OnInlineQuery(rules ...Rule) *Matcher { return On("InlineQuery", rules...) }
+
+// OnChosenInlineResult ...
+func (e *Engine) OnChosenInlineResult(rules ...Rule) *Matcher {
+	return e.On("ChosenInlineResult", rules...)
+}
+
+// OnChosenInlineResult ...
+func OnChosenInlineResult(rules ...Rule) *Matcher { return On("ChosenInlineResult", rules...) }
+
+// OnCallbackQuery ...
+func (e *Engine) OnCallbackQuery(rules ...Rule) *Matcher { return e.On("CallbackQuery", rules...) }
+
+// OnCallbackQuery ...
+func OnCallbackQuery(rules ...Rule) *Matcher { return On("CallbackQuery", rules...) }
+
+// OnShippingQuery ...
+func (e *Engine) OnShippingQuery(rules ...Rule) *Matcher { return e.On("ShippingQuery", rules...) }
+
+// OnShippingQuery ...
+func OnShippingQuery(rules ...Rule) *Matcher { return On("ShippingQuery", rules...) }
+
+// OnPreCheckoutQuery ...
+func (e *Engine) OnPreCheckoutQuery(rules ...Rule) *Matcher {
+	return e.On("PreCheckoutQuery", rules...)
+}
+
+// OnPreCheckoutQuery ...
+func OnPreCheckoutQuery(rules ...Rule) *Matcher { return On("PreCheckoutQuery", rules...) }
+
+// OnPoll ...
+func (e *Engine) OnPoll(rules ...Rule) *Matcher { return e.On("Poll", rules...) }
+
+// OnPoll ...
+func OnPoll(rules ...Rule) *Matcher { return On("Poll", rules...) }
+
+// OnPollAnswer ...
+func (e *Engine) OnPollAnswer(rules ...Rule) *Matcher { return e.On("PollAnswer", rules...) }
+
+// OnPollAnswer ...
+func OnPollAnswer(rules ...Rule) *Matcher { return On("PollAnswer", rules...) }
+
+// OnMyChatMember ...
+func (e *Engine) OnMyChatMember(rules ...Rule) *Matcher { return e.On("MyChatMember", rules...) }
+
+// OnMyChatMember ...
+func OnMyChatMember(rules ...Rule) *Matcher { return On("MyChatMember", rules...) }
+
+// OnChatMember ...
+func (e *Engine) OnChatMember(rules ...Rule) *Matcher { return e.On("ChatMember", rules...) }
+
+// OnChatMember ...
+func OnChatMember(rules ...Rule) *Matcher { return On("ChatMember", rules...) }
+
+// OnChatJoinRequest ...
+func (e *Engine) OnChatJoinRequest(rules ...Rule) *Matcher { return e.On("ChatJoinRequest", rules...) }
+
+// OnChatJoinRequest ...
+func OnChatJoinRequest(rules ...Rule) *Matcher { return On("ChatJoinRequest", rules...) }
+
+// OnPrefix 前缀触发器
+func OnMessagePrefix(prefix string, rules ...Rule) *Matcher {
+	return defaultEngine.OnMessagePrefix(prefix, rules...)
+}
+
+// OnPrefix 前缀触发器
+func (e *Engine) OnMessagePrefix(prefix string, rules ...Rule) *Matcher {
+	matcher := &Matcher{
+		Type:   "Message",
+		Rules:  append([]Rule{PrefixRule(prefix)}, rules...),
+		Engine: e,
+	}
+	e.matchers = append(e.matchers, matcher)
+	return StoreMatcher(matcher)
+}