Merge pull request #247 from fjl5/master

Update recipe for element

3 files changed, 15 insertions, 12 deletions

diff --git a/uncompressed/element/README.md b/uncompressed/element/README.md
index affdf4b..221864f 100644
--- a/uncompressed/element/README.md
+++ b/uncompressed/element/README.md
@@ -1,2 +1,2 @@
-# Element for Franz
-This is a Franz recipe/plugin for Element
+# Element for Franz and Ferdi
+This is a Franz and Ferdi recipe/plugin for Element
diff --git a/uncompressed/element/package.json b/uncompressed/element/package.json
index 6774ce3..3b2f59f 100644
--- a/uncompressed/element/package.json
+++ b/uncompressed/element/package.json
@@ -1,10 +1,10 @@
 {
   "id": "element",
   "name": "Element",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "Decentralized, encrypted chat & collaboration powered by [matrix]",
   "main": "index.js",
-  "author": "Omove <61330514+omove@users.noreply.github.com>",
+  "author": "Sylvain Cecchetto <cecchetto.sylvain@me.com>, Omove <61330514+omove@users.noreply.github.com>, Achim Leitner <1033549+fjl5@users.noreply.github.com>",
   "license": "MIT",
   "config": {
     "serviceURL": "https://app.element.io",
diff --git a/uncompressed/element/webview.js b/uncompressed/element/webview.js
index 6201f27..aab6038 100644
--- a/uncompressed/element/webview.js
+++ b/uncompressed/element/webview.js
@@ -2,17 +2,20 @@
 
 module.exports = Franz => {
   function getMessages() {
-    const indirect_badge = document.getElementsByClassName('mx_RoomSubList_badge');
-    let indirect_count = 0;
-    if (indirect_badge.length !== 0) {
-      indirect_count = indirect_badge.length;
-    }
+    const badges = document.getElementsByClassName('mx_RoomSublist_badgeContainer');
 
-    const direct_badge = document.getElementsByClassName('mx_RoomSubList_badgeHighlight');
+    // Number of messages from People
     let direct_count = 0;
-    if (direct_badge.length !== 0) {
-      direct_count = direct_badge.length;
+    if (badges.length > 0 && badges[0].children.length > 0) {
+      direct_count = parseInt(badges[0].textContent)
+    }
+
+    // Number of messages from Rooms
+    let indirect_count = 0;
+    if (badges.length > 1 && badges[1].children.length > 0) {
+      indirect_count = parseInt(badges[1].textContent)
     }
+
     // set Franz badge
     Franz.setBadge(direct_count, indirect_count);
   }