mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:50:09 +08:00
288 lines
8.9 KiB
Go
288 lines
8.9 KiB
Go
// Package score 签到，答题得分
|
|
package score
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/Coloured-glaze/gg"
|
|
"github.com/golang/freetype"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/tidwall/gjson"
|
|
"github.com/wcharczuk/go-chart/v2"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
|
|
ctrl "github.com/FloatTech/zbpctrl"
|
|
"github.com/FloatTech/zbputils/binary"
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/FloatTech/zbputils/file"
|
|
"github.com/FloatTech/zbputils/img"
|
|
"github.com/FloatTech/zbputils/img/text"
|
|
"github.com/FloatTech/zbputils/img/writer"
|
|
"github.com/FloatTech/zbputils/web"
|
|
)
|
|
|
|
const (
|
|
backgroundURL = "https://mirlkoi.ifast3.vipnps.vip/api.php?sort=pc&type=json"
|
|
referer = "https://iw233.cn/main.html"
|
|
ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36"
|
|
signinMax = 1
|
|
// SCOREMAX 分数上限定为120
|
|
SCOREMAX = 120
|
|
)
|
|
|
|
var levelArray = [...]int{0, 1, 2, 5, 10, 20, 35, 55, 75, 100, 120}
|
|
|
|
func init() {
|
|
engine := control.Register("score", &ctrl.Options[*zero.Ctx]{
|
|
DisableOnDefault: false,
|
|
Help: "签到得分\n- 签到\n- 获得签到背景[@xxx] | 获得签到背景\n- 查看分数排名",
|
|
PrivateDataFolder: "score",
|
|
})
|
|
cachePath := engine.DataFolder() + "cache/"
|
|
go func() {
|
|
_ = os.RemoveAll(cachePath)
|
|
err := os.MkdirAll(cachePath, 0755)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
sdb = initialize(engine.DataFolder() + "score.db")
|
|
}()
|
|
engine.OnFullMatch("签到", zero.OnlyGroup).Limit(ctxext.LimitByGroup).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
uid := ctx.Event.UserID
|
|
now := time.Now()
|
|
today := now.Format("20060102")
|
|
si := sdb.GetSignInByUID(uid)
|
|
siUpdateTimeStr := si.UpdatedAt.Format("20060102")
|
|
drawedFile := cachePath + strconv.FormatInt(uid, 10) + today + "signin.png"
|
|
|
|
picFile := cachePath + strconv.FormatInt(uid, 10) + today + ".png"
|
|
|
|
if si.Count >= signinMax && siUpdateTimeStr == today {
|
|
ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("今天你已经签到过了！"))
|
|
if file.IsExist(drawedFile) {
|
|
ctx.SendChain(message.Image("file:///" + file.BOTPATH + "/" + drawedFile))
|
|
}
|
|
return
|
|
}
|
|
err := initPic(picFile)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
back, err := gg.LoadImage(picFile)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
if siUpdateTimeStr != today {
|
|
_ = sdb.InsertOrUpdateSignInCountByUID(uid, 0)
|
|
}
|
|
|
|
_ = sdb.InsertOrUpdateSignInCountByUID(uid, si.Count+1)
|
|
|
|
// 避免图片过大，最大 1280*720
|
|
back = img.Limit(back, 1280, 720)
|
|
|
|
canvas := gg.NewContext(back.Bounds().Size().X, int(float64(back.Bounds().Size().Y)*1.7))
|
|
canvas.SetRGB(1, 1, 1)
|
|
canvas.Clear()
|
|
canvas.DrawImage(back, 0, 0)
|
|
|
|
monthWord := now.Format("01/02")
|
|
hourWord := getHourWord(now)
|
|
_, err = file.GetLazyData(text.BoldFontFile, true)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
if err = canvas.LoadFontFace(text.BoldFontFile, float64(back.Bounds().Size().X)*0.1); err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
canvas.SetRGB(0, 0, 0)
|
|
canvas.DrawString(hourWord, float64(back.Bounds().Size().X)*0.1, float64(back.Bounds().Size().Y)*1.2)
|
|
canvas.DrawString(monthWord, float64(back.Bounds().Size().X)*0.6, float64(back.Bounds().Size().Y)*1.2)
|
|
nickName := ctx.CardOrNickName(uid)
|
|
_, err = file.GetLazyData(text.FontFile, true)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
if err = canvas.LoadFontFace(text.FontFile, float64(back.Bounds().Size().X)*0.04); err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
add := 1
|
|
canvas.DrawString(nickName+fmt.Sprintf(" 小熊饼干+%d", add), float64(back.Bounds().Size().X)*0.1, float64(back.Bounds().Size().Y)*1.3)
|
|
score := sdb.GetScoreByUID(uid).Score
|
|
score += add
|
|
if score > SCOREMAX {
|
|
score = SCOREMAX
|
|
ctx.SendChain(message.At(uid), message.Text("你获得的小熊饼干已经达到上限"))
|
|
}
|
|
_ = sdb.InsertOrUpdateScoreByUID(uid, score)
|
|
level := getLevel(score)
|
|
canvas.DrawString("当前小熊饼干:"+strconv.FormatInt(int64(score), 10), float64(back.Bounds().Size().X)*0.1, float64(back.Bounds().Size().Y)*1.4)
|
|
canvas.DrawString("LEVEL:"+strconv.FormatInt(int64(level), 10), float64(back.Bounds().Size().X)*0.1, float64(back.Bounds().Size().Y)*1.5)
|
|
canvas.DrawRectangle(float64(back.Bounds().Size().X)*0.1, float64(back.Bounds().Size().Y)*1.55, float64(back.Bounds().Size().X)*0.6, float64(back.Bounds().Size().Y)*0.1)
|
|
canvas.SetRGB255(150, 150, 150)
|
|
canvas.Fill()
|
|
var nextLevelScore int
|
|
if level < 10 {
|
|
nextLevelScore = levelArray[level+1]
|
|
} else {
|
|
nextLevelScore = SCOREMAX
|
|
}
|
|
canvas.SetRGB255(0, 0, 0)
|
|
canvas.DrawRectangle(float64(back.Bounds().Size().X)*0.1, float64(back.Bounds().Size().Y)*1.55, float64(back.Bounds().Size().X)*0.6*float64(score)/float64(nextLevelScore), float64(back.Bounds().Size().Y)*0.1)
|
|
canvas.SetRGB255(102, 102, 102)
|
|
canvas.Fill()
|
|
canvas.DrawString(fmt.Sprintf("%d/%d", score, nextLevelScore), float64(back.Bounds().Size().X)*0.75, float64(back.Bounds().Size().Y)*1.62)
|
|
|
|
f, err := os.Create(drawedFile)
|
|
if err != nil {
|
|
log.Errorln("[score]", err)
|
|
data, cl := writer.ToBytes(canvas.Image())
|
|
ctx.SendChain(message.ImageBytes(data))
|
|
cl()
|
|
return
|
|
}
|
|
_, err = writer.WriteTo(canvas.Image(), f)
|
|
_ = f.Close()
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Image("file:///" + file.BOTPATH + "/" + drawedFile))
|
|
})
|
|
engine.OnPrefix("获得签到背景", zero.OnlyGroup).Limit(ctxext.LimitByGroup).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
param := ctx.State["args"].(string)
|
|
var uidStr string
|
|
if len(ctx.Event.Message) > 1 && ctx.Event.Message[1].Type == "at" {
|
|
uidStr = ctx.Event.Message[1].Data["qq"]
|
|
} else if param == "" {
|
|
uidStr = strconv.FormatInt(ctx.Event.UserID, 10)
|
|
}
|
|
picFile := cachePath + uidStr + time.Now().Format("20060102") + ".png"
|
|
if file.IsNotExist(picFile) {
|
|
ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("请先签到！"))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Image("file:///" + file.BOTPATH + "/" + picFile))
|
|
})
|
|
engine.OnFullMatch("查看分数排名", zero.OnlyGroup).Limit(ctxext.LimitByGroup).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
today := time.Now().Format("20060102")
|
|
drawedFile := cachePath + today + "scoreRank.png"
|
|
if file.IsExist(drawedFile) {
|
|
ctx.SendChain(message.Image("file:///" + file.BOTPATH + "/" + drawedFile))
|
|
return
|
|
}
|
|
st, err := sdb.GetScoreRankByTopN(10)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
if len(st) == 0 {
|
|
ctx.SendChain(message.Text("ERROR:目前还没有人签到过"))
|
|
return
|
|
}
|
|
_, err = file.GetLazyData(text.FontFile, true)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
b, err := os.ReadFile(text.FontFile)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
font, err := freetype.ParseFont(b)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
f, err := os.Create(drawedFile)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
bars := make([]chart.Value, len(st))
|
|
for i, v := range st {
|
|
bars[i].Value = float64(v.Score)
|
|
bars[i].Label = ctx.CardOrNickName(v.UID)
|
|
}
|
|
err = chart.BarChart{
|
|
Font: font,
|
|
Title: "饼干排名",
|
|
Background: chart.Style{
|
|
Padding: chart.Box{
|
|
Top: 40,
|
|
},
|
|
},
|
|
Height: 500,
|
|
BarWidth: 50,
|
|
Bars: bars,
|
|
}.Render(chart.PNG, f)
|
|
_ = f.Close()
|
|
if err != nil {
|
|
_ = os.Remove(drawedFile)
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Image("file:///" + file.BOTPATH + "/" + drawedFile))
|
|
})
|
|
}
|
|
|
|
func getHourWord(t time.Time) string {
|
|
h := t.Hour()
|
|
switch {
|
|
case 6 <= h && h < 12:
|
|
return "早上好"
|
|
case 12 <= h && h < 14:
|
|
return "中午好"
|
|
case 14 <= h && h < 19:
|
|
return "下午好"
|
|
case 19 <= h && h < 24:
|
|
return "晚上好"
|
|
case 0 <= h && h < 6:
|
|
return "凌晨好"
|
|
default:
|
|
return ""
|
|
}
|
|
}
|
|
|
|
func getLevel(count int) int {
|
|
for k, v := range levelArray {
|
|
if count == v {
|
|
return k
|
|
} else if count < v {
|
|
return k - 1
|
|
}
|
|
}
|
|
return -1
|
|
}
|
|
|
|
func initPic(picFile string) error {
|
|
if file.IsExist(picFile) {
|
|
return nil
|
|
}
|
|
data, err := web.RequestDataWith(web.NewDefaultClient(), backgroundURL, "GET", referer, ua)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
picURL := gjson.Get(binary.BytesToString(data), "pic.0").Str
|
|
data, err = web.RequestDataWith(web.NewDefaultClient(), picURL, "GET", "", ua)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return os.WriteFile(picFile, data, 0644)
|
|
}
|