mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 22:00:11 +08:00
793cac09cb
* fix:捕获协程异常 * feat:添加睡眠时间 * feat:添加睡眠管理功能 * fix:修改对标时间 * fix:修改注释函数名，使用一个时间 * feat:让大家能够更加清楚的看到每一个功能 * feat:把群管放在control,增加退群信息 * fix:不会recover * fix:原本的正则限制机器人回复回复的消息，故去掉正则 * fix:添加服务详情,整理manager * feat:chat进control Co-authored-by: Guohuiyuan <haibaraguo@yeahka.com>
514 lines
17 KiB
Go
514 lines
17 KiB
Go
// Package manager 群管
|
||
package manager
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/FloatTech/ZeroBot-Plugin/control"
|
||
"io"
|
||
"math/rand"
|
||
"os"
|
||
"sort"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/RomiChan/protobuf/proto"
|
||
"github.com/sirupsen/logrus"
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
"github.com/wdvxdr1123/ZeroBot/extension/rate"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
|
||
"github.com/FloatTech/ZeroBot-Plugin/plugin_manager/timer"
|
||
"github.com/FloatTech/ZeroBot-Plugin/utils/file"
|
||
"github.com/FloatTech/ZeroBot-Plugin/utils/math"
|
||
)
|
||
|
||
const (
|
||
datapath = "data/manager/"
|
||
confile = datapath + "config.pb"
|
||
timerfile = datapath + "timers.pb"
|
||
hint = "====群管====\n" +
|
||
"- 禁言@QQ 1分钟\n" +
|
||
"- 解除禁言 @QQ\n" +
|
||
"- 我要自闭 1分钟\n" +
|
||
"- 开启全员禁言\n" +
|
||
"- 解除全员禁言\n" +
|
||
"- 升为管理@QQ\n" +
|
||
"- 取消管理@QQ\n" +
|
||
"- 修改名片@QQ XXX\n" +
|
||
"- 修改头衔@QQ XXX\n" +
|
||
"- 申请头衔 XXX\n" +
|
||
"- 踢出群聊@QQ\n" +
|
||
"- 退出群聊 1234@bot\n" +
|
||
"- 群聊转发 1234 XXX\n" +
|
||
"- 私聊转发 0000 XXX\n" +
|
||
"- 在MM月dd日的hh点mm分时(用http://url)提醒大家XXX\n" +
|
||
"- 在MM月[每周|周几]的hh点mm分时(用http://url)提醒大家XXX\n" +
|
||
"- 取消在MM月dd日的hh点mm分的提醒\n" +
|
||
"- 取消在MM月[每周|周几]的hh点mm分的提醒\n" +
|
||
"- 在\"cron\"时(用[url])提醒大家[xxx]\n" +
|
||
"- 取消在\"cron\"的提醒\n" +
|
||
"- 列出所有提醒\n" +
|
||
"- 翻牌\n" +
|
||
"- 设置欢迎语XXX\n" +
|
||
"- [开启|关闭]入群验证"
|
||
)
|
||
|
||
var (
|
||
config Config
|
||
limit = rate.NewManager(time.Minute*5, 2)
|
||
clock timer.Clock
|
||
)
|
||
var engine = control.Register("manager", &control.Options{
|
||
DisableOnDefault: false,
|
||
Help: hint,
|
||
})
|
||
|
||
func init() { // 插件主体
|
||
loadConfig()
|
||
go func() {
|
||
time.Sleep(time.Second + time.Millisecond*time.Duration(rand.Intn(1000)))
|
||
clock = timer.NewClock(timerfile)
|
||
}()
|
||
// 升为管理
|
||
engine.OnRegex(`^升为管理.*?(\d+)`, zero.OnlyGroup, zero.SuperUserPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupAdmin(
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 被升为管理的人的qq
|
||
true,
|
||
)
|
||
nickname := ctx.GetGroupMemberInfo( // 被升为管理的人的昵称
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 被升为管理的人的qq
|
||
false,
|
||
).Get("nickname").Str
|
||
ctx.SendChain(message.Text(nickname + " 升为了管理~"))
|
||
})
|
||
// 取消管理
|
||
engine.OnRegex(`^取消管理.*?(\d+)`, zero.OnlyGroup, zero.SuperUserPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupAdmin(
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 被取消管理的人的qq
|
||
false,
|
||
)
|
||
nickname := ctx.GetGroupMemberInfo( // 被取消管理的人的昵称
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 被取消管理的人的qq
|
||
false,
|
||
).Get("nickname").Str
|
||
ctx.SendChain(message.Text("残念~ " + nickname + " 暂时失去了管理员的资格"))
|
||
})
|
||
// 踢出群聊
|
||
engine.OnRegex(`^踢出群聊.*?(\d+)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupKick(
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 被踢出群聊的人的qq
|
||
false,
|
||
)
|
||
nickname := ctx.GetGroupMemberInfo( // 被踢出群聊的人的昵称
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 被踢出群聊的人的qq
|
||
false,
|
||
).Get("nickname").Str
|
||
ctx.SendChain(message.Text("残念~ " + nickname + " 被放逐"))
|
||
})
|
||
// 退出群聊
|
||
engine.OnRegex(`^退出群聊.*?(\d+)`, zero.OnlyToMe, zero.SuperUserPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupLeave(
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 要退出的群的群号
|
||
true,
|
||
)
|
||
})
|
||
// 开启全体禁言
|
||
engine.OnRegex(`^开启全员禁言$`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupWholeBan(
|
||
ctx.Event.GroupID,
|
||
true,
|
||
)
|
||
ctx.SendChain(message.Text("全员自闭开始~"))
|
||
})
|
||
// 解除全员禁言
|
||
engine.OnRegex(`^解除全员禁言$`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupWholeBan(
|
||
ctx.Event.GroupID,
|
||
false,
|
||
)
|
||
ctx.SendChain(message.Text("全员自闭结束~"))
|
||
})
|
||
// 禁言
|
||
engine.OnRegex(`^禁言.*?(\d+).*?\s(\d+)(.*)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
duration := strToInt(ctx.State["regex_matched"].([]string)[2])
|
||
switch ctx.State["regex_matched"].([]string)[3] {
|
||
case "分钟":
|
||
//
|
||
case "小时":
|
||
duration *= 60
|
||
case "天":
|
||
duration *= 60 * 24
|
||
default:
|
||
//
|
||
}
|
||
if duration >= 43200 {
|
||
duration = 43199 // qq禁言最大时长为一个月
|
||
}
|
||
ctx.SetGroupBan(
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 要禁言的人的qq
|
||
duration*60, // 要禁言的时间（分钟）
|
||
)
|
||
ctx.SendChain(message.Text("小黑屋收留成功~"))
|
||
})
|
||
// 解除禁言
|
||
engine.OnRegex(`^解除禁言.*?(\d+)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupBan(
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 要解除禁言的人的qq
|
||
0,
|
||
)
|
||
ctx.SendChain(message.Text("小黑屋释放成功~"))
|
||
})
|
||
// 自闭禁言
|
||
engine.OnRegex(`^(我要自闭|禅定).*?(\d+)(.*)`, zero.OnlyGroup).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
duration := strToInt(ctx.State["regex_matched"].([]string)[2])
|
||
switch ctx.State["regex_matched"].([]string)[3] {
|
||
case "分钟", "min", "mins", "m":
|
||
break
|
||
case "小时", "hour", "hours", "h":
|
||
duration *= 60
|
||
case "天", "day", "days", "d":
|
||
duration *= 60 * 24
|
||
default:
|
||
break
|
||
}
|
||
if duration >= 43200 {
|
||
duration = 43199 // qq禁言最大时长为一个月
|
||
}
|
||
ctx.SetGroupBan(
|
||
ctx.Event.GroupID,
|
||
ctx.Event.UserID,
|
||
duration*60, // 要自闭的时间（分钟）
|
||
)
|
||
ctx.SendChain(message.Text("那我就不手下留情了~"))
|
||
})
|
||
// 修改名片
|
||
engine.OnRegex(`^修改名片.*?(\d+).*?\s(.*)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupCard(
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 被修改群名片的人
|
||
ctx.State["regex_matched"].([]string)[2], // 修改成的群名片
|
||
)
|
||
ctx.SendChain(message.Text("嗯！已经修改了"))
|
||
})
|
||
// 修改头衔
|
||
engine.OnRegex(`^修改头衔.*?(\d+).*?\s(.*)`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupSpecialTitle(
|
||
ctx.Event.GroupID,
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 被修改群头衔的人
|
||
ctx.State["regex_matched"].([]string)[2], // 修改成的群头衔
|
||
)
|
||
ctx.SendChain(message.Text("嗯！已经修改了"))
|
||
})
|
||
// 申请头衔
|
||
engine.OnRegex(`^申请头衔(.*)`, zero.OnlyGroup).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SetGroupSpecialTitle(
|
||
ctx.Event.GroupID,
|
||
ctx.Event.UserID, // 被修改群头衔的人
|
||
ctx.State["regex_matched"].([]string)[1], // 修改成的群头衔
|
||
)
|
||
ctx.SendChain(message.Text("嗯！不错的头衔呢~"))
|
||
})
|
||
// 群聊转发
|
||
engine.OnRegex(`^群聊转发.*?(\d+)\s(.*)`, zero.SuperUserPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
// 对CQ码进行反转义
|
||
content := ctx.State["regex_matched"].([]string)[2]
|
||
content = strings.ReplaceAll(content, "[", "[")
|
||
content = strings.ReplaceAll(content, "]", "]")
|
||
ctx.SendGroupMessage(
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 需要发送的群
|
||
content, // 需要发送的信息
|
||
)
|
||
ctx.SendChain(message.Text("📧 --> " + ctx.State["regex_matched"].([]string)[1]))
|
||
})
|
||
// 私聊转发
|
||
engine.OnRegex(`^私聊转发.*?(\d+)\s(.*)`, zero.SuperUserPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
// 对CQ码进行反转义
|
||
content := ctx.State["regex_matched"].([]string)[2]
|
||
content = strings.ReplaceAll(content, "[", "[")
|
||
content = strings.ReplaceAll(content, "]", "]")
|
||
ctx.SendPrivateMessage(
|
||
strToInt(ctx.State["regex_matched"].([]string)[1]), // 需要发送的人的qq
|
||
content, // 需要发送的信息
|
||
)
|
||
ctx.SendChain(message.Text("📧 --> " + ctx.State["regex_matched"].([]string)[1]))
|
||
})
|
||
// 定时提醒
|
||
engine.OnRegex(`^在(.{1,2})月(.{1,3}日|每?周.?)的(.{1,3})点(.{1,3})分时(用.+)?提醒大家(.*)`, zero.AdminPermission, zero.OnlyGroup).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
dateStrs := ctx.State["regex_matched"].([]string)
|
||
ts := timer.GetFilledTimer(dateStrs, ctx.Event.SelfID, false)
|
||
if ts.En() {
|
||
go clock.RegisterTimer(ts, ctx.Event.GroupID, true)
|
||
ctx.SendChain(message.Text("记住了~"))
|
||
} else {
|
||
ctx.SendChain(message.Text("参数非法:" + ts.Alert))
|
||
}
|
||
})
|
||
// 定时 cron 提醒
|
||
engine.OnRegex(`^在"(.*)"时(用.+)?提醒大家(.*)`, zero.AdminPermission, zero.OnlyGroup).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
dateStrs := ctx.State["regex_matched"].([]string)
|
||
var url, alert string
|
||
switch len(dateStrs) {
|
||
case 4:
|
||
url = dateStrs[2]
|
||
alert = dateStrs[3]
|
||
case 3:
|
||
alert = dateStrs[2]
|
||
default:
|
||
ctx.SendChain(message.Text("参数非法!"))
|
||
return
|
||
}
|
||
logrus.Debugln("[manager] cron:", dateStrs[1])
|
||
ts := timer.GetFilledCronTimer(dateStrs[1], alert, url, ctx.Event.SelfID)
|
||
if clock.RegisterTimer(ts, ctx.Event.GroupID, true) {
|
||
ctx.SendChain(message.Text("记住了~"))
|
||
} else {
|
||
ctx.SendChain(message.Text("参数非法:" + ts.Alert))
|
||
}
|
||
})
|
||
// 取消定时
|
||
engine.OnRegex(`^取消在(.{1,2})月(.{1,3}日|每?周.?)的(.{1,3})点(.{1,3})分的提醒`, zero.AdminPermission, zero.OnlyGroup).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
dateStrs := ctx.State["regex_matched"].([]string)
|
||
ts := timer.GetFilledTimer(dateStrs, ctx.Event.SelfID, true)
|
||
ti := ts.GetTimerInfo(ctx.Event.GroupID)
|
||
ok := clock.CancelTimer(ti)
|
||
if ok {
|
||
ctx.SendChain(message.Text("取消成功~"))
|
||
} else {
|
||
ctx.SendChain(message.Text("没有这个定时器哦~"))
|
||
}
|
||
})
|
||
// 取消 cron 定时
|
||
engine.OnRegex(`^取消在"(.*)"的提醒`, zero.AdminPermission, zero.OnlyGroup).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
dateStrs := ctx.State["regex_matched"].([]string)
|
||
ts := timer.Timer{Cron: dateStrs[1]}
|
||
ti := ts.GetTimerInfo(ctx.Event.GroupID)
|
||
ok := clock.CancelTimer(ti)
|
||
if ok {
|
||
ctx.SendChain(message.Text("取消成功~"))
|
||
} else {
|
||
ctx.SendChain(message.Text("没有这个定时器哦~"))
|
||
}
|
||
})
|
||
// 列出本群所有定时
|
||
engine.OnFullMatch("列出所有提醒", zero.AdminPermission, zero.OnlyGroup).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
ctx.SendChain(message.Text(clock.ListTimers(ctx.Event.GroupID)))
|
||
})
|
||
// 随机点名
|
||
engine.OnFullMatchGroup([]string{"翻牌"}, zero.OnlyGroup).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
if !limit.Load(ctx.Event.UserID).Acquire() {
|
||
ctx.SendChain(message.Text("少女祈祷中......"))
|
||
return
|
||
}
|
||
// 无缓存获取群员列表
|
||
list := ctx.CallAction("get_group_member_list", zero.Params{
|
||
"group_id": ctx.Event.GroupID,
|
||
"no_cache": true,
|
||
}).Data
|
||
temp := list.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)-10):]
|
||
rand.Seed(time.Now().UnixNano())
|
||
who := temp[rand.Intn(len(temp))]
|
||
if who.Get("user_id").Int() == ctx.Event.SelfID {
|
||
ctx.SendChain(message.Text("幸运儿居然是我自己"))
|
||
return
|
||
}
|
||
if who.Get("user_id").Int() == ctx.Event.UserID {
|
||
ctx.SendChain(message.Text("哎呀，就是你自己了"))
|
||
return
|
||
}
|
||
nick := who.Get("card").Str
|
||
if nick == "" {
|
||
nick = who.Get("nickname").Str
|
||
}
|
||
ctx.SendChain(
|
||
message.Text(
|
||
nick,
|
||
" 就是你啦！",
|
||
),
|
||
)
|
||
})
|
||
// 入群欢迎
|
||
engine.OnNotice().SetBlock(false).FirstPriority().
|
||
Handle(func(ctx *zero.Ctx) {
|
||
if ctx.Event.NoticeType == "group_increase" && ctx.Event.SelfID != ctx.Event.UserID {
|
||
word, ok := config.Welcome[ctx.Event.GroupID]
|
||
if ok {
|
||
ctx.SendChain(message.Text(word))
|
||
} else {
|
||
ctx.SendChain(message.Text("欢迎~"))
|
||
}
|
||
enable, ok1 := config.Checkin[ctx.Event.GroupID]
|
||
if ok1 && enable {
|
||
uid := ctx.Event.UserID
|
||
a := rand.Intn(100)
|
||
b := rand.Intn(100)
|
||
r := a + b
|
||
ctx.SendChain(message.At(uid), message.Text(fmt.Sprintf("考你一道题：%d+%d=?\n如果60秒之内答不上来，%s就要把你踢出去了哦~", a, b, zero.BotConfig.NickName[0])))
|
||
// 匹配发送者进行验证
|
||
rule := func(ctx *zero.Ctx) bool {
|
||
for _, elem := range ctx.Event.Message {
|
||
if elem.Type == "text" {
|
||
text := strings.ReplaceAll(elem.Data["text"], " ", "")
|
||
ans, err := strconv.Atoi(text)
|
||
if err == nil {
|
||
if ans != r {
|
||
ctx.SendChain(message.Text("答案不对哦，再想想吧~"))
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
next := zero.NewFutureEvent("message", 999, false, zero.CheckUser(ctx.Event.UserID), rule)
|
||
recv, cancel := next.Repeat()
|
||
select {
|
||
case <-time.After(time.Minute):
|
||
ctx.SendChain(message.Text("拜拜啦~"))
|
||
ctx.SetGroupKick(ctx.Event.GroupID, uid, false)
|
||
cancel()
|
||
case <-recv:
|
||
cancel()
|
||
ctx.SendChain(message.Text("答对啦~"))
|
||
}
|
||
}
|
||
}
|
||
})
|
||
// 退群提醒
|
||
engine.OnNotice().SetBlock(false).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
if ctx.Event.NoticeType == "group_decrease" {
|
||
userid := ctx.Event.UserID
|
||
nickname := ctx.GetStrangerInfo(userid, false).Get("nickname").String()
|
||
ctx.SendChain(message.Text(nickname, "(", userid, ")", "离开了我们~"))
|
||
}
|
||
})
|
||
// 设置欢迎语
|
||
engine.OnRegex(`^设置欢迎语([\s\S]*)$`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
config.Welcome[ctx.Event.GroupID] = ctx.State["regex_matched"].([]string)[1]
|
||
if saveConfig() == nil {
|
||
ctx.SendChain(message.Text("记住啦!"))
|
||
} else {
|
||
ctx.SendChain(message.Text("出错啦!"))
|
||
}
|
||
})
|
||
// 入群验证开关
|
||
engine.OnRegex(`^(.*)入群验证$`, zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(40).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
option := ctx.State["regex_matched"].([]string)[1]
|
||
switch option {
|
||
case "开启":
|
||
config.Checkin[ctx.Event.GroupID] = true
|
||
case "关闭":
|
||
config.Checkin[ctx.Event.GroupID] = false
|
||
default:
|
||
return
|
||
}
|
||
if saveConfig() == nil {
|
||
ctx.SendChain(message.Text("已", option))
|
||
} else {
|
||
ctx.SendChain(message.Text("出错啦!"))
|
||
}
|
||
})
|
||
// 运行 CQ 码
|
||
engine.OnRegex(`^run(.*)$`, zero.SuperUserPermission).SetBlock(true).SetPriority(0).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
var cmd = ctx.State["regex_matched"].([]string)[1]
|
||
cmd = strings.ReplaceAll(cmd, "[", "[")
|
||
cmd = strings.ReplaceAll(cmd, "]", "]")
|
||
// 可注入，权限为主人
|
||
ctx.Send(cmd)
|
||
})
|
||
// 自动同意加好友，被邀请入群(从qingyunke移过来，先注释)
|
||
/*
|
||
engine.OnRequest().SetBlock(false).FirstPriority().Handle(func(ctx *zero.Ctx) {
|
||
if ctx.Event.RequestType == "friend" {
|
||
ctx.SetFriendAddRequest(ctx.Event.Flag, true, "")
|
||
}
|
||
if ctx.Event.RequestType == "group" && ctx.Event.SubType == "invite" {
|
||
ctx.SetGroupAddRequest(ctx.Event.Flag, "invite", true, "我爱你，mua~")
|
||
}
|
||
})
|
||
*/
|
||
}
|
||
|
||
func strToInt(str string) int64 {
|
||
val, _ := strconv.ParseInt(str, 10, 64)
|
||
return val
|
||
}
|
||
|
||
// loadConfig 加载设置，没有则手动初始化
|
||
func loadConfig() {
|
||
mkdirerr := os.MkdirAll(datapath, 0755)
|
||
if mkdirerr == nil {
|
||
if file.IsExist(confile) {
|
||
f, err := os.Open(confile)
|
||
if err == nil {
|
||
data, err1 := io.ReadAll(f)
|
||
if err1 == nil {
|
||
if len(data) > 0 {
|
||
if proto.Unmarshal(data, &config) == nil {
|
||
return
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
config.Checkin = make(map[int64]bool)
|
||
config.Welcome = make(map[int64]string)
|
||
} else {
|
||
panic(mkdirerr)
|
||
}
|
||
}
|
||
|
||
// saveConfig 保存设置，无此文件则新建
|
||
func saveConfig() error {
|
||
data, err := proto.Marshal(&config)
|
||
if err != nil {
|
||
return err
|
||
} else if file.IsExist(datapath) {
|
||
f, err1 := os.OpenFile(confile, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
|
||
if err1 != nil {
|
||
return err1
|
||
}
|
||
defer f.Close()
|
||
_, err2 := f.Write(data)
|
||
return err2
|
||
}
|
||
return nil
|
||
}
|