Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2025-12-19 13:59:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修复CD共用问题，新增功能 (#229)

Browse Source
This commit is contained in:
方柳煜 2022-05-16 13:39:48 +08:00 committed by GitHub
parent fbb387bd9f
commit ed2ed8d968
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 428 additions and 326 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

754
plugin/qqwife/qqmapwife.go
View File

@ -1,326 +1,428 @@
// Package qqwife 娶群友 基于“翻牌”和江林大佬的“群老婆”插件魔改作品 // Package qqwife 娶群友 基于“翻牌”和江林大佬的“群老婆”插件魔改作品
package qqwife package qqwife
import ( import (
"math/rand" "math/rand"
"sort" "sort"
"strconv" "strconv"
"sync" "sync"
"time" "time"
zero "github.com/wdvxdr1123/ZeroBot" zero "github.com/wdvxdr1123/ZeroBot"
"github.com/wdvxdr1123/ZeroBot/message" "github.com/wdvxdr1123/ZeroBot/message"
"github.com/FloatTech/zbputils/binary" "github.com/FloatTech/zbputils/binary"
control "github.com/FloatTech/zbputils/control" control "github.com/FloatTech/zbputils/control"
"github.com/FloatTech/zbputils/ctxext" "github.com/FloatTech/zbputils/ctxext"
"github.com/FloatTech/zbputils/math" "github.com/FloatTech/zbputils/math"
) "github.com/wdvxdr1123/ZeroBot/extension/rate"
)
//nolint: asciicheck
type 婚姻登记 struct { //nolint: asciicheck
sync.Mutex type 婚姻登记 struct {
mp map[int64]map[int64]int64 sync.Mutex
} mp map[int64]map[int64]*userinfo
}
//nolint: asciicheck
func 新登记处() (db 婚姻登记) { // 结婚证信息
db.mp = make(map[int64]map[int64]int64, 64) type userinfo struct {
return target int64 //对象身份证号
} username string //户主名称
targetname string //对象名称
func (db *婚姻登记) 重置() { }
db.Lock()
defer db.Unlock() //nolint: asciicheck
for k := range db.mp { func 新登记处() (db 婚姻登记) {
delete(db.mp, k) db.mp = make(map[int64]map[int64]*userinfo, 64)
} return
} }
func (db *婚姻登记) 有登记在(gid int64) (ok bool) { //nolint: asciicheck
db.Lock() func (db *婚姻登记) 重置() {
defer db.Unlock() db.Lock()
mp, ok := db.mp[gid] defer db.Unlock()
if !ok { for k := range db.mp {
return delete(db.mp, k)
} }
for range mp { }
return true
} //nolint: asciicheck
return func (db *婚姻登记) 离婚休妻(gid, wife int64) {
} db.Lock()
defer db.Unlock()
func (db *婚姻登记) 登记情况(gid int64, ctx *zero.Ctx) string { delete(db.mp[gid], -wife)
db.Lock() }
defer db.Unlock()
mp, ok := db.mp[gid] //nolint: asciicheck
if !ok { func (db *婚姻登记) 离婚休夫(gid, husband int64) {
return "" db.Lock()
} defer db.Unlock()
return binary.BytesToString(binary.NewWriterF(func(w *binary.Writer) { delete(db.mp[gid], husband)
w.WriteString("群老公←———→群老婆\n-----------") }
for husband, wife := range mp {
if husband > 0 { //nolint: asciicheck
_ = w.WriteByte('\n') func (db *婚姻登记) 有登记(gid int64) (ok bool) {
w.WriteString(ctx.CardOrNickName(husband)) db.Lock()
w.WriteString(" & ") defer db.Unlock()
w.WriteString(ctx.CardOrNickName(wife)) mp, ok := db.mp[gid]
} if !ok {
} return
})) }
} for range mp {
return true
func (db *婚姻登记) 有妻子(gid, uid int64) (ok bool) { }
db.Lock() return
defer db.Unlock() }
mp, ok := db.mp[gid]
if !ok { //nolint: asciicheck
return func (db *婚姻登记) 花名册(ctx *zero.Ctx, gid int64) string {
} db.Lock()
_, ok = mp[uid] defer db.Unlock()
return mp, ok := db.mp[gid]
} if !ok {
return "民政局的花名册出问题了额..."
func (db *婚姻登记) 查询妻子(gid, uid int64) (wife int64) { }
db.Lock() return binary.BytesToString(binary.NewWriterF(func(w *binary.Writer) {
defer db.Unlock() w.WriteString("群老公←———→群老婆\n-----------")
mp, ok := db.mp[gid] for uid, userinfo := range mp {
if !ok { if uid > 0 {
return _ = w.WriteByte('\n')
} w.WriteString(userinfo.username)
return mp[uid] w.WriteString(" & ")
} w.WriteString(userinfo.targetname)
}
func (db *婚姻登记) 有丈夫(gid, uid int64) (ok bool) { }
db.Lock() }))
defer db.Unlock() }
mp, ok := db.mp[gid]
if !ok { //nolint: asciicheck
return func (db *婚姻登记) 查户口(gid, uid int64) (userinfo *userinfo, gender int, ok bool) {
} db.Lock()
_, ok = mp[-uid] defer db.Unlock()
return gender = 0
} mp, ok := db.mp[gid]
if !ok {
func (db *婚姻登记) 查询丈夫(gid, uid int64) (husband int64) { return
db.Lock() }
defer db.Unlock() userinfo, ok = mp[uid]
mp, ok := db.mp[gid] if !ok {
if !ok { gender = 1
return userinfo, ok = mp[-uid]
} }
return mp[-uid] return
} }
func (db *婚姻登记) 登记(gid, wife, husband int64) { //nolint: asciicheck
db.Lock() func (db *婚姻登记) 登记(gid, uid, target int64, username, targetname string) {
defer db.Unlock() db.Lock()
mp, ok := db.mp[gid] defer db.Unlock()
if !ok { _,ok := db.mp[gid]
mp = make(map[int64]int64, 32) if !ok{
db.mp[gid] = mp db.mp[gid] = make(map[int64]*userinfo, 32)
} }
// 绑定CP // 填写夫妻信息
mp[husband] = wife uidinfo := &userinfo{
mp[-wife] = husband target: target,
} username: username,
targetname: targetname,
var ( }
//nolint: asciicheck targetinfo := &userinfo{
民政局 = 新登记处() target: uid,
lastdate time.Time username: targetname,
sendtext = [...][]string{ targetname: username,
{ }
"今天你向ta表白了, ta羞涩的点了点头同意了!\n", // 民政局登记数据
"你对ta说“以我之名, 冠你指间, 一天相伴, 一天相随”. ta捂着嘴点了点头\n\n", db.mp[gid][uid] = uidinfo
}, db.mp[gid][-target] = targetinfo
{ }
"今天你向ta表白了, ta毫无感情的拒绝了你\n",
"今天你向ta表白了, ta对你说“你是一个非常好的人”\n", var (
"今天你向ta表白了, ta给了你一个拥抱后擦肩而过\n", //nolint: asciicheck
}, 民政局 = 新登记处()
} skillCD = rate.NewManager[string](time.Hour*24, 1)
) lastdate time.Time
sendtext = [...][]string{
func init() { { // 表白成功
engine := control.Register("qqwife", &control.Options{ "今天你向ta表白了，ta羞涩的点了点头同意了！\n",
DisableOnDefault: false, "你对ta说“以我之名，冠你指间，一天相伴，一天相随”.ta捂着嘴点了点头\n\n",
Help: "一群一天一夫一妻制群老婆\n" + },
"- 娶群友\n" + { // 表白失败
"- 娶[老婆QQ号|@老婆QQ]\n(注:单身专属技能,CD24H,不跨天刷新)\n" + "今天你向ta表白了，ta毫无感情的拒绝了你",
"- 群老婆列表", "今天你向ta表白了，ta对你说“你是一个非常好的人”",
}) "今天你向ta表白了，ta给了你一个拥抱后擦肩而过",
engine.OnFullMatch("娶群友", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser). },
Handle(func(ctx *zero.Ctx) { { //ntr成功
if time.Now().Day() != lastdate.Day() { "你处心积虑的接近ta，ta最终选择跟随你\n",
民政局.重置() },
// 更新时间 }
lastdate = time.Now() )
}
// 先判断是否已经娶过或者被娶 func init() {
gid := ctx.Event.GroupID engine := control.Register("qqwife", &control.Options{
uid := ctx.Event.UserID DisableOnDefault: false,
// 如果娶过 Help: "一群一天一夫一妻制群老婆\n（每天凌晨刷新CP）\n" +
wife := 民政局.查询妻子(gid, uid) "- 娶群友\n- 群老婆列表\n" +
if wife > 0 { "--------------------------------\n以下技能每人只能二选一\n CD24H，不跨天刷新\n--------------------------------\n" +
ctx.SendChain( "- (娶|嫁)@对方QQ\n- 当[对方Q号|@对方QQ]的小三\n",
message.At(uid), })
message.Text("今天你的群老婆是"), engine.OnFullMatch("娶群友", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(wife, 10)+"&s=640").Add("cache", 0), Handle(func(ctx *zero.Ctx) {
message.Text( if time.Now().Day() != lastdate.Day() {
"\n", 民政局.重置()
"[", ctx.CardOrNickName(wife), "]", // 更新时间
"(", wife, ")哒", lastdate = time.Now()
), }
) gid := ctx.Event.GroupID
return uid := ctx.Event.UserID
} targetinfo, status, ok := 民政局.查户口(gid, uid)
// 如果被娶过 if ok {
husband := 民政局.查询丈夫(gid, uid) switch status {
if husband > 0 { case 0: // 娶过别人
ctx.SendChain( ctx.SendChain(
message.At(uid), message.At(uid),
message.Text("今天你被娶了, 群老公是"), message.Text("今天你的群老婆是"),
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(husband, 10)+"&s=640").Add("cache", 0), message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(targetinfo.target, 10)+"&s=640").Add("cache", 0),
message.Text( message.Text(
"\n", "\n",
"[", ctx.CardOrNickName(husband), "]", "[", targetinfo.targetname, "]",
"(", husband, ")哒", "(", targetinfo.target, ")哒",
), ),
) )
return default: // 嫁给别人
} ctx.SendChain(
// 无缓存获取群员列表 message.At(uid),
temp := ctx.GetThisGroupMemberListNoCache().Array() message.Text("今天你的群老公是"),
sort.SliceStable(temp, func(i, j int) bool { message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(targetinfo.target, 10)+"&s=640").Add("cache", 0),
return temp[i].Get("last_sent_time").Int() < temp[j].Get("last_sent_time").Int() message.Text(
}) "\n",
temp = temp[math.Max(0, len(temp)-30):] "[", targetinfo.targetname, "]",
// 将已经娶过的人剔除 "(", targetinfo.target, ")哒",
qqgrouplist := make([]int64, 0, len(temp)) ),
for k := 0; k < len(temp); k++ { )
usr := temp[k].Get("user_id").Int() }
if 民政局.有妻子(gid, usr) || 民政局.有丈夫(gid, usr) { return
continue }
} // 无缓存获取群员列表
qqgrouplist = append(qqgrouplist, usr) temp := ctx.GetThisGroupMemberListNoCache().Array()
} sort.SliceStable(temp, func(i, j int) bool {
// 没有人（只剩自己）的时候 return temp[i].Get("last_sent_time").Int() < temp[j].Get("last_sent_time").Int()
if len(qqgrouplist) == 0 { })
ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧")) temp = temp[math.Max(0, len(temp)-30):]
return // 将已经娶过的人剔除
} qqgrouplist := make([]int64, 0, len(temp))
// 随机抽娶 for k := 0; k < len(temp); k++ {
wife = qqgrouplist[rand.Intn(len(qqgrouplist))] usr := temp[k].Get("user_id").Int()
if wife == uid { // 如果是自己 _, _, ok := 民政局.查户口(gid, usr)
ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧")) if ok {
return continue
} }
// 绑定CP qqgrouplist = append(qqgrouplist, usr)
民政局.登记(gid, wife, uid) }
// 输出结果 // 没有人（只剩自己）的时候
ctx.SendChain( if len(qqgrouplist) == 0 {
message.At(uid), ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧"))
message.Text("今天你的群老婆是"), return
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(wife, 10)+"&s=640").Add("cache", 0), }
message.Text( // 随机抽娶
"\n", fiancee := qqgrouplist[rand.Intn(len(qqgrouplist))]
"[", ctx.CardOrNickName(wife), "]", if fiancee == uid { // 如果是自己
"(", wife, ")哒", ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧"))
), return
) }
}) // 去民政局办证
// 单身狗专属技能 民政局.登记(gid, uid, fiancee, ctx.CardOrNickName(uid), ctx.CardOrNickName(fiancee))
engine.OnRegex(`^娶(\d+|\[CQ:at,qq=(\d+)\])`, zero.OnlyGroup, checkdog).SetBlock(true). // 请大家吃席
Limit(ctxext.NewLimiterManager(time.Hour*12, 1).LimitByUser). ctx.SendChain(
Handle(func(ctx *zero.Ctx) { message.At(uid),
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64) message.Text("今天你的群老婆是"),
if err != nil { message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(fiancee, 10)+"&s=640").Add("cache", 0),
fiancee, _ = strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64) message.Text(
} "\n",
gid := ctx.Event.GroupID "[", ctx.CardOrNickName(fiancee), "]",
uid := ctx.Event.UserID "(", fiancee, ")哒",
if rand.Intn(2) == 1 { ),
民政局.登记(gid, fiancee, uid) )
// 输出结果 })
ctx.SendChain( // 单身技能
message.Text(sendtext[0][rand.Intn(len(sendtext[0]))]), engine.OnRegex(`^(娶|嫁)\[CQ:at,qq=(\d+)\]`, zero.OnlyGroup, checkdog).SetBlock(true).Limit(cdcheck, iscding).
message.At(uid), Handle(func(ctx *zero.Ctx) {
message.Text("今天你的群老婆是"), choice := ctx.State["regex_matched"].([]string)[1]
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(fiancee, 10)+"&s=640").Add("cache", 0), fiancee, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
message.Text( uid := ctx.Event.UserID
"\n", if uid == fiancee { // 如果是自己
"[", ctx.CardOrNickName(fiancee), "]", ctx.SendChain(message.Text("今日获得成就：自恋狂"))
"(", fiancee, ")哒", return
), }
) if rand.Intn(2) == 1 { // 二分之一的概率表白成功
return gid := ctx.Event.GroupID
} // 去民政局登记
ctx.SendChain(message.Text(sendtext[1][rand.Intn(len(sendtext[1]))])) var choicetext string
}) switch choice {
engine.OnFullMatch("群老婆列表", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser). case "娶":
Handle(func(ctx *zero.Ctx) { 民政局.登记(gid, uid, fiancee, ctx.CardOrNickName(uid), ctx.CardOrNickName(fiancee))
if !民政局.有登记在(ctx.Event.GroupID) { choicetext = "今天你的群老婆是"
ctx.SendChain(message.Text("你群并没有任何的CP额")) default:
return 民政局.登记(gid, fiancee, uid, ctx.CardOrNickName(fiancee), ctx.CardOrNickName(uid))
} choicetext = "今天你的群老公是"
ctx.SendChain(message.Text(民政局.登记情况(ctx.Event.GroupID, ctx))) }
}) // 请大家吃席
} ctx.SendChain(
message.Text(sendtext[0][rand.Intn(len(sendtext[0]))]),
// 注入判断 是否为单身狗 message.At(uid),
func checkdog(ctx *zero.Ctx) bool { message.Text(choicetext),
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64) message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(fiancee, 10)+"&s=640").Add("cache", 0),
// fmt.Println("1:", fiancee) message.Text(
if err != nil { "\n",
fiancee, _ = strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64) "[", ctx.CardOrNickName(fiancee), "]",
// fmt.Println("2:", fiancee) "(", fiancee, ")哒",
} ),
gid := ctx.Event.GroupID )
uid := ctx.Event.UserID return
if uid == fiancee { }
ctx.SendChain(message.Text("今日获得成就：自恋狂")) ctx.SendChain(message.Text(sendtext[1][rand.Intn(len(sendtext[1]))]))
return false })
} // NTR技能
// 如果用户娶过 engine.OnRegex(`^当(\[CQ:at,qq=(\d+)\] |(\d+))的小三`, zero.OnlyGroup, checkcp).SetBlock(true).Limit(cdcheck, iscding).
if o := 民政局.查询妻子(gid, uid); o > 0 { Handle(func(ctx *zero.Ctx) {
switch o { fid := ctx.State["regex_matched"].([]string)
case fiancee: fiancee, _ := strconv.ParseInt(fid[2]+fid[3], 10, 64)
ctx.SendChain(message.Text("笨蛋~你明明已经娶了啊w")) if rand.Intn(10)/4 != 0 { // 十分之三的概率NTR成功
default: ctx.SendChain(message.Text("你的ntr计划失败了"))
ctx.SendChain(message.Text("笨蛋~你家里还有个吃白饭的w")) return
} }
return false gid := ctx.Event.GroupID
} uid := ctx.Event.UserID
// 如果用户被娶过 // 判断target是老公还是老婆
if o := 民政局.查询丈夫(gid, uid); o > 0 { var choicetext string
switch o { targetinfo, gender, _ := 民政局.查户口(gid, fiancee)
case fiancee: switch gender{
ctx.SendChain(message.Text("笨蛋~你明明已经嫁给他了啊w")) case 0:
default: // 让对象离婚
ctx.SendChain(message.Text("该是0就是0, 当0有什么不好")) 民政局.离婚休妻(gid, targetinfo.target)
} // 和对象结婚登记
return false choicetext = "老公"
} 民政局.登记(gid, fiancee, uid, ctx.CardOrNickName(fiancee), ctx.CardOrNickName(uid))
// 如果未婚妻娶过 default:
if o := 民政局.查询妻子(gid, fiancee); o > 0 { // 让对象离婚
switch o { 民政局.离婚休夫(gid, targetinfo.target)
case uid: // 和对象结婚登记
ctx.SendChain(message.Text("笨蛋~你明明已经嫁给他了啊w")) choicetext = "老婆"
default: 民政局.登记(gid, uid, fiancee, ctx.CardOrNickName(uid), ctx.CardOrNickName(fiancee))
ctx.SendChain(message.Text("他有别的女人了, 你该放下了")) }
} // 输出结果
return false ctx.SendChain(
} message.Text(sendtext[2][rand.Intn(len(sendtext[2]))]),
// 如果未婚妻被娶过 message.At(uid),
if o := 民政局.查询丈夫(gid, fiancee); o > 0 { message.Text("今天你的群"+choicetext+"是"),
switch o { message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(fiancee, 10)+"&s=640").Add("cache", 0),
case uid: message.Text(
ctx.SendChain(message.Text("笨蛋~你明明已经娶了啊w")) "\n",
default: "[", ctx.CardOrNickName(fiancee), "]",
ctx.SendChain(message.Text("这是一个纯爱的世界, 拒绝NTR")) "(", fiancee, ")哒",
} ),
return false )
} })
return true engine.OnFullMatch("群老婆列表", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
} Handle(func(ctx *zero.Ctx) {
if !民政局.有登记(ctx.Event.GroupID) {
ctx.SendChain(message.Text("你群并没有任何的CP额"))
return
}
ctx.SendChain(message.Text(民政局.花名册(ctx, ctx.Event.GroupID)))
})
}
// 以群号和昵称为限制
func cdcheck(ctx *zero.Ctx) *rate.Limiter {
limitID := strconv.FormatInt(ctx.Event.GroupID, 10) + strconv.FormatInt(ctx.Event.UserID, 10)
return skillCD.Load(limitID)
}
func iscding(ctx *zero.Ctx) {
ctx.SendChain(message.Text("你的技能现在正在CD中"))
}
// 注入判断 是否为单身
func checkdog(ctx *zero.Ctx) bool {
gid := ctx.Event.GroupID
if !民政局.有登记(gid) {
return true // 如果没有人登记，说明全是单身
}
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
if err != nil {
ctx.SendChain(message.Text("额，你的target好像不存在？"))
return false
}
uid := ctx.Event.UserID
if uid == fiancee {
ctx.SendChain(message.Text("今日获得成就：自恋狂"))
return false
}
// 获取用户信息
uidtarget, uidstatus, ok1 := 民政局.查户口(gid, uid)
_, fianceestatus, ok2 := 民政局.查户口(gid, fiancee)
if !ok1 && !ok2 { // 必须是两个单身
return true
}
if uidtarget.target == fiancee { // 如果本就是一块
ctx.SendChain(message.Text("笨蛋~你们明明已经在一起了啊w"))
return false
}
if ok1 {
switch uidstatus {
case 0: // 如果如为攻
ctx.SendChain(message.Text("笨蛋~你家里还有个吃白饭的w"))
default: // 如果为受
ctx.SendChain(message.Text("该是0就是0，当0有什么不好"))
}
return false
}
if ok2 {
switch fianceestatus {
case 0: // 如果如为攻
ctx.SendChain(message.Text("他有别的女人了，你该放下了"))
default: // 如果为受
ctx.SendChain(message.Text("这是一个纯爱的世界，拒绝NTR"))
}
return false
}
return true
}
// 注入判断 是否满足小三要求
func checkcp(ctx *zero.Ctx) bool {
// 检查群内是否有人登记了
gid := ctx.Event.GroupID
if !民政局.有登记(gid) {
ctx.SendChain(message.Text("ta无法达成你当小三的条件"))
return false
}
// 检查target
fid := ctx.State["regex_matched"].([]string)
fiancee, err := strconv.ParseInt(fid[2]+fid[3], 10, 64)
if err != nil {
ctx.SendChain(message.Text("额，你的对象好像不存在?"))
return false
}
// 检查用户是否登记过
uid := ctx.Event.UserID
userinfo, uidstatus, ok := 民政局.查户口(gid, uid)
if ok {
if userinfo.target == fiancee { // 如果本就是一块
ctx.SendChain(message.Text("笨蛋~你们明明已经在一起了啊w"))
return false
}
switch uidstatus {
case 0: // 如果如为攻
ctx.SendChain(message.Text("抱歉，建国之后不支持后宫"))
default: //如果为受
ctx.SendChain(message.Text("该是0就是0，当0有什么不好"))
}
return false
}
_, _, ok = 民政局.查户口(gid, fiancee)
if !ok {
ctx.SendChain(message.Text("ta无法达成你当小三的条件"))
return false
}
return true
}