[vfk/v1.1] Added count to solutions pages and slightly reworked styling

1 files changed, 20 insertions, 8 deletions

diff --git a/vim-fmi-keystrokes.js b/vim-fmi-keystrokes.js
index c8fc6bd..87da7b9 100644
--- a/vim-fmi-keystrokes.js
+++ b/vim-fmi-keystrokes.js
@@ -2,7 +2,7 @@
 // @name        vim-fmi точки
 // @description Добавя брой натиснати клавиши във всяка страница на решения
 //
-// @version     1.0
+// @version     1.1
 // @license     MIT
 // @author      Syndamia
 //
@@ -16,10 +16,22 @@
 // @run-at document-body
 // ==/UserScript==
 
-document.getElementsByTagName('ul')[3].childNodes.forEach((x) => {
-	if (x.tagName != 'LI') return;
-	var countElem = document.createElement('span');
-    countElem.style.verticalAlign = 'bottom';
-	countElem.innerText = x.children[0].children.length;
-	x.insertBefore(countElem, x.firstChild);
-});
+if (window.location.pathname.endsWith('solutions')) {
+    document.getElementsByTagName('tbody')[0].childNodes.forEach((x) => {
+        if (x.tagName != "TR") return;
+	    var countElem = document.createElement('div');
+	    countElem.style.display = 'inline-block';
+        countElem.style.fontWeight = 'bold';
+        countElem.innerText = x.children[1].children[0].children.length;
+	    x.children[1].insertBefore(countElem, x.children[1].firstChild);
+    });
+}
+else {
+    document.getElementsByTagName('ul')[3].childNodes.forEach((x) => {
+        if (x.tagName != 'LI') return;
+        var countElem = document.createElement('span');
+        countElem.style.verticalAlign = 'middle';
+        countElem.innerText = x.children[0].children.length;
+        x.insertBefore(countElem, x.firstChild);
+    });
+}