ZeroBot-Plugin/utils/txt2img/txt2img.go

// 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
}