fumiama/NanoBot
1
0
Fork 0
mirror of https://github.com/fumiama/NanoBot.git synced 2026-06-10 13:10:26 +08:00
Code Activity

finish

Browse Source
This commit is contained in:
源文雨
2023-10-16 00:26:04 +09:00
parent 6a1019d201
commit ce34c996ef
25 changed files with 2844 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

368
rule.go Normal file
View File

@@ -0,0 +1,368 @@
package nano
import (
"fmt"
"strconv"
"strings"
"time"
"unsafe"
"github.com/FloatTech/floatbox/process"
ctrl "github.com/FloatTech/zbpctrl"
"github.com/wdvxdr1123/ZeroBot/extension"
"github.com/wdvxdr1123/ZeroBot/extension/rate"
)
func newctrl(service string, o *ctrl.Options[*Ctx]) Rule {
c := m.NewControl(service, o)
return func(ctx *Ctx) bool {
ctx.State["manager"] = c
gid, _ := strconv.ParseUint(ctx.Message.ChannelID, 10, 64)
uid, _ := strconv.ParseUint(ctx.Message.Author.ID, 10, 64)
return c.Handler(uintptr(unsafe.Pointer(ctx)), int64(gid), int64(uid))
}
}
func Lookup(service string) (*ctrl.Control[*Ctx], bool) {
return m.Lookup(service)
}
// respLimiterManager 请求响应限速器管理
//
// 每 1d 4次触发
var respLimiterManager = rate.NewManager[string](time.Hour*24, 4)
func init() {
process.NewCustomOnce(&m).Do(func() {
OnMessageCommandGroup([]string{
"响应", "response", "沉默", "silence",
}, UserOrGrpAdmin).SetBlock(true).Limit(func(ctx *Ctx) *rate.Limiter {
return respLimiterManager.Load(ctx.Message.ChannelID)
}).secondPriority().Handle(func(ctx *Ctx) {
grp, _ := strconv.ParseUint(ctx.Message.ChannelID, 10, 64)
msg := ""
switch ctx.State["command"] {
case "响应", "response":
if m.CanResponse(int64(grp)) {
msg = ctx.Caller.ready.User.Username + "已经在工作了哦~"
break
}
if SuperUserPermission(ctx) {
err := m.Response(int64(grp))
if err == nil {
msg = ctx.Caller.ready.User.Username + "将开始在此工作啦~"
} else {
msg = "ERROR: " + err.Error()
}
break
}
case "沉默", "silence":
if !m.CanResponse(int64(grp)) {
msg = ctx.Caller.ready.User.Username + "已经在休息了哦~"
break
}
err := m.Silence(int64(grp))
if err == nil {
msg = ctx.Caller.ready.User.Username + "将开始休息啦~"
} else {
msg = "ERROR: " + err.Error()
}
if SuperUserPermission(ctx) {
break
}
default:
msg = "ERROR: bad command\"" + fmt.Sprint(ctx.State["command"]) + "\""
}
_, _ = ctx.SendPlainMessage(false, msg)
})
OnMessageCommandGroup([]string{
"全局响应", "allresponse", "全局沉默", "allsilence",
}, SuperUserPermission).SetBlock(true).secondPriority().Handle(func(ctx *Ctx) {
msg := ""
cmd := ctx.State["command"].(string)
switch {
case strings.Contains(cmd, "响应") || strings.Contains(cmd, "response"):
err := m.Response(0)
if err == nil {
msg = ctx.Caller.ready.User.Username + "将开始在此工作啦~"
} else {
msg = "ERROR: " + err.Error()
}
case strings.Contains(cmd, "沉默") || strings.Contains(cmd, "silence"):
err := m.Silence(0)
if err == nil {
msg = ctx.Caller.ready.User.Username + "将开始休息啦~"
} else {
msg = "ERROR: " + err.Error()
}
default:
msg = "ERROR: bad command\"" + cmd + "\""
}
_, _ = ctx.SendPlainMessage(false, msg)
})
OnMessageCommandGroup([]string{
"启用", "enable", "禁用", "disable",
}, UserOrGrpAdmin).SetBlock(true).secondPriority().Handle(func(ctx *Ctx) {
grp, _ := strconv.ParseUint(ctx.Message.ChannelID, 10, 64)
if !m.CanResponse(int64(grp)) {
return
}
model := extension.CommandModel{}
_ = ctx.Parse(&model)
service, ok := Lookup(model.Args)
if !ok {
_, _ = ctx.SendPlainMessage(false, "没有找到指定服务!")
return
}
if strings.Contains(model.Command, "启用") || strings.Contains(model.Command, "enable") {
service.Enable(int64(grp))
if service.Options.OnEnable != nil {
service.Options.OnEnable(ctx)
} else {
_, _ = ctx.SendPlainMessage(false, "已启用服务: ", model.Args)
}
} else {
service.Disable(int64(grp))
if service.Options.OnDisable != nil {
service.Options.OnDisable(ctx)
} else {
_, _ = ctx.SendPlainMessage(false, "已禁用服务: ", model.Args)
}
}
})
OnMessageCommandGroup([]string{
"全局启用", "allenable", "全局禁用", "alldisable",
}, OnlyToMe, SuperUserPermission).SetBlock(true).secondPriority().Handle(func(ctx *Ctx) {
model := extension.CommandModel{}
_ = ctx.Parse(&model)
service, ok := Lookup(model.Args)
if !ok {
_, _ = ctx.SendPlainMessage(false, "没有找到指定服务!")
return
}
if strings.Contains(model.Command, "启用") || strings.Contains(model.Command, "enable") {
service.Enable(0)
_, _ = ctx.SendPlainMessage(false, "已全局启用服务: ", model.Args)
} else {
service.Disable(0)
_, _ = ctx.SendPlainMessage(false, "已全局禁用服务: ", model.Args)
}
})
OnMessageCommandGroup([]string{"还原", "reset"}, UserOrGrpAdmin).SetBlock(true).secondPriority().Handle(func(ctx *Ctx) {
grp, _ := strconv.ParseUint(ctx.Message.ChannelID, 10, 64)
if !m.CanResponse(int64(grp)) {
return
}
model := extension.CommandModel{}
_ = ctx.Parse(&model)
service, ok := Lookup(model.Args)
if !ok {
_, _ = ctx.SendPlainMessage(false, "没有找到指定服务!")
return
}
service.Reset(int64(grp))
_, _ = ctx.SendPlainMessage(false, "已还原服务的默认启用状态: ", model.Args)
})
OnMessageCommandGroup([]string{
"禁止", "ban", "允许", "permit",
}, AdminPermission).SetBlock(true).secondPriority().Handle(func(ctx *Ctx) {
grp, _ := strconv.ParseUint(ctx.Message.ChannelID, 10, 64)
if !m.CanResponse(int64(grp)) {
return
}
model := extension.CommandModel{}
_ = ctx.Parse(&model)
args := strings.Split(model.Args, " ")
if len(args) >= 2 {
service, ok := Lookup(args[0])
if !ok {
_, _ = ctx.SendPlainMessage(false, "没有找到指定服务!")
return
}
grp, _ := strconv.ParseUint(ctx.Message.ChannelID, 10, 64)
msg := "*" + args[0] + "报告*"
issu := SuperUserPermission(ctx)
if strings.Contains(model.Command, "允许") || strings.Contains(model.Command, "permit") {
for _, usr := range args[1:] {
uid, err := strconv.ParseInt(usr, 10, 64)
if err == nil {
if issu {
service.Permit(uid, int64(grp))
msg += "\n+ 已允许" + usr
} else {
member, err := ctx.Caller.GetGuildMemberOf(ctx.Message.GuildID, usr)
if err == nil && !member.Pending {
service.Permit(uid, int64(grp))
msg += "\n+ 已允许" + usr
} else {
msg += "\nx " + usr + " 不在本群"
}
}
}
}
} else {
for _, usr := range args[1:] {
uid, err := strconv.ParseInt(usr, 10, 64)
if err == nil {
if issu {
service.Ban(uid, int64(grp))
msg += "\n- 已禁止" + usr
} else {
member, err := ctx.Caller.GetGuildMemberOf(ctx.Message.GuildID, usr)
if err == nil && !member.Pending {
service.Ban(uid, int64(grp))
msg += "\n- 已禁止" + usr
} else {
msg += "\nx " + usr + " 不在本群"
}
}
}
}
}
_, _ = ctx.SendPlainMessage(false, msg)
return
}
_, _ = ctx.SendPlainMessage(false, "参数错误!")
})
OnMessageCommandGroup([]string{
"全局禁止", "allban", "全局允许", "allpermit",
}, SuperUserPermission).SetBlock(true).secondPriority().Handle(func(ctx *Ctx) {
model := extension.CommandModel{}
_ = ctx.Parse(&model)
args := strings.Split(model.Args, " ")
if len(args) >= 2 {
service, ok := Lookup(args[0])
if !ok {
_, _ = ctx.SendPlainMessage(false, "没有找到指定服务!")
return
}
msg := "*" + args[0] + "全局报告*"
if strings.Contains(model.Command, "允许") || strings.Contains(model.Command, "permit") {
for _, usr := range args[1:] {
uid, err := strconv.ParseInt(usr, 10, 64)
if err == nil {
service.Permit(uid, 0)
msg += "\n+ 已允许" + usr
}
}
} else {
for _, usr := range args[1:] {
uid, err := strconv.ParseInt(usr, 10, 64)
if err == nil {
service.Ban(uid, 0)
msg += "\n- 已禁止" + usr
}
}
}
_, _ = ctx.SendPlainMessage(false, msg)
return
}
_, _ = ctx.SendPlainMessage(false, "参数错误!")
})
OnMessageCommandGroup([]string{
"封禁", "block", "解封", "unblock",
}, SuperUserPermission).SetBlock(true).secondPriority().Handle(func(ctx *Ctx) {
model := extension.CommandModel{}
_ = ctx.Parse(&model)
args := strings.Split(model.Args, " ")
if len(args) >= 1 {
msg := "*报告*"
if strings.Contains(model.Command, "解") || strings.Contains(model.Command, "un") {
for _, usr := range args {
uid, err := strconv.ParseInt(usr, 10, 64)
if err == nil {
if m.DoUnblock(uid) == nil {
msg += "\n- 已解封" + usr
}
}
}
} else {
for _, usr := range args {
uid, err := strconv.ParseInt(usr, 10, 64)
if err == nil {
if m.DoBlock(uid) == nil {
msg += "\n+ 已封禁" + usr
}
}
}
}
_, _ = ctx.SendPlainMessage(false, msg)
return
}
_, _ = ctx.SendPlainMessage(false, "参数错误!")
})
OnMessageCommandGroup([]string{
"改变默认启用状态", "allflip",
}, SuperUserPermission).SetBlock(true).secondPriority().Handle(func(ctx *Ctx) {
model := extension.CommandModel{}
_ = ctx.Parse(&model)
service, ok := Lookup(model.Args)
if !ok {
_, _ = ctx.SendPlainMessage(false, "没有找到指定服务!")
return
}
err := service.Flip()
if err != nil {
_, _ = ctx.SendPlainMessage(false, "ERROR: ", err)
return
}
_, _ = ctx.SendPlainMessage(false, "已改变全局默认启用状态: ", model.Args)
})
OnMessageCommandGroup([]string{"用法", "usage"}, UserOrGrpAdmin).SetBlock(true).secondPriority().
Handle(func(ctx *Ctx) {
model := extension.CommandModel{}
_ = ctx.Parse(&model)
service, ok := Lookup(model.Args)
if !ok {
_, _ = ctx.SendPlainMessage(false, "没有找到指定服务!")
return
}
if service.Options.Help != "" {
gid := ctx.Message.ChannelID
grp, _ := strconv.ParseUint(gid, 10, 64)
_, _ = ctx.SendPlainMessage(false, service.EnableMarkIn(int64(grp)), " ", service)
} else {
_, _ = ctx.SendPlainMessage(false, "该服务无帮助!")
}
})
OnMessageCommandGroup([]string{"服务列表", "service_list"}, UserOrGrpAdmin).SetBlock(true).secondPriority().
Handle(func(ctx *Ctx) {
gid := ctx.Message.ChannelID
m.RLock()
msg := make([]any, 1, len(m.M)*4+1)
m.RUnlock()
msg[0] = "--------服务列表--------\n发送\"/用法 name\"查看详情\n发送\"/响应\"启用会话"
ForEachByPrio(func(i int, service *ctrl.Control[*Ctx]) bool {
grp, _ := strconv.ParseUint(gid, 10, 64)
msg = append(msg, "\n", i+1, ": ", service.EnableMarkIn(int64(grp)), service.Service)
return true
})
_, _ = ctx.SendPlainMessage(false, msg...)
})
OnMessageCommandGroup([]string{"服务详情", "service_detail"}, UserOrGrpAdmin).SetBlock(true).secondPriority().
Handle(func(ctx *Ctx) {
gid := ctx.Message.ChannelID
m.RLock()
msgs := make([]any, 1, len(m.M)*7+1)
m.RUnlock()
msgs[0] = "---服务详情---\n"
ForEachByPrio(func(i int, service *ctrl.Control[*Ctx]) bool {
grp, _ := strconv.ParseUint(gid, 10, 64)
msgs = append(msgs, i+1, ": ", service.EnableMarkIn(int64(grp)), service.Service, "\n", service, "\n\n")
return true
})
_, _ = ctx.SendPlainMessage(false, msgs...)
})
})
}