diff --git a/main.go b/main.go index 661bf17..e3c520e 100644 --- a/main.go +++ b/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()