1 files changed, 31 insertions, 7 deletions

diff --git a/recipes/outlook/webview.js b/recipes/outlook/webview.js
index 7e30fe4..6d4edf2 100644
--- a/recipes/outlook/webview.js
+++ b/recipes/outlook/webview.js
@@ -1,10 +1,34 @@
-"use strict";
+'use strict';
 
-module.exports = Franz => {
-  const getMessages = function getMessages() {
-    const unreadMail = parseInt(jQuery("i[data-icon-name='Inbox'] + span + span > span").first().text(), 10);
-    Franz.setBadge(unreadMail);
+module.exports = Ferdi => {
+  function getMessages() {
+    let unreadMail = 0;
+
+    if (location.pathname.match(/\/owa/)) {
+      // classic app
+      unreadMail = parseInt(
+        jQuery("span[title*='Inbox'] + div > span")
+          .first()
+          .text(),
+        10
+      );
+    } else {
+      // new app
+      const favorites = document.querySelector('div[title="Favorites"]');
+      if (!favorites) {
+        return;
+      }
+      const folders = Array.from(favorites.nextSibling.childNodes);
+
+      unreadMail = folders.reduce((count, child) => {
+        const unread = child.querySelector('.screenReaderOnly');
+        return unread && unread.textContent === 'unread'
+          ? count + parseInt(unread.previousSibling.textContent, 10)
+          : count;
+      }, 0);
+    }
+
+    Ferdi.setBadge(unreadMail);
   };
-  Franz.loop(getMessages);
+  Ferdi.loop(getMessages);
 };
-//# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbIm9mZmljZTM2NS1vd2Evd2Vidmlldy5qcyJdLCJuYW1lcyI6WyJtb2R1bGUiLCJleHBvcnRzIiwiRnJhbnoiLCJnZXRNZXNzYWdlcyIsInVucmVhZE1haWwiLCJwYXJzZUludCIsImpRdWVyeSIsImZpcnN0IiwidGV4dCIsInNldEJhZGdlIiwibG9vcCJdLCJtYXBwaW5ncyI6Ijs7QUFBQSxPQUFPLEFBQVAsVUFBa0IsQUFBRCxTQUFXLEFBQzFCO1FBQU0sY0FBYyxTQUFTLEFBQVQsY0FBdUIsQUFDekM7VUFBTSxhQUFhLFNBQVMsT0FBTyxBQUFQLEFBQTRDLEFBQTVDLEFBQW9ELEFBQXBELGtEQUFULEFBQXFFLEFBQXJFLEFBQW5CLGdCQUNBO1VBQU0sQUFBTixTQUFlLEFBQWYsQUFDRDtBQUhELEFBSUE7UUFBTSxBQUFOLEtBQVcsQUFBWCxBQUNEO0FBTkQiLCJmaWxlIjoib2ZmaWNlMzY1LW93YS93ZWJ2aWV3LmpzIiwic291cmNlc0NvbnRlbnQiOlsibW9kdWxlLmV4cG9ydHMgPSAoRnJhbnopID0+IHtcbiAgY29uc3QgZ2V0TWVzc2FnZXMgPSBmdW5jdGlvbiBnZXRNZXNzYWdlcygpIHtcbiAgICBjb25zdCB1bnJlYWRNYWlsID0gcGFyc2VJbnQoalF1ZXJ5KFwic3Bhblt0aXRsZSo9J0luYm94J10gKyBkaXYgPiBzcGFuXCIpLmZpcnN0KCkudGV4dCgpLCAxMCk7XG4gICAgRnJhbnouc2V0QmFkZ2UodW5yZWFkTWFpbCk7XG4gIH07XG4gIEZyYW56Lmxvb3AoZ2V0TWVzc2FnZXMpO1xufTtcbiJdfQ==
\ No newline at end of file