pushbits/internal/authentication/token.go

package authentication

import (
	"crypto/rand"
	"math/big"
)

var (
	tokenCharacters        = []byte("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
	regularTokenLength     = 64 // This length includes the prefix (one character).
	compatTokenLength      = 15 // This length includes the prefix (one character).
	applicationTokenPrefix = "A"
)

func randIntn(n int) int {
	maxValue := big.NewInt(int64(n))

	res, err := rand.Int(rand.Reader, maxValue)
	if err != nil {
		panic("random source is not available")
	}

	return int(res.Int64())
}

// GenerateNotExistingToken receives a token generation function and a function to check whether the token exists, returns a unique token.
func GenerateNotExistingToken(generateToken func(bool) string, compat bool, tokenExists func(token string) bool) string {
	for {
		token := generateToken(compat)

		if !tokenExists(token) {
			return token
		}
	}
}

func generateRandomString(length int) string {
	res := make([]byte, length)

	for i := range res {
		index := randIntn(len(tokenCharacters))
		res[i] = tokenCharacters[index]
	}

	return string(res)
}

// GenerateApplicationToken generates a token for an application.
func GenerateApplicationToken(compat bool) string {
	tokenLength := regularTokenLength

	if compat {
		tokenLength = compatTokenLength
	}

	tokenLength -= len(applicationTokenPrefix)

	return applicationTokenPrefix + generateRandomString(tokenLength)
}