1 files changed, 17 insertions, 0 deletions

diff --git a/uncompressed/twitter/webview.js b/uncompressed/twitter/webview.js
new file mode 100644
index 0000000..75655f1
--- /dev/null
+++ b/uncompressed/twitter/webview.js
@@ -0,0 +1,17 @@
+"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.
+    direct += document.querySelector('[data-testid=AppTabBar_Notifications_Link] div div div').innerHTML;
+    direct += document.querySelector('[data-testid=AppTabBar_DirectMessage_Link] div div div').innerHTML;
+
+    Franz.setBadge(direct);
+  };
+
+  Franz.loop(getMessages);
+
+};
\ No newline at end of file