1 files changed, 9 insertions, 6 deletions

diff --git a/recipes/stackoverflow-chat/webview.js b/recipes/stackoverflow-chat/webview.js
index 044c773..cf487b5 100644
--- a/recipes/stackoverflow-chat/webview.js
+++ b/recipes/stackoverflow-chat/webview.js
@@ -3,12 +3,15 @@ var _path = _interopRequireDefault(require('path'));
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
 module.exports = Ferdi => {
-  // TODO: If your stackoverflow-chat service has unread messages, uncomment these lines to implement the logic for updating the badges
-  // const getMessages = () => {
-  //   // TODO: Insert your notification-finding code here
-  //   Ferdi.setBadge(0, 0);
-  // };
-  // Ferdi.loop(getMessages);
+  const getMessages = () => {
+    const unreadSpan = document.querySelector('span.flag-count.message-count.unread-count');
+    let directCount = 0;
+    if (unreadSpan) {
+      directCount = Ferdi.safeParseInt(unreadSpan.innerText);
+    }
+    Ferdi.setBadge(directCount);
+  };
+  Ferdi.loop(getMessages);
 
   Ferdi.injectCSS(_path.default.join(__dirname, 'service.css'));
 };