108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package image
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"planA/planB/config"
|
||
"syscall"
|
||
"unsafe"
|
||
)
|
||
|
||
var (
|
||
gImageDll *ImageDLL
|
||
)
|
||
|
||
// ImageDLL 图片工具DLL结构
|
||
type ImageDLL struct {
|
||
Dll *syscall.DLL
|
||
AddWatermarkFromURLEx *syscall.Proc // 打水印
|
||
}
|
||
|
||
// InitImageDll 初始化 imageDLL
|
||
func InitImageDll() (*ImageDLL, error) {
|
||
fileConfig, getDllFileConfigErr := config.GetFileUrlConfig()
|
||
if getDllFileConfigErr != nil {
|
||
return nil, getDllFileConfigErr
|
||
}
|
||
dllPath := filepath.Join(fileConfig.ImageDll, "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地址
|
||
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
|
||
}
|
||
|
||
// AddWatermarkFromURLExs 添加水印
|
||
func (m *ImageDLL) AddWatermarkFromURLExs(sourceImageUrl, watermarkUrl string) (string, error) {
|
||
|
||
watermarkConfig := WatermarkConfig{
|
||
SourceImageURL: sourceImageUrl,
|
||
WatermarkURL: watermarkUrl,
|
||
Position: "center",
|
||
Opacity: 1.0,
|
||
Scale: 1.0,
|
||
TileSpacing: 50,
|
||
Timeout: 30,
|
||
OutputFormat: "jpeg",
|
||
JPEGQuality: 95,
|
||
}
|
||
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)
|
||
}
|
||
watermarkConfigJsonPtr, _ := syscall.BytePtrFromString(string(watermarkConfigJson))
|
||
|
||
resultPtr, _, _ := proc.Call(
|
||
uintptr(unsafe.Pointer(watermarkConfigJsonPtr)),
|
||
)
|
||
result := cStr(resultPtr)
|
||
return result, nil
|
||
}
|
||
|
||
// 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)
|
||
}
|