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)
}