cdle
135 lines
3.0 KiB
Go
135 lines
3.0 KiB
Go
package core
|
|
|
|
import (
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/beego/beego/v2/adapter/logs"
|
|
cron "github.com/robfig/cron/v3"
|
|
)
|
|
|
|
var c *cron.Cron
|
|
|
|
func init() {
|
|
c = cron.New()
|
|
c.Start()
|
|
}
|
|
|
|
type Function struct {
|
|
Rules []string
|
|
FindAll bool
|
|
Admin bool
|
|
Handle func(s Sender) interface{}
|
|
Cron string
|
|
}
|
|
|
|
var pname = regexp.MustCompile(`/([^/\s]+)`).FindStringSubmatch(os.Args[0])[1]
|
|
|
|
var name = func() string {
|
|
return sillyGirl.Get("name", "傻妞")
|
|
}
|
|
|
|
var functions = []Function{}
|
|
|
|
var Senders chan Sender
|
|
|
|
func initToHandleMessage() {
|
|
Senders = make(chan Sender)
|
|
go func() {
|
|
for {
|
|
go handleMessage(<-Senders)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func AddCommand(prefix string, cmds []Function) {
|
|
for j := range cmds {
|
|
for i := range cmds[j].Rules {
|
|
if strings.Contains(cmds[j].Rules[i], "raw ") {
|
|
cmds[j].Rules[i] = strings.Replace(cmds[j].Rules[i], "raw ", "", -1)
|
|
continue
|
|
}
|
|
if strings.Contains(cmds[j].Rules[i], "$") {
|
|
continue
|
|
}
|
|
if prefix != "" {
|
|
cmds[j].Rules[i] = prefix + `\s+` + cmds[j].Rules[i]
|
|
}
|
|
|
|
cmds[j].Rules[i] = strings.Replace(cmds[j].Rules[i], "(", `[(]`, -1)
|
|
cmds[j].Rules[i] = strings.Replace(cmds[j].Rules[i], ")", `[)]`, -1)
|
|
cmds[j].Rules[i] = regexp.MustCompile(`\?$`).ReplaceAllString(cmds[j].Rules[i], `(.+)`)
|
|
cmds[j].Rules[i] = strings.Replace(cmds[j].Rules[i], " ", `\s+`, -1)
|
|
cmds[j].Rules[i] = strings.Replace(cmds[j].Rules[i], "?", `(\S+)`, -1)
|
|
cmds[j].Rules[i] = "^" + cmds[j].Rules[i] + "$"
|
|
}
|
|
functions = append(functions, cmds[j])
|
|
if cmds[j].Cron != "" {
|
|
cmd := cmds[j]
|
|
if _, err := c.AddFunc(cmds[j].Cron, func() {
|
|
cmd.Handle(&Faker{})
|
|
}); err != nil {
|
|
logs.Warn("任务%v添加失败%v", cmds[j].Rules[0], err)
|
|
} else {
|
|
logs.Warn("任务%v添加成功", cmds[j].Rules[0])
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func handleMessage(sender 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 {
|
|
if function.Admin && !sender.IsAdmin() {
|
|
sender.Delete()
|
|
sender.Disappear()
|
|
sender.Reply("没有权限操作")
|
|
sender.Finish()
|
|
return
|
|
}
|
|
rt := function.Handle(sender)
|
|
if rt != nil {
|
|
sender.Reply(rt)
|
|
}
|
|
sender.Finish()
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func FetchCookieValue(ps ...string) string {
|
|
var key, cookies string
|
|
if len(ps) == 2 {
|
|
if len(ps[0]) > len(ps[1]) {
|
|
key, cookies = ps[1], ps[0]
|
|
} else {
|
|
key, cookies = ps[0], ps[1]
|
|
}
|
|
}
|
|
match := regexp.MustCompile(key + `=([^;]*);{0,1}`).FindStringSubmatch(cookies)
|
|
if len(match) == 2 {
|
|
return match[1]
|
|
} else {
|
|
return ""
|
|
}
|
|
}
|