Finish offline detection

1 files changed, 10 insertions, 3 deletions

diff --git a/src/assets/javascripts/imdb.js b/src/assets/javascripts/imdb.js
index 9981f1b9..1f6c7a5f 100644
--- a/src/assets/javascripts/imdb.js
+++ b/src/assets/javascripts/imdb.js
@@ -17,10 +17,10 @@ for (let i = 0; i < frontends.length; i++) {
 }
 
 function setRedirects(val) {
-	browser.storage.local.get("cloudflareBlackList", r => {
+	browser.storage.local.get(["cloudflareBlackList", "offlineBlackList"], r => {
 		redirects.libremdb = val
 		libremdbNormalRedirectsChecks = [...redirects.libremdb.normal]
-		for (const instance of r.cloudflareBlackList) {
+		for (const instance of [...r.cloudflareBlackList, ...r.offlineBlackList]) {
 			const a = libremdbNormalRedirectsChecks.indexOf(instance)
 			if (a > -1) libremdbNormalRedirectsChecks.splice(a, 1)
 		}
@@ -172,12 +172,18 @@ function initDefaults() {
 				for (let i = 0; i < frontends.length; i++) {
 					redirects[frontends[i]] = dataJson[frontends[i]]
 				}
+	browser.storage.local.get(["cloudflareBlackList", "offlineBlackList"], async r => {
+		libremdbNormalRedirectsChecks = [...redirects.libremdb.normal]
+		for (const instance of [...r.cloudflareBlackList, ...r.offlineBlackList]) {
+			const a = libremdbNormalRedirectsChecks.indexOf(instance)
+			if (a > -1) libremdbNormalRedirectsChecks.splice(a, 1)
+		}
 				browser.storage.local.set(
 					{
 						disableImdb: true,
 						imdbRedirects: redirects,
 
-						libremdbNormalRedirectsChecks: [...redirects.libremdb.normal],
+						libremdbNormalRedirectsChecks,
 						libremdbNormalCustomRedirects: [],
 
 						libremdbTorRedirectsChecks: [...redirects.libremdb.tor],
@@ -192,6 +198,7 @@ function initDefaults() {
 					() => resolve()
 				)
 			})
+			})
 	})
 }