142 lines
3.9 KiB
Go
142 lines
3.9 KiB
Go
package image
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"syscall"
|
||
"unsafe"
|
||
)
|
||
|
||
var (
|
||
gImageDll *ImageDLL
|
||
|
||
// Windows API - 使用 C 运行时库
|
||
libc = syscall.NewLazyDLL("msvcrt.dll")
|
||
procFree = libc.NewProc("free")
|
||
procMalloc = libc.NewProc("malloc")
|
||
)
|
||
|
||
// ImageDLL 图片工具DLL结构
|
||
type ImageDLL struct {
|
||
Dll *syscall.DLL
|
||
AddWatermarkFromURLEx *syscall.Proc // 打水印
|
||
}
|
||
|
||
// InitImageDll 初始化 imageDLL
|
||
func InitImageDll(url string) (*ImageDLL, error) {
|
||
dllPath := filepath.Join(url, "image.dll")
|
||
if _, err := os.Stat(dllPath); os.IsNotExist(err) {
|
||
return nil, fmt.Errorf("Image DLL 不存在: %s", dllPath)
|
||
}
|
||
dll, err := syscall.LoadDLL(dllPath)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("加载Image DLL 失败: %s", err)
|
||
}
|
||
gImageDll = &ImageDLL{
|
||
Dll: dll,
|
||
AddWatermarkFromURLEx: dll.MustFindProc("AddWatermarkFromURLEx"),
|
||
}
|
||
return gImageDll, nil
|
||
}
|
||
|
||
// WatermarkConfig 添加水印
|
||
type WatermarkConfig struct {
|
||
SourceImageURL string // 源图片URL地址
|
||
WatermarkURL string // 水印图片URL地址
|
||
WatermarkBase64 string // 水印图片base64编码字符串（新增，优先使用）
|
||
Opacity float64 // 不透明度 (0.0-1.0)
|
||
Position string // 位置: center, top-left, top-right, bottom-left, bottom-right, tile
|
||
TileSpacing int // 平铺时的间距
|
||
Scale float64 // 水印缩放比例 (0.0-1.0)
|
||
Rotation float64 // 旋转角度 (度数)
|
||
XOffset int // X轴偏移量
|
||
YOffset int // Y轴偏移量
|
||
Timeout int // 下载超时时间（秒），默认30秒
|
||
OutputFormat string // 输出格式: "jpeg", "png", "auto"（默认auto，根据源图片格式）auto
|
||
JPEGQuality int // JPEG质量 (1-100)，默认95
|
||
TargetWidth int // 目标宽度（0表示不缩放）
|
||
TargetHeight int // 目标高度（0表示不缩放）
|
||
ResizeMode string // 缩放模式: "fit"（适应，保持比例，可能有黑边）, "fill"（填充，裁剪）, "stretch"（拉伸）
|
||
}
|
||
|
||
// AddWatermarkFromURLExs 添加水印
|
||
func (m *ImageDLL) AddWatermarkFromURLExs(sourceImageUrl, watermarkUrl string) (string, error) {
|
||
watermarkConfig := WatermarkConfig{
|
||
SourceImageURL: sourceImageUrl,
|
||
WatermarkBase64: watermarkUrl,
|
||
Position: "center",
|
||
Opacity: 1.0,
|
||
Scale: 1.0,
|
||
TileSpacing: 50,
|
||
Timeout: 30,
|
||
OutputFormat: "jpeg",
|
||
JPEGQuality: 95,
|
||
TargetWidth: 800,
|
||
TargetHeight: 800,
|
||
ResizeMode: "fit",
|
||
}
|
||
watermarkConfigJson, err := json.Marshal(watermarkConfig)
|
||
if err != nil {
|
||
return "", fmt.Errorf("JSON序列化失败: %v", err)
|
||
}
|
||
|
||
proc, err := m.Dll.FindProc("AddWatermarkFromURLEx")
|
||
if err != nil {
|
||
return "", fmt.Errorf("找不到函数 AddWatermarkFromURLEx: %v", err)
|
||
}
|
||
|
||
// 分配内存并确保释放
|
||
jsonStr := string(watermarkConfigJson)
|
||
jsonPtr := cString(jsonStr)
|
||
defer freeCString(jsonPtr)
|
||
|
||
// 调用 DLL 函数
|
||
resultPtr, _, _ := proc.Call(
|
||
uintptr(unsafe.Pointer(jsonPtr)),
|
||
)
|
||
result := cStr(resultPtr)
|
||
return result, nil
|
||
}
|
||
|
||
// cString 分配 C 字符串（使用 malloc）
|
||
func cString(str string) unsafe.Pointer {
|
||
// 计算需要的内存大小
|
||
size := len(str) + 1
|
||
ptr, _, _ := procMalloc.Call(uintptr(size))
|
||
if ptr == 0 {
|
||
return nil
|
||
}
|
||
// 复制字符串内容
|
||
for i := 0; i < len(str); i++ {
|
||
*(*byte)(unsafe.Pointer(ptr + uintptr(i))) = str[i]
|
||
}
|
||
*(*byte)(unsafe.Pointer(ptr + uintptr(len(str)))) = 0 // 结尾加 \0
|
||
return unsafe.Pointer(ptr)
|
||
}
|
||
|
||
// freeCString 释放 C 字符串
|
||
func freeCString(ptr unsafe.Pointer) {
|
||
if ptr != nil {
|
||
procFree.Call(uintptr(ptr))
|
||
}
|
||
}
|
||
|
||
// cStr 将 C 字符串指针转换为 Go 字符串
|
||
func cStr(ptr uintptr) string {
|
||
if ptr == 0 {
|
||
return ""
|
||
}
|
||
var b []byte
|
||
for {
|
||
c := *(*byte)(unsafe.Pointer(ptr))
|
||
if c == 0 {
|
||
break
|
||
}
|
||
b = append(b, c)
|
||
ptr++
|
||
}
|
||
return string(b)
|
||
}
|