添加微信小程序二维码logo替换函数

imgutils/imgutils.go

@@ -11,6 +11,7 @@ import (
 	"image/color"
 	"image/draw"
 	_ "image/gif"
+	"image/jpeg"
 	_ "image/jpeg"
 	_ "image/png"
 	"io"
@@ -276,3 +277,36 @@ func GetBase64FromUrl(url string) (string, error) {
 
 	return base64.StdEncoding.EncodeToString(data), nil
 }
+
+func ReplaceMiniQrLogo(qr image.Image, logo image.Image) ([]byte, error) {
+	logoSize := qr.Bounds().Dx() / 3
+	resizeLogo := resize.Resize(uint(logoSize), uint(logoSize), logo, resize.Bilinear)
+
+	// 计算放置图像的位置
+	posX := (qr.Bounds().Dx() - resizeLogo.Bounds().Dx()) / 2
+	posY := (qr.Bounds().Dy() - resizeLogo.Bounds().Dy()) / 2
+
+	bg := image.NewRGBA(image.Rect(0, 0, 150, 148))
+
+	// 将每个像素设置为白色
+	for x := 0; x < 150; x++ {
+		for y := 0; y < 148; y++ {
+			bg.Set(x, y, color.White)
+		}
+	}
+
+	b := qr.Bounds()
+	m := image.NewRGBA(b)
+	draw.Draw(m, b, qr, image.Point{}, draw.Src)
+	draw.Draw(m, bg.Bounds().Add(image.Point{X: 140, Y: 141}), bg, image.Point{}, draw.Src)
+	draw.Draw(m, resizeLogo.Bounds().Add(image.Point{X: posX, Y: posY}),
+		resizeLogo, image.Point{}, draw.Over)
+
+	var bs bytes.Buffer
+	err := jpeg.Encode(&bs, m, &jpeg.Options{Quality: jpeg.DefaultQuality})
+	if err != nil {
+		return nil, err
+	}
+
+	return bs.Bytes(), nil
+}