v1.14.5/sdk/database/helper/credsutil/usernames.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0

package credsutil

import (
	"fmt"
	"strings"
	"time"
)

type CaseOp int

const (
	KeepCase CaseOp = iota
	Uppercase
	Lowercase
)

type usernameBuilder struct {
	displayName string
	roleName    string
	separator   string

	maxLen        int
	caseOperation CaseOp
}

func (ub usernameBuilder) makeUsername() (string, error) {
	userUUID, err := RandomAlphaNumeric(20, false)
	if err != nil {
		return "", err
	}

	now := fmt.Sprint(time.Now().Unix())

	username := joinNonEmpty(ub.separator,
		"v",
		ub.displayName,
		ub.roleName,
		userUUID,
		now)
	username = trunc(username, ub.maxLen)
	switch ub.caseOperation {
	case Lowercase:
		username = strings.ToLower(username)
	case Uppercase:
		username = strings.ToUpper(username)
	}

	return username, nil
}

type UsernameOpt func(*usernameBuilder)

func DisplayName(dispName string, maxLength int) UsernameOpt {
	return func(b *usernameBuilder) {
		b.displayName = trunc(dispName, maxLength)
	}
}

func RoleName(roleName string, maxLength int) UsernameOpt {
	return func(b *usernameBuilder) {
		b.roleName = trunc(roleName, maxLength)
	}
}

func Separator(sep string) UsernameOpt {
	return func(b *usernameBuilder) {
		b.separator = sep
	}
}

func MaxLength(maxLen int) UsernameOpt {
	return func(b *usernameBuilder) {
		b.maxLen = maxLen
	}
}

func Case(c CaseOp) UsernameOpt {
	return func(b *usernameBuilder) {
		b.caseOperation = c
	}
}

func ToLower() UsernameOpt {
	return Case(Lowercase)
}

func ToUpper() UsernameOpt {
	return Case(Uppercase)
}

func GenerateUsername(opts ...UsernameOpt) (string, error) {
	b := usernameBuilder{
		separator:     "_",
		maxLen:        100,
		caseOperation: KeepCase,
	}

	for _, opt := range opts {
		opt(&b)
	}

	return b.makeUsername()
}

func trunc(str string, l int) string {
	switch {
	case l > 0:
		if l > len(str) {
			return str
		}
		return str[:l]
	case l == 0:
		return str
	default:
		return ""
	}
}

func joinNonEmpty(sep string, vals ...string) string {
	if sep == "" {
		return strings.Join(vals, sep)
	}
	switch len(vals) {
	case 0:
		return ""
	case 1:
		return vals[0]
	}
	builder := &strings.Builder{}
	for _, val := range vals {
		if val == "" {
			continue
		}
		if builder.Len() > 0 {
			builder.WriteString(sep)
		}
		builder.WriteString(val)
	}
	return builder.String()
}