1 files changed, 12 insertions, 4 deletions

diff --git a/uncompressed/instagram/webview.js b/uncompressed/instagram/webview.js
index aede5a5..f4f7f94 100644
--- a/uncompressed/instagram/webview.js
+++ b/uncompressed/instagram/webview.js
@@ -1,7 +1,15 @@
 const path = require("path");
 
-module.exports = Franz => {
-  // Apply fixes suggested in
-  // https://github.com/getferdi/recipes/issues/1
-  Franz.injectCSS(path.join(__dirname, "service.css"));
+module.exports = (Franz) => {
+  const getMessages = function getMessages() {
+    const element = document.querySelector('a[href^="/direct/inbox"]');
+
+    if (element) {
+      Franz.setBadge(parseInt(element.innerText, 10));
+    }
+  };
+
+  Franz.loop(getMessages);
+
+  Franz.injectCSS(path.join(__dirname, 'service.css'));
 };