ZeroBot-Plugin/plugin_curse/curse.go
github-actions[bot] 25ef2ca220 🎨 改进代码样式
2021-12-21 13:25:34 +00:00

58 lines
1.6 KiB
Go

package curse
import (
"regexp"
"strconv"
"time"
log "github.com/sirupsen/logrus"
zero "github.com/wdvxdr1123/ZeroBot"
"github.com/wdvxdr1123/ZeroBot/extension/rate"
"github.com/wdvxdr1123/ZeroBot/message"
"github.com/wdvxdr1123/ZeroBot/utils/helper"
"github.com/FloatTech/ZeroBot-Plugin/control"
"github.com/FloatTech/ZeroBot-Plugin/utils/web"
)
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"
)
var (
engine = control.Register("curse", &control.Options{
DisableOnDefault: false,
Help: "骂人\n" +
"- 骂他[@xxx]|骂他[qq号]\n",
})
limit = rate.NewManager(time.Minute, 20)
)
func init() {
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)
}
text = helper.BytesToString(data)
ctx.SendChain(message.At(uid), message.Text(text))
})
}