mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
324 lines
10 KiB
Go
324 lines
10 KiB
Go
// Package niuniu 牛牛大作战
|
||
package niuniu
|
||
|
||
import (
|
||
"fmt"
|
||
"math/rand"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/FloatTech/AnimeAPI/wallet"
|
||
ctrl "github.com/FloatTech/zbpctrl"
|
||
"github.com/FloatTech/zbputils/control"
|
||
"github.com/FloatTech/zbputils/ctxext"
|
||
"github.com/RomiChan/syncx"
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
"github.com/wdvxdr1123/ZeroBot/extension/rate"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
)
|
||
|
||
type lastLength struct {
|
||
TimeLimit time.Time
|
||
Count int
|
||
Length float64
|
||
}
|
||
|
||
var (
|
||
en = control.AutoRegister(&ctrl.Options[*zero.Ctx]{
|
||
DisableOnDefault: false,
|
||
Brief: "牛牛大作战",
|
||
Help: "- 打胶\n" +
|
||
"- 赎牛牛 （cd：30分钟）\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)
|
||
jjCount = syncx.Map[string, *lastLength]{}
|
||
)
|
||
|
||
func init() {
|
||
en.OnFullMatch("赎牛牛", zero.OnlyGroup, getdb).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
||
gid := ctx.Event.GroupID
|
||
uid := ctx.Event.UserID
|
||
last, ok := jjCount.Load(fmt.Sprintf("%d_%d", gid, uid))
|
||
if !ok {
|
||
ctx.SendChain(message.Text("你还没有被厥呢"))
|
||
return
|
||
}
|
||
if time.Since(last.TimeLimit) > time.Minute*30 {
|
||
ctx.SendChain(message.Text("时间已经过期了,牛牛已被收回!"))
|
||
jjCount.Delete(fmt.Sprintf("%d_%d", gid, uid))
|
||
return
|
||
}
|
||
if last.Count < 6 {
|
||
ctx.SendChain(message.Text("你还没有被厥够6次呢,不能赎牛牛"))
|
||
return
|
||
}
|
||
money := wallet.GetWalletOf(uid)
|
||
if money < 100 {
|
||
ctx.SendChain(message.Text("赎牛牛需要100ATRI币，快去赚钱吧"))
|
||
return
|
||
}
|
||
err := wallet.InsertWalletOf(uid, -100)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR:", err))
|
||
return
|
||
}
|
||
u := &userInfo{
|
||
UID: uid,
|
||
Length: last.Length,
|
||
UserCount: 0,
|
||
}
|
||
err = db.insertniuniu(u, gid)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR:", err))
|
||
return
|
||
}
|
||
jjCount.Delete(fmt.Sprintf("%d_%d", gid, uid))
|
||
ctx.SendChain(message.At(uid), message.Text(fmt.Sprintf("恭喜你!成功赎回牛牛,当前长度为:%.2fcm", last.Length)))
|
||
})
|
||
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 m.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 m.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\s?(\[CQ:at,(?:\S*,)?qq=(\d+)(?:,\S*)?\]|(\d+))$`, 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) {
|
||
fiancee := ctx.State["regex_matched"].([]string)
|
||
adduser, err := strconv.ParseInt(fiancee[2]+fiancee[3], 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, UserCount: 0}, gid)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR:", err))
|
||
return
|
||
}
|
||
err = db.insertniuniu(&userInfo{UID: adduser, Length: f1, UserCount: 0}, gid)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR:", err))
|
||
return
|
||
}
|
||
ctx.SendChain(message.At(uid), message.Text(" ", fencingResult))
|
||
j := fmt.Sprintf("%d_%d", gid, adduser)
|
||
count, ok := jjCount.Load(j)
|
||
var c lastLength
|
||
if !ok {
|
||
c = lastLength{
|
||
TimeLimit: time.Now(),
|
||
Count: 1,
|
||
Length: adduserniuniu,
|
||
}
|
||
} else {
|
||
c = lastLength{
|
||
TimeLimit: c.TimeLimit,
|
||
Count: count.Count + 1,
|
||
Length: count.Length,
|
||
}
|
||
}
|
||
jjCount.Store(j, &c)
|
||
if c.Count > 5 {
|
||
ctx.SendChain(message.Text(randomChoice([]string{fmt.Sprintf("你们太厉害了，对方已经被你们打了%d次了，你们可以继续找他🤺", c.Count),
|
||
"你们不要再找ta🤺啦！"})))
|
||
if c.Count < 7 {
|
||
id := ctx.SendPrivateMessage(adduser,
|
||
message.Text(fmt.Sprintf("你在%d群里已经被厥冒烟了，快去群里赎回你原本的牛牛!\n发送:`赎牛牛`即可！", gid)))
|
||
if id == 0 {
|
||
ctx.SendChain(message.At(adduser), message.Text("快发送`赎牛牛`来赎回你原本的牛牛!"))
|
||
}
|
||
}
|
||
}
|
||
})
|
||
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))]
|
||
}
|