mirror of https://github.com/mickael-menu/zk
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1019 B
Go
44 lines
1019 B
Go
package rand
|
|
|
|
import (
|
|
"math/rand"
|
|
"time"
|
|
"unicode"
|
|
|
|
"github.com/zk-org/zk/internal/core"
|
|
)
|
|
|
|
// NewIDGenerator returns a function generating string IDs using the given options.
|
|
// Inspired by https://www.calhoun.io/creating-random-strings-in-go/
|
|
func NewIDGenerator(options core.IDOptions) func() string {
|
|
if options.Length < 1 {
|
|
panic("IDOptions.Length must be at least 1")
|
|
}
|
|
|
|
var charset []rune
|
|
for _, char := range options.Charset {
|
|
switch options.Case {
|
|
case core.CaseLower:
|
|
charset = append(charset, unicode.ToLower(char))
|
|
case core.CaseUpper:
|
|
charset = append(charset, unicode.ToUpper(char))
|
|
case core.CaseMixed:
|
|
charset = append(charset, unicode.ToLower(char))
|
|
charset = append(charset, unicode.ToUpper(char))
|
|
default:
|
|
panic("unknown zk.Case value")
|
|
}
|
|
}
|
|
|
|
rand := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
|
|
return func() string {
|
|
buf := make([]rune, options.Length)
|
|
for i := range buf {
|
|
buf[i] = charset[rand.Intn(len(charset))]
|
|
}
|
|
|
|
return string(buf)
|
|
}
|
|
}
|