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

修复CD共用问题，新增功能 (#229)

Browse Source
This commit is contained in:
方柳煜 2022-05-16 13:39:48 +08:00 committed by GitHub
parent fbb387bd9f
commit ed2ed8d968
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 428 additions and 326 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

376
plugin/qqwife/qqmapwife.go
View File

@ -15,20 +15,29 @@ import (
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"
) )
//nolint: asciicheck //nolint: asciicheck
type 婚姻登记 struct { type 婚姻登记 struct {
sync.Mutex sync.Mutex
mp map[int64]map[int64]int64 mp map[int64]map[int64]*userinfo
}
// 结婚证信息
type userinfo struct {
target int64 //对象身份证号
username string //户主名称
targetname string //对象名称
} }
//nolint: asciicheck //nolint: asciicheck
func 新登记处() (db 婚姻登记) { func 新登记处() (db 婚姻登记) {
db.mp = make(map[int64]map[int64]int64, 64) db.mp = make(map[int64]map[int64]*userinfo, 64)
return return
} }
//nolint: asciicheck
func (db *婚姻登记) 重置() { func (db *婚姻登记) 重置() {
db.Lock() db.Lock()
defer db.Unlock() defer db.Unlock()
@ -37,7 +46,22 @@ func (db *婚姻登记) 重置() {
} }
} }
func (db *婚姻登记) 有登记在(gid int64) (ok bool) { //nolint: asciicheck
func (db *婚姻登记) 离婚休妻(gid, wife int64) {
db.Lock()
defer db.Unlock()
delete(db.mp[gid], -wife)
}
//nolint: asciicheck
func (db *婚姻登记) 离婚休夫(gid, husband int64) {
db.Lock()
defer db.Unlock()
delete(db.mp[gid], husband)
}
//nolint: asciicheck
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]
@ -50,94 +74,85 @@ func (db *婚姻登记) 有登记在(gid int64) (ok bool) {
return return
} }
func (db *婚姻登记) 登记情况(gid int64, ctx *zero.Ctx) string { //nolint: asciicheck
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 husband, wife := range mp { for uid, userinfo := range mp {
if husband > 0 { if uid > 0 {
_ = w.WriteByte('\n') _ = w.WriteByte('\n')
w.WriteString(ctx.CardOrNickName(husband)) w.WriteString(userinfo.username)
w.WriteString(" & ") w.WriteString(" & ")
w.WriteString(ctx.CardOrNickName(wife)) w.WriteString(userinfo.targetname)
} }
} }
})) }))
} }
func (db *婚姻登记) 有妻子(gid, uid int64) (ok bool) { //nolint: asciicheck
func (db *婚姻登记) 查户口(gid, uid int64) (userinfo *userinfo, gender int, ok bool) {
db.Lock() db.Lock()
defer db.Unlock() defer db.Unlock()
gender = 0
mp, ok := db.mp[gid] mp, ok := db.mp[gid]
if !ok { if !ok {
return return
} }
_, ok = mp[uid] userinfo, ok = mp[uid]
if !ok {
gender = 1
userinfo, ok = mp[-uid]
}
return return
} }
func (db *婚姻登记) 查询妻子(gid, uid int64) (wife int64) { //nolint: asciicheck
func (db *婚姻登记) 登记(gid, uid, target int64, username, targetname string) {
db.Lock() db.Lock()
defer db.Unlock() defer db.Unlock()
mp, ok := db.mp[gid] _,ok := db.mp[gid]
if !ok{ if !ok{
return db.mp[gid] = make(map[int64]*userinfo, 32)
} }
return mp[uid] // 填写夫妻信息
uidinfo := &userinfo{
target: target,
username: username,
targetname: targetname,
} }
targetinfo := &userinfo{
func (db *婚姻登记) 有丈夫(gid, uid int64) (ok bool) { target: uid,
db.Lock() username: targetname,
defer db.Unlock() targetname: username,
mp, ok := db.mp[gid]
if !ok {
return
} }
_, ok = mp[-uid] // 民政局登记数据
return db.mp[gid][uid] = uidinfo
} db.mp[gid][-target] = targetinfo
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 ( var (
//nolint: asciicheck //nolint: asciicheck
民政局 = 新登记处() 民政局 = 新登记处()
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毫无感情的拒绝了你\n", "今天你向ta表白了，ta毫无感情的拒绝了你",
"今天你向ta表白了, ta对你说“你是一个非常好的人”\n", "今天你向ta表白了，ta对你说“你是一个非常好的人”",
"今天你向ta表白了, ta给了你一个拥抱后擦肩而过\n", "今天你向ta表白了，ta给了你一个拥抱后擦肩而过",
},
{ //ntr成功
"你处心积虑的接近ta，ta最终选择跟随你\n",
}, },
} }
) )
@ -145,10 +160,10 @@ var (
func init() { func init() {
engine := control.Register("qqwife", &control.Options{ engine := control.Register("qqwife", &control.Options{
DisableOnDefault: false, DisableOnDefault: false,
Help: "一群一天一夫一妻制群老婆\n" + Help: "一群一天一夫一妻制群老婆\n（每天凌晨刷新CP）\n" +
"- 娶群友\n" + "- 娶群友\n- 群老婆列表\n" +
"- 娶[老婆QQ号|@老婆QQ]\n(注:单身专属技能,CD24H,不跨天刷新)\n" + "--------------------------------\n以下技能每人只能二选一\n CD24H，不跨天刷新\n--------------------------------\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) {
@ -157,37 +172,34 @@ func init() {
// 更新时间 // 更新时间
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)
wife := 民政局.查询妻子(gid, uid) if ok {
if wife > 0 { switch status {
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(wife, 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",
"[", ctx.CardOrNickName(wife), "]", "[", targetinfo.targetname, "]",
"(", wife, ")哒", "(", targetinfo.target, ")哒",
), ),
) )
return default: // 嫁给别人
}
// 如果被娶过
husband := 民政局.查询丈夫(gid, uid)
if husband > 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(husband, 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",
"[", ctx.CardOrNickName(husband), "]", "[", targetinfo.targetname, "]",
"(", husband, ")哒", "(", targetinfo.target, ")哒",
), ),
) )
}
return return
} }
// 无缓存获取群员列表 // 无缓存获取群员列表
@ -200,7 +212,8 @@ func init() {
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()
if 民政局.有妻子(gid, usr) || 民政局.有丈夫(gid, usr) { _, _, ok := 民政局.查户口(gid, usr)
if ok {
continue continue
} }
qqgrouplist = append(qqgrouplist, usr) qqgrouplist = append(qqgrouplist, usr)
@ -211,42 +224,52 @@ func init() {
return return
} }
// 随机抽娶 // 随机抽娶
wife = qqgrouplist[rand.Intn(len(qqgrouplist))] fiancee := qqgrouplist[rand.Intn(len(qqgrouplist))]
if wife == uid { // 如果是自己 if fiancee == uid { // 如果是自己
ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧")) ctx.SendChain(message.Text("噢, 此时此刻你还是一只单身狗, 等待下一次情缘吧"))
return return
} }
// 绑定CP // 去民政局办证
民政局.登记(gid, wife, uid) 民政局.登记(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(wife, 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(wife), "]", "[", ctx.CardOrNickName(fiancee), "]",
"(", wife, ")哒", "(", fiancee, ")哒",
), ),
) )
}) })
// 单身狗专属技能 // 单身技能
engine.OnRegex(`^娶(\d+|\[CQ:at,qq=(\d+)\])`, zero.OnlyGroup, checkdog).SetBlock(true). engine.OnRegex(`^(娶|嫁)\[CQ:at,qq=(\d+)\]`, zero.OnlyGroup, checkdog).SetBlock(true).Limit(cdcheck, iscding).
Limit(ctxext.NewLimiterManager(time.Hour*12, 1).LimitByUser).
Handle(func(ctx *zero.Ctx) { Handle(func(ctx *zero.Ctx) {
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64) choice := ctx.State["regex_matched"].([]string)[1]
if err != nil { fiancee, _ := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
fiancee, _ = strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
}
gid := ctx.Event.GroupID
uid := ctx.Event.UserID uid := ctx.Event.UserID
if rand.Intn(2) == 1 { if uid == fiancee { // 如果是自己
民政局.登记(gid, fiancee, uid) 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( 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("今天你的群老婆是"), 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",
@ -258,69 +281,148 @@ func init() {
} }
ctx.SendChain(message.Text(sendtext[1][rand.Intn(len(sendtext[1]))])) 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.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.Event.GroupID, ctx))) ctx.SendChain(message.Text(民政局.花名册(ctx, ctx.Event.GroupID)))
}) })
} }
// 注入判断 是否为单身狗 // 以群号和昵称为限制
func checkdog(ctx *zero.Ctx) bool { func cdcheck(ctx *zero.Ctx) *rate.Limiter {
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64) limitID := strconv.FormatInt(ctx.Event.GroupID, 10) + strconv.FormatInt(ctx.Event.UserID, 10)
// fmt.Println("1:", fiancee) return skillCD.Load(limitID)
if err != nil {
fiancee, _ = strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
// fmt.Println("2:", fiancee)
} }
func iscding(ctx *zero.Ctx) {
ctx.SendChain(message.Text("你的技能现在正在CD中"))
}
// 注入判断 是否为单身
func checkdog(ctx *zero.Ctx) bool {
gid := ctx.Event.GroupID 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 uid := ctx.Event.UserID
if uid == fiancee { if uid == fiancee {
ctx.SendChain(message.Text("今日获得成就：自恋狂")) ctx.SendChain(message.Text("今日获得成就：自恋狂"))
return false return false
} }
// 如果用户娶过 // 获取用户信息
if o := 民政局.查询妻子(gid, uid); o > 0 { uidtarget, uidstatus, ok1 := 民政局.查户口(gid, uid)
switch o { _, fianceestatus, ok2 := 民政局.查户口(gid, fiancee)
case fiancee: if !ok1 && !ok2 { // 必须是两个单身
ctx.SendChain(message.Text("笨蛋~你明明已经娶了啊w")) return true
default: }
if uidtarget.target == fiancee { // 如果本就是一块
ctx.SendChain(message.Text("笨蛋~你们明明已经在一起了啊w"))
return false
}
if ok1 {
switch uidstatus {
case 0: // 如果如为攻
ctx.SendChain(message.Text("笨蛋~你家里还有个吃白饭的w")) ctx.SendChain(message.Text("笨蛋~你家里还有个吃白饭的w"))
default: // 如果为受
ctx.SendChain(message.Text("该是0就是0，当0有什么不好"))
} }
return false return false
} }
// 如果用户被娶过 if ok2 {
if o := 民政局.查询丈夫(gid, uid); o > 0 { switch fianceestatus {
switch o { case 0: // 如果如为攻
case fiancee: ctx.SendChain(message.Text("他有别的女人了，你该放下了"))
ctx.SendChain(message.Text("笨蛋~你明明已经嫁给他了啊w")) default: // 如果为受
default: ctx.SendChain(message.Text("这是一个纯爱的世界，拒绝NTR"))
ctx.SendChain(message.Text("该是0就是0, 当0有什么不好"))
}
return false
}
// 如果未婚妻娶过
if o := 民政局.查询妻子(gid, fiancee); o > 0 {
switch o {
case uid:
ctx.SendChain(message.Text("笨蛋~你明明已经嫁给他了啊w"))
default:
ctx.SendChain(message.Text("他有别的女人了, 你该放下了"))
}
return false
}
// 如果未婚妻被娶过
if o := 民政局.查询丈夫(gid, fiancee); o > 0 {
switch o {
case uid:
ctx.SendChain(message.Text("笨蛋~你明明已经娶了啊w"))
default:
ctx.SendChain(message.Text("这是一个纯爱的世界, 拒绝NTR"))
} }
return false return false
} }
return true 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
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
}