lixiangwuxian/qq_bot
0
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 喜报不要求保证同一条消息

Browse Source
This commit is contained in:
lixiangwuxian 2024-11-01 22:35:39 +08:00
parent 74cb2d19ca
commit 71b3fe8bbf
4 changed files with 71 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -35,3 +35,4 @@ config.yml
tmp/
__debug*
Dockerfile
resource/emoji.ttf

2
go.mod
View File

@ -15,6 +15,7 @@ require (
github.com/mattn/go-sqlite3 v1.14.23
github.com/sashabaranov/go-openai v1.30.3
golang.org/x/exp v0.0.0-20241009180824-f66d83c29e7c
golang.org/x/image v0.21.0
golang.org/x/net v0.30.0
gopkg.in/yaml.v3 v3.0.1
)
@ -69,7 +70,6 @@ require (
go.opentelemetry.io/otel/trace v1.31.0 // indirect
golang.org/x/arch v0.8.0 // indirect
golang.org/x/crypto v0.28.0 // indirect
golang.org/x/image v0.21.0 // indirect
golang.org/x/sys v0.26.0 // indirect
golang.org/x/text v0.19.0 // indirect
golang.org/x/time v0.7.0 // indirect

35
handler/xibao/xibao.go
View File

@ -21,13 +21,13 @@ func init() {
func xiBao(msg model.Message) (reply model.Reply) {
fileName := uuid.New().String()
filePath := "./tmp/" + fileName + ".png"
re := regexp.MustCompile(`\s+`)
tokens := re.Split(msg.RawMsg, 2)
if len(tokens) < 2 {
handler.RegisterLiveHandler(msg.GroupInfo.GroupId, msg.UserId, xiBaoTemp)
return model.Reply{
ReplyMsg: "参数不足, 请使用\"喜报 [内容]\"生成喜报图片",
ReferOriginMsg: true,
ReplyMsg: "",
ReferOriginMsg: false,
FromMsg: msg,
}
}
@ -39,15 +39,28 @@ func xiBao(msg model.Message) (reply model.Reply) {
}
}
func xiBaoTemp(msg model.Message) (reply model.Reply, isTrigger bool) {
fileName := uuid.New().String()
filePath := "./tmp/" + fileName + ".png"
xibao.GenerateCongratulationImage(msg.RawMsg, "./resource/xibao_background.png", filePath, true)
handler.UnRegisterLiveHandler(msg.GroupInfo.GroupId, msg.UserId)
return model.Reply{
ReplyMsg: fmt.Sprintf("[CQ:image,file=file:///root/qqbot/tmp/%s]", fileName+".png"),
ReferOriginMsg: true,
FromMsg: msg,
}, true
}
func beiBao(msg model.Message) (reply model.Reply) {
fileName := uuid.New().String()
filePath := "./tmp/" + fileName + ".png"
re := regexp.MustCompile(`\s+`)
tokens := re.Split(msg.RawMsg, 2)
if len(tokens) < 2 {
handler.RegisterLiveHandler(msg.GroupInfo.GroupId, msg.UserId, beiBaoTemp)
return model.Reply{
ReplyMsg: "参数不足, 请使用\"悲报 [内容]\"生成悲报图片",
ReferOriginMsg: true,
ReplyMsg: "",
ReferOriginMsg: false,
FromMsg: msg,
}
}
@ -58,3 +71,15 @@ func beiBao(msg model.Message) (reply model.Reply) {
FromMsg: msg,
}
}
func beiBaoTemp(msg model.Message) (reply model.Reply, isTrigger bool) {
fileName := uuid.New().String()
filePath := "./tmp/" + fileName + ".png"
xibao.GenerateCongratulationImage(msg.RawMsg, "./resource/beibao_background.png", filePath, false)
handler.UnRegisterLiveHandler(msg.GroupInfo.GroupId, msg.UserId)
return model.Reply{
ReplyMsg: fmt.Sprintf("[CQ:image,file=file:///root/qqbot/tmp/%s]", fileName+".png"),
ReferOriginMsg: true,
FromMsg: msg,
}, true
}

44
service/xibao/image_gen.go
View File

@ -1,14 +1,15 @@
package xibao
import (
"fmt"
"log"
"os"
"strings"
"git.lxtend.com/qqbot/model"
"git.lxtend.com/qqbot/util"
"github.com/fogleman/gg"
"github.com/google/uuid"
"golang.org/x/image/font/opentype"
)
func GenerateCongratulationImage(text string, inputFile, outputFile string, isGood bool) {
@ -51,8 +52,8 @@ func GenerateCongratulationImage(text string, inputFile, outputFile string, isGo
scaleFactor := 1.0
// 根据高度比例进行缩放
if float64(heightC) > float64(height)/1.5 {
scaleFactor = (float64(height) / 1.5) / float64(heightC)
if float64(heightC) > float64(height)/1.8 {
scaleFactor = (float64(height) / 1.8) / float64(heightC)
}
// 根据宽度比例进行缩放
@ -64,10 +65,10 @@ func GenerateCongratulationImage(text string, inputFile, outputFile string, isGo
if heightC < height/5 && widthC < width/4 {
if float64(widthC)*(float64(height)/5)/float64(heightC) > float64(width/4) {
scaleFactor = (float64(height) / 5) / float64(heightC)
log.Print(fmt.Sprintf("图片高度过小，已缩放至高度的 %f 倍", scaleFactor))
// log.Print(fmt.Sprintf("图片高度过小，已缩放至高度的 %f 倍", scaleFactor))
} else {
scaleFactor = (float64(width) / 4) / float64(widthC)
log.Print(fmt.Sprintf("图片宽度过小，已缩放至宽度的 %f 倍", scaleFactor))
// log.Print(fmt.Sprintf("图片宽度过小，已缩放至宽度的 %f 倍", scaleFactor))
}
}
@ -90,10 +91,43 @@ func GenerateCongratulationImage(text string, inputFile, outputFile string, isGo
}
// 设置字体和大小，字体文件需要自备，放在合适的路径
// if err := dc.LoadFontFace("./resource/emoji.ttf", 96); err != nil {
// log.Print("无法加载字体:", err)
// return
// }
fontBytes, err := os.ReadFile("./resource/emoji.ttf")
if err != nil {
log.Print("failed to load font: %v", err)
}
// 解析字体
font, err := opentype.Parse(fontBytes)
if err != nil {
log.Print("failed to parse font: %v", err)
}
// 设置字体大小和 DPI
const fontSize = 48
const dpi = 72
// 创建 font.Face，用于渲染文本
face, err := opentype.NewFace(font, &opentype.FaceOptions{
Size: fontSize,
DPI: dpi,
Hinting: 0,
})
if err != nil {
log.Print("failed to create font face: %v", err)
return
}
dc.SetFontFace(face)
if err := dc.LoadFontFace("./resource/font.ttf", 96); err != nil {
log.Print("无法加载字体:", err)
return
}
if isGood {
// 设置文本颜色为红色
dc.SetRGB(0.8, 0, 0)