mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2026-01-02 08:19:04 +08:00
106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
// Package txt2img 文字转图片
|
|
package txt2img
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"image/jpeg"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/fogleman/gg"
|
|
"github.com/mattn/go-runewidth"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/FloatTech/ZeroBot-Plugin/utils/file"
|
|
"github.com/FloatTech/ZeroBot-Plugin/utils/process"
|
|
)
|
|
|
|
const (
|
|
whitespace = "\t\n\r\x0b\x0c"
|
|
// FontPath 通用字体路径
|
|
FontPath = "data/Font/"
|
|
// FontFile 苹方字体
|
|
FontFile = FontPath + "regular.ttf"
|
|
// BoldFontFile 粗体苹方字体
|
|
BoldFontFile = FontPath + "regular-bold.ttf"
|
|
)
|
|
|
|
// 加载数据库
|
|
func init() {
|
|
go func() {
|
|
process.SleepAbout1sTo2s()
|
|
_ = os.MkdirAll(FontPath, 0755)
|
|
_, _ = file.GetLazyData(FontFile, false, true)
|
|
_, _ = file.GetLazyData(BoldFontFile, false, true)
|
|
}()
|
|
}
|
|
|
|
// RenderToBase64 文字转base64
|
|
func RenderToBase64(text string, width, fontSize int) (base64Bytes []byte, err error) {
|
|
canvas, err := Render(text, width, fontSize)
|
|
if err != nil {
|
|
log.Println("[txt2img]:", err)
|
|
return nil, err
|
|
}
|
|
base64Bytes, err = CanvasToBase64(canvas)
|
|
if err != nil {
|
|
log.Println("[txt2img]:", err)
|
|
return nil, err
|
|
}
|
|
return
|
|
}
|
|
|
|
// Render 文字转图片
|
|
func Render(text string, width, fontSize int) (canvas *gg.Context, err error) {
|
|
buff := make([]string, 0)
|
|
line := ""
|
|
count := 0
|
|
for _, v := range text {
|
|
c := string(v)
|
|
if strings.Contains(whitespace, c) {
|
|
buff = append(buff, strings.TrimSpace(line))
|
|
count = 0
|
|
line = ""
|
|
continue
|
|
}
|
|
if count <= width {
|
|
line += c
|
|
count += runewidth.StringWidth(c)
|
|
} else {
|
|
buff = append(buff, line)
|
|
line = c
|
|
count = runewidth.StringWidth(c)
|
|
}
|
|
}
|
|
|
|
canvas = gg.NewContext((fontSize+4)*width/2, (len(buff)+2)*fontSize)
|
|
canvas.SetRGB(1, 1, 1)
|
|
canvas.Clear()
|
|
canvas.SetRGB(0, 0, 0)
|
|
if err = canvas.LoadFontFace(FontFile, float64(fontSize)); err != nil {
|
|
log.Println("[txt2img]:", err)
|
|
return nil, err
|
|
}
|
|
for i, v := range buff {
|
|
if v != "" {
|
|
canvas.DrawString(v, float64(width/2), float64((i+2)*fontSize))
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// CanvasToBase64 gg内容转为base64
|
|
func CanvasToBase64(canvas *gg.Context) (base64Bytes []byte, err error) {
|
|
buffer := new(bytes.Buffer)
|
|
encoder := base64.NewEncoder(base64.StdEncoding, buffer)
|
|
var opt jpeg.Options
|
|
opt.Quality = 70
|
|
if err = jpeg.Encode(encoder, canvas.Image(), &opt); err != nil {
|
|
return nil, err
|
|
}
|
|
encoder.Close()
|
|
base64Bytes = buffer.Bytes()
|
|
return
|
|
}
|