修复单身狗技能逻辑 (#224)

* ⬆️ update deps

* 更改源酱提出的问题

* Update qqmapwife.go

Co-authored-by: 源文雨 <41315874+fumiama@users.noreply.github.com>

Author: fangliuyu

plugin/qqwife/qqmapwife.go

@@ -1,4 +1,4 @@
-//  Package qqwife 娶群友  基于“翻牌”和江林大佬的“群老婆”插件魔改作品
+// Package qqwife 娶群友  基于“翻牌”和江林大佬的“群老婆”插件魔改作品
 package qqwife
 
 import (
@@ -21,13 +21,25 @@ 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() {
 	engine := control.Register("qqwife", &control.Options{
 		DisableOnDefault: false,
 		Help: "一群一天一夫一妻制群老婆\n" +
 			"- 娶群友\n" +
+			"- 娶[老婆QQ号|@老婆QQ]\n(注:单身专属技能,CD24H,不跨天刷新)\n" +
 			"- 群老婆列表",
 	})
 	engine.OnFullMatch("娶群友", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
@@ -37,16 +49,20 @@ 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]
 			if ok {
 				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
+}