update

core/function.go

@@ -0,0 +1,86 @@
+package core
+
+import (
+	"fmt"
+	"regexp"
+
+	"github.com/cdle/sillyGirl/im"
+	"github.com/cdle/sillyGirl/im/tg"
+	tb "gopkg.in/tucnak/telebot.v2"
+)
+
+type Function struct {
+	Rules   []string
+	FindAll bool
+	Admin   bool
+	Handle  func(s im.Sender) bool
+}
+
+var functions = []Function{
+	{
+		Rules: []string{"^傻妞 (.*)$", "^傻妞$"},
+		Handle: func(s im.Sender) bool {
+			m := s.Get()
+			if m != "" {
+				s.Reply(fmt.Sprintf("哎呀，傻妞不懂%s是什么意思啦。", m))
+			} else {
+				s.Reply("请说，我在。")
+			}
+			return true
+		},
+	},
+}
+
+var Senders chan im.Sender
+
+func initToHandleMessage() {
+	for _, im := range Config.Im {
+		switch im.Type {
+		case "tg":
+			tg.Handler = func(message *tb.Message) {
+				Senders <- &tg.Sender{
+					Message: message,
+				}
+			}
+			go tg.RunBot(&im)
+		case "qq":
+
+		}
+	}
+	Senders = make(chan im.Sender)
+	go func() {
+		for {
+			go handleMessage(<-Senders)
+		}
+	}()
+}
+
+func AddCommand(cmd *Function) {
+	functions = append(functions, *cmd)
+}
+
+func handleMessage(sender im.Sender) {
+	for _, function := range functions {
+		for _, rule := range function.Rules {
+			var matched bool
+			if function.FindAll {
+				if res := regexp.MustCompile(rule).FindAllStringSubmatch(sender.GetContent(), -1); len(res) > 0 {
+					tmp := [][]string{}
+					for i := range res {
+						tmp = append(tmp, res[i][1:])
+					}
+					sender.SetAllMatch(tmp)
+					matched = true
+				}
+			} else {
+				if res := regexp.MustCompile(rule).FindStringSubmatch(sender.GetContent()); len(res) > 0 {
+					sender.SetMatch(res[1:])
+					matched = true
+				}
+			}
+			if matched && function.Handle(sender) {
+				return
+			}
+		}
+	}
+}