feat(util): update generate random str and simple password functions

pkg/util/random.go

@@ -1,46 +1,38 @@
 package util
 
 import (
-	"crypto/rand"
-	"encoding/base64"
-	"errors"
-	mathRand "math/rand"
+	"math/rand"
 	"regexp"
 )
 
 const (
-	minioSecretIDLength = 40
+	letterBytes  = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+	specialBytes = "!@#$%^&*()_+-=[]{}\\|;':\",.<>/?`~"
+	numBytes     = "0123456789"
 )
 
-// GenerateSecretAccessKey - generate random base64 numeric value from a random seed.
-func GenerateSecretAccessKey() (string, error) {
-	rb := make([]byte, minioSecretIDLength)
-	if _, e := rand.Read(rb); e != nil {
-		return "", errors.New("could not generate Secret Key")
-	}
-
-	return string(Encode(rb)), nil
-}
-
-// Encode queues message
-func Encode(value []byte) []byte {
-	length := len(value)
-	encoded := make([]byte, base64.URLEncoding.EncodedLen(length))
-	base64.URLEncoding.Encode(encoded, value)
-	return encoded
-}
-
-// GenerateRandomStr - generate random string for given length
-func GenerateRandomStr(length int) string {
-	letterBytes := "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+// GenerateRandomStr generates a random string of the specified length.
+// The string can contain letters, special characters, and numbers based on the flags provided.
+func GenerateRandomStr(length int, useLetters bool, useSpecial bool, useNum bool) string {
 	bytes := make([]byte, length)
 	for i := range bytes {
-		bytes[i] = letterBytes[mathRand.Intn(len(letterBytes))]
+		if useLetters {
+			bytes[i] = letterBytes[rand.Intn(len(letterBytes))]
+		} else if useSpecial {
+			bytes[i] = specialBytes[rand.Intn(len(specialBytes))]
+		} else if useNum {
+			bytes[i] = numBytes[rand.Intn(len(numBytes))]
+		}
 	}
-
 	return string(bytes)
 }
 
+// GenerateSimplePassword generates a simple password of the specified length.
+// The password will contain only letters and numbers.
+func GenerateSimplePassword(length int) string {
+	return GenerateRandomStr(length, true, false, true)
+}
+
 // RemoveSpecialCharacter - remove special character
 func RemoveSpecialCharacter(str string) string {
 	regex := regexp.MustCompile("[^a-zA-Z0-9]+")

pkg/util/random_test.go

@@ -1,8 +1,35 @@
 package util
 
-import "testing"
+import (
+	"testing"
+)
 
-func Test_generateRandomStr(t *testing.T) {
-	str := GenerateRandomStr(5)
-	t.Log(str)
+func TestGenerateRandomStr(t *testing.T) {
+	length := 10
+	useLetters := true
+	useSpecial := true
+	useNum := true
+
+	randomStr := GenerateRandomStr(length, useLetters, useSpecial, useNum)
+
+	if len(randomStr) != length {
+		t.Errorf("Generated random string length is incorrect. Expected: %d, Got: %d", length, len(randomStr))
+	}
+
+	// Add additional assertions here if needed
+
+	t.Logf("Generated random string: %s", randomStr)
+}
+
+func TestGenerateSimplePassword(t *testing.T) {
+	length := 8
+	password := GenerateSimplePassword(length)
+
+	if len(password) != length {
+		t.Errorf("Generated password length is incorrect. Expected: %d, Got: %d", length, len(password))
+	}
+
+	// Add additional assertions here if needed
+
+	t.Logf("Generated password: %s", password)
 }