Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2025-12-23 01:36:36 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

🎨 改进代码样式

Browse Source
This commit is contained in:
github-actions[bot] 2022-05-16 05:40:05 +00:00
parent ed2ed8d968
commit 4b4cafcda8
1 changed files with 428 additions and 428 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

856
plugin/qqwife/qqmapwife.go
View File

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