✨ 插件名称自动注册

2026-02-06 15:20:22 +00:00 · 2023-09-03 13:53:25 +08:00
parent 5485cc3be9
commit 6bff62b91b
108 changed files with 124 additions and 126 deletions
--- a/plugin/nwife/main.go
+++ b/plugin/nwife/main.go
@@ -0,0 +1,166 @@
+// Package nativewife 本地老婆
+package nativewife
+
+import (
+	"crypto/md5"
+	"encoding/binary"
+	"errors"
+	"fmt"
+	"math/rand"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+	"github.com/wdvxdr1123/ZeroBot/utils/helper"
+
+	"github.com/FloatTech/floatbox/file"
+	"github.com/FloatTech/floatbox/process"
+	ctrl "github.com/FloatTech/zbpctrl"
+	"github.com/FloatTech/zbputils/control"
+)
+
+func init() {
+	engine := control.AutoRegister(&ctrl.Options[*zero.Ctx]{
+		DisableOnDefault:  false,
+		Brief:             "本地老婆",
+		Help:              "- 抽wife[@xxx]\n- 添加wife[名字][图片]\n- 删除wife[名字]\n- [让 | 不让]所有人均可添加wife",
+		PrivateDataFolder: "nwife",
+	})
+	base := engine.DataFolder()
+	baseuri := "file:///" + file.BOTPATH + "/" + base
+	engine.OnPrefix("抽wife", zero.OnlyGroup).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			grpf := strconv.FormatInt(ctx.Event.GroupID, 36)
+			wifes, err := os.ReadDir(base + grpf)
+			if err != nil {
+				ctx.SendChain(message.Text("一个wife也没有哦~"))
+				return
+			}
+			switch len(wifes) {
+			case 0:
+				ctx.SendChain(message.Text("一个wife也没有哦~"))
+			case 1:
+				wn := wifes[0].Name()
+				ctx.SendChain(message.Text("大家的wife都是", wn, "\n"), message.Image(baseuri+grpf+"/"+wn), message.Text("\n哦~"))
+			default:
+				// 获取名字
+				name := ctx.NickName()
+				now := time.Now()
+				s := md5.Sum(helper.StringToBytes(fmt.Sprintf("%s%d%d%d", name, now.Year(), now.Month(), now.Day())))
+				r := rand.New(rand.NewSource(int64(binary.LittleEndian.Uint64(s[:]))))
+				n := r.Intn(len(wifes))
+				wn := wifes[n].Name()
+				ctx.SendChain(message.Text(name, "的wife是", wn, "\n"), message.Image(baseuri+grpf+"/"+wn), message.Text("\n哦~"))
+			}
+		})
+	// 上传一张图
+	engine.OnPrefix("添加wife", zero.OnlyGroup, chkAddWifePermission, zero.MustProvidePicture).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			name := ""
+			for _, elem := range ctx.Event.Message {
+				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
+				}
+			}
+			if name != "" {
+				url := ctx.State["image_url"].([]string)[0]
+				grpfolder := base + strconv.FormatInt(ctx.Event.GroupID, 36)
+				if file.IsNotExist(grpfolder) {
+					err := os.Mkdir(grpfolder, 0755)
+					if err != nil {
+						ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("错误：", err.Error()))
+						return
+					}
+				}
+				err := file.DownloadTo(url, grpfolder+"/"+name)
+				if err == nil {
+					process.SleepAbout1sTo2s()
+					ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("成功！"))
+				} else {
+					ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("错误：", err.Error()))
+				}
+			} else {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("没有找到wife的名字！"))
+			}
+		})
+	engine.OnPrefix("删除wife", zero.OnlyGroup, zero.AdminPermission).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			name := ""
+			for _, elem := range ctx.Event.Message {
+				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
+				}
+			}
+			if name != "" {
+				grpfolder := base + strconv.FormatInt(ctx.Event.GroupID, 36)
+				err := os.Remove(grpfolder + "/" + name)
+				if err == nil {
+					ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("成功！"))
+				} else {
+					ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("错误：", err.Error()))
+				}
+			} else {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("没有找到wife的名字！"))
+			}
+		})
+	engine.OnSuffix("所有人均可添加wife", zero.SuperUserPermission, zero.OnlyGroup).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			text := ""
+			for _, elem := range ctx.Event.Message {
+				if elem.Type == "text" {
+					text = strings.ReplaceAll(elem.Data["text"], " ", "")
+					text = text[:strings.LastIndex(text, "所有人均可添加wife")]
+					break
+				}
+			}
+			var err error
+			switch text {
+			case "设置", "授予", "让":
+				err = setEveryoneCanAddWife(ctx, true)
+			case "取消", "撤销", "不让":
+				err = setEveryoneCanAddWife(ctx, false)
+			}
+			if err == nil {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("成功！"))
+			} else {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("错误：", err.Error()))
+			}
+		})
+}
+
+func chkAddWifePermission(ctx *zero.Ctx) bool {
+	gid := ctx.Event.GroupID
+	if gid > 0 {
+		m, ok := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
+		if ok {
+			data := m.GetData(gid)
+			if data&1 == 1 {
+				return true
+			}
+			return zero.AdminPermission(ctx)
+		}
+	}
+	return false
+}
+
+func setEveryoneCanAddWife(ctx *zero.Ctx, canadd bool) error {
+	m, ok := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
+	if ok {
+		if canadd {
+			return m.SetData(ctx.Event.GroupID, 1)
+		}
+		return m.SetData(ctx.Event.GroupID, 0)
+	}
+	return errors.New("no such plugin")
+}