From 5e0ab1515ab1c399a6405f579c3b931563e4020d Mon Sep 17 00:00:00 2001
From: hackademix <giorgio@maone.net>
Date: Mon, 30 Jul 2018 16:18:17 +0200
Subject: Implement early whitelisting / blacklisting logic.

---
 bg/ResponseProcessor.js | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

(limited to 'bg/ResponseProcessor.js')

diff --git a/bg/ResponseProcessor.js b/bg/ResponseProcessor.js
index 3f3151b..400cb42 100644
--- a/bg/ResponseProcessor.js
+++ b/bg/ResponseProcessor.js
@@ -52,6 +52,13 @@ class ResponseProcessor {
   }
 }
 
+Object.assign(ResponseProcessor, {
+  // control flow values to be returned by handler.pre() callbacks
+	ACCEPT: {},
+	REJECT: {cancel: true},
+	CONTINUE: null
+});
+
 class ResponseTextFilter {
   constructor(request) {
     this.request = request;
@@ -64,9 +71,16 @@ class ResponseTextFilter {
   }
 
   process(handler) {
-    if (!this.canProcess) return {};
-    let metaData = this.metaData;
-    let {requestId, responseHeaders} = this.request;
+    if (!this.canProcess) return ResponseProcessor.ACCEPT;
+    let {metaData, request} = this;
+    if (typeof handler.pre === "function") {
+      let res = handler.pre({request, metaData});
+      if (res) return res;
+      if (handler.post) handler = handler.post;
+      if (typeof handler !== "function") ResponseProcessor.ACCEPT;
+    }
+    
+    let {requestId, responseHeaders} = request;
     let filter = browser.webRequest.filterResponseData(requestId);
     let buffer = [];
 
@@ -83,7 +97,7 @@ class ResponseTextFilter {
       let editedText = null;
       try {
         let response = {
-          request: this.request,
+          request,
           metaData,
           text,
         };
@@ -103,7 +117,7 @@ class ResponseTextFilter {
       filter.disconnect();
     }
 
-    return metaData.forceUTF8() ? {responseHeaders} : {};
+    return metaData.forceUTF8() ? {responseHeaders} : ResponseProcessor.ACCEPT;;
   }
 }
 
-- 
cgit v1.2.3