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
1.0 KiB
Go
44 lines
1.0 KiB
Go
4 years ago
|
package rand
|
||
|
|
||
4 years ago
|
import (
|
||
|
"math/rand"
|
||
|
"time"
|
||
|
"unicode"
|
||
4 years ago
|
|
||
3 years ago
|
"github.com/mickael-menu/zk/internal/core"
|
||
4 years ago
|
)
|
||
|
|
||
4 years ago
|
// NewIDGenerator returns a function generating string IDs using the given options.
|
||
4 years ago
|
// Inspired by https://www.calhoun.io/creating-random-strings-in-go/
|
||
3 years ago
|
func NewIDGenerator(options core.IDOptions) func() string {
|
||
4 years ago
|
if options.Length < 1 {
|
||
4 years ago
|
panic("IDOptions.Length must be at least 1")
|
||
4 years ago
|
}
|
||
|
|
||
4 years ago
|
var charset []rune
|
||
|
for _, char := range options.Charset {
|
||
|
switch options.Case {
|
||
3 years ago
|
case core.CaseLower:
|
||
4 years ago
|
charset = append(charset, unicode.ToLower(char))
|
||
3 years ago
|
case core.CaseUpper:
|
||
4 years ago
|
charset = append(charset, unicode.ToUpper(char))
|
||
3 years ago
|
case core.CaseMixed:
|
||
4 years ago
|
charset = append(charset, unicode.ToLower(char))
|
||
|
charset = append(charset, unicode.ToUpper(char))
|
||
|
default:
|
||
4 years ago
|
panic("unknown zk.Case value")
|
||
4 years ago
|
}
|
||
|
}
|
||
|
|
||
|
rand := rand.New(rand.NewSource(time.Now().UnixNano()))
|
||
|
|
||
4 years ago
|
return func() string {
|
||
|
buf := make([]rune, options.Length)
|
||
|
for i := range buf {
|
||
|
buf[i] = charset[rand.Intn(len(charset))]
|
||
|
}
|
||
|
|
||
|
return string(buf)
|
||
|
}
|
||
4 years ago
|
}
|