Fix bug menu

src/kaplus.js

@@ -32,22 +32,16 @@ function createRow(key, value) {
 }
 
 function shortcutElementReplace(elt, img, text) {
-	let imgElements = elt.getElementsByTagName("img");
-	for (let i=0; i<imgElements.length; i++) {
-		if (i === 0) {
-			elt.getElementsByTagName("a")[0].classList.add("has_notif");
-		}
-		imgElements[i].remove();
-		i --;
-	}
-
+    // If more than one hyperlink element found,
+    // then keep the last one but with href of the first one.
 	let aElements = elt.getElementsByTagName("a");
-	for (i=0; i<aElements.length; i++) {
-		if (i > 0) {
-			aElements[i].remove();
-			i --;
-		}
-	}
+    if (aElements.length > 1) {
+        aElements[aElements.length - 1].classList.add("has_notif");
+        aElements[aElements.length - 1].setAttribute("href", aElements[0].getAttribute("href"));
+        for (let i = aElements.length - 2; i >= 0; i --) {
+            aElements[i].remove();
+        }
+    }
 	
 	let spanImageElement = document.createElement("span");
 	spanImageElement.classList.add("shortcut_element_image");