1 files changed, 23 insertions, 0 deletions

diff --git a/recipes/twitter/webview.js b/recipes/twitter/webview.js
new file mode 100644
index 0000000..81f0ba5
--- /dev/null
+++ b/recipes/twitter/webview.js
@@ -0,0 +1,23 @@
+"use strict";
+
+module.exports = Franz => {
+  const getMessages = () => {
+    var direct = 0;
+
+    // "Notifications" and "Messages" - aria-label ending in
+    // "unread items". Sum the values for direct badge.
+    const notificationsElement = document.querySelector('[data-testid=AppTabBar_Notifications_Link] div div div');
+    if (notificationsElement) {
+      direct += notificationsElement.innerHTML;
+    }
+    const DMElement = document.querySelector('[data-testid=AppTabBar_DirectMessage_Link] div div div');
+    if (DMElement) {
+      direct += DMElement.innerHTML;
+    }
+    
+    Franz.setBadge(direct);
+  };
+
+  Franz.loop(getMessages);
+
+};
\ No newline at end of file