94 lines
1.9 KiB
Go
94 lines
1.9 KiB
Go
package utils
|
||
|
||
import (
|
||
"math/rand"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/pquerna/otp/totp"
|
||
)
|
||
|
||
type RandDateParams struct {
|
||
Year int
|
||
Month int
|
||
Day int
|
||
}
|
||
|
||
const letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||
const (
|
||
letterBits = 6 // 52个字母需6位（2^6=64）
|
||
letterMask = 1<<letterBits - 1 // 二进制掩码（63个1）
|
||
)
|
||
|
||
func BoolToInt(b bool) int64 {
|
||
if b {
|
||
return 1
|
||
}
|
||
return 0
|
||
}
|
||
|
||
func RandString(n int) string {
|
||
b := make([]byte, n)
|
||
for i := 0; i < n; {
|
||
if idx := int(rand.Int63() & letterMask); idx < len(letters) {
|
||
b[i] = letters[idx]
|
||
i++
|
||
} // 丢弃无效索引，重试
|
||
}
|
||
return string(b)
|
||
}
|
||
|
||
func RandDate(minAge, maxAge int) RandDateParams {
|
||
// 创建本地随机生成器（避免全局 rand.Seed 弃用问题）
|
||
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
||
|
||
// 计算年份范围（当前年-100岁 到 当前年-18岁）
|
||
currentTime := time.Now()
|
||
currentYear := currentTime.Year()
|
||
minYear := currentYear - maxAge
|
||
maxYear := currentYear - minAge
|
||
|
||
// 生成随机年份
|
||
year := minYear + r.Intn(maxYear-minYear+1)
|
||
|
||
// 生成随机月份（1-12）
|
||
month := r.Intn(12) + 1
|
||
|
||
// 动态计算当月最大天数（考虑闰年）
|
||
daysInMonth := 31
|
||
switch month {
|
||
case 2: // 二月特殊处理
|
||
if (year%4 == 0 && year%100 != 0) || year%400 == 0 {
|
||
daysInMonth = 29 // 闰年
|
||
} else {
|
||
daysInMonth = 28 // 平年
|
||
}
|
||
case 4, 6, 9, 11: // 30 天的月份
|
||
daysInMonth = 30
|
||
}
|
||
|
||
// 生成随机日期（1 ~ daysInMonth）
|
||
day := r.Intn(daysInMonth) + 1
|
||
|
||
return RandDateParams{
|
||
Year: year,
|
||
Month: month,
|
||
Day: day,
|
||
}
|
||
}
|
||
|
||
// 获取FA2码
|
||
func GetFA2Code(secret string) (string, error) {
|
||
// 去除多余的字符串
|
||
replacer := strings.NewReplacer(
|
||
"-", "",
|
||
" ", "",
|
||
)
|
||
result := replacer.Replace(secret)
|
||
code, err := totp.GenerateCode(result, time.Now())
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
return code, nil
|
||
}
|