mirror of https://github.com/gofiber/fiber.git
76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package earlydata
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
const (
|
|
DefaultHeaderName = "Early-Data"
|
|
DefaultHeaderTrueValue = "1"
|
|
)
|
|
|
|
// Config defines the config for middleware.
|
|
type Config struct {
|
|
// Next defines a function to skip this middleware when returned true.
|
|
//
|
|
// Optional. Default: nil
|
|
Next func(c *fiber.Ctx) bool
|
|
|
|
// IsEarlyData returns whether the request is an early-data request.
|
|
//
|
|
// Optional. Default: a function which checks if the "Early-Data" request header equals "1".
|
|
IsEarlyData func(c *fiber.Ctx) bool
|
|
|
|
// AllowEarlyData returns whether the early-data request should be allowed or rejected.
|
|
//
|
|
// Optional. Default: a function which rejects the request on unsafe and allows the request on safe HTTP request methods.
|
|
AllowEarlyData func(c *fiber.Ctx) bool
|
|
|
|
// Error is returned in case an early-data request is rejected.
|
|
//
|
|
// Optional. Default: fiber.ErrTooEarly.
|
|
Error error
|
|
}
|
|
|
|
// ConfigDefault is the default config
|
|
//
|
|
//nolint:gochecknoglobals // Using a global var is fine here
|
|
var ConfigDefault = Config{
|
|
IsEarlyData: func(c *fiber.Ctx) bool {
|
|
return c.Get(DefaultHeaderName) == DefaultHeaderTrueValue
|
|
},
|
|
|
|
AllowEarlyData: func(c *fiber.Ctx) bool {
|
|
return fiber.IsMethodSafe(c.Method())
|
|
},
|
|
|
|
Error: fiber.ErrTooEarly,
|
|
}
|
|
|
|
// Helper function to set default values
|
|
func configDefault(config ...Config) Config {
|
|
// Return default config if nothing provided
|
|
if len(config) < 1 {
|
|
return ConfigDefault
|
|
}
|
|
|
|
// Override default config
|
|
cfg := config[0]
|
|
|
|
// Set default values
|
|
|
|
if cfg.IsEarlyData == nil {
|
|
cfg.IsEarlyData = ConfigDefault.IsEarlyData
|
|
}
|
|
|
|
if cfg.AllowEarlyData == nil {
|
|
cfg.AllowEarlyData = ConfigDefault.AllowEarlyData
|
|
}
|
|
|
|
if cfg.Error == nil {
|
|
cfg.Error = ConfigDefault.Error
|
|
}
|
|
|
|
return cfg
|
|
}
|