feat: 喜报/悲报支持图片

service/xibao/image_gen.go

@@ -1,10 +1,14 @@
 package xibao
 
 import (
+	"fmt"
 	"log"
 	"strings"
 
+	"git.lxtend.com/qqbot/model"
+	"git.lxtend.com/qqbot/util"
 	"github.com/fogleman/gg"
+	"github.com/google/uuid"
 )
 
 func GenerateCongratulationImage(text string, inputFile, outputFile string, isGood bool) {
@@ -22,6 +26,68 @@ func GenerateCongratulationImage(text string, inputFile, outputFile string, isGo
 
 	// 将背景图片绘制到画布上
 	dc.DrawImage(im, 0, 0)
+	// 判断是否为图片
+	if imgUrl, ok := isImageCQ(text); ok {
+		fileName := uuid.New().String() + ".jpg"
+		filePath := "./tmp/" + fileName
+
+		if err := util.DownloadFile(imgUrl, filePath); err != nil {
+			log.Print("无法下载图片:", err)
+			return
+		}
+		imC, err := gg.LoadImage(filePath)
+		if err != nil {
+			log.Print("无法加载图片:", err)
+			return
+		}
+		widthC := imC.Bounds().Dx()
+		heightC := imC.Bounds().Dy()
+		dcC := gg.NewContext(widthC, heightC)
+		dcC.DrawImage(imC, 0, 0)
+
+		// 保存原始状态
+		dcC.Push()
+
+		scaleFactor := 1.0
+
+		// 根据高度比例进行缩放
+		if float64(heightC) > float64(height)/1.5 {
+			scaleFactor = (float64(height) / 1.5) / float64(heightC)
+		}
+
+		// 根据宽度比例进行缩放
+		if float64(widthC) > float64(width)/1.1 {
+			scaleFactor = (float64(width) / 1.1) / float64(widthC)
+		}
+
+		// 针对宽高较小的情况进行缩放
+		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))
+			} else {
+				scaleFactor = (float64(width) / 4) / float64(widthC)
+				log.Print(fmt.Sprintf("图片宽度过小，已缩放至宽度的 %f 倍", scaleFactor))
+			}
+		}
+
+		// 应用缩放比例
+		// 计算缩放后的图像尺寸
+		newWidth := int(float64(widthC) * scaleFactor)
+		newHeight := int(float64(heightC) * scaleFactor)
+
+		// 创建一个新的上下文，使用缩放后的尺寸
+		dcScaledC := gg.NewContext(newWidth, newHeight)
+
+		// 将原始图像缩放并绘制到新的上下文中
+		dcScaledC.Scale(scaleFactor, scaleFactor)
+		dcScaledC.DrawImage(imC, 0, 0)
+
+		// 在最终画布上绘制缩放后的图像
+		dc.DrawImageAnchored(dcScaledC.Image(), width/2, height/2, 0.5, 0.5)
+		dc.SavePNG(outputFile)
+		return
+	}
 
 	// 设置字体和大小，字体文件需要自备，放在合适的路径
 	if err := dc.LoadFontFace("./resource/font.ttf", 96); err != nil {
@@ -30,9 +96,9 @@ func GenerateCongratulationImage(text string, inputFile, outputFile string, isGo
 	}
 	if isGood {
 		// 设置文本颜色为红色
-		dc.SetRGB(1, 0.1, 0.1)
+		dc.SetRGB(0.8, 0, 0)
 	} else {
-		dc.SetRGB(0.1, 0.1, 1)
+		dc.SetRGB(0, 0, 0.8)
 	}
 	// 将文本按 \n 分割为多行
 	lines := strings.Split(text, "\n")
@@ -43,6 +109,7 @@ func GenerateCongratulationImage(text string, inputFile, outputFile string, isGo
 
 	// 居中绘制每一行
 	for i, line := range lines {
+		i = i - len(lines)/2
 		x := float64(width) / 2
 		y := startY + float64(i)*lineHeight
 		dc.DrawStringAnchored(line, x, y, 0.5, 0.5)
@@ -54,3 +121,10 @@ func GenerateCongratulationImage(text string, inputFile, outputFile string, isGo
 		log.Print("无法保存生成的图片:", err)
 	}
 }
+
+func isImageCQ(text string) (string, bool) {
+	if img, err := model.ParseCQImageMessage(text); err == nil {
+		return img.URL, true
+	}
+	return "", false
+}