diff --git a/plugin/qqwife/qqmapwife.go b/plugin/qqwife/qqmapwife.go index e535daab..f9b45dd9 100644 --- a/plugin/qqwife/qqmapwife.go +++ b/plugin/qqwife/qqmapwife.go @@ -1,4 +1,4 @@ -// Package qqwife 娶群友 基于“翻牌”和江林大佬的“群老婆”插件魔改作品 +// Package qqwife 娶群友 基于“翻牌”和江林大佬的“群老婆”插件魔改作品 package qqwife import ( @@ -21,6 +21,17 @@ var ( qqwifegroup = make(map[int64]map[int64]int64, 64) // 64个群的预算大小 lastdate time.Time mu sync.Mutex + sendtext = [...][]string{ + { + "今天你向ta表白了，ta羞涩的点了点头同意了！\n", + "你对ta说“以我之名，冠你指间，一天相伴，一天相随”.ta捂着嘴点了点头\n\n", + }, + { + "今天你向ta表白了，ta毫无感情的拒绝了你\n", + "今天你向ta表白了，ta对你说“你是一个非常好的人”\n", + "今天你向ta表白了，ta给了你一个拥抱后擦肩而过\n", + }, + } ) func init() { @@ -28,6 +39,7 @@ func init() { DisableOnDefault: false, Help: "一群一天一夫一妻制群老婆\n" + "- 娶群友\n" + + "- 娶[老婆QQ号|@老婆QQ]\n(注:单身专属技能,CD24H,不跨天刷新)\n" + "- 群老婆列表", }) engine.OnFullMatch("娶群友", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser). @@ -37,8 +49,12 @@ func init() { 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) + } + // 先判断是否已经娶过或者被娶 uid := ctx.Event.UserID // 如果娶过 wife, ok := qqwifegroup[gid][uid] @@ -46,7 +62,7 @@ func init() { ctx.SendChain( message.At(uid), message.Text("今天你的群老婆是"), - message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(wife, 10)+"&s=640"), + message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(wife, 10)+"&s=640").Add("cache", 0), message.Text( "\n", "[", ctx.CardOrNickName(wife), "]", @@ -56,20 +72,19 @@ func init() { return } // 如果被娶过 - for husband, wife := range qqwifegroup[gid] { - if wife == uid { // 如果为0且是在本群抽的就输出 - ctx.SendChain( - message.At(uid), - message.Text("今天你被娶了，群老公是"), - message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(husband, 10)+"&s=640"), - message.Text( - "\n", - "[", ctx.CardOrNickName(husband), "]", - "(", husband, ")哒", - ), - ) - return - } + husband, ok := qqwifegroup[gid][-uid] + if ok { + ctx.SendChain( + message.At(uid), + message.Text("今天你被娶了，群老公是"), + message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(husband, 10)+"&s=640").Add("cache", 0), + message.Text( + "\n", + "[", ctx.CardOrNickName(husband), "]", + "(", husband, ")哒", + ), + ) + return } // 无缓存获取群员列表 temp := ctx.GetThisGroupMemberListNoCache().Array() @@ -86,9 +101,14 @@ func init() { } else { for k := 0; k < len(temp); k++ { _, ok := qqwifegroup[gid][temp[k].Get("user_id").Int()] - if !ok { - qqgrouplist = append(qqgrouplist, 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()) } } // 没有人（只剩自己）的时候 @@ -103,15 +123,13 @@ func init() { return } // 绑定CP - if qqwifegroup[gid] == nil { - qqwifegroup[gid] = make(map[int64]int64, 32) - } qqwifegroup[gid][uid] = wife + qqwifegroup[gid][-wife] = uid // 输出结果 ctx.SendChain( message.At(uid), message.Text("今天你的群老婆是"), - message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(wife, 10)+"&s=640"), + message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(wife, 10)+"&s=640").Add("cache", 0), message.Text( "\n", "[", ctx.CardOrNickName(wife), "]", @@ -121,6 +139,41 @@ func init() { // 更新时间 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). + 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) + } + 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 + // 输出结果 + ctx.SendChain( + message.Text(sendtext[0][rand.Intn(len(sendtext[0]))]), + message.At(uid), + message.Text("今天你的群老婆是"), + message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(fiancee, 10)+"&s=640").Add("cache", 0), + message.Text( + "\n", + "[", ctx.CardOrNickName(fiancee), "]", + "(", fiancee, ")哒", + ), + ) + } else { + 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() @@ -135,10 +188,73 @@ func init() { return } cplist := make([]string, 1, len(group)+1) - cplist[0] = "群老公←———→群老婆\n--------------------------" + cplist[0] = "群老公←———→群老婆\n----------------------" for husband, wife := range group { - cplist = append(cplist, ctx.CardOrNickName(husband)+" & "+ctx.CardOrNickName(wife)) + if husband > 0 { + cplist = append(cplist, ctx.CardOrNickName(husband)+" & "+ctx.CardOrNickName(wife)) + } } ctx.SendChain(message.Text(strings.Join(cplist, "\n"))) }) } + +//注入判断 是否为单身狗 +func checkdog(ctx *zero.Ctx) bool { + fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64) + //fmt.Println("1:", fiancee) + if err != nil { + fiancee, _ = strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64) + //fmt.Println("2:", fiancee) + } + gid := ctx.Event.GroupID + uid := ctx.Event.UserID + if uid == fiancee { + ctx.SendChain(message.Text("今日获得成就：自恋狂")) + return false + } + // 如果用户娶过 + husband, ok := qqwifegroup[gid][uid] + if ok { + switch husband { + case fiancee: + ctx.SendChain(message.Text("笨蛋~你明明已经娶了啊w")) + default: + ctx.SendChain(message.Text("笨蛋~你家里还有个吃白饭的w")) + } + return false + } + // 如果用户被娶过 + wife, ok := qqwifegroup[gid][-uid] + if ok { + switch wife { + case fiancee: + ctx.SendChain(message.Text("笨蛋~你明明已经嫁给他了啊w")) + default: + ctx.SendChain(message.Text("该是0就是0，当0有什么不好")) + } + return false + } + // 如果未婚妻娶过 + wife, ok = qqwifegroup[gid][fiancee] + if ok { + switch wife { + case uid: + ctx.SendChain(message.Text("笨蛋~你明明已经嫁给他了啊w")) + default: + ctx.SendChain(message.Text("他有别的女人了，你该放下了")) + } + return false + } + // 如果未婚妻被娶过 + husband, ok = qqwifegroup[gid][-fiancee] + if ok { + switch husband { + case uid: + ctx.SendChain(message.Text("笨蛋~你明明已经娶了啊w")) + default: + ctx.SendChain(message.Text("这是一个纯爱的世界，拒绝NTR")) + } + return false + } + return true +}