mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 05:30:07 +08:00
381 lines
12 KiB
Go
381 lines
12 KiB
Go
// Package score 签到，答题得分
|
||
package score
|
||
|
||
import (
|
||
"bytes"
|
||
"image"
|
||
"image/color"
|
||
"math"
|
||
"math/rand"
|
||
"os"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/disintegration/imaging"
|
||
|
||
"github.com/FloatTech/ZeroBot-Plugin/kanban/banner"
|
||
|
||
"github.com/FloatTech/AnimeAPI/bilibili"
|
||
"github.com/FloatTech/AnimeAPI/wallet"
|
||
"github.com/FloatTech/floatbox/file"
|
||
"github.com/FloatTech/floatbox/process"
|
||
"github.com/FloatTech/floatbox/web"
|
||
"github.com/FloatTech/gg"
|
||
"github.com/FloatTech/imgfactory"
|
||
ctrl "github.com/FloatTech/zbpctrl"
|
||
"github.com/FloatTech/zbputils/control"
|
||
"github.com/FloatTech/zbputils/ctxext"
|
||
"github.com/FloatTech/zbputils/img/text"
|
||
"github.com/golang/freetype"
|
||
"github.com/wcharczuk/go-chart/v2"
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
)
|
||
|
||
const (
|
||
backgroundURL = "https://iw233.cn/api.php?sort=pc"
|
||
referer = "https://weibo.com/"
|
||
signinMax = 1
|
||
// SCOREMAX 分数上限定为1200
|
||
SCOREMAX = 1200
|
||
)
|
||
|
||
var (
|
||
rankArray = [...]int{0, 10, 20, 50, 100, 200, 350, 550, 750, 1000, 1200}
|
||
engine = control.Register("score", &ctrl.Options[*zero.Ctx]{
|
||
DisableOnDefault: false,
|
||
Brief: "签到",
|
||
Help: "- 签到\n- 获得签到背景[@xxx] | 获得签到背景\n- 查看等级排名\n注:为跨群排名\n- 查看我的钱包\n- 查看钱包排名\n注:为本群排行，若群人数太多不建议使用该功能!!!",
|
||
PrivateDataFolder: "score",
|
||
})
|
||
)
|
||
|
||
func init() {
|
||
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("签到").Limit(ctxext.LimitByUser).SetBlock(true).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
uid := ctx.Event.UserID
|
||
now := time.Now()
|
||
today := now.Format("20060102")
|
||
// 签到图片
|
||
drawedFile := cachePath + strconv.FormatInt(uid, 10) + today + "signin.png"
|
||
picFile := cachePath + strconv.FormatInt(uid, 10) + today + ".png"
|
||
// 获取签到时间
|
||
si := sdb.GetSignInByUID(uid)
|
||
siUpdateTimeStr := si.UpdatedAt.Format("20060102")
|
||
switch {
|
||
case 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
|
||
case siUpdateTimeStr != today:
|
||
// 如果是跨天签到就清数据
|
||
err := sdb.InsertOrUpdateSignInCountByUID(uid, 0)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
}
|
||
// 更新签到次数
|
||
err := sdb.InsertOrUpdateSignInCountByUID(uid, si.Count+1)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
// 更新经验
|
||
level := sdb.GetScoreByUID(uid).Score + 1
|
||
if level > SCOREMAX {
|
||
level = SCOREMAX
|
||
ctx.SendChain(message.At(uid), message.Text("你的等级已经达到上限"))
|
||
}
|
||
err = sdb.InsertOrUpdateScoreByUID(uid, level)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
// 更新钱包
|
||
rank := getrank(level)
|
||
add := 1 + rand.Intn(10) + rank*5 // 等级越高获得的钱越高
|
||
err = wallet.InsertWalletOf(uid, add)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
score := wallet.GetWalletOf(uid)
|
||
// 绘图
|
||
getAvatar, err := initPic(picFile, uid)
|
||
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
|
||
}
|
||
// 避免图片过大，最大 1280*720
|
||
back = imgfactory.Limit(back, 1280, 720)
|
||
imgDX := back.Bounds().Dx()
|
||
imgDY := back.Bounds().Dy()
|
||
canvas := gg.NewContext(imgDX, imgDY)
|
||
// draw Aero Style
|
||
aeroStyle := gg.NewContext(imgDX-202, imgDY-202)
|
||
aeroStyle.DrawImage(imaging.Blur(back, 2.5), -100, -100)
|
||
// aero draw image.
|
||
aeroStyle.DrawRoundedRectangle(0, 0, float64(imgDX-200), float64(imgDY-200), 16)
|
||
// SideLine
|
||
aeroStyle.SetLineWidth(3)
|
||
aeroStyle.SetRGBA255(255, 255, 255, 100)
|
||
aeroStyle.StrokePreserve()
|
||
aeroStyle.SetRGBA255(255, 255, 255, 140)
|
||
// fill
|
||
aeroStyle.Fill()
|
||
// draw background
|
||
canvas.DrawImage(back, 0, 0)
|
||
// Aero style combine
|
||
canvas.DrawImage(aeroStyle.Image(), 100, 100)
|
||
canvas.Fill()
|
||
hourWord := getHourWord(now)
|
||
avatar, _, err := image.Decode(bytes.NewReader(getAvatar))
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
avatarf := imgfactory.Size(avatar, 200, 200)
|
||
canvas.DrawImage(avatarf.Circle(0).Image(), 120, 120)
|
||
// draw info(name,coin,etc)
|
||
canvas.SetRGB255(0, 0, 0)
|
||
data, err := file.GetLazyData(text.BoldFontFile, control.Md5File, true)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
if err = canvas.ParseFontFace(data, 50); err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
// draw head
|
||
nickName := ctx.CardOrNickName(uid)
|
||
canvas.DrawStringWrapped(nickName, 350, 180, 0.5, 0.5, 0.5, 0.5, gg.AlignLeft)
|
||
canvas.Fill()
|
||
// main draw
|
||
data, err = file.GetLazyData(text.FontFile, control.Md5File, true)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
if err = canvas.ParseFontFace(data, 30); err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
canvas.DrawStringAnchored(hourWord, 350, 280, 0, 0)
|
||
canvas.DrawStringAnchored("ATRI币 + "+strconv.Itoa(add), 350, 350, 0, 0)
|
||
canvas.DrawStringAnchored("当前ATRI币："+strconv.Itoa(score), 350, 400, 0, 0)
|
||
canvas.DrawStringAnchored("LEVEL: "+strconv.Itoa(getrank(level)), 350, 450, 0, 0)
|
||
// draw Info(Time,etc.)
|
||
getTime := time.Now().Format("2006-01-02 15:04:05")
|
||
getTimeLengthWidth, getTimeLengthHight := canvas.MeasureString(getTime)
|
||
canvas.DrawStringAnchored(getTime, float64(imgDX)-100-20-getTimeLengthWidth/2, float64(imgDY)-100-getTimeLengthHight, 0.5, 0.5) // time
|
||
var nextrankScore int
|
||
if rank < 10 {
|
||
nextrankScore = rankArray[rank+1]
|
||
} else {
|
||
nextrankScore = SCOREMAX
|
||
}
|
||
nextLevelStyle := strconv.Itoa(level) + "/" + strconv.Itoa(nextrankScore)
|
||
getLevelLength, _ := canvas.MeasureString(nextLevelStyle)
|
||
canvas.DrawStringAnchored(nextLevelStyle, 100+getLevelLength, float64(imgDY)-100-getTimeLengthHight, 0.5, 0.5) // time
|
||
canvas.Fill()
|
||
canvas.SetRGB255(255, 255, 255)
|
||
if err = canvas.ParseFontFace(data, 20); err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
canvas.DrawStringAnchored("Created By Zerobot-Plugin "+banner.Version, float64(imgDX)/2, float64(imgDY)-20, 0.5, 0.5) // zbp
|
||
canvas.SetRGB255(0, 0, 0)
|
||
canvas.DrawStringAnchored("Created By Zerobot-Plugin "+banner.Version, float64(imgDX)/2-3, float64(imgDY)-19, 0.5, 0.5) // zbp
|
||
canvas.SetRGB255(255, 255, 255)
|
||
// Gradient
|
||
grad := gg.NewLinearGradient(20, 320, 400, 20)
|
||
grad.AddColorStop(0, color.RGBA{G: 255, A: 255})
|
||
grad.AddColorStop(1, color.RGBA{B: 255, A: 255})
|
||
grad.AddColorStop(0.5, color.RGBA{R: 255, A: 255})
|
||
canvas.SetStrokeStyle(grad)
|
||
canvas.SetLineWidth(4)
|
||
// level array with rectangle work.
|
||
gradLineLength := float64(imgDX-120) - 120
|
||
renderLine := (float64(level) / float64(nextrankScore)) * gradLineLength
|
||
canvas.MoveTo(120, float64(imgDY)-102)
|
||
canvas.LineTo(120+renderLine, float64(imgDY)-102)
|
||
canvas.ClosePath()
|
||
canvas.Stroke()
|
||
// done.
|
||
f, err := os.Create(drawedFile)
|
||
if err != nil {
|
||
data, err := imgfactory.ToBytes(canvas.Image())
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR: ", err))
|
||
return
|
||
}
|
||
ctx.SendChain(message.ImageBytes(data))
|
||
return
|
||
}
|
||
_, err = imgfactory.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, control.Md5File, 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
|
||
}
|
||
var bars []chart.Value
|
||
for _, v := range st {
|
||
if v.Score != 0 {
|
||
bars = append(bars, chart.Value{
|
||
Label: ctx.CardOrNickName(v.UID),
|
||
Value: float64(v.Score),
|
||
})
|
||
}
|
||
}
|
||
err = chart.BarChart{
|
||
Font: font,
|
||
Title: "等级排名(1天只刷新1次)",
|
||
Background: chart.Style{
|
||
Padding: chart.Box{
|
||
Top: 40,
|
||
},
|
||
},
|
||
YAxis: chart.YAxis{
|
||
Range: &chart.ContinuousRange{
|
||
Min: 0,
|
||
Max: math.Ceil(bars[0].Value/10) * 10,
|
||
},
|
||
},
|
||
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 getrank(count int) int {
|
||
for k, v := range rankArray {
|
||
if count == v {
|
||
return k
|
||
} else if count < v {
|
||
return k - 1
|
||
}
|
||
}
|
||
return -1
|
||
}
|
||
|
||
func initPic(picFile string, uid int64) (avatar []byte, err error) {
|
||
if file.IsExist(picFile) {
|
||
return nil, nil
|
||
}
|
||
defer process.SleepAbout1sTo2s()
|
||
url, err := bilibili.GetRealURL(backgroundURL)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
data, err := web.RequestDataWith(web.NewDefaultClient(), url, "", referer, "", nil)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
avatar, err = web.GetData("http://q4.qlogo.cn/g?b=qq&nk=" + strconv.FormatInt(uid, 10) + "&s=640")
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return avatar, os.WriteFile(picFile, data, 0644)
|
||
}
|