sonr/pkg/common/middleware/cookie/cookie.go

package cookie

import (
	"encoding/base64"
	"net/http"
	"time"

	"github.com/labstack/echo/v4"
)

func Exists(c echo.Context, key Key) bool {
	ck, err := c.Cookie(key.String())
	if err != nil {
		return false
	}
	return ck != nil
}

func Read(c echo.Context, key Key) (string, error) {
	cookie, err := c.Cookie(key.String())
	if err != nil {
		// Cookie not found or other error
		return "", err
	}
	if cookie == nil || cookie.Value == "" {
		// Cookie is empty
		return "", http.ErrNoCookie
	}
	return cookie.Value, nil
}

func ReadBytes(c echo.Context, key Key) ([]byte, error) {
	cookie, err := c.Cookie(key.String())
	if err != nil {
		// Cookie not found or other error
		return nil, err
	}
	if cookie == nil || cookie.Value == "" {
		// Cookie is empty
		return nil, http.ErrNoCookie
	}
	return base64.RawURLEncoding.DecodeString(cookie.Value)
}

func ReadUnsafe(c echo.Context, key Key) string {
	ck, err := c.Cookie(key.String())
	if err != nil {
		return ""
	}
	return ck.Value
}

func Write(c echo.Context, key Key, value string) error {
	cookie := &http.Cookie{
		Name:     key.String(),
		Value:    value,
		Expires:  time.Now().Add(24 * time.Hour),
		HttpOnly: true,
		Path:     "/",
		// Add Secure and SameSite attributes as needed
	}
	c.SetCookie(cookie)
	return nil
}

func WriteBytes(c echo.Context, key Key, value []byte) error {
	cookie := &http.Cookie{
		Name:     key.String(),
		Value:    base64.RawURLEncoding.EncodeToString(value),
		Expires:  time.Now().Add(24 * time.Hour),
		HttpOnly: true,
		Path:     "/",
		// Add Secure and SameSite attributes as needed
	}
	c.SetCookie(cookie)
	return nil
}