golang 添加文字水印并旋转

注意：如水印文字是中文，需要中文字体
字体下载链接：https://fonts.google.com

package imgUtils

import (
	"golang.org/x/image/font"
	"golang.org/x/image/math/fixed"
	"math"

	"image"
	"image/color"
	"image/draw"
	"io/ioutil"

	"github.com/golang/freetype/truetype"
	"github.com/disintegration/imaging"
)

/**
	添加文本水印到图片
	img：图片数据
	watermarkText：水印内容
	fontPath：字体路径
	angle：旋转角度
 */
func AddWatermark(img image.Image, watermarkText string, fontPath string, angle float64) (image.Image, error) {
	// 加载字体
	fontBytes, err := ioutil.ReadFile(fontPath)
	if err != nil {
		return nil, err
	}
	f, err := truetype.Parse(fontBytes)
	if err != nil {
		return nil, err
	}

	// 创建面部字体
	face := truetype.NewFace(f, &truetype.Options{
		Size:    24,
		DPI:     72,
		Hinting: font.HintingNone,
	})

	// 计算文本大小（调整密度）
	txtSize := int(math.Ceil(45.0 * 90.0 / 96.0))

	// 创建水印图像
	bounds := img.Bounds()
	wm := image.NewRGBA(bounds)
	draw.Draw(wm, bounds, img, bounds.Min, draw.Src)

	// 设置字体颜色
	col := color.RGBA{R: 0, G: 0, B: 0, A: 20}

	// 循环绘制水印
	for y := txtSize; y < bounds.Dy(); y += txtSize * 4 {
		for x := txtSize; x < bounds.Dx(); x += txtSize * 4 {
			// 创建临时图像
			temp := image.NewRGBA(image.Rect(0, 0, txtSize*4, txtSize*4))
			d := &font.Drawer{
				Dst:  temp,
				Src:  image.NewUniform(col),
				Face: face,
				Dot:  fixed.P(0, txtSize),
			}
			d.DrawString(watermarkText)

			// 旋转临时图像
			rotated := imaging.Rotate(temp, angle, color.NRGBA{0, 0, 0, 0})

			// 绘制旋转后的图像到原始图像
			draw.Draw(wm, image.Rect(x, y, x+rotated.Bounds().Dx(), y+rotated.Bounds().Dy()), rotated, image.Point{}, draw.Over)
		}
	}

	return wm, nil
}