package jiguang

import (
	"crypto/hmac"
	"crypto/md5"
	"encoding/hex"
	"fmt"
	"strings"
)

// SignMethod 签名方法类型
type SignMethod string

const (
	SignMethodMD5     SignMethod = "md5"
	SignMethodHMACMD5 SignMethod = "hmac"
)

// GenerateSign 生成签名
// 根据 signMethod 参数选择使用 MD5 或 HMAC-MD5 算法
// MD5: md5(timestamp + "&appSecret=" + appSecret)，然后转大写十六进制
// HMAC-MD5: hmac_md5(timestamp, appSecret)，然后转大写十六进制
func GenerateSign(timestamp string, appSecret string, signMethod SignMethod) (string, error) {
	var hashBytes []byte

	switch signMethod {
	case SignMethodMD5:
		// MD5算法：在待签名字符串后面加上 &appSecret=xxx 再进行计算
		signStr := timestamp + "&appSecret=" + appSecret
		hash := md5.Sum([]byte(signStr))
		hashBytes = hash[:]
	case SignMethodHMACMD5:
		// HMAC-MD5算法：使用 appSecret 初始化摘要算法再进行计算
		mac := hmac.New(md5.New, []byte(appSecret))
		mac.Write([]byte(timestamp))
		hashBytes = mac.Sum(nil)
	default:
		return "", fmt.Errorf("不支持的签名方法: %s", signMethod)
	}

	// 将二进制转化为大写的十六进制（正确签名应该为32大写字符串）
	return strings.ToUpper(hex.EncodeToString(hashBytes)), nil
}

// GenerateSignWithDefault 使用默认的 HMAC-MD5 方法生成签名
func GenerateSignWithDefault(timestamp string, appSecret string) (string, error) {
	return GenerateSign(timestamp, appSecret, SignMethodHMACMD5)
}