🎨 优化目录结构

2026-02-12 18:20:27 +00:00 · 2022-02-25 22:15:14 +08:00
parent 5ccf753af3
commit 0cfb2e4e06
101 changed files with 116 additions and 116 deletions
--- a/plugin/nativewife/main.go
+++ b/plugin/nativewife/main.go
@@ -0,0 +1,165 @@
+// 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/zbputils/control/order"
+
+	control "github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
+	"github.com/FloatTech/zbputils/file"
+)
+
+func init() {
+	engine := control.Register("nwife", order.AcquirePrio(), &control.Options{
+		DisableOnDefault:  false,
+		Help:              "nativewife\n- 抽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 := ctxext.NickName(ctx)
+				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, ctxext.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, true)
+				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.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.Event.GroupID, true)
+			case "取消", "撤销", "不让":
+				err = setEveryoneCanAddWife(ctx.Event.GroupID, 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 := control.Lookup("nwife")
+		if ok {
+			data := m.GetData(gid)
+			if data&1 == 1 {
+				return true
+			}
+			return zero.AdminPermission(ctx)
+		}
+	}
+	return false
+}
+
+func setEveryoneCanAddWife(gid int64, canadd bool) error {
+	m, ok := control.Lookup("nwife")
+	if ok {
+		if canadd {
+			return m.SetData(gid, 1)
+		}
+		return m.SetData(gid, 0)
+	}
+	return errors.New("no such plugin")
+}