chore: 初始化 Cloudreve 源码仓库
This commit is contained in:
@@ -0,0 +1,360 @@
|
||||
package util
|
||||
|
||||
import (
|
||||
"context"
|
||||
cryptoRand "crypto/rand"
|
||||
"fmt"
|
||||
"math/big"
|
||||
"math/rand"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
"unicode/utf8"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/gofrs/uuid"
|
||||
)
|
||||
|
||||
func init() {
|
||||
rand.Seed(time.Now().UnixNano())
|
||||
}
|
||||
|
||||
var (
|
||||
RandomVariantAll = []rune("1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
|
||||
RandomLowerCases = []rune("1234567890abcdefghijklmnopqrstuvwxyz")
|
||||
)
|
||||
|
||||
// RandStringRunes 返回随机字符串
|
||||
func RandStringRunes(n int) string {
|
||||
b := make([]rune, n)
|
||||
for i := range b {
|
||||
b[i] = RandomVariantAll[rand.Intn(len(RandomVariantAll))]
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
|
||||
func RandStringRunesCrypto(n int) string {
|
||||
b := make([]rune, n)
|
||||
for i := range b {
|
||||
num, err := cryptoRand.Int(cryptoRand.Reader, big.NewInt(int64(len(RandomVariantAll))))
|
||||
if err != nil {
|
||||
// fallback to math/rand on crypto failure
|
||||
b[i] = RandomVariantAll[rand.Intn(len(RandomVariantAll))]
|
||||
} else {
|
||||
b[i] = RandomVariantAll[num.Int64()]
|
||||
}
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
|
||||
// RandString returns random string in given length and variant
|
||||
func RandString(n int, variant []rune) string {
|
||||
b := make([]rune, n)
|
||||
for i := range b {
|
||||
b[i] = variant[rand.Intn(len(variant))]
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
|
||||
// ContainsUint 返回list中是否包含
|
||||
func ContainsUint(s []uint, e uint) bool {
|
||||
for _, a := range s {
|
||||
if a == e {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// IsInExtensionList 返回文件的扩展名是否在给定的列表范围内
|
||||
func IsInExtensionList(extList []string, fileName string) bool {
|
||||
ext := Ext(fileName)
|
||||
// 无扩展名时
|
||||
if len(ext) == 0 {
|
||||
return false
|
||||
}
|
||||
|
||||
if ContainsString(extList, ext) {
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// IsInExtensionList 返回文件的扩展名是否在给定的列表范围内
|
||||
func IsInExtensionListExt(extList []string, ext string) bool {
|
||||
// 无扩展名时
|
||||
if len(ext) == 0 {
|
||||
return false
|
||||
}
|
||||
|
||||
if ContainsString(extList, ext) {
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// ContainsString 返回list中是否包含
|
||||
func ContainsString(s []string, e string) bool {
|
||||
for _, a := range s {
|
||||
if a == e {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// Replace 根据替换表执行批量替换
|
||||
func Replace(table map[string]string, s string) string {
|
||||
for key, value := range table {
|
||||
s = strings.Replace(s, key, value, -1)
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
// ReplaceMagicVar 动态替换字符串中的魔法变量
|
||||
func ReplaceMagicVar(rawString string, fsSeparator string, pathAvailable bool, blobAvailable bool,
|
||||
timeConst time.Time, userId int, originName string, originPath string, completeBlobPath string) string {
|
||||
re := regexp.MustCompile(`\{[^{}]+\}`)
|
||||
return re.ReplaceAllStringFunc(rawString, func(match string) string {
|
||||
switch match {
|
||||
case "{randomkey16}":
|
||||
return RandStringRunes(16)
|
||||
case "{randomkey8}":
|
||||
return RandStringRunes(8)
|
||||
case "{timestamp}":
|
||||
return strconv.FormatInt(timeConst.Unix(), 10)
|
||||
case "{timestamp_nano}":
|
||||
return strconv.FormatInt(timeConst.UnixNano(), 10)
|
||||
case "{randomnum2}":
|
||||
return strconv.Itoa(rand.Intn(2))
|
||||
case "{randomnum3}":
|
||||
return strconv.Itoa(rand.Intn(3))
|
||||
case "{randomnum4}":
|
||||
return strconv.Itoa(rand.Intn(4))
|
||||
case "{randomnum8}":
|
||||
return strconv.Itoa(rand.Intn(8))
|
||||
case "{uid}":
|
||||
return strconv.Itoa(userId)
|
||||
case "{datetime}":
|
||||
return timeConst.Format("20060102150405")
|
||||
case "{date}":
|
||||
return timeConst.Format("20060102")
|
||||
case "{year}":
|
||||
return timeConst.Format("2006")
|
||||
case "{month}":
|
||||
return timeConst.Format("01")
|
||||
case "{day}":
|
||||
return timeConst.Format("02")
|
||||
case "{hour}":
|
||||
return timeConst.Format("15")
|
||||
case "{minute}":
|
||||
return timeConst.Format("04")
|
||||
case "{second}":
|
||||
return timeConst.Format("05")
|
||||
case "{uuid}":
|
||||
return uuid.Must(uuid.NewV4()).String()
|
||||
case "{ext}":
|
||||
return filepath.Ext(originName)
|
||||
case "{originname}":
|
||||
return originName
|
||||
case "{originname_without_ext}":
|
||||
return strings.TrimSuffix(originName, filepath.Ext(originName))
|
||||
case "{path}":
|
||||
if pathAvailable {
|
||||
return originPath + fsSeparator
|
||||
}
|
||||
return match
|
||||
case "{blob_name}":
|
||||
if blobAvailable {
|
||||
return filepath.Base(completeBlobPath)
|
||||
}
|
||||
return match
|
||||
case "{blob_name_without_ext}":
|
||||
if blobAvailable {
|
||||
return strings.TrimSuffix(filepath.Base(completeBlobPath), filepath.Ext(completeBlobPath))
|
||||
}
|
||||
return match
|
||||
case "{blob_path}":
|
||||
if blobAvailable {
|
||||
return path.Dir(completeBlobPath) + fsSeparator
|
||||
}
|
||||
return match
|
||||
default:
|
||||
return match
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// BuildRegexp 构建用于SQL查询用的多条件正则
|
||||
func BuildRegexp(search []string, prefix, suffix, condition string) string {
|
||||
var res string
|
||||
for key, value := range search {
|
||||
res += prefix + regexp.QuoteMeta(value) + suffix
|
||||
if key < len(search)-1 {
|
||||
res += condition
|
||||
}
|
||||
}
|
||||
return res
|
||||
}
|
||||
|
||||
// BuildConcat 根据数据库类型构建字符串连接表达式
|
||||
func BuildConcat(str1, str2 string, DBType string) string {
|
||||
switch DBType {
|
||||
case "mysql":
|
||||
return "CONCAT(" + str1 + "," + str2 + ")"
|
||||
default:
|
||||
return str1 + "||" + str2
|
||||
}
|
||||
}
|
||||
|
||||
// SliceIntersect 求两个切片交集
|
||||
func SliceIntersect(slice1, slice2 []string) []string {
|
||||
m := make(map[string]int)
|
||||
nn := make([]string, 0)
|
||||
for _, v := range slice1 {
|
||||
m[v]++
|
||||
}
|
||||
|
||||
for _, v := range slice2 {
|
||||
times, _ := m[v]
|
||||
if times == 1 {
|
||||
nn = append(nn, v)
|
||||
}
|
||||
}
|
||||
return nn
|
||||
}
|
||||
|
||||
// SliceDifference 求两个切片差集
|
||||
func SliceDifference(slice1, slice2 []string) []string {
|
||||
m := make(map[string]int)
|
||||
nn := make([]string, 0)
|
||||
inter := SliceIntersect(slice1, slice2)
|
||||
for _, v := range inter {
|
||||
m[v]++
|
||||
}
|
||||
|
||||
for _, value := range slice1 {
|
||||
times, _ := m[value]
|
||||
if times == 0 {
|
||||
nn = append(nn, value)
|
||||
}
|
||||
}
|
||||
return nn
|
||||
}
|
||||
|
||||
// WithValue inject key-value pair into request context.
|
||||
func WithValue(c *gin.Context, key any, value any) {
|
||||
c.Request = c.Request.WithContext(context.WithValue(c.Request.Context(), key, value))
|
||||
}
|
||||
|
||||
// BoolToString transform bool to string
|
||||
func BoolToString(b bool) string {
|
||||
if b {
|
||||
return "1"
|
||||
}
|
||||
return "0"
|
||||
}
|
||||
|
||||
func EncodeTimeFlowString(str string, timeNow int64) string {
|
||||
timeNow = timeNow / 1000
|
||||
timeDigits := []int{}
|
||||
timeDigitIndex := 0
|
||||
|
||||
if len(str) == 0 {
|
||||
return ""
|
||||
}
|
||||
|
||||
str = fmt.Sprintf("%d|%s", timeNow, str)
|
||||
|
||||
res := make([]int32, 0, utf8.RuneCountInString(str))
|
||||
for timeNow > 0 {
|
||||
timeDigits = append(timeDigits, int(timeNow%int64(10)))
|
||||
timeNow = timeNow / 10
|
||||
}
|
||||
|
||||
add := false
|
||||
for pos, rune := range str {
|
||||
// take single digit with index timeDigitIndex from timeNow
|
||||
newIndex := pos
|
||||
if add {
|
||||
newIndex = pos + timeDigits[timeDigitIndex]*timeDigitIndex
|
||||
} else {
|
||||
newIndex = 2*timeDigitIndex*timeDigits[timeDigitIndex] - pos
|
||||
}
|
||||
|
||||
if newIndex < 0 {
|
||||
newIndex = newIndex * -1
|
||||
}
|
||||
|
||||
res = append(res, rune)
|
||||
newIndex = newIndex % len(res)
|
||||
|
||||
res[newIndex], res[len(res)-1] = res[len(res)-1], res[newIndex]
|
||||
|
||||
add = !add
|
||||
// Add timeDigitIndex by 1, but does not exceed total digits in timeNow
|
||||
timeDigitIndex++
|
||||
timeDigitIndex = timeDigitIndex % len(timeDigits)
|
||||
}
|
||||
|
||||
return string(res)
|
||||
}
|
||||
|
||||
func DecodeTimeFlowStringTime(str string, timeNow int64) string {
|
||||
timeNow = timeNow / 1000
|
||||
timeDigits := []int{}
|
||||
|
||||
if len(str) == 0 {
|
||||
return ""
|
||||
}
|
||||
|
||||
for timeNow > 0 {
|
||||
timeDigits = append(timeDigits, int(timeNow%int64(10)))
|
||||
timeNow = timeNow / 10
|
||||
}
|
||||
|
||||
res := make([]int32, utf8.RuneCountInString(str))
|
||||
secret := []rune(str)
|
||||
add := false
|
||||
if len(secret)%2 == 0 {
|
||||
add = true
|
||||
}
|
||||
timeDigitIndex := (len(secret) - 1) % len(timeDigits)
|
||||
for pos := range secret {
|
||||
// take single digit with index timeDigitIndex from timeNow
|
||||
newIndex := len(res) - 1 - pos
|
||||
if add {
|
||||
newIndex = newIndex + timeDigits[timeDigitIndex]*timeDigitIndex
|
||||
} else {
|
||||
newIndex = 2*timeDigitIndex*timeDigits[timeDigitIndex] - newIndex
|
||||
}
|
||||
|
||||
if newIndex < 0 {
|
||||
newIndex = newIndex * -1
|
||||
}
|
||||
|
||||
newIndex = newIndex % len(secret)
|
||||
|
||||
res[len(res)-1-pos] = secret[newIndex]
|
||||
secret[newIndex], secret[len(res)-1-pos] = secret[len(res)-1-pos], secret[newIndex]
|
||||
secret = secret[:len(secret)-1]
|
||||
|
||||
add = !add
|
||||
// Add timeDigitIndex by 1, but does not exceed total digits in timeNow
|
||||
timeDigitIndex--
|
||||
if timeDigitIndex < 0 {
|
||||
timeDigitIndex = len(timeDigits) - 1
|
||||
}
|
||||
}
|
||||
|
||||
return string(res)
|
||||
}
|
||||
|
||||
func ToPtr[T any](v T) *T {
|
||||
return &v
|
||||
}
|
||||
Reference in New Issue
Block a user