From 93284105a5b4720f75a7c778ce189e266bb190f4 Mon Sep 17 00:00:00 2001
From: wanhose <wanhose.development@gmail.com>
Date: Sat, 25 Apr 2020 13:18:00 +0200
Subject: [PATCH] feat(content): code improvements

---
 js/content.js | 18 +++++-------------
 1 file changed, 5 insertions(+), 13 deletions(-)

diff --git a/js/content.js b/js/content.js
index 670bcc2..6c55cbd 100644
--- a/js/content.js
+++ b/js/content.js
@@ -18,8 +18,12 @@ if (typeof chrome.app.isInstalled !== 'undefined') {
     ]);
 
     const removableElements = Array.from([
+      // ENS
+      document.getElementById("ensNotifyBanner"),
       // Quantcast
       ...document.getElementsByClassName("qc-cmp-ui-container"),
+      // OneTrust
+      document.getElementById("onetrust-consent-sdk"),
       // Optanon
       ...document.getElementsByClassName("optanon-alert-box-wrapper"),
     ]);
@@ -29,19 +33,7 @@ if (typeof chrome.app.isInstalled !== 'undefined') {
       element.classList.remove("qc-cmp-ui-showing");
     });
     
-    removableElements.forEach(element => {
-      element.remove();
-    });
-
-    let removableElement = null;
-
-    // ENS
-    removableElement = document.getElementById("ensNotifyBanner");
-    if (!!removableElement) removableElement.remove();
-
-    // OneTrust
-    removableElement = document.getElementById("onetrust-consent-sdk");
-    if (!!removableElement) removableElement.remove();
+    removableElements.forEach(element => !!element && element.remove());
   };
 
   // Observer starts observe when call this function