44 lines
931 B
Go
44 lines
931 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"web-proxy/redis_rate"
|
||
|
|
)
|
||
|
|
|
||
|
|
func is_api_ratelimited(rw http.ResponseWriter, r *http.Request) bool {
|
||
|
|
var limit int
|
||
|
|
var key string
|
||
|
|
|
||
|
|
if r.Header.Get("X-PluralKit-App") == token2 {
|
||
|
|
limit = 20
|
||
|
|
key = "token2"
|
||
|
|
} else {
|
||
|
|
limit = 2
|
||
|
|
key = r.Header.Get("Fly-Client-IP")
|
||
|
|
}
|
||
|
|
|
||
|
|
res, err := limiter.Allow(r.Context(), "ratelimit:"+key, redis_rate.Limit{
|
||
|
|
Period: time.Second,
|
||
|
|
Rate: limit,
|
||
|
|
Burst: 5,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
rw.Header().Set("X-RateLimit-Limit", fmt.Sprint(limit))
|
||
|
|
rw.Header().Set("X-RateLimit-Remaining", fmt.Sprint(res.Remaining))
|
||
|
|
rw.Header().Set("X-RateLimit-Reset", fmt.Sprint(time.Now().Add(res.ResetAfter).UnixNano()/1_000_000))
|
||
|
|
|
||
|
|
if res.Allowed < 1 {
|
||
|
|
rw.WriteHeader(429)
|
||
|
|
rw.Write([]byte(`{"message":"429: too many requests","retry_after":` + fmt.Sprint(res.RetryAfter.Milliseconds()) + `,"code":0}`))
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
return false
|
||
|
|
}
|