Merge 5c55fb0 into b3e3f46

wswebcreation · Aug 25, 2019 · c7860c6 · c7860c6
2 parents b3e3f46 + 5c55fb0
commit c7860c6
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/lib/clientSideScripts/hideRemoveElements.ts b/lib/clientSideScripts/hideRemoveElements.ts
@@ -2,6 +2,20 @@
  * Hide or remove elements on the page
  */
 export default function hideRemoveElements(hideRemoveElements: { hide: HTMLElement[], remove: HTMLElement[] }, hideRemove: boolean): void {
-  hideRemoveElements.hide.forEach(element => element.style.visibility = hideRemove ? 'hidden' : '');
-  hideRemoveElements.remove.forEach(element => element.style.display = hideRemove ? 'none' : '');
+  hideRemoveElements.hide.forEach(element => {
+    if (Array.isArray(element)) {
+      return element.forEach(singleElement => singleElement.style.visibility = hideRemove ? 'hidden' : '');
+    }
+
+    return element.style.visibility = hideRemove ? 'hidden' : '';
+  });
+
+
+  hideRemoveElements.remove.forEach(element => {
+    if (Array.isArray(element)) {
+      return element.forEach(singleElement => singleElement.style.display = hideRemove ? 'none' : '');
+    }
+
+    return element.style.display = hideRemove ? 'none' : '';
+  });
 }