package utility

import (
	"crypto/rand"
	"fmt"
	"math/big"
	"strings"
	"time"
)

// GenerateUserID 根据传入的用户类型生成唯一的用户ID。
// 用户ID格式为：<PREFIX>_<17位数字字符串>，例如：QQ_20250527123456789。
// 前缀由 userType 参数指定（例如 "qq"、"wx"），自动转换为大写。
// 数字部分由当前时间戳（毫秒级）与4位加密随机数字拼接而成，确保唯一性。
//
// 参数:
//   - userType: 用户类型，如 "qq" 或 "wx"
//
// 返回值:
//   - string: 格式为 <大写前缀>_<17位唯一数字> 的用户ID
func GenerateUserID(userType string) string {
	prefix := strings.ToUpper(userType)

	// 当前时间戳（毫秒级），长度为13位
	timestamp := fmt.Sprintf("%d", time.Now().UnixNano()/1e6)

	// 生成一个0~9999之间的随机数，补齐4位
	suffixNum, _ := rand.Int(rand.Reader, big.NewInt(10000))
	suffix := fmt.Sprintf("%04d", suffixNum.Int64())

	// 拼接最终ID
	idNum := timestamp + suffix
	return fmt.Sprintf("%s_%s", prefix, idNum)
}