Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2025-12-19 13:59:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✏️ nwife 防注入

Browse Source
This commit is contained in:
fumiama 2021-12-06 19:33:47 +08:00
parent 1d9813691c
commit b20cb1c0a0
3 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
control/web/gui.go
View File

@ -60,7 +60,7 @@ type request struct {
// InitGui 初始化gui
func InitGui(addr string) {
// 将日志重定向到前端hook
writer := io.MultiWriter(l, os.Stderr)
writer := io.MultiWriter(l, os.Stdout)
log.SetOutput(writer)
// 监听后端
go controller(addr)

11
main.go
View File

@ -80,8 +80,8 @@ func init() {
d := flag.Bool("d", false, "Enable debug level log and higher.")
w := flag.Bool("w", false, "Enable warning level log and higher.")
h := flag.Bool("h", false, "Display this help.")
// 解析命令行参数，输入 `-g 监听地址:端口` 即可启用 gui
// g := flag.String("g", "127.0.0.1:3000", "Enable web gui.")
// 解析命令行参数，输入 `-g 监听地址:端口` 指定 gui 访问地址，默认 127.0.0.1:3000
// g := flag.String("g", "127.0.0.1:3000", "Set web gui listening address.")
// 直接写死 AccessToken 时，请更改下面第二个参数
token = flag.String("t", "", "Set AccessToken of WSClient.")
@ -102,10 +102,8 @@ func init() {
logrus.SetLevel(logrus.WarnLevel)
}
}
// 解析命令行参数，输入 `-g` 即可启用 gui
// if *g {
// 启用 gui
// webctrl.InitGui(*g)
// }
}
func printBanner() {
@ -121,8 +119,7 @@ func main() {
// 帮助
zero.OnFullMatchGroup([]string{"/help", ".help", "菜单"}, zero.OnlyToMe).SetBlock(true).FirstPriority().
Handle(func(ctx *zero.Ctx) {
ctx.SendChain(message.Text("可发送\"/服务详情\"等指令,查看bot功能"))
ctx.SendChain(message.Text(banner))
ctx.SendChain(message.Text(banner, "\n可发送\"/服务详情\"查看 bot 功能"))
})
zero.RunAndBlock(
zero.Config{

6
plugin_nativewife/main.go
View File

@ -33,7 +33,7 @@ func init() {
DisableOnDefault: false,
Help: "nativewife\n- 抽wife[@xxx]\n- 添加wife[名字][图片]\n- 删除wife[名字]\n- [让|不让]所有人均可添加wife",
})
engine.OnPrefix("抽wife", zero.OnlyGroup).SetBlock(true).SetPriority(20).
engine.OnFullMatch("抽wife", zero.OnlyGroup).SetBlock(true).SetPriority(20).
Handle(func(ctx *zero.Ctx) {
grpf := strconv.FormatInt(ctx.Event.GroupID, 36)
wifes, err := os.ReadDir(base + "/" + grpf)
@ -72,6 +72,8 @@ func init() {
if elem.Type == "text" {
name = strings.ReplaceAll(elem.Data["text"], " ", "")
name = name[strings.LastIndex(name, "添加wife")+10:]
name = strings.ReplaceAll(name, "/", "")
name = strings.ReplaceAll(name, "\\", "")
break
}
}
@ -98,6 +100,8 @@ func init() {
if elem.Type == "text" {
name = strings.ReplaceAll(elem.Data["text"], " ", "")
name = name[strings.LastIndex(name, "删除wife")+10:]
name = strings.ReplaceAll(name, "/", "")
name = strings.ReplaceAll(name, "\\", "")
break
}
}