mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
352 lines
9.7 KiB
Go
352 lines
9.7 KiB
Go
package qqwife
|
||
|
||
import (
|
||
"strconv"
|
||
"sync"
|
||
"time"
|
||
|
||
sql "github.com/FloatTech/sqlite"
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
"github.com/wdvxdr1123/ZeroBot/extension/rate"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
|
||
// 画图
|
||
"github.com/Coloured-glaze/gg"
|
||
)
|
||
|
||
//nolint: asciicheck
|
||
// nolint: asciicheck
|
||
type 婚姻登记 struct {
|
||
db *sql.Sqlite
|
||
dbmu sync.RWMutex
|
||
}
|
||
|
||
// 结婚证信息
|
||
type userinfo struct {
|
||
User int64 // 用户身份证
|
||
Target int64 // 对象身份证号
|
||
Username string // 户主名称
|
||
Targetname string // 对象名称
|
||
Updatetime string // 登记时间
|
||
|
||
}
|
||
|
||
// 民政局的当前时间
|
||
type updateinfo struct {
|
||
GID int64
|
||
Updatetime string // 登记时间
|
||
|
||
}
|
||
|
||
func (sql *婚姻登记) 开门时间(gid int64) (ok bool, err error) {
|
||
sql.dbmu.Lock()
|
||
defer sql.dbmu.Unlock()
|
||
ok = false
|
||
err = sql.db.Create("updateinfo", &updateinfo{})
|
||
if err != nil {
|
||
return
|
||
}
|
||
gidstr := strconv.FormatInt(gid, 10)
|
||
dbinfo := updateinfo{}
|
||
// 获取表格更新的时间
|
||
err = sql.db.Find("updateinfo", &dbinfo, "where gid is "+gidstr)
|
||
if err != nil {
|
||
// 如果没有登记过就记录
|
||
err = sql.db.Insert("updateinfo", &updateinfo{
|
||
GID: gid,
|
||
Updatetime: time.Now().Format("2006/01/02"),
|
||
})
|
||
if err == nil {
|
||
ok = true
|
||
}
|
||
return
|
||
}
|
||
// 开门了就拿新的花名册
|
||
if time.Now().Format("2006/01/02") == dbinfo.Updatetime {
|
||
return
|
||
}
|
||
err = sql.db.Drop(gidstr)
|
||
if err != nil {
|
||
return
|
||
}
|
||
updateinfo := updateinfo{
|
||
GID: gid,
|
||
Updatetime: time.Now().Format("2006/01/02"),
|
||
}
|
||
err = sql.db.Insert("updateinfo", &updateinfo)
|
||
if err == nil {
|
||
ok = true
|
||
}
|
||
return
|
||
}
|
||
|
||
func (sql *婚姻登记) 清理花名册(gid string) error {
|
||
sql.dbmu.Lock()
|
||
defer sql.dbmu.Unlock()
|
||
grouplist, err := sql.db.ListTables()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if gid != "0" {
|
||
grouplist = []string{gid}
|
||
}
|
||
for _, gid := range grouplist {
|
||
err = sql.db.Drop(gid)
|
||
if err != nil {
|
||
continue
|
||
}
|
||
gidint, _ := strconv.ParseInt(gid, 10, 64)
|
||
updateinfo := updateinfo{
|
||
GID: gidint,
|
||
Updatetime: time.Now().Format("2006/01/02"),
|
||
}
|
||
err = sql.db.Insert("updateinfo", &updateinfo)
|
||
}
|
||
return err
|
||
}
|
||
|
||
func (sql *婚姻登记) 查户口(gid, uid int64) (info userinfo, status string, err error) {
|
||
sql.dbmu.Lock()
|
||
defer sql.dbmu.Unlock()
|
||
gidstr := strconv.FormatInt(gid, 10)
|
||
uidstr := strconv.FormatInt(uid, 10)
|
||
status = "单"
|
||
err = sql.db.Create(gidstr, &userinfo{})
|
||
if err != nil {
|
||
status = "错"
|
||
return
|
||
}
|
||
err = sql.db.Find(gidstr, &info, "where user = "+uidstr)
|
||
if err == nil {
|
||
status = "攻"
|
||
return
|
||
}
|
||
err = sql.db.Find(gidstr, &info, "where target = "+uidstr)
|
||
if err == nil {
|
||
status = "受"
|
||
}
|
||
return
|
||
}
|
||
|
||
func (sql *婚姻登记) 登记(gid, uid, target int64, username, targetname string) error {
|
||
sql.dbmu.Lock()
|
||
defer sql.dbmu.Unlock()
|
||
gidstr := strconv.FormatInt(gid, 10)
|
||
err := sql.db.Create(gidstr, &userinfo{})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
updatetime := time.Now().Format("2006/01/02")
|
||
// 填写夫妻信息
|
||
uidinfo := userinfo{
|
||
User: uid,
|
||
Username: username,
|
||
Target: target,
|
||
Targetname: targetname,
|
||
Updatetime: updatetime,
|
||
}
|
||
// 民政局登记数据
|
||
err = sql.db.Insert(gidstr, &uidinfo)
|
||
return err
|
||
}
|
||
|
||
func (sql *婚姻登记) 离婚休妻(gid, wife int64) error {
|
||
sql.dbmu.Lock()
|
||
defer sql.dbmu.Unlock()
|
||
gidstr := strconv.FormatInt(gid, 10)
|
||
wifestr := strconv.FormatInt(wife, 10)
|
||
return sql.db.Del(gidstr, "where target = "+wifestr)
|
||
}
|
||
|
||
func (sql *婚姻登记) 离婚休夫(gid, husband int64) error {
|
||
sql.dbmu.Lock()
|
||
defer sql.dbmu.Unlock()
|
||
gidstr := strconv.FormatInt(gid, 10)
|
||
husbandstr := strconv.FormatInt(husband, 10)
|
||
return sql.db.Del(gidstr, "where user = "+husbandstr)
|
||
}
|
||
|
||
func (sql *婚姻登记) 花名册(gid int64) (list [][4]string, number int, err error) {
|
||
sql.dbmu.Lock()
|
||
defer sql.dbmu.Unlock()
|
||
gidstr := strconv.FormatInt(gid, 10)
|
||
err = sql.db.Create(gidstr, &userinfo{})
|
||
if err != nil {
|
||
return
|
||
}
|
||
number, err = sql.db.Count(gidstr)
|
||
if err != nil || number <= 0 {
|
||
return
|
||
}
|
||
var info userinfo
|
||
list = make([][4]string, 0, number)
|
||
err = sql.db.FindFor(gidstr, &info, "GROUP BY user", func() error {
|
||
if info.Target == 0 {
|
||
return nil
|
||
}
|
||
dbinfo := [4]string{
|
||
info.Username,
|
||
strconv.FormatInt(info.User, 10),
|
||
info.Targetname,
|
||
strconv.FormatInt(info.Target, 10),
|
||
}
|
||
list = append(list, dbinfo)
|
||
return nil
|
||
})
|
||
if len(list) == 0 {
|
||
number = 0
|
||
}
|
||
return
|
||
}
|
||
|
||
func slicename(name string, canvas *gg.Context) (resultname string) {
|
||
usermane := []rune(name) // 将每个字符单独放置
|
||
widthlen := 0
|
||
numberlen := 0
|
||
for i, v := range usermane {
|
||
width, _ := canvas.MeasureString(string(v)) // 获取单个字符的宽度
|
||
widthlen += int(width)
|
||
if widthlen > 350 {
|
||
break // 总宽度不能超过350
|
||
}
|
||
numberlen = i
|
||
}
|
||
if widthlen > 350 {
|
||
resultname = string(usermane[:numberlen-1]) + "......" // 名字切片
|
||
} else {
|
||
resultname = name
|
||
}
|
||
return
|
||
}
|
||
|
||
// 以群号和昵称为限制
|
||
func cdcheck(ctx *zero.Ctx) *rate.Limiter {
|
||
limitID := strconv.FormatInt(ctx.Event.GroupID, 10) + strconv.FormatInt(ctx.Event.UserID, 10) + "1"
|
||
return skillCD.Load(limitID)
|
||
}
|
||
func cdcheck2(ctx *zero.Ctx) *rate.Limiter {
|
||
limitID := strconv.FormatInt(ctx.Event.GroupID, 10) + strconv.FormatInt(ctx.Event.UserID, 10) + "2"
|
||
return skillCD.Load(limitID)
|
||
}
|
||
func cdcheck3(ctx *zero.Ctx) *rate.Limiter {
|
||
limitID := strconv.FormatInt(ctx.Event.GroupID, 10) + strconv.FormatInt(ctx.Event.UserID, 10) + "3"
|
||
return skillCD.Load(limitID)
|
||
}
|
||
func iscding(ctx *zero.Ctx) {
|
||
ctx.SendChain(message.Text("你的技能现在正在CD中"))
|
||
}
|
||
func iscding2(ctx *zero.Ctx) {
|
||
ctx.SendChain(message.Text("打灭，禁止离婚 (你的技能正在CD中)"))
|
||
}
|
||
|
||
// 注入判断 是否为单身
|
||
func checkdog(ctx *zero.Ctx) bool {
|
||
// 得先判断用户是否存在才行在，再重置
|
||
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("额，你的target好像不存在？"))
|
||
return false
|
||
}
|
||
// 判断是否需要重置
|
||
gid := ctx.Event.GroupID
|
||
ok, err := 民政局.开门时间(gid)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("群状态查询失败\n[error]", err))
|
||
return false
|
||
}
|
||
if ok {
|
||
return true // 重置后也全是单身
|
||
}
|
||
// 获取用户信息
|
||
uid := ctx.Event.UserID
|
||
uidtarget, uidstatus, err := 民政局.查户口(gid, uid)
|
||
switch {
|
||
case uidstatus == "错":
|
||
ctx.SendChain(message.Text("用户状态查询失败\n[error]", err))
|
||
return false
|
||
case uidstatus != "单" && (uidtarget.Target == 0 || uidtarget.User == 0): // 如果是单身贵族
|
||
ctx.SendChain(message.Text("今天的你是单身贵族噢"))
|
||
return false
|
||
case (uidstatus == "攻" && uidtarget.Target == fiancee) ||
|
||
(uidstatus == "受" && uidtarget.User == fiancee):
|
||
ctx.SendChain(message.Text("笨蛋！你们已经在一起了！"))
|
||
return false
|
||
case uidstatus == "攻": // 如果如为攻
|
||
ctx.SendChain(message.Text("笨蛋~你家里还有个吃白饭的w"))
|
||
return false
|
||
case uidstatus == "受": // 如果为受
|
||
ctx.SendChain(message.Text("该是0就是0，当0有什么不好"))
|
||
return false
|
||
}
|
||
fianceeinfo, fianceestatus, err := 民政局.查户口(gid, fiancee)
|
||
switch {
|
||
case fianceestatus == "错":
|
||
ctx.SendChain(message.Text("对象状态查询失败\n[error]", err))
|
||
case fianceestatus == "单": // 如果为单身狗
|
||
return true
|
||
case fianceestatus != "单" && (fianceeinfo.Target == 0 || fianceeinfo.User == 0): // 如果是单身贵族
|
||
ctx.SendChain(message.Text("今天的ta是单身贵族噢"))
|
||
case fianceestatus == "攻": // 如果如为攻
|
||
ctx.SendChain(message.Text("他有别的女人了，你该放下了"))
|
||
case fianceestatus == "受": // 如果为受
|
||
ctx.SendChain(message.Text("ta被别人娶了，你来晚力"))
|
||
}
|
||
return false
|
||
}
|
||
|
||
// 注入判断 是否满足小三要求
|
||
func checkcp(ctx *zero.Ctx) bool {
|
||
// 得先判断用户是否存在才行在，再重置
|
||
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("额，你的target好像不存在？"))
|
||
return false
|
||
}
|
||
// 判断是否需要重置
|
||
gid := ctx.Event.GroupID
|
||
ok, err := 民政局.开门时间(gid)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("群状态查询失败\n[error]", err))
|
||
return false
|
||
}
|
||
if ok {
|
||
ctx.SendChain(message.Text("ta现在还是单身哦，快向ta表白吧！"))
|
||
return false // 重置后也全是单身
|
||
}
|
||
uid := ctx.Event.UserID
|
||
fianceeinfo, fianceestatus, err := 民政局.查户口(gid, fiancee)
|
||
switch {
|
||
case fianceestatus == "错":
|
||
ctx.SendChain(message.Text("对象状态查询失败\n[error]", err))
|
||
return false
|
||
case fianceestatus == "单": // 如果为单身狗
|
||
if fiancee == uid {
|
||
return true
|
||
}
|
||
ctx.SendChain(message.Text("ta现在还是单身哦，快向ta表白吧！"))
|
||
return false
|
||
case fianceestatus != "单" && (fianceeinfo.Target == 0 || fianceeinfo.User == 0): // 如果是单身贵族
|
||
ctx.SendChain(message.Text("今天的ta是单身贵族噢"))
|
||
return false
|
||
case (fianceestatus == "攻" && fianceeinfo.Target == fiancee) ||
|
||
(fianceestatus == "受" && fianceeinfo.User == fiancee):
|
||
ctx.SendChain(message.Text("笨蛋！你们已经在一起了！"))
|
||
return false
|
||
}
|
||
// 获取用户信息
|
||
uidtarget, uidstatus, err := 民政局.查户口(gid, uid)
|
||
switch {
|
||
case uidstatus == "错":
|
||
ctx.SendChain(message.Text("用户状态查询失败\n[error]", err))
|
||
case uidstatus == "单": // 如果为单身狗
|
||
return true
|
||
case uidstatus != "单" && (uidtarget.Target == 0 || uidtarget.User == 0): // 如果是单身贵族
|
||
ctx.SendChain(message.Text("今天的你是单身贵族噢"))
|
||
case uidstatus == "攻": // 如果如为攻
|
||
ctx.SendChain(message.Text("打灭，不给纳小妾！"))
|
||
case uidstatus == "受": // 如果为受
|
||
ctx.SendChain(message.Text("该是0就是0，当0有什么不好"))
|
||
}
|
||
return false
|
||
}
|