From 25719ddc5e4caf28e8443c09fdba6008d5628f9d Mon Sep 17 00:00:00 2001
From: hackademix
Date: Thu, 12 Jul 2018 22:58:52 +0200
Subject: Fixed cleaning up reloadingTabs in StreamFilter.onstart() was still
 breaking some feeds sometimes.

---
 src/bg/RequestUtil.js | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/src/bg/RequestUtil.js b/src/bg/RequestUtil.js
index e151f5b..557293d 100644
--- a/src/bg/RequestUtil.js
+++ b/src/bg/RequestUtil.js
@@ -8,11 +8,15 @@
 
   let pendingRequests = new Map();
 
-  let reloadingTabs = new Set();
+  let reloadingTabs = new Map();
   let tabKey = (tabId, url) => `${tabId}:${url}`;
 
   let cleanup = r => {
     pendingRequests.delete(r.requestId);
+    let key = tabKey(r.tabId, r.url);
+    if (reloadingTabs.get(key) === false) {
+      reloadingTabs.delete(key);
+    }
   };
   let filter = {
     urls: ["<all_urls>"],
@@ -66,14 +70,8 @@
       if (frameId === 0) {
         let key = tabKey(tabId, url);
         debug("Checking whether %s is a reloading tab...", key);
-        if (reloadingTabs.has(key)) {
-          debug("Skipping dynamic script injection for reloading feed tab", key);
-          let filter = browser.webRequest.filterResponseData(requestId);
-          filter.onstart = e => {
-            reloadingTabs.delete(key);
-            filter.write(NULL);
-            filter.disconnect();
-          }
+        if (reloadingTabs.get(key)) {
+          reloadingTabs.set(key, false); // doom it for removal in cleanup
           return;
         }
       }
@@ -97,7 +95,7 @@
         if (mustCheckFeed && !scriptsRan) {
           mustReload = true;
           debug(`Marking as "must reload"`, tabId, url);
-          reloadingTabs.add(tabKey(tabId, url));
+          reloadingTabs.set(tabKey(tabId, url), true);
         }
         if (buffer && buffer.length) {
           debug("Flushing %s buffer chunks", buffer.length);
-- 
cgit v1.2.3