package captcha

import (
	"sync"

	"anticaptcha/pkg/opencv"
)

type Handler struct {
	modelPath string
	mu        sync.RWMutex
}

func NewHandler(modelPath string) *Handler {
	return &Handler{
		modelPath: modelPath,
	}
}

// OCR 文字识别（需要 ONNX 模型）
func (h *Handler) OCR(imageBase64 string) (string, error) {
	// 暂时返回模拟结果
	// 实际实现需要加载 OCR 模型
	return "OCR result", nil
}

// Math 数学计算识别
func (h *Handler) Math(imageBase64 string) (string, error) {
	// 暂时返回模拟结果
	return "0", nil
}

// SliderMatch 滑块缺口匹配
func (h *Handler) SliderMatch(targetBase64, backgroundBase64 string) (int, error) {
	target, err := opencv.DecodeFromBase64(targetBase64)
	if err != nil {
		return 0, err
	}
	defer target.Free()

	background, err := opencv.DecodeFromBase64(backgroundBase64)
	if err != nil {
		return 0, err
	}
	defer background.Free()

	return opencv.SliderMatch(target, background)
}

// SliderComparison 阴影滑块匹配
func (h *Handler) SliderComparison(targetBase64, backgroundBase64 string) (int, error) {
	target, err := opencv.DecodeFromBase64(targetBase64)
	if err != nil {
		return 0, err
	}
	defer target.Free()

	background, err := opencv.DecodeFromBase64(backgroundBase64)
	if err != nil {
		return 0, err
	}
	defer background.Free()

	return opencv.SliderComparison(target, background)
}

// CompareSimilarity 图片相似度对比
func (h *Handler) CompareSimilarity(img1Base64, img2Base64 string) (float32, error) {
	img1, err := opencv.DecodeFromBase64(img1Base64)
	if err != nil {
		return 0, err
	}
	defer img1.Free()

	img2, err := opencv.DecodeFromBase64(img2Base64)
	if err != nil {
		return 0, err
	}
	defer img2.Free()

	return opencv.CompareSimilarity(img1, img2)
}

// SingleRotate 单图旋转验证码
func (h *Handler) SingleRotate(imageBase64 string) (float32, error) {
	img, err := opencv.DecodeFromBase64(imageBase64)
	if err != nil {
		return 0, err
	}
	defer img.Free()

	return opencv.DetectRotation(img)
}

// DoubleRotate 双图旋转验证码
func (h *Handler) DoubleRotate(insideBase64, outsideBase64 string) (float32, error) {
	// 简化处理
	inside, err := opencv.DecodeFromBase64(insideBase64)
	if err != nil {
		return 0, err
	}
	defer inside.Free()

	outside, err := opencv.DecodeFromBase64(outsideBase64)
	if err != nil {
		return 0, err
	}
	defer outside.Free()

	angleInside, err := opencv.DetectRotation(inside)
	if err != nil {
		return 0, err
	}

	angleOutside, err := opencv.DetectRotation(outside)
	if err != nil {
		return 0, err
	}

	return angleInside - angleOutside, nil
}

// DetectionIcon 图标检测
func (h *Handler) DetectionIcon(imageBase64 string) ([]map[string]int, error) {
	// 暂时返回空结果
	// 实际需要目标检测模型
	return []map[string]int{}, nil
}

// DetectionText 文字检测
func (h *Handler) DetectionText(imageBase64 string) ([]map[string]int, error) {
	// 暂时返回空结果
	return []map[string]int{}, nil
}

// DetectionIconOrder 按序检测图标
func (h *Handler) DetectionIconOrder(orderImgBase64, targetImgBase64 string) ([]map[string]int, error) {
	// 暂时返回空结果
	return []map[string]int{}, nil
}

// DetectionTextOrder 按序检测文字
func (h *Handler) DetectionTextOrder(orderImgBase64, targetImgBase64 string) ([]map[string]int, error) {
	// 暂时返回空结果
	return []map[string]int{}, nil
}