1 files changed, 17 insertions, 11 deletions

diff --git a/recipes/hey/webview.js b/recipes/hey/webview.js
index 686e811..67e87d4 100644
--- a/recipes/hey/webview.js
+++ b/recipes/hey/webview.js
@@ -1,23 +1,29 @@
-module.exports = (Ferdi) => {
+module.exports = Ferdi => {
   const getMessages = () => {
-    if (document.location.href == "https://app.hey.com/") {
+    if (document.location.href == 'https://app.hey.com/') {
       let screener = 0;
       let unread = 0;
 
-      if (document.getElementsByClassName('btn--icon-screener').length > 0) {
-        let text = document.getElementsByClassName('btn--icon-screener')[0].innerText;
-
-        screener = Ferdi.safeParseInt(/[0-9]+/.exec(text)[0]);
+      if (document.querySelectorAll('.btn--icon-screener').length > 0) {
+        let text = document.querySelectorAll('.btn--icon-screener')[0]
+          .textContent;
+        if (text) {
+          const parsedText = Ferdi.safeParseInt(/\d+/.exec(text));
+          screener = parsedText[0];
+        }
       }
 
-      let postings = document.getElementsByClassName('posting');
+      let postings = document.querySelectorAll('.posting');
 
       if (postings.length > 0) {
-        Array.from(postings).forEach(p => {
-          if (p.nodeName == "ARTICLE" && p.getAttribute("data-seen") !== "true") {
+        for (const p of postings) {
+          if (
+            p.nodeName == 'ARTICLE' &&
+            p.getAttribute('data-seen') !== 'true'
+          ) {
             unread++;
           }
-        });
+        }
       }
 
       Ferdi.setBadge(unread, screener);
@@ -25,4 +31,4 @@ module.exports = (Ferdi) => {
   };
 
   Ferdi.loop(getMessages);
-}
+};