From 01bb440c87d9f872acdf2c2b9ca4ed463af1e698 Mon Sep 17 00:00:00 2001
From: Jonathan Cobb <jonathan@kyuss.org>
Date: Wed, 5 Feb 2020 22:29:34 -0500
Subject: [PATCH] allow rate limit to decide unlimited use based on full
 request context

---
 .../java/org/cobbzilla/wizard/filters/RateLimitFilter.java    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/wizard-server/src/main/java/org/cobbzilla/wizard/filters/RateLimitFilter.java b/wizard-server/src/main/java/org/cobbzilla/wizard/filters/RateLimitFilter.java
index d945b42..462a8be 100644
--- a/wizard-server/src/main/java/org/cobbzilla/wizard/filters/RateLimitFilter.java
+++ b/wizard-server/src/main/java/org/cobbzilla/wizard/filters/RateLimitFilter.java
@@ -59,7 +59,7 @@ public abstract class RateLimitFilter implements ContainerRequestFilter {
         String key;
         final Principal user = empty(request.getSecurityContext()) ? null : request.getSecurityContext().getUserPrincipal();
         if (!empty(user)) {
-            if (allowUnlimitedUse(user)) return null;
+            if (allowUnlimitedUse(user, request)) return null;
             key = user.getName();
         }
         else {
@@ -77,7 +77,7 @@ public abstract class RateLimitFilter implements ContainerRequestFilter {
         }
     }
 
-    protected boolean allowUnlimitedUse(Principal user) { return false; }
+    protected boolean allowUnlimitedUse(Principal user, ContainerRequestContext request) { return false; }
 
     @Getter(lazy=true) private final List<ApiRateLimit> limits = initLimits();
     private List<ApiRateLimit> initLimits() {