mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-20 22:40:23 +08:00
93 lines
2.9 KiB
Go
93 lines
2.9 KiB
Go
package nativewife
|
||
|
||
import (
|
||
"crypto/md5"
|
||
"encoding/binary"
|
||
"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[名字][图片]",
|
||
})
|
||
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, "\n的wife是", wn, "\n"), message.Image(baseuri+"/"+grpf+"/"+wn), message.Text("\n哦~"))
|
||
}
|
||
})
|
||
// 上传一张图
|
||
engine.OnKeyword("添加wife", zero.OnlyGroup, zero.AdminPermission, 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:]
|
||
}
|
||
}
|
||
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.Text("成功！"))
|
||
} else {
|
||
ctx.SendChain(message.Text("错误：", err.Error()))
|
||
}
|
||
} else {
|
||
ctx.SendChain(message.Text("没有找到wife的名字！"))
|
||
}
|
||
})
|
||
}
|