src/javascript/enhancements/languageDisplay.js

@@ -10,7 +10,6 @@ export function init() {
     }, "^/anime/[0-9]*$");
 }
 
-function updateLanguageDisplay(node) {
 const langPrefix = 'ep.lang.';
 const dubSuffix = 'dub';
 const subSuffix = 'sub';
@@ -19,13 +18,19 @@ function updateLanguageDisplay(node) {
 const subIcon = 'closed_caption';
 const zeroWidthSpace = ''; // ​
 
-    let listItems = document.querySelectorAll('md-list-item');
+function updateLanguageDisplay(node) {
+    const listNodeName = 'MD-LIST-ITEM';
 
-    listItems.forEach(item => {
+    if (node.nodeName === listNodeName) {
+        updateLanguageDisplayListMode(node);
+    }
+}
+
+function updateLanguageDisplayListMode(node) {
     // last column with flags
-        let col = item.querySelector('h3.layout-align-end-center');
+    let col = node.querySelector('h3.layout-align-end-center');
 
-        if (col.eaManipulated) {
+    if (typeof col === 'undefined' || col.eaManipulated) {
         return;
     }
 
@@ -181,15 +186,14 @@ function updateLanguageDisplay(node) {
         col.appendChild(div);
     });
 
-        item.querySelectorAll('.layout-column:not(:last-child)').forEach(div => {
+    node.querySelectorAll('.layout-column:not(:last-child)').forEach(div => {
         div.style.borderRight = '1px solid rgba(155,155,155, 0.2)';
     })
 
-        item.querySelectorAll('.layout-column').forEach(div => {
+    node.querySelectorAll('.layout-column').forEach(div => {
         div.style.paddingLeft = '2px';
         div.style.paddingRight = '2px';
     })
 
     col.eaManipulated = true;
-    });
 }