This commit is contained in:
38
main.go
38
main.go
@@ -71,6 +71,7 @@ func main() {
|
||||
}
|
||||
|
||||
// Server
|
||||
http.HandleFunc("/whitelist", handleWhitelist)
|
||||
http.HandleFunc("/check/", handleCheck)
|
||||
http.Handle("/metrics", promhttp.Handler())
|
||||
|
||||
@@ -176,6 +177,43 @@ func normalizePrefix(s string) (string, bool) {
|
||||
return s, err == nil
|
||||
}
|
||||
|
||||
func handleWhitelist(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
var body struct {
|
||||
IP string `json:"ip"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
|
||||
http.Error(w, "bad request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
addr, err := netip.ParseAddr(body.IP)
|
||||
if err != nil {
|
||||
http.Error(w, "invalid IP", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
key := "wl:" + addr.String()
|
||||
if err := rdb.Set(ctx, key, "1", 0).Err(); err != nil {
|
||||
http.Error(w, "redis error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Optional: Cache leeren für die IP
|
||||
prefixCacheMu.Lock()
|
||||
defer prefixCacheMu.Unlock()
|
||||
// Kein spezifischer IP-Cache in deinem Design, aber hier könnte man Cache invalidieren falls nötig
|
||||
|
||||
writeJSON(w, map[string]string{
|
||||
"status": "whitelisted",
|
||||
"ip": addr.String(),
|
||||
})
|
||||
}
|
||||
|
||||
// Check-Handler
|
||||
func handleCheck(w http.ResponseWriter, r *http.Request) {
|
||||
checkRequests.Inc()
|
||||
|
Reference in New Issue
Block a user