ZeroBot-Plugin/plugin_nativewife/main.go

package nativewife

import (
	"crypto/md5"
	"encoding/binary"
	"errors"
	"fmt"
	"math/rand"
	"os"
	"strconv"
	"strings"
	"time"

	"github.com/FloatTech/AnimeAPI/picture"
	zero "github.com/wdvxdr1123/ZeroBot"
	"github.com/wdvxdr1123/ZeroBot/message"
	"github.com/wdvxdr1123/ZeroBot/utils/helper"

	"github.com/FloatTech/ZeroBot-Plugin/control"
	"github.com/FloatTech/ZeroBot-Plugin/utils/file"
)

const base = "data/nwife"

var baseuri = "file:///" + file.BOT_PATH + "/" + base

func init() {
	err := os.MkdirAll(base, 0755)
	if err != nil {
		panic(err)
	}
	engine := control.Register("nwife", &control.Options{
		DisableOnDefault: false,
		Help:             "nativewife\n- 抽wife[@xxx]\n- 添加wife[名字][图片]\n- 删除wife[名字]\n- [让|不让]所有人均可添加wife",
	})
	engine.OnPrefix("抽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)
			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.State["args"].(string)
				if len(ctx.Event.Message) > 1 && ctx.Event.Message[1].Type == "at" {
					qq, _ := strconv.ParseInt(ctx.Event.Message[1].Data["qq"], 10, 64)
					name = ctx.GetGroupMemberInfo(ctx.Event.GroupID, qq, false).Get("nickname").Str
				} else if name == "" {
					name = ctx.Event.Sender.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, picture.MustGiven).SetBlock(true).SetPriority(20).
		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) {
					os.Mkdir(grpfolder, 0755)
				}
				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).SetPriority(20).
		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).SetPriority(20).
		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")
}