1 files changed, 12 insertions, 21 deletions
diff --git a/session/functions.go b/session/functions.go
index 589091f..776fddd 100644
--- a/session/functions.go
+++ b/session/functions.go
@@ -2,35 +2,26 @@ package session
 
 import "github.com/gofiber/fiber/v2"
 
-func CreateSession(context *fiber.Ctx, email string) error {
-	sess, err := Store.Get(context)
-	if err != nil {
-		return err
-	}
+const emailKey = "email"
 
-	sess.Set("email", email)
-	return sess.Save()
+func CreateSession(ctx *fiber.Ctx, email string) error {
+	return Set(ctx, emailKey, email)
 }
 
-func DestroySession(context *fiber.Ctx) error {
-	sess, err := Store.Get(context)
-	if err != nil {
-		return err
-	}
-
-	return sess.Destroy()
+func DestroySession(ctx *fiber.Ctx) error {
+	return Delete(ctx, emailKey)
 }
 
-func GetSessionEmail(context *fiber.Ctx) (string, error) {
-	sess, err := Store.Get(context)
-	if err != nil {
+func GetSessionEmail(ctx *fiber.Ctx) (string, error) {
+	value, err := Get(ctx, emailKey)
+	if err != nil || value == nil {
 		return "", err
 	}
 
-	email := sess.Get("email")
-	if emailStr, ok := email.(string); ok {
-		return emailStr, nil
+	email, ok := value.(string)
+	if !ok {
+		return "", nil
 	}
 
-	return "", nil
+	return email, nil
 }