package fqapi

import (
	"crypto/md5"
	"encoding/hex"
	"fanqie/internal/algos"
	"fanqie/internal/common"
)

// CreateMD5 创建MD5哈希
func CreateMD5(s string) string {
	hasher := md5.New()
	hasher.Write([]byte(s))
	return hex.EncodeToString(hasher.Sum(nil))
}

// setCommonHeaders 注入番茄客户端通用请求头（与签名头不冲突，对齐书源实现降低风控特征）
func setCommonHeaders(headers map[string]string) {
	if _, ok := headers["x-tt-store-region"]; ok {
		return
	}
	headers["x-tt-store-region"] = "cn-zj"
	headers["x-tt-store-region-src"] = "did"
	headers["sdk-version"] = "2"
}

// GorgonRequest Gorgon请求（使用签名切换器）
func GorgonRequest(urlStr string, options map[string]interface{}) (*common.ClientResponse, error) {
	headers, ok := options["headers"].(map[string]string)
	if !ok {
		headers = make(map[string]string)
		options["headers"] = headers
	}
	setCommonHeaders(headers)

	dataMD5 := ""
	if body, ok := options["body"].(string); ok && body != "" {
		dataMD5 = CreateMD5(body)
	} else if body, ok := options["body"].([]byte); ok && len(body) > 0 {
		dataMD5 = CreateMD5(string(body))
	}

	if dataMD5 != "" {
		headers["x-ss-stub"] = dataMD5
	}

	// 使用签名切换器（sign2 或 sign4）生成签名
	signedHeaders := algos.GlobalSignatureSwitcher.Sign(urlStr, headers)

	// 更新headers
	for k, v := range signedHeaders {
		headers[k] = v
	}

	// 调用实际的HTTP请求
	client := common.NewHTTPClient()
	return client.Request(urlStr, options)
}

// GorgonRequestSync Gorgon同步请求（使用签名切换器）
func GorgonRequestSync(urlStr string, options map[string]interface{}) (*common.ClientResponse, error) {
	headers, ok := options["headers"].(map[string]string)
	if !ok {
		headers = make(map[string]string)
		options["headers"] = headers
	}
	setCommonHeaders(headers)

	dataMD5 := ""
	if body, ok := options["body"].(string); ok && body != "" {
		dataMD5 = CreateMD5(body)
	} else if body, ok := options["body"].([]byte); ok && len(body) > 0 {
		dataMD5 = CreateMD5(string(body))
	}

	if dataMD5 != "" {
		headers["x-ss-stub"] = dataMD5
	}

	// 使用签名切换器（sign2 或 sign4）生成签名
	signedHeaders := algos.GlobalSignatureSwitcher.Sign(urlStr, headers)

	// 删除x-ss-stub（如果存在）
	delete(signedHeaders, "x-ss-stub")

	// 更新headers
	for k, v := range signedHeaders {
		headers[k] = v
	}

	// 调用实际的HTTP请求
	client := common.NewHTTPClient()
	return client.Request(urlStr, options)
}

// ContentRequest Content请求（硬编码使用Sign2，不受配置影响）
func ContentRequest(urlStr string, options map[string]interface{}) (*common.ClientResponse, error) {
	headers, ok := options["headers"].(map[string]string)
	if !ok {
		headers = make(map[string]string)
		options["headers"] = headers
	}
	setCommonHeaders(headers)

	dataMD5 := ""
	if body, ok := options["body"].(string); ok && body != "" {
		dataMD5 = CreateMD5(body)
	} else if body, ok := options["body"].([]byte); ok && len(body) > 0 {
		dataMD5 = CreateMD5(string(body))
	}

	if dataMD5 != "" {
		headers["x-ss-stub"] = dataMD5
	}

	// 硬编码使用 Sign2（不受配置影响）
	signedHeaders := algos.Sign2Wrapper(urlStr, headers)

	// 更新headers
	for k, v := range signedHeaders {
		headers[k] = v
	}

	// 调用实际的HTTP请求
	client := common.NewHTTPClient()
	return client.Request(urlStr, options)
}