增加文字转图片&拟声鸟 (#96)

* feat:添加文字转图片,改造长文字 * fix:修改log * fix:修改条件 * fix:不保存图片 * fix:增加block和优先级 * fix:文件夹首字母大写 * fix:修改解签为图片 * feat:添加拟声鸟 * fix:清理缓存 * fix:换一个音频 * fix:小修格式 * fix:修一下lint * fix:修一下lint * fix:修一下lint * fix:修一下lint * fix:修一下lint * fix:修一下lint * fix:修一下lint * fix:修一下lint * fix:10s一次 * fix:10s一次 * fix:修lint
2026-02-12 18:20:27 +00:00 · 2022-01-03 00:26:00 +08:00
parent a69d985139
commit 045ee091a7
21 changed files with 497 additions and 193 deletions
--- a/utils/txt2img/txt2img.go
+++ b/utils/txt2img/txt2img.go
@@ -0,0 +1,89 @@
+// Package txt2img 文字转图片
+package txt2img
+
+import (
+	"bytes"
+	"encoding/base64"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/file"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/process"
+	"github.com/fogleman/gg"
+	"github.com/mattn/go-runewidth"
+	log "github.com/sirupsen/logrus"
+	"image/jpeg"
+	"os"
+	"strings"
+)
+
+const (
+	whitespace = "\t\n\r\x0b\x0c"
+	fontpath   = "data/Font/"
+	fontfile   = fontpath + "regular.ttf"
+)
+
+// 加载数据库
+func init() {
+	go func() {
+		process.SleepAbout1sTo2s()
+		_ = os.MkdirAll(fontpath, 0755)
+		_, _ = file.GetLazyData(fontfile, 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("err:", err)
+		return nil, err
+	}
+	// 转成 base64
+	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
+}
+
+// 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+3)*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("err:", err)
+		return nil, err
+	}
+	for i, v := range buff {
+		if v != "" {
+			canvas.DrawString(v, float64(width/2), float64((i+2)*fontSize))
+		}
+	}
+	return
+}