Skip to content

Commit f775b43

Browse files
committed
Allow to list whitelist
1 parent 27296da commit f775b43

File tree

3 files changed

+21
-2
lines changed

3 files changed

+21
-2
lines changed

src/main/java/net/lightbody/bmp/proxy/ProxyServer.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -292,6 +292,10 @@ public void blacklistRequests(String pattern, int responseCode) {
292292
public List<BlacklistEntry> getBlacklistedRequests() {
293293
return client.getBlacklistedRequests();
294294
}
295+
296+
public WhitelistEntry getWhitelistRequests() {
297+
return client.getWhitelistRequests();
298+
}
295299

296300
public void clearBlacklist() {
297301
client.clearBlacklist();

src/main/java/net/lightbody/bmp/proxy/bricks/ProxyResource.java

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -160,8 +160,19 @@ public Reply<?> clearBlacklist(@Named("port") int port, Request request) {
160160
return Reply.saying().notFound();
161161
}
162162

163-
proxy.clearBlacklist();
164-
return Reply.saying().ok();
163+
proxy.clearBlacklist();
164+
return Reply.saying().ok();
165+
}
166+
167+
@Get
168+
@At("/:port/whitelist")
169+
public Reply<?> getWhitelist(@Named("port") int port, Request request) {
170+
ProxyServer proxy = proxyManager.get(port);
171+
if (proxy == null) {
172+
return Reply.saying().notFound();
173+
}
174+
175+
return Reply.with(proxy.getWhitelistRequests()).as(Json.class);
165176
}
166177

167178
@Put

src/main/java/net/lightbody/bmp/proxy/http/BrowserMobHttpClient.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -974,6 +974,10 @@ public void clearBlacklist() {
974974
blacklistEntries.clear();
975975
}
976976

977+
public WhitelistEntry getWhitelistRequests() {
978+
return whitelistEntry;
979+
}
980+
977981
public synchronized void whitelistRequests(String[] patterns, int responseCode) {
978982
// synchronized to guard against concurrent modification
979983
whitelistEntry = new WhitelistEntry(patterns, responseCode);

0 commit comments

Comments
 (0)