1 files changed, 8 insertions, 6 deletions

diff --git a/recipes/snapchat/webview.js b/recipes/snapchat/webview.js
index 86bb422..78a2455 100644
--- a/recipes/snapchat/webview.js
+++ b/recipes/snapchat/webview.js
@@ -5,12 +5,14 @@ function _interopRequireDefault(obj) {
 }
 
 module.exports = Ferdium => {
-  // TODO: If your SNAME service has unread messages, uncomment these lines to implement the logic for updating the badges
-  // const getMessages = () => {
-  //   // TODO: Insert your notification-finding code here
-  //   Ferdium.setBadge(0, 0);
-  // };
-  // Ferdium.loop(getMessages);
+  const getMessages = () => {
+    const title = document.querySelector('title').textContent.match(/\d+/);
+    const count = title !== null ? title[0] : 0;
+
+    Ferdium.setBadge(count);
+  };
+
+  Ferdium.loop(getMessages);
 
   Ferdium.injectCSS(_path.default.join(__dirname, 'service.css'));
 };