ZeroBot-Plugin/plugin/nativewife/main.go

// 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"
	ctrl "github.com/FloatTech/zbpctrl"
	"github.com/FloatTech/zbputils/control"
)

func init() {
	engine := control.Register("nwife", &ctrl.Options[*zero.Ctx]{
		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 := 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, 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, 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")
}