mirror of https://github.com/gofiber/fiber.git
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package encryptcookie
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
// New creates a new middleware handler
|
|
func New(config ...Config) fiber.Handler {
|
|
// Set default config
|
|
cfg := configDefault(config...)
|
|
|
|
// Return new handler
|
|
return func(c *fiber.Ctx) error {
|
|
// Don't execute middleware if Next returns true
|
|
if cfg.Next != nil && cfg.Next(c) {
|
|
return c.Next()
|
|
}
|
|
|
|
// Decrypt request cookies
|
|
c.Request().Header.VisitAllCookie(func(key, value []byte) {
|
|
keyString := string(key)
|
|
if !isDisabled(keyString, cfg.Except) {
|
|
decryptedValue, err := cfg.Decryptor(string(value), cfg.Key)
|
|
if err != nil {
|
|
c.Request().Header.SetCookieBytesKV(key, nil)
|
|
} else {
|
|
c.Request().Header.SetCookie(string(key), decryptedValue)
|
|
}
|
|
}
|
|
})
|
|
|
|
// Continue stack
|
|
err := c.Next()
|
|
|
|
// Encrypt response cookies
|
|
c.Response().Header.VisitAllCookie(func(key, value []byte) {
|
|
keyString := string(key)
|
|
if !isDisabled(keyString, cfg.Except) {
|
|
cookieValue := fasthttp.Cookie{}
|
|
cookieValue.SetKeyBytes(key)
|
|
if c.Response().Header.Cookie(&cookieValue) {
|
|
encryptedValue, err := cfg.Encryptor(string(cookieValue.Value()), cfg.Key)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
cookieValue.SetValue(encryptedValue)
|
|
c.Response().Header.SetCookie(&cookieValue)
|
|
}
|
|
}
|
|
})
|
|
|
|
return err
|
|
}
|
|
}
|