提交新插件牛牛大作战

README.md

@@ -1009,22 +1009,6 @@ print("run[CQ:image,file="+j["img"]+"]")
 
   - 注：刷新文件夹较慢，请耐心等待刷新完成，会提示“成功”。
 
-</details>
-<details>
-  <summary>抽wife</summary>
-
-  `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/nativewife"`
-
-  - [x] 抽wife[@xxx]
-
-  - [x] 添加wife[名字][图片]
-
-  - [x] 删除wife[名字]
-
-  - [x] [让 | 不让]所有人均可添加wife
-
-  - 注：不同群添加后不会重叠
-
 </details>
 <details>
   <summary>拼音首字母释义工具</summary>
@@ -1067,6 +1051,42 @@ print("run[CQ:image,file="+j["img"]+"]")
 
   - [x] 当图片属于非 neutral 类别时自动发送评价(默认禁用，启用输入 /启用 nsfwauto)
 
+</details>
+<details>
+  <summary>抽wife</summary>
+
+  `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/nwife"`
+
+  - [x] 抽wife[@xxx]
+
+  - [x] 添加wife[名字][图片]
+
+  - [x] 删除wife[名字]
+
+  - [x] [让 | 不让]所有人均可添加wife
+
+  - 注：不同群添加后不会重叠
+
+</details>
+<details>
+  <summary>牛牛大作战</summary>
+
+`import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/niuniu" `
+
+- [x] 打胶
+
+- [x] jj[@xxx]
+
+- [x] 注册牛牛
+
+- [x] 注销牛牛
+
+- [x] 牛子长度排行
+
+- [x] 牛子深度排行
+
+- [x] 查看我的牛牛
+
 </details>
 <details>
   <summary>浅草寺求签</summary>

main.go

@@ -116,6 +116,7 @@ import (
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/novel"            // 铅笔小说网搜索
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/nsfw"             // nsfw图片识别
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/nwife"            // 本地老婆
+	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/niuniu"           // 牛牛大作战
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/omikuji"          // 浅草寺求签
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/poker"            // 抽扑克
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/qqwife"           // 一群一天一夫一妻制群老婆

plugin/niuniu/main.go

@@ -0,0 +1,244 @@
+// Package niuniu 牛牛大作战
+package niuniu
+
+import (
+	"fmt"
+	ctrl "github.com/FloatTech/zbpctrl"
+	"github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/extension/rate"
+	"github.com/wdvxdr1123/ZeroBot/message"
+	"math/rand"
+	"strconv"
+	"strings"
+	"time"
+)
+
+var (
+	en = control.AutoRegister(&ctrl.Options[*zero.Ctx]{
+		DisableOnDefault: false,
+		Brief:            "牛牛大作战",
+		Help: "- 打胶\n" +
+			"- 注册牛牛\n" +
+			"- 注销牛牛\n" +
+			"- 查看我的牛牛\n" +
+			"- jj@xxx\n" +
+			"- 牛子长度排行\n" +
+			"- 牛子深度排行\n",
+		PrivateDataFolder: "niuniu",
+	})
+	dajiaoLimiter = rate.NewManager[string](time.Second*90, 1)
+	jjLimiter     = rate.NewManager[string](time.Second*150, 1)
+)
+
+func init() {
+	en.OnFullMatch("牛子长度排行", zero.OnlyGroup, getdb).SetBlock(true).Handle(func(ctx *zero.Ctx) {
+		gid := ctx.Event.GroupID
+		niuniuList, err := db.readAllTable(gid)
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR: ", err))
+			return
+		}
+		m := niuniuList.positive()
+		if m == nil {
+			ctx.SendChain(message.Text("暂时没有男孩子哦"))
+			return
+		}
+		var messages strings.Builder
+		messages.WriteString("牛子长度排行\n")
+		for i, user := range niuniuList.sort(true) {
+			messages.WriteString(fmt.Sprintf("第%d名  id:%s  长度:%.2fcm\n", i+1,
+				ctx.CardOrNickName(user.UID), user.Length))
+		}
+		msg := ctxext.FakeSenderForwardNode(ctx, message.Text(&messages))
+		if id := ctx.Send(message.Message{msg}).ID(); id == 0 {
+			ctx.Send(message.Text("发送排行失败"))
+		}
+	})
+	en.OnFullMatch("牛子深度排行", zero.OnlyGroup, getdb).SetBlock(true).Handle(func(ctx *zero.Ctx) {
+		gid := ctx.Event.GroupID
+		niuniuList, err := db.readAllTable(gid)
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR: ", err))
+			return
+		}
+		m := niuniuList.negative()
+		if m == nil {
+			ctx.SendChain(message.Text("暂时没有女孩子哦"))
+			return
+		}
+		var messages strings.Builder
+		messages.WriteString("牛牛深度排行榜\n")
+		for i, user := range niuniuList.sort(false) {
+			messages.WriteString(fmt.Sprintf("第%d名  id:%s  长度:%.2fcm\n", i+1,
+				ctx.CardOrNickName(user.UID), user.Length))
+		}
+		msg := ctxext.FakeSenderForwardNode(ctx, message.Text(&messages))
+		if id := ctx.Send(message.Message{msg}).ID(); id == 0 {
+			ctx.Send(message.Text("发送排行失败"))
+		}
+	})
+	en.OnFullMatch("查看我的牛牛", getdb, zero.OnlyGroup).SetBlock(true).Handle(func(ctx *zero.Ctx) {
+		uid := ctx.Event.UserID
+		gid := ctx.Event.GroupID
+		niuniu, err := db.findniuniu(gid, uid)
+		if err != nil {
+			ctx.SendChain(message.Text("你还没有牛牛呢不能查看!"))
+			return
+		}
+		var result strings.Builder
+		sexLong := "长"
+		sex := "♂️"
+		if niuniu < 0 {
+			sexLong = "深"
+			sex = "♀️"
+		}
+		niuniuList, err := db.readAllTable(gid)
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return
+		}
+		result.WriteString(fmt.Sprintf("\n📛%s<%s>的牛牛信息\n⭕性别:%s\n⭕%s度:%.2fcm\n⭕排行:%d\n⭕%s ",
+			ctx.CardOrNickName(uid), strconv.FormatInt(uid, 10),
+			sex, sexLong, niuniu, niuniuList.ranking(niuniu, uid), generateRandomString(niuniu)))
+		ctx.SendChain(message.At(uid), message.Text(&result))
+	})
+	en.OnFullMatchGroup([]string{"dj", "打胶"}, zero.OnlyGroup,
+		getdb).SetBlock(true).Limit(func(ctx *zero.Ctx) *rate.Limiter {
+		lt := dajiaoLimiter.Load(fmt.Sprintf("%d_%d", ctx.Event.GroupID, ctx.Event.UserID))
+		ctx.State["dajiao_last_touch"] = lt.LastTouch()
+		return lt
+	}, func(ctx *zero.Ctx) {
+		timePass := int(time.Since(time.Unix(ctx.State["dajiao_last_touch"].(int64), 0)).Seconds())
+		ctx.SendChain(message.Text(randomChoice([]string{
+			fmt.Sprintf("才过去了%ds时间,你就又要打🦶了，身体受得住吗", timePass),
+			fmt.Sprintf("不行不行，你的身体会受不了的，歇%ds再来吧", 90-timePass),
+			fmt.Sprintf("休息一下吧，会炸膛的！%ds后再来吧", 90-timePass),
+			fmt.Sprintf("打咩哟，你的牛牛会爆炸的，休息%ds再来吧", 90-timePass),
+		})))
+	}).Handle(func(ctx *zero.Ctx) {
+		// 获取群号和用户ID
+		gid := ctx.Event.GroupID
+		uid := ctx.Event.UserID
+		niuniu, err := db.findniuniu(gid, uid)
+		if err != nil {
+			ctx.SendChain(message.Text("请先注册牛牛！"))
+			dajiaoLimiter.Delete(fmt.Sprintf("%d_%d", gid, uid))
+			return
+		}
+		messages, f := generateRandomStingTwo(niuniu)
+		u := userInfo{
+			UID:    uid,
+			Length: f,
+		}
+		ctx.SendChain(message.Text(messages))
+		if err = db.insertniuniu(&u, gid); err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return
+		}
+	})
+	en.OnFullMatch("注册牛牛", zero.OnlyGroup, getdb).SetBlock(true).Handle(func(ctx *zero.Ctx) {
+		gid := ctx.Event.GroupID
+		uid := ctx.Event.UserID
+		if _, err := db.findniuniu(gid, uid); err == nil {
+			ctx.SendChain(message.Text("你已经注册过了"))
+			return
+		}
+		//获取初始长度
+		long := db.randLength()
+		u := userInfo{
+			UID:       uid,
+			Length:    long,
+			UserCount: 0,
+		}
+		//添加数据进入表
+		err := db.insertniuniu(&u, gid)
+		if err != nil {
+			err = db.createGIDTable(gid)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+			err = db.insertniuniu(&u, gid)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR:", err))
+				return
+			}
+		}
+		ctx.SendChain(message.Reply(ctx.Event.GroupID),
+			message.Text("注册成功,你的牛牛现在有", u.Length, "cm"))
+	})
+	en.OnRegex(`jj\[CQ:at,qq=(\d+),name=[\s\S]*\]$`, getdb,
+		zero.OnlyGroup).SetBlock(true).Limit(func(ctx *zero.Ctx) *rate.Limiter {
+		lt := jjLimiter.Load(fmt.Sprintf("%d_%d", ctx.Event.GroupID, ctx.Event.UserID))
+		ctx.State["jj_last_touch"] = lt.LastTouch()
+		return lt
+	}, func(ctx *zero.Ctx) {
+		timePass := int(time.Since(time.Unix(ctx.State["jj_last_touch"].(int64), 0)).Seconds())
+		ctx.SendChain(message.Text(randomChoice([]string{
+			fmt.Sprintf("才过去了%ds时间,你就又要击剑了，真是饥渴难耐啊", timePass),
+			fmt.Sprintf("不行不行，你的身体会受不了的，歇%ds再来吧", 150-timePass),
+			fmt.Sprintf("你这种男同就应该被送去集中营！等待%ds再来吧", 150-timePass),
+			fmt.Sprintf("打咩哟！你的牛牛会炸的，休息%ds再来吧", 150-timePass),
+		})))
+	},
+	).Handle(func(ctx *zero.Ctx) {
+		adduser, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64)
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return
+		}
+		uid := ctx.Event.UserID
+		gid := ctx.Event.GroupID
+		myniuniu, err := db.findniuniu(gid, uid)
+		if err != nil {
+			ctx.SendChain(message.Text("你还没有牛牛快去注册一个吧!"))
+			jjLimiter.Delete(fmt.Sprintf("%d_%d", gid, uid))
+			return
+		}
+		adduserniuniu, err := db.findniuniu(gid, adduser)
+		if err != nil {
+			ctx.SendChain(message.At(uid), message.Text("对方还没有牛牛呢，不能🤺"))
+			jjLimiter.Delete(fmt.Sprintf("%d_%d", gid, uid))
+			return
+		}
+		if uid == adduser {
+			ctx.SendChain(message.Text("你要和谁🤺？你自己吗？"))
+			jjLimiter.Delete(fmt.Sprintf("%d_%d", gid, uid))
+			return
+		}
+		fencingResult, f, f1 := fencing(myniuniu, adduserniuniu)
+		err = db.insertniuniu(&userInfo{UID: uid, Length: f}, gid)
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return
+		}
+		err = db.insertniuniu(&userInfo{UID: adduser, Length: f1}, gid)
+		if err != nil {
+			ctx.SendChain(message.Text("ERROR:", err))
+			return
+		}
+		ctx.SendChain(message.At(uid), message.Text(fencingResult))
+	})
+	en.OnFullMatch("注销牛牛", getdb, zero.OnlyGroup).SetBlock(true).Handle(func(ctx *zero.Ctx) {
+		uid := ctx.Event.UserID
+		gid := ctx.Event.GroupID
+		_, err := db.findniuniu(gid, uid)
+		if err != nil {
+			ctx.SendChain(message.Text("你还没有牛牛呢，咋的你想凭空造一个啊"))
+			return
+		}
+		err = db.deleteniuniu(gid, uid)
+		if err != nil {
+			ctx.SendChain(message.Text("注销失败"))
+			return
+		}
+		ctx.SendChain(message.Text("注销成功,你已经没有牛牛了"))
+	})
+}
+
+func randomChoice(options []string) string {
+	return options[rand.Intn(len(options))]
+}
+