扩展插件功能 (#422)

This commit is contained in:
方柳煜
2022-09-22 10:40:26 +08:00
committed by GitHub
parent 76cd2afbd3
commit 11f7f07009
3 changed files with 708 additions and 176 deletions

View File

@@ -1,21 +1,21 @@
package qqwife
import (
"errors"
"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
//nolint: asciicheck
type 婚姻登记 struct {
db *sql.Sqlite
dbmu sync.RWMutex
@@ -34,8 +34,24 @@ type userinfo struct {
// 民政局的当前时间
type updateinfo struct {
GID int64
Updatetime string // 登记时间
Updatetime string // 登记时间
CanMatch int // 订婚开关
CanNtr int // Ntr技能开关
CDtime float64 // CD时间
}
// 好感度系统
type favorability struct {
Userinfo string // 记录用户
Favor int // 好感度
}
// 技能CD记录表
type cdsheet struct {
Time int64 // 时间
GroupID int64 // 群号
UserID int64 // 用户
ModeID int64 // 技能类型
}
func (sql *婚姻登记) 开门时间(gid int64) (ok bool, err error) {
@@ -44,7 +60,12 @@ func (sql *婚姻登记) 开门时间(gid int64) (ok bool, err error) {
ok = false
err = sql.db.Create("updateinfo", &updateinfo{})
if err != nil {
return
if err = sql.db.Drop("updateinfo"); err == nil {
err = sql.db.Create("updateinfo", &updateinfo{})
}
if err != nil {
return
}
}
gidstr := strconv.FormatInt(gid, 10)
dbinfo := updateinfo{}
@@ -55,31 +76,102 @@ func (sql *婚姻登记) 开门时间(gid int64) (ok bool, err error) {
err = sql.db.Insert("updateinfo", &updateinfo{
GID: gid,
Updatetime: time.Now().Format("2006/01/02"),
CanMatch: 1,
CanNtr: 1,
CDtime: 12,
})
if err == nil {
ok = true
}
return
}
// 开门了就拿新的花名册
if time.Now().Format("2006/01/02") == dbinfo.Updatetime {
return
}
err = sql.db.Drop(gidstr)
// 开门了就拿新的花名册
err = sql.db.Drop("group" + gidstr)
if err != nil {
err = sql.db.Create("group"+gidstr, &userinfo{})
return
}
updateinfo := updateinfo{
GID: gid,
Updatetime: time.Now().Format("2006/01/02"),
}
err = sql.db.Insert("updateinfo", &updateinfo)
dbinfo.Updatetime = time.Now().Format("2006/01/02")
err = sql.db.Insert("updateinfo", &dbinfo)
if err == nil {
ok = true
}
return
}
func (sql *婚姻登记) 营业模式(gid int64) (canMatch, canNtr int, err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
err = sql.db.Create("updateinfo", &updateinfo{})
if err != nil {
if err = sql.db.Drop("updateinfo"); err == nil {
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 {
canMatch = 1
canNtr = 1
err = sql.db.Insert("updateinfo", &updateinfo{
GID: gid,
CanMatch: canMatch,
CanNtr: canNtr,
CDtime: 12,
})
return
}
canMatch = dbinfo.CanMatch
canNtr = dbinfo.CanNtr
return
}
func (sql *婚姻登记) 修改模式(gid int64, mode string, stauts int) (err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
err = sql.db.Create("updateinfo", &updateinfo{})
if err != nil {
if err = sql.db.Drop("updateinfo"); err == nil {
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)
switch mode {
case "自由恋爱":
dbinfo.CanMatch = stauts
case "牛头人":
dbinfo.CanNtr = stauts
default:
return errors.New("错误:修改内容不匹配!")
}
if err != nil {
dbinfo.GID = gid
switch mode {
case "自由恋爱":
dbinfo.CanNtr = 1
case "牛头人":
dbinfo.CanMatch = 1
}
dbinfo.CDtime = 12
err = sql.db.Insert("updateinfo", &dbinfo)
return
}
err = sql.db.Insert("updateinfo", &dbinfo)
return
}
func (sql *婚姻登记) 清理花名册(gid string) error {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
@@ -91,7 +183,7 @@ func (sql *婚姻登记) 清理花名册(gid string) error {
grouplist = []string{gid}
}
for _, gid := range grouplist {
err = sql.db.Drop(gid)
err = sql.db.Drop("group" + gid)
if err != nil {
continue
}
@@ -108,7 +200,7 @@ func (sql *婚姻登记) 清理花名册(gid string) error {
func (sql *婚姻登记) 查户口(gid, uid int64) (info userinfo, status string, err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
gidstr := strconv.FormatInt(gid, 10)
gidstr := "group" + strconv.FormatInt(gid, 10)
uidstr := strconv.FormatInt(uid, 10)
status = "单"
err = sql.db.Create(gidstr, &userinfo{})
@@ -131,7 +223,7 @@ func (sql *婚姻登记) 查户口(gid, uid int64) (info userinfo, status string
func (sql *婚姻登记) 登记(gid, uid, target int64, username, targetname string) error {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
gidstr := strconv.FormatInt(gid, 10)
gidstr := "group" + strconv.FormatInt(gid, 10)
err := sql.db.Create(gidstr, &userinfo{})
if err != nil {
return err
@@ -153,7 +245,7 @@ func (sql *婚姻登记) 登记(gid, uid, target int64, username, targetname str
func (sql *婚姻登记) 离婚休妻(gid, wife int64) error {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
gidstr := strconv.FormatInt(gid, 10)
gidstr := "group" + strconv.FormatInt(gid, 10)
wifestr := strconv.FormatInt(wife, 10)
return sql.db.Del(gidstr, "where target = "+wifestr)
}
@@ -161,7 +253,7 @@ func (sql *婚姻登记) 离婚休妻(gid, wife int64) error {
func (sql *婚姻登记) 离婚休夫(gid, husband int64) error {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
gidstr := strconv.FormatInt(gid, 10)
gidstr := "group" + strconv.FormatInt(gid, 10)
husbandstr := strconv.FormatInt(husband, 10)
return sql.db.Del(gidstr, "where user = "+husbandstr)
}
@@ -169,7 +261,7 @@ func (sql *婚姻登记) 离婚休夫(gid, husband int64) error {
func (sql *婚姻登记) 花名册(gid int64) (list [][4]string, number int, err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
gidstr := strconv.FormatInt(gid, 10)
gidstr := "group" + strconv.FormatInt(gid, 10)
err = sql.db.Create(gidstr, &userinfo{})
if err != nil {
return
@@ -219,28 +311,210 @@ func slicename(name string, canvas *gg.Context) (resultname string) {
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 (sql *婚姻登记) getFavorability(uid, target int64) (favor int, err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
err = sql.db.Create("favorability", &favorability{})
if err != nil {
return
}
info := favorability{}
uidstr := strconv.FormatInt(uid, 10)
targstr := strconv.FormatInt(target, 10)
err = sql.db.Find("favorability", &info, "where Userinfo glob '*"+uidstr+"+"+targstr+"*'")
if err != nil {
err = sql.db.Insert("favorability", &favorability{
Userinfo: uidstr + "+" + targstr + "+" + uidstr,
Favor: 0,
})
return
}
favor = info.Favor
return
}
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 (sql *婚姻登记) setFavorability(uid, target int64, score int) (favor int, err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
err = sql.db.Create("favorability", &favorability{})
if err != nil {
return
}
info := favorability{}
uidstr := strconv.FormatInt(uid, 10)
targstr := strconv.FormatInt(target, 10)
err = sql.db.Find("favorability", &info, "where Userinfo glob '*"+uidstr+"+"+targstr+"*'")
if err != nil {
err = sql.db.Insert("favorability", &favorability{
Userinfo: uidstr + "+" + targstr + "+" + uidstr,
Favor: score,
})
return score, err
}
info.Favor += score
if info.Favor > 100 {
info.Favor = 100
} else if info.Favor < 0 {
info.Favor = 0
}
err = sql.db.Insert("favorability", &info)
return info.Favor, err
}
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 (sql *婚姻登记) getCDtime(gid int64) (skillCD float64, err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
skillCD = 12
err = sql.db.Create("updateinfo", &updateinfo{})
if err != nil {
if err = sql.db.Drop("updateinfo"); err == nil {
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,
CanMatch: 1,
CanNtr: 1,
CDtime: 12,
})
return
}
return dbinfo.CDtime, nil
}
func iscding(ctx *zero.Ctx) {
ctx.SendChain(message.Text("你的技能现在正在CD中"))
// 设置技能时长
func (sql *婚姻登记) setCDtime(gid int64, cdTime float64) (err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
err = sql.db.Create("updateinfo", &updateinfo{})
if err != nil {
if err = sql.db.Drop("updateinfo"); err == nil {
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,
CanMatch: 1,
CanNtr: 1,
CDtime: cdTime,
})
return
}
dbinfo.CDtime = cdTime
err = sql.db.Insert("updateinfo", &dbinfo)
return
}
func iscding2(ctx *zero.Ctx) {
ctx.SendChain(message.Text("打灭,禁止离婚 (你的技能正在CD中)"))
// 记录CD
func (sql *婚姻登记) writeCDtime(gid, uid, mun int64) error {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
err := sql.db.Create("cdsheet", &cdsheet{})
if err != nil {
if err = sql.db.Drop("cdsheet"); err == nil {
err = sql.db.Create("cdsheet", &cdsheet{})
}
if err != nil {
return err
}
}
err = sql.db.Insert("cdsheet", &cdsheet{
Time: time.Now().Unix(),
GroupID: gid,
UserID: uid,
ModeID: mun,
})
return err
}
// 判断CD是否过时
func (sql *婚姻登记) compareCDtime(gid, uid, mun int64, cdtime float64) (ok bool, err error) {
sql.dbmu.Lock()
defer sql.dbmu.Unlock()
ok = false
err = sql.db.Create("cdsheet", &cdsheet{})
if err != nil {
if err = sql.db.Drop("cdsheet"); err == nil {
err = sql.db.Create("cdsheet", &cdsheet{})
}
if err != nil {
return
}
}
limitID := "where GroupID is " + strconv.FormatInt(gid, 10) +
" and UserID is " + strconv.FormatInt(uid, 10) +
" and ModeID is " + strconv.FormatInt(mun, 10)
exist := sql.db.CanFind("cdsheet", limitID)
if !exist {
return true, nil
}
cdinfo := cdsheet{}
err = sql.db.Find("cdsheet", &cdinfo, limitID)
if err != nil {
return
}
getTime := time.Unix(cdinfo.Time, 0)
if time.Since(getTime).Hours() > cdtime {
// 如果CD已过就删除
err = sql.db.Del("cdsheet", limitID)
return true, err
}
return
}
// 注入判断 是否为单身
func checkdog(ctx *zero.Ctx) bool {
mode := int64(1) // 指代技能1
gid := ctx.Event.GroupID
uid := ctx.Event.UserID
// 获取CD
cdTime, err := 民政局.getCDtime(gid)
if err != nil {
ctx.SendChain(message.Text("[qqwife]获取该群技能CD错误(将以CD12H计算)\n", err))
}
ok, err := 民政局.compareCDtime(gid, uid, mode, cdTime)
if err != nil {
ctx.SendChain(message.Text("[qqwife]查询用户CD状态失败,请重试\n", err))
return false
}
if !ok {
ctx.SendChain(message.Text("你的技能还在CD中..."))
return false
}
// 写入CD
err = 民政局.writeCDtime(gid, uid, mode)
if err != nil {
ctx.SendChain(message.At(uid), message.Text("[qqwife]你的技能CD记录失败\n", err))
}
// 判断是否符合条件
stauts, _, err := 民政局.营业模式(gid)
if err != nil {
ctx.SendChain(message.Text("[qqwife]", err))
return false
}
if stauts == 0 {
ctx.SendChain(message.Text("你群包分配,别在娶妻上面下功夫,好好水群"))
return false
}
// 得先判断用户是否存在才行在,再重置
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
if err != nil {
@@ -248,21 +522,19 @@ func checkdog(ctx *zero.Ctx) bool {
return false
}
// 判断是否需要重置
gid := ctx.Event.GroupID
ok, err := 民政局.开门时间(gid)
ok, err = 民政局.开门时间(gid)
if err != nil {
ctx.SendChain(message.Text("群状态查询失败\n[error]", err))
ctx.SendChain(message.Text("[qqwife]群状态查询失败\n", 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))
ctx.SendChain(message.Text("[qqwife]用户状态查询失败\n", err))
return false
case uidstatus != "单" && (uidtarget.Target == 0 || uidtarget.User == 0): // 如果是单身贵族
ctx.SendChain(message.Text("今天的你是单身贵族噢"))
@@ -281,7 +553,7 @@ func checkdog(ctx *zero.Ctx) bool {
fianceeinfo, fianceestatus, err := 民政局.查户口(gid, fiancee)
switch {
case fianceestatus == "错":
ctx.SendChain(message.Text("对象状态查询失败\n[error]", err))
ctx.SendChain(message.Text("[qqwife]对象状态查询失败\n", err))
case fianceestatus == "单": // 如果为单身狗
return true
case fianceestatus != "单" && (fianceeinfo.Target == 0 || fianceeinfo.User == 0): // 如果是单身贵族
@@ -296,6 +568,38 @@ func checkdog(ctx *zero.Ctx) bool {
// 注入判断 是否满足小三要求
func checkcp(ctx *zero.Ctx) bool {
mode := int64(2) // 指代技能2
gid := ctx.Event.GroupID
uid := ctx.Event.UserID
// 获取CD
cdTime, err := 民政局.getCDtime(gid)
if err != nil {
ctx.SendChain(message.Text("[qqwife]获取该群技能CD错误(将以CD12H计算)\n", err))
}
ok, err := 民政局.compareCDtime(gid, uid, mode, cdTime)
if err != nil {
ctx.SendChain(message.Text("[qqwife]查询用户CD状态失败,请重试\n", err))
return false
}
if !ok {
ctx.SendChain(message.Text("你的技能还在CD中..."))
return false
}
// 写入CD
err = 民政局.writeCDtime(gid, uid, mode)
if err != nil {
ctx.SendChain(message.At(uid), message.Text("[qqwife]你的技能CD记录失败\n", err))
}
// 判断是否符合条件
_, stauts, err := 民政局.营业模式(gid)
if err != nil {
ctx.SendChain(message.Text("[qqwife]", err))
return false
}
if stauts == 0 {
ctx.SendChain(message.Text("你群发布了牛头人禁止令,放弃吧"))
return false
}
// 得先判断用户是否存在才行在,再重置
fiancee, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
if err != nil {
@@ -303,21 +607,19 @@ func checkcp(ctx *zero.Ctx) bool {
return false
}
// 判断是否需要重置
gid := ctx.Event.GroupID
ok, err := 民政局.开门时间(gid)
ok, err = 民政局.开门时间(gid)
if err != nil {
ctx.SendChain(message.Text("群状态查询失败\n[error]", err))
ctx.SendChain(message.Text("[qqwife]群状态查询失败\n", 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))
ctx.SendChain(message.Text("[qqwife]对象状态查询失败\n", err))
return false
case fianceestatus == "单": // 如果为单身狗
if fiancee == uid {
@@ -337,7 +639,7 @@ func checkcp(ctx *zero.Ctx) bool {
uidtarget, uidstatus, err := 民政局.查户口(gid, uid)
switch {
case uidstatus == "错":
ctx.SendChain(message.Text("用户状态查询失败\n[error]", err))
ctx.SendChain(message.Text("[qqwife]用户状态查询失败\n", err))
case uidstatus == "单": // 如果为单身狗
return true
case uidstatus != "单" && (uidtarget.Target == 0 || uidtarget.User == 0): // 如果是单身贵族
@@ -349,3 +651,123 @@ func checkcp(ctx *zero.Ctx) bool {
}
return false
}
// 注入判断 是否满足离婚要求
func checkdivorce(ctx *zero.Ctx) bool {
mode := int64(3) // 指代技能3
gid := ctx.Event.GroupID
uid := ctx.Event.UserID
// 获取CD
cdTime, err := 民政局.getCDtime(gid)
if err != nil {
ctx.SendChain(message.Text("[qqwife]获取该群技能CD错误(将以CD12H计算)\n", err))
}
ok, err := 民政局.compareCDtime(gid, uid, mode, cdTime)
if err != nil {
ctx.SendChain(message.Text("[qqwife]查询用户CD状态失败,请重试\n", err))
return false
}
if !ok {
ctx.SendChain(message.Text("你的技能还在CD中..."))
return false
}
// 写入CD
err = 民政局.writeCDtime(gid, uid, mode)
if err != nil {
ctx.SendChain(message.At(uid), message.Text("[qqwife]你的技能CD记录失败\n", err))
}
// 判断是否符合条件
_, uidstatus, err := 民政局.查户口(gid, uid)
switch uidstatus {
case "错":
ctx.SendChain(message.Text("[qqwife]数据库发生问题力\n", err))
return false
case "单":
ctx.SendChain(message.Text("今天你还没结婚哦"))
return false
}
return true
}
// 注入判断 是否满足做媒要求
func checkCondition(ctx *zero.Ctx) bool {
mode := int64(4) // 指代技能4
gid := ctx.Event.GroupID
uid := ctx.Event.UserID
// 获取CD
cdTime, err := 民政局.getCDtime(gid)
if err != nil {
ctx.SendChain(message.Text("[qqwife]获取该群技能CD错误(将以CD12H计算)\n", err))
}
ok, err := 民政局.compareCDtime(gid, uid, mode, cdTime)
if err != nil {
ctx.SendChain(message.Text("[qqwife]查询用户CD状态失败,请重试\n", err))
return false
}
if !ok {
ctx.SendChain(message.Text("你的技能还在CD中..."))
return false
}
// 写入CD
err = 民政局.writeCDtime(gid, uid, mode)
if err != nil {
ctx.SendChain(message.At(uid), message.Text("[qqwife]你的技能CD记录失败\n", err))
}
// 得先判断用户是否存在才行在,再重置
gayOne, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[1], 10, 64)
if err != nil {
ctx.SendChain(message.Text("额,攻方好像不存在?"))
return false
}
gayZero, err := strconv.ParseInt(ctx.State["regex_matched"].([]string)[2], 10, 64)
if err != nil {
ctx.SendChain(message.Text("额,受方好像不存在?"))
return false
}
if gayOne == uid || gayZero == uid {
ctx.SendChain(message.Text("禁止自己给自己做媒!"))
return false
}
if gayOne == gayZero {
ctx.SendChain(message.Text("你这个媒人XP很怪咧不能这样噢"))
return false
}
// 判断是否需要重置
ok, err = 民政局.开门时间(gid)
if err != nil {
ctx.SendChain(message.Text("[qqwife]群状态查询失败\n", err))
return false
}
if ok {
return true // 重置后也全是单身
}
fianceeinfo, fianceestatus, err := 民政局.查户口(gid, gayOne)
switch {
case fianceestatus == "错":
ctx.SendChain(message.Text("[qqwife]对象状态查询失败\n", err))
return false
case fianceestatus != "单" && (fianceeinfo.Target == 0 || fianceeinfo.User == 0): // 如果是单身贵族
ctx.SendChain(message.Text("今天的攻方是单身贵族噢"))
return false
case (fianceestatus == "攻" && fianceeinfo.Target == gayZero) ||
(fianceestatus == "受" && fianceeinfo.User == gayZero):
ctx.SendChain(message.Text("笨蛋ta们已经在一起了"))
return false
case fianceestatus != "单":
ctx.SendChain(message.Text("攻方不是单身,不允许给这种人做媒!"))
return false
}
// 获取用户信息
uidtarget, uidstatus, err := 民政局.查户口(gid, gayZero)
switch {
case uidstatus == "错":
ctx.SendChain(message.Text("[qqwife]用户状态查询失败\n", err))
case uidstatus == "单": // 如果为单身狗
return true
case uidstatus != "单" && (uidtarget.Target == 0 || uidtarget.User == 0): // 如果是单身贵族
ctx.SendChain(message.Text("今天的你是单身贵族噢"))
case uidstatus != "单":
ctx.SendChain(message.Text("受方不是单身,不允许给这种人做媒!"))
}
return false
}