Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2026-02-12 18:20:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

✏️ 优化代码结构

Browse Source
This commit is contained in:
fumiama
2021-12-21 22:09:43 +08:00
parent b953385bc5
commit 1d3a61386d
13 changed files with 136 additions and 364 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
plugin_curse/curse.go
View File

@@ -1,13 +1,11 @@
package curse
import (
"regexp"
"strconv"
"time"
"github.com/FloatTech/ZeroBot-Plugin/control"
"github.com/FloatTech/ZeroBot-Plugin/utils/math"
"github.com/FloatTech/ZeroBot-Plugin/utils/web"
log "github.com/sirupsen/logrus"
zero "github.com/wdvxdr1123/ZeroBot"
"github.com/wdvxdr1123/ZeroBot/extension/rate"
"github.com/wdvxdr1123/ZeroBot/message"
@@ -15,7 +13,6 @@ import (
)
const (
qqReg = `\d+`
ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.107 Safari/537.36"
curseURL = "https://zuanbot.com/api.php?level=min&lang=zh_cn"
)
@@ -30,27 +27,27 @@ var (
)
func init() {
engine.OnPrefix("骂他").SetBlock(true).FirstPriority().Handle(func(ctx *zero.Ctx) {
engine.OnPrefix("骂我").SetBlock(true).FirstPriority().Handle(func(ctx *zero.Ctx) {
if !limit.Load(ctx.Event.GroupID).Acquire() {
ctx.SendChain(message.Text("请稍后重试0x0..."))
return
}
var uid int64
var text string
reg := regexp.MustCompile(qqReg)
if len(ctx.Event.Message) > 1 && ctx.Event.Message[1].Type == "at" {
uid, _ = strconv.ParseInt(ctx.Event.Message[1].Data["qq"], 10, 64)
} else if reg.MatchString(ctx.Event.RawMessage) {
result := reg.FindAllString(ctx.Event.RawMessage, -1)
uid, _ = strconv.ParseInt(result[0], 10, 64)
} else if uid == 0 {
uid = ctx.Event.UserID
}
data, err := web.ReqWith(curseURL, "GET", "", ua)
if err != nil {
log.Println("err为:", err)
ctx.SendChain(message.Text("ERROR:", err))
return
}
text = helper.BytesToString(data)
ctx.SendChain(message.At(uid), message.Text(text))
ctx.SendChain(message.At(ctx.Event.UserID), message.Text(helper.BytesToString(data)))
})
engine.OnRegex(`^骂他.*?(\d+)`, zero.OnlyGroup).SetBlock(true).SetPriority(40).
Handle(func(ctx *zero.Ctx) {
if !limit.Load(ctx.Event.GroupID).Acquire() {
return
}
data, err := web.ReqWith(curseURL, "GET", "", ua)
if err != nil {
ctx.SendChain(message.Text("ERROR:", err))
return
}
ctx.SendChain(message.At(math.Str2Int64(ctx.State["regex_matched"].([]string)[1])), message.Text(helper.BytesToString(data)))
})
}