diff --git a/plugin/qqwife/qqmapwife.go b/plugin/qqwife/qqmapwife.go index 1043a05f..e9097468 100644 --- a/plugin/qqwife/qqmapwife.go +++ b/plugin/qqwife/qqmapwife.go @@ -5,31 +5,136 @@ import ( "math/rand" "sort" "strconv" - "strings" "sync" "time" zero "github.com/wdvxdr1123/ZeroBot" "github.com/wdvxdr1123/ZeroBot/message" + "github.com/FloatTech/zbputils/binary" control "github.com/FloatTech/zbputils/control" "github.com/FloatTech/zbputils/ctxext" "github.com/FloatTech/zbputils/math" ) +type 婚姻登记 struct { + sync.Mutex + mp map[int64]map[int64]int64 +} + +func 新登记处() (db 婚姻登记) { + db.mp = make(map[int64]map[int64]int64, 64) + return +} + +func (db *婚姻登记) 重置() { + db.Lock() + defer db.Unlock() + for k := range db.mp { + delete(db.mp, k) + } +} + +func (db *婚姻登记) 有登记在(gid int64) (ok bool) { + db.Lock() + defer db.Unlock() + mp, ok := db.mp[gid] + if !ok { + return + } + for range mp { + return true + } + return +} + +func (db *婚姻登记) 登记情况(gid int64, ctx *zero.Ctx) string { + db.Lock() + defer db.Unlock() + mp, ok := db.mp[gid] + if !ok { + return "" + } + return binary.BytesToString(binary.NewWriterF(func(w *binary.Writer) { + w.WriteString("群老公←———→群老婆\n-----------") + for husband, wife := range mp { + if husband > 0 { + w.WriteByte('\n') + w.WriteString(ctx.CardOrNickName(husband)) + w.WriteString(" & ") + w.WriteString(ctx.CardOrNickName(wife)) + } + } + })) +} + +func (db *婚姻登记) 有妻子(gid, uid int64) (ok bool) { + db.Lock() + defer db.Unlock() + mp, ok := db.mp[gid] + if !ok { + return + } + _, ok = mp[uid] + return +} + +func (db *婚姻登记) 查询妻子(gid, uid int64) (wife int64) { + db.Lock() + defer db.Unlock() + mp, ok := db.mp[gid] + if !ok { + return + } + return mp[uid] +} + +func (db *婚姻登记) 有丈夫(gid, uid int64) (ok bool) { + db.Lock() + defer db.Unlock() + mp, ok := db.mp[gid] + if !ok { + return + } + _, ok = mp[-uid] + return +} + +func (db *婚姻登记) 查询丈夫(gid, uid int64) (husband int64) { + db.Lock() + defer db.Unlock() + mp, ok := db.mp[gid] + if !ok { + return + } + return mp[-uid] +} + +func (db *婚姻登记) 登记(gid, wife, husband int64) { + db.Lock() + defer db.Unlock() + mp, ok := db.mp[gid] + if !ok { + mp = make(map[int64]int64, 32) + db.mp[gid] = mp + } + // 绑定CP + mp[husband] = wife + mp[-wife] = husband +} + var ( - qqwifegroup = make(map[int64]map[int64]int64, 64) // 64个群的预算大小 - lastdate time.Time - mu sync.Mutex - sendtext = [...][]string{ + 民政局 = 新登记处() + lastdate time.Time + sendtext = [...][]string{ { - "今天你向ta表白了,ta羞涩的点了点头同意了!\n", - "你对ta说“以我之名,冠你指间,一天相伴,一天相随”.ta捂着嘴点了点头\n\n", + "今天你向ta表白了, ta羞涩的点了点头同意了!\n", + "你对ta说“以我之名, 冠你指间, 一天相伴, 一天相随”. ta捂着嘴点了点头\n\n", }, { - "今天你向ta表白了,ta毫无感情的拒绝了你\n", - "今天你向ta表白了,ta对你说“你是一个非常好的人”\n", - "今天你向ta表白了,ta给了你一个拥抱后擦肩而过\n", + "今天你向ta表白了, ta毫无感情的拒绝了你\n", + "今天你向ta表白了, ta对你说“你是一个非常好的人”\n", + "今天你向ta表白了, ta给了你一个拥抱后擦肩而过\n", }, } ) @@ -44,21 +149,17 @@ func init() { }) engine.OnFullMatch("娶群友", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser). Handle(func(ctx *zero.Ctx) { - mu.Lock() - defer mu.Unlock() if time.Now().Day() != lastdate.Day() { - qqwifegroup = make(map[int64]map[int64]int64, 64) // 跨天就重新初始化数据 - } - // 看列表是为为空 - gid := ctx.Event.GroupID - if qqwifegroup[gid] == nil { - qqwifegroup[gid] = make(map[int64]int64, 32) + 民政局.重置() + // 更新时间 + lastdate = time.Now() } // 先判断是否已经娶过或者被娶 + gid := ctx.Event.GroupID uid := ctx.Event.UserID // 如果娶过 - wife, ok := qqwifegroup[gid][uid] - if ok { + wife := 民政局.查询妻子(gid, uid) + if wife > 0 { ctx.SendChain( message.At(uid), message.Text("今天你的群老婆是"), @@ -72,11 +173,11 @@ func init() { return } // 如果被娶过 - husband, ok := qqwifegroup[gid][-uid] - if ok { + husband := 民政局.查询丈夫(gid, uid) + if husband > 0 { ctx.SendChain( 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.Text( "\n", @@ -94,37 +195,26 @@ func init() { temp = temp[math.Max(0, len(temp)-30):] // 将已经娶过的人剔除 qqgrouplist := make([]int64, 0, len(temp)) - if len(qqwifegroup) == 0 { - for k := 0; k < len(temp); k++ { - qqgrouplist = append(qqgrouplist, temp[k].Get("user_id").Int()) - } - } else { - for k := 0; k < len(temp); k++ { - _, ok := qqwifegroup[gid][temp[k].Get("user_id").Int()] - if ok { - continue - } - _, ok = qqwifegroup[gid][-temp[k].Get("user_id").Int()] - if ok { - continue - } - qqgrouplist = append(qqgrouplist, temp[k].Get("user_id").Int()) + for k := 0; k < len(temp); k++ { + usr := temp[k].Get("user_id").Int() + if 民政局.有妻子(gid, usr) || 民政局.有丈夫(gid, usr) { + continue } + qqgrouplist = append(qqgrouplist, usr) } // 没有人(只剩自己)的时候 if len(qqgrouplist) == 0 { - ctx.SendChain(message.Text("噢,此时此刻你还是一只单身狗,等待下一次情缘吧")) + ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧")) return } // 随机抽娶 wife = qqgrouplist[rand.Intn(len(qqgrouplist))] if wife == uid { // 如果是自己 - ctx.SendChain(message.Text("噢,此时此刻你还是一只单身狗,等待下一次情缘吧")) + ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧")) return } // 绑定CP - qqwifegroup[gid][uid] = wife - qqwifegroup[gid][-wife] = uid + 民政局.登记(gid, wife, uid) // 输出结果 ctx.SendChain( message.At(uid), @@ -136,15 +226,11 @@ func init() { "(", wife, ")哒", ), ) - // 更新时间 - lastdate = time.Now() }) - // 单生狗专属技能 - var singledogCD = ctxext.NewLimiterManager(time.Hour*24, 1) - engine.OnRegex(`^娶(\d+|\[CQ:at,qq=(\d+)\])`, zero.OnlyGroup, checkdog).SetBlock(true).Limit(singledogCD.LimitByUser). + // 单身狗专属技能 + engine.OnRegex(`^娶(\d+|\[CQ:at,qq=(\d+)\])`, zero.OnlyGroup, checkdog).SetBlock(true). + Limit(ctxext.NewLimiterManager(time.Hour*12, 1).LimitByUser). Handle(func(ctx *zero.Ctx) { - mu.Lock() - defer mu.Unlock() fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64) if err != nil { fiancee, _ = strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64) @@ -152,12 +238,7 @@ func init() { gid := ctx.Event.GroupID uid := ctx.Event.UserID if rand.Intn(2) == 1 { - // 绑定CP - if qqwifegroup[gid] == nil { - qqwifegroup[gid] = make(map[int64]int64, 32) - } - qqwifegroup[gid][uid] = fiancee - qqwifegroup[gid][-fiancee] = uid + 民政局.登记(gid, fiancee, uid) // 输出结果 ctx.SendChain( message.Text(sendtext[0][rand.Intn(len(sendtext[0]))]), @@ -170,31 +251,17 @@ func init() { "(", fiancee, ")哒", ), ) - } else { - ctx.SendChain(message.Text(sendtext[1][rand.Intn(len(sendtext[1]))])) + return } + ctx.SendChain(message.Text(sendtext[1][rand.Intn(len(sendtext[1]))])) }) engine.OnFullMatch("群老婆列表", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser). Handle(func(ctx *zero.Ctx) { - mu.Lock() - defer mu.Unlock() - group, ok := qqwifegroup[ctx.Event.GroupID] - if !ok { + if !民政局.有登记在(ctx.Event.GroupID) { ctx.SendChain(message.Text("你群并没有任何的CP额")) return } - if len(group) == 0 { - ctx.SendChain(message.Text("你群并没有任何的CP额")) - return - } - cplist := make([]string, 1, len(group)+1) - cplist[0] = "群老公←———→群老婆\n----------------------" - for husband, wife := range group { - if husband > 0 { - cplist = append(cplist, ctx.CardOrNickName(husband)+" & "+ctx.CardOrNickName(wife)) - } - } - ctx.SendChain(message.Text(strings.Join(cplist, "\n"))) + ctx.SendChain(message.Text(民政局.登记情况(ctx.Event.GroupID, ctx))) }) } @@ -213,9 +280,8 @@ func checkdog(ctx *zero.Ctx) bool { return false } // 如果用户娶过 - husband, ok := qqwifegroup[gid][uid] - if ok { - switch husband { + if o := 民政局.查询妻子(gid, uid); o > 0 { + switch o { case fiancee: ctx.SendChain(message.Text("笨蛋~你明明已经娶了啊w")) default: @@ -224,35 +290,32 @@ func checkdog(ctx *zero.Ctx) bool { return false } // 如果用户被娶过 - wife, ok := qqwifegroup[gid][-uid] - if ok { - switch wife { + if o := 民政局.查询丈夫(gid, uid); o > 0 { + switch o { case fiancee: ctx.SendChain(message.Text("笨蛋~你明明已经嫁给他了啊w")) default: - ctx.SendChain(message.Text("该是0就是0,当0有什么不好")) + ctx.SendChain(message.Text("该是0就是0, 当0有什么不好")) } return false } // 如果未婚妻娶过 - wife, ok = qqwifegroup[gid][fiancee] - if ok { - switch wife { + if o := 民政局.查询妻子(gid, fiancee); o > 0 { + switch o { case uid: ctx.SendChain(message.Text("笨蛋~你明明已经嫁给他了啊w")) default: - ctx.SendChain(message.Text("他有别的女人了,你该放下了")) + ctx.SendChain(message.Text("他有别的女人了, 你该放下了")) } return false } // 如果未婚妻被娶过 - husband, ok = qqwifegroup[gid][-fiancee] - if ok { - switch husband { + if o := 民政局.查询丈夫(gid, fiancee); o > 0 { + switch o { case uid: ctx.SendChain(message.Text("笨蛋~你明明已经娶了啊w")) default: - ctx.SendChain(message.Text("这是一个纯爱的世界,拒绝NTR")) + ctx.SendChain(message.Text("这是一个纯爱的世界, 拒绝NTR")) } return false }