Add fixes for Nextcloud-related webapps (#66)

1 files changed, 10 insertions, 5 deletions

diff --git a/recipes/nextcloud-tasks/webview.js b/recipes/nextcloud-tasks/webview.js
index e9e6433..717fed7 100644
--- a/recipes/nextcloud-tasks/webview.js
+++ b/recipes/nextcloud-tasks/webview.js
@@ -1,15 +1,20 @@
 const _path = _interopRequireDefault(require('path'));
 
-function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+function _interopRequireDefault(obj) {
+  return obj && obj.__esModule ? obj : { default: obj };
+}
 
 module.exports = Ferdium => {
   const getMessages = () => {
-    const direct = document.querySelectorAll(
+    const directSelector = document.querySelectorAll(
       '.app-navigation-entry-utils-counter.highlighted',
-    ).length;
-    const indirect = document.querySelectorAll(
+    );
+    const direct = directSelector ? Ferdium.safeParseInt(directSelector.length) : 0;
+
+    const indirectSelector = document.querySelectorAll(
       '.app-navigation-entry-utils-counter:not(.highlighted)',
-    ).length;
+    );
+    const indirect = indirectSelector ? Ferdium.safeParseInt(indirectSelector.length) : 0;
 
     Ferdium.setBadge(direct, indirect);
   };