package utils

import (
	"crypto/rand"
	"encoding/hex"
	"strings"

	"github.com/google/uuid"
	"go.uber.org/zap"
)

// FormatUUID 将UUID格式化为带连字符的标准格式
// 如果输入已经是标准格式，直接返回
// 如果输入是32位十六进制字符串，添加连字符
// 如果输入格式无效，返回错误
func FormatUUID(uuid string) string {
	// 如果为空，直接返回
	if uuid == "" {
		return uuid
	}

	// 如果已经是标准格式（8-4-4-4-12），直接返回
	if len(uuid) == 36 && uuid[8] == '-' && uuid[13] == '-' && uuid[18] == '-' && uuid[23] == '-' {
		return uuid
	}

	// 如果是32位十六进制字符串，添加连字符
	if len(uuid) == 32 {
		// 预分配容量以提高性能
		var b strings.Builder
		b.Grow(36) // 最终长度为36(32个字符 + 4个连字符)

		// 使用WriteString和WriteByte优化性能
		b.WriteString(uuid[0:8])
		b.WriteByte('-')
		b.WriteString(uuid[8:12])
		b.WriteByte('-')
		b.WriteString(uuid[12:16])
		b.WriteByte('-')
		b.WriteString(uuid[16:20])
		b.WriteByte('-')
		b.WriteString(uuid[20:32])

		return b.String()
	}

	// 如果长度不是32或36，说明格式无效，直接返回原值
	var logger *zap.Logger
	logger.Warn("[WARN] UUID格式无效: ", zap.String("uuid:", uuid))
	return uuid
}

// GenerateUUID 生成无符号UUID（32位十六进制字符串，不带连字符）
// 使用github.com/google/uuid库生成标准UUID，然后移除连字符
// 返回格式示例: "123e4567e89b12d3a456426614174000"
func GenerateUUID() string {
	return strings.ReplaceAll(uuid.New().String(), "-", "")
}

// FormatUUIDToNoDash 将带连字符的UUID转换为无符号格式（移除连字符）
// 输入: "123e4567-e89b-12d3-a456-426614174000"
// 输出: "123e4567e89b12d3a456426614174000"
// 如果输入已经是32位格式，直接返回
// 如果输入格式无效，返回原值并记录警告
func FormatUUIDToNoDash(uuid string) string {
	// 如果为空，直接返回
	if uuid == "" {
		return uuid
	}

	// 如果已经是32位格式（无连字符），直接返回
	if len(uuid) == 32 {
		return uuid
	}

	// 如果是36位标准格式，移除连字符
	if len(uuid) == 36 && uuid[8] == '-' && uuid[13] == '-' && uuid[18] == '-' && uuid[23] == '-' {
		return strings.ReplaceAll(uuid, "-", "")
	}

	// 如果格式无效，记录警告并返回原值
	var logger *zap.Logger
	logger.Warn("[WARN] UUID格式无效，无法转换为无符号格式: ", zap.String("uuid:", uuid))
	return uuid
}

// RandomHex 生成指定长度的随机十六进制字符串
// 参数 n: 需要生成的十六进制字符数量（每个字节生成2个十六进制字符）
// 返回: 长度为 2*n 的十六进制字符串
// 示例: RandomHex(16) 返回 "a1b2c3d4e5f67890abcdef1234567890" (32字符)
func RandomHex(n uint) (string, error) {
	bytes := make([]byte, n)
	if _, err := rand.Read(bytes); err != nil {
		return "", err
	}
	return hex.EncodeToString(bytes), nil
}