src/javascript/enhancements/languageDisplay.js

@@ -10,22 +10,27 @@ export function init() {
     }, "^/anime/[0-9]*$");
 }
 
+const langPrefix = 'ep.lang.';
+const dubSuffix = 'dub';
+const subSuffix = 'sub';
+
+const dubIcon = 'volume_up';
+const subIcon = 'closed_caption';
+const zeroWidthSpace = ''; // ​
+
 function updateLanguageDisplay(node) {
-    const langPrefix = 'ep.lang.';
+    const listNodeName = 'MD-LIST-ITEM';
-    const dubSuffix = 'dub';
-    const subSuffix = 'sub';
 
-    const dubIcon = 'volume_up';
+    if (node.nodeName === listNodeName) {
-    const subIcon = 'closed_caption';
+        updateLanguageDisplayListMode(node);
-    const zeroWidthSpace = ''; // ​
+    }
+}
 
-    let listItems = document.querySelectorAll('md-list-item');
+function updateLanguageDisplayListMode(node) {
-
-    listItems.forEach(item => {
     // 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;
-    });
 }