56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package auth
|
|
|
|
import (
|
|
"math/rand"
|
|
)
|
|
|
|
var (
|
|
tokenCharacters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789.-_")
|
|
randomTokenLength = 14
|
|
applicationPrefix = "A"
|
|
clientPrefix = "C"
|
|
pluginPrefix = "P"
|
|
)
|
|
|
|
// GenerateNotExistingToken receives a token generation func and a func to check whether the token exists, returns a unique token.
|
|
func GenerateNotExistingToken(generateToken func() string, tokenExists func(token string) bool) string {
|
|
for {
|
|
token := generateToken()
|
|
if !tokenExists(token) {
|
|
return token
|
|
}
|
|
}
|
|
}
|
|
|
|
// GenerateApplicationToken generates an application token.
|
|
func GenerateApplicationToken() string {
|
|
return generateRandomToken(applicationPrefix)
|
|
}
|
|
|
|
// GenerateClientToken generates a client token.
|
|
func GenerateClientToken() string {
|
|
return generateRandomToken(clientPrefix)
|
|
}
|
|
|
|
// GeneratePluginToken generates a plugin token.
|
|
func GeneratePluginToken() string {
|
|
return generateRandomToken(pluginPrefix)
|
|
}
|
|
|
|
// GenerateImageName generates an image name.
|
|
func GenerateImageName() string {
|
|
return generateRandomString(25)
|
|
}
|
|
|
|
func generateRandomToken(prefix string) string {
|
|
return prefix + generateRandomString(randomTokenLength)
|
|
}
|
|
|
|
func generateRandomString(length int) string {
|
|
b := make([]rune, length)
|
|
for i := range b {
|
|
b[i] = tokenCharacters[rand.Intn(len(tokenCharacters))]
|
|
}
|
|
return string(b)
|
|
}
|