summaryrefslogtreecommitdiff
path: root/vendor/github.com/ulule
diff options
context:
space:
mode:
authorLibravatar dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>2023-08-28 06:59:08 +0000
committerLibravatar GitHub <noreply@github.com>2023-08-28 06:59:08 +0000
commite6407ec95c1c0b134866d7bd4b2d17f4b0c3bf39 (patch)
tree4085dc4a80680a2f480e0466a984e3097f4c58b4 /vendor/github.com/ulule
parent[chore/frontend] Make line-height a wee little bit bigger (#2159) (diff)
downloadgotosocial-e6407ec95c1c0b134866d7bd4b2d17f4b0c3bf39.tar.xz
[chore]: Bump github.com/KimMachineGun/automemlimit from 0.2.6 to 0.3.0 (#2165)
Diffstat (limited to 'vendor/github.com/ulule')
-rw-r--r--vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go65
-rw-r--r--vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go71
2 files changed, 0 insertions, 136 deletions
diff --git a/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go b/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go
deleted file mode 100644
index 23bad417a..000000000
--- a/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/middleware.go
+++ /dev/null
@@ -1,65 +0,0 @@
-package gin
-
-import (
- "strconv"
-
- "github.com/gin-gonic/gin"
-
- "github.com/ulule/limiter/v3"
-)
-
-// Middleware is the middleware for gin.
-type Middleware struct {
- Limiter *limiter.Limiter
- OnError ErrorHandler
- OnLimitReached LimitReachedHandler
- KeyGetter KeyGetter
- ExcludedKey func(string) bool
-}
-
-// NewMiddleware return a new instance of a gin middleware.
-func NewMiddleware(limiter *limiter.Limiter, options ...Option) gin.HandlerFunc {
- middleware := &Middleware{
- Limiter: limiter,
- OnError: DefaultErrorHandler,
- OnLimitReached: DefaultLimitReachedHandler,
- KeyGetter: DefaultKeyGetter,
- ExcludedKey: nil,
- }
-
- for _, option := range options {
- option.apply(middleware)
- }
-
- return func(ctx *gin.Context) {
- middleware.Handle(ctx)
- }
-}
-
-// Handle gin request.
-func (middleware *Middleware) Handle(c *gin.Context) {
- key := middleware.KeyGetter(c)
- if middleware.ExcludedKey != nil && middleware.ExcludedKey(key) {
- c.Next()
- return
- }
-
- context, err := middleware.Limiter.Get(c, key)
- if err != nil {
- middleware.OnError(c, err)
- c.Abort()
- return
- }
-
- c.Header("X-RateLimit-Limit", strconv.FormatInt(context.Limit, 10))
- c.Header("X-RateLimit-Remaining", strconv.FormatInt(context.Remaining, 10))
- c.Header("X-RateLimit-Reset", strconv.FormatInt(context.Reset, 10))
-
- if context.Reached {
- middleware.OnLimitReached(c)
- c.Abort()
- return
- }
-
- c.Next()
-}
diff --git a/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go b/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go
deleted file mode 100644
index 604c6bc68..000000000
--- a/vendor/github.com/ulule/limiter/v3/drivers/middleware/gin/options.go
+++ /dev/null
@@ -1,71 +0,0 @@
-package gin
-
-import (
- "net/http"
-
- "github.com/gin-gonic/gin"
-)
-
-// Option is used to define Middleware configuration.
-type Option interface {
- apply(*Middleware)
-}
-
-type option func(*Middleware)
-
-func (o option) apply(middleware *Middleware) {
- o(middleware)
-}
-
-// ErrorHandler is an handler used to inform when an error has occurred.
-type ErrorHandler func(c *gin.Context, err error)
-
-// WithErrorHandler will configure the Middleware to use the given ErrorHandler.
-func WithErrorHandler(handler ErrorHandler) Option {
- return option(func(middleware *Middleware) {
- middleware.OnError = handler
- })
-}
-
-// DefaultErrorHandler is the default ErrorHandler used by a new Middleware.
-func DefaultErrorHandler(c *gin.Context, err error) {
- panic(err)
-}
-
-// LimitReachedHandler is an handler used to inform when the limit has exceeded.
-type LimitReachedHandler func(c *gin.Context)
-
-// WithLimitReachedHandler will configure the Middleware to use the given LimitReachedHandler.
-func WithLimitReachedHandler(handler LimitReachedHandler) Option {
- return option(func(middleware *Middleware) {
- middleware.OnLimitReached = handler
- })
-}
-
-// DefaultLimitReachedHandler is the default LimitReachedHandler used by a new Middleware.
-func DefaultLimitReachedHandler(c *gin.Context) {
- c.String(http.StatusTooManyRequests, "Limit exceeded")
-}
-
-// KeyGetter will define the rate limiter key given the gin Context.
-type KeyGetter func(c *gin.Context) string
-
-// WithKeyGetter will configure the Middleware to use the given KeyGetter.
-func WithKeyGetter(handler KeyGetter) Option {
- return option(func(middleware *Middleware) {
- middleware.KeyGetter = handler
- })
-}
-
-// DefaultKeyGetter is the default KeyGetter used by a new Middleware.
-// It returns the Client IP address.
-func DefaultKeyGetter(c *gin.Context) string {
- return c.ClientIP()
-}
-
-// WithExcludedKey will configure the Middleware to ignore key(s) using the given function.
-func WithExcludedKey(handler func(string) bool) Option {
- return option(func(middleware *Middleware) {
- middleware.ExcludedKey = handler
- })
-}