mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-28 04:41:20 +08:00
422 lines
12 KiB
Go
422 lines
12 KiB
Go
// Package qqwife 娶群友 基于“翻牌”和江林大佬的“群老婆”插件魔改作品
|
||
package qqwife
|
||
|
||
import (
|
||
"math/rand"
|
||
"sort"
|
||
"strconv"
|
||
"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"
|
||
"github.com/wdvxdr1123/ZeroBot/extension/rate"
|
||
)
|
||
|
||
//nolint: asciicheck
|
||
type 婚姻登记 struct {
|
||
sync.Mutex
|
||
mp map[int64]map[int64]*userinfo
|
||
}
|
||
|
||
// 结婚证信息
|
||
type userinfo struct {
|
||
target int64 // 对象身份证号
|
||
username string // 户主名称
|
||
targetname string // 对象名称
|
||
}
|
||
|
||
//nolint: asciicheck
|
||
func 新登记处() (db 婚姻登记) {
|
||
db.mp = make(map[int64]map[int64]*userinfo, 64)
|
||
return
|
||
}
|
||
|
||
func (db *婚姻登记) 重置() {
|
||
db.Lock()
|
||
defer db.Unlock()
|
||
for k := range db.mp {
|
||
delete(db.mp, k)
|
||
}
|
||
}
|
||
|
||
func (db *婚姻登记) 离婚休妻(gid, wife int64) {
|
||
db.Lock()
|
||
defer db.Unlock()
|
||
delete(db.mp[gid], -wife)
|
||
}
|
||
|
||
func (db *婚姻登记) 离婚休夫(gid, husband int64) {
|
||
db.Lock()
|
||
defer db.Unlock()
|
||
delete(db.mp[gid], husband)
|
||
}
|
||
|
||
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 *婚姻登记) 花名册(ctx *zero.Ctx, gid int64) 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 uid, userinfo := range mp {
|
||
if uid > 0 {
|
||
_ = w.WriteByte('\n')
|
||
w.WriteString(userinfo.username)
|
||
w.WriteString(" & ")
|
||
w.WriteString(userinfo.targetname)
|
||
}
|
||
}
|
||
}))
|
||
}
|
||
|
||
func (db *婚姻登记) 查户口(gid, uid int64) (userinfo *userinfo, gender int, ok bool) {
|
||
db.Lock()
|
||
defer db.Unlock()
|
||
gender = 0
|
||
mp, ok := db.mp[gid]
|
||
if !ok {
|
||
return
|
||
}
|
||
userinfo, ok = mp[uid]
|
||
if !ok {
|
||
gender = 1
|
||
userinfo, ok = mp[-uid]
|
||
}
|
||
return
|
||
}
|
||
|
||
func (db *婚姻登记) 登记(gid, uid, target int64, username, targetname string) {
|
||
db.Lock()
|
||
defer db.Unlock()
|
||
_, ok := db.mp[gid]
|
||
if !ok {
|
||
db.mp[gid] = make(map[int64]*userinfo, 32)
|
||
}
|
||
// 填写夫妻信息
|
||
uidinfo := &userinfo{
|
||
target: target,
|
||
username: username,
|
||
targetname: targetname,
|
||
}
|
||
targetinfo := &userinfo{
|
||
target: uid,
|
||
username: targetname,
|
||
targetname: username,
|
||
}
|
||
// 民政局登记数据
|
||
db.mp[gid][uid] = uidinfo
|
||
db.mp[gid][-target] = targetinfo
|
||
}
|
||
|
||
var (
|
||
//nolint: asciicheck
|
||
民政局 = 新登记处()
|
||
skillCD = rate.NewManager[string](time.Hour*24, 1)
|
||
lastdate time.Time
|
||
sendtext = [...][]string{
|
||
{ // 表白成功
|
||
"今天你向ta表白了，ta羞涩的点了点头同意了！\n",
|
||
"你对ta说“以我之名，冠你指间，一天相伴，一天相随”.ta捂着嘴点了点头\n\n",
|
||
},
|
||
{ // 表白失败
|
||
"今天你向ta表白了，ta毫无感情的拒绝了你",
|
||
"今天你向ta表白了，ta对你说“你是一个非常好的人”",
|
||
"今天你向ta表白了，ta给了你一个拥抱后擦肩而过",
|
||
},
|
||
{ // ntr成功
|
||
"你处心积虑的接近ta，ta最终选择跟随你\n",
|
||
},
|
||
}
|
||
)
|
||
|
||
func init() {
|
||
engine := control.Register("qqwife", &control.Options{
|
||
DisableOnDefault: false,
|
||
Help: "一群一天一夫一妻制群老婆\n（每天凌晨刷新CP）\n" +
|
||
"- 娶群友\n- 群老婆列表\n" +
|
||
"--------------------------------\n以下技能每人只能二选一\n CD24H，不跨天刷新\n--------------------------------\n" +
|
||
"- (娶|嫁)@对方QQ\n- 当[对方Q号|@对方QQ]的小三\n",
|
||
})
|
||
engine.OnFullMatch("娶群友", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByGroup).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
if time.Now().Day() != lastdate.Day() {
|
||
民政局.重置()
|
||
// 更新时间
|
||
lastdate = time.Now()
|
||
}
|
||
gid := ctx.Event.GroupID
|
||
uid := ctx.Event.UserID
|
||
targetinfo, status, ok := 民政局.查户口(gid, uid)
|
||
if ok {
|
||
switch status {
|
||
case 0: // 娶过别人
|
||
ctx.SendChain(
|
||
message.At(uid),
|
||
message.Text("今天你的群老婆是"),
|
||
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(targetinfo.target, 10)+"&s=640").Add("cache", 0),
|
||
message.Text(
|
||
"\n",
|
||
"[", targetinfo.targetname, "]",
|
||
"(", targetinfo.target, ")哒",
|
||
),
|
||
)
|
||
default: // 嫁给别人
|
||
ctx.SendChain(
|
||
message.At(uid),
|
||
message.Text("今天你的群老公是"),
|
||
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(targetinfo.target, 10)+"&s=640").Add("cache", 0),
|
||
message.Text(
|
||
"\n",
|
||
"[", targetinfo.targetname, "]",
|
||
"(", targetinfo.target, ")哒",
|
||
),
|
||
)
|
||
}
|
||
return
|
||
}
|
||
// 无缓存获取群员列表
|
||
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()
|
||
})
|
||
temp = temp[math.Max(0, len(temp)-30):]
|
||
// 将已经娶过的人剔除
|
||
qqgrouplist := make([]int64, 0, len(temp))
|
||
for k := 0; k < len(temp); k++ {
|
||
usr := temp[k].Get("user_id").Int()
|
||
_, _, ok := 民政局.查户口(gid, usr)
|
||
if ok {
|
||
continue
|
||
}
|
||
qqgrouplist = append(qqgrouplist, usr)
|
||
}
|
||
// 没有人（只剩自己）的时候
|
||
if len(qqgrouplist) == 0 {
|
||
ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧"))
|
||
return
|
||
}
|
||
// 随机抽娶
|
||
fiancee := qqgrouplist[rand.Intn(len(qqgrouplist))]
|
||
if fiancee == uid { // 如果是自己
|
||
ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧"))
|
||
return
|
||
}
|
||
// 去民政局办证
|
||
民政局.登记(gid, uid, fiancee, ctx.CardOrNickName(uid), ctx.CardOrNickName(fiancee))
|
||
// 请大家吃席
|
||
ctx.SendChain(
|
||
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, ")哒",
|
||
),
|
||
)
|
||
})
|
||
// 单身技能
|
||
engine.OnRegex(`^(娶|嫁)\[CQ:at,qq=(\d+)\]`, zero.OnlyGroup, checkdog).SetBlock(true).Limit(cdcheck, iscding).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
choice := ctx.State["regex_matched"].([]string)[1]
|
||
fiancee, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
|
||
uid := ctx.Event.UserID
|
||
if uid == fiancee { // 如果是自己
|
||
ctx.SendChain(message.Text("今日获得成就：自恋狂"))
|
||
return
|
||
}
|
||
if rand.Intn(2) == 1 { // 二分之一的概率表白成功
|
||
gid := ctx.Event.GroupID
|
||
// 去民政局登记
|
||
var choicetext string
|
||
switch choice {
|
||
case "娶":
|
||
民政局.登记(gid, uid, fiancee, ctx.CardOrNickName(uid), ctx.CardOrNickName(fiancee))
|
||
choicetext = "今天你的群老婆是"
|
||
default:
|
||
民政局.登记(gid, fiancee, uid, ctx.CardOrNickName(fiancee), ctx.CardOrNickName(uid))
|
||
choicetext = "今天你的群老公是"
|
||
}
|
||
// 请大家吃席
|
||
ctx.SendChain(
|
||
message.Text(sendtext[0][rand.Intn(len(sendtext[0]))]),
|
||
message.At(uid),
|
||
message.Text(choicetext),
|
||
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, ")哒",
|
||
),
|
||
)
|
||
return
|
||
}
|
||
ctx.SendChain(message.Text(sendtext[1][rand.Intn(len(sendtext[1]))]))
|
||
})
|
||
// NTR技能
|
||
engine.OnRegex(`^当(\[CQ:at,qq=(\d+)\] |(\d+))的小三`, zero.OnlyGroup, checkcp).SetBlock(true).Limit(cdcheck, iscding).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
fid := ctx.State["regex_matched"].([]string)
|
||
fiancee, _ := strconv.ParseInt(fid[2]+fid[3], 10, 64)
|
||
if rand.Intn(10)/4 != 0 { // 十分之三的概率NTR成功
|
||
ctx.SendChain(message.Text("你的ntr计划失败了"))
|
||
return
|
||
}
|
||
gid := ctx.Event.GroupID
|
||
uid := ctx.Event.UserID
|
||
// 判断target是老公还是老婆
|
||
var choicetext string
|
||
targetinfo, gender, _ := 民政局.查户口(gid, fiancee)
|
||
switch gender {
|
||
case 0:
|
||
// 让对象离婚
|
||
民政局.离婚休妻(gid, targetinfo.target)
|
||
// 和对象结婚登记
|
||
choicetext = "老公"
|
||
民政局.登记(gid, fiancee, uid, ctx.CardOrNickName(fiancee), ctx.CardOrNickName(uid))
|
||
default:
|
||
// 让对象离婚
|
||
民政局.离婚休夫(gid, targetinfo.target)
|
||
// 和对象结婚登记
|
||
choicetext = "老婆"
|
||
民政局.登记(gid, uid, fiancee, ctx.CardOrNickName(uid), ctx.CardOrNickName(fiancee))
|
||
}
|
||
// 输出结果
|
||
ctx.SendChain(
|
||
message.Text(sendtext[2][rand.Intn(len(sendtext[2]))]),
|
||
message.At(uid),
|
||
message.Text("今天你的群"+choicetext+"是"),
|
||
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, ")哒",
|
||
),
|
||
)
|
||
})
|
||
engine.OnFullMatch("群老婆列表", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByGroup).
|
||
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
|
||
uidtarget, uidstatus, ok1 := 民政局.查户口(gid, uid)
|
||
_, fianceestatus, ok2 := 民政局.查户口(gid, fiancee)
|
||
if !ok1 && !ok2 { // 必须是两个单身
|
||
return true
|
||
}
|
||
if ok1 {
|
||
if uidtarget.target == fiancee { // 如果本就是一块
|
||
ctx.SendChain(message.Text("笨蛋~你们明明已经在一起了啊w"))
|
||
return false
|
||
}
|
||
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
|
||
if fiancee == uid {
|
||
ctx.SendChain(message.Text("自我攻略?"))
|
||
return false
|
||
}
|
||
// 检查用户是否登记过
|
||
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
|
||
}
|