mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
🎨 改进代码样式
This commit is contained in:
github-actions[bot]
parent
92c9d1d2ce
commit
dec42cc72b
@ -1,146 +1,146 @@
|
||||
// Package qqwife 娶群友 基于“翻牌”和江林大佬的“群老婆”插件魔改作品
|
||||
package qqwife
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
zero "github.com/wdvxdr1123/ZeroBot"
|
||||
"github.com/wdvxdr1123/ZeroBot/message"
|
||||
|
||||
control "github.com/FloatTech/zbputils/control"
|
||||
"github.com/FloatTech/zbputils/ctxext"
|
||||
"github.com/FloatTech/zbputils/math"
|
||||
)
|
||||
|
||||
var (
|
||||
setcp = make(map[int64]int64, 60) //初始化中间的map元素，虽然取了30个人数，以防万一翻个倍储存
|
||||
qqwifegroup = make(map[int64]map[int64]int64, 50) //50个群的预算大小
|
||||
lastdate time.Time
|
||||
rock sync.Mutex
|
||||
)
|
||||
|
||||
func init() {
|
||||
engine := control.Register("qqwife", &control.Options{
|
||||
DisableOnDefault: false,
|
||||
Help: "一群一天一夫一妻制群老婆\n" +
|
||||
"-娶群友",
|
||||
})
|
||||
engine.OnFullMatch("娶群友", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
rock.Lock()
|
||||
defer rock.Unlock()
|
||||
if time.Now().Day() != lastdate.Day() {
|
||||
qqwifegroup = make(map[int64]map[int64]int64, 50) //跨天就重新初始化数据
|
||||
}
|
||||
//先判断是否已经娶过或者被娶
|
||||
groupid := ctx.Event.GroupID
|
||||
uid := ctx.Event.UserID
|
||||
var status string
|
||||
//如果娶过
|
||||
userinfo, ok := qqwifegroup[groupid][uid]
|
||||
if ok {
|
||||
ctx.SendChain(
|
||||
message.At(uid),
|
||||
message.Text("今天你的群老婆是"),
|
||||
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(userinfo, 10)+"&s=640"),
|
||||
message.Text(
|
||||
"\n",
|
||||
"[", ctx.CardOrNickName(userinfo), "]",
|
||||
"(", strconv.FormatInt(userinfo, 10), ")哒",
|
||||
),
|
||||
)
|
||||
return
|
||||
}
|
||||
//如果被娶过
|
||||
for k, v := range qqwifegroup[groupid] {
|
||||
if v == uid { //如果为0且是在本群抽的就输出
|
||||
ctx.SendChain(
|
||||
message.At(uid),
|
||||
message.Text("今天你被娶了，群老公是"),
|
||||
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(k, 10)+"&s=640"),
|
||||
message.Text(
|
||||
"\n",
|
||||
"[", ctx.CardOrNickName(k), "]",
|
||||
"(", strconv.FormatInt(k, 10), ")哒",
|
||||
),
|
||||
)
|
||||
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):]
|
||||
//将已经娶过的人剔除
|
||||
var qqgrouplist = []int64{}
|
||||
if len(qqwifegroup) == 0 {
|
||||
for k := 0; k < len(temp); k++ {
|
||||
qqgrouplist = append(qqgrouplist, temp[k].Get("user_id").Int())
|
||||
}
|
||||
} else {
|
||||
for k := 0; k < len(temp); k++ {
|
||||
_, ok := qqwifegroup[groupid][temp[k].Get("user_id").Int()]
|
||||
if !ok {
|
||||
qqgrouplist = append(qqgrouplist, temp[k].Get("user_id").Int())
|
||||
}
|
||||
}
|
||||
}
|
||||
//没有人（只剩自己）的时候
|
||||
if len(qqgrouplist) == 0 {
|
||||
ctx.SendChain(message.Text("噢，此时此刻你还是一只单身狗，等待下一次情缘吧"))
|
||||
return
|
||||
}
|
||||
//随机抽娶
|
||||
who := qqgrouplist[rand.Intn(len(qqgrouplist))]
|
||||
if who == uid { //如果是自己
|
||||
ctx.SendChain(message.Text("噢，此时此刻你还是一只单身狗，等待下一次情缘吧"))
|
||||
return
|
||||
}
|
||||
//绑定CP
|
||||
setcp[uid] = who
|
||||
qqwifegroup[groupid] = setcp
|
||||
//输出结果
|
||||
ctx.SendChain(
|
||||
message.Text(status),
|
||||
message.At(uid),
|
||||
message.Text("今天你的群老婆是"),
|
||||
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(who, 10)+"&s=640"),
|
||||
message.Text(
|
||||
"\n",
|
||||
"[", ctx.CardOrNickName(who), "]",
|
||||
"(", strconv.FormatInt(who, 10), ")哒",
|
||||
),
|
||||
)
|
||||
//更新时间
|
||||
lastdate = time.Now()
|
||||
})
|
||||
engine.OnFullMatch("群老婆列表", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
var cplist = []string{"群老公←———→群老婆\n--------------------------"}
|
||||
rock.Lock()
|
||||
defer rock.Unlock()
|
||||
group, ok := qqwifegroup[ctx.Event.GroupID]
|
||||
if !ok {
|
||||
ctx.SendChain(message.Text("你群并没有任何的CP额"))
|
||||
return
|
||||
}
|
||||
for k, v := range group {
|
||||
if k != 0 {
|
||||
husband := ctx.CardOrNickName(k)
|
||||
wife := ctx.CardOrNickName(v)
|
||||
cplist = append(cplist, husband+" & "+wife)
|
||||
} else {
|
||||
ctx.SendChain(message.Text("你群并没有任何的CP额"))
|
||||
return
|
||||
}
|
||||
}
|
||||
ctx.SendChain(message.Text(strings.Join(cplist, "\n")))
|
||||
})
|
||||
}
|
||||
// Package qqwife 娶群友 基于“翻牌”和江林大佬的“群老婆”插件魔改作品
|
||||
package qqwife
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
zero "github.com/wdvxdr1123/ZeroBot"
|
||||
"github.com/wdvxdr1123/ZeroBot/message"
|
||||
|
||||
control "github.com/FloatTech/zbputils/control"
|
||||
"github.com/FloatTech/zbputils/ctxext"
|
||||
"github.com/FloatTech/zbputils/math"
|
||||
)
|
||||
|
||||
var (
|
||||
setcp = make(map[int64]int64, 60) //初始化中间的map元素，虽然取了30个人数，以防万一翻个倍储存
|
||||
qqwifegroup = make(map[int64]map[int64]int64, 50) //50个群的预算大小
|
||||
lastdate time.Time
|
||||
rock sync.Mutex
|
||||
)
|
||||
|
||||
func init() {
|
||||
engine := control.Register("qqwife", &control.Options{
|
||||
DisableOnDefault: false,
|
||||
Help: "一群一天一夫一妻制群老婆\n" +
|
||||
"-娶群友",
|
||||
})
|
||||
engine.OnFullMatch("娶群友", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
rock.Lock()
|
||||
defer rock.Unlock()
|
||||
if time.Now().Day() != lastdate.Day() {
|
||||
qqwifegroup = make(map[int64]map[int64]int64, 50) //跨天就重新初始化数据
|
||||
}
|
||||
//先判断是否已经娶过或者被娶
|
||||
groupid := ctx.Event.GroupID
|
||||
uid := ctx.Event.UserID
|
||||
var status string
|
||||
//如果娶过
|
||||
userinfo, ok := qqwifegroup[groupid][uid]
|
||||
if ok {
|
||||
ctx.SendChain(
|
||||
message.At(uid),
|
||||
message.Text("今天你的群老婆是"),
|
||||
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(userinfo, 10)+"&s=640"),
|
||||
message.Text(
|
||||
"\n",
|
||||
"[", ctx.CardOrNickName(userinfo), "]",
|
||||
"(", strconv.FormatInt(userinfo, 10), ")哒",
|
||||
),
|
||||
)
|
||||
return
|
||||
}
|
||||
//如果被娶过
|
||||
for k, v := range qqwifegroup[groupid] {
|
||||
if v == uid { //如果为0且是在本群抽的就输出
|
||||
ctx.SendChain(
|
||||
message.At(uid),
|
||||
message.Text("今天你被娶了，群老公是"),
|
||||
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(k, 10)+"&s=640"),
|
||||
message.Text(
|
||||
"\n",
|
||||
"[", ctx.CardOrNickName(k), "]",
|
||||
"(", strconv.FormatInt(k, 10), ")哒",
|
||||
),
|
||||
)
|
||||
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):]
|
||||
//将已经娶过的人剔除
|
||||
var qqgrouplist = []int64{}
|
||||
if len(qqwifegroup) == 0 {
|
||||
for k := 0; k < len(temp); k++ {
|
||||
qqgrouplist = append(qqgrouplist, temp[k].Get("user_id").Int())
|
||||
}
|
||||
} else {
|
||||
for k := 0; k < len(temp); k++ {
|
||||
_, ok := qqwifegroup[groupid][temp[k].Get("user_id").Int()]
|
||||
if !ok {
|
||||
qqgrouplist = append(qqgrouplist, temp[k].Get("user_id").Int())
|
||||
}
|
||||
}
|
||||
}
|
||||
//没有人（只剩自己）的时候
|
||||
if len(qqgrouplist) == 0 {
|
||||
ctx.SendChain(message.Text("噢，此时此刻你还是一只单身狗，等待下一次情缘吧"))
|
||||
return
|
||||
}
|
||||
//随机抽娶
|
||||
who := qqgrouplist[rand.Intn(len(qqgrouplist))]
|
||||
if who == uid { //如果是自己
|
||||
ctx.SendChain(message.Text("噢，此时此刻你还是一只单身狗，等待下一次情缘吧"))
|
||||
return
|
||||
}
|
||||
//绑定CP
|
||||
setcp[uid] = who
|
||||
qqwifegroup[groupid] = setcp
|
||||
//输出结果
|
||||
ctx.SendChain(
|
||||
message.Text(status),
|
||||
message.At(uid),
|
||||
message.Text("今天你的群老婆是"),
|
||||
message.Image("http://q4.qlogo.cn/g?b=qq&nk="+strconv.FormatInt(who, 10)+"&s=640"),
|
||||
message.Text(
|
||||
"\n",
|
||||
"[", ctx.CardOrNickName(who), "]",
|
||||
"(", strconv.FormatInt(who, 10), ")哒",
|
||||
),
|
||||
)
|
||||
//更新时间
|
||||
lastdate = time.Now()
|
||||
})
|
||||
engine.OnFullMatch("群老婆列表", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
|
||||
Handle(func(ctx *zero.Ctx) {
|
||||
var cplist = []string{"群老公←———→群老婆\n--------------------------"}
|
||||
rock.Lock()
|
||||
defer rock.Unlock()
|
||||
group, ok := qqwifegroup[ctx.Event.GroupID]
|
||||
if !ok {
|
||||
ctx.SendChain(message.Text("你群并没有任何的CP额"))
|
||||
return
|
||||
}
|
||||
for k, v := range group {
|
||||
if k != 0 {
|
||||
husband := ctx.CardOrNickName(k)
|
||||
wife := ctx.CardOrNickName(v)
|
||||
cplist = append(cplist, husband+" & "+wife)
|
||||
} else {
|
||||
ctx.SendChain(message.Text("你群并没有任何的CP额"))
|
||||
return
|
||||
}
|
||||
}
|
||||
ctx.SendChain(message.Text(strings.Join(cplist, "\n")))
|
||||
})
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user