✨ 新增 抽wife 插件

README.md

@@ -165,6 +165,10 @@ zerobot -h -t token -u url [-d|w] [-g 监听地址:端口] qq1 qq2 qq3 ...
     - [x] 清理缓存
     - [ ] 简易语音
     - [ ] 爬图合成 [@xxx]
+- **抽wife** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin_nativewife"`
+    - [x] 抽wife[@xxx]
+    - [x] 添加wife[名字][图片]
+    - 注：不同群添加后不会重叠
 - **minecraft** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin_minecraft"`
     - [x] /mcstart xxx
 	- [x] /mcstop xxx

main.go

@@ -49,6 +49,7 @@ import (
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_image_finder"  // 关键字搜图
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_lolicon"       // lolicon 随机图片
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_nativesetu"    // 本地涩图
+	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_nativewife"    // 本地老婆
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_saucenao"      // 以图搜图
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_setutime"      // 来份涩图
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin_tracemoe"      // 搜番

plugin_nativewife/main.go

@@ -0,0 +1,92 @@
+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的名字！"))
+			}
+		})
+}