Refactor CreateToken function to accept duration parameter for token expiration, enhancing flexibility in token management

Author: L4B0MB4

pkg/auth/token_manager.go

@@ -42,7 +42,7 @@ func GetUserUuidFromToken(tokenString string) (uuid.UUID, error) {
 
 }
 
-func CreateToken(userUuid uuid.UUID) (string, error) {
+func CreateToken(userUuid uuid.UUID, duration time.Duration) (string, error) {
 
 	signingSecret, err := getSigningSecret()
 	if err != nil {
@@ -53,7 +53,7 @@ func CreateToken(userUuid uuid.UUID) (string, error) {
 		"sub": userUuid,
 		"iss": "pryvt",
 		"aud": "local-audience",
-		"exp": time.Now().Add(time.Minute * 30).Unix(),
+		"exp": time.Now().Add(duration).Unix(),
 		"iat": time.Now().Unix(),
 	})
 

pkg/auth/token_manager_test.go

@@ -3,6 +3,7 @@ package auth
 import (
 	"os"
 	"testing"
+	"time"
 
 	"github.com/google/uuid"
 	"github.com/stretchr/testify/assert"
@@ -15,7 +16,7 @@ func TestGetUserUuidFromToken(t *testing.T) {
 
 	// Create a test UUID
 	testUuid := uuid.New()
-	tokenString, err := CreateToken(testUuid)
+	tokenString, err := CreateToken(testUuid, 1*time.Second)
 	assert.NoError(t, err)
 
 	returnedUuid, err := GetUserUuidFromToken(tokenString)